Error in fuzz count when using repeated FUZZ keywords (fixes #411)

qinwentu · Mar 23, 2015 · 5075ea0 · 5075ea0
1 parent 5e60f67
commit 5075ea0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/framework/fuzzer/fuzzobjects.py b/framework/fuzzer/fuzzobjects.py
@@ -170,13 +170,13 @@ def from_seed(seed, payload):
 	http_method = None
 
 	marker_regex = re.compile("FUZ\d*Z",re.MULTILINE|re.DOTALL)
-	fuzz_words = len(marker_regex.findall(rawReq))
+	fuzz_words = len(set(marker_regex.findall(rawReq)))
 
 	if seed.wf_fuzz_methods:
 	    fuzz_words += 1
 
 	if method:
-	    fuzz_words += len(marker_regex.findall(userpass))
+	    fuzz_words += len(set(marker_regex.findall(userpass)))
 
 	if len(payload) != fuzz_words:
 	    raise FuzzException(FuzzException.FATAL, "FUZZ words and number of payloads do not match!")