General improvements

(nothing special, mainly cleanup and docs)

1 files changed, 24 insertions, 1 deletions

diff --git a/src/term.c b/src/term.c
index 50333c7..63bbb05 100644
--- a/src/term.c
+++ b/src/term.c
@@ -12,7 +12,30 @@ struct term *new_term(term_type type)
 {
 	struct term *term = malloc(sizeof(*term));
 	if (!term)
-		fatal("Out of memory!\n");
+		fatal("out of memory!\n");
 	term->type = type;
 	return term;
 }
+
+void free_term(struct term *term)
+{
+	switch (term->type) {
+	case ABS:
+		free_term(term->u.abs.term);
+		free(term);
+		break;
+	case APP:
+		free_term(term->u.app.lhs);
+		free_term(term->u.app.rhs);
+		free(term);
+		break;
+	case VAR:
+		free(term);
+		break;
+	case REF:
+		free(term);
+		break;
+	default:
+		fatal("invalid type %d\n", term->type);
+	}
+}