热门标签 | HotTags
当前位置:  开发笔记 > 编程语言 > 正文

打桩时不修改源码_说实话,你工作5年,不知道什么是JavaAgent技术,让我很吃惊。。。...

来源:cnblogs.comkokovp12120033.html1、引子2、思路2.1朴素思路2.2略成熟思路3、javaagent介绍3.1什么是javaagen

来源:cnblogs.com/kokov/p/12120033.html

  • 1、引子
  • 2、思路
    • 2.1 朴素思路
    • 2.2 略成熟思路
  • 3、 java agent介绍
    • 3.1 什么是java agent?
    • 3.2 java Instrumentation API
  • 4、java agent原理简述
    • 4.1 启动时修改
    • 4.2 运行时修改
    • 4.3 ClassFileLoadHook和TransFormClassFile
    • 4.4 何时进行运行时替换?

注:本文定义-在函数执行前后增加对应的逻辑的操作统称为MOCK

1、引子

在某天与QA同学进行沟通时,发现QA同学有针对某个方法调用时,有让该方法停止一段时间的需求,我对这部分的功能实现非常好奇,因此决定对原理进行一些深入的了解,力争找到一种使用者尽可能少的对原有代码进行修改的方式,以达到对应的MOCK要求。

整体的感知程度可以分为三个级别:

  • 硬编码
  • 增加配置
  • 无需任何修改
2、思路

在对方法进行mock,暂停以及异常模拟,在不知道其原理的情况下,进行猜想,思考其具体的实现原理,整体来说,最简单的实现模型无外乎两种:

2.1 朴素思路

假设存在如下的函数

public Object targetMethod(){
    System.out.println("运行");
}

若想要在函数执行后暂停一段时间、返回特定mock值或抛出特定异常,那么可以考虑修改对应的函数内容:

public Object targetMethod(){
      //在此处加入Sleep return 或 throw逻辑
    System.out.println("运行");
}

或使用类似代理的方法把对应的函数进行代理:

public Object proxy(){
    //执行Sleep return 或 throw逻辑
      return targetMethod();
}
public Object targetMethod(){
    System.out.println("运行");
}

2.2 略成熟思路

在朴素思路的基础上,我们可以看出,实现类似的暂停、mock和异常功能整体实现方案无外乎两种:

  • 代理模式
  • 深入修改内部函数

在这两种思路的基础上,我们从代理模式开始考虑(主要是代理使用的比较多,更熟悉)

2.2.1 动态代理

说起代理,最常想到的两个词语就是静态代理和动态代理,二者却别不进行详述,对于静态代理模式由于需要大量硬编码,所以完全可以不用考虑。

针对动态代理来看,开始考虑最具代表性的CGLIB进行调研。

下面的代码为一个典型的使用CGLIB进行动态代理的样例(代理的函数为PersonService.setPerson):

public class CGlibDynamicProxy implements MethodInterceptor {
    public Object intercept(Object o, Method method, Object[] objects, MethodProxy methodProxy) throws Throwable {
        System.out.println("执行前...");
        Object object = methodProxy.invokeSuper(o, objects);
        System.out.println("执行后...");
        return object;
    }

    static class PersonService {
        public PersonService() {
            System.out.println("PersonService构造");
        }

        public void setPerson() {
            System.out.println("PersonService:setPerson");
        }
    }

    public static void main(String[] args) {
        Enhancer enhancer = new Enhancer();
        enhancer.setSuperclass(PersonService.class);
        enhancer.setCallback(new CGlibDynamicProxy());
        PersonService proxy= (PersonService)  enhancer.create();
        proxy.setPerson();
    }
}

从上面代码可以看出,对于CGLIB的动态代理而言,需要在原有代码中进行硬编码,且需要在对象初始化的时候,使用特定的方式进行初始化。因此若使用CGLIB完成MOCK,需要对应代码的的感知程度最高,达到了硬编码的程度。

2.2.2 AspectJ

由于使用代理方式无法在不对代码进行修改的情况下完成MOCK,因此我们抛弃代理方式,考虑使用修改方法内部代码的方式进行MOCK。

基于这种思路,将目光转向了AspectJ。

在使用AspectJ时,需要定义方法执行前的函数以及方法执行后的函数:

