Improve input error display when fuzzing a single value

RedTeamPentesting · Jun 21, 2023 · 85bbc02 · 85bbc02
1 parent f1c0fe6
commit 85bbc02
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/reporter/reporter.go b/reporter/reporter.go
@@ -158,7 +158,7 @@ next_response:
 			var reqErr response.InvalidRequest
 			if errors.As(resp.Error, &reqErr) {
 				errString := cleanedErrorString(reqErr.Err)
-				stats.InvalidInputData[errString] = append(stats.InvalidInputData[errString], fmt.Sprintf("%q", resp.Values))
+				stats.InvalidInputData[errString] = append(stats.InvalidInputData[errString], formatValues(resp.Values))
 
 				continue
 			}
@@ -190,6 +190,14 @@ next_response:
 	return nil
 }
 
+func formatValues(values []string) string {
+	if len(values) == 1 {
+		return fmt.Sprintf("%q", values[0])
+	}
+
+	return fmt.Sprintf("%q", values)
+}
+
 func sortedKeys(m map[string][]string) []string {
 	keys := make([]string, 0, len(m))