Agent agentmain-运行期恢复类修改，热替换

最新推荐文章于 2024-05-28 18:22:24 发布

ljz2016

最新推荐文章于 2024-05-28 18:22:24 发布

阅读量1.7k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：字节码

本文链接：https://blog.youkuaiyun.com/ljz2016/article/details/84334636

字节码专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种在Java中恢复类原始状态的方法，通过重新从源码文件读取并加载类信息到JVM，实现了运行时类修改的擦除与热替换功能，特别适用于取消先前添加的类监控。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前文讲过运行期修改类，但是有的时候想去除以前的修改，比如曾经添加了类监控，但是现在不想监控了，希望恢复成原生代码。

使用以下代码，重新从源码文件中读取类信息，然后加载到jvm中。

这既可以擦除原来的修改，也可以实现热替换功能。这里只贴出了关键代码，其它部分和前文一致。

public class RestoreTransformer implements ClassFileTransformer {
    public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
        try {
            if (className.contains(UdAgent.paths));
                return readStream(ClassLoader.getSystemResourceAsStream(className.replace('.', '/') + ".class"), true);
        } catch (IOException e) {
            e.printStackTrace();
        }

        return null;
    }

    private static byte[] readStream(InputStream inputStream, boolean close) throws IOException {
        if(inputStream == null) {
            throw new IOException("Class not found");
        } else {
            try {
                ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
                byte[] data = new byte[4096];

                int bytesRead;
                while((bytesRead = inputStream.read(data, 0, data.length)) != -1) {
                    outputStream.write(data, 0, bytesRead);
                }

                outputStream.flush();
                byte[] var5 = outputStream.toByteArray();
                return var5;
            } finally {
                if(close) {
                    inputStream.close();
                }

            }
        }
    }
}