Print memory for iniquity.

Author: dvanhorn

langs/iniquity/compile-ops.rkt

@@ -21,7 +21,15 @@
                      unpad-stack)]
     ['peek-byte (seq pad-stack
                      (Call 'peek_byte)
-                     unpad-stack)]))
+                     unpad-stack)]
+    ['dump-memory-stats
+     (seq (Mov rdi rsp)
+          (Mov 'rsi 'rbp)
+          (Mov 'rdx rbx)
+          pad-stack
+          (Call 'print_memory)
+          (Mov rax 0)
+          unpad-stack)]))
 
 ;; Op1 -> Asm
 (define (compile-op1 p)

langs/iniquity/compile.rkt

@@ -17,7 +17,8 @@
      (prog (externs)
            (Global 'entry)
            (Label 'entry)
-           (Mov rbx rdi) ; recv heap pointer
+           (Mov 'rbp 'rsp) ; save stack base pointer
+           (Mov rbx rdi)   ; recv heap pointer
            (compile-e e '())
            (Ret)
            (compile-defines ds)
@@ -29,7 +30,8 @@
   (seq (Extern 'peek_byte)
        (Extern 'read_byte)
        (Extern 'write_byte)
-       (Extern 'raise_error)))
+       (Extern 'raise_error)
+       (Extern 'print_memory)))
 
 ;; [Listof Defn] -> Asm
 (define (compile-defines ds)

langs/iniquity/main.c

@@ -1,5 +1,6 @@
 #include <stdio.h>
 #include <stdlib.h>
+#include <inttypes.h>
 #include "values.h"
 #include "print.h"
 #include "runtime.h"
@@ -38,3 +39,39 @@ int main(int argc, char** argv)
   free(heap);
   return 0;
 }
+
+const char* val_typeof_string(int64_t t) {
+  switch (val_typeof(t)) {
+  case T_INT: return "INT";
+  case T_BOOL: return "BOOL";
+  case T_CHAR: return "CHAR";
+  case T_EOF: return "EOF";
+  case T_VOID: return "VOID";
+  case T_EMPTY: return "EMPTY";
+  case T_BOX: return "BOX";
+  case T_CONS: return "CONS";
+  case T_VECT: return "VECT";
+  case T_STR: return "STR";
+  default: return "UNKNOWN";
+  }
+}
+
+
+
+void print_memory(int64_t* rsp, int64_t rbp, int64_t rbx) {
+  int stack_count = (rbp-(int64_t)rsp) / 8;
+  int heap_count = (rbx-(int64_t)heap) / 8;
+
+  printf("----------------------------------------------------------------\n");
+  //  printf("rsp: %" PRIx64 ", rbp: %" PRIx64 ", stack count: %d, heap count: %d\n", (int64_t)rsp, rbp, stack_count, heap_count);
+  int i;
+
+  printf("STACK:\n");
+  for (i = 0; i < stack_count; i++) {
+    printf("[%" PRIx64 "] = %016" PRIx64 ", %s\n", (int64_t)rsp + i, rsp[i], val_typeof_string(rsp[i]));
+  }
+  printf("HEAP:\n");
+  for (i = 0; i < heap_count; i++) {
+    printf("[%" PRIx64 "] = %016" PRIx64 ", %s\n", (int64_t)heap + i, heap[i], val_typeof_string(heap[i]));
+  }
+}

langs/iniquity/parse.rkt

@@ -47,7 +47,7 @@
     [_ (error "Parse error" s)]))
 
 (define op0
-  '(read-byte peek-byte void))
+  '(read-byte peek-byte void dump-memory-stats))
 
 (define op1
   '(add1 sub1 zero? char? write-byte eof-object?