forked from openrs2/openrs2
parent
2ebcb3507b
commit
c2809a16f8
@ -0,0 +1,56 @@ |
||||
package dev.openrs2.deob.transform |
||||
|
||||
import com.github.michaelbull.logging.InlineLogger |
||||
import dev.openrs2.asm.MemberDesc |
||||
import dev.openrs2.asm.classpath.ClassPath |
||||
import dev.openrs2.asm.classpath.Library |
||||
import dev.openrs2.asm.transform.Transformer |
||||
import org.objectweb.asm.Opcodes |
||||
import org.objectweb.asm.Type |
||||
import org.objectweb.asm.tree.AnnotationNode |
||||
import org.objectweb.asm.tree.ClassNode |
||||
import org.objectweb.asm.tree.MethodNode |
||||
|
||||
class OverrideTransformer : Transformer() { |
||||
private var overrides = 0 |
||||
|
||||
override fun preTransform(classPath: ClassPath) { |
||||
overrides = 0 |
||||
} |
||||
|
||||
override fun transformCode( |
||||
classPath: ClassPath, |
||||
library: Library, |
||||
clazz: ClassNode, |
||||
method: MethodNode |
||||
): Boolean { |
||||
if (method.name == "<init>" || method.name == "<clinit>" || method.access and Opcodes.ACC_STATIC != 0) { |
||||
return false |
||||
} |
||||
|
||||
if (!classPath[clazz.name].isOverride(MemberDesc(method))) { |
||||
return false |
||||
} |
||||
|
||||
if (method.visibleAnnotations != null && method.visibleAnnotations.any { it.desc == OVERRIDE_DESC }) { |
||||
return false |
||||
} |
||||
|
||||
if (method.visibleAnnotations == null) { |
||||
method.visibleAnnotations = mutableListOf() |
||||
} |
||||
method.visibleAnnotations.add(AnnotationNode(OVERRIDE_DESC)) |
||||
overrides++ |
||||
|
||||
return false |
||||
} |
||||
|
||||
override fun postTransform(classPath: ClassPath) { |
||||
logger.info { "Added $overrides override annotations" } |
||||
} |
||||
|
||||
companion object { |
||||
val logger = InlineLogger() |
||||
val OVERRIDE_DESC: String = Type.getDescriptor(Override::class.java) |
||||
} |
||||
} |
Loading…
Reference in new issue