feat: support probe shell generation

web/src/components/memshell/results/agent.tsx

@@ -0,0 +1,69 @@
+import { ScrollTextIcon } from "lucide-react";
+import { useTranslation } from "react-i18next";
+import { Button } from "@/components/ui/button";
+import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card";
+import { Separator } from "@/components/ui/separator";
+import { downloadBytes, formatBytes } from "@/lib/utils";
+import type { GenerateResult } from "@/types/shell";
+
+export function AgentResult({
+  packMethod,
+  packResult,
+  generateResult,
+}: Readonly<{ packMethod: string; packResult: string; generateResult?: GenerateResult }>) {
+  const { t } = useTranslation();
+  const isPureAgent = packMethod === "AgentJar";
+  return (
+    <Card>
+      <CardHeader>
+        <CardTitle className="text-md flex items-center gap-2">
+          <ScrollTextIcon className="h-5" />
+          <span>{t("generateResult.usage")}</span>
+        </CardTitle>
+      </CardHeader>
+      <CardContent>
+        <ol className="list-decimal list-inside space-y-4 text-sm">
+          <li className="flex items-center justify-between">
+            <span>
+              {t("download")} MemShellAgent.jar ({formatBytes(atob(packResult).length)})
+            </span>
+            <Button
+              size="sm"
+              variant="outline"
+              className="w-28"
+              type="button"
+              onClick={() =>
+                downloadBytes(
+                  packResult,
+                  undefined,
+                  `${generateResult?.shellConfig.server}${generateResult?.shellConfig.shellTool}MemShellAgent`,
+                )
+              }
+            >
+              {t("download")}
+            </Button>
+          </li>
+          {isPureAgent && (
+            <li className="flex items-center justify-between">
+              <span>{t("tips.download-jattach")}</span>
+              <Button
+                size="sm"
+                variant="outline"
+                className="w-28"
+                type="button"
+                onClick={() => window.open("https://github.com/jattach/jattach/releases")}
+              >
+                {t("download")}
+              </Button>
+            </li>
+          )}
+          <Separator />
+          <li>{isPureAgent ? t("tips.agent-move-to-target") : t("tips.agent-move-to-target1")}</li>
+          <li>{t("tips.get-pid")}</li>
+          <li>{isPureAgent ? t("tips.execute-command") : t("tips.execute-command1")}</li>
+          <li>{t("tips.try-to-use-shell")}</li>
+        </ol>
+      </CardContent>
+    </Card>
+  );
+}

web/src/components/memshell/results/basic-info.tsx