@Aspect
public class AspectJFrame {
    private Object before() {
        System.out.println("before");
        return new Object();
    }

    private Object after() {
        System.out.println("after");
        return new Object();
    }

    @Around("aroundPoint()")
    public Object doMock(ProceedingJoinPoint joinPoint) {
        Object object=null;
        before();
        try {
            object = joinPoint.proceed();
        } catch (Throwable throwable) {
            throwable.printStackTrace();
        }
        after();
        return object;
    }
}

并通过aop.xml指定对应的切点以及对应的环绕函数

"com.test.framework.AspectJFrame">"" pointcut=""/>

但是基于以上的实现方式,需要对原有项目进行一定侵入,主要包含两部分内容:

  • 在META-INF路径下增加aop.xml
  • 引入对应的切面定义的jar包

通过aspectj可以完成在硬编码的情况下实现MOCK,但是这种实现方式受限于Aspectj自身局限,MOCK的功能代码在编译期就已经添加到对应的函数中了,最晚可在运行时完成MOCK功能代码的添加。这种方式主要有两个缺点:

  • 对于运行中的java进行无法在不重启的条件下执行新增MOCK
  • MOCK功能代码嵌入到目标函数中,无法对MOCK功能代码进行卸载,可能带来稳定性风险
3、 java agent介绍

由于在上述提到的各种技术都难以很好的支持在对原有项目无任何修改下完成MOCK功能的需求,在查阅资料后,将目光放至了java agent技术。

3.1 什么是java agent?

java agent本质上可以理解为一个插件,该插件就是一个精心提供的jar包,这个jar包通过JVMTI(JVM Tool Interface)完成加载,最终借助JPLISAgent(Java Programming Language Instrumentation Services Agent)完成对目标代码的修改。

java agent技术的主要功能如下:

  • 可以在加载java文件之前做拦截把字节码做修改
  • 可以在运行期将已经加载的类的字节码做变更
  • 还有其他的一些小众的功能
    • 获取所有已经被加载过的类
    • 获取所有已经被初始化过了的类
    • 获取某个对象的大小
    • 将某个jar加入到bootstrapclasspath里作为高优先级被bootstrapClassloader加载
    • 将某个jar加入到classpath里供AppClassloard去加载
    • 设置某些native方法的前缀,主要在查找native方法的时候做规则匹配

3.2 java Instrumentation API

通过java agent技术进行类的字节码修改最主要使用的就是Java Instrumentation API。下面将介绍如何使用Java Instrumentation API进行字节码修改。

3.2.1 实现agent启动方法

Java Agent支持目标JVM启动时加载,也支持在目标JVM运行时加载,这两种不同的加载模式会使用不同的入口函数,如果需要在目标JVM启动的同时加载Agent,那么可以选择实现下面的方法:

[1] public static void premain(String agentArgs, Instrumentation inst);
[2] public static void premain(String agentArgs);

JVM将首先寻找[1],如果没有发现[1],再寻找[2]。如果希望在目标JVM运行时加载Agent,则需要实现下面的方法:

[1] public static void agentmain(String agentArgs, Instrumentation inst);
[2] public static void agentmain(String agentArgs);

这两组方法的第一个参数AgentArgs是随同 “–javaagent”一起传入的程序参数,如果这个字符串代表了多个参数,就需要自己解析这些参数。inst是Instrumentation类型的对象,是JVM自动传入的,我们可以拿这个参数进行类增强等操作。

3.2.2 指定Main-Class

Agent需要打包成一个jar包,在ManiFest属性中指定“Premain-Class”或者“Agent-Class”,且需根据需求定义Can-Redefine-Classes和Can-Retransform-Classes:

Manifest-Version: 1.0
preMain-Class: com.test.AgentClass
Archiver-Version: Plexus Archiver
Agent-Class: com.test.AgentClass
Can-Redefine-Classes: true
Can-Retransform-Classes: true
Created-By: Apache Maven 3.3.9
Build-Jdk: 1.8.0_112

3.2.3 agent加载

  • 启动时加载
    • 启动参数增加-javaagent:[path],其中path为对应的agent的jar包路径
  • 运行中加载
    • 使用com.sun.tools.attach.VirtualMachine加载

