java插桩-javaassist_插桩 java

│   ├── main
│   └── test
└── tutorial
    ├── brown.css
    ├── tutorial2.html
    ├── tutorial3.html
    └── tutorial.html

• sample  样例
• src         源代码
• tutorial   教程
• Readme.html 自述文件。
• License.html  许可证文件
• tutorial 教程目录
• javassist.jar  jar文件（类文件）
• src        源代码目录

源代码级别

ClassPool

CtClass对象的容器。

1. getDefault (): 返回默认的ClassPool ，单例模式，一般通过该方法创建我们的ClassPool；
2. appendClassPath(ClassPath cp), insertClassPath(ClassPath cp) : 将一个ClassPath加到类搜索路径的末尾位置或插入到起始位置。通常通过该方法写入额外的类搜索路径，以解决多个类加载器环境中找不到类问题；
3. importPackage(String packageName):导入包；
4. makeClass(String classname):创建一个空类，没有变量和方法，后序通过CtClass的函数进行添加；
5. get(String classname)、getCtClass(String classname) : 根据类路径名获取该类的CtClass对象，用于后续的编辑。

CtClass

类

1. debugDump;String类型，如果生成。class文件，保存在这个目录下。
2. setName(String name):给类重命名;
3. setSuperclass(CtClass clazz):设置父类;
4. addField(CtField f, Initializer init):添加字段（属性），初始值见CtField；
5. addMethod(CtMethod m):添加方法（函数）；
6. toBytecode(): 返回修改后的字节码。需要注意的是一旦调用该方法，则无法继续修改CtClass；
7. toClass(): 将修改后的CtClass加载至当前线程的上下文类加载器中，CtClass的toClass方法是通过调用本方法实现。需要注意的是一旦调用该方法，则无法继续修改已经被加载的CtClass；
8. writeFile(String directoryName):根据CtClass生成 .class 文件；
9. defrost():解冻类，用于使用了toclass()、toBytecode、writeFile()，类已经被JVM加载，Javassist冻结CtClass后；
10. detach():避免内存溢出，从ClassPool中移除一些不需要的CtClass。

Loader

类加载器

1. loadClass(String name):加载类

CtField

字段

1. CtField(CtClass type, String name, CtClass declaring) :构造函数，添加字段类型，名称，所属的类;
2. CtField.Initializer constant():CtClass使用addField时初始值的设置;
3. setModifiers(int mod):设置访问级别，一般使用Modifier调用常量。

CtMethod

方法

1. insertBefore(String src):在方法的起始位置插入代码；
2. insertAfter(String src):在方法的所有 return 语句前插入代码以确保语句能够被执行，除非遇到exception；
3. insertAt(int lineNum, String src):在指定的位置插入代码；
4. addCatch(String src, CtClass exceptionType):将方法内语句作为try的代码块，插入catch代码块src;
5. setBody(String src):将方法的内容设置为要写入的代码，当方法被 abstract修饰时，该修饰符被移除；
6. setModifiers(int mod):设置访问级别，一般使用Modifier调用常量;
7. invoke(Object obj, Object… args):反射调用字节码生成类的方法。

对于setBody $0代表this $1、$2、…代表方法的第几个参数

setBody("{$0.name = $1;}");

$符号含义

$符号含义

符号	含义
$0, $1, $2, …	this,第几个参数
$args	参数列表. $args的类型是Object[].
$$	所有实参.例如, m($$) 等价于 m($1,$2,…)
$cflow(…)	cflow变量
$r	结果类型. 用于表达式转换.
$w	包装类型. 用于表达式转换.
$_	结果值
$sig	java.lang.Class列表，代表正式入参类型
$type	java.lang.Class对象，代表正式入参值.
$class	java.lang.Class对象，代表传入的代码段.

使用步骤

使用步骤

举例

新建Java项目，导入javassist.jar。

项目结构1

项目结构

代码1

Main无所谓，是我建项目时选择了Hello World模板

Test.java

package bupt.edu.cn;
import javassist.*;

import java.lang.reflect.Method;

