diff --git a/boot/src/main/java/com/reajason/javaweb/boot/dto/MemShellGenerateResponse.java b/boot/src/main/java/com/reajason/javaweb/boot/dto/MemShellGenerateResponse.java index f23c0d3f..ebeacb12 100644 --- a/boot/src/main/java/com/reajason/javaweb/boot/dto/MemShellGenerateResponse.java +++ b/boot/src/main/java/com/reajason/javaweb/boot/dto/MemShellGenerateResponse.java @@ -1,29 +1,18 @@ package com.reajason.javaweb.boot.dto; import com.reajason.javaweb.memshell.MemShellResult; +import lombok.AllArgsConstructor; import lombok.Data; import lombok.NoArgsConstructor; -import java.util.Map; - /** * @author ReaJason * @since 2024/12/18 */ @Data @NoArgsConstructor +@AllArgsConstructor public class MemShellGenerateResponse { private MemShellResult memShellResult; private String packResult; - private Map allPackResults; - - public MemShellGenerateResponse(MemShellResult memShellResult, String packResult) { - this.memShellResult = memShellResult; - this.packResult = packResult; - } - - public MemShellGenerateResponse(MemShellResult memShellResult, Map allPackResults) { - this.allPackResults = allPackResults; - this.memShellResult = memShellResult; - } } diff --git a/boot/src/main/java/com/reajason/javaweb/boot/dto/ProbeShellGenerateResponse.java b/boot/src/main/java/com/reajason/javaweb/boot/dto/ProbeShellGenerateResponse.java index 0ad792ae..b21a6ada 100644 --- a/boot/src/main/java/com/reajason/javaweb/boot/dto/ProbeShellGenerateResponse.java +++ b/boot/src/main/java/com/reajason/javaweb/boot/dto/ProbeShellGenerateResponse.java @@ -1,29 +1,18 @@ package com.reajason.javaweb.boot.dto; import com.reajason.javaweb.probe.ProbeShellResult; +import lombok.AllArgsConstructor; import lombok.Data; import lombok.NoArgsConstructor; -import java.util.Map; - /** * @author ReaJason * @since 2025/8/10 */ @Data @NoArgsConstructor +@AllArgsConstructor public class ProbeShellGenerateResponse { private ProbeShellResult probeShellResult; private String packResult; - private Map allPackResults; - - public ProbeShellGenerateResponse(ProbeShellResult probeShellResult, String packResult) { - this.probeShellResult = probeShellResult; - this.packResult = packResult; - } - - public ProbeShellGenerateResponse(ProbeShellResult probeShellResult, Map allPackResults) { - this.allPackResults = allPackResults; - this.probeShellResult = probeShellResult; - } } diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/MemShellResult.java b/generator/src/main/java/com/reajason/javaweb/memshell/MemShellResult.java index 48d1d544..d60266d2 100644 --- a/generator/src/main/java/com/reajason/javaweb/memshell/MemShellResult.java +++ b/generator/src/main/java/com/reajason/javaweb/memshell/MemShellResult.java @@ -52,7 +52,7 @@ public class MemShellResult { } public JarPackerConfig toJarPackerConfig() { - JarPackerConfig jarPackerConfig = new JarPackerConfig(); + JarPackerConfig jarPackerConfig = new JarPackerConfig<>(); jarPackerConfig.setMainClassName(injectorClassName); Map bytes = new HashMap<>(); bytes.put(shellClassName, shellBytes); @@ -65,7 +65,7 @@ public class MemShellResult { } public ClassPackerConfig toClassPackerConfig() { - ClassPackerConfig classPackerConfig = new ClassPackerConfig(); + ClassPackerConfig classPackerConfig = new ClassPackerConfig<>(); classPackerConfig.setClassName(injectorClassName); classPackerConfig.setClassBytes(injectorBytes); classPackerConfig.setClassBytesBase64Str(injectorBytesBase64Str); diff --git a/web/app/types/memshell.ts b/web/app/types/memshell.ts index 3890ea83..64d13949 100644 --- a/web/app/types/memshell.ts +++ b/web/app/types/memshell.ts @@ -139,7 +139,6 @@ export type PackerConfig = Array; export interface MemShellGenerateResponse { memShellResult: MemShellResult; packResult?: string; - allPackResults?: Map; } export interface APIErrorResponse {