save function args

1 files changed, 9 insertions, 0 deletions

diff --git a/src/main.c b/src/main.c
index 3944956..dbf39df 100644
--- a/src/main.c
+++ b/src/main.c
@@ -643,6 +643,15 @@ gen_function(Node *node)
 	printf("\tLIT %04x ADD\n", locals_size);
 	printf("\tLIT ,__stack_ptr STA\n");
 
+	/* TODO: optimize this using DUP */
+	cur = locals;
+	for (int i = 0; i < argcount; i++) {
+		printf("\tLIT ,__stack_ptr LDA ( %.*s )\n", cur->len, cur->loc);
+		printf("\tLIT %04x SUB\n", locals_size - i - 1);
+		printf("\tSWP STA\n");
+		cur = cur->next;
+	}
+
 	while (node != NULL) {
 		switch (node->type) {
 		case NOD_EXPR_STMT: