feat: support probe shell generation

boot/src/main/java/com/reajason/javaweb/boot/controller/GeneratorController.java

@@ -3,7 +3,7 @@ package com.reajason.javaweb.boot.controller;
 import com.reajason.javaweb.boot.dto.GenerateRequest;
 import com.reajason.javaweb.boot.dto.GenerateResponse;
 import com.reajason.javaweb.memshell.MemShellGenerator;
-import com.reajason.javaweb.memshell.config.GenerateResult;
+import com.reajason.javaweb.memshell.MemShellResult;
 import com.reajason.javaweb.memshell.config.InjectorConfig;
 import com.reajason.javaweb.memshell.config.ShellConfig;
 import com.reajason.javaweb.memshell.config.ShellToolConfig;
@@ -27,7 +27,7 @@ public class GeneratorController {
         ShellConfig shellConfig = request.getShellConfig();
         ShellToolConfig shellToolConfig = request.parseShellToolConfig();
         InjectorConfig injectorConfig = request.getInjectorConfig();
-        GenerateResult generateResult = MemShellGenerator.generate(shellConfig, injectorConfig, shellToolConfig);
+        MemShellResult generateResult = MemShellGenerator.generate(shellConfig, injectorConfig, shellToolConfig);
         Packer packer = request.getPacker().getInstance();
         if (packer instanceof JarPacker) {
             return new GenerateResponse(generateResult, Base64.getEncoder().encodeToString(((JarPacker) packer).packBytes(generateResult.toJarPackerConfig())));

boot/src/main/java/com/reajason/javaweb/boot/controller/ProbeGeneratorController.java

@@ -0,0 +1,33 @@
+package com.reajason.javaweb.boot.controller;
+
+import com.reajason.javaweb.boot.dto.ProbeGenerateRequest;
+import com.reajason.javaweb.boot.dto.ProbeGenerateResponse;
+import com.reajason.javaweb.packer.AggregatePacker;
+import com.reajason.javaweb.packer.Packer;
+import com.reajason.javaweb.probe.ProbeGenerator;
+import com.reajason.javaweb.probe.ProbeResult;
+import com.reajason.javaweb.probe.config.ProbeConfig;
+import com.reajason.javaweb.probe.config.ProbeContentConfig;
+import org.springframework.web.bind.annotation.*;
+
+/**
+ * @author ReaJason
+ * @since 2025/8/10
+ */
+@RestController
+@RequestMapping("/probe/generate")
+@CrossOrigin("*")
+public class ProbeGeneratorController {
+    @PostMapping
+    public ProbeGenerateResponse generate(@RequestBody ProbeGenerateRequest request) {
+        ProbeConfig probeConfig = request.getProbeConfig();
+        ProbeContentConfig probeContentConfig = request.parseProbeContentConfig();
+        ProbeResult generateResult = ProbeGenerator.generate(probeConfig, probeContentConfig);
+        Packer packer = request.getPacker().getInstance();
+        if (packer instanceof AggregatePacker) {
+            return new ProbeGenerateResponse(generateResult, ((AggregatePacker) packer).packAll(generateResult.toClassPackerConfig()));
+        } else {
+            return new ProbeGenerateResponse(generateResult, packer.pack(generateResult.toClassPackerConfig()));
+        }
+    }
+}

boot/src/main/java/com/reajason/javaweb/boot/dto/GenerateResponse.java

@@ -1,6 +1,6 @@
 package com.reajason.javaweb.boot.dto;
 
-import com.reajason.javaweb.memshell.config.GenerateResult;
+import com.reajason.javaweb.memshell.MemShellResult;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 
@@ -13,16 +13,16 @@ import java.util.Map;
 @Data
 @NoArgsConstructor
 public class GenerateResponse {
-    private GenerateResult generateResult;
+    private MemShellResult generateResult;
     private String packResult;
     private Map<String, String> allPackResults;
 
-    public GenerateResponse(GenerateResult generateResult, String packResult) {
+    public GenerateResponse(MemShellResult generateResult, String packResult) {
         this.generateResult = generateResult;
         this.packResult = packResult;
     }
 
-    public GenerateResponse(GenerateResult generateResult, Map<String, String> allPackResults) {
+    public GenerateResponse(MemShellResult generateResult, Map<String, String> allPackResults) {
         this.allPackResults = allPackResults;
         this.generateResult = generateResult;
     }

boot/src/main/java/com/reajason/javaweb/boot/dto/ProbeGenerateRequest.java

@@ -0,0 +1,44 @@
+package com.reajason.javaweb.boot.dto;
+
+import com.reajason.javaweb.packer.Packers;
+import com.reajason.javaweb.probe.config.*;
+import lombok.Data;
+
+/**
+ * @author ReaJason
+ * @since 2025/8/10
+ */
+@Data
+public class ProbeGenerateRequest {
+    private ProbeConfig probeConfig;
+    private ProbeContentConfigDTO probeContentConfig;
+    private Packers packer;
+
+    @Data
+    static class ProbeContentConfigDTO {
+        private String host;
+        private int seconds;
+        private String server;
+        private String sleepServer;
+        private String reqParamName;
+        private String reqHeaderName;
+    }
+
+    public ProbeContentConfig parseProbeContentConfig() {
+        return switch (probeConfig.getProbeMethod()) {
+            case DNSLog -> DnsLogConfig.builder()
+                    .host(probeContentConfig.host)
+                    .build();
+            case Sleep -> SleepConfig.builder()
+                    .seconds(probeContentConfig.seconds)
+                    .server(probeContentConfig.sleepServer)
+                    .build();
+            case ResponseBody -> ResponseBodyConfig.builder()
+                    .reqParamName(probeContentConfig.reqParamName)
+                    .reqHeaderName(probeContentConfig.reqHeaderName)
+                    .server(probeContentConfig.server)
+                    .build();
+            default -> throw new UnsupportedOperationException("unknown probe method: " + probeConfig.getProbeMethod());
+        };
+    }
+}

boot/src/main/java/com/reajason/javaweb/boot/dto/ProbeGenerateResponse.java

@@ -0,0 +1,29 @@
+package com.reajason.javaweb.boot.dto;
+
+import com.reajason.javaweb.probe.ProbeResult;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import java.util.Map;
+
+/**
+ * @author ReaJason
+ * @since 2025/8/10
+ */
+@Data
+@NoArgsConstructor
+public class ProbeGenerateResponse {
+    private ProbeResult probeResult;
+    private String packResult;
+    private Map<String, String> allPackResults;
+
+    public ProbeGenerateResponse(ProbeResult probeResult, String packResult) {
+        this.probeResult = probeResult;
+        this.packResult = packResult;
+    }
+
+    public ProbeGenerateResponse(ProbeResult probeResult, Map<String, String> allPackResults) {
+        this.allPackResults = allPackResults;
+        this.probeResult = probeResult;
+    }
+}