From dc2f25ef2a5fd4959869fe9c0e2757107510e4c2 Mon Sep 17 00:00:00 2001
From: Yuichi Nishiwaki <yuichi.nishiwaki@gmail.com>
Date: Sun, 13 Oct 2013 19:00:12 +0900
Subject: [PATCH] initial GC commit

---
 include/picrin/gc.h |   6 ++
 src/gc.c            | 134 +++++++++++++++++++++++++++++++++++++-------
 src/state.c         |   1 +
 3 files changed, 122 insertions(+), 19 deletions(-)

diff --git a/include/picrin/gc.h b/include/picrin/gc.h
index 121a3a02..ccc6271c 100644
--- a/include/picrin/gc.h
+++ b/include/picrin/gc.h
@@ -1,10 +1,16 @@
 #ifndef GC_H__
 #define GC_H__
 
+enum pic_gc_mark {
+  PIC_GC_MARK,
+  PIC_GC_UNMARK
+};
+
 union header {
   struct {
     union header *ptr;
     size_t size;
+    enum pic_gc_mark mark;
   } s;
   long alignment;   /* force alignment for headers to be allocated at the size of long */
 };
diff --git a/src/gc.c b/src/gc.c
index 482b95f8..c584de57 100644
--- a/src/gc.c
+++ b/src/gc.c
@@ -74,7 +74,7 @@ pic_gc_arena_restore(pic_state *pic, int state)
 }
 
 static void *
-pic_gc_alloc(pic_state *pic, size_t size)
+gc_alloc(pic_state *pic, size_t size)
 {
   union header *freep, *p, *prevp;
   size_t nunits;
@@ -100,34 +100,122 @@ pic_gc_alloc(pic_state *pic, size_t size)
   }
   pic->heap->freep = prevp;
 
+  p->s.mark = PIC_GC_UNMARK;
   return (void *)(p + 1);
 }
 
+static void gc_mark(pic_state *, pic_value);
+
 static void
-pic_gc_free(pic_state *pic, void *ptr)
+gc_mark_object(pic_state *pic, struct pic_object *obj)
 {
-  union header *bp, *p;
+  union header *p;
 
-  bp = (union header *)ptr - 1;
+  p = (union header *)obj - 1;
+  p->s.mark = PIC_GC_MARK;
 
-  for (p = pic->heap->freep; !(p < bp && bp < p->s.ptr); p = p->s.ptr) {
-    if (p >= p->s.ptr && (bp > p || bp < p->s.ptr))
-	break;
+  switch (obj->tt) {
+  case PIC_TT_PAIR: {
+    gc_mark(pic, ((struct pic_pair *)obj)->car);
+    gc_mark(pic, ((struct pic_pair *)obj)->cdr);
+    break;
   }
-  if (bp + bp->s.size == p->s.ptr) {
-    bp->s.size += p->s.ptr->s.size;
-    bp->s.ptr = p->s.ptr->s.ptr;
+  case PIC_TT_SYMBOL: {
+    break;
   }
-  else {
-    bp->s.ptr = p->s.ptr;
+  case PIC_TT_PROC: {
+    break;
   }
-  if (p + p->s.size == bp) {
-    p->s.size += bp->s.size;
-    p->s.ptr = bp->s.ptr;
-  } else {
-    p->s.ptr  =  bp;
+  default:
+    pic_raise(pic, "logic flaw");
   }
-  pic->heap->freep = p;
+}
+
+static void
+gc_mark(pic_state *pic, pic_value v)
+{
+  struct pic_object *obj;
+
+  if (v.type != PIC_VTYPE_HEAP)
+    return;
+  obj = pic_object_ptr(v);
+
+  gc_mark_object(pic, obj);
+}
+
+static void
+gc_mark_phase(pic_state *pic)
+{
+  pic_value *stack;
+  struct pic_env *env;
+  int i;
+
+  /* stack */
+  for (stack = pic->stbase; stack != pic->sp; ++stack) {
+    gc_mark(pic, *stack);
+  }
+  gc_mark(pic, *stack);
+
+  /* arena */
+  for (i = 0; i < pic->arena_idx; ++i) {
+    gc_mark_object(pic, pic->arena[i]);
+  }
+
+  /* global env */
+  env = pic->global_env;
+  do {
+    gc_mark(pic, env->assoc);
+  } while ((env = env->parent) != NULL);
+}
+
+static bool
+is_marked(union header *p)
+{
+  return p->s.mark == PIC_GC_MARK;
+}
+
+static void
+gc_unmark(union header *p)
+{
+  p->s.mark = PIC_GC_UNMARK;
+}
+
+static void
+gc_sweep_phase(pic_state *pic)
+{
+  union header *freep, *bp, *p;
+
+  freep = pic->heap->freep;
+  for (p = freep; p != freep; p = p->s.ptr) {
+    for (bp = p + p->s.size; bp != p->s.ptr; bp += bp->s.size) {
+      if (is_marked(bp)) {
+	gc_unmark(bp);
+	continue;
+      }
+      /* free! */
+      if (bp + bp->s.size == p->s.ptr) {
+	bp->s.size += p->s.ptr->s.size;
+	bp->s.ptr = p->s.ptr->s.ptr;
+      }
+      else {
+	bp->s.ptr = p->s.ptr;
+      }
+      if (p + p->s.size == bp) {
+	p->s.size += bp->s.size;
+	p->s.ptr = bp->s.ptr;
+      }
+      else {
+	p->s.ptr = bp;
+      }
+    }
+  }
+}
+
+static void
+pic_gc_run(pic_state *pic)
+{
+  gc_mark_phase(pic);
+  gc_sweep_phase(pic);
 }
 
 struct pic_object *
@@ -135,7 +223,15 @@ pic_obj_alloc(pic_state *pic, size_t size, enum pic_tt tt)
 {
   struct pic_object *obj;
 
-  obj = (struct pic_object *)malloc(size);
+  pic_gc_run(pic);
+  obj = (struct pic_object *)gc_alloc(pic, size);
+  if (obj == NULL) {
+    puts("gc run!");
+    pic_gc_run(pic);
+    obj = (struct pic_object *)gc_alloc(pic, size);
+    if (obj == NULL)
+      pic_raise(pic, "memory exhausted");
+  }
   obj->tt = tt;
 
   gc_protect(pic, obj);
diff --git a/src/state.c b/src/state.c
index 04021827..5d96b76a 100644
--- a/src/state.c
+++ b/src/state.c
@@ -25,6 +25,7 @@ pic_open()
   /* prepare VM stack */
   pic->stbase = pic->sp = (pic_value *)malloc(sizeof(pic_value) * 1024);
   pic->stend = pic->stbase + 1024;
+  *pic->sp = pic_nil_value();	/* push dummy object */
 
   /* memory heap */
   pic->heap = (struct heap_page *)malloc(sizeof(struct heap_page));