|
|
|
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
|
|
|
|
import javax.inject.Singleton
|
|
|
|
|
|
|
|
@Singleton
|
|
|
|
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.invisibleAnnotations != null && method.invisibleAnnotations.any { it.desc == OVERRIDE_DESC }) {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
if (method.invisibleAnnotations == null) {
|
|
|
|
method.invisibleAnnotations = mutableListOf()
|
|
|
|
}
|
|
|
|
method.invisibleAnnotations.add(AnnotationNode(OVERRIDE_DESC))
|
|
|
|
overrides++
|
|
|
|
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
override fun postTransform(classPath: ClassPath) {
|
|
|
|
logger.info { "Added $overrides override annotations" }
|
|
|
|
}
|
|
|
|
|
|
|
|
private companion object {
|
|
|
|
private val logger = InlineLogger()
|
|
|
|
private val OVERRIDE_DESC: String = Type.getDescriptor(Override::class.java)
|
|
|
|
}
|
|
|
|
}
|