Skip to content

Commit

Permalink
feat(TU Dortmund): Add Show on lockscreen patch (ReVanced#2947)
Browse files Browse the repository at this point in the history
Co-authored-by: oSumAtrIX <[email protected]>
  • Loading branch information
Traktores and oSumAtrIX authored Sep 14, 2023
1 parent a4d6b4e commit 9a18326
Show file tree
Hide file tree
Showing 2 changed files with 108 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
package app.revanced.patches.tudortmund.lockscreen.fingerprints

import app.revanced.patcher.fingerprint.method.impl.MethodFingerprint
import com.android.tools.smali.dexlib2.AccessFlags

object BrightnessFingerprint : MethodFingerprint(
returnType = "V",
accessFlags = AccessFlags.PUBLIC.value,
parameters = listOf(),
customFingerprint = { method, classDef ->
method.name == "run"
&& method.definingClass.contains("/ScreenPlugin\$")
&& classDef.fields.any { it.type == "Ljava/lang/Float;" }
}
)
Original file line number Diff line number Diff line change
@@ -0,0 +1,93 @@
package app.revanced.patches.tudortmund.lockscreen.patch

import app.revanced.extensions.exception
import app.revanced.patcher.annotation.Compatibility
import app.revanced.patcher.annotation.Description
import app.revanced.patcher.annotation.Name
import app.revanced.patcher.annotation.Package
import app.revanced.patcher.data.BytecodeContext
import app.revanced.patcher.extensions.InstructionExtensions.addInstruction
import app.revanced.patcher.extensions.InstructionExtensions.addInstructions
import app.revanced.patcher.extensions.InstructionExtensions.replaceInstruction
import app.revanced.patcher.patch.BytecodePatch
import app.revanced.patcher.patch.annotations.Patch
import app.revanced.patcher.patch.annotations.RequiresIntegrations
import app.revanced.patches.tudortmund.lockscreen.fingerprints.BrightnessFingerprint
import com.android.tools.smali.dexlib2.Opcode
import com.android.tools.smali.dexlib2.iface.instruction.formats.Instruction22c
import com.android.tools.smali.dexlib2.iface.instruction.formats.Instruction35c
import com.android.tools.smali.dexlib2.iface.reference.FieldReference
import com.android.tools.smali.dexlib2.iface.reference.MethodReference

@Patch
@Name("Show on lockscreen")
@Description("Shows student id and student ticket on lockscreen.")
@Compatibility([Package("de.tudortmund.app")])
@RequiresIntegrations
class ShowOnLockscreenPatch : BytecodePatch(
listOf(BrightnessFingerprint)
) {
override fun execute(context: BytecodeContext) {
BrightnessFingerprint.result?.mutableMethod?.apply {
// Find the instruction where the brightness value is loaded into a register
val brightnessInstruction = implementation!!.instructions.firstNotNullOf { instruction ->
if (instruction.opcode != Opcode.IGET_OBJECT) return@firstNotNullOf null

val getInstruction = instruction as Instruction22c
val fieldReference = getInstruction.reference as FieldReference

if (fieldReference.type != "Ljava/lang/Float;") return@firstNotNullOf null

instruction
}

// Search for the instruction where we get the android.view.Window via the Activity.
// Gets the index of that instruction and the register of the Activity.
val (windowIndex, activityRegister) = implementation!!.instructions.withIndex()
.firstNotNullOf { (index, instruction) ->
if (instruction.opcode != Opcode.INVOKE_VIRTUAL)
return@firstNotNullOf null

val invokeInstruction = instruction as Instruction35c
val methodRef = invokeInstruction.reference as MethodReference

if (methodRef.name != "getWindow" || methodRef.returnType != "Landroid/view/Window;")
return@firstNotNullOf null

Pair(index, invokeInstruction.registerC)
}

// The register in which the brightness value is loaded
val brightnessRegister = brightnessInstruction.registerA

// Replaces the getWindow call with our custom one to run the lockscreen code
replaceInstruction(
windowIndex,
"invoke-static { v$activityRegister, v$brightnessRegister }, " +
"$INTEGRATIONS_CLASS_DESCRIPTOR->" +
"getWindow" +
"(Landroidx/appcompat/app/AppCompatActivity;F)" +
"Landroid/view/Window;"
)

// Normally, the brightness is loaded into a register after the getWindow call.
// In order to pass the brightness value to our custom getWindow implementation,
// we need to add the same instructions before the getWindow call.
// The Float object is loaded into the brightness register and gets converted to a float.
addInstructions(
windowIndex,
"""
invoke-virtual { v$brightnessRegister }, Ljava/lang/Float;->floatValue()F
move-result v$brightnessRegister
"""
)

addInstruction(windowIndex, brightnessInstruction)

} ?: throw BrightnessFingerprint.exception
}

private companion object {
const val INTEGRATIONS_CLASS_DESCRIPTOR = "Lapp/revanced/tudortmund/lockscreen/ShowOnLockscreenPatch;"
}
}

0 comments on commit 9a18326

Please sign in to comment.