try {
  String jvmPid = 目标进行的pid;
  logger.info("Attaching to target JVM with PID: " + jvmPid);
  VirtualMachine jvm = VirtualMachine.attach(jvmPid);
  jvm.loadAgent(agentFilePath);//agentFilePath为agent的路径
  jvm.detach();
  logger.info("Attached to target JVM and loaded Java agent successfully");
} catch (Exception e) {
  throw new RuntimeException(e);
}

3.2.4 Instrument

instrument是JVM提供的一个可以修改已加载类的类库,专门为Java语言编写的插桩服务提供支持。它需要依赖JVMTI的Attach API机制实现。在JDK 1.6以前,instrument只能在JVM刚启动开始加载类时生效,而在JDK 1.6之后,instrument支持了在运行时对类定义的修改。要使用instrument的类修改功能,我们需要实现它提供的ClassFileTransformer接口,定义一个类文件转换器。接口中的transform()方法会在类文件被加载时调用,而在transform方法里,我们可以利用上文中的ASM或Javassist对传入的字节码进行改写或替换,生成新的字节码数组后返回。

首先可以定义如下的类转换器:

public class TestTransformer implements ClassFileTransformer {
      //目标类名称,  .分隔
      private String targetClassName;
    //目标类名称,  /分隔
    private String targetVMClassName;
    private String targetMethodName;


    public TestTransformer(String className,String methodName){
        this.targetVMClassName = new String(className).replaceAll("\\.","\\/");
        this.targetMethodName = methodName;
        this.targetClassName=className;
    }
    //类加载时会执行该函数,其中参数 classfileBuffer为类原始字节码,返回值为目标字节码,className为/分隔
    public byte[] transform(ClassLoader loader, String className, Class> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
          //判断类名是否为目标类名
        if(!className.equals(targetVMClassName)){
            return classfileBuffer;
        }
        try {
            ClassPool classPool = ClassPool.getDefault();
            CtClass cls = classPool.get(this.targetClassName);
            CtMethod ctMethod = cls.getDeclaredMethod(this.targetMethodName);
            ctMethod.insertBefore("{ System.out.println(\"start\"); }");
            ctMethod.insertAfter("{ System.out.println(\"end\"); }");
            return cls.toBytecode();
        } catch (Exception e) {

        }
        return classfileBuffer;
    }
}

类转换器定义完毕后,需要将定义好的类转换器添加到对应的instrmentation中,对于已经加载过的类使用retransformClasses对类进行重新加载:

public class AgentDemo {

    private static String className = "hello.GreetingController";
    private static String methodName = "getDomain";

    public static void agentmain(String args, Instrumentation instrumentation) {

        try {
            List needRetransFormClasses &#61; new LinkedList<>();
            Class[] loadedClass &#61; instrumentation.getAllLoadedClasses();for (int i &#61; 0; i                 if (loadedClass[i].getName().equals(className)) {
                    needRetransFormClasses.add(loadedClass[i]);
                }
            }
            instrumentation.addTransformer(new TestTransformer(className, methodName));
            instrumentation.retransformClasses(needRetransFormClasses.toArray(new Class[0]));
        } catch (Exception e) {
        }
    }public static void premain(String args, Instrumentation instrumentation) {
        instrumentation.addTransformer(new TestTransformer(className, methodName));
    }
}

从上图的代码可以看出&#xff0c;主方法实现了两个&#xff0c;分别为agentmain和premain&#xff0c;其中

  • premain
    • 用于在启动时&#xff0c;类加载前定义类的TransFormer&#xff0c;在类加载的时候更新对应的类的字节码
  • agentmain
    • 注册类的TransFormer
    • 调用retransformClasses函数进行类的重加载
    • 用于在运行时进行类的字节码的修改&#xff0c;步骤整体分为两步
4、java agent原理简述

4.1 启动时修改

6438548ea264915681a942c1902aa896.png

启动时修改主要是在jvm启动时&#xff0c;执行native函数的Agent_OnLoad方法&#xff0c;在方法执行时&#xff0c;执行如下步骤&#xff1a;

  • 创建InstrumentationImpl对象
  • 监听ClassFileLoadHook事件
  • 调用InstrumentationImpl的loadClassAndCallPremain方法&#xff0c;在这个方法里会去调用javaagent里MANIFEST.MF里指定的Premain-Class类的premain方法

