lexer: add error

1 files changed, 7 insertions, 1 deletions

diff --git a/src/main.cc b/src/main.cc
index 61d853b..c6dcb81 100644
--- a/src/main.cc
+++ b/src/main.cc
@@ -31,7 +31,14 @@ main(int argc, char **argv)
   Lexer l;
   l.init(buffer);
 
+  f.close();
+
   List *tokens = l.scan();
+  if (tokens == NULL) {
+    l.clean();
+    return 1;
+  }
+
   for (size_t i = 0; i < tokens->size(); i++) {
     Token *t = (Token *) tokens->get(i);
     printf("%s\n", t->to_string());
@@ -39,6 +46,5 @@ main(int argc, char **argv)
 
   l.clean();
 
-  f.close();
   return 0;
 }