feat: add memory loads via ^reg

2 files changed, 35 insertions, 0 deletions

diff --git a/tests/codegen_test.c b/tests/codegen_test.c
index b69d79e..1cb0071 100644
--- a/tests/codegen_test.c
+++ b/tests/codegen_test.c
@@ -372,8 +372,36 @@ static void test_generate_logical_disabled(struct TestContext* context)
 	free_program(&program);
 }
 
+static void test_generate_load(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer(
+		"proc main\n{\nrax = ^rsi\nrbx = ^byte rsi\nrcx = ^dword rsi\nrdx = ^rsi + 4\n}\n");
+	struct Program program;
+	check(context, parse_program(&lexer, &program));
+
+	FILE* out = tmpfile();
+	generate_nasm(&program, out);
+	fflush(out);
+	rewind(out);
+
+	char buffer[1024];
+	size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
+	buffer[read] = '\0';
+	fclose(out);
+
+	check(context, strstr(buffer, "mov rax, [rsi]") != NULL);
+	check(context, strstr(buffer, "movzx rbx, byte [rsi]") != NULL);
+	check(context, strstr(buffer, "mov ecx, [rsi]") != NULL);
+	check(context, strstr(buffer, "mov rdx, [rsi]") != NULL);
+	check(context, strstr(buffer, "add rdx, 4") != NULL);
+	check(context, strstr(buffer, "; TODO") == NULL);
+
+	free_program(&program);
+}
+
 void run_codegen_tests(struct TestContext* context)
 {
+	test_generate_load(context);
 	test_generate_logical_registers(context);
 	test_generate_logical_disabled(context);
 	test_generate_entry_and_bits(context);
diff --git a/tests/sema_test.c b/tests/sema_test.c
index 095321d..2757e6c 100644
--- a/tests/sema_test.c
+++ b/tests/sema_test.c
@@ -76,6 +76,12 @@ static void test_assign_to_const(struct TestContext* context)
 	check(context, !analyze_source("const K = 5\nproc main\n{\nK = 1\n}\n"));
 }
 
+static void test_deref_needs_register(struct TestContext* context)
+{
+	check(context, !analyze_source("proc main\n{\nrax = ^MISSING\n}\n"));
+	check(context, analyze_source("proc main\n{\nrax = ^rsi\n}\n"));
+}
+
 static void test_references_resolve(struct TestContext* context)
 {
 	// registers, params, consts, data (+ .len), stack buffers, labels, calls
@@ -108,5 +114,6 @@ void run_sema_tests(struct TestContext* context)
 	test_undefined_call(context);
 	test_call_arg_count(context);
 	test_assign_to_const(context);
+	test_deref_needs_register(context);
 	test_references_resolve(context);
 }