TheArchive/Magisk
1
0
Fork 0
mirror of https://github.com/topjohnwu/Magisk.git synced 2024-12-04 15:35:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Only desugar ZipEntry's methods

Browse Source
This commit is contained in:
LoveSy 2024-12-02 23:16:00 +08:00 committed by John Wu
parent a34c04f999
commit 06616659b8
1 changed files with 39 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
buildSrc/src/main/java/DesugarClassVisitorFactory.kt
View File

@ -9,19 +9,21 @@ import org.objectweb.asm.Opcodes.ASM9
private const val DESUGAR_CLASS_NAME = "com.topjohnwu.magisk.core.utils.Desugar" private const val DESUGAR_CLASS_NAME = "com.topjohnwu.magisk.core.utils.Desugar"
private const val ZIP_ENTRY_GET_TIME_DESC = "()Ljava/nio/file/attribute/FileTime;" private const val ZIP_ENTRY_GET_TIME_DESC = "()Ljava/nio/file/attribute/FileTime;"
private const val DESUGAR_GET_TIME_DESC = "(Ljava/util/zip/ZipEntry;)Ljava/nio/file/attribute/FileTime;" private const val DESUGAR_GET_TIME_DESC =
"(Ljava/util/zip/ZipEntry;)Ljava/nio/file/attribute/FileTime;"
abstract class DesugarClassVisitorFactory : AsmClassVisitorFactory<InstrumentationParameters.None> { abstract class DesugarClassVisitorFactory : AsmClassVisitorFactory<InstrumentationParameters.None> {
override fun createClassVisitor( override fun createClassVisitor(
classContext: ClassContext, classContext: ClassContext,
nextClassVisitor: ClassVisitor nextClassVisitor: ClassVisitor
): ClassVisitor { ): ClassVisitor {
return DesugarClassVisitor(nextClassVisitor) return DesugarClassVisitor(classContext, nextClassVisitor)
} }
override fun isInstrumentable(classData: ClassData) = classData.className != DESUGAR_CLASS_NAME override fun isInstrumentable(classData: ClassData) = classData.className != DESUGAR_CLASS_NAME
class DesugarClassVisitor(cv: ClassVisitor) : ClassVisitor(ASM9, cv) { class DesugarClassVisitor(private val classContext: ClassContext, cv: ClassVisitor) :
ClassVisitor(ASM9, cv) {
override fun visitMethod( override fun visitMethod(
access: Int, access: Int,
name: String?, name: String?,
@ -29,39 +31,44 @@ abstract class DesugarClassVisitorFactory : AsmClassVisitorFactory<Instrumentati
signature: String?, signature: String?,
exceptions: Array<out String>? exceptions: Array<out String>?
): MethodVisitor { ): MethodVisitor {
return DesugarMethodVisitor(super.visitMethod(access, name, descriptor, signature, exceptions)) return DesugarMethodVisitor(
} super.visitMethod(access, name, descriptor, signature, exceptions)
} )
class DesugarMethodVisitor(mv: MethodVisitor?) : MethodVisitor(ASM9, mv) {
override fun visitMethodInsn(
opcode: Int,
owner: String,
name: String,
descriptor: String,
isInterface: Boolean
) {
if (!process(name, descriptor)) {
super.visitMethodInsn(opcode, owner, name, descriptor, isInterface)
}
} }
private fun process(name: String, descriptor: String): Boolean { inner class DesugarMethodVisitor(mv: MethodVisitor?) :
if (descriptor != ZIP_ENTRY_GET_TIME_DESC) MethodVisitor(ASM9, mv) {
return false override fun visitMethodInsn(
when (name) { opcode: Int,
"getLastModifiedTime", "getLastAccessTime", "getCreationTime" -> { owner: String,
mv.visitMethodInsn( name: String,
Opcodes.INVOKESTATIC, descriptor: String,
DESUGAR_CLASS_NAME.replace('.', '/'), isInterface: Boolean
name, ) {
DESUGAR_GET_TIME_DESC, if (!classContext.loadClassData(owner.replace("/", "."))?.superClasses.orEmpty()
false .contains("java.util.zip.ZipEntry") || !process(name, descriptor)
) ) {
super.visitMethodInsn(opcode, owner, name, descriptor, isInterface)
}
}
private fun process(name: String, descriptor: String): Boolean {
if (descriptor != ZIP_ENTRY_GET_TIME_DESC)
return false
return when (name) {
"getLastModifiedTime", "getLastAccessTime", "getCreationTime" -> {
mv.visitMethodInsn(
Opcodes.INVOKESTATIC,
DESUGAR_CLASS_NAME.replace('.', '/'),
name,
DESUGAR_GET_TIME_DESC,
false
)
true
}
else -> false
} }
else -> return false
} }
return true
} }
} }
} }