Exception when there is a different number of Baseline {} and FUZZ ke…

…ywords (fixes #416)
qinwentu · Mar 23, 2015 · 574caa5 · 574caa5
1 parent 5075ea0
commit 574caa5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/framework/fuzzer/fuzzobjects.py b/framework/fuzzer/fuzzobjects.py
@@ -238,7 +238,10 @@ def from_baseline(seed):
 	seed.parseRequest(rawReq, schema)
 	if seed.wf_fuzz_methods: seed.method = "FUZZ"
 
-	baseline_req = FuzzRequest.from_seed(seed, baseline_payload)
+	try:
+	    baseline_req = FuzzRequest.from_seed(seed, baseline_payload)
+	except FuzzException:
+	    raise FuzzException(FuzzException.FATAL, "You must supply a baseline value for all the FUZZ words.")
 	baseline_req.wf_is_baseline = True
 
 	return baseline_req