(fix) limit screenshot filename lengths

fixes #121
sensepost · Jun 10, 2022 · 19b4e15 · 19b4e15
1 parent 2f87924
commit 19b4e15
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/lib/helpers.go b/lib/helpers.go
@@ -123,3 +123,18 @@ func SliceContainsString(s []string, e string) bool {
 
 	return false
 }
+
+// TruncateString truncates a string for l characters
+func TruncateString(s string, l int) string {
+
+	if len(s) <= l {
+		return s
+	}
+
+	var n string
+	for _, char := range s {
+		n = n + string(char)
+	}
+
+	return n
+}
diff --git a/lib/processor.go b/lib/processor.go
@@ -83,6 +83,9 @@ func (p *Processor) init() {
 		p.fn = SafeFileName(p.URL.String())
 	}
 
+	// limit filename length
+	p.fn = TruncateString(p.fn, 30)
+
 	// set the extention depending on the screenshot format
 	if p.Chrome.AsPDF {
 		p.fn = p.fn + ".pdf"