java 字节码替换

java agent

spinrgboot+Javassist
aop运行于应用程序内的方法级别,而agent能够作用于虚拟机级别

对静态方法的改造
 <dependency>
            <groupId>org.javassist</groupId>
            <artifactId>javassist</artifactId>
            <version>3.28.0-GA</version> <!-- 版本号可能会更新 -->
        </dependency>

@Slf4j
public class AgentUtil {

    public static String getToken(){

        return "123"+ UUID.randomUUID();
    }

    public static String doThing(){
        log.info("====>>> start dong thing");
        return "doing things";
    }
}

public class Agenttest001Application {

    public static void main(String[] args) throws Exception{
        SpringApplication.run(Agenttest001Application.class, args);

        ClassPool pool = ClassPool.getDefault();

        // 获取目标类的 CtClass 对象
        CtClass ctClass = pool.get("com.jin.agenttest001.utils.AgentUtil");

        // 获取目标静态方法的 CtMethod 对象
        CtMethod method = ctClass.getDeclaredMethod("getToken");

        // 修改方法体为返回新的字符串
        method.setBody("{ return \"abc\"; }");

        // 将修改后的类重新加载到 JVM 中
          ctClass.toClass();

    }
}

在 Javassist 中, 1 是一个占位符,用于表示传递给方法或构造函数的第一个参数。它是在修改方法体时用来引用原始方法签名中的参数的一种方式。更一般地, 1 是一个占位符,用于表示传递给方法或构造函数的第一个参数。它是在修改方法体时用来引用原始方法签名中的参数的一种方式。更一般地, 1是一个占位符,用于表示传递给方法或构造函数的第一个参数。它是在修改方法体时用来引用原始方法签名中的参数的一种方式。更一般地,n 表示第 n 个参数(从1开始计数),而 $0 则代表当前对象实例(类似于 Java 中的 this) ,对修改构造方法注意下

 public static void mockStudent(ClassPool pool)throws Exception{
        CtClass ctClass = pool.get("com.jin.agenttest001.dto.Student");
        // 获取目标构造方法的 CtConstructor 对象
        CtConstructor[] constructors = ctClass.getDeclaredConstructors();
        for (CtConstructor constructor : constructors) {
            if (constructor.getParameterTypes().length == 1 &&
                    constructor.getParameterTypes()[0].getName().equals("int")) {

                // 修改构造方法体
                constructor.setBody("{ this.name = \"jin\"; this.age = $1; }");

                break;
            }
        }
        CtMethod fun2 = ctClass.getDeclaredMethod("fun2");
        fun2.setBody("{ return \"fun2\"; }");
        ctClass.toClass();
    }

对注入的类,顺序稍微调整下
在这里插入图片描述

完整代码:
https://gitee.com/jinfei602917226/agent/tree/master/src/main/java/com/jin/agenttest001

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值