Add display functionality for lambda objects

2 files changed, 28 insertions, 7 deletions

diff --git a/src/compiler.h b/src/compiler.h
index 897b7bb..0223269 100644
--- a/src/compiler.h
+++ b/src/compiler.h
@@ -509,8 +509,15 @@ emit_bss_section(void) {
 void
 emit_data_section(void) {
     printf("section .data\n");
-    printf("true_str: db \"true\", 10, 0, 0, 0\n");
-    printf("false_str: db \"false\", 10, 0, 0\n");
+    printf("true_str:\n    db \"true\", 10\n");
+    printf("    alignb 8\n");
+    printf("false_str:\n    db \"false\", 10\n");
+    printf("    alignb 8\n");
+    printf("lambda_str:\n");
+    char lambda_str[] = "#{lambda}";
+    printf("    dq %ld\n", sizeof(lambda_str));
+    printf("    db \"%s\", 10\n", lambda_str);
+    printf("    alignb 8\n");
     for (size_t i = 0; i < array_size(constants); i++) {
         // NOTE: Only supporting string constants for now.
         Constant c = constants[i];
@@ -520,11 +527,7 @@ emit_data_section(void) {
         printf("%s:\n", c.label);
         printf("    dq %d\n", n + 1);
         printf("    db \"%.*s\", 10\n", n, c.obj->text.start);
-        // Ensure alignment to 8 bytes.
-        int remainder = (n + 1) % 8;
-        if (remainder != 0) {
-            printf("    times %d db 0\n", 8 - (n + 1) % 8);
-        }
+        printf("    alignb 8\n");
     }
     printf("\n");
 }
diff --git a/src/x86_64/prelude.asm b/src/x86_64/prelude.asm
index abb069b..81a2633 100644
--- a/src/x86_64/prelude.asm
+++ b/src/x86_64/prelude.asm
@@ -82,6 +82,15 @@ printstring:
     syscall
     ret
 
+printlambda:
+    mov     rsi, lambda_str
+    mov     rdx, [rsi]
+    add     rsi, 8
+    mov     rax, 1
+    mov     rdi, 1
+    syscall
+    ret
+
 display:
     ;; is nil?
     mov     rax, rdi
@@ -106,6 +115,15 @@ not_bool:
     ret
 not_string:
 
+    ;; is lambda?
+    mov     rax, rdi
+    and     rax, LAMBDA_MASK
+    cmp     rax, LAMBDA_TAG
+    jne     not_lambda
+    call    printlambda
+    ret
+not_lambda:
+
     ;; is fixnum?
     mov     rax, rdi
     and     rax, FIXNUM_MASK