archive/MemShellParty
1
0
Fork 0
mirror of https://github.com/ReaJason/MemShellParty.git synced 2026-09-21 22:50:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: support GroovyTransformJar packer

Browse Source
This commit is contained in:
ReaJason
2025-12-08 01:43:41 +08:00
parent 8b8cc7173d
commit 102173b924
10 changed files with 259 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packer/src/main/java/com/reajason/javaweb/packer/Packers.java
+1
View File
@@ -177,6 +177,7 @@ public enum Packers {
Jar(new DefaultJarPacker()),
ScriptEngineJar(new ScriptEngineJarPacker()),
GroovyTransformJar(new GroovyTransformJarPacker()),
XxlJob(new XxlJobPacker()),
;
packer/src/main/java/com/reajason/javaweb/packer/jar/GroovyTransformJarPacker.java
+40
View File
@@ -0,0 +1,40 @@
package com.reajason.javaweb.packer.jar;
import com.reajason.javaweb.asm.ClassAnnotationUtils;
import com.reajason.javaweb.asm.ClassInterfaceUtils;
import com.reajason.javaweb.packer.JarPacker;
import com.reajason.javaweb.packer.JarPackerConfig;
import lombok.SneakyThrows;
import java.io.ByteArrayOutputStream;
import java.nio.charset.StandardCharsets;
import java.util.jar.JarEntry;
import java.util.jar.JarOutputStream;
import java.util.jar.Manifest;
/**
* @author ReaJason
* @since 2025/12/6
*/
public class GroovyTransformJarPacker implements JarPacker {
@Override
@SneakyThrows
public byte[] packBytes(JarPackerConfig config) {
String mainClassName = config.getMainClassName();
byte[] mainClassBytes = config.getClassBytes().get(mainClassName);
mainClassBytes = ClassInterfaceUtils.addInterface(mainClassBytes, "org.codehaus.groovy.transform.ASTTransformation");
mainClassBytes = ClassAnnotationUtils.setAnnotation(mainClassBytes, "org.codehaus.groovy.transform.GroovyASTTransformation");
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
try (JarOutputStream targetJar = new JarOutputStream(outputStream, new Manifest())) {
targetJar.putNextEntry(new JarEntry(mainClassName.replace('.', '/') + ".class"));
targetJar.write(mainClassBytes);
targetJar.closeEntry();
targetJar.putNextEntry(new JarEntry("META-INF/services/org.codehaus.groovy.transform.ASTTransformation"));
targetJar.write(mainClassName.getBytes(StandardCharsets.UTF_8));
targetJar.closeEntry();
}
return outputStream.toByteArray();
}
}