Fix random failure for WafFuncGen

Marven11 · Aug 25, 2024 · 2deb07d · 2deb07d
1 parent c8c0d76
commit 2deb07d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fenjing/waf_func_gen.py b/fenjing/waf_func_gen.py
@@ -363,6 +363,8 @@ def generate(self) -> WafFunc:
             "".join(random.choices(string.ascii_lowercase, k=4)),
             False,
         )
+        while any(w in extra_content for w in replaced_keyword):
+            extra_content = "".join(random.choices(string.ascii_lowercase, k=4))
 
         # WAF函数，只有在payload一定可以通过WAF时才返回True
         @lru_cache(10000)