1 files changed, 8 insertions, 0 deletions
diff --git a/src/ast.c b/src/ast.c
index acada6f..3069924 100644
--- a/src/ast.c
+++ b/src/ast.c
@@ -76,6 +76,9 @@ static struct ast *create_empty_ast()
 	}
 
 	struct ast *n = calloc(1, sizeof(struct ast));
+	if (!n)
+		return NULL;
+
 	/* just to be safe */
 	n->k = AST_EMPTY;
 	vect_append(struct ast *, nodes, &n);
@@ -105,6 +108,9 @@ struct ast *gen_ast(enum ast_kind kind,
                     struct src_loc loc)
 {
 	struct ast *n = create_empty_ast();
+	if (!n)
+		return NULL;
+
 	n->k = kind;
 	n->a0 = a0;
 	n->a1 = a1;
@@ -548,6 +554,8 @@ void fix_closures(struct ast *root)
 		if (!next)
 			next = gen_empty(root->loc);
 
+		assert(next);
+
 		struct ast *block = gen_block(next, root->loc);
 		closure_body(arg) = block;
 		root->n = NULL;