Merge pull request #711 from SSWConsulting/sanitise-csv

Sanitize strings before writing tab-delimited CSV
SSWConsulting · Oct 19, 2023 · e37f335 · e37f335
2 parents 2438d9a + 072bfe7
commit e37f335
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/docker/sswlinkauditor.go b/docker/sswlinkauditor.go
@@ -208,12 +208,23 @@ func writeResultFile(allUrls map[string]LinkStatus) {
 
 	f.WriteString("Source" + "\t" + "Destination" + "\t" + "Status" + "\t" + "StatusCode" + "\t" + "Anchor" + "\n")
 	for _, v := range allUrls {
-		f.WriteString(v.srcUrl + "\t" + v.url + "\t" + v.status + "\t" + strconv.Itoa(v.statusCode) + "\t" + strings.ReplaceAll(v.anchor,"\"","") + "\n")
+		f.WriteString(sanitizeString(v.srcUrl) + "\t" + sanitizeString(v.url) + "\t" + sanitizeString(v.status) + "\t" + strconv.Itoa(v.statusCode) + "\t" + sanitizeString(v.anchor) + "\n")
 	}
 
 	f.Close()
 }
 
+func sanitizeString(s string) string {
+	replacer := strings.NewReplacer(
+		"\"", "",
+		"\t", " ",
+		"\r\n", "",
+		"\n", "",
+	);
+
+	return replacer.Replace(s);
+}
+
 func isLinkUnscannable(a string, unscannableLinks []string) bool {
 	for _, b := range unscannableLinks {
 		if strings.HasPrefix(strings.ToLower(a), strings.ToLower(b)) {