字节码插桩(三): ASM 字节码插桩(1)

class MkAnalyticsClassVisitor extends ClassVisitor implements Opcodes {
   private final static String SDK_API_CLASS = "com/github/microkibaco/asm_sdk/SensorsDataAutoTrackHelper"
   private String[] mInterfaces
   private ClassVisitor classVisitor
   private HashMap<String, MkAnalyticsMethodCell> mLambdaMethodCells = new HashMap<>()
   MkAnalyticsClassVisitor(final ClassVisitor classVisitor) {
       super(Opcodes.ASM6, classVisitor)
       this.classVisitor = classVisitor
   }
   private
   static void visitMethodWithLoadedParams(MethodVisitor methodVisitor, int opcode, String owner, String methodName, String methodDesc, int start, int count, List<Integer> paramOpcodes) {
       for (int i = start; i < start + count; i++) {
           methodVisitor.visitVarInsn(paramOpcodes[i - start], i)
       }
       methodVisitor.visitMethodInsn(opcode, owner, methodName, methodDesc, false)
   }
   /**
    *  visit 可以拿到关于 .class 的所有信息,比如: 当前类所实现的接口列表
    * @param version JDK的版本
    * @param access 类的修饰符
    * @param name 类的名称
    * @param signature 当前类所继承的父类
    * @param superName
    * @param interfaces 类所实现的接口列表,在java中,一个类是可以实现多个不同的接口,因此该参数是一个数组类型
    */
   @Override
   void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
       super.visit(version, access, name, signature, superName, interfaces)
       mInterfaces = interfaces
   }
   /**
    * 可以拿到关于 method所有的信息,比如: 方法名 方法的参数描述
    * @param access 类的修饰符
    * @param name 类的名称
    * @param desc 方法签名
    * @param signature 类签名
    * @param exceptions 异常信息
    * @return MethodVisitor
    */
   @Override
   MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
       MethodVisitor methodVisitor = super.visitMethod(access, name, desc, signature, exceptions)
       return methodVisitor
   }
   /**
    * 获取方法参数下标为 index 的对应 ASM index
    * @param types 方法参数类型数组
    * @param index 方法中参数下标，从 0 开始
    * @param isStaticMethod 该方法是否为静态方法
    * @return 访问该方法的 index 位参数的 ASM index
    */
   int getVisitPosition(Type[] types, int index, boolean isStaticMethod) {
       if (types == null || index < 0 || index >= types.length) {
           throw new Error("getVisitPosition error")
       }
       if (index == 0) {
           return isStaticMethod ? 0 : 1
       } else {
           return getVisitPosition(types, index - 1, isStaticMethod) + types[index - 1].getSize()
       }
   }
}

methodVisitor = new MkAnalyticsDefaultMethodVisitor(methodVisitor, access, name, desc) {
           boolean isSensorsDataTrackViewOnClickAnnotation = false
           /**
            * 退出方法前可以退出字节码
            */
           @Override
           void visitEnd() {
               super.visitEnd()
               if (mLambdaMethodCells.containsKey(nameDesc)) {
                   mLambdaMethodCells.remove(nameDesc)
               }
           }
           /**
            * 进入方法时插入字节码
            */
           @Override
           protected void onMethodEnter() {
               super.onMethodEnter()
               }
/**
* 可以在这里通过注解的方式操作字节码
* @param s 访问的注解名
* @param b 是否方法
* @return  methodVisitor
*/
           @Override
           groovyjarjarasm.asm.AnnotationVisitor visitAnnotation(String s, boolean b) {
               if (s == 'Lcom/sensorsdata/analytics/android/sdk/SensorsDataTrackViewOnClick;') {
                   isSensorsDataTrackViewOnClickAnnotation = true
               }
               return super.visitAnnotation(s, b)
           }
       }

class MkAnalyticsDefaultMethodVisitor extends AdviceAdapter {
   MkAnalyticsDefaultMethodVisitor(MethodVisitor mv, int access, String name, String desc) {
       super(Opcodes.ASM6, mv, access, name, desc)
   }
   /**
    * 表示 ASM 开始扫描这个方法
    */
   @Override
   void visitCode() {
       super.visitCode()
   }
   @Override
   void visitMethodInsn(int opcode, String owner, String name, String desc) {
       super.visitMethodInsn(opcode, owner, name, desc)
   }
   @Override
   void visitAttribute(Attribute attribute) {
       super.visitAttribute(attribute)
   }
   /**
    * 表示方法输出完毕
    */
   @Override
   void visitEnd() {
       super.visitEnd()
   }
   @Override
   void visitFieldInsn(int opcode, String owner, String name, String desc) {
       super.visitFieldInsn(opcode, owner, name, desc)
   }
   @Override
   void visitIincInsn(int var, int increment) {
       super.visitIincInsn(var, increment)
   }
   @Override
   void visitIntInsn(int i, int i1) {
       super.visitIntInsn(i, i1)
   }
   /**
    * 该方法是 visitEnd 之前调用的方法，可以反复调用。用以确定类方法在执行时候的堆栈大小。
    * @param maxStack
    * @param maxLocals
    */
   @Override
   void visitMaxs(int maxStack, int maxLocals) {
       super.visitMaxs(maxStack, maxLocals)
   }
   @Override
   void visitVarInsn(int opcode, int var) {
       super.visitVarInsn(opcode, var)
   }
   @Override
   void visitJumpInsn(int opcode, Label label) {
       super.visitJumpInsn(opcode, label)
   }
   @Override
   void visitLookupSwitchInsn(Label label, int[] ints, Label[] labels) {
       super.visitLookupSwitchInsn(label, ints, labels)
   }
   @Override
   void visitMultiANewArrayInsn(String s, int i) {
       super.visitMultiANewArrayInsn(s, i)
   }
   @Override
   void visitTableSwitchInsn(int i, int i1, Label label, Label[] labels) {
       super.visitTableSwitchInsn(i, i1, label, labels)
   }
   @Override
   void visitTryCatchBlock(Label label, Label label1, Label label2, String s) {
       super.visitTryCatchBlock(label, label1, label2, s)
   }
   @Override
   void visitTypeInsn(int opcode, String s) {
       super.visitTypeInsn(opcode, s)
   }
   @Override
   void visitLocalVariable(String s, String s1, String s2, Label label, Label label1, int i) {
       super.visitLocalVariable(s, s1, s2, label, label1, i)
   }
   @Override
   void visitInsn(int opcode) {
       super.visitInsn(opcode)
   }
   @Override
   AnnotationVisitor visitAnnotation(String s, boolean b) {
       return super.visitAnnotation(s, b)
   }
   @Override
   protected void onMethodEnter() {
       super.onMethodEnter()
   }
   /**
    * 使用 onMethodExit 这样就不会影响到应用程序原有点击事件的响应速度
    * @param opcode
    */
   @Override
   protected void onMethodExit(int opcode) {
       super.onMethodExit(opcode)
   }
}