archive/java-memshell-generator
1
0
Fork 0
mirror of https://github.com/pen4uin/java-memshell-generator.git synced 2026-09-22 01:30:43 +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
+9 -5
View File
@@ -18,18 +18,22 @@ 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();
String simpleName = TomcatAgentTransformer.class.getSimpleName(); String simpleName = TomcatAgentTransformer.class.getSimpleName();
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 包中的文件不能通过文件路径读取，需要通过流读取
@@ -53,7 +57,7 @@ public class JARAgentFormater implements IFormater {
copyJarEntries(jar, tempJar); copyJarEntries(jar, tempJar);
addModifiedClassToJar(pool, className, simpleName, classFileName, tempJar, config.getPass(),CommonUtil.encodeBase64(clazzbyte)); addModifiedClassToJar(pool, className, simpleName, classFileName, tempJar, config.getPass(), CommonUtil.encodeBase64(clazzbyte));
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
} }
@@ -86,7 +90,7 @@ public class JARAgentFormater implements IFormater {
} }
} }
private void addModifiedClassToJar(ClassPool pool, String className, String simpleName, String classFileName, JarOutputStream tempJar,String injectFlag, String injectorCode) throws Exception { private void addModifiedClassToJar(ClassPool pool, String className, String simpleName, String classFileName, JarOutputStream tempJar, String injectFlag, String injectorCode) throws Exception {
CtClass ctClass = pool.get(className); CtClass ctClass = pool.get(className);
ctClass.getClassFile().setVersionToJava5(); ctClass.getClassFile().setVersionToJava5();
ctClass.setName(simpleName); ctClass.setName(simpleName);