forked from openrs2/openrs2
parent
1d889357a9
commit
4a8704f566
@ -1,90 +0,0 @@ |
|||||||
package dev.openrs2.bundler.transform; |
|
||||||
|
|
||||||
import dev.openrs2.asm.InsnMatcher; |
|
||||||
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.ClassNode; |
|
||||||
import org.objectweb.asm.tree.InsnNode; |
|
||||||
import org.objectweb.asm.tree.MethodInsnNode; |
|
||||||
import org.objectweb.asm.tree.MethodNode; |
|
||||||
import org.slf4j.Logger; |
|
||||||
import org.slf4j.LoggerFactory; |
|
||||||
|
|
||||||
public final class HostCheckTransformer extends Transformer { |
|
||||||
private static final Logger logger = LoggerFactory.getLogger(HostCheckTransformer.class); |
|
||||||
|
|
||||||
private static final InsnMatcher GET_HOST_MATCHER = InsnMatcher.compile("INVOKEVIRTUAL INVOKEVIRTUAL INVOKEVIRTUAL"); |
|
||||||
|
|
||||||
private int hostChecks; |
|
||||||
|
|
||||||
@Override |
|
||||||
protected void preTransform(ClassPath classPath) { |
|
||||||
hostChecks = 0; |
|
||||||
} |
|
||||||
|
|
||||||
@Override |
|
||||||
protected boolean transformCode(ClassPath classPath, Library library, ClassNode clazz, MethodNode method) { |
|
||||||
if (Type.getReturnType(method.desc).getSort() != Type.BOOLEAN) { |
|
||||||
return false; |
|
||||||
} |
|
||||||
|
|
||||||
GET_HOST_MATCHER.match(method).forEach(match -> { |
|
||||||
var insn1 = (MethodInsnNode) match.get(0); |
|
||||||
if (!insn1.owner.equals(clazz.name)) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
if (!insn1.name.equals("getDocumentBase")) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
if (!insn1.desc.equals("()Ljava/net/URL;")) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
var insn2 = (MethodInsnNode) match.get(1); |
|
||||||
if (!insn2.owner.equals("java/net/URL")) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
if (!insn2.name.equals("getHost")) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
if (!insn2.desc.equals("()Ljava/lang/String;")) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
var insn3 = (MethodInsnNode) match.get(2); |
|
||||||
if (!insn3.owner.equals("java/lang/String")) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
if (!insn3.name.equals("toLowerCase")) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
if (!insn3.desc.equals("()Ljava/lang/String;")) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
method.instructions.clear(); |
|
||||||
method.tryCatchBlocks.clear(); |
|
||||||
|
|
||||||
method.instructions.add(new InsnNode(Opcodes.ICONST_1)); |
|
||||||
method.instructions.add(new InsnNode(Opcodes.IRETURN)); |
|
||||||
|
|
||||||
hostChecks++; |
|
||||||
}); |
|
||||||
|
|
||||||
return false; |
|
||||||
} |
|
||||||
|
|
||||||
@Override |
|
||||||
protected void postTransform(ClassPath classPath) { |
|
||||||
logger.info("Removed {} host checks", hostChecks); |
|
||||||
} |
|
||||||
} |
|
||||||
@ -0,0 +1,63 @@ |
|||||||
|
package dev.openrs2.bundler.transform |
||||||
|
|
||||||
|
import com.github.michaelbull.logging.InlineLogger |
||||||
|
import dev.openrs2.asm.InsnMatcher |
||||||
|
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.ClassNode |
||||||
|
import org.objectweb.asm.tree.InsnNode |
||||||
|
import org.objectweb.asm.tree.MethodInsnNode |
||||||
|
import org.objectweb.asm.tree.MethodNode |
||||||
|
|
||||||
|
class HostCheckTransformer : Transformer() { |
||||||
|
private var hostChecks = 0 |
||||||
|
|
||||||
|
override fun preTransform(classPath: ClassPath) { |
||||||
|
hostChecks = 0 |
||||||
|
} |
||||||
|
|
||||||
|
override fun transformCode(classPath: ClassPath, library: Library, clazz: ClassNode, method: MethodNode): Boolean { |
||||||
|
if (Type.getReturnType(method.desc).sort != Type.BOOLEAN) { |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
GET_HOST_MATCHER.match(method).forEach { |
||||||
|
val insn1 = it[0] as MethodInsnNode |
||||||
|
if (insn1.owner != clazz.name || insn1.name != "getDocumentBase" || insn1.desc != "()Ljava/net/URL;") { |
||||||
|
return@forEach |
||||||
|
} |
||||||
|
|
||||||
|
val insn2 = it[1] as MethodInsnNode |
||||||
|
if (insn2.owner != "java/net/URL" || insn2.name != "getHost" || insn2.desc != "()Ljava/lang/String;") { |
||||||
|
return@forEach |
||||||
|
} |
||||||
|
|
||||||
|
val insn3 = it[2] as MethodInsnNode |
||||||
|
if (insn3.owner != "java/lang/String" || insn3.name != "toLowerCase" || insn3.desc != "()Ljava/lang/String;") { |
||||||
|
return@forEach |
||||||
|
} |
||||||
|
|
||||||
|
method.instructions.clear() |
||||||
|
method.tryCatchBlocks.clear() |
||||||
|
|
||||||
|
method.instructions.add(InsnNode(Opcodes.ICONST_1)) |
||||||
|
method.instructions.add(InsnNode(Opcodes.IRETURN)) |
||||||
|
|
||||||
|
hostChecks++ |
||||||
|
} |
||||||
|
|
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
override fun postTransform(classPath: ClassPath) { |
||||||
|
logger.info { "Removed $hostChecks host checks" } |
||||||
|
} |
||||||
|
|
||||||
|
companion object { |
||||||
|
private val logger = InlineLogger() |
||||||
|
private val GET_HOST_MATCHER = InsnMatcher.compile("INVOKEVIRTUAL INVOKEVIRTUAL INVOKEVIRTUAL") |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue