parser+codegen: Handle function epilogue

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

core/codegen.c

@@ -13,6 +13,10 @@ resolve_func(struct rifle_state *state, struct ast_node *root)
         return -1;
     }
 
+    if (root->epilogue) {
+        return mu_gen_ret(state);
+    }
+
     if ((symbol = root->symbol) == NULL) {
         trace_error(state, "internal: no symbol associated with func node\n");
         return -1;

core/parser.c

@@ -495,6 +495,9 @@ parse_func(struct rifle_state *state, struct token *tok, struct ast_node **res)
 static int
 parse_rbrace(struct rifle_state *state, struct token *tok, struct ast_node **res)
 {
+    struct ast_node *root = NULL;
+    tt_t scope;
+
     if (state == NULL || tok == NULL) {
         return -1;
     }
@@ -507,7 +510,21 @@ parse_rbrace(struct rifle_state *state, struct token *tok, struct ast_node **res
         return -1;
     }
 
-    scope_pop(state);
+    scope = scope_pop(state);
+    switch (scope) {
+    case TT_F:
+        if (ast_alloc_node(state, AST_FUNC, &root) < 0) {
+            trace_error(state, "failed to allocate AST_FUNC\n");
+            return -1;
+        }
+
+        root->epilogue = 1;
+        break;
+    default:
+        break;
+    }
+
+    *res = root;
     return 0;
 }
 
@@ -537,11 +554,11 @@ parse_begin(struct rifle_state *state)
             utok1(state, &tok);
             return -1;
         }
-    }
 
-    if (root != NULL) {
-        if (cg_resolve_node(state, root) < 0)
-            return -1;
+        if (root != NULL) {
+            if (cg_resolve_node(state, root) < 0)
+                return -1;
+        }
     }
 
     return 0;