Java Agent的agentmain和premain方式

Java Agent的agentmain和premain方式
https://www.yht7.com/news/30041
https://zhuanlan.zhihu.com/p/214093661 --jar跟cp的区别
Java Agent的agentmain和premain方式
作者:佚名 来源: 网络转载 时间:2020-03-14

首先假如我们有个应用1启动类为task,

package com.st;

import java.lang.instrument.Instrumentation;
import java.util.Scanner;

public class Task {

public static void main (String[] args) {
    Scanner scanner = new Scanner(System.in);
    scanner.hasNext();
}

}
需要执行的代码

package com.st;

import java.lang.instrument.Instrumentation;

public class AgentMainTest {

public static void  agentmain(String agentArgs, Instrumentation inst){
    System.out.println("打印全部加载的类:");
    Class[] allLoadedClasses = inst.getAllLoadedClasses();
    for (Class allLoadedClass : allLoadedClasses) {
        System.out.println(allLoadedClass.getName());
    }
}

}

maven打包配置:

<plugins>
  <plugin>
    <artifactId>maven-jar-plugin</artifactId>
    <version>3.0.2</version>
    <configuration>
      <archive>
        <manifest>
          <addClasspath>true</addClasspath>
          <mainClass>com.st.Task</mainClass>
        </manifest>
        <manifestEntries>
          <Agent-Class>
            com.st.AgentMainTest
          </Agent-Class>
        </manifestEntries>
      </archive>
    </configuration>
  </plugin>
</plugins>

打包成ks-demo-1.0-SNAPSHOT.jar,将其运行起来java -jar ks-demo-1.0-SNAPSHOT.jar

创建一个java类做应用2,导入jdk目录下的lib/tools.jar包,运行main方法

package com.st;

import com.sun.tools.attach.AgentInitializationException;
import com.sun.tools.attach.AgentLoadException;
import com.sun.tools.attach.AttachNotSupportedException;
import com.sun.tools.attach.VirtualMachine;

import java.io.IOException;

/**

  • Hello world!
    **/
    public class App
    {
    public static void main( String[] args )
    {

     try {
         //VirtualMachine 来自tools.jar
         // VirtualMachine.attach("9444") 9444为线程PID
         VirtualMachine vm = VirtualMachine.attach("9444");
         //指定应用路径
         vm.loadAgent("D:\work\ks-demo\target\ks-demo-1.0-SNAPSHOT.jar");
     } catch (AttachNotSupportedException e) {
         e.printStackTrace();
     } catch (IOException e) {
         e.printStackTrace();
     } catch (AgentLoadException e) {
         e.printStackTrace();
     } catch (AgentInitializationException e) {
         e.printStackTrace();
     }
    

    }
    }

该功能多用于收集系统信息,以及在系统加载后对class文件重新编辑等功能。

premain方式

package com.st;

public class Task {

public static void main (String[] args) {
    System.out.println("mian执行");
}

}
package com.st;

import java.lang.instrument.Instrumentation;

public class AgentMainTest {

public static void  premain(String agentArgs, Instrumentation inst){
    System.out.println("premain执行:");
}

}

maven-jar-plugin 3.0.2 true com.st.Task com.st.AgentMainTest 运行命令:java -javaagent:ks-demo-1.0-SNAPSHOT.jar -jar ks-demo-1.0-SNAPSHOT.jar

运行结果:

premain在应用加载前执行,通常用于加载重新编辑类

若本号内容有做得不到位的地方(比如:涉及版权或其他问题),请及时联系我们进行整改即可,会在第一时间进行处理。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Java Agent 是指在程序运行时,通过 Java Instrumentation API 动态修改字节码的一种技术。Java Agent 可以用于实现一些高级功能,例如监控、代码注入等。下面是一个简单的 Java Agent 的设计和编写demo: 1. 创建一个 Maven 项目,并添加以下依赖: ```xml <dependency> <groupId>org.ow2.asm</groupId> <artifactId>asm-all</artifactId> <version>5.2</version> </dependency> ``` 2. 创建一个 Java Agent 类,并实现 premain 方法: ```java public class MyAgent { public static void premain(String agentArgs, Instrumentation inst) { // 在这里添加字节码转换器 } } ``` 3. 在 premain 方法中添加字节码转换器: ```java public static void premain(String agentArgs, Instrumentation inst) { inst.addTransformer(new MyClassFileTransformer()); } ``` 4. 创建一个 MyClassFileTransformer 类,并实现 transform 方法,该方法将在类被加载时被调用: ```java public class MyClassFileTransformer implements ClassFileTransformer { public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) { // 在这里修改字节码 return classfileBuffer; } } ``` 5. 在 transform 方法中,可以使用 ASM 库修改字节码。例如,下面的代码将在每个类的构造函数中添加一行输出语句: ```java public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) { ClassReader classReader = new ClassReader(classfileBuffer); ClassWriter classWriter = new ClassWriter(ClassWriter.COMPUTE_FRAMES); // 创建一个 ClassVisitor,用于修改字节码 ClassVisitor classVisitor = new ClassVisitor(Opcodes.ASM7, classWriter) { @Override public MethodVisitor visitMethod(int access, String name, String descriptor, String signature, String[] exceptions) { MethodVisitor methodVisitor = super.visitMethod(access, name, descriptor, signature, exceptions); if ("<init>".equals(name)) { // 找到构造函数 methodVisitor.visitFieldInsn(Opcodes.GETSTATIC, "java/lang/System", "out", "Ljava/io/PrintStream;"); methodVisitor.visitLdcInsn("Hello, World!"); methodVisitor.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/io/PrintStream", "println", "(Ljava/lang/String;)V", false); } return methodVisitor; } }; // 使用 ClassVisitor 修改字节码 classReader.accept(classVisitor, ClassReader.EXPAND_FRAMES); return classWriter.toByteArray(); } ``` 6. 编译项目,并将生成的 jar 文件作为 Java Agent 运行: ```bash java -javaagent:myagent.jar -jar myprogram.jar ``` 以上就是一个简单的 Java Agent 的设计和编写demo。在实际应用中,需要根据具体的需求进行进一步的开发和优化。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值