如何通过 Instrumentation 接口实现一个能够在线动态修改代码逻辑的 Java Agent

陌明同学_7964

陌明同学_7964

2026-04-21

257人浏览

原创

必须在premain中注册,因为jvm类加载不可逆,关键类(如arraylist、beanfactory)早已加载完毕;延迟注册将导致transformer对已加载类无效,仅对后续新类生效。

如何通过 instrumentation 接口实现一个能够在线动态修改代码逻辑的 java agent

Instrumentation.addTransformer 为什么必须在 premain 中注册

因为 JVM 启动后,类加载过程已经进入不可逆阶段;Instrumentation 只允许在类被首次定义(defineClass)前插入字节码转换器。如果等到应用运行中再注册 transformer,它对已加载的类完全无效,且对后续新加载类也仅在下一次类加载时生效——但很多关键类(如 java.util.ArrayList、Spring 的 BeanFactory)早在 agent 加载前就已被加载完毕。

所以动态修改逻辑的前提是:提前介入类加载链路。这意味着你必须把 transformer 注册写死在 premain 方法里,哪怕你后续想“开关”逻辑,也只能靠 transformer 内部判断条件,而非重新注册。

  • premain 是唯一能保证捕获绝大多数目标类的时机
  • 使用 Instrumentation.retransformClasses(Class...) 可以重定义已加载类,但要求目标类此前未被 ClassLoader.defineClass 过(即不能是 bootstrap class loader 加载的核心类),且 JVM 必须开启 -XX:+HotswapAgent 或支持 retransformation(JDK 6+ 默认支持,但部分 JDK 8u 版本有 bug)
  • 若用 agentmain 方式热加载 agent,仍需在 agentmain 中调用 instrumentation.addTransformer(..., true) 并立即对目标类调用 retransformClasses,否则 transformer 不生效

Byte Buddy 和 ASM 哪个更适合 runtime 修改逻辑

Byte Buddy 封装了大部分字节码操作细节,适合快速实现「方法入口/出口增强」或「字段读写拦截」;ASM 更底层,适合精细控制指令流(比如跳过某段 if 判断、替换常量池中的字符串)。但两者都不解决「语义一致性」问题:你改了字节码,不代表业务逻辑还能跑通。

例如,想把 UserService.getUserById(long id) 的返回值强制改成 new User(0, "mock"),Byte Buddy 可以一行写完:.method(named("getUserById")).intercept(FixedValue.value(new User(0, "mock")));但如果你要插入一段含 try-catch + 外部 HTTP 调用的新逻辑,就得自己处理异常表、局部变量表扩容、栈帧校验——这时 ASM 更可控,但开发成本陡增。

java-enterprise
java-enterprise

Java企业开发专家,专注于Spring Boot、微服务架构和JVM优化。适用于:Java语言掌握、

下载
  • 优先用 Byte Buddy 实现「绕过」「固定返回」「日志注入」等简单场景
  • 涉及控制流修改(如跳过校验分支)、或需要访问原方法参数/局部变量并参与计算时,用 ASM 手写 MethodVisitor 更稳妥
  • 无论选哪个,都得在 transformer 的 transform 方法里加 if (!className.equals("com/example/UserService")) return null; 防止无差别增强所有类导致 OOM 或 VerifyError

retransformClasses 报错 java.lang.UnsupportedOperationException: class redefinition failed: attempted to change the schema (add/remove fields)

这个错误说明你试图通过 retransformClasses 添加字段、删除方法、改变签名——JVM 明确禁止这类结构性变更。retransform 只允许修改方法体(method body),包括指令序列、异常表、栈映射帧,但不能动类结构本身。

常见踩坑点:用 Byte Buddy 的 .defineField() 或 ASM 的 visitField() 在 transformer 中尝试加字段;或者误以为 Advice 的 @OnMethodEnter 可以自动分配新局部变量而没手动调用 visitVarInsn 分配槽位。

  • 所有字段、方法、接口实现,必须在类首次 define 时就确定;retransform 阶段只能改已有方法的字节码
  • 若真需要新增状态,改用 ThreadLocal 或静态 Map 存储,避免动类结构
  • JDK 9+ 引入了 ClassFileTransformer 的 canRedefineClasses() 和 isModifiableClass(Class),可在 transform 前做预检,避免无效 retransform

如何让 agent 支持运行时开关和参数传递

agent 本身没有标准配置机制,得自己搭通信通道。最轻量的方式是监听本地文件(如 /tmp/myagent.conf),定期检查时间戳变化;更可靠的是开一个嵌入式 HTTP 端点(用 Jetty 或 Undertow),接收 PUT 请求更新开关状态。

关键是:这些开关变量必须是 static volatile 或 AtomicBoolean,并在 transformer 的 transform 方法开头做快速判断,避免每次方法调用都走 I/O 或网络。例如:

public byte[] transform(ClassLoader loader, String className, Class> classBeingRedefined,
                        ProtectionDomain protectionDomain, byte[] classfileBuffer) {
    if (!ENABLED.get()) return null;
    if (!TARGET_CLASS.matcher(className).matches()) return null;
    // ... real transformation
}
  • 不要在 transformer 里做耗时操作(如读文件、发 HTTP、查 DB),会拖慢所有类加载
  • 开关变量建议用 AtomicBoolean + Unsafe 的 lazySet,避免 volatile 写的内存屏障开销
  • 如果要用 JSON 配置,别用 Jackson/Gson 解析——它们自身类可能还没加载完,优先用正则或 Properties.load()
Java Agent 动态改逻辑这事,难点从来不在字节码怎么写,而在于你改完之后,JVM 还认不认这个类、其他线程还在不在用旧版本对象、GC 是否已回收相关元数据——这些边界条件,比 transformer 本身难 debug 得多。

Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

10197

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

7322

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

6432

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1104

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

908

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1336

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

2689

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

20011

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1195

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习