ASM:
操作Java字节码的框架,按照Class文件的格式,解析、修改、生成Class,可以动态生成类或者增强现有类的功能。
字节码插桩的具体实现:
a.添加依赖:
implementation 'org.ow2.asm:asm:7.1' implementation 'org.ow2.asm:asm-commons:7.1'
b.对某个class文件进行插桩测试代码:
public void test() throws IOException { /** * 1、获得待插桩的字节码数据,某个.class文件 */ FileInputStream fis = new FileInputStream(fisPath); /** * 2、执行插桩 修改class数据 */ ClassReader cr = new ClassReader(fis); ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS); //ClassVisitor: 类访问者,接收类信息的回调 //MethodVisitor: 方法访问者,操作方法体 cr.accept(new ClassVisitor(Opcodes.ASM7, cw) { //其实在classVisitor方法内,还有其它的继承方法供访问外部类、内部类或属性 //在解析到一个方法时候回调一次 @Override public MethodVisitor visitMethod(int access, String name, String descriptor, String signature, String[] exceptions) { //判断方法是否为main方法,我的.class文件中是一个Java的main()方法 if (name.equals("main")) { // 执行插桩... MethodVisitor mv = super.visitMethod(access, name, descriptor, signature, exceptions); return new MyMethodVisitor(api, mv, access, name, descriptor); } else { return super.visitMethod(access, name, descriptor, signature, exceptions); } } @Override public FieldVisitor visitField(int access, String name, String descriptor, String signature, Object value) { return super.visitField(access, name, descriptor, signature, value); } }, ClassReader.EXPAND_FRAMES); /** * 3、输出结果 */ byte[] bytes = cw.toByteArray(); //写到另一个位置存放插桩后的class文件 FileOutputStream fos = new FileOutputStream(fosPath); fos.write(bytes); fos.close(); System.out.println("======================="); }
把要修改的代码修改成功后,再进行编译,拿到.class文件并用ASMPlugin插件查看其字节码:
// access flags 0x9 public static main([Ljava/lang/String;)V throws java/lang/InterruptedException L0 LINENUMBER 8 L0 //需要插入 INVOKESTATIC com/enjoy/asminject/AppMethodBeta.i ()V L1 LINENUMBER 9 L1 //需要插入 GETSTATIC java/lang/System.out : Ljava/io/PrintStream; //需要插入 LDC "Lance!!!!" //需要插入 INVOKEVIRTUAL java/io/PrintStream.println (Ljava/lang/String;)V L2 LINENUMBER 10 L2 INVOKESTATIC com/enjoy/asminject/AppMethodBeta.i ()V L3 LINENUMBER 11 L3 GETSTATIC java/lang/System.out : Ljava/io/PrintStream; LDC "\u6267\u884c\u4e1a\u52a1...." INVOKEVIRTUAL java/io/PrintStream.println (Ljava/lang/String;)V L4 LINENUMBER 12 L4 INVOKESTATIC com/enjoy/asminject/AppMethodBeta.o ()V L5 LINENUMBER 13 L5 //需要插入 INVOKESTATIC com/enjoy/asminject/AppMethodBeta.o ()V L6 LINENUMBER 14 L6 RETURN L7 LOCALVARIABLE args [Ljava/lang/String; L0 L7 0 MAXSTACK = 2 MAXLOCALS = 1 }
可以查看上面的字节码编写插桩的代码:
class MyMethodVisitor extends AdviceAdapter { protected MyMethodVisitor(int api, MethodVisitor methodVisitor, int access, String name, String descriptor) { super(api, methodVisitor, access, name, descriptor); } //修改system.out.println("开始执行")为system.out.println("Modified print statement") //"开始执行"编译后在.class文件里为"鎵ц\ue511涓氬姟....",其在.class文件中是 //LDC "\u93b5\u0446\ue511\u6d93\u6c2c\u59df....",应该就是访问LdcInsn @Override public void visitLdcInsn(Object value) { // 检查是否为指定字符串常量 if (value instanceof String && ((String) value).equals("鎵ц\ue511涓氬姟....")) { // 修改为新的字符串常量 super.visitLdcInsn("Modified print statement"); return; } super.visitLdcInsn(value); } // 进入方法 回调;在main()方法里最前面插入 @Override protected void onMethodEnter() { super.onMethodEnter(); //插入代码:AppMethodBeta.i(); invokeStatic(Type.getType("Lcom/enjoy/asminject/AppMethodBeta;"), new Method("i", "()V")); //插入代码:System.out.println(lance!!!!); getStatic(Type.getType("Ljava/lang/System;"), "out", Type.getType("Ljava/io/PrintStream;")); visitLdcInsn("Lance!!!!"); invokeVirtual(Type.getType("Ljava/io/PrintStream;"), new Method("println", "(Ljava/lang/String;)V")); } /** * 退出方法,在main()方法里的最后插入 * * @param opcode */ @Override protected void onMethodExit(int opcode) { super.onMethodExit(opcode); // 插入代码:AppMethodBeta.o(); invokeStatic(Type.getType("Lcom/enjoy/asminject/AppMethodBeta;"), new Method("o", "()V")); } }
c.最后得到的.class文件打开显示:
public static void main(String[] var0) throws InterruptedException { AppMethodBeta.i(); System.out.println("Lance!!!!"); AppMethodBeta.i(); System.out.println("Modified print statement"); AppMethodBeta.o(); AppMethodBeta.o(); }
d.之前未插桩的代码:
public static void main(String[] args) throws InterruptedException { AppMethodBeta.i(); System.out.println("执行业务...."); AppMethodBeta.o(); }