4.2 运行时修改

7fe3fe8bbee40aae8301e854655b0633.png

运行时修改主要是通过jvm的attach机制来请求目标jvm加载对应的agent&#xff0c;执行native函数的Agent_OnAttach方法&#xff0c;在方法执行时&#xff0c;执行如下步骤&#xff1a;

  • 创建InstrumentationImpl对象
  • 监听ClassFileLoadHook事件
  • 调用InstrumentationImpl的loadClassAndCallAgentmain方法&#xff0c;在这个方法里会去调用javaagent里MANIFEST.MF里指定的Agentmain-Class类的agentmain方法

4.3 ClassFileLoadHook和TransFormClassFile

在4.1和4.2节中&#xff0c;可以看出整体流程中有两个部分是具有共性的&#xff0c;分别为&#xff1a;

  • ClassFileLoadHook
  • TranFormClassFile

ClassFileLoadHook是一个jvmti事件&#xff0c;该事件是instrument agent的一个核心事件&#xff0c;主要是在读取字节码文件回调时调用&#xff0c;内部调用了TransFormClassFile函数。

TransFormClassFile的主要作用是调用java.lang.instrument.ClassFileTransformer的tranform方法&#xff0c;该方法由开发者实现&#xff0c;通过instrument的addTransformer方法进行注册。

通过以上描述可以看出在字节码文件加载的时候&#xff0c;会触发ClassFileLoadHook事件&#xff0c;该事件调用TransFormClassFile&#xff0c;通过经由instrument的addTransformer注册的方法完成整体的字节码修改。

对于已加载的类&#xff0c;需要调用retransformClass函数&#xff0c;然后经由redefineClasses函数&#xff0c;在读取已加载的字节码文件后&#xff0c;若该字节码文件对应的类关注了ClassFileLoadHook事件&#xff0c;则调用ClassFileLoadHook事件。后续流程与类加载时字节码替换一致。

4.4 何时进行运行时替换&#xff1f;

在类加载完毕后&#xff0c;对应的想要替换函数可能正在执行&#xff0c;那么何时进行类字节码的替换呢&#xff1f;

由于运行时类字节码替换依赖于redefineClasses&#xff0c;那么可以看一下该方法的定义&#xff1a;

jvmtiError
JvmtiEnv::RedefineClasses(jint class_count, const jvmtiClassDefinition* class_definitions) {
//TODO: add locking
  VM_RedefineClasses op(class_count, class_definitions, jvmti_class_load_kind_redefine);
  VMThread::execute(&op);
  return (op.check_error());
} /* end RedefineClasses */

其中整体的执行依赖于VMThread&#xff0c;VMThread是一个在虚拟机创建时生成的单例原生线程&#xff0c;这个线程能派生出其他线程。同时&#xff0c;这个线程的主要的作用是维护一个vm操作队列(VMOperationQueue)&#xff0c;用于处理其他线程提交的vm operation&#xff0c;比如执行GC等。

VmThread在执行一个vm操作时&#xff0c;先判断这个操作是否需要在safepoint下执行。若需要safepoint下执行且当前系统不在safepoint下&#xff0c;则调用SafepointSynchronize的方法驱使所有线程进入safepoint中&#xff0c;再执行vm操作。执行完后再唤醒所有线程。若此操作不需要在safepoint下&#xff0c;或者当前系统已经在safepoint下&#xff0c;则可以直接执行该操作了。所以&#xff0c;在safepoint的vm操作下&#xff0c;只有vm线程可以执行具体的逻辑&#xff0c;其他线程都要进入safepoint下并被挂起&#xff0c;直到完成此次操作。

因此&#xff0c;在执行字节码替换的时候需要在safepoint下执行&#xff0c;因此整体会触发stop-the-world。




