From 276e139ec69f4d2ae3f7618316a7f584da5d2c17 Mon Sep 17 00:00:00 2001
From: Yuichi Nishiwaki <yuichi.nishiwaki@gmail.com>
Date: Mon, 14 Oct 2013 18:21:56 +0900
Subject: [PATCH] fix bugs around GC

---
 include/picrin/gc.h | 2 +-
 src/gc.c            | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/include/picrin/gc.h b/include/picrin/gc.h
index ccc6271c..4318671e 100644
--- a/include/picrin/gc.h
+++ b/include/picrin/gc.h
@@ -16,7 +16,7 @@ union header {
 };
 
 struct heap_page {
-  union header *base, *freep;
+  union header *base, *freep, *endp;
   size_t heap_size;
 };
 
diff --git a/src/gc.c b/src/gc.c
index 058224ae..7b105a15 100644
--- a/src/gc.c
+++ b/src/gc.c
@@ -20,6 +20,8 @@ init_heap_page(struct heap_page *heap)
   heap->freep = freep = base->s.ptr;
   freep->s.ptr = base;
   freep->s.size = ((char *)p + PIC_HEAP_SIZE - (char *)freep) / sizeof(union header);
+
+  heap->endp = freep + freep->s.size;
 }
 
 void *
@@ -217,6 +219,8 @@ gc_sweep_phase(pic_state *pic)
   base = pic->heap->base;
   for (p = base->s.ptr; p != base; p = p->s.ptr) {
     for (bp = p + p->s.size; bp != p->s.ptr; bp += bp->s.size) {
+      if (p >= p->s.ptr && bp == pic->heap->endp)
+	break;
       if (is_marked(bp)) {
 	gc_unmark(bp);
 	continue;
@@ -237,6 +241,7 @@ gc_sweep_phase(pic_state *pic)
       else {
 	p->s.ptr = bp;
       }
+      break;
     }
   }
 }