fix: handle unterminated strings gracefully

1 files changed, 7 insertions, 1 deletions

diff --git a/src/lisp/lex.rs b/src/lisp/lex.rs
index 1a34e53..e514d7f 100644
--- a/src/lisp/lex.rs
+++ b/src/lisp/lex.rs
@@ -210,6 +210,7 @@ fn parse_number<'a>(mut input: &'a str) -> Result<(usize, Token<'a>), ParseError
 fn parse_string<'a>(input: &'a str) -> Result<(usize, Token<'a>), ParseErrorKind> {
     // count opening quote
     let mut size = 1;
+    let mut closed = false;
     let mut chars = input.char_indices().skip(1);
     while let Some((ind, chr)) = chars.next() {
         match chr {
@@ -218,12 +219,17 @@ fn parse_string<'a>(input: &'a str) -> Result<(usize, Token<'a>), ParseErrorKind
             }
             '"' => {
                 size += ind;
+                closed = true;
                 break;
             }
             _ => (),
         }
     }
-    return Ok((size, Token::String(&input[..size])));
+    if !closed {
+        Err(ParseErrorKind::UnterminatedString)
+    } else {
+        Ok((size, Token::String(&input[..size])))
+    }
 }
 
 fn is_ident(ch: char) -> bool {