diff --git a/asm/src/main/java/dev/openrs2/asm/InsnNodeUtils.java b/asm/src/main/java/dev/openrs2/asm/InsnNodeUtils.java index 047726b2..5654f798 100644 --- a/asm/src/main/java/dev/openrs2/asm/InsnNodeUtils.java +++ b/asm/src/main/java/dev/openrs2/asm/InsnNodeUtils.java @@ -308,22 +308,28 @@ public final class InsnNodeUtils { throw new IllegalArgumentException(); } - public static boolean deleteSimpleExpression(InsnList list, AbstractInsnNode last) { + public static boolean replaceSimpleExpression(InsnList list, AbstractInsnNode last, AbstractInsnNode replacement) { var deadInsns = new ArrayList(); var height = 0; var insn = last; do { - deadInsns.add(insn); - var metadata = StackMetadata.get(insn); if (insn != last) { + deadInsns.add(insn); height -= metadata.getPushes(); } height += metadata.getPops(); if (height == 0) { deadInsns.forEach(list::remove); + + if (replacement != null) { + list.set(last, replacement); + } else { + list.remove(last); + } + return true; } @@ -333,6 +339,10 @@ public final class InsnNodeUtils { return false; } + public static boolean deleteSimpleExpression(InsnList list, AbstractInsnNode last) { + return replaceSimpleExpression(list, last, null); + } + private InsnNodeUtils() { /* empty */ } diff --git a/deob/src/main/java/dev/openrs2/deob/transform/DummyArgTransformer.java b/deob/src/main/java/dev/openrs2/deob/transform/DummyArgTransformer.java index 1ee55a71..a25422f2 100644 --- a/deob/src/main/java/dev/openrs2/deob/transform/DummyArgTransformer.java +++ b/deob/src/main/java/dev/openrs2/deob/transform/DummyArgTransformer.java @@ -11,6 +11,7 @@ import com.google.common.base.Preconditions; import com.google.common.collect.HashMultimap; import com.google.common.collect.ImmutableSet; import com.google.common.collect.Multimap; +import dev.openrs2.asm.InsnNodeUtils; import dev.openrs2.asm.MemberRef; import dev.openrs2.asm.classpath.ClassPath; import dev.openrs2.asm.transform.Transformer; @@ -20,7 +21,6 @@ import dev.openrs2.util.collect.DisjointSet; 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.JumpInsnNode; import org.objectweb.asm.tree.MethodInsnNode; import org.objectweb.asm.tree.MethodNode; @@ -187,11 +187,8 @@ public final class DummyArgTransformer extends Transformer { var changed = false; - var alwaysTakenUnaryBranches = new ArrayList(); - var neverTakenUnaryBranches = new ArrayList(); - - var alwaysTakenBinaryBranches = new ArrayList(); - var neverTakenBinaryBranches = new ArrayList(); + var alwaysTakenBranches = new ArrayList(); + var neverTakenBranches = new ArrayList(); for (var i = 0; i < frames.length; i++) { var frame = frames[i]; @@ -230,10 +227,10 @@ public final class DummyArgTransformer extends Transformer { var result = evaluateUnaryBranch(insn.getOpcode(), value.getIntValues()); switch (result) { case ALWAYS_TAKEN: - alwaysTakenUnaryBranches.add((JumpInsnNode) insn); + alwaysTakenBranches.add((JumpInsnNode) insn); break; case NEVER_TAKEN: - neverTakenUnaryBranches.add((JumpInsnNode) insn); + neverTakenBranches.add((JumpInsnNode) insn); break; } break; @@ -252,42 +249,28 @@ public final class DummyArgTransformer extends Transformer { result = evaluateBinaryBranch(insn.getOpcode(), value1.getIntValues(), value2.getIntValues()); switch (result) { case ALWAYS_TAKEN: - alwaysTakenBinaryBranches.add((JumpInsnNode) insn); + alwaysTakenBranches.add((JumpInsnNode) insn); break; case NEVER_TAKEN: - neverTakenBinaryBranches.add((JumpInsnNode) insn); + neverTakenBranches.add((JumpInsnNode) insn); break; } break; } } - for (var insn : alwaysTakenUnaryBranches) { - method.instructions.insertBefore(insn, new InsnNode(Opcodes.POP)); - method.instructions.set(insn, new JumpInsnNode(Opcodes.GOTO, insn.label)); - branchesSimplified++; - changed = true; - } - - for (var insn : neverTakenUnaryBranches) { - method.instructions.set(insn, new InsnNode(Opcodes.POP)); - branchesSimplified++; - changed = true; - } - - for (var insn : alwaysTakenBinaryBranches) { - method.instructions.insertBefore(insn, new InsnNode(Opcodes.POP)); - method.instructions.insertBefore(insn, new InsnNode(Opcodes.POP)); - method.instructions.set(insn, new JumpInsnNode(Opcodes.GOTO, insn.label)); - branchesSimplified++; - changed = true; + for (var insn : alwaysTakenBranches) { + if (InsnNodeUtils.replaceSimpleExpression(method.instructions, insn, new JumpInsnNode(Opcodes.GOTO, insn.label))) { + branchesSimplified++; + changed = true; + } } - for (var insn : neverTakenBinaryBranches) { - method.instructions.insertBefore(insn, new InsnNode(Opcodes.POP)); - method.instructions.set(insn, new InsnNode(Opcodes.POP)); - branchesSimplified++; - changed = true; + for (var insn : neverTakenBranches) { + if (InsnNodeUtils.deleteSimpleExpression(method.instructions, insn)) { + branchesSimplified++; + changed = true; + } } return changed;