From 21f2c58a5309487ce3bfc3f3b3bb96c37d76ff59 Mon Sep 17 00:00:00 2001
From: Yuichi Nishiwaki <yuichi.nishiwaki@gmail.com>
Date: Fri, 29 Nov 2013 12:05:22 +0900
Subject: [PATCH] [gc][bugfix] sweeping failed when the heap gets completely
 exhausted (i.e. all blocks' size are zero)

---
 include/picrin/gc.h |  2 +-
 src/gc.c            | 14 +++++++-------
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/include/picrin/gc.h b/include/picrin/gc.h
index 76c029ca..10872222 100644
--- a/include/picrin/gc.h
+++ b/include/picrin/gc.h
@@ -16,7 +16,7 @@ union header {
 };
 
 struct pic_heap {
-  union header base, *freep;
+  union header base, *freep, *endp;
   size_t heap_size;
 };
 
diff --git a/src/gc.c b/src/gc.c
index ab86bdef..b4719f8b 100644
--- a/src/gc.c
+++ b/src/gc.c
@@ -27,13 +27,11 @@ init_heap(struct pic_heap *heap)
   heap->base.s.mark = PIC_GC_UNMARK;
 
   heap->freep = heap->base.s.ptr;
-  heap->freep->s.size = nu - 1;
-  heap->freep->s.ptr = heap->freep + heap->freep->s.size;
+  heap->freep->s.size = nu;
+  heap->freep->s.ptr = &heap->base;
   heap->freep->s.mark = PIC_GC_UNMARK;
 
-  heap->freep->s.ptr->s.size = 0;
-  heap->freep->s.ptr->s.ptr = &heap->base;
-  heap->freep->s.ptr->s.mark = PIC_GC_UNMARK;
+  heap->endp = heap->freep + heap->freep->s.size;
 
 #if GC_DEBUG
   printf("freep = %p\n", heap->freep);
@@ -482,9 +480,10 @@ gc_sweep_phase(pic_state *pic)
 
   basep = &pic->heap->base;
   for (bp = basep->s.ptr; bp != basep; bp = bp->s.ptr) {
-    if (bp->s.size == 0)	/* end of page */
-      continue;
     for (p = bp + bp->s.size; p != bp->s.ptr; p += p->s.size) {
+      if (p == pic->heap->endp) {
+	goto end;
+      }
       if (! gc_is_marked(p)) {
 	if (s == NULL) {
 	  s = p;
@@ -498,6 +497,7 @@ gc_sweep_phase(pic_state *pic)
       gc_unmark(p);
     }
   }
+ end:
 
   /* free! */
   while (s) {