public class Test {
    public static void createStudent() throws  Exception {
        ClassPool pool = ClassPool.getDefault();
        CtClass cc = pool.makeClass("E:\\Workspace\\IDEA_workspace\\JavassistTest\\src\\bupt.edu.cn.Student");

        // 字段名为name
        CtField param = new CtField(pool.get("java.lang.String"),"name", cc);
        // 访问级别是 private
        param.setModifiers(Modifier.PRIVATE);
        // 初始值是 "Frankyu"
        cc.addField(param, CtField.Initializer.constant("Frankyu"));

        // 生成 getter、setter 方法
        cc.addMethod(CtNewMethod.setter("setName", param));
        cc.addMethod(CtNewMethod.getter("getName", param));

        // 添加无参的构造函数
        CtConstructor cons = new CtConstructor(new CtClass[]{}, cc);
        cons.setBody("{name = \"yubo\";}");
        cc.addConstructor(cons);

        // 5. 添加有参的构造函数
        cons = new CtConstructor(new CtClass[]{pool.get("java.lang.String")}, cc);
        // $0=this / $1,$2,$3... 代表方法参数
        cons.setBody("{$0.name = $1;}");
        cc.addConstructor(cons);

        // 6. 创建一个名为printName方法，无参数，无返回值，输出name值
        CtMethod ctMethod = new CtMethod(CtClass.voidType, "printName", new CtClass[]{}, cc);
        ctMethod.setModifiers(Modifier.PUBLIC);
        ctMethod.setBody("{System.out.println(name);}");
        cc.addMethod(ctMethod);

        //这里会将这个创建的类对象编译为.class文件
        cc.writeFile("");
    }
    public static void usingStudent() throws Exception{
        ClassPool pool = ClassPool.getDefault();
        CtClass cc = pool.getCtClass("E:\\Workspace\\IDEA_workspace\\JavassistTest\\src\\bupt.edu.cn.Student");
        // 实例化
        Object student = cc.toClass().newInstance();
        // 设置值
        Method setName = student.getClass().getMethod("setName", String.class);
        setName.invoke(student, "junjie");
        // 输出值
        Method execute = student.getClass().getMethod("printName");
        execute.invoke(student);
    }
    public static void main(String[] args) throws Exception{
        createStudent();
        usingStudent();
    }
}

注意：.class文件的绝对路径需要修改

Student.class

//
// Source code recreated from a .class file by IntelliJ IDEA
// (powered by Fernflower decompiler)
//

package E:\Workspace\IDEA_workspace\JavassistTest\src\bupt.edu.cn;

public class Student {
    private String name = "Frankyu";

    public void setName(String var1) {
        this.name = var1;
    }

    public String getName() {
        return this.name;
    }

    public Student() {
        this.name = "yubo";
    }

    public Student(String var1) {
        this.name = var1;
    }

    public void printName() {
        System.out.println(this.name);
    }
}

结果

项目结构2

项目结构

代码2

Student.java

package bupt.edu.cn;

import java.util.Random;

public class Student {
    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    private int age;

    public int getGrade() {
        return grade;
    }

    public void setGrade(int grade) {
        this.grade = grade;
    }

    private int grade;

    @Override
    public String toString() {
        return "Student{" +
                "age=" + age +
                ", grade=" + grade +
                '}';
    }

    public void display() {
        System.out.println(this.toString());
    }
    public void learn() throws Exception{
//        long start = System.currentTimeMillis();
        Random rand = new Random();
        int time =  rand.nextInt(1000)+3000;
        Thread.sleep(time);
//        long end = System.currentTimeMillis();
//        System.out.println(end-start);
    }
    public static void main(String[] args) {
        Student s  = new Student();
        s.display();
        try{
            s.learn();
        }
        catch (Exception e){

        }

    }
}

Monitor.java

import bupt.edu.cn.Student;
import javassist.*;

public class Monitor {
   final static ClassPool pool = ClassPool.getDefault();
   final static String classname = "bupt.edu.cn.Student";
    public void studentLearnMonitor() throws Exception{
        CtClass ss = pool.getCtClass(classname);
        CtClass.debugDump="./dump";
        String methodname = "learn";
        CtMethod learn_ori = ss.getDeclaredMethod(methodname);
        //拷贝一份learn方法
        CtMethod learn_cp = CtNewMethod.copy(learn_ori,learn_ori.getName()+"_cp",ss,null);
        //添加拷贝后的方法
        ss.addMethod(learn_cp);
        //修改learn方法：原代码前后添加时间
        String src = "{"+
                "long start = System.currentTimeMillis();" +
                learn_ori.getName()+"_cp($$);"+
                "long end = System.currentTimeMillis();"+
                "System.out.println(end-start);"+
                "}";
        learn_ori.setBody(src);
        ss.toClass();
        //生成.class文件，主要用于调试，查看是否有代码片段被忽略
        //ss.writeFile();
        Student s = new Student();
        s.learn();
    }

    public void studentDisplayMonitor() throws Exception{
        CtClass ss = pool.getCtClass(classname);
        CtClass.debugDump="./dump";
        //添加字段name
        CtField param = new CtField(pool.get("java.lang.String"),"name", ss);
        // 访问级别是 private
        param.setModifiers(Modifier.PRIVATE);
        // 初始值是 "Frankyu"
        ss.addField(param, CtField.Initializer.constant("Frankyu"));
        String methodname = "display";
        CtMethod display = ss.getDeclaredMethod(methodname);
        String src = "{"+
                "System.out.println($0.age);" +
                "$0.name=\"frankyu\";" +
                "System.out.println($0.name);"+
                "}";
        display.insertBefore(src);
        if(true){
            System.out.println("Hello javassist");
        }
        src ="if(true){" +
                "System.out.println(\"Hello javassist\");" +
                "}";
        display.insertAt(7,src);
        src = "{"+
                "System.out.println($0.grade);"+
                "}";
        display.insertAfter(src);
        ss.toClass();
        //ss.writeFile();
        Student s = new Student();
        s.display();
    }

    public static void main(String[] args) throws Exception{
            Monitor m = new Monitor();
            // 由于类冻结问题，两个方法不可同时调用
//            m.studentLearnMonitor();
            m.studentDisplayMonitor();
    }

}

studentLearnMonitor对learn方法插桩，通过拷贝的方式，使start和end属于同一个代码块，解决不同代码块之间变量无法使用问题。

studentDisplayMonitor对display方法插桩，演示了添加变量，访问变量，修改变量，方法前插，后插，任意位置插。

learn插桩结果

display插桩结果

调用studentDisplayMonitor时，dump目录下,Student.class部分代码

    public void display() {
        System.out.println(this.age);
        this.name = "frankyu";
        System.out.println(this.name);
        if (true) {
            System.out.println("Hello javassist");
        }

        System.out.println(this.toString());
        Object var2 = null;
        System.out.println(this.grade);
    }

调用studentLearnMonitor时，dump目录下,Student.class部分代码

    public void learn() throws Exception {
        long var1 = System.currentTimeMillis();
        this.learn_cp();
        long var3 = System.currentTimeMillis();
        System.out.println(var3 - var1);
    }

字节码级别

ClassFile

类

1. getFields():返回字段列表；
2. addField(FieldInfo finfo):添加字段；
3. addMethod(MethodInfo minfo) :添加方法；
4. getMethod(String name):根据方法名返回MethodInfo对象；

FieldInfo

字段

最后

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数网络安全工程师，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年网络安全全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上网络安全知识点！真正的体系化！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

因此收集整理了一份《2024年网络安全全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。

[外链图片转存中…(img-QZwjWtvN-1715501804221)]

[外链图片转存中…(img-JrWuoCzX-1715501804222)]

[外链图片转存中…(img-28bNTPyv-1715501804222)]

[外链图片转存中…(img-IukfWgnV-1715501804222)]

[外链图片转存中…(img-oU7UWhqx-1715501804222)]

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上网络安全知识点！真正的体系化！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！