@@ -0,0 +1,141 @@
+import { FileTextIcon } from "lucide-react";
+import { useTranslation } from "react-i18next";
+import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card";
+import { Separator } from "@/components/ui/separator";
+import { shouldHidden } from "@/lib/utils";
+import {
+  type AntSwordShellToolConfig,
+  type BehinderShellToolConfig,
+  type CommandShellToolConfig,
+  type GenerateResult,
+  type GodzillaShellToolConfig,
+  type NeoreGeorgShellToolConfig,
+  ShellToolType,
+  type Suo5ShellToolConfig,
+} from "@/types/shell";
+import { CopyableField } from "../../copyable-field";
+import { FeedbackAlert } from "./feedback-alert";
+
+export function BasicInfo({ generateResult }: Readonly<{ generateResult?: GenerateResult }>) {
+  const { t } = useTranslation();
+  return (
+    <Card>
+      <CardHeader>
+        <CardTitle className="flex items-center justify-between">
+          <div className="text-md flex items-center gap-2">
+            <FileTextIcon className="h-5" />
+            <span>{t("generateResult.basicInfo")}</span>
+          </div>
+          <FeedbackAlert />
+        </CardTitle>
+      </CardHeader>
+      <CardContent>
+        <div className="grid grid-cols-1 md:grid-cols-2 gap-2">
+          <CopyableField label={t("mainConfig.server")} text={generateResult?.shellConfig.server} />
+          <CopyableField label={t("mainConfig.shellTool")} text={generateResult?.shellConfig.shellTool} />
+          <CopyableField label={t("mainConfig.shellMountType")} text={generateResult?.shellConfig.shellType} />
+          {!shouldHidden(generateResult?.shellConfig?.shellType) && (
+            <CopyableField
+              label={t("mainConfig.urlPattern")}
+              text={generateResult?.injectorConfig.urlPattern}
+              value={generateResult?.injectorConfig.urlPattern}
+            />
+          )}
+        </div>
+        {generateResult?.shellConfig.shellTool !== ShellToolType.Custom && <Separator className="my-1" />}
+        <div className="grid grid-cols-1 md:grid-cols-2 gap-2">
+          {generateResult?.shellConfig.shellTool === ShellToolType.Behinder && (
+            <>
+              <CopyableField label={t("shellToolConfig.behinderScriptType")} text="jsp" />
+              <CopyableField
+                label={t("shellToolConfig.behinderEncryptType")}
+                text={t("shellToolConfig.behinderDefaultEncryptType")}
+              />
+              <CopyableField
+                label={t("shellToolConfig.behinderPass")}
+                text={(generateResult?.shellToolConfig as BehinderShellToolConfig).pass}
+                value={(generateResult?.shellToolConfig as BehinderShellToolConfig).pass}
+              />
+              <CopyableField
+                label={t("shellToolConfig.customHeader")}
+                text={`${(generateResult?.shellToolConfig as BehinderShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as BehinderShellToolConfig).headerValue}`}
+                value={`${(generateResult?.shellToolConfig as BehinderShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as BehinderShellToolConfig).headerValue}`}
+              />
+            </>
+          )}
+          {generateResult?.shellConfig.shellTool === ShellToolType.Godzilla && (
+            <>
+              <CopyableField
+                label={t("shellToolConfig.pass")}
+                text={(generateResult?.shellToolConfig as GodzillaShellToolConfig).pass}
+                value={(generateResult?.shellToolConfig as GodzillaShellToolConfig).pass}
+              />
+              <CopyableField
+                label={t("shellToolConfig.key")}
+                text={(generateResult?.shellToolConfig as GodzillaShellToolConfig).key}
+                value={(generateResult?.shellToolConfig as GodzillaShellToolConfig).key}
+              />
+              <CopyableField label={t("shellToolConfig.godzillaEncryptor")} text="JAVA_AES_BASE64" />
+              <CopyableField
+                label={t("shellToolConfig.godzillaHeader")}
+                text={`${(generateResult?.shellToolConfig as GodzillaShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as GodzillaShellToolConfig).headerValue}`}
+                value={`${(generateResult?.shellToolConfig as GodzillaShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as GodzillaShellToolConfig).headerValue}`}
+              />
+            </>
+          )}
+          {generateResult?.shellConfig.shellTool === ShellToolType.Command && (
+            <CopyableField
+              label={t("shellToolConfig.paramName")}
+              text={(generateResult?.shellToolConfig as CommandShellToolConfig).paramName}
+              value={(generateResult?.shellToolConfig as CommandShellToolConfig).paramName}
+            />
+          )}
+          {generateResult?.shellConfig.shellTool === ShellToolType.Suo5 && (
+            <CopyableField
+              label={t("shellToolConfig.suo5Header")}
+              text={`${(generateResult?.shellToolConfig as Suo5ShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as Suo5ShellToolConfig).headerValue}`}
+              value={`${(generateResult?.shellToolConfig as Suo5ShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as Suo5ShellToolConfig).headerValue}`}
+            />
+          )}
+          {generateResult?.shellConfig.shellTool === ShellToolType.AntSword && (
+            <>
+              <CopyableField
+                label={t("shellToolConfig.antSwordPass")}
+                text={(generateResult?.shellToolConfig as AntSwordShellToolConfig).pass}
+                value={(generateResult?.shellToolConfig as AntSwordShellToolConfig).pass}
+              />
+              <CopyableField
+                label={t("shellToolConfig.httpHeader")}
+                text={`${(generateResult?.shellToolConfig as AntSwordShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as AntSwordShellToolConfig).headerValue}`}
+                value={`${(generateResult?.shellToolConfig as AntSwordShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as AntSwordShellToolConfig).headerValue}`}
+              />
+            </>
+          )}
+          {generateResult?.shellConfig.shellTool === ShellToolType.NeoreGeorg && (
+            <>
+              <CopyableField label={t("shellToolConfig.neoreGeorgKey")} text="key" value="key" />
+              <CopyableField
+                label={t("shellToolConfig.neoreGeorgHeader")}
+                text={`${(generateResult?.shellToolConfig as NeoreGeorgShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as NeoreGeorgShellToolConfig).headerValue}`}
+                value={`${(generateResult?.shellToolConfig as NeoreGeorgShellToolConfig).headerName}: ${(generateResult?.shellToolConfig as NeoreGeorgShellToolConfig).headerValue}`}
+              />
+            </>
+          )}
+        </div>
+        <Separator className="my-1" />
+        <div className="grid grid-cols-1 md:grid-cols-2 gap-2">
+          <CopyableField
+            label={t("mainConfig.injectorClassName")}
+            value={generateResult?.injectorClassName}
+            text={`${generateResult?.injectorClassName} (${generateResult?.injectorSize} bytes)`}
+          />
+          <CopyableField
+            label={t("mainConfig.shellClassName")}
+            value={generateResult?.shellClassName}
+            text={`${generateResult?.shellClassName} (${generateResult?.shellSize} bytes)`}
+          />
+        </div>
+      </CardContent>
+    </Card>
+  );
+}

