首页 > 云计算 > 虚拟化 > 字节码增强:原理与实战
本文由一个拦截器逻辑的使用场景及演变历程,引入字节码增强技术。介绍字节码的本质,字节码增强的原理及JVM 启动过程中的 Agent 加载、生效流程,并对常见字节码操作工具进行了简单应用。
注:本文仅讨论 javaagent “启动时加载”。
技术是工具,是解决问题的途径。针对不同的业务需求场景,可以使用不同的技术实现。
通过一部拦截器的流浪史来引入主题:
被调用方

调用方

常规开发:
被调用方

调用方

面向对象设计原则,对象应该尽可能专注自己职责范围内的事情,狗只负责叫,不负责统计自己叫了多长时间,因此统计代码应该移出Dog类。






至此,非业务逻辑由从被调用方剥离出来了,同时我们也发现调用方代码却遭到改变,Main class里面需要添加动态代理类的处理逻辑。假如不允许改变调用方代码,进一步处理。
切面

被调方

调用方

注意:此时直接运行Main class切面不会生效,运行前先进行编译期织入 java -jar $ASPECTJ_TOOLS -cp $ASPECTJ_RT -sourceroots src/main/java/ -d target/classes ...
至此,调用方不用显式地调用动态代理逻辑,编译期织入到class中去了(这里已经闻到了代码增强的气味了)。
切面逻辑虽然与具体的业务逻辑解耦合了,独立出切面类。但是是否生效仍然由业务代码(切面类)去控制。无论如何,都需要业务方改造,添加切面逻辑代码。
能不能更进一步,连切面都不写,也让切面逻辑生效呢?

MANIFEST.MF文件。

待替换的新class文件(忽略中文乱码)。

class转换器,将新的Dog.class替换旧的Dog.class。

maven打包输出Agent.jar

上面的javaagent实现细节可以先存疑,后面会深入描述,只需要知道按照这样的步骤可以实现我们的需求。
对于业务方而言:代码完全没有变化:
被调用方

调用方

想要使切面逻辑生效,只需要在启动命令参数中加入-javaagent 选项,指向 Agent 的 jar 包。

这样,拦截器逻辑以一种插件的形式抽取出来了,使用的时候加载插件就可以了。
byte[]

在我们刚开始学习 Java 语言时候的 demo 运行:
编写原始 Java 文件:



使用 Javac 编译字节码文件:

Javac生产的 class 文件有什么作用呢?
Java 语言一次编译,到处运行的核心基础-JVM。

先用文本编辑器暴力打开看看:

看不懂?换个方式:

想看个明白?继续整:使用010editor打开。

各个数据项按顺序紧密的从前向后排列, 相邻的项之间没有间隙, 这样可以使得class文件非常紧凑, 体积轻巧, 可以被JVM快速的加载至内存, 并且占据较少的内存空间。
主要包含的信息:
(1)魔数
(2)版本号(参考文末例子:JRE版本错误)
(3)常量池容量
(4)常量池:
(5)其他属性
常量池如何索引:
相互索引:
例如方法索引,获取classIndex和nameAndTypeIndex,通过数组下标,可以找到该方法所属的class和方法名称。
MethodRef
|-----|classIndex |-----|-----|nameIndex --→ classNmae |-----|nameAndTypeIndex |-----|-----|nameIndex --→ methodName
常量池索引和字节码指令的执行。
使用jre自带工具javap反编译class文件如下:
Main.class字节码:

Dog.class字节码:

可以看到字节码具备一定的可读性,对照着源码,可以按照执行逻辑走一遍字节码执行流程,相关指令的含义很容易从网上查询到。

至此,我们通过一个简单的demo执行流程,大致了解了常量的引用以及一个简单java方法对应的字节码指令执行过程。
注:
stack:最大操作数栈,JVM运行时会根据这个值来分配栈帧(Frame)中的操作栈深度。
locals:局部变量所需的存储空间,单位为Slot。
args_size: 方法参数的个数。
压栈:字节码指令执行过程中涉及到了很多压栈操作:JVM是一个基于栈的架构。方法执行的时候(包括main方法),在栈上会分配一个新的帧,这个栈帧包含一组局部变量。
这组局部变量包含了方法运行过程中用到的所有变量,包括this引用,所有的方法参数,以及其它局部定义的变量。
byte[] → byte[]
思路:
工具集:/ASM/javaassist/ByteBuddy 等等。

