From 0e5c802b0b3e896420e7b2c2406495c72c2e8e50 Mon Sep 17 00:00:00 2001
From: Mike Lowis <mike.lowis@gentex.com>
Date: Mon, 14 Dec 2015 12:04:06 +0000
Subject: [PATCH] Added reference count handling to new AST  functions

---
 Makefile     | 2 +-
 source/ast.c | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/Makefile b/Makefile
index 281ffba..a6495c6 100644
--- a/Makefile
+++ b/Makefile
@@ -10,7 +10,7 @@ LD = ${CC}
 # flags
 INCS      = -Isource/
 CPPFLAGS  = -D_XOPEN_SOURCE=700
-CFLAGS   += ${INCS} ${CPPFLAGS}
+CFLAGS   += ${INCS} ${CPPFLAGS} -Wall -Wextra
 LDFLAGS  += ${LIBS}
 
 #------------------------------------------------------------------------------
diff --git a/source/ast.c b/source/ast.c
index c7f6964..6eb8f66 100644
--- a/source/ast.c
+++ b/source/ast.c
@@ -225,7 +225,7 @@ AST* Block(void)
 
 void block_append(AST* block, AST* expr)
 {
-    vec_push_back(&(block->value.exprs), expr);
+    vec_push_back(&(block->value.exprs), gc_addref(expr));
 }
 
 size_t block_size(AST* block)
@@ -258,12 +258,12 @@ AST* func_body(AST* func)
 
 void func_add_arg(AST* func, AST* arg)
 {
-    vec_push_back(func_args(func), arg);
+    vec_push_back(func_args(func), gc_addref(arg));
 }
 
 void func_set_body(AST* func, AST* body)
 {
-    func->value.func.body = body;
+    func->value.func.body = (AST*)gc_addref(body);
 }
 
 
-- 
2.52.0