feat: support appendLambdaSuffix

ReaJason · ReaJason · commit 78f6098e6c4f · 2025-12-06T14:09:40.000+08:00
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/MemShellGenerator.java b/generator/src/main/java/com/reajason/javaweb/memshell/MemShellGenerator.java
@@ -46,6 +46,11 @@ public static MemShellResult generate(ShellConfig shellConfig, InjectorConfig in
             injectorConfig.setInjectorClassName(CommonUtil.generateInjectorClassName());
         }
 
+        if (shellConfig.isLambdaSuffix()) {
+            shellToolConfig.setShellClassName(CommonUtil.appendLambdaSuffix(shellToolConfig.getShellClassName()));
+            injectorConfig.setInjectorClassName(CommonUtil.appendLambdaSuffix(injectorConfig.getInjectorClassName()));
+        }
+
         byte[] shellBytes = ShellToolFactory.generateBytes(shellConfig, shellToolConfig);
 
         injectorConfig.setInjectorClass(injectorClass);
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/config/ShellConfig.java b/generator/src/main/java/com/reajason/javaweb/memshell/config/ShellConfig.java
@@ -61,6 +61,12 @@ public class ShellConfig {
     @Builder.Default
     private boolean shrink = false;
 
+    /**
+     * 追加 Lambda 类名后缀
+     */
+    @Builder.Default
+    private boolean lambdaSuffix = false;
+
     public boolean isDebugOff() {
         return !debug;
     }
diff --git a/generator/src/main/java/com/reajason/javaweb/probe/ProbeShellGenerator.java b/generator/src/main/java/com/reajason/javaweb/probe/ProbeShellGenerator.java
@@ -15,6 +15,9 @@ public static ProbeShellResult generate(ProbeConfig probeConfig, ProbeContentCon
         if (StringUtils.isBlank(probeConfig.getShellClassName())) {
             probeConfig.setShellClassName(CommonUtil.generateInjectorClassName());
         }
+        if (probeConfig.isLambdaSuffix()) {
+            probeConfig.setShellClassName(CommonUtil.appendLambdaSuffix(probeConfig.getShellClassName()));
+        }
         byte[] bytes = probeConfig.getProbeMethod().generateBytes(probeConfig, contentConfig);
         return ProbeShellResult.builder()
                 .shellClassName(probeConfig.getShellClassName())
diff --git a/generator/src/main/java/com/reajason/javaweb/probe/config/ProbeConfig.java b/generator/src/main/java/com/reajason/javaweb/probe/config/ProbeConfig.java
@@ -53,6 +53,12 @@ public class ProbeConfig {
     @Builder.Default
     private boolean staticInitialize = false;
 
+    /**
+     * 追加 Lambda 类名后缀
+     */
+    @Builder.Default
+    private boolean lambdaSuffix = false;
+
     public boolean isDebugOff() {
         return !debug;
     }
diff --git a/generator/src/main/java/com/reajason/javaweb/utils/CommonUtil.java b/generator/src/main/java/com/reajason/javaweb/utils/CommonUtil.java
@@ -101,6 +101,13 @@ public static String generateInjectorClassName() {
         return getRandomPackageName() + "." + INJECTOR_CLASS_NAMES[new Random().nextInt(INJECTOR_CLASS_NAMES.length)];
     }
 
+    public static String appendLambdaSuffix(String className) {
+        if (className.contains("$Lambda$")) {
+            return className;
+        }
+        return className + "$Proxy0$$Lambda$1";
+    }
+
     public static String generateShellClassName(String server, String shellType) {
         String packageName;
         switch (server) {
diff --git a/web/app/components/memshell/main-config-card.tsx b/web/app/components/memshell/main-config-card.tsx
@@ -358,7 +358,7 @@ export default function MainConfigCard({
               )}
             />
           </div>
-          <div className="flex gap-4 mt-4 flex-col sm:flex-row xl:grid xl:grid-cols-2 2xl:flex 2xl:flex-row">
+          <div className="flex gap-4 mt-4 flex-col sm:flex-row xl:grid xl:grid-cols-2 2xl:grid 2xl:grid-cols-3">
             <FormField
               control={form.control}
               name="debug"
@@ -391,6 +391,24 @@ export default function MainConfigCard({
                 </FormItem>
               )}
             />
+            <FormField
+              control={form.control}
+              name="lambdaSuffix"
+              render={({ field }) => (
+                <FormItem className="flex items-center space-x-2  space-y-0">
+                  <FormControl>
+                    <Switch
+                      id="lambdaSuffix"
+                      checked={field.value}
+                      onCheckedChange={field.onChange}
+                    />
+                  </FormControl>
+                  <Label htmlFor="lambdaSuffix">
+                    {t("common:lambdaSuffix")}
+                  </Label>
+                </FormItem>
+              )}
+            />
             <FormField
               control={form.control}
               name="shrink"
diff --git a/web/app/components/probeshell/main-config-card.tsx b/web/app/components/probeshell/main-config-card.tsx
@@ -279,7 +279,7 @@ export default function MainConfigCard({ form, servers }: MainConfigCardProps) {
 
   const SwitchGroup = useMemo(
     () => (
-      <div className="flex gap-4 mt-4 flex-col sm:flex-row">
+      <div className="flex gap-4 mt-4 flex-col sm:flex-row xl:grid xl:grid-cols-2 2xl:grid 2xl:grid-cols-3">
         <FormField
           control={form.control}
           name="debug"
@@ -312,6 +312,22 @@ export default function MainConfigCard({ form, servers }: MainConfigCardProps) {
             </FormItem>
           )}
         />
+        <FormField
+          control={form.control}
+          name="lambdaSuffix"
+          render={({ field }) => (
+            <FormItem className="flex items-center space-x-2  space-y-0">
+              <FormControl>
+                <Switch
+                  id="lambdaSuffix"
+                  checked={field.value}
+                  onCheckedChange={field.onChange}
+                />
+              </FormControl>
+              <Label htmlFor="lambdaSuffix">{t("common:lambdaSuffix")}</Label>
+            </FormItem>
+          )}
+        />
         <FormField
           control={form.control}
           name="shrink"
diff --git a/web/app/i18n/common/en.json b/web/app/i18n/common/en.json
@@ -35,5 +35,7 @@
   "usage": "Usage",
   "version.updateAvailable": "Update Available",
   "version.updateAvailableTooltip": "Click to Open Github Release Page ( v{{currentVersion}} -> v{{latestVersion}})",
-  "shellTool": "Shell Tool"
+  "shellTool": "Shell Tool",
+  "lambdaSuffix": "LambdaSuffix",
+  "probe": "Probe Mode"
 }
diff --git a/web/app/i18n/common/zh-CN.json b/web/app/i18n/common/zh-CN.json
@@ -35,5 +35,7 @@
   "usage": "使用指南",
   "version.updateAvailable": "有可用升级",
   "version.updateAvailableTooltip": "点击前往 GitHub Release ( v{{currentVersion}} -> v{{latestVersion}})",
-  "shellTool": "内存马工具"
+  "shellTool": "内存马工具",
+  "lambdaSuffix": "Lambda 类名后缀",
+  "probe": "回显模式"
 }
diff --git a/web/app/types/memshell.ts b/web/app/types/memshell.ts
@@ -8,6 +8,8 @@ export interface ShellConfig {
   byPassJavaModule?: boolean;
   obfuscate?: boolean;
   shrink?: boolean;
+  probe?: boolean;
+  lambdaSuffix?:boolean;
 }
 
 export interface ShellToolConfig {
diff --git a/web/app/types/probeshell.ts b/web/app/types/probeshell.ts
@@ -17,6 +17,7 @@ export interface ProbeConfig {
   byPassJavaModule?: boolean;
   shrink?: boolean;
   staticInitialize?: boolean;
+  lambdaSuffix?: boolean;
 }
 
 export interface ProbeContentConfig {
diff --git a/web/app/types/schema.ts b/web/app/types/schema.ts
@@ -26,6 +26,8 @@ export const memShellFormSchema = yup.object({
   injectorClassName: yup.string().optional(),
   packingMethod: yup.string().required().min(1),
   shrink: yup.boolean().optional(),
+  lambdaSuffix: yup.boolean().optional(),
+  probe: yup.boolean().optional(),
   shellClassBase64: yup.string().optional(),
   encryptor: yup.string().optional(),
 });
@@ -165,6 +167,7 @@ export const probeShellFormSchema = yup.object().shape({
   byPassJavaModule: yup.boolean().optional(),
   shrink: yup.boolean().optional(),
   staticInitialize: yup.boolean().optional(),
+  lambdaSuffix: yup.boolean().optional(),
 });
 
 type ProbeValidationResult = ResolverResult<ProbeShellFormSchema>;
diff --git a/web/app/utils/transformer.ts b/web/app/utils/transformer.ts
@@ -19,6 +19,8 @@ export function transformToPostData(formValue: MemShellFormSchema) {
     targetJreVersion: formValue.targetJdkVersion,
     byPassJavaModule: formValue.byPassJavaModule,
     shrink: formValue.shrink,
+    lambdaSuffix: formValue.lambdaSuffix,
+    probe: formValue.probe
   };
   const shellToolConfig: ShellToolConfig = {
     shellClassName: formValue.shellClassName,
@@ -56,6 +58,7 @@ export function transformToProbePostData(formValue: ProbeShellFormSchema) {
     debug: formValue.debug,
     byPassJavaModule: formValue.byPassJavaModule,
     staticInitialize: formValue.staticInitialize,
+    lambdaSuffix: formValue.lambdaSuffix,
   };
   const probeContentConfig: ProbeContentConfig = {
     host: formValue.host,

Original file line number	Diff line number	Diff line change
`@@ -35,5 +35,7 @@`
`35`	`35`	`"usage": "Usage",`
`36`	`36`	`"version.updateAvailable": "Update Available",`
`37`	`37`	`"version.updateAvailableTooltip": "Click to Open Github Release Page ( v{{currentVersion}} -> v{{latestVersion}})",`
`38`		`- "shellTool": "Shell Tool"`
	`38`	`+ "shellTool": "Shell Tool",`
	`39`	`+ "lambdaSuffix": "LambdaSuffix",`
	`40`	`+ "probe": "Probe Mode"`
`39`	`41`	`}`
Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,8 @@ export interface ShellConfig {`
`8`	`8`	`byPassJavaModule?: boolean;`
`9`	`9`	`obfuscate?: boolean;`
`10`	`10`	`shrink?: boolean;`
	`11`	`+ probe?: boolean;`
	`12`	`+ lambdaSuffix?:boolean;`
`11`	`13`	`}`
`12`	`14`
`13`	`15`	`export interface ShellToolConfig {`