Convert BitShiftTransformer to Kotlin

pull/48/head
Graham 4 years ago
parent 368df81bc8
commit 77d3489b23
  1. 50
      deob/src/main/java/dev/openrs2/deob/transform/BitShiftTransformer.java
  2. 53
      deob/src/main/java/dev/openrs2/deob/transform/BitShiftTransformer.kt

@ -1,50 +0,0 @@
package dev.openrs2.deob.transform;
import com.google.common.collect.ImmutableSet;
import dev.openrs2.asm.InsnMatcher;
import dev.openrs2.asm.InsnNodeUtils;
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.tree.ClassNode;
import org.objectweb.asm.tree.MethodNode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public final class BitShiftTransformer extends Transformer {
private static final Logger logger = LoggerFactory.getLogger(BitShiftTransformer.class);
private static final InsnMatcher CONST_SHIFT_MATCHER = InsnMatcher.compile("(ICONST | BIPUSH | SIPUSH | LDC) (ISHL | ISHR | IUSHR | LSHL | LSHR | LUSHR)");
private static final ImmutableSet<Integer> LONG_SHIFTS = ImmutableSet.of(Opcodes.LSHL, Opcodes.LSHR, Opcodes.LUSHR);
private int simplified;
@Override
public void preTransform(ClassPath classPath) {
simplified = 0;
}
@Override
public boolean transformCode(ClassPath classPath, Library library, ClassNode clazz, MethodNode method) {
CONST_SHIFT_MATCHER.match(method).forEach(match -> {
var push = match.get(0);
var bits = InsnNodeUtils.getIntConstant(push);
var opcode = match.get(1).getOpcode();
var simplifiedBits = bits & (LONG_SHIFTS.contains(opcode) ? 63 : 31);
if (bits != simplifiedBits) {
method.instructions.set(push, InsnNodeUtils.createIntConstant(simplifiedBits));
simplified++;
}
});
return false;
}
@Override
public void postTransform(ClassPath classPath) {
logger.info("Simplified {} bit shifts", simplified);
}
}

@ -0,0 +1,53 @@
package dev.openrs2.deob.transform
import com.github.michaelbull.logging.InlineLogger
import dev.openrs2.asm.InsnMatcher
import dev.openrs2.asm.InsnNodeUtils
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.tree.ClassNode
import org.objectweb.asm.tree.MethodNode
class BitShiftTransformer : Transformer() {
private var simplified = 0
public override fun preTransform(classPath: ClassPath) {
simplified = 0
}
public override fun transformCode(
classPath: ClassPath,
library: Library,
clazz: ClassNode,
method: MethodNode
): Boolean {
CONST_SHIFT_MATCHER.match(method).forEach {
val push = it[0]
val bits = InsnNodeUtils.getIntConstant(push)
val opcode = it[1].opcode
val mask = if (LONG_SHIFTS.contains(opcode)) 63 else 31
val simplifiedBits = bits and mask
if (bits != simplifiedBits) {
method.instructions[push] = InsnNodeUtils.createIntConstant(simplifiedBits)
simplified++
}
}
return false
}
public override fun postTransform(classPath: ClassPath) {
logger.info { "Simplified $simplified bit shifts" }
}
companion object {
private val logger = InlineLogger()
private val CONST_SHIFT_MATCHER =
InsnMatcher.compile("(ICONST | BIPUSH | SIPUSH | LDC) (ISHL | ISHR | IUSHR | LSHL | LSHR | LUSHR)")
private val LONG_SHIFTS = setOf(Opcodes.LSHL, Opcodes.LSHR, Opcodes.LUSHR)
}
}
Loading…
Cancel
Save