X-Git-Url: http://git.lukelau.me/?p=scheme.git;a=blobdiff_plain;f=codegen.scm;h=b403939bc2122c2a4c9b5e24980955a2e329402e;hp=4848d4ddfa98731899f6b2ca92413b45eb3086ce;hb=6684c39cf92a0fa6c21339c6fec8a6e4cf4f463d;hpb=dea4c4e4d1be8c0a90bb165bde1e362dc204e061

diff --git a/codegen.scm b/codegen.scm
index 4848d4d..b403939 100644
--- a/codegen.scm
+++ b/codegen.scm
@@ -105,8 +105,21 @@
   (set! cur-lambda (+ 1 cur-lambda))
   (format "_lambda~a" (- cur-lambda 1)))
 
+(define (codegen-closure label captured si env)
+;; (define (codegen-closure label captured si env)
+;;   (let* ((stack-offsets (map (lambda (x) (- si (* x wordsize))))
+;; 			(range 0 (length captured)))
+;; 	 (inner-si (- si (* (length captured) wordsize))))
+;;     (for-each (lambda (var-name new-offset)
+;; 		(emit "movq ~a(%rbp), ~a(%rbp)" ; todo: do we need to copy this?
+;; 		      (cdr (assoc var-name env))
+;; 		      new-offset))
+;; 	      captured
+;; 	      stack-offsets)
+;; )
 					; for now we can only call closures
 (define (codegen-call closure args si env)
+;  (codegen-expr closure si env)
   (when (not (eq? (ast-type closure) 'closure))
     (error #f (format "~a is not a closure" closure)))
   (let* ((captured (caddr closure))
@@ -121,7 +134,6 @@
 	     (param-register i)))
      captured (range 0 (length captured)))
 
-
 					; then codegen the arguments and move them into the next param registers
     (for-each
      (lambda (e i)
@@ -191,7 +203,7 @@
 (define (codegen-expr e si env)
   (case (ast-type e)
     ('builtin e)
-    ('closure e)
+    ('closure (codegen-closure (cadr e) (caddr e) si env))
     ('app
      (let ((callee (codegen-expr (car e) si env)))
        (case callee
@@ -343,6 +355,21 @@
     (for-each codegen-lambda lambdas)
 
     (emit "_start:")
+
+    ; allocate some heap memory
+    (emit "mov $9, %rax") ; mmap
+    (emit "xor %rdi, %rdi")  ; addr = null
+    (emit "movq $1024, %rsi")   ; length = 1kb
+    (emit "movq $0x3, %rdx") ; prot = read | write = 0x2 | 0x1
+    (emit "movq $0x22, %r10") ; flags = anonymous | private = 0x20 | 0x02
+    (emit "movq $-1, %r8") ; fd = -1
+    (emit "xor %r9, %r9") ; offset = 0
+    (emit "syscall")
+
+    ; %rax now contains pointer to the start of the heap
+    ; keep track of it
+    (emit "movq %rax, (heap_start)")
+    
     (emit "movq %rsp, %rbp")            ; set up the base pointer
     (codegen-expr xform-prog 0 '())
 
@@ -351,7 +378,10 @@
     (emit "mov $60, %rax")
     (emit "syscall")
 
-    (emit "\t.data")
+    (emit ".data")
+
+    (emit "heap_start:")
+    (emit "\t.quad 0")
 
     (for-each emit-string-data strings)))
 
@@ -362,3 +392,13 @@
     (with-output-to-file tmp-path
       (lambda () (codegen program)))
     (system (format "clang -nostdlib /tmp/a.s -o ~a" output))))
+
+; NOTES
+; syscalls in linux use the following arguments for syscall instruction:
+; %rax = syscall #
+; %rdi = 1st arg
+; %rsi = 2nd arg
+; %rdx = 3rd arg
+; %r10 = 4th arg
+; %r8  = 5th arg
+; %r9  = 6th arg