lexer: Add token for 'return' keyword

Signed-off-by: Ian Moffett <ian@mirocom.org>

core/lexer.c

@@ -261,6 +261,13 @@ lexer_check_kw(struct rifle_state *state, struct token *tok)
             return;
         }
 
+        break;
+    case 'r':
+        if (strcmp(tok->s, "return") == 0) {
+            tok->type = TT_RETURN;
+            return;
+        }
+
         break;
     case '.':
         lexer_check_direc(state, tok);

core/parser.c

@@ -74,6 +74,7 @@ static const char *toktab[] = {
     [TT_STRUCT] = qtok(".struct"),
     [TT_EXTERN] = qtok(".extern"),
     [TT_PUB]    = qtok(".pub"),
+    [TT_RETURN] = qtok("return"),
     [TT_VOID]   = qtok("void"),
     [TT_U8]     = qtok("u8"),
     [TT_U16]    = qtok("u16"),

inc/rifle/token.h

@@ -24,6 +24,7 @@ typedef enum {
     TT_STRUCT,  /* '.struct' */
     TT_EXTERN,  /* '.extern' */
     TT_PUB,     /* '.pub' */
+    TT_RETURN,  /* 'return' */
     TT_VOID,    /* 'void' */
     TT_U8,      /* 'u8' */
     TT_U16,     /* 'u16' */