示例:
指令级别的字节码操作(性能强悍)。


指令→ASM api 对应关系(这里将原始类做了简化,将字符串拼接逻辑去掉,仅仅输出时间。因为一个简单的字符串拼接过程,转换成字节码指令可能需要很多行)。
先看看 目标源码与字节码指令的一一映射关系。

再看看 增强字节码逻辑与目标源码的字节码的一一映射关系。

通过对比我们可以发现,ASM的API精确到字节码指令级别,所有的临时变量存储,压栈操作,静态/实例方法的调用都有对应的API操作。
提供字节码级别的API,类似ASM,不再赘述。
提供源码级别的API,针对本文的案例,实现如下:

基于ASM的高级API,使我们对字节码的操作提升到更抽象层次。开发者只需要知道要实现什么目标,如何使用对应的API,不用关心底层的字节码指令排列,甚至可以不用了解字节码指令。


关于相关框架的API不详细说,有兴趣的同学可以自行查询相关资料。
各种级别的API可以帮助开发者轻松实现字节码增强,实现特定逻辑。
不论什么奇技淫巧,都离不开Instrumentation机制。
Event --> CallBack
由前文总结,引入Instrumentation机制。
JVMTI 是基于事件驱动的,JVM 每执行到一定的逻辑就会调用一些事件的回调接口(如果有的话),这些接口可以供开发者扩展自己的逻辑。
JVMTIAgent 使用JVMTI来查询或控制JVM,JVMTIAgent与目标JVM运行在同一个进程中,通过JVMTI进行通信,最大化控制能力,最小化通信成本。
典型场景下,JVMTI代理会被实现的非常紧凑,其他的进程会与JVMTI代理进行通信。比如jdwp(IDEA远程调试)。
表现形式:
(1)linux: .so文件
(2)JPLISAgent: .jar文件
命令行参数
(1)-agentlib:agent-lib-name=options
(2)-agentpath:path-to-agent=options
(3)-javaagent:/data/../../Agent.jar
可加载多个,通过options区分
实现接口
(1)JNIEXPORT jint JNICALL
Agent_OnLoad(JavaVM *vm, char *options, void *reserved);
(2)JNIEXPORT jint JNICALL
Agent_OnAttach(JavaVM vm, char options, void* reserved);
(3)JNIEXPORT void JNICALL
Agent_OnUnload(JavaVM *vm);
JPLISAgent(Java Programming Language Instrumentation Services Agent)-- Instrumentation机制
(1)JavaSE1.5 启动时加载(本文重点)。
(2)JavaSE1.6 运行时加载。
命令参数:-javaagent:/data/../../Agent.jar=optoions。
虚拟机创建-构建并初始化Agent-注册VMInit事件。
虚拟机初始化-触发VMInit事件-Agent start方法-注册回调函数并监听ClassFileLoadHook。
类加载-触发jvmtiEventClassFileLoadHook事件-替换byt[]-ClassLoader解析。

启动时读取jvm命令,-agentlib -agentpath -javaagent,并构建了Agent Library链表构建了Agent Library链表。
对agent链表中的每个agent,加载所指定的动态库(如instrument.so), 并调用里面的Agent_OnLoad方法。
创建并初始化 JPLISAgent,初始化了Premain class和包里的配置文件。
注册VMInit事件。







实际上是调用 java 类 sun.instrument.InstrumentationImpl 类里的方法loadClassAndCallPremain。




实际调用的java方法 Instrumentationimpl.transform。

debug过程中通过ClassFileTransformer的transform函数的执行堆栈印证。

到这里,增强的byte[]如何生效并影响运行时class的过程基本可以串起来。
非常规应用:IDEA破解。
部分破解教程里面下载插件jar后,会要求你在IDEA的启动参数文件idea.vmoptions中添加一行,就是javaagent参数。

我们可以反编译这个插件jar包看看,发现很多class因为加了混淆,反编译后无法正常识别,但是核心入口Agent.class的主要工作就是注册Transformer,可以推测这些Transformer的功能就是在IDEA启动时之前修改某些鉴定Lisence的逻辑。

通过介绍字节码,字节码操作工具以及openJDK关于Instrumention机制的部分源码,探索了字节码增强的实现原理。
简单介绍了相关技术的应用场景。
SpringMVC-Interceptor


IDEA 远程调试


JRE版本错误

作者: Neo
来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/69912579/viewspace-2728048/ ,如需转载,请注明出处,否则将追究法律责任。