LiteLoader / LiteLoader · Commits

Commit a6a6084a00b237a9539ff52c653587b25aec2570

Authored by Mumfrey
1 parent 1bfcfd90

add new convenience function replaceConstructors to ByteCodeUtilities

Inline Side-by-side
Showing 1 changed file with 51 additions and 11 deletions
java/common/com/mumfrey/liteloader/transformers/ByteCodeUtilities.java
  View file @a6a6084
... ... @@ -9,22 +9,15 @@ import java.util.Map;
9 9  
10 10 import org.objectweb.asm.Opcodes;
11 11 import org.objectweb.asm.Type;
12   -import org.objectweb.asm.tree.AbstractInsnNode;
13   -import org.objectweb.asm.tree.AnnotationNode;
14   -import org.objectweb.asm.tree.ClassNode;
15   -import org.objectweb.asm.tree.FieldNode;
16   -import org.objectweb.asm.tree.FrameNode;
17   -import org.objectweb.asm.tree.InsnList;
18   -import org.objectweb.asm.tree.LabelNode;
19   -import org.objectweb.asm.tree.LocalVariableNode;
20   -import org.objectweb.asm.tree.MethodNode;
21   -import org.objectweb.asm.tree.VarInsnNode;
  12 +import org.objectweb.asm.tree.*;
22 13 import org.objectweb.asm.tree.analysis.Analyzer;
23 14 import org.objectweb.asm.tree.analysis.AnalyzerException;
24 15 import org.objectweb.asm.tree.analysis.BasicValue;
25 16 import org.objectweb.asm.tree.analysis.Frame;
26 17 import org.objectweb.asm.tree.analysis.SimpleVerifier;
27 18  
  19 +import com.mumfrey.liteloader.core.runtime.Obf;
  20 +
28 21 /**
29 22 * Utility methods for working with bytecode using ASM
30 23 *
... ... @@ -35,7 +28,54 @@ public abstract class ByteCodeUtilities
35 28 private static Map<String, List<LocalVariableNode>> calculatedLocalVariables = new HashMap<String, List<LocalVariableNode>>();
36 29  
37 30 private ByteCodeUtilities() {}
38   -
  31 +
  32 + /**
  33 + * Replace all constructor invokations for the target class in the supplied classNode with invokations of the replacement class
  34 + *
  35 + * @param classNode Class to search in
  36 + * @param target Target type
  37 + * @param replacement Replacement type
  38 + */
  39 + public static void replaceConstructors(ClassNode classNode, Obf target, Obf replacement)
  40 + {
  41 + for (MethodNode method : classNode.methods)
  42 + {
  43 + ByteCodeUtilities.replaceConstructors(method, target, replacement);
  44 + }
  45 + }
  46 +
  47 + /**
  48 + * Replace all constructor invokations for the target class in the supplied method with invokations of the replacement class
  49 + *
  50 + * @param method Method to look in
  51 + * @param target Target type
  52 + * @param replacement Replacement type
  53 + */
  54 + public static void replaceConstructors(MethodNode method, Obf target, Obf replacement)
  55 + {
  56 + Iterator<AbstractInsnNode> iter = method.instructions.iterator();
  57 + while (iter.hasNext())
  58 + {
  59 + AbstractInsnNode insn = iter.next();
  60 + if (insn.getOpcode() == Opcodes.NEW)
  61 + {
  62 + TypeInsnNode typeInsn = (TypeInsnNode)insn;
  63 + if (target.obf.equals(typeInsn.desc) || target.name.equals(typeInsn.desc))
  64 + {
  65 + typeInsn.desc = replacement.name;
  66 + }
  67 + }
  68 + else if (insn instanceof MethodInsnNode && insn.getOpcode() == Opcodes.INVOKESPECIAL)
  69 + {
  70 + MethodInsnNode methodInsn = (MethodInsnNode)insn;
  71 + if ((target.obf.equals(methodInsn.owner) || target.name.equals(methodInsn.owner)) && "<init>".equals(methodInsn.name))
  72 + {
  73 + methodInsn.owner = replacement.name;
  74 + }
  75 + }
  76 + }
  77 + }
  78 +
39 79 /**
40 80 * Injects appropriate LOAD opcodes into the supplied InsnList appropriate for each entry in the args array starting at pos
41 81 *
... ...