web/src/components/memshell/results/feedback-alert.tsx

@@ -0,0 +1,50 @@
+import { CircleHelpIcon } from "lucide-react";
+import { useTranslation } from "react-i18next";
+import {
+  AlertDialog,
+  AlertDialogAction,
+  AlertDialogCancel,
+  AlertDialogContent,
+  AlertDialogDescription,
+  AlertDialogFooter,
+  AlertDialogHeader,
+  AlertDialogTitle,
+  AlertDialogTrigger,
+} from "@/components/ui/alert-dialog";
+import { Button } from "@/components/ui/button";
+
+export function FeedbackAlert() {
+  const { t } = useTranslation();
+  return (
+    <AlertDialog>
+      <AlertDialogTrigger asChild>
+        <Button variant="outline" type="button">
+          <CircleHelpIcon /> {t("shellNotWork.title")}
+        </Button>
+      </AlertDialogTrigger>
+      <AlertDialogContent>
+        <AlertDialogHeader>
+          <AlertDialogTitle>{t("shellNotWork.title")}</AlertDialogTitle>
+          <AlertDialogDescription>
+            <ol>
+              <li>{t("shellNotWork.step1")}</li>
+              <li>{t("shellNotWork.step2")}</li>
+            </ol>
+          </AlertDialogDescription>
+        </AlertDialogHeader>
+        <AlertDialogFooter>
+          <AlertDialogCancel>{t("cancel")}</AlertDialogCancel>
+          <AlertDialogAction
+            onClick={() =>
+              window.open(
+                "https://github.com/ReaJason/MemShellParty/issues/new?template=%E5%86%85%E5%AD%98%E9%A9%AC%E7%94%9F%E6%88%90-bug-%E4%B8%8A%E6%8A%A5.md",
+              )
+            }
+          >
+            {t("feedback")}
+          </AlertDialogAction>
+        </AlertDialogFooter>
+      </AlertDialogContent>
+    </AlertDialog>
+  );
+}

web/src/components/memshell/results/jar-result.tsx

@@ -0,0 +1,27 @@
+import { useTranslation } from "react-i18next";
+import { Button } from "@/components/ui/button";
+import { downloadBytes } from "@/lib/utils";
+import type { GenerateResult } from "@/types/shell";
+
+export function JarResult({
+  packResult,
+  generateResult,
+}: Readonly<{ packResult: string; generateResult?: GenerateResult }>) {
+  const { t } = useTranslation();
+  return (
+    <div className="flex items-center justify-center">
+      <Button
+        type="button"
+        onClick={() =>
+          downloadBytes(
+            packResult,
+            undefined,
+            `${generateResult?.shellConfig.server}${generateResult?.shellConfig.shellTool}MemShell`,
+          )
+        }
+      >
+        {t("download")} Jar
+      </Button>
+    </div>
+  );
+}

web/src/components/memshell/results/multi-packer.tsx

@@ -0,0 +1,58 @@
+import { useEffect, useState } from "react";
+import { useTranslation } from "react-i18next";
+import CodeViewer from "@/components/code-viewer";
+import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from "@/components/ui/select";
+export function MultiPackResult({
+  allPackResults,
+  packMethod,
+}: Readonly<{
+  allPackResults: object | undefined;
+  packMethod: string;
+}>) {
+  const showCode = packMethod === "JSP";
+  const {t} = useTranslation();
+  const packMethods = Object.keys(allPackResults ?? {});
+  const [selectedMethod, setSelectedMethod] = useState(packMethods[0]);
+  const [packResult, setPackResult] = useState(allPackResults?.[selectedMethod as keyof typeof allPackResults] ?? "");
+
+  useEffect(() => {
+    const methods = Object.keys(allPackResults ?? {});
+    const firstMethod = methods[0];
+    setSelectedMethod(firstMethod);
+    setPackResult(allPackResults?.[firstMethod as keyof typeof allPackResults] ?? "");
+  }, [allPackResults]);
+
+  return (
+    <CodeViewer
+      code={packResult ?? ""}
+      header={
+        <div className="flex items-center justify-between text-xs gap-2">
+          <Select
+            onValueChange={(value) => {
+              setSelectedMethod(value);
+              setPackResult(allPackResults?.[value as keyof typeof allPackResults] ?? "");
+            }}
+            value={selectedMethod}
+          >
+            <SelectTrigger className="h-7 text-xs [&_svg]:h-4 [&_svg]:w-4">
+              <span className="text-muted-foreground">{t("packageConfig.title")}:&nbsp;</span>
+              <SelectValue />
+            </SelectTrigger>
+            <SelectContent>
+              {packMethods.map((method) => (
+                <SelectItem key={method} value={method} className="text-xs">
+                  {method}
+                </SelectItem>
+              ))}
+            </SelectContent>
+          </Select>
+          <span className="text-muted-foreground">({packResult?.length})</span>
+        </div>
+      }
+      wrapLongLines={!showCode}
+      showLineNumbers={showCode}
+      language={showCode ? "java" : "text"}
+      height={350}
+    />
+  );
+}

web/src/components/memshell/results/result-component.tsx

@@ -0,0 +1,53 @@
+import { useTranslation } from "react-i18next";
+import CodeViewer from "@/components/code-viewer";
+import type { GenerateResult } from "@/types/shell";
+import { AgentResult } from "./agent";
+import { JarResult } from "./jar-result";
+import { MultiPackResult } from "./multi-packer";
+
+export function ResultComponent({
+  packResult,
+  allPackResults,
+  packMethod,
+  generateResult,
+}: Readonly<{
+  packResult: string | undefined;
+  allPackResults: Map<string, string> | undefined;
+  packMethod: string;
+  generateResult?: GenerateResult;
+}>) {
+  const showCode = packMethod === "JSP";
+  const isAgent = packMethod.startsWith("Agent");
+  const isJar = packMethod === "Jar";
+  const { t } =  useTranslation();
+  if (allPackResults) {
+    return <MultiPackResult allPackResults={allPackResults} packMethod={packMethod} />;
+  }
+
+  if (isAgent) {
+    return <AgentResult packMethod={packMethod} packResult={packResult ?? ""} generateResult={generateResult} />;
+  }
+  if (isJar) {
+    return <JarResult packResult={packResult ?? ""} generateResult={generateResult} />;
+  }
+  if (!isAgent && !isJar) {
+    return (
+      <CodeViewer
+        code={packResult ?? ""}
+        header={
+          <div className="flex items-center justify-between text-xs gap-2">
+            <span>
+              {t("packageConfig.title")}：{packMethod}
+            </span>
+            <span className="text-muted-foreground">({packResult?.length})</span>
+          </div>
+        }
+        wrapLongLines={!showCode}
+        showLineNumbers={showCode}
+        language={showCode ? "java" : "text"}
+        height={350}
+      />
+    );
+  }
+  return null;
+}