前文讲过运行期修改类,但是有的时候想去除以前的修改,比如曾经添加了类监控,但是现在不想监控了,希望恢复成原生代码。
使用以下代码,重新从源码文件中读取类信息,然后加载到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();
}
}
}
}
}