Java ASM学习

时间：2020-04-23 19:07:28 阅读：78 评论：0 收藏：0 [点我收藏+]

标签：ide rri exception 编写 row get 优势 adapter 部分

ASM也是字节码编辑库，如果我们的目的仅仅是为目标类添加某些功能，也可以考虑动态代理，但是动态代理是面向接口的，因为proxy.newinstance实际上是对某个接口定义一个invocaionHandler，那么这样限制就比较大，并且对代理的每一次函数调用都将被invocationHandler处理，加上handlder中反射的应用，因此动态代理整体来说和直接改变目标class的内部结构来说性能并没有太多优势

ASM采用树这种结构来描述字节码，通过push模型（访问者模式）遍历树的过程中对字节码进行修改,ASM提供classReader可以从字节数组或者class文件中去获得字节码，如何访问字节码中所表示的类结构：

通过调用该类的accept方法传入一个classVisitor的实例来进行class字节码的访问，另一个参数就是解析选项，定义在解析class过程中是否跳过某些区域的解析

技术图片

接着在accept方法中调用传入的classsVisitor接口的各个方法，把字节码中不同的区域想成树上不同的位置，每个位置都有对应的vistor，我们只需要提供不同的vistor就能访问字节码不同偏移位置的所实际代表的成员变量、方法、修饰符等，如下图所示第一步将解析class源码技术图片

比如可以用ClassVistor、AnnotationVistor、FieldVistor、MethodVistor(都是抽象类)对不同区域做处理，classReader解析到不同位置时，将自动调用这些vistor，这里接用绿盟的一张字节码解析流程：

技术图片

其中通过classAdaptor类实现Classvistor中定义的函数，因为处理class是有顺序的，因此在声明classAdaptor时传入下一个访问区域的vistor，这里把这种机制成为责任链，所以定义责任链的时候从后往前声明

比如：

ClassWriter  classWriter = new ClassWriter(ClassWriter.COMPUTE_MAXS); 
ClassAdaptor delLoginClassAdaptor = new DelLoginClassAdapter(classWriter); 
ClassAdaptor accessClassAdaptor = new AccessClassAdaptor(delLoginClassAdaptor); 
ClassReader classReader = new ClassReader(strFileName); 
classReader.accept(accessClassAdapter, ClassReader.SKIP_DEBUG);

classWrite作为责任链的最后一部分，其中每一步的ClassAdaptor都是链接起来的，asm提供的toByteArray就能转为字节数组存入class文件实现hotspot或者直接返回(结合插桩agentmain)，如下图的时序图，从左到右，就可以看到第一步是ClassReader->ClassReader.accept(av)->av(av是个classAdaptor，接着依次读取class字节码文件所构成的那颗树的不同区域，我们复写了哪个访问方法就调用我们的方法去访问某个区域)->cw(责任链最后一步份)，整个这样设计的话我们就不用去管class字节码文件到底是某个偏移具体是什么含义，只需要根据需要用ClassAdaptor中来重写ClassVistor的某个方法即可

技术图片

people.java

package asm;

public class people {
    public void eat(){
        System.out.println("i like eat");
    }

    public static void main(String[] args) {
        people w = new people();
        w.eat();
    }
}

drink.java

package asm;
public class drink {
    public static void appale(){
        System.out.println("add appale");
    }
}

如果目前需要在eat方法中添加调用drink.appale的话，则需要修改peope.class对应的字节码，并到eat对应的方法区修改其方法块代码，所以根据上面的学习，要继承ClassAdaptor重写其visitMethod方法

技术图片

其返回的是一个MethodVisitor，默认情况下这里调用this.cv的visitMethod，那么这里的this.cv就是传入的信任链的下一个节点，即传入的ClassWriter(ClassWriter也是继承自ClassVistor)

asmtest.java

package asm;

import org.objectweb.asm.ClassAdapter;
import org.objectweb.asm.ClassReader;
import org.objectweb.asm.ClassWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;

public class asmTest {
    public static void main(String[] args) throws IOException {
    FileInputStream fi = new FileInputStream(System.getProperty("user.dir")+"/target/classes/asm/people.class");
    byte[] fib = new byte[fi.available()];
    fi.read(fib);
    ClassReader clar = new ClassReader(fib);
    ClassWriter claw =  new ClassWriter(ClassWriter.COMPUTE_MAXS);
    ClassAdapter clap = new asmadd(claw);
    clar.accept(clap,ClassReader.SKIP_DEBUG);
    byte[] fo = claw.toByteArray();
    File file;
    file = new File("people1.class");
    FileOutputStream fof = new FileOutputStream(file);
    fof.write(fo);
    fof.close();

    }
}

那么原始的this.cv.visitMethod实际是来自ClassVisitor，先看该方法的作用即访问一个类的方法区，该方法返回一个CodeVistor实例或null（返回null，即当前访问者并不想进一步访问该方法区的具体代码块）,这里返回的必须是之前未返回的visitor，因此这样规定说明实际上责任链逻辑上是一定顺序的。access：方法的访问标识符(public protected private etc.)，以及当前方法区的名字(有了这个就能精确的匹配并修改某个方法了)，desc即方法描述符即返回值类型包括其参数类型都能进行精准匹配，还有exceptions异常和attrs

技术图片

所以目前就要定义我们自己的ClassAdaptor去匹配eat方法，然后进一步访问其代码块，此时访问代码块使用MethodAdaptor是实现了MethodVistor接口，使用该Adaptor就能重写visitCode在eat方法中添加代码块

技术图片

由其构造方法看所以此时传入的也应该是MethodVistor，即已经匹配到的方法，把其当作树中的某一个父节点，那么此时进入其子节点，通过MethodAdaptor的visitcode就可以访问方法体，所以在这里面加入调用drink.apple的代码即可

技术图片

asmadd.java

package asm;
import org.objectweb.asm.ClassAdapter;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.MethodVisitor;
public class asmadd extends ClassAdapter {
    public asmadd(ClassVisitor cv) {
        super(cv);
    }
    @Override
    public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
        MethodVisitor mv = this.cv.visitMethod(access, name, desc, signature, exceptions);
        MethodVisitor mvwrapper = mv;
        if(name.equals("eat")){
             mvwrapper = new methodWrapper(mv);
        }
        return mvwrapper;
    }
}

methodWrapper.java

package asm;

import org.objectweb.asm.MethodAdapter;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;

public class methodWrapper extends MethodAdapter {
    public methodWrapper(MethodVisitor mv) {
        super(mv);
    }
    @Override
    public void visitCode(){
    visitMethodInsn(Opcodes.INVOKESTATIC,"drink","eat","()V");
    }
}

技术图片

植入代码块用的api为visitMethodInsn，即调用某个类的某个方法，比如这里调用的类名为drink，方法名为eat，描述符为void，opcode为invokestatic

技术图片

因为此时desc是个字符串，因此在Type类中就定义了如何去扫描给定的字符串来判断目标方法的参数类型以及返回值类型技术图片

跟一下整个过程：

技术图片

首先加载classReader，先从appclassloader到extclassloader中找，再从extclassloader到bootstrapclassloader中找，boot没找到，ext没找到，则到app中findclass，利用URLClasspath中存储的jar包来进行查找，这里就包括maven中引入的class文件，找到之后就defineclass来生成一个class类型的实例了，最后再将其放入保护域中，后面几个new依然进行相同操作，

技术图片