Fix hang in lexer if file ends in a comment without a newline (#184)

Found by fuzzer!
asmaloney · Jun 9, 2022 · 67ce086 · 67ce086
1 parent 2acde1d
commit 67ce086
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/amod/lex.go b/amod/lex.go
@@ -419,7 +419,14 @@ func lexSpace(l *lexer_amod) stateFn {
 func lexComment(l *lexer_amod) stateFn {
 	l.pos += len(commentDelim)
 	i := strings.Index(l.input[l.pos:], "\n")
-	l.pos += i
+
+	// If we are at the end of file there may not be a newline,
+	// so take the rest of the input.
+	if i == -1 {
+		l.pos = len(l.input)
+	} else {
+		l.pos += i
+	}
 
 	l.emit(lexemeComment)