mirror of
https://github.com/revanced/revanced-patches
synced 2024-06-02 11:26:20 +02:00
73 lines
3.4 KiB
Kotlin
73 lines
3.4 KiB
Kotlin
package app.revanced.patches.reddit.customclients.syncforreddit.api.patch
|
|
|
|
import app.revanced.extensions.toErrorResult
|
|
import app.revanced.patcher.annotation.Compatibility
|
|
import app.revanced.patcher.annotation.Package
|
|
import app.revanced.patcher.data.BytecodeContext
|
|
import app.revanced.patcher.extensions.InstructionExtensions.addInstructions
|
|
import app.revanced.patcher.extensions.InstructionExtensions.getInstruction
|
|
import app.revanced.patcher.extensions.InstructionExtensions.replaceInstruction
|
|
import app.revanced.patcher.fingerprint.method.impl.MethodFingerprint.Companion.resolve
|
|
import app.revanced.patcher.fingerprint.method.impl.MethodFingerprintResult
|
|
import app.revanced.patcher.patch.PatchResult
|
|
import app.revanced.patcher.patch.PatchResultSuccess
|
|
import app.revanced.patches.reddit.customclients.AbstractChangeOAuthClientIdPatch
|
|
import app.revanced.patches.reddit.customclients.ChangeOAuthClientIdPatchAnnotation
|
|
import app.revanced.patches.reddit.customclients.syncforreddit.api.fingerprints.GetAuthorizationStringFingerprint
|
|
import app.revanced.patches.reddit.customclients.syncforreddit.api.fingerprints.GetBearerTokenFingerprint
|
|
import org.jf.dexlib2.iface.instruction.OneRegisterInstruction
|
|
import org.jf.dexlib2.iface.instruction.ReferenceInstruction
|
|
import org.jf.dexlib2.iface.reference.StringReference
|
|
import java.util.*
|
|
|
|
@ChangeOAuthClientIdPatchAnnotation
|
|
@Compatibility(
|
|
[
|
|
Package("com.laurencedawson.reddit_sync"),
|
|
Package("com.laurencedawson.reddit_sync.pro"),
|
|
Package("com.laurencedawson.reddit_sync.dev")
|
|
]
|
|
)
|
|
class ChangeOAuthClientIdPatch : AbstractChangeOAuthClientIdPatch(
|
|
"http://redditsync/auth", Options, listOf(GetAuthorizationStringFingerprint)
|
|
) {
|
|
override fun List<MethodFingerprintResult>.patch(context: BytecodeContext): PatchResult {
|
|
forEach { fingerprintResult ->
|
|
fingerprintResult.also { result ->
|
|
GetBearerTokenFingerprint.also { it.resolve(context, result.classDef) }.result?.mutableMethod?.apply {
|
|
val auth = Base64.getEncoder().encodeToString("$clientId:".toByteArray(Charsets.UTF_8))
|
|
addInstructions(
|
|
0,
|
|
"""
|
|
const-string v0, "Basic $auth"
|
|
return-object v0
|
|
"""
|
|
)
|
|
} ?: return GetBearerTokenFingerprint.toErrorResult()
|
|
}.let {
|
|
val occurrenceIndex = it.scanResult.stringsScanResult!!.matches.first().index
|
|
|
|
it.mutableMethod.apply {
|
|
val authorizationStringInstruction = getInstruction<ReferenceInstruction>(occurrenceIndex)
|
|
val targetRegister = (authorizationStringInstruction as OneRegisterInstruction).registerA
|
|
val reference = authorizationStringInstruction.reference as StringReference
|
|
|
|
val newAuthorizationUrl = reference.string.replace(
|
|
"client_id=.*?&".toRegex(),
|
|
"client_id=$clientId&"
|
|
)
|
|
|
|
replaceInstruction(
|
|
occurrenceIndex,
|
|
"const-string v$targetRegister, \"$newAuthorizationUrl\""
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
return PatchResultSuccess()
|
|
}
|
|
|
|
companion object Options : AbstractChangeOAuthClientIdPatch.Options.ChangeOAuthClientIdOptionsContainer()
|
|
}
|