feat(web): support download button for some packers

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

@@ -1,6 +1,8 @@
+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,
@@ -8,32 +10,67 @@ import {
   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 methods = Object.keys(allPackResults ?? {});
-    const firstMethod = methods[0];
-    setSelectedMethod(firstMethod);
-    setPackResult(
-      allPackResults?.[firstMethod as keyof typeof allPackResults] ?? "",
-    );
-  }, [allPackResults]);
+    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
@@ -66,6 +103,22 @@ export function MultiPackResult({
           <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"}

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

@@ -25,6 +25,7 @@ export function ResultComponent({
       <MultiPackResult
         allPackResults={allPackResults}
         packMethod={packMethod}
+        shellClassName={generateResult?.injectorClassName}
       />
     );
   }

web/src/components/probeshell/shell-result.tsx

@@ -40,6 +40,7 @@ export default function ShellResult({
         {allPackResults && (
           <MultiPackResult
             allPackResults={allPackResults}
+            shellClassName={generateResult?.shellClassName}
             packMethod={packMethod}
             height={height}
           />