javassist 使用笔记

javassist

Javassist 是一个开源的分析、编辑和创建Java字节码的类库。其主要的优点，在于简单，而且快速。直接使用 java 编码的形式，而不需要了解虚拟机指令，就能动态改变类的结构，或者动态生成类。

• ClassPool：一个基于 Hashtable 实现的 CtClass 对象容器，其中键是类名称，值是表示该类的 CtClass 对象。
• CtClass：CtClass 表示类，一个 CtClass (编译时类）对象可以处理一个 class 文件，这些 CtClass 对象可以从 ClassPool 获得。
• CtMethods：表示类中的方法。
• CtFields ：表示类中的字段。

ClassPool

获取 classpool 对象

// 获取 ClassPool 对象，使用系统默认类路径
ClassPool pool = new ClassPool(true);
// 效果与 new ClassPool(true) 一致
ClassPool pool1 = ClassPool.getDefault();

获取类

// 通过类名获取 CtClass，未找到会抛出异常
CtClass ctClass = pool.get("org.test.demo.DemoService");
// 通过类名获取 CtClass，未找到返回 null，不会抛出异常
CtClass ctClass1 = pool.getOrNull("org.test.demo.DemoService");

创建类

// 复制一个类，创建一个新类
CtClass ctClass2 = pool.getAndRename("org.test.demo.DemoService", "org.test.demo.DemoCopyService");
// 通过类名，创建一个新类
CtClass ctClass3 = pool.makeClass("org.test.demo.NewDemoService");
// 通过文件流，创建一个新类，注意文件必须是编译后的 class 文件，不是源代码文件。
CtClass ctClass4 = pool.makeClass(new FileInputStream(new File("./customize/DemoBeforeHandler.class")));

添加类搜索路径

// 将类搜索路径插入到搜索路径之前
pool.insertClassPath(new ClassClassPath(this.getClass()));
// 将类搜索路径添加到搜索路径之后
pool.appendClassPath(new ClassClassPath(this.getClass()));
// 将一个目录作为类搜索路径
pool.insertClassPath("/usr/local/javalib");

CtClass

public static void main(String[] args) throws Exception {

    ClassPool pool = ClassPool.getDefault();
    CtClass ctClass = pool.get("javassist.test02.Person");
    //类名
    String name = ctClass.getName();
    //包名
    String packageName = ctClass.getPackageName();
    //父类
    CtClass superclass = ctClass.getSuperclass();
    //接口
    CtClass[] interfaces = ctClass.getInterfaces();

    System.out.println(name);
    System.out.println(packageName);
    System.out.println(superclass.getName());
    System.out.println(interfaces[0].getName());
}

CtMethod

// 在方法体前插入代码块
ctMethod.insertBefore("");
// 在方法体后插入代码块
ctMethod.insertAfter("");
// 在某行 字节码 后插入代码块
ctMethod.insertAt(10, "");
// 添加参数
ctMethod.addParameter(CtClass);
// 设置方法名
ctMethod.setName("newName");
// 设置方法体
ctMethod.setBody("System.out.println(123);");

创建新类和调用

package javassist.test;

import javassist.*;

import java.lang.reflect.Constructor;
import java.lang.reflect.Method;

/**
 * Created by ssr on 2020/9/15.
 */
public class JavassistTest01 {
    public static void main(String[] args) throws Exception{

        // 获取 ClassPool 对象，使用系统默认类路径
        ClassPool classPool = ClassPool.getDefault();
        // 创建一个空类
        CtClass ctClass = classPool.makeClass("Test");

        // 新增 String 类型字段 name
        CtField name = new CtField(classPool.get(String.class.getName()), "name", ctClass);
        // 新增 int 类型字段 age
        CtField age = new CtField(classPool.get(int.class.getName()), "age", ctClass);

        // 设置修饰符 private
        name.setModifiers(Modifier.PRIVATE);
        age.setModifiers(Modifier.PRIVATE);

        //添加到类里
        ctClass.addField(name);
        ctClass.addField(age);

        // 无参构造方法
        CtConstructor ctConstructor = new CtConstructor(new CtClass[]{},ctClass);
        // 有参构造方法
        CtConstructor ctConstructor1 = new CtConstructor(new CtClass[]{classPool.get(String.class.getName()), CtClass.intType}, ctClass);

        // 方法体
        ctConstructor.setBody("{name="test";age=12;}");
        // 方法体  $0 代表this  $1 $2 方法参数 name age
        ctConstructor1.setBody("{$0.name = $1;$0.age = $2;}");

        ctClass.addConstructor(ctConstructor);
        ctClass.addConstructor(ctConstructor1);

        //创建 getter  setter 方法
        CtMethod setName = CtNewMethod.setter("setName", name);
        CtMethod getName = CtNewMethod.getter("getName", name);

        CtMethod getAge = CtNewMethod.getter("getAge", age);
        CtMethod setAge = CtNewMethod.setter("setAge", age);

        ctClass.addMethod(setName);
        ctClass.addMethod(getName);

        ctClass.addMethod(getAge);
        ctClass.addMethod(setAge);

        //新增方法
        CtMethod printName = new CtMethod(new CtClass(String.class.getName()) {
            @Override
            public String toString() {
                return super.toString();
            }
        }, "printInfo", new CtClass[]{}, ctClass);

        // 设置方法修饰符 public
        printName.setModifiers(Modifier.PUBLIC);
        // 设置方法体
        printName.setBody("{return "my name is "+name+"," + "age is "+age;}");
        ctClass.addMethod(printName);

        // 写入文件
        ctClass.writeFile("D:\漏洞分析\Commons Collections 3.1\src\main\java\javassist\test");

        //将创建的ctClass加载至当前线程的上下文类加载器中
        Class clz = ctClass.toClass();

        //反射调用
        Constructor declaredConstructor = clz.getDeclaredConstructor(String.class, int.class);
        Object obj = declaredConstructor.newInstance("liangzi", 100);

        Method printInfo = clz.getMethod("printInfo");
        System.out.println(printInfo.invoke(obj));


    }
}

修改类

• 原始类

package javassist.test03;

public class Test {
    private void info(String name){
        System.out.println(name);
    }

}

• 修改

package javassist.test03;

import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.Modifier;

import java.lang.reflect.Method;

public class JavassistTest01 {

    public static void main(String[] args) throws Exception{

        ClassPool pool = ClassPool.getDefault();
        // 通过类名获取 CtClass，未找到会抛出异常
        CtClass ctClass = pool.get("javassist.test03.Test");
        // 获取类中的info方法
        CtMethod info = ctClass.getDeclaredMethod("info");
        // 修改info方法修饰符为 public
        info.setModifiers(Modifier.PUBLIC);
        // 方法开头添加语句
        info.insertBefore("{System.out.println("000");}");
        // 方法结尾添加语句
        info.insertAfter("{System.out.println("111");}");

        Class cls = ctClass.toClass();
        Object o = cls.newInstance();
        Method info1 = cls.getMethod("info", String.class);
        info1.invoke(o,"bbb");
    }
}