archive/java-memshell-generator
1
0
Fork 0
mirror of https://github.com/pen4uin/java-memshell-generator.git synced 2026-09-25 03:01:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update: 优化异常处理

Browse Source
This commit is contained in:
pen4uin
2024-08-24 00:45:45 +08:00
parent e5994d4813
commit 916d12982f
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jmg-core/src/main/java/jmg/core/format/JARAgentFormater.java
+6 -2
View File
@@ -18,6 +18,8 @@ import java.util.jar.JarFile;
import java.util.jar.JarOutputStream; import java.util.jar.JarOutputStream;
import java.util.jar.Manifest; import java.util.jar.Manifest;
// todo: 其他中间件
public class JARAgentFormater implements IFormater { public class JARAgentFormater implements IFormater {
public byte[] transform(byte[] clazzbyte, AbstractConfig config) throws Exception { public byte[] transform(byte[] clazzbyte, AbstractConfig config) throws Exception {
String className = TomcatAgentTransformer.class.getName(); String className = TomcatAgentTransformer.class.getName();
@@ -25,11 +27,13 @@ public class JARAgentFormater implements IFormater {
if (config.getServerType().equals(Constants.SERVER_TOMCAT)) { if (config.getServerType().equals(Constants.SERVER_TOMCAT)) {
className = TomcatAgentTransformer.class.getName(); className = TomcatAgentTransformer.class.getName();
simpleName = TomcatAgentTransformer.class.getSimpleName(); simpleName = TomcatAgentTransformer.class.getSimpleName();
} } else if (config.getServerType().equals(Constants.SERVER_SPRING_MVC)) {
if (config.getServerType().equals(Constants.SERVER_SPRING_MVC)){
className = SpringMVCAgentTransformer.class.getName(); className = SpringMVCAgentTransformer.class.getName();
simpleName = SpringMVCAgentTransformer.class.getSimpleName(); simpleName = SpringMVCAgentTransformer.class.getSimpleName();
} else {
throw new RuntimeException(String.format("Java Agent 暂时只支持 %s、%s",Constants.SERVER_TOMCAT,Constants.SERVER_SPRING_MVC));
} }
String classFileName = simpleName.replace('.', '/') + ".class"; String classFileName = simpleName.replace('.', '/') + ".class";
ClassPool pool = ClassPool.getDefault(); ClassPool pool = ClassPool.getDefault();
// Note: jar 包中的文件不能通过文件路径读取，需要通过流读取 // Note: jar 包中的文件不能通过文件路径读取，需要通过流读取