Java Agent Premain & Agentmain

概念

premain是在jvm启动的时候类加载到虚拟机之前执行的 

agentmain是可以在jvm启动后类已经加载到jvm中了，才去转换类。 这种方式会转换会有一些限制，比如不能增加或移除字段。

具体的做法,两者的实际做法是差不多的：

premain

定义个静态方法public static void premain(String args, Instrumentation inst),

在java 的启动参数中添加 -javaagent:/jar包路径[=agentArgs] 这样定义了后jvm启动时，就会去加载javaagent中指定的jar包，查找MANIFEST.MF文件中Premain-Class属性的类，执行premain方法。

Manifest-Version: 1.0
Can-Redefine-Classes: true
Premain-Class: com.premian.MyAgent
Can-Retransform-Classes: true

import java.lang.instrument.Instrumentation;

import java.lang.management.ManagementFactory;


@Slf4j
public class PreMain {
    public static void premain(String agentArgs, Instrumentation inst) {
        log.debug("--> during jvm pre main run... <--");
        // agentLogic(agentArgs, inst);
        String jvmPid = jvmPid();
        try {
            FileUtil.writeToFile(System.getProperty("user.dir") + "/pid", jvmPid);
        } catch (IOException e) {
            e.printStackTrace();
            log.error("[ERROR] write pid to file error.");
        }
        log.debug("--> premain get class end <--\n");
    }

    private static String jvmPid() {
        String thisJvmName = ManagementFactory.getRuntimeMXBean().getName();
        String thisJvmPid = thisJvmName.split("@")[0];
        log.debug("--> this jvm process pid: " + thisJvmPid + "\n");
        return thisJvmPid;
    }
}

agentmain

定义个静态方法public static void agentmain(String agentOps, Instrumentation instrumentation),

在生成jar包中MANIFEST.MF文件中需要有Agent-Class: xxx.xxx (xxx.xxx就是上面agentmain方法所在的类名)
Can-Retransform-Classes: true

Manifest-Version: 1.0
Can-Redefine-Classes: true
Agent-Class: cn.think.in.java.clazz.loader.asm.agent.AgentMainTraceAgent 
Can-Retransform-Classes: true

使用下面代码，将agent添加到指定java进程
public class AgentMain {
    public static void agentmain(String agentArgs, Instrumentation inst) {
         vm = VirtualMachine.attach(pid);
        try {
            vm.loadAgent("D:\\tmp\\my-java-agent-1.0-jar-with-dependencies.jar", null);
        } finally {
            vm.detach();
        }
}