推荐阅读
  • 开机自启动的几种方式
    0x01快速自启动目录快速启动目录自启动方式源于Windows中的一个目录,这个目录一般叫启动或者Startup。位于该目录下的PE文件会在开机后进行自启动 ... [详细]
  • 原文网址:https:www.cnblogs.comysoceanp7476379.html目录1、AOP什么?2、需求3、解决办法1:使用静态代理4 ... [详细]
  • Python 程序转换为 EXE 文件:详细解析 .py 脚本打包成独立可执行文件的方法与技巧
    在开发了几个简单的爬虫 Python 程序后,我决定将其封装成独立的可执行文件以便于分发和使用。为了实现这一目标,首先需要解决的是如何将 Python 脚本转换为 EXE 文件。在这个过程中,我选择了 Qt 作为 GUI 框架,因为之前对此并不熟悉,希望通过这个项目进一步学习和掌握 Qt 的基本用法。本文将详细介绍从 .py 脚本到 EXE 文件的整个过程,包括所需工具、具体步骤以及常见问题的解决方案。 ... [详细]
  • 本文将深入探讨 iOS 中的 Grand Central Dispatch (GCD),并介绍如何利用 GCD 进行高效多线程编程。如果你对线程的基本概念还不熟悉,建议先阅读相关基础资料。 ... [详细]
  • 本文详细介绍了 Spark 中的弹性分布式数据集(RDD)及其常见的操作方法,包括 union、intersection、cartesian、subtract、join、cogroup 等转换操作,以及 count、collect、reduce、take、foreach、first、saveAsTextFile 等行动操作。 ... [详细]
  • 本文详细介绍了如何使用Python的多进程技术来高效地分块读取超大文件,并将其输出为多个文件。通过这种方式,可以显著提高读取速度和处理效率。 ... [详细]
  • 本文节选自《NLTK基础教程——用NLTK和Python库构建机器学习应用》一书的第1章第1.2节,作者Nitin Hardeniya。本文将带领读者快速了解Python的基础知识,为后续的机器学习应用打下坚实的基础。 ... [详细]
  • IOS Run loop详解
    为什么80%的码农都做不了架构师?转自http:blog.csdn.netztp800201articledetails9240913感谢作者分享Objecti ... [详细]
  • 利用python爬取豆瓣电影Top250的相关信息,包括电影详情链接,图片链接,影片中文名,影片外国名,评分,评价数,概况,导演,主演,年份,地区,类别这12项内容,然后将爬取的信息写入Exce ... [详细]
  • Ihavetwomethodsofgeneratingmdistinctrandomnumbersintherange[0..n-1]我有两种方法在范围[0.n-1]中生 ... [详细]
  • 本文介绍如何使用 Python 的 DOM 和 SAX 方法解析 XML 文件,并通过示例展示了如何动态创建数据库表和处理大量数据的实时插入。 ... [详细]
  • Java高并发与多线程(二):线程的实现方式详解
    本文将深入探讨Java中线程的三种主要实现方式,包括继承Thread类、实现Runnable接口和实现Callable接口,并分析它们之间的异同及其应用场景。 ... [详细]
  • 本文详细解析了使用C++实现的键盘输入记录程序的源代码,该程序在Windows应用程序开发中具有很高的实用价值。键盘记录功能不仅在远程控制软件中广泛应用,还为开发者提供了强大的调试和监控工具。通过具体实例,本文深入探讨了C++键盘记录程序的设计与实现,适合需要相关技术的开发者参考。 ... [详细]
  • 为了确保iOS应用能够安全地访问网站数据,本文介绍了如何在Nginx服务器上轻松配置CertBot以实现SSL证书的自动化管理。通过这一过程,可以确保应用始终使用HTTPS协议,从而提升数据传输的安全性和可靠性。文章详细阐述了配置步骤和常见问题的解决方法,帮助读者快速上手并成功部署SSL证书。 ... [详细]
  • 在Linux系统中,网络配置是至关重要的任务之一。本文详细解析了Firewalld和Netfilter机制,并探讨了iptables的应用。通过使用`ip addr show`命令来查看网卡IP地址(需要安装`iproute`包),当网卡未分配IP地址或处于关闭状态时,可以通过`ip link set`命令进行配置和激活。此外,文章还介绍了如何利用Firewalld和iptables实现网络流量控制和安全策略管理,为系统管理员提供了实用的操作指南。 ... [详细]
author-avatar
lajallan608
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有