在获取载荷时校验探针有效性

phplaber · Aug 4, 2024 · 8aef810 · 8aef810
1 parent fe0ce50
commit 8aef810
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/core/fuzzer.py b/core/fuzzer.py
@@ -53,9 +53,6 @@ def do_fuzz(self, requests, fuzz_results, flag, load_page):
                 for probe in self.probes:
                     if hasattr(Probe, probe) and callable(getattr(Probe, probe)):
                         getattr(Probe, probe)(probe_ins)
-                    else:
-                        if probe not in ['jsonp']:
-                            print(f'[*] invalid probe: {probe}')
 
         # 关闭 Chrome 浏览器
         if chrome is not None:

diff --git a/yawf.py b/yawf.py
@@ -381,6 +381,8 @@
         payload_file = os.path.join(payload_path, f'{probe}.txt')
         if check_file(payload_file):
             probes_payload[probe] = read_file(payload_file)
+        elif probe not in ['jsonp']:
+            print(f'[*] invalid probe: {probe}')
 
     # 初始化 dnslog 实例
     dnslog = None

diff --git a/yawf_bulk.py b/yawf_bulk.py
@@ -75,6 +75,8 @@
         payload_file = os.path.join(payload_path, f'{probe}.txt')
         if check_file(payload_file):
             probes_payload[probe] = read_file(payload_file)
+        elif probe not in ['jsonp']:
+            print(f'[*] invalid probe: {probe}')
 
     # 初始化 dnslog 实例
     dnslog = None