ASM插桩修改代码

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();
    }