added simple renamer for local variable table

1 files changed, 32 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/bytecode/LocalVariableRenamer.java b/src/cuchaz/enigma/bytecode/LocalVariableRenamer.java
new file mode 100644
index 0000000..53f207c
--- /dev/null
+++ b/src/cuchaz/enigma/bytecode/LocalVariableRenamer.java
@@ -0,0 +1,32 @@
+package cuchaz.enigma.bytecode;
+
+import javassist.CtBehavior;
+import javassist.CtClass;
+import javassist.bytecode.ByteArray;
+import javassist.bytecode.ConstPool;
+import javassist.bytecode.LocalVariableAttribute;
+
+
+public class LocalVariableRenamer {
+
+        public void rename(CtClass c) {
+                for (CtBehavior behavior : c.getDeclaredBehaviors()) {
+                        
+                        // if there's a local variable table, just rename everything to v1, v2, v3, ... for now
+                        LocalVariableAttribute table = (LocalVariableAttribute)behavior.getMethodInfo().getCodeAttribute().getAttribute(LocalVariableAttribute.tag);
+                        if (table == null) {
+                                continue;
+                        }
+                        
+                        ConstPool constants = c.getClassFile().getConstPool();
+                        for (int i=0; i<table.tableLength(); i++) {
+                                renameVariable(table, i, constants.addUtf8Info("v" + i));
+                        }
+                }
+        }
+
+        private void renameVariable(LocalVariableAttribute table, int i, int stringId) {
+                // based off of LocalVariableAttribute.nameIndex()
+                ByteArray.write16bit(stringId, table.get(), i*10 + 6);
+        }
+}