(feat) add immediate mode probe

sensepost · Sep 22, 2024 · bb5cf4f · bb5cf4f
1 parent 1db81b3
commit bb5cf4f
Show file tree

Hide file tree

Showing 6 changed files with 356 additions and 130 deletions.
diff --git a/web/ui/dist/index.html b/web/ui/dist/index.html
@@ -6,8 +6,8 @@
   <link rel="icon" type="image/svg+xml" href="/favico.ico" />
   <meta name="viewport" content="width=device-width, initial-scale=1.0" />
   <title>gowitness - a golang screenshotting tool by @leonjza</title>
-  <script type="module" crossorigin src="/assets/index-1RGl8SFg.js"></script>
-  <link rel="stylesheet" crossorigin href="/assets/index-BctVAbJi.css">
+  <script type="module" crossorigin src="/assets/index-DVkfmUJF.js"></script>
+  <link rel="stylesheet" crossorigin href="/assets/index-iQg3vwxl.css">
 </head>
 
 <body>

diff --git a/web/ui/src/lib/api/api.ts b/web/ui/src/lib/api/api.ts
@@ -54,6 +54,10 @@ const endpoints = {
   submit: {
     path: `/submit`,
     returnas: "" as string
+  },
+  submitsingle: {
+    path: `/submit/single`,
+    returnas: {} as detail
   }
 };
 

diff --git a/web/ui/src/lib/api/types.ts b/web/ui/src/lib/api/types.ts
@@ -160,6 +160,7 @@ interface detail {
   is_pdf: boolean;
   failed: boolean;
   failed_reason: string;
+  screenshot: string;
   tls: tls;
   technologies: technology[];
   headers: header[];

diff --git a/web/ui/src/main.tsx b/web/ui/src/main.tsx
@@ -16,7 +16,7 @@ import JobSubmissionPage from '@/pages/submit/Submit';
 import { searchAction } from '@/pages/search/action';
 import { searchLoader } from '@/pages/search/loader';
 import { deleteAction } from '@/pages/detail/actions';
-import { submitAction } from '@/pages/submit/action';
+import { submitImmediateAction, submitJobAction } from '@/pages/submit/action';
 
 const router = createBrowserRouter([
   {
@@ -50,7 +50,20 @@ const router = createBrowserRouter([
       {
         path: 'submit',
         element: <JobSubmissionPage />,
-        action: submitAction,
+        action: async ({ request }) => {
+          const formData = await request.formData();
+          const action = formData.get('action');
+
+          switch (action) {
+            case 'job':
+              return submitJobAction({ formData });
+            case 'immediate':
+              return submitImmediateAction({ formData });
+
+            default:
+              throw new Error('unknown action for job submit route');
+          }
+        },
       },
     ]
   }