forked from openrs2/openrs2
parent
405c33c1a1
commit
a1f7cf5ce4
@ -0,0 +1,90 @@ |
|||||||
|
package dev.openrs2.bundler.transform; |
||||||
|
|
||||||
|
import dev.openrs2.asm.InsnMatcher; |
||||||
|
import dev.openrs2.asm.classpath.ClassPath; |
||||||
|
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.objectweb.asm.tree.analysis.AnalyzerException; |
||||||
|
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) throws AnalyzerException { |
||||||
|
hostChecks = 0; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
protected boolean transformCode(ClassNode clazz, MethodNode method) throws AnalyzerException { |
||||||
|
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) throws AnalyzerException { |
||||||
|
logger.info("Removed {} host checks", hostChecks); |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue