Merge pull request yaklang#1584 from yaklang/song/feat/webFuzzerTable…

…MoreThan20000Data Song/feat/web fuzzer table more than20000 data
Box-CFT · Apr 18, 2024 · d2dfc42 · d2dfc42
2 parents 77a992d + a04f702
commit d2dfc42
Show file tree

Hide file tree

Showing 16 changed files with 1,481 additions and 1,129 deletions.
diff --git a/app/protos/grpc.proto b/app/protos/grpc.proto
@@ -4098,6 +4098,7 @@ message FuzzerRequest {
   bool IsPause = 53;
 
   repeated MutateMethod MutateMethods = 54;
+  bool SetPauseStatus = 55;
 }
 
 message MutateMethod {
@@ -4210,6 +4211,8 @@ message FuzzerResponse {
   string TooLargeResponseHeaderFile = 50;
   string TooLargeResponseBodyFile = 51;
   bool DisableRenderStyles = 52;
+
+  string RuntimeID = 53;
 }
 
 message RedirectHTTPFlow {
@@ -4288,6 +4291,9 @@ message QueryHTTPFlowRequest {
   repeated string IncludeSuffix = 36;
   repeated string ExcludeSuffix = 37;
   repeated string ExcludeContentType = 38;
+
+  bool WithPayload = 39;
+  repeated string RuntimeIDs = 40;
 }
 
 message ExportHTTPFlowsRequest {
@@ -4388,6 +4394,9 @@ message HTTPFlow {
   string TooLargeResponseHeaderFile = 44;
   string TooLargeResponseBodyFile = 45;
   bool DisableRenderStyles = 46;
+
+  // http_flows 与 web_fuzzer_responses 表关联下取到Payload
+  repeated string Payloads = 47;
 }
 
 message FuzzableParam {

diff --git a/app/renderer/src/main/src/components/DataExport/DataExport.tsx b/app/renderer/src/main/src/components/DataExport/DataExport.tsx
@@ -310,17 +310,18 @@ interface ExportSelectProps {
     fileName?: string
     /* limit */
     pageSize?: number
+    initCheckValue?: string[]
 }
 // 导出字段选择
 export const ExportSelect: React.FC<ExportSelectProps> = (props) => {
-    const {exportValue, fileName, setExportTitle, exportKey, getData, pageSize} = props
+    const {exportValue, fileName, setExportTitle, exportKey, getData, pageSize, initCheckValue} = props
     const [checkValue, setCheckValue] = useState<CheckboxValueType[]>([])
     useEffect(() => {
         getRemoteValue(exportKey).then((setting) => {
             if (!setting) {
                 // 第一次进入 默认勾选所有导出字段
                 setExportTitle(exportValue as string[])
-                setCheckValue(exportValue)
+                setCheckValue(initCheckValue || exportValue)
             } else {
                 const values = JSON.parse(setting)
                 setCheckValue(values?.checkedValues)