diff options
| author | 2015-02-08 21:29:25 -0500 | |
|---|---|---|
| committer | 2015-02-08 21:29:25 -0500 | |
| commit | ed9b5cdfc648e86fd463bfa8d86b94c41671e14c (patch) | |
| tree | 2619bbc7e04dfa3b82f8dfd3b1d31f529766cd4b /src/cuchaz/enigma/bytecode | |
| download | enigma-fork-ed9b5cdfc648e86fd463bfa8d86b94c41671e14c.tar.gz enigma-fork-ed9b5cdfc648e86fd463bfa8d86b94c41671e14c.tar.xz enigma-fork-ed9b5cdfc648e86fd463bfa8d86b94c41671e14c.zip | |
switch all classes to new signature/type system
Diffstat (limited to 'src/cuchaz/enigma/bytecode')
17 files changed, 1846 insertions, 0 deletions
diff --git a/src/cuchaz/enigma/bytecode/CheckCastIterator.java b/src/cuchaz/enigma/bytecode/CheckCastIterator.java new file mode 100644 index 0000000..5284557 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/CheckCastIterator.java | |||
| @@ -0,0 +1,127 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode; | ||
| 12 | |||
| 13 | import java.util.Iterator; | ||
| 14 | |||
| 15 | import javassist.bytecode.BadBytecode; | ||
| 16 | import javassist.bytecode.CodeAttribute; | ||
| 17 | import javassist.bytecode.CodeIterator; | ||
| 18 | import javassist.bytecode.ConstPool; | ||
| 19 | import javassist.bytecode.Descriptor; | ||
| 20 | import javassist.bytecode.Opcode; | ||
| 21 | import cuchaz.enigma.bytecode.CheckCastIterator.CheckCast; | ||
| 22 | import cuchaz.enigma.mapping.ClassEntry; | ||
| 23 | import cuchaz.enigma.mapping.MethodEntry; | ||
| 24 | import cuchaz.enigma.mapping.Signature; | ||
| 25 | |||
| 26 | public class CheckCastIterator implements Iterator<CheckCast> { | ||
| 27 | |||
| 28 | public static class CheckCast { | ||
| 29 | |||
| 30 | public String className; | ||
| 31 | public MethodEntry prevMethodEntry; | ||
| 32 | |||
| 33 | public CheckCast(String className, MethodEntry prevMethodEntry) { | ||
| 34 | this.className = className; | ||
| 35 | this.prevMethodEntry = prevMethodEntry; | ||
| 36 | } | ||
| 37 | } | ||
| 38 | |||
| 39 | private ConstPool m_constants; | ||
| 40 | private CodeAttribute m_attribute; | ||
| 41 | private CodeIterator m_iter; | ||
| 42 | private CheckCast m_next; | ||
| 43 | |||
| 44 | public CheckCastIterator(CodeAttribute codeAttribute) throws BadBytecode { | ||
| 45 | m_constants = codeAttribute.getConstPool(); | ||
| 46 | m_attribute = codeAttribute; | ||
| 47 | m_iter = m_attribute.iterator(); | ||
| 48 | |||
| 49 | m_next = getNext(); | ||
| 50 | } | ||
| 51 | |||
| 52 | @Override | ||
| 53 | public boolean hasNext() { | ||
| 54 | return m_next != null; | ||
| 55 | } | ||
| 56 | |||
| 57 | @Override | ||
| 58 | public CheckCast next() { | ||
| 59 | CheckCast out = m_next; | ||
| 60 | try { | ||
| 61 | m_next = getNext(); | ||
| 62 | } catch (BadBytecode ex) { | ||
| 63 | throw new Error(ex); | ||
| 64 | } | ||
| 65 | return out; | ||
| 66 | } | ||
| 67 | |||
| 68 | @Override | ||
| 69 | public void remove() { | ||
| 70 | throw new UnsupportedOperationException(); | ||
| 71 | } | ||
| 72 | |||
| 73 | private CheckCast getNext() throws BadBytecode { | ||
| 74 | int prevPos = 0; | ||
| 75 | while (m_iter.hasNext()) { | ||
| 76 | int pos = m_iter.next(); | ||
| 77 | int opcode = m_iter.byteAt(pos); | ||
| 78 | switch (opcode) { | ||
| 79 | case Opcode.CHECKCAST: | ||
| 80 | |||
| 81 | // get the type of this op code (next two bytes are a classinfo index) | ||
| 82 | MethodEntry prevMethodEntry = getMethodEntry(prevPos); | ||
| 83 | if (prevMethodEntry != null) { | ||
| 84 | return new CheckCast(m_constants.getClassInfo(m_iter.s16bitAt(pos + 1)), prevMethodEntry); | ||
| 85 | } | ||
| 86 | break; | ||
| 87 | } | ||
| 88 | prevPos = pos; | ||
| 89 | } | ||
| 90 | return null; | ||
| 91 | } | ||
| 92 | |||
| 93 | private MethodEntry getMethodEntry(int pos) { | ||
| 94 | switch (m_iter.byteAt(pos)) { | ||
| 95 | case Opcode.INVOKEVIRTUAL: | ||
| 96 | case Opcode.INVOKESTATIC: | ||
| 97 | case Opcode.INVOKEDYNAMIC: | ||
| 98 | case Opcode.INVOKESPECIAL: { | ||
| 99 | int index = m_iter.s16bitAt(pos + 1); | ||
| 100 | return new MethodEntry( | ||
| 101 | new ClassEntry(Descriptor.toJvmName(m_constants.getMethodrefClassName(index))), | ||
| 102 | m_constants.getMethodrefName(index), | ||
| 103 | new Signature(m_constants.getMethodrefType(index)) | ||
| 104 | ); | ||
| 105 | } | ||
| 106 | |||
| 107 | case Opcode.INVOKEINTERFACE: { | ||
| 108 | int index = m_iter.s16bitAt(pos + 1); | ||
| 109 | return new MethodEntry( | ||
| 110 | new ClassEntry(Descriptor.toJvmName(m_constants.getInterfaceMethodrefClassName(index))), | ||
| 111 | m_constants.getInterfaceMethodrefName(index), | ||
| 112 | new Signature(m_constants.getInterfaceMethodrefType(index)) | ||
| 113 | ); | ||
| 114 | } | ||
| 115 | } | ||
| 116 | return null; | ||
| 117 | } | ||
| 118 | |||
| 119 | public Iterable<CheckCast> casts() { | ||
| 120 | return new Iterable<CheckCast>() { | ||
| 121 | @Override | ||
| 122 | public Iterator<CheckCast> iterator() { | ||
| 123 | return CheckCastIterator.this; | ||
| 124 | } | ||
| 125 | }; | ||
| 126 | } | ||
| 127 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/ClassRenamer.java b/src/cuchaz/enigma/bytecode/ClassRenamer.java new file mode 100644 index 0000000..a5fea92 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/ClassRenamer.java | |||
| @@ -0,0 +1,110 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode; | ||
| 12 | |||
| 13 | import java.util.Map; | ||
| 14 | import java.util.Set; | ||
| 15 | |||
| 16 | import javassist.ClassMap; | ||
| 17 | import javassist.CtClass; | ||
| 18 | import javassist.bytecode.ConstPool; | ||
| 19 | import javassist.bytecode.Descriptor; | ||
| 20 | import javassist.bytecode.InnerClassesAttribute; | ||
| 21 | |||
| 22 | import com.google.common.collect.Maps; | ||
| 23 | import com.google.common.collect.Sets; | ||
| 24 | |||
| 25 | import cuchaz.enigma.mapping.ClassEntry; | ||
| 26 | |||
| 27 | public class ClassRenamer { | ||
| 28 | |||
| 29 | public static void renameClasses(CtClass c, Map<ClassEntry,ClassEntry> map) { | ||
| 30 | |||
| 31 | // build the map used by javassist | ||
| 32 | ClassMap nameMap = new ClassMap(); | ||
| 33 | for (Map.Entry<ClassEntry,ClassEntry> entry : map.entrySet()) { | ||
| 34 | nameMap.put(entry.getKey().getName(), entry.getValue().getName()); | ||
| 35 | } | ||
| 36 | |||
| 37 | c.replaceClassName(nameMap); | ||
| 38 | |||
| 39 | // replace simple names in the InnerClasses attribute too | ||
| 40 | ConstPool constants = c.getClassFile().getConstPool(); | ||
| 41 | InnerClassesAttribute attr = (InnerClassesAttribute)c.getClassFile().getAttribute(InnerClassesAttribute.tag); | ||
| 42 | if (attr != null) { | ||
| 43 | for (int i = 0; i < attr.tableLength(); i++) { | ||
| 44 | ClassEntry classEntry = new ClassEntry(Descriptor.toJvmName(attr.innerClass(i))); | ||
| 45 | if (attr.innerNameIndex(i) != 0) { | ||
| 46 | attr.setInnerNameIndex(i, constants.addUtf8Info(classEntry.getInnerClassName())); | ||
| 47 | } | ||
| 48 | |||
| 49 | /* DEBUG | ||
| 50 | System.out.println(String.format("\tDEOBF: %s-> ATTR: %s,%s,%s", classEntry, attr.outerClass(i), attr.innerClass(i), attr.innerName(i))); | ||
| 51 | */ | ||
| 52 | } | ||
| 53 | } | ||
| 54 | } | ||
| 55 | |||
| 56 | public static Set<ClassEntry> getAllClassEntries(final CtClass c) { | ||
| 57 | |||
| 58 | // get the classes that javassist knows about | ||
| 59 | final Set<ClassEntry> entries = Sets.newHashSet(); | ||
| 60 | ClassMap map = new ClassMap() { | ||
| 61 | @Override | ||
| 62 | public Object get(Object obj) { | ||
| 63 | if (obj instanceof String) { | ||
| 64 | String str = (String)obj; | ||
| 65 | |||
| 66 | // javassist throws a lot of weird things at this map | ||
| 67 | // I either have to implement my on class scanner, or just try to filter out the weirdness | ||
| 68 | // I'm opting to filter out the weirdness for now | ||
| 69 | |||
| 70 | // skip anything with generic arguments | ||
| 71 | if (str.indexOf('<') >= 0 || str.indexOf('>') >= 0 || str.indexOf(';') >= 0) { | ||
| 72 | return null; | ||
| 73 | } | ||
| 74 | |||
| 75 | // convert path/to/class.inner to path/to/class$inner | ||
| 76 | str = str.replace('.', '$'); | ||
| 77 | |||
| 78 | // remember everything else | ||
| 79 | entries.add(new ClassEntry(str)); | ||
| 80 | } | ||
| 81 | return null; | ||
| 82 | } | ||
| 83 | |||
| 84 | private static final long serialVersionUID = -202160293602070641L; | ||
| 85 | }; | ||
| 86 | c.replaceClassName(map); | ||
| 87 | |||
| 88 | return entries; | ||
| 89 | } | ||
| 90 | |||
| 91 | public static void moveAllClassesOutOfDefaultPackage(CtClass c, String newPackageName) { | ||
| 92 | Map<ClassEntry,ClassEntry> map = Maps.newHashMap(); | ||
| 93 | for (ClassEntry classEntry : ClassRenamer.getAllClassEntries(c)) { | ||
| 94 | if (classEntry.isInDefaultPackage()) { | ||
| 95 | map.put(classEntry, new ClassEntry(newPackageName + "/" + classEntry.getName())); | ||
| 96 | } | ||
| 97 | } | ||
| 98 | ClassRenamer.renameClasses(c, map); | ||
| 99 | } | ||
| 100 | |||
| 101 | public static void moveAllClassesIntoDefaultPackage(CtClass c, String oldPackageName) { | ||
| 102 | Map<ClassEntry,ClassEntry> map = Maps.newHashMap(); | ||
| 103 | for (ClassEntry classEntry : ClassRenamer.getAllClassEntries(c)) { | ||
| 104 | if (classEntry.getPackageName().equals(oldPackageName)) { | ||
| 105 | map.put(classEntry, new ClassEntry(classEntry.getSimpleName())); | ||
| 106 | } | ||
| 107 | } | ||
| 108 | ClassRenamer.renameClasses(c, map); | ||
| 109 | } | ||
| 110 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/ClassTranslator.java b/src/cuchaz/enigma/bytecode/ClassTranslator.java new file mode 100644 index 0000000..afd3a77 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/ClassTranslator.java | |||
| @@ -0,0 +1,144 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode; | ||
| 12 | |||
| 13 | import java.util.Map; | ||
| 14 | |||
| 15 | import javassist.CtBehavior; | ||
| 16 | import javassist.CtClass; | ||
| 17 | import javassist.CtField; | ||
| 18 | import javassist.CtMethod; | ||
| 19 | import javassist.bytecode.ConstPool; | ||
| 20 | import javassist.bytecode.Descriptor; | ||
| 21 | import javassist.bytecode.SourceFileAttribute; | ||
| 22 | |||
| 23 | import com.google.common.collect.Maps; | ||
| 24 | |||
| 25 | import cuchaz.enigma.mapping.BehaviorEntry; | ||
| 26 | import cuchaz.enigma.mapping.BehaviorEntryFactory; | ||
| 27 | import cuchaz.enigma.mapping.ClassEntry; | ||
| 28 | import cuchaz.enigma.mapping.FieldEntry; | ||
| 29 | import cuchaz.enigma.mapping.JavassistUtil; | ||
| 30 | import cuchaz.enigma.mapping.MethodEntry; | ||
| 31 | import cuchaz.enigma.mapping.Signature; | ||
| 32 | import cuchaz.enigma.mapping.Translator; | ||
| 33 | import cuchaz.enigma.mapping.Type; | ||
| 34 | |||
| 35 | public class ClassTranslator { | ||
| 36 | |||
| 37 | private Translator m_translator; | ||
| 38 | |||
| 39 | public ClassTranslator(Translator translator) { | ||
| 40 | m_translator = translator; | ||
| 41 | } | ||
| 42 | |||
| 43 | public void translate(CtClass c) { | ||
| 44 | |||
| 45 | // NOTE: the order of these translations is very important | ||
| 46 | |||
| 47 | // translate all the field and method references in the code by editing the constant pool | ||
| 48 | ConstPool constants = c.getClassFile().getConstPool(); | ||
| 49 | ConstPoolEditor editor = new ConstPoolEditor(constants); | ||
| 50 | for (int i = 1; i < constants.getSize(); i++) { | ||
| 51 | switch (constants.getTag(i)) { | ||
| 52 | |||
| 53 | case ConstPool.CONST_Fieldref: { | ||
| 54 | |||
| 55 | // translate the name | ||
| 56 | FieldEntry entry = new FieldEntry( | ||
| 57 | new ClassEntry(Descriptor.toJvmName(constants.getFieldrefClassName(i))), | ||
| 58 | constants.getFieldrefName(i) | ||
| 59 | ); | ||
| 60 | FieldEntry translatedEntry = m_translator.translateEntry(entry); | ||
| 61 | |||
| 62 | // translate the type | ||
| 63 | Type type = new Type(constants.getFieldrefType(i)); | ||
| 64 | Type translatedType = m_translator.translateType(type); | ||
| 65 | |||
| 66 | if (!entry.equals(translatedEntry) || !type.equals(translatedType)) { | ||
| 67 | editor.changeMemberrefNameAndType(i, translatedEntry.getName(), translatedType.toString()); | ||
| 68 | } | ||
| 69 | } | ||
| 70 | break; | ||
| 71 | |||
| 72 | case ConstPool.CONST_Methodref: | ||
| 73 | case ConstPool.CONST_InterfaceMethodref: { | ||
| 74 | |||
| 75 | // translate the name and type | ||
| 76 | BehaviorEntry entry = BehaviorEntryFactory.create( | ||
| 77 | Descriptor.toJvmName(editor.getMemberrefClassname(i)), | ||
| 78 | editor.getMemberrefName(i), | ||
| 79 | editor.getMemberrefType(i) | ||
| 80 | ); | ||
| 81 | BehaviorEntry translatedEntry = m_translator.translateEntry(entry); | ||
| 82 | |||
| 83 | if (!entry.getName().equals(translatedEntry.getName()) || !entry.getSignature().equals(translatedEntry.getSignature())) { | ||
| 84 | editor.changeMemberrefNameAndType(i, translatedEntry.getName(), translatedEntry.getSignature().toString()); | ||
| 85 | } | ||
| 86 | } | ||
| 87 | break; | ||
| 88 | } | ||
| 89 | } | ||
| 90 | |||
| 91 | ClassEntry classEntry = new ClassEntry(Descriptor.toJvmName(c.getName())); | ||
| 92 | |||
| 93 | // translate all the fields | ||
| 94 | for (CtField field : c.getDeclaredFields()) { | ||
| 95 | |||
| 96 | // translate the name | ||
| 97 | FieldEntry entry = new FieldEntry(classEntry, field.getName()); | ||
| 98 | String translatedName = m_translator.translate(entry); | ||
| 99 | if (translatedName != null) { | ||
| 100 | field.setName(translatedName); | ||
| 101 | } | ||
| 102 | |||
| 103 | // translate the type | ||
| 104 | Type translatedType = m_translator.translateType(new Type(field.getFieldInfo().getDescriptor())); | ||
| 105 | field.getFieldInfo().setDescriptor(translatedType.toString()); | ||
| 106 | } | ||
| 107 | |||
| 108 | // translate all the methods and constructors | ||
| 109 | for (CtBehavior behavior : c.getDeclaredBehaviors()) { | ||
| 110 | if (behavior instanceof CtMethod) { | ||
| 111 | CtMethod method = (CtMethod)behavior; | ||
| 112 | |||
| 113 | // translate the name | ||
| 114 | MethodEntry entry = JavassistUtil.getMethodEntry(method); | ||
| 115 | String translatedName = m_translator.translate(entry); | ||
| 116 | if (translatedName != null) { | ||
| 117 | method.setName(translatedName); | ||
| 118 | } | ||
| 119 | } | ||
| 120 | |||
| 121 | // translate the type | ||
| 122 | Signature translatedSignature = m_translator.translateSignature(new Signature(behavior.getMethodInfo().getDescriptor())); | ||
| 123 | behavior.getMethodInfo().setDescriptor(translatedSignature.toString()); | ||
| 124 | } | ||
| 125 | |||
| 126 | // translate all the class names referenced in the code | ||
| 127 | // the above code only changed method/field/reference names and types, but not the class names themselves | ||
| 128 | Map<ClassEntry,ClassEntry> map = Maps.newHashMap(); | ||
| 129 | for (ClassEntry obfClassEntry : ClassRenamer.getAllClassEntries(c)) { | ||
| 130 | ClassEntry deobfClassEntry = m_translator.translateEntry(obfClassEntry); | ||
| 131 | if (!obfClassEntry.equals(deobfClassEntry)) { | ||
| 132 | map.put(obfClassEntry, deobfClassEntry); | ||
| 133 | } | ||
| 134 | } | ||
| 135 | ClassRenamer.renameClasses(c, map); | ||
| 136 | |||
| 137 | // translate the source file attribute too | ||
| 138 | ClassEntry deobfClassEntry = map.get(classEntry); | ||
| 139 | if (deobfClassEntry != null) { | ||
| 140 | String sourceFile = Descriptor.toJvmName(deobfClassEntry.getOuterClassName()) + ".java"; | ||
| 141 | c.getClassFile().addAttribute(new SourceFileAttribute(constants, sourceFile)); | ||
| 142 | } | ||
| 143 | } | ||
| 144 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/ConstPoolEditor.java b/src/cuchaz/enigma/bytecode/ConstPoolEditor.java new file mode 100644 index 0000000..2dec3b7 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/ConstPoolEditor.java | |||
| @@ -0,0 +1,263 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode; | ||
| 12 | |||
| 13 | import java.io.DataInputStream; | ||
| 14 | import java.io.DataOutputStream; | ||
| 15 | import java.lang.reflect.Constructor; | ||
| 16 | import java.lang.reflect.Field; | ||
| 17 | import java.lang.reflect.Method; | ||
| 18 | import java.util.HashMap; | ||
| 19 | |||
| 20 | import javassist.bytecode.ConstPool; | ||
| 21 | import javassist.bytecode.Descriptor; | ||
| 22 | import cuchaz.enigma.bytecode.accessors.ClassInfoAccessor; | ||
| 23 | import cuchaz.enigma.bytecode.accessors.ConstInfoAccessor; | ||
| 24 | import cuchaz.enigma.bytecode.accessors.MemberRefInfoAccessor; | ||
| 25 | |||
| 26 | public class ConstPoolEditor { | ||
| 27 | |||
| 28 | private static Method m_getItem; | ||
| 29 | private static Method m_addItem; | ||
| 30 | private static Method m_addItem0; | ||
| 31 | private static Field m_items; | ||
| 32 | private static Field m_cache; | ||
| 33 | private static Field m_numItems; | ||
| 34 | private static Field m_objects; | ||
| 35 | private static Field m_elements; | ||
| 36 | private static Method m_methodWritePool; | ||
| 37 | private static Constructor<ConstPool> m_constructorPool; | ||
| 38 | |||
| 39 | static { | ||
| 40 | try { | ||
| 41 | m_getItem = ConstPool.class.getDeclaredMethod("getItem", int.class); | ||
| 42 | m_getItem.setAccessible(true); | ||
| 43 | |||
| 44 | m_addItem = ConstPool.class.getDeclaredMethod("addItem", Class.forName("javassist.bytecode.ConstInfo")); | ||
| 45 | m_addItem.setAccessible(true); | ||
| 46 | |||
| 47 | m_addItem0 = ConstPool.class.getDeclaredMethod("addItem0", Class.forName("javassist.bytecode.ConstInfo")); | ||
| 48 | m_addItem0.setAccessible(true); | ||
| 49 | |||
| 50 | m_items = ConstPool.class.getDeclaredField("items"); | ||
| 51 | m_items.setAccessible(true); | ||
| 52 | |||
| 53 | m_cache = ConstPool.class.getDeclaredField("itemsCache"); | ||
| 54 | m_cache.setAccessible(true); | ||
| 55 | |||
| 56 | m_numItems = ConstPool.class.getDeclaredField("numOfItems"); | ||
| 57 | m_numItems.setAccessible(true); | ||
| 58 | |||
| 59 | m_objects = Class.forName("javassist.bytecode.LongVector").getDeclaredField("objects"); | ||
| 60 | m_objects.setAccessible(true); | ||
| 61 | |||
| 62 | m_elements = Class.forName("javassist.bytecode.LongVector").getDeclaredField("elements"); | ||
| 63 | m_elements.setAccessible(true); | ||
| 64 | |||
| 65 | m_methodWritePool = ConstPool.class.getDeclaredMethod("write", DataOutputStream.class); | ||
| 66 | m_methodWritePool.setAccessible(true); | ||
| 67 | |||
| 68 | m_constructorPool = ConstPool.class.getDeclaredConstructor(DataInputStream.class); | ||
| 69 | m_constructorPool.setAccessible(true); | ||
| 70 | } catch (Exception ex) { | ||
| 71 | throw new Error(ex); | ||
| 72 | } | ||
| 73 | } | ||
| 74 | |||
| 75 | private ConstPool m_pool; | ||
| 76 | |||
| 77 | public ConstPoolEditor(ConstPool pool) { | ||
| 78 | m_pool = pool; | ||
| 79 | } | ||
| 80 | |||
| 81 | public void writePool(DataOutputStream out) { | ||
| 82 | try { | ||
| 83 | m_methodWritePool.invoke(m_pool, out); | ||
| 84 | } catch (Exception ex) { | ||
| 85 | throw new Error(ex); | ||
| 86 | } | ||
| 87 | } | ||
| 88 | |||
| 89 | public static ConstPool readPool(DataInputStream in) { | ||
| 90 | try { | ||
| 91 | return m_constructorPool.newInstance(in); | ||
| 92 | } catch (Exception ex) { | ||
| 93 | throw new Error(ex); | ||
| 94 | } | ||
| 95 | } | ||
| 96 | |||
| 97 | public String getMemberrefClassname(int memberrefIndex) { | ||
| 98 | return Descriptor.toJvmName(m_pool.getClassInfo(m_pool.getMemberClass(memberrefIndex))); | ||
| 99 | } | ||
| 100 | |||
| 101 | public String getMemberrefName(int memberrefIndex) { | ||
| 102 | return m_pool.getUtf8Info(m_pool.getNameAndTypeName(m_pool.getMemberNameAndType(memberrefIndex))); | ||
| 103 | } | ||
| 104 | |||
| 105 | public String getMemberrefType(int memberrefIndex) { | ||
| 106 | return m_pool.getUtf8Info(m_pool.getNameAndTypeDescriptor(m_pool.getMemberNameAndType(memberrefIndex))); | ||
| 107 | } | ||
| 108 | |||
| 109 | public ConstInfoAccessor getItem(int index) { | ||
| 110 | try { | ||
| 111 | Object entry = m_getItem.invoke(m_pool, index); | ||
| 112 | if (entry == null) { | ||
| 113 | return null; | ||
| 114 | } | ||
| 115 | return new ConstInfoAccessor(entry); | ||
| 116 | } catch (Exception ex) { | ||
| 117 | throw new Error(ex); | ||
| 118 | } | ||
| 119 | } | ||
| 120 | |||
| 121 | public int addItem(Object item) { | ||
| 122 | try { | ||
| 123 | return (Integer)m_addItem.invoke(m_pool, item); | ||
| 124 | } catch (Exception ex) { | ||
| 125 | throw new Error(ex); | ||
| 126 | } | ||
| 127 | } | ||
| 128 | |||
| 129 | public int addItemForceNew(Object item) { | ||
| 130 | try { | ||
| 131 | return (Integer)m_addItem0.invoke(m_pool, item); | ||
| 132 | } catch (Exception ex) { | ||
| 133 | throw new Error(ex); | ||
| 134 | } | ||
| 135 | } | ||
| 136 | |||
| 137 | @SuppressWarnings("rawtypes") | ||
| 138 | public void removeLastItem() { | ||
| 139 | try { | ||
| 140 | // remove the item from the cache | ||
| 141 | HashMap cache = getCache(); | ||
| 142 | if (cache != null) { | ||
| 143 | Object item = getItem(m_pool.getSize() - 1); | ||
| 144 | cache.remove(item); | ||
| 145 | } | ||
| 146 | |||
| 147 | // remove the actual item | ||
| 148 | // based off of LongVector.addElement() | ||
| 149 | Object items = m_items.get(m_pool); | ||
| 150 | Object[][] objects = (Object[][])m_objects.get(items); | ||
| 151 | int numElements = (Integer)m_elements.get(items) - 1; | ||
| 152 | int nth = numElements >> 7; | ||
| 153 | int offset = numElements & (128 - 1); | ||
| 154 | objects[nth][offset] = null; | ||
| 155 | |||
| 156 | // decrement the number of items | ||
| 157 | m_elements.set(items, numElements); | ||
| 158 | m_numItems.set(m_pool, (Integer)m_numItems.get(m_pool) - 1); | ||
| 159 | } catch (Exception ex) { | ||
| 160 | throw new Error(ex); | ||
| 161 | } | ||
| 162 | } | ||
| 163 | |||
| 164 | @SuppressWarnings("rawtypes") | ||
| 165 | public HashMap getCache() { | ||
| 166 | try { | ||
| 167 | return (HashMap)m_cache.get(m_pool); | ||
| 168 | } catch (Exception ex) { | ||
| 169 | throw new Error(ex); | ||
| 170 | } | ||
| 171 | } | ||
| 172 | |||
| 173 | @SuppressWarnings({ "rawtypes", "unchecked" }) | ||
| 174 | public void changeMemberrefNameAndType(int memberrefIndex, String newName, String newType) { | ||
| 175 | // NOTE: when changing values, we always need to copy-on-write | ||
| 176 | try { | ||
| 177 | // get the memberref item | ||
| 178 | Object item = getItem(memberrefIndex).getItem(); | ||
| 179 | |||
| 180 | // update the cache | ||
| 181 | HashMap cache = getCache(); | ||
| 182 | if (cache != null) { | ||
| 183 | cache.remove(item); | ||
| 184 | } | ||
| 185 | |||
| 186 | new MemberRefInfoAccessor(item).setNameAndTypeIndex(m_pool.addNameAndTypeInfo(newName, newType)); | ||
| 187 | |||
| 188 | // update the cache | ||
| 189 | if (cache != null) { | ||
| 190 | cache.put(item, item); | ||
| 191 | } | ||
| 192 | } catch (Exception ex) { | ||
| 193 | throw new Error(ex); | ||
| 194 | } | ||
| 195 | |||
| 196 | // make sure the change worked | ||
| 197 | assert (newName.equals(getMemberrefName(memberrefIndex))); | ||
| 198 | assert (newType.equals(getMemberrefType(memberrefIndex))); | ||
| 199 | } | ||
| 200 | |||
| 201 | @SuppressWarnings({ "rawtypes", "unchecked" }) | ||
| 202 | public void changeClassName(int classNameIndex, String newName) { | ||
| 203 | // NOTE: when changing values, we always need to copy-on-write | ||
| 204 | try { | ||
| 205 | // get the class item | ||
| 206 | Object item = getItem(classNameIndex).getItem(); | ||
| 207 | |||
| 208 | // update the cache | ||
| 209 | HashMap cache = getCache(); | ||
| 210 | if (cache != null) { | ||
| 211 | cache.remove(item); | ||
| 212 | } | ||
| 213 | |||
| 214 | // add the new name and repoint the name-and-type to it | ||
| 215 | new ClassInfoAccessor(item).setNameIndex(m_pool.addUtf8Info(newName)); | ||
| 216 | |||
| 217 | // update the cache | ||
| 218 | if (cache != null) { | ||
| 219 | cache.put(item, item); | ||
| 220 | } | ||
| 221 | } catch (Exception ex) { | ||
| 222 | throw new Error(ex); | ||
| 223 | } | ||
| 224 | } | ||
| 225 | |||
| 226 | public static ConstPool newConstPool() { | ||
| 227 | // const pool expects the name of a class to initialize itself | ||
| 228 | // but we want an empty pool | ||
| 229 | // so give it a bogus name, and then clear the entries afterwards | ||
| 230 | ConstPool pool = new ConstPool("a"); | ||
| 231 | |||
| 232 | ConstPoolEditor editor = new ConstPoolEditor(pool); | ||
| 233 | int size = pool.getSize(); | ||
| 234 | for (int i = 0; i < size - 1; i++) { | ||
| 235 | editor.removeLastItem(); | ||
| 236 | } | ||
| 237 | |||
| 238 | // make sure the pool is actually empty | ||
| 239 | // although, in this case "empty" means one thing in it | ||
| 240 | // the JVM spec says index 0 should be reserved | ||
| 241 | assert (pool.getSize() == 1); | ||
| 242 | assert (editor.getItem(0) == null); | ||
| 243 | assert (editor.getItem(1) == null); | ||
| 244 | assert (editor.getItem(2) == null); | ||
| 245 | assert (editor.getItem(3) == null); | ||
| 246 | |||
| 247 | // also, clear the cache | ||
| 248 | editor.getCache().clear(); | ||
| 249 | |||
| 250 | return pool; | ||
| 251 | } | ||
| 252 | |||
| 253 | public String dump() { | ||
| 254 | StringBuilder buf = new StringBuilder(); | ||
| 255 | for (int i = 1; i < m_pool.getSize(); i++) { | ||
| 256 | buf.append(String.format("%4d", i)); | ||
| 257 | buf.append(" "); | ||
| 258 | buf.append(getItem(i).toString()); | ||
| 259 | buf.append("\n"); | ||
| 260 | } | ||
| 261 | return buf.toString(); | ||
| 262 | } | ||
| 263 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/InfoType.java b/src/cuchaz/enigma/bytecode/InfoType.java new file mode 100644 index 0000000..deaf623 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/InfoType.java | |||
| @@ -0,0 +1,317 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode; | ||
| 12 | |||
| 13 | import java.util.Collection; | ||
| 14 | import java.util.List; | ||
| 15 | import java.util.Map; | ||
| 16 | |||
| 17 | import com.google.common.collect.Lists; | ||
| 18 | import com.google.common.collect.Maps; | ||
| 19 | |||
| 20 | import cuchaz.enigma.bytecode.accessors.ClassInfoAccessor; | ||
| 21 | import cuchaz.enigma.bytecode.accessors.ConstInfoAccessor; | ||
| 22 | import cuchaz.enigma.bytecode.accessors.InvokeDynamicInfoAccessor; | ||
| 23 | import cuchaz.enigma.bytecode.accessors.MemberRefInfoAccessor; | ||
| 24 | import cuchaz.enigma.bytecode.accessors.MethodHandleInfoAccessor; | ||
| 25 | import cuchaz.enigma.bytecode.accessors.MethodTypeInfoAccessor; | ||
| 26 | import cuchaz.enigma.bytecode.accessors.NameAndTypeInfoAccessor; | ||
| 27 | import cuchaz.enigma.bytecode.accessors.StringInfoAccessor; | ||
| 28 | |||
| 29 | public enum InfoType { | ||
| 30 | |||
| 31 | Utf8Info( 1, 0 ), | ||
| 32 | IntegerInfo( 3, 0 ), | ||
| 33 | FloatInfo( 4, 0 ), | ||
| 34 | LongInfo( 5, 0 ), | ||
| 35 | DoubleInfo( 6, 0 ), | ||
| 36 | ClassInfo( 7, 1 ) { | ||
| 37 | |||
| 38 | @Override | ||
| 39 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 40 | ClassInfoAccessor accessor = new ClassInfoAccessor(entry.getItem()); | ||
| 41 | gatherIndexTree(indices, editor, accessor.getNameIndex()); | ||
| 42 | } | ||
| 43 | |||
| 44 | @Override | ||
| 45 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 46 | ClassInfoAccessor accessor = new ClassInfoAccessor(entry.getItem()); | ||
| 47 | accessor.setNameIndex(remapIndex(map, accessor.getNameIndex())); | ||
| 48 | } | ||
| 49 | |||
| 50 | @Override | ||
| 51 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 52 | ClassInfoAccessor accessor = new ClassInfoAccessor(entry.getItem()); | ||
| 53 | ConstInfoAccessor nameEntry = pool.getItem(accessor.getNameIndex()); | ||
| 54 | return nameEntry != null && nameEntry.getTag() == Utf8Info.getTag(); | ||
| 55 | } | ||
| 56 | }, | ||
| 57 | StringInfo( 8, 1 ) { | ||
| 58 | |||
| 59 | @Override | ||
| 60 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 61 | StringInfoAccessor accessor = new StringInfoAccessor(entry.getItem()); | ||
| 62 | gatherIndexTree(indices, editor, accessor.getStringIndex()); | ||
| 63 | } | ||
| 64 | |||
| 65 | @Override | ||
| 66 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 67 | StringInfoAccessor accessor = new StringInfoAccessor(entry.getItem()); | ||
| 68 | accessor.setStringIndex(remapIndex(map, accessor.getStringIndex())); | ||
| 69 | } | ||
| 70 | |||
| 71 | @Override | ||
| 72 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 73 | StringInfoAccessor accessor = new StringInfoAccessor(entry.getItem()); | ||
| 74 | ConstInfoAccessor stringEntry = pool.getItem(accessor.getStringIndex()); | ||
| 75 | return stringEntry != null && stringEntry.getTag() == Utf8Info.getTag(); | ||
| 76 | } | ||
| 77 | }, | ||
| 78 | FieldRefInfo( 9, 2 ) { | ||
| 79 | |||
| 80 | @Override | ||
| 81 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 82 | MemberRefInfoAccessor accessor = new MemberRefInfoAccessor(entry.getItem()); | ||
| 83 | gatherIndexTree(indices, editor, accessor.getClassIndex()); | ||
| 84 | gatherIndexTree(indices, editor, accessor.getNameAndTypeIndex()); | ||
| 85 | } | ||
| 86 | |||
| 87 | @Override | ||
| 88 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 89 | MemberRefInfoAccessor accessor = new MemberRefInfoAccessor(entry.getItem()); | ||
| 90 | accessor.setClassIndex(remapIndex(map, accessor.getClassIndex())); | ||
| 91 | accessor.setNameAndTypeIndex(remapIndex(map, accessor.getNameAndTypeIndex())); | ||
| 92 | } | ||
| 93 | |||
| 94 | @Override | ||
| 95 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 96 | MemberRefInfoAccessor accessor = new MemberRefInfoAccessor(entry.getItem()); | ||
| 97 | ConstInfoAccessor classEntry = pool.getItem(accessor.getClassIndex()); | ||
| 98 | ConstInfoAccessor nameAndTypeEntry = pool.getItem(accessor.getNameAndTypeIndex()); | ||
| 99 | return classEntry != null && classEntry.getTag() == ClassInfo.getTag() && nameAndTypeEntry != null && nameAndTypeEntry.getTag() == NameAndTypeInfo.getTag(); | ||
| 100 | } | ||
| 101 | }, | ||
| 102 | // same as FieldRefInfo | ||
| 103 | MethodRefInfo( 10, 2 ) { | ||
| 104 | |||
| 105 | @Override | ||
| 106 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 107 | FieldRefInfo.gatherIndexTree(indices, editor, entry); | ||
| 108 | } | ||
| 109 | |||
| 110 | @Override | ||
| 111 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 112 | FieldRefInfo.remapIndices(map, entry); | ||
| 113 | } | ||
| 114 | |||
| 115 | @Override | ||
| 116 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 117 | return FieldRefInfo.subIndicesAreValid(entry, pool); | ||
| 118 | } | ||
| 119 | }, | ||
| 120 | // same as FieldRefInfo | ||
| 121 | InterfaceMethodRefInfo( 11, 2 ) { | ||
| 122 | |||
| 123 | @Override | ||
| 124 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 125 | FieldRefInfo.gatherIndexTree(indices, editor, entry); | ||
| 126 | } | ||
| 127 | |||
| 128 | @Override | ||
| 129 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 130 | FieldRefInfo.remapIndices(map, entry); | ||
| 131 | } | ||
| 132 | |||
| 133 | @Override | ||
| 134 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 135 | return FieldRefInfo.subIndicesAreValid(entry, pool); | ||
| 136 | } | ||
| 137 | }, | ||
| 138 | NameAndTypeInfo( 12, 1 ) { | ||
| 139 | |||
| 140 | @Override | ||
| 141 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 142 | NameAndTypeInfoAccessor accessor = new NameAndTypeInfoAccessor(entry.getItem()); | ||
| 143 | gatherIndexTree(indices, editor, accessor.getNameIndex()); | ||
| 144 | gatherIndexTree(indices, editor, accessor.getTypeIndex()); | ||
| 145 | } | ||
| 146 | |||
| 147 | @Override | ||
| 148 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 149 | NameAndTypeInfoAccessor accessor = new NameAndTypeInfoAccessor(entry.getItem()); | ||
| 150 | accessor.setNameIndex(remapIndex(map, accessor.getNameIndex())); | ||
| 151 | accessor.setTypeIndex(remapIndex(map, accessor.getTypeIndex())); | ||
| 152 | } | ||
| 153 | |||
| 154 | @Override | ||
| 155 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 156 | NameAndTypeInfoAccessor accessor = new NameAndTypeInfoAccessor(entry.getItem()); | ||
| 157 | ConstInfoAccessor nameEntry = pool.getItem(accessor.getNameIndex()); | ||
| 158 | ConstInfoAccessor typeEntry = pool.getItem(accessor.getTypeIndex()); | ||
| 159 | return nameEntry != null && nameEntry.getTag() == Utf8Info.getTag() && typeEntry != null && typeEntry.getTag() == Utf8Info.getTag(); | ||
| 160 | } | ||
| 161 | }, | ||
| 162 | MethodHandleInfo( 15, 3 ) { | ||
| 163 | |||
| 164 | @Override | ||
| 165 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 166 | MethodHandleInfoAccessor accessor = new MethodHandleInfoAccessor(entry.getItem()); | ||
| 167 | gatherIndexTree(indices, editor, accessor.getTypeIndex()); | ||
| 168 | gatherIndexTree(indices, editor, accessor.getMethodRefIndex()); | ||
| 169 | } | ||
| 170 | |||
| 171 | @Override | ||
| 172 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 173 | MethodHandleInfoAccessor accessor = new MethodHandleInfoAccessor(entry.getItem()); | ||
| 174 | accessor.setTypeIndex(remapIndex(map, accessor.getTypeIndex())); | ||
| 175 | accessor.setMethodRefIndex(remapIndex(map, accessor.getMethodRefIndex())); | ||
| 176 | } | ||
| 177 | |||
| 178 | @Override | ||
| 179 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 180 | MethodHandleInfoAccessor accessor = new MethodHandleInfoAccessor(entry.getItem()); | ||
| 181 | ConstInfoAccessor typeEntry = pool.getItem(accessor.getTypeIndex()); | ||
| 182 | ConstInfoAccessor methodRefEntry = pool.getItem(accessor.getMethodRefIndex()); | ||
| 183 | return typeEntry != null && typeEntry.getTag() == Utf8Info.getTag() && methodRefEntry != null && methodRefEntry.getTag() == MethodRefInfo.getTag(); | ||
| 184 | } | ||
| 185 | }, | ||
| 186 | MethodTypeInfo( 16, 1 ) { | ||
| 187 | |||
| 188 | @Override | ||
| 189 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 190 | MethodTypeInfoAccessor accessor = new MethodTypeInfoAccessor(entry.getItem()); | ||
| 191 | gatherIndexTree(indices, editor, accessor.getTypeIndex()); | ||
| 192 | } | ||
| 193 | |||
| 194 | @Override | ||
| 195 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 196 | MethodTypeInfoAccessor accessor = new MethodTypeInfoAccessor(entry.getItem()); | ||
| 197 | accessor.setTypeIndex(remapIndex(map, accessor.getTypeIndex())); | ||
| 198 | } | ||
| 199 | |||
| 200 | @Override | ||
| 201 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 202 | MethodTypeInfoAccessor accessor = new MethodTypeInfoAccessor(entry.getItem()); | ||
| 203 | ConstInfoAccessor typeEntry = pool.getItem(accessor.getTypeIndex()); | ||
| 204 | return typeEntry != null && typeEntry.getTag() == Utf8Info.getTag(); | ||
| 205 | } | ||
| 206 | }, | ||
| 207 | InvokeDynamicInfo( 18, 2 ) { | ||
| 208 | |||
| 209 | @Override | ||
| 210 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 211 | InvokeDynamicInfoAccessor accessor = new InvokeDynamicInfoAccessor(entry.getItem()); | ||
| 212 | gatherIndexTree(indices, editor, accessor.getBootstrapIndex()); | ||
| 213 | gatherIndexTree(indices, editor, accessor.getNameAndTypeIndex()); | ||
| 214 | } | ||
| 215 | |||
| 216 | @Override | ||
| 217 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 218 | InvokeDynamicInfoAccessor accessor = new InvokeDynamicInfoAccessor(entry.getItem()); | ||
| 219 | accessor.setBootstrapIndex(remapIndex(map, accessor.getBootstrapIndex())); | ||
| 220 | accessor.setNameAndTypeIndex(remapIndex(map, accessor.getNameAndTypeIndex())); | ||
| 221 | } | ||
| 222 | |||
| 223 | @Override | ||
| 224 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 225 | InvokeDynamicInfoAccessor accessor = new InvokeDynamicInfoAccessor(entry.getItem()); | ||
| 226 | ConstInfoAccessor bootstrapEntry = pool.getItem(accessor.getBootstrapIndex()); | ||
| 227 | ConstInfoAccessor nameAndTypeEntry = pool.getItem(accessor.getNameAndTypeIndex()); | ||
| 228 | return bootstrapEntry != null && bootstrapEntry.getTag() == Utf8Info.getTag() && nameAndTypeEntry != null && nameAndTypeEntry.getTag() == NameAndTypeInfo.getTag(); | ||
| 229 | } | ||
| 230 | }; | ||
| 231 | |||
| 232 | private static Map<Integer,InfoType> m_types; | ||
| 233 | |||
| 234 | static { | ||
| 235 | m_types = Maps.newTreeMap(); | ||
| 236 | for (InfoType type : values()) { | ||
| 237 | m_types.put(type.getTag(), type); | ||
| 238 | } | ||
| 239 | } | ||
| 240 | |||
| 241 | private int m_tag; | ||
| 242 | private int m_level; | ||
| 243 | |||
| 244 | private InfoType(int tag, int level) { | ||
| 245 | m_tag = tag; | ||
| 246 | m_level = level; | ||
| 247 | } | ||
| 248 | |||
| 249 | public int getTag() { | ||
| 250 | return m_tag; | ||
| 251 | } | ||
| 252 | |||
| 253 | public int getLevel() { | ||
| 254 | return m_level; | ||
| 255 | } | ||
| 256 | |||
| 257 | public void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, ConstInfoAccessor entry) { | ||
| 258 | // by default, do nothing | ||
| 259 | } | ||
| 260 | |||
| 261 | public void remapIndices(Map<Integer,Integer> map, ConstInfoAccessor entry) { | ||
| 262 | // by default, do nothing | ||
| 263 | } | ||
| 264 | |||
| 265 | public boolean subIndicesAreValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 266 | // by default, everything is good | ||
| 267 | return true; | ||
| 268 | } | ||
| 269 | |||
| 270 | public boolean selfIndexIsValid(ConstInfoAccessor entry, ConstPoolEditor pool) { | ||
| 271 | ConstInfoAccessor entryCheck = pool.getItem(entry.getIndex()); | ||
| 272 | if (entryCheck == null) { | ||
| 273 | return false; | ||
| 274 | } | ||
| 275 | return entryCheck.getItem().equals(entry.getItem()); | ||
| 276 | } | ||
| 277 | |||
| 278 | public static InfoType getByTag(int tag) { | ||
| 279 | return m_types.get(tag); | ||
| 280 | } | ||
| 281 | |||
| 282 | public static List<InfoType> getByLevel(int level) { | ||
| 283 | List<InfoType> types = Lists.newArrayList(); | ||
| 284 | for (InfoType type : values()) { | ||
| 285 | if (type.getLevel() == level) { | ||
| 286 | types.add(type); | ||
| 287 | } | ||
| 288 | } | ||
| 289 | return types; | ||
| 290 | } | ||
| 291 | |||
| 292 | public static List<InfoType> getSortedByLevel() { | ||
| 293 | List<InfoType> types = Lists.newArrayList(); | ||
| 294 | types.addAll(getByLevel(0)); | ||
| 295 | types.addAll(getByLevel(1)); | ||
| 296 | types.addAll(getByLevel(2)); | ||
| 297 | types.addAll(getByLevel(3)); | ||
| 298 | return types; | ||
| 299 | } | ||
| 300 | |||
| 301 | public static void gatherIndexTree(Collection<Integer> indices, ConstPoolEditor editor, int index) { | ||
| 302 | // add own index | ||
| 303 | indices.add(index); | ||
| 304 | |||
| 305 | // recurse | ||
| 306 | ConstInfoAccessor entry = editor.getItem(index); | ||
| 307 | entry.getType().gatherIndexTree(indices, editor, entry); | ||
| 308 | } | ||
| 309 | |||
| 310 | private static int remapIndex(Map<Integer,Integer> map, int index) { | ||
| 311 | Integer newIndex = map.get(index); | ||
| 312 | if (newIndex == null) { | ||
| 313 | newIndex = index; | ||
| 314 | } | ||
| 315 | return newIndex; | ||
| 316 | } | ||
| 317 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/InnerClassWriter.java b/src/cuchaz/enigma/bytecode/InnerClassWriter.java new file mode 100644 index 0000000..817500b --- /dev/null +++ b/src/cuchaz/enigma/bytecode/InnerClassWriter.java | |||
| @@ -0,0 +1,102 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode; | ||
| 12 | |||
| 13 | import java.util.Collection; | ||
| 14 | |||
| 15 | import javassist.CtClass; | ||
| 16 | import javassist.bytecode.AccessFlag; | ||
| 17 | import javassist.bytecode.ConstPool; | ||
| 18 | import javassist.bytecode.Descriptor; | ||
| 19 | import javassist.bytecode.EnclosingMethodAttribute; | ||
| 20 | import javassist.bytecode.InnerClassesAttribute; | ||
| 21 | import cuchaz.enigma.Constants; | ||
| 22 | import cuchaz.enigma.analysis.JarIndex; | ||
| 23 | import cuchaz.enigma.mapping.BehaviorEntry; | ||
| 24 | import cuchaz.enigma.mapping.ClassEntry; | ||
| 25 | |||
| 26 | public class InnerClassWriter { | ||
| 27 | |||
| 28 | private JarIndex m_jarIndex; | ||
| 29 | |||
| 30 | public InnerClassWriter(JarIndex jarIndex) { | ||
| 31 | m_jarIndex = jarIndex; | ||
| 32 | } | ||
| 33 | |||
| 34 | public void write(CtClass c) { | ||
| 35 | |||
| 36 | // is this an inner or outer class? | ||
| 37 | String obfInnerClassName = new ClassEntry(Descriptor.toJvmName(c.getName())).getSimpleName(); | ||
| 38 | String obfOuterClassName = m_jarIndex.getOuterClass(obfInnerClassName); | ||
| 39 | if (obfOuterClassName == null) { | ||
| 40 | // this is an outer class | ||
| 41 | obfOuterClassName = Descriptor.toJvmName(c.getName()); | ||
| 42 | } else { | ||
| 43 | // this is an inner class, rename it to outer$inner | ||
| 44 | ClassEntry obfClassEntry = new ClassEntry(obfOuterClassName + "$" + obfInnerClassName); | ||
| 45 | c.setName(obfClassEntry.getName()); | ||
| 46 | |||
| 47 | BehaviorEntry caller = m_jarIndex.getAnonymousClassCaller(obfInnerClassName); | ||
| 48 | if (caller != null) { | ||
| 49 | // write the enclosing method attribute | ||
| 50 | if (caller.getName().equals("<clinit>")) { | ||
| 51 | c.getClassFile().addAttribute(new EnclosingMethodAttribute(c.getClassFile().getConstPool(), caller.getClassName())); | ||
| 52 | } else { | ||
| 53 | c.getClassFile().addAttribute(new EnclosingMethodAttribute(c.getClassFile().getConstPool(), caller.getClassName(), caller.getName(), caller.getSignature().toString())); | ||
| 54 | } | ||
| 55 | } | ||
| 56 | } | ||
| 57 | |||
| 58 | // write the inner classes if needed | ||
| 59 | Collection<String> obfInnerClassNames = m_jarIndex.getInnerClasses(obfOuterClassName); | ||
| 60 | if (obfInnerClassNames != null && !obfInnerClassNames.isEmpty()) { | ||
| 61 | writeInnerClasses(c, obfOuterClassName, obfInnerClassNames); | ||
| 62 | } | ||
| 63 | } | ||
| 64 | |||
| 65 | private void writeInnerClasses(CtClass c, String obfOuterClassName, Collection<String> obfInnerClassNames) { | ||
| 66 | InnerClassesAttribute attr = new InnerClassesAttribute(c.getClassFile().getConstPool()); | ||
| 67 | c.getClassFile().addAttribute(attr); | ||
| 68 | for (String obfInnerClassName : obfInnerClassNames) { | ||
| 69 | // get the new inner class name | ||
| 70 | ClassEntry obfClassEntry = new ClassEntry(obfOuterClassName + "$" + obfInnerClassName); | ||
| 71 | |||
| 72 | // here's what the JVM spec says about the InnerClasses attribute | ||
| 73 | // append( inner, outer of inner if inner is member of outer 0 ow, name after $ if inner not anonymous 0 ow, flags ); | ||
| 74 | |||
| 75 | // update the attribute with this inner class | ||
| 76 | ConstPool constPool = c.getClassFile().getConstPool(); | ||
| 77 | int innerClassIndex = constPool.addClassInfo(obfClassEntry.getName()); | ||
| 78 | int outerClassIndex = 0; | ||
| 79 | int innerClassSimpleNameIndex = 0; | ||
| 80 | if (!m_jarIndex.isAnonymousClass(obfInnerClassName)) { | ||
| 81 | outerClassIndex = constPool.addClassInfo(obfClassEntry.getOuterClassName()); | ||
| 82 | innerClassSimpleNameIndex = constPool.addUtf8Info(obfClassEntry.getInnerClassName()); | ||
| 83 | } | ||
| 84 | |||
| 85 | attr.append(innerClassIndex, outerClassIndex, innerClassSimpleNameIndex, c.getClassFile().getAccessFlags() & ~AccessFlag.SUPER); | ||
| 86 | |||
| 87 | /* DEBUG | ||
| 88 | System.out.println(String.format("\tOBF: %s -> ATTR: %s,%s,%s (replace %s with %s)", | ||
| 89 | obfClassEntry, | ||
| 90 | attr.outerClass(attr.tableLength() - 1), | ||
| 91 | attr.innerClass(attr.tableLength() - 1), | ||
| 92 | attr.innerName(attr.tableLength() - 1), | ||
| 93 | Constants.NonePackage + "/" + obfInnerClassName, | ||
| 94 | obfClassEntry.getName() | ||
| 95 | )); | ||
| 96 | */ | ||
| 97 | |||
| 98 | // make sure the outer class references only the new inner class names | ||
| 99 | c.replaceClassName(Constants.NonePackage + "/" + obfInnerClassName, obfClassEntry.getName()); | ||
| 100 | } | ||
| 101 | } | ||
| 102 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/MethodParameterWriter.java b/src/cuchaz/enigma/bytecode/MethodParameterWriter.java new file mode 100644 index 0000000..5d4ca1a --- /dev/null +++ b/src/cuchaz/enigma/bytecode/MethodParameterWriter.java | |||
| @@ -0,0 +1,53 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode; | ||
| 12 | |||
| 13 | import java.util.ArrayList; | ||
| 14 | import java.util.List; | ||
| 15 | |||
| 16 | import javassist.CtBehavior; | ||
| 17 | import javassist.CtClass; | ||
| 18 | import cuchaz.enigma.mapping.ArgumentEntry; | ||
| 19 | import cuchaz.enigma.mapping.BehaviorEntry; | ||
| 20 | import cuchaz.enigma.mapping.BehaviorEntryFactory; | ||
| 21 | import cuchaz.enigma.mapping.Translator; | ||
| 22 | |||
| 23 | public class MethodParameterWriter { | ||
| 24 | |||
| 25 | private Translator m_translator; | ||
| 26 | |||
| 27 | public MethodParameterWriter(Translator translator) { | ||
| 28 | m_translator = translator; | ||
| 29 | } | ||
| 30 | |||
| 31 | public void writeMethodArguments(CtClass c) { | ||
| 32 | |||
| 33 | // Procyon will read method arguments from the "MethodParameters" attribute, so write those | ||
| 34 | for (CtBehavior behavior : c.getDeclaredBehaviors()) { | ||
| 35 | BehaviorEntry behaviorEntry = BehaviorEntryFactory.create(behavior); | ||
| 36 | |||
| 37 | // get the number of arguments | ||
| 38 | int numParams = behaviorEntry.getSignature().getArgumentTypes().size(); | ||
| 39 | if (numParams <= 0) { | ||
| 40 | continue; | ||
| 41 | } | ||
| 42 | |||
| 43 | // get the list of argument names | ||
| 44 | List<String> names = new ArrayList<String>(numParams); | ||
| 45 | for (int i = 0; i < numParams; i++) { | ||
| 46 | names.add(m_translator.translate(new ArgumentEntry(behaviorEntry, i, ""))); | ||
| 47 | } | ||
| 48 | |||
| 49 | // save the mappings to the class | ||
| 50 | MethodParametersAttribute.updateClass(behavior.getMethodInfo(), names); | ||
| 51 | } | ||
| 52 | } | ||
| 53 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/MethodParametersAttribute.java b/src/cuchaz/enigma/bytecode/MethodParametersAttribute.java new file mode 100644 index 0000000..bf95956 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/MethodParametersAttribute.java | |||
| @@ -0,0 +1,85 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode; | ||
| 12 | |||
| 13 | import java.io.ByteArrayOutputStream; | ||
| 14 | import java.io.DataOutputStream; | ||
| 15 | import java.io.IOException; | ||
| 16 | import java.util.ArrayList; | ||
| 17 | import java.util.List; | ||
| 18 | |||
| 19 | import javassist.bytecode.AttributeInfo; | ||
| 20 | import javassist.bytecode.ConstPool; | ||
| 21 | import javassist.bytecode.MethodInfo; | ||
| 22 | |||
| 23 | public class MethodParametersAttribute extends AttributeInfo { | ||
| 24 | |||
| 25 | private MethodParametersAttribute(ConstPool pool, List<Integer> parameterNameIndices) { | ||
| 26 | super(pool, "MethodParameters", writeStruct(parameterNameIndices)); | ||
| 27 | } | ||
| 28 | |||
| 29 | public static void updateClass(MethodInfo info, List<String> names) { | ||
| 30 | // add the names to the class const pool | ||
| 31 | ConstPool constPool = info.getConstPool(); | ||
| 32 | List<Integer> parameterNameIndices = new ArrayList<Integer>(); | ||
| 33 | for (String name : names) { | ||
| 34 | if (name != null) { | ||
| 35 | parameterNameIndices.add(constPool.addUtf8Info(name)); | ||
| 36 | } else { | ||
| 37 | parameterNameIndices.add(0); | ||
| 38 | } | ||
| 39 | } | ||
| 40 | |||
| 41 | // add the attribute to the method | ||
| 42 | info.addAttribute(new MethodParametersAttribute(constPool, parameterNameIndices)); | ||
| 43 | } | ||
| 44 | |||
| 45 | private static byte[] writeStruct(List<Integer> parameterNameIndices) { | ||
| 46 | // JVM 8 Spec says the struct looks like this: | ||
| 47 | // http://cr.openjdk.java.net/~mr/se/8/java-se-8-fr-spec-01/java-se-8-jvms-fr-diffs.pdf | ||
| 48 | // uint8 num_params | ||
| 49 | // for each param: | ||
| 50 | // uint16 name_index -> points to UTF8 entry in constant pool, or 0 for no entry | ||
| 51 | // uint16 access_flags -> don't care, just set to 0 | ||
| 52 | |||
| 53 | ByteArrayOutputStream buf = new ByteArrayOutputStream(); | ||
| 54 | DataOutputStream out = new DataOutputStream(buf); | ||
| 55 | |||
| 56 | // NOTE: java hates unsigned integers, so we have to be careful here | ||
| 57 | // the writeShort(), writeByte() methods will read 16,8 low-order bits from the int argument | ||
| 58 | // as long as the int argument is in range of the unsigned short/byte type, it will be written as an unsigned short/byte | ||
| 59 | // if the int is out of range, the byte stream won't look the way we want and weird things will happen | ||
| 60 | final int SIZEOF_UINT8 = 1; | ||
| 61 | final int SIZEOF_UINT16 = 2; | ||
| 62 | final int MAX_UINT8 = (1 << 8) - 1; | ||
| 63 | final int MAX_UINT16 = (1 << 16) - 1; | ||
| 64 | |||
| 65 | try { | ||
| 66 | assert (parameterNameIndices.size() >= 0 && parameterNameIndices.size() <= MAX_UINT8); | ||
| 67 | out.writeByte(parameterNameIndices.size()); | ||
| 68 | |||
| 69 | for (Integer index : parameterNameIndices) { | ||
| 70 | assert (index >= 0 && index <= MAX_UINT16); | ||
| 71 | out.writeShort(index); | ||
| 72 | |||
| 73 | // just write 0 for the access flags | ||
| 74 | out.writeShort(0); | ||
| 75 | } | ||
| 76 | |||
| 77 | out.close(); | ||
| 78 | byte[] data = buf.toByteArray(); | ||
| 79 | assert (data.length == SIZEOF_UINT8 + parameterNameIndices.size() * (SIZEOF_UINT16 + SIZEOF_UINT16)); | ||
| 80 | return data; | ||
| 81 | } catch (IOException ex) { | ||
| 82 | throw new Error(ex); | ||
| 83 | } | ||
| 84 | } | ||
| 85 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/accessors/ClassInfoAccessor.java b/src/cuchaz/enigma/bytecode/accessors/ClassInfoAccessor.java new file mode 100644 index 0000000..d76f056 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/accessors/ClassInfoAccessor.java | |||
| @@ -0,0 +1,55 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode.accessors; | ||
| 12 | |||
| 13 | import java.lang.reflect.Field; | ||
| 14 | |||
| 15 | public class ClassInfoAccessor { | ||
| 16 | |||
| 17 | private static Class<?> m_class; | ||
| 18 | private static Field m_nameIndex; | ||
| 19 | |||
| 20 | static { | ||
| 21 | try { | ||
| 22 | m_class = Class.forName("javassist.bytecode.ClassInfo"); | ||
| 23 | m_nameIndex = m_class.getDeclaredField("name"); | ||
| 24 | m_nameIndex.setAccessible(true); | ||
| 25 | } catch (Exception ex) { | ||
| 26 | throw new Error(ex); | ||
| 27 | } | ||
| 28 | } | ||
| 29 | |||
| 30 | public static boolean isType(ConstInfoAccessor accessor) { | ||
| 31 | return m_class.isAssignableFrom(accessor.getItem().getClass()); | ||
| 32 | } | ||
| 33 | |||
| 34 | private Object m_item; | ||
| 35 | |||
| 36 | public ClassInfoAccessor(Object item) { | ||
| 37 | m_item = item; | ||
| 38 | } | ||
| 39 | |||
| 40 | public int getNameIndex() { | ||
| 41 | try { | ||
| 42 | return (Integer)m_nameIndex.get(m_item); | ||
| 43 | } catch (Exception ex) { | ||
| 44 | throw new Error(ex); | ||
| 45 | } | ||
| 46 | } | ||
| 47 | |||
| 48 | public void setNameIndex(int val) { | ||
| 49 | try { | ||
| 50 | m_nameIndex.set(m_item, val); | ||
| 51 | } catch (Exception ex) { | ||
| 52 | throw new Error(ex); | ||
| 53 | } | ||
| 54 | } | ||
| 55 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/accessors/ConstInfoAccessor.java b/src/cuchaz/enigma/bytecode/accessors/ConstInfoAccessor.java new file mode 100644 index 0000000..d00c102 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/accessors/ConstInfoAccessor.java | |||
| @@ -0,0 +1,156 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode.accessors; | ||
| 12 | |||
| 13 | import java.io.ByteArrayInputStream; | ||
| 14 | import java.io.ByteArrayOutputStream; | ||
| 15 | import java.io.DataInputStream; | ||
| 16 | import java.io.DataOutputStream; | ||
| 17 | import java.io.IOException; | ||
| 18 | import java.io.PrintWriter; | ||
| 19 | import java.lang.reflect.Constructor; | ||
| 20 | import java.lang.reflect.Field; | ||
| 21 | import java.lang.reflect.Method; | ||
| 22 | |||
| 23 | import cuchaz.enigma.bytecode.InfoType; | ||
| 24 | |||
| 25 | public class ConstInfoAccessor { | ||
| 26 | |||
| 27 | private static Class<?> m_class; | ||
| 28 | private static Field m_index; | ||
| 29 | private static Method m_getTag; | ||
| 30 | |||
| 31 | static { | ||
| 32 | try { | ||
| 33 | m_class = Class.forName("javassist.bytecode.ConstInfo"); | ||
| 34 | m_index = m_class.getDeclaredField("index"); | ||
| 35 | m_index.setAccessible(true); | ||
| 36 | m_getTag = m_class.getMethod("getTag"); | ||
| 37 | m_getTag.setAccessible(true); | ||
| 38 | } catch (Exception ex) { | ||
| 39 | throw new Error(ex); | ||
| 40 | } | ||
| 41 | } | ||
| 42 | |||
| 43 | private Object m_item; | ||
| 44 | |||
| 45 | public ConstInfoAccessor(Object item) { | ||
| 46 | if (item == null) { | ||
| 47 | throw new IllegalArgumentException("item cannot be null!"); | ||
| 48 | } | ||
| 49 | m_item = item; | ||
| 50 | } | ||
| 51 | |||
| 52 | public ConstInfoAccessor(DataInputStream in) throws IOException { | ||
| 53 | try { | ||
| 54 | // read the entry | ||
| 55 | String className = in.readUTF(); | ||
| 56 | int oldIndex = in.readInt(); | ||
| 57 | |||
| 58 | // NOTE: ConstInfo instances write a type id (a "tag"), but they don't read it back | ||
| 59 | // so we have to read it here | ||
| 60 | in.readByte(); | ||
| 61 | |||
| 62 | Constructor<?> constructor = Class.forName(className).getConstructor(DataInputStream.class, int.class); | ||
| 63 | constructor.setAccessible(true); | ||
| 64 | m_item = constructor.newInstance(in, oldIndex); | ||
| 65 | } catch (IOException ex) { | ||
| 66 | throw ex; | ||
| 67 | } catch (Exception ex) { | ||
| 68 | throw new Error(ex); | ||
| 69 | } | ||
| 70 | } | ||
| 71 | |||
| 72 | public Object getItem() { | ||
| 73 | return m_item; | ||
| 74 | } | ||
| 75 | |||
| 76 | public int getIndex() { | ||
| 77 | try { | ||
| 78 | return (Integer)m_index.get(m_item); | ||
| 79 | } catch (Exception ex) { | ||
| 80 | throw new Error(ex); | ||
| 81 | } | ||
| 82 | } | ||
| 83 | |||
| 84 | public void setIndex(int val) { | ||
| 85 | try { | ||
| 86 | m_index.set(m_item, val); | ||
| 87 | } catch (Exception ex) { | ||
| 88 | throw new Error(ex); | ||
| 89 | } | ||
| 90 | } | ||
| 91 | |||
| 92 | public int getTag() { | ||
| 93 | try { | ||
| 94 | return (Integer)m_getTag.invoke(m_item); | ||
| 95 | } catch (Exception ex) { | ||
| 96 | throw new Error(ex); | ||
| 97 | } | ||
| 98 | } | ||
| 99 | |||
| 100 | public ConstInfoAccessor copy() { | ||
| 101 | return new ConstInfoAccessor(copyItem()); | ||
| 102 | } | ||
| 103 | |||
| 104 | public Object copyItem() { | ||
| 105 | // I don't know of a simpler way to copy one of these silly things... | ||
| 106 | try { | ||
| 107 | // serialize the item | ||
| 108 | ByteArrayOutputStream buf = new ByteArrayOutputStream(); | ||
| 109 | DataOutputStream out = new DataOutputStream(buf); | ||
| 110 | write(out); | ||
| 111 | |||
| 112 | // deserialize the item | ||
| 113 | DataInputStream in = new DataInputStream(new ByteArrayInputStream(buf.toByteArray())); | ||
| 114 | Object item = new ConstInfoAccessor(in).getItem(); | ||
| 115 | in.close(); | ||
| 116 | |||
| 117 | return item; | ||
| 118 | } catch (Exception ex) { | ||
| 119 | throw new Error(ex); | ||
| 120 | } | ||
| 121 | } | ||
| 122 | |||
| 123 | public void write(DataOutputStream out) throws IOException { | ||
| 124 | try { | ||
| 125 | out.writeUTF(m_item.getClass().getName()); | ||
| 126 | out.writeInt(getIndex()); | ||
| 127 | |||
| 128 | Method method = m_item.getClass().getMethod("write", DataOutputStream.class); | ||
| 129 | method.setAccessible(true); | ||
| 130 | method.invoke(m_item, out); | ||
| 131 | } catch (IOException ex) { | ||
| 132 | throw ex; | ||
| 133 | } catch (Exception ex) { | ||
| 134 | throw new Error(ex); | ||
| 135 | } | ||
| 136 | } | ||
| 137 | |||
| 138 | @Override | ||
| 139 | public String toString() { | ||
| 140 | try { | ||
| 141 | ByteArrayOutputStream buf = new ByteArrayOutputStream(); | ||
| 142 | PrintWriter out = new PrintWriter(buf); | ||
| 143 | Method print = m_item.getClass().getMethod("print", PrintWriter.class); | ||
| 144 | print.setAccessible(true); | ||
| 145 | print.invoke(m_item, out); | ||
| 146 | out.close(); | ||
| 147 | return buf.toString().replace("\n", ""); | ||
| 148 | } catch (Exception ex) { | ||
| 149 | throw new Error(ex); | ||
| 150 | } | ||
| 151 | } | ||
| 152 | |||
| 153 | public InfoType getType() { | ||
| 154 | return InfoType.getByTag(getTag()); | ||
| 155 | } | ||
| 156 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/accessors/InvokeDynamicInfoAccessor.java b/src/cuchaz/enigma/bytecode/accessors/InvokeDynamicInfoAccessor.java new file mode 100644 index 0000000..0d780ea --- /dev/null +++ b/src/cuchaz/enigma/bytecode/accessors/InvokeDynamicInfoAccessor.java | |||
| @@ -0,0 +1,74 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode.accessors; | ||
| 12 | |||
| 13 | import java.lang.reflect.Field; | ||
| 14 | |||
| 15 | public class InvokeDynamicInfoAccessor { | ||
| 16 | |||
| 17 | private static Class<?> m_class; | ||
| 18 | private static Field m_bootstrapIndex; | ||
| 19 | private static Field m_nameAndTypeIndex; | ||
| 20 | |||
| 21 | static { | ||
| 22 | try { | ||
| 23 | m_class = Class.forName("javassist.bytecode.InvokeDynamicInfo"); | ||
| 24 | m_bootstrapIndex = m_class.getDeclaredField("bootstrap"); | ||
| 25 | m_bootstrapIndex.setAccessible(true); | ||
| 26 | m_nameAndTypeIndex = m_class.getDeclaredField("nameAndType"); | ||
| 27 | m_nameAndTypeIndex.setAccessible(true); | ||
| 28 | } catch (Exception ex) { | ||
| 29 | throw new Error(ex); | ||
| 30 | } | ||
| 31 | } | ||
| 32 | |||
| 33 | public static boolean isType(ConstInfoAccessor accessor) { | ||
| 34 | return m_class.isAssignableFrom(accessor.getItem().getClass()); | ||
| 35 | } | ||
| 36 | |||
| 37 | private Object m_item; | ||
| 38 | |||
| 39 | public InvokeDynamicInfoAccessor(Object item) { | ||
| 40 | m_item = item; | ||
| 41 | } | ||
| 42 | |||
| 43 | public int getBootstrapIndex() { | ||
| 44 | try { | ||
| 45 | return (Integer)m_bootstrapIndex.get(m_item); | ||
| 46 | } catch (Exception ex) { | ||
| 47 | throw new Error(ex); | ||
| 48 | } | ||
| 49 | } | ||
| 50 | |||
| 51 | public void setBootstrapIndex(int val) { | ||
| 52 | try { | ||
| 53 | m_bootstrapIndex.set(m_item, val); | ||
| 54 | } catch (Exception ex) { | ||
| 55 | throw new Error(ex); | ||
| 56 | } | ||
| 57 | } | ||
| 58 | |||
| 59 | public int getNameAndTypeIndex() { | ||
| 60 | try { | ||
| 61 | return (Integer)m_nameAndTypeIndex.get(m_item); | ||
| 62 | } catch (Exception ex) { | ||
| 63 | throw new Error(ex); | ||
| 64 | } | ||
| 65 | } | ||
| 66 | |||
| 67 | public void setNameAndTypeIndex(int val) { | ||
| 68 | try { | ||
| 69 | m_nameAndTypeIndex.set(m_item, val); | ||
| 70 | } catch (Exception ex) { | ||
| 71 | throw new Error(ex); | ||
| 72 | } | ||
| 73 | } | ||
| 74 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/accessors/MemberRefInfoAccessor.java b/src/cuchaz/enigma/bytecode/accessors/MemberRefInfoAccessor.java new file mode 100644 index 0000000..9fe945f --- /dev/null +++ b/src/cuchaz/enigma/bytecode/accessors/MemberRefInfoAccessor.java | |||
| @@ -0,0 +1,74 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode.accessors; | ||
| 12 | |||
| 13 | import java.lang.reflect.Field; | ||
| 14 | |||
| 15 | public class MemberRefInfoAccessor { | ||
| 16 | |||
| 17 | private static Class<?> m_class; | ||
| 18 | private static Field m_classIndex; | ||
| 19 | private static Field m_nameAndTypeIndex; | ||
| 20 | |||
| 21 | static { | ||
| 22 | try { | ||
| 23 | m_class = Class.forName("javassist.bytecode.MemberrefInfo"); | ||
| 24 | m_classIndex = m_class.getDeclaredField("classIndex"); | ||
| 25 | m_classIndex.setAccessible(true); | ||
| 26 | m_nameAndTypeIndex = m_class.getDeclaredField("nameAndTypeIndex"); | ||
| 27 | m_nameAndTypeIndex.setAccessible(true); | ||
| 28 | } catch (Exception ex) { | ||
| 29 | throw new Error(ex); | ||
| 30 | } | ||
| 31 | } | ||
| 32 | |||
| 33 | public static boolean isType(ConstInfoAccessor accessor) { | ||
| 34 | return m_class.isAssignableFrom(accessor.getItem().getClass()); | ||
| 35 | } | ||
| 36 | |||
| 37 | private Object m_item; | ||
| 38 | |||
| 39 | public MemberRefInfoAccessor(Object item) { | ||
| 40 | m_item = item; | ||
| 41 | } | ||
| 42 | |||
| 43 | public int getClassIndex() { | ||
| 44 | try { | ||
| 45 | return (Integer)m_classIndex.get(m_item); | ||
| 46 | } catch (Exception ex) { | ||
| 47 | throw new Error(ex); | ||
| 48 | } | ||
| 49 | } | ||
| 50 | |||
| 51 | public void setClassIndex(int val) { | ||
| 52 | try { | ||
| 53 | m_classIndex.set(m_item, val); | ||
| 54 | } catch (Exception ex) { | ||
| 55 | throw new Error(ex); | ||
| 56 | } | ||
| 57 | } | ||
| 58 | |||
| 59 | public int getNameAndTypeIndex() { | ||
| 60 | try { | ||
| 61 | return (Integer)m_nameAndTypeIndex.get(m_item); | ||
| 62 | } catch (Exception ex) { | ||
| 63 | throw new Error(ex); | ||
| 64 | } | ||
| 65 | } | ||
| 66 | |||
| 67 | public void setNameAndTypeIndex(int val) { | ||
| 68 | try { | ||
| 69 | m_nameAndTypeIndex.set(m_item, val); | ||
| 70 | } catch (Exception ex) { | ||
| 71 | throw new Error(ex); | ||
| 72 | } | ||
| 73 | } | ||
| 74 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/accessors/MethodHandleInfoAccessor.java b/src/cuchaz/enigma/bytecode/accessors/MethodHandleInfoAccessor.java new file mode 100644 index 0000000..4c95b22 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/accessors/MethodHandleInfoAccessor.java | |||
| @@ -0,0 +1,74 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode.accessors; | ||
| 12 | |||
| 13 | import java.lang.reflect.Field; | ||
| 14 | |||
| 15 | public class MethodHandleInfoAccessor { | ||
| 16 | |||
| 17 | private static Class<?> m_class; | ||
| 18 | private static Field m_kindIndex; | ||
| 19 | private static Field m_indexIndex; | ||
| 20 | |||
| 21 | static { | ||
| 22 | try { | ||
| 23 | m_class = Class.forName("javassist.bytecode.MethodHandleInfo"); | ||
| 24 | m_kindIndex = m_class.getDeclaredField("refKind"); | ||
| 25 | m_kindIndex.setAccessible(true); | ||
| 26 | m_indexIndex = m_class.getDeclaredField("refIndex"); | ||
| 27 | m_indexIndex.setAccessible(true); | ||
| 28 | } catch (Exception ex) { | ||
| 29 | throw new Error(ex); | ||
| 30 | } | ||
| 31 | } | ||
| 32 | |||
| 33 | public static boolean isType(ConstInfoAccessor accessor) { | ||
| 34 | return m_class.isAssignableFrom(accessor.getItem().getClass()); | ||
| 35 | } | ||
| 36 | |||
| 37 | private Object m_item; | ||
| 38 | |||
| 39 | public MethodHandleInfoAccessor(Object item) { | ||
| 40 | m_item = item; | ||
| 41 | } | ||
| 42 | |||
| 43 | public int getTypeIndex() { | ||
| 44 | try { | ||
| 45 | return (Integer)m_kindIndex.get(m_item); | ||
| 46 | } catch (Exception ex) { | ||
| 47 | throw new Error(ex); | ||
| 48 | } | ||
| 49 | } | ||
| 50 | |||
| 51 | public void setTypeIndex(int val) { | ||
| 52 | try { | ||
| 53 | m_kindIndex.set(m_item, val); | ||
| 54 | } catch (Exception ex) { | ||
| 55 | throw new Error(ex); | ||
| 56 | } | ||
| 57 | } | ||
| 58 | |||
| 59 | public int getMethodRefIndex() { | ||
| 60 | try { | ||
| 61 | return (Integer)m_indexIndex.get(m_item); | ||
| 62 | } catch (Exception ex) { | ||
| 63 | throw new Error(ex); | ||
| 64 | } | ||
| 65 | } | ||
| 66 | |||
| 67 | public void setMethodRefIndex(int val) { | ||
| 68 | try { | ||
| 69 | m_indexIndex.set(m_item, val); | ||
| 70 | } catch (Exception ex) { | ||
| 71 | throw new Error(ex); | ||
| 72 | } | ||
| 73 | } | ||
| 74 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/accessors/MethodTypeInfoAccessor.java b/src/cuchaz/enigma/bytecode/accessors/MethodTypeInfoAccessor.java new file mode 100644 index 0000000..e151117 --- /dev/null +++ b/src/cuchaz/enigma/bytecode/accessors/MethodTypeInfoAccessor.java | |||
| @@ -0,0 +1,55 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode.accessors; | ||
| 12 | |||
| 13 | import java.lang.reflect.Field; | ||
| 14 | |||
| 15 | public class MethodTypeInfoAccessor { | ||
| 16 | |||
| 17 | private static Class<?> m_class; | ||
| 18 | private static Field m_descriptorIndex; | ||
| 19 | |||
| 20 | static { | ||
| 21 | try { | ||
| 22 | m_class = Class.forName("javassist.bytecode.MethodTypeInfo"); | ||
| 23 | m_descriptorIndex = m_class.getDeclaredField("descriptor"); | ||
| 24 | m_descriptorIndex.setAccessible(true); | ||
| 25 | } catch (Exception ex) { | ||
| 26 | throw new Error(ex); | ||
| 27 | } | ||
| 28 | } | ||
| 29 | |||
| 30 | public static boolean isType(ConstInfoAccessor accessor) { | ||
| 31 | return m_class.isAssignableFrom(accessor.getItem().getClass()); | ||
| 32 | } | ||
| 33 | |||
| 34 | private Object m_item; | ||
| 35 | |||
| 36 | public MethodTypeInfoAccessor(Object item) { | ||
| 37 | m_item = item; | ||
| 38 | } | ||
| 39 | |||
| 40 | public int getTypeIndex() { | ||
| 41 | try { | ||
| 42 | return (Integer)m_descriptorIndex.get(m_item); | ||
| 43 | } catch (Exception ex) { | ||
| 44 | throw new Error(ex); | ||
| 45 | } | ||
| 46 | } | ||
| 47 | |||
| 48 | public void setTypeIndex(int val) { | ||
| 49 | try { | ||
| 50 | m_descriptorIndex.set(m_item, val); | ||
| 51 | } catch (Exception ex) { | ||
| 52 | throw new Error(ex); | ||
| 53 | } | ||
| 54 | } | ||
| 55 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/accessors/NameAndTypeInfoAccessor.java b/src/cuchaz/enigma/bytecode/accessors/NameAndTypeInfoAccessor.java new file mode 100644 index 0000000..6e82f3e --- /dev/null +++ b/src/cuchaz/enigma/bytecode/accessors/NameAndTypeInfoAccessor.java | |||
| @@ -0,0 +1,74 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode.accessors; | ||
| 12 | |||
| 13 | import java.lang.reflect.Field; | ||
| 14 | |||
| 15 | public class NameAndTypeInfoAccessor { | ||
| 16 | |||
| 17 | private static Class<?> m_class; | ||
| 18 | private static Field m_nameIndex; | ||
| 19 | private static Field m_typeIndex; | ||
| 20 | |||
| 21 | static { | ||
| 22 | try { | ||
| 23 | m_class = Class.forName("javassist.bytecode.NameAndTypeInfo"); | ||
| 24 | m_nameIndex = m_class.getDeclaredField("memberName"); | ||
| 25 | m_nameIndex.setAccessible(true); | ||
| 26 | m_typeIndex = m_class.getDeclaredField("typeDescriptor"); | ||
| 27 | m_typeIndex.setAccessible(true); | ||
| 28 | } catch (Exception ex) { | ||
| 29 | throw new Error(ex); | ||
| 30 | } | ||
| 31 | } | ||
| 32 | |||
| 33 | public static boolean isType(ConstInfoAccessor accessor) { | ||
| 34 | return m_class.isAssignableFrom(accessor.getItem().getClass()); | ||
| 35 | } | ||
| 36 | |||
| 37 | private Object m_item; | ||
| 38 | |||
| 39 | public NameAndTypeInfoAccessor(Object item) { | ||
| 40 | m_item = item; | ||
| 41 | } | ||
| 42 | |||
| 43 | public int getNameIndex() { | ||
| 44 | try { | ||
| 45 | return (Integer)m_nameIndex.get(m_item); | ||
| 46 | } catch (Exception ex) { | ||
| 47 | throw new Error(ex); | ||
| 48 | } | ||
| 49 | } | ||
| 50 | |||
| 51 | public void setNameIndex(int val) { | ||
| 52 | try { | ||
| 53 | m_nameIndex.set(m_item, val); | ||
| 54 | } catch (Exception ex) { | ||
| 55 | throw new Error(ex); | ||
| 56 | } | ||
| 57 | } | ||
| 58 | |||
| 59 | public int getTypeIndex() { | ||
| 60 | try { | ||
| 61 | return (Integer)m_typeIndex.get(m_item); | ||
| 62 | } catch (Exception ex) { | ||
| 63 | throw new Error(ex); | ||
| 64 | } | ||
| 65 | } | ||
| 66 | |||
| 67 | public void setTypeIndex(int val) { | ||
| 68 | try { | ||
| 69 | m_typeIndex.set(m_item, val); | ||
| 70 | } catch (Exception ex) { | ||
| 71 | throw new Error(ex); | ||
| 72 | } | ||
| 73 | } | ||
| 74 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/accessors/StringInfoAccessor.java b/src/cuchaz/enigma/bytecode/accessors/StringInfoAccessor.java new file mode 100644 index 0000000..6665ffe --- /dev/null +++ b/src/cuchaz/enigma/bytecode/accessors/StringInfoAccessor.java | |||
| @@ -0,0 +1,55 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode.accessors; | ||
| 12 | |||
| 13 | import java.lang.reflect.Field; | ||
| 14 | |||
| 15 | public class StringInfoAccessor { | ||
| 16 | |||
| 17 | private static Class<?> m_class; | ||
| 18 | private static Field m_stringIndex; | ||
| 19 | |||
| 20 | static { | ||
| 21 | try { | ||
| 22 | m_class = Class.forName("javassist.bytecode.StringInfo"); | ||
| 23 | m_stringIndex = m_class.getDeclaredField("string"); | ||
| 24 | m_stringIndex.setAccessible(true); | ||
| 25 | } catch (Exception ex) { | ||
| 26 | throw new Error(ex); | ||
| 27 | } | ||
| 28 | } | ||
| 29 | |||
| 30 | public static boolean isType(ConstInfoAccessor accessor) { | ||
| 31 | return m_class.isAssignableFrom(accessor.getItem().getClass()); | ||
| 32 | } | ||
| 33 | |||
| 34 | private Object m_item; | ||
| 35 | |||
| 36 | public StringInfoAccessor(Object item) { | ||
| 37 | m_item = item; | ||
| 38 | } | ||
| 39 | |||
| 40 | public int getStringIndex() { | ||
| 41 | try { | ||
| 42 | return (Integer)m_stringIndex.get(m_item); | ||
| 43 | } catch (Exception ex) { | ||
| 44 | throw new Error(ex); | ||
| 45 | } | ||
| 46 | } | ||
| 47 | |||
| 48 | public void setStringIndex(int val) { | ||
| 49 | try { | ||
| 50 | m_stringIndex.set(m_item, val); | ||
| 51 | } catch (Exception ex) { | ||
| 52 | throw new Error(ex); | ||
| 53 | } | ||
| 54 | } | ||
| 55 | } | ||
diff --git a/src/cuchaz/enigma/bytecode/accessors/Utf8InfoAccessor.java b/src/cuchaz/enigma/bytecode/accessors/Utf8InfoAccessor.java new file mode 100644 index 0000000..2abf60b --- /dev/null +++ b/src/cuchaz/enigma/bytecode/accessors/Utf8InfoAccessor.java | |||
| @@ -0,0 +1,28 @@ | |||
| 1 | /******************************************************************************* | ||
| 2 | * Copyright (c) 2014 Jeff Martin. | ||
| 3 | * All rights reserved. This program and the accompanying materials | ||
| 4 | * are made available under the terms of the GNU Public License v3.0 | ||
| 5 | * which accompanies this distribution, and is available at | ||
| 6 | * http://www.gnu.org/licenses/gpl.html | ||
| 7 | * | ||
| 8 | * Contributors: | ||
| 9 | * Jeff Martin - initial API and implementation | ||
| 10 | ******************************************************************************/ | ||
| 11 | package cuchaz.enigma.bytecode.accessors; | ||
| 12 | |||
| 13 | public class Utf8InfoAccessor { | ||
| 14 | |||
| 15 | private static Class<?> m_class; | ||
| 16 | |||
| 17 | static { | ||
| 18 | try { | ||
| 19 | m_class = Class.forName("javassist.bytecode.Utf8Info"); | ||
| 20 | } catch (Exception ex) { | ||
| 21 | throw new Error(ex); | ||
| 22 | } | ||
| 23 | } | ||
| 24 | |||
| 25 | public static boolean isType(ConstInfoAccessor accessor) { | ||
| 26 | return m_class.isAssignableFrom(accessor.getItem().getClass()); | ||
| 27 | } | ||
| 28 | } | ||