Merge pull request #2156 from yaklang/song/feat/webFuzzerInsertLabel

Song/feat/web fuzzer insert label
yaklang · Nov 29, 2024 · 2ca8273 · 2ca8273
2 parents 3667b1f + 9b0a049
commit 2ca8273
Show file tree

Hide file tree

Showing 14 changed files with 661 additions and 813 deletions.
diff --git a/app/main/handlers/httpFuzzer.js b/app/main/handlers/httpFuzzer.js
@@ -82,6 +82,37 @@ module.exports = (win, getClient) => {
         })
     })
 
+
+    const asyncGetAllFuzztagInfo = (params) => {
+        return new Promise((resolve, reject) => {
+            getClient().GetAllFuzztagInfo(params, (err, data) => {
+                if (err) {
+                    reject(err)
+                    return
+                }
+                resolve(data)
+            })
+        })
+    }
+    ipcMain.handle("GetAllFuzztagInfo", async (e, params) => {
+        return await asyncGetAllFuzztagInfo(params)
+    })
+
+    const asyncGenerateFuzztag = (params) => {
+        return new Promise((resolve, reject) => {
+            getClient().GenerateFuzztag(params, (err, data) => {
+                if (err) {
+                    reject(err)
+                    return
+                }
+                resolve(data)
+            })
+        })
+    }
+    ipcMain.handle("GenerateFuzztag", async (e, params) => {
+        return await asyncGenerateFuzztag(params)
+    })
+
     const handlerHelper = require("./handleStreamWithContext");
     const streamHTTPFuzzerMap = new Map();
     ipcMain.handle("cancel-HTTPFuzzer", handlerHelper.cancelHandler(streamHTTPFuzzerMap));