---

yaml
---
r: 8135
b: refs/heads/snap-stage3
c: a6dde77
h: refs/heads/master
i:
  8133: de30aca
  8131: dac5e7b
  8127: 25e8274
v: v3

Author: nikomatsakis

[refs]

@@ -1,6 +1,6 @@
 ---
 refs/heads/master: 2898dcc5d97da9427ac367542382b6239d9c0bbf
 refs/heads/snap-stage1: e33de59e47c5076a89eadeb38f4934f58a3618a6
-refs/heads/snap-stage3: 5f8e0ddccfe4d9797b1e2b293788d023d9916e31
+refs/heads/snap-stage3: a6dde7766ee9c2a397ef05da5c87c064f9f06118
 refs/heads/try: 2898dcc5d97da9427ac367542382b6239d9c0bbf
 refs/tags/release-0.1: 1f5c5126e96c79d22cb7862f75304136e204f105

branches/snap-stage3/src/rt/rust_obstack.cpp

@@ -18,7 +18,7 @@
 #endif
 
 #undef DPRINT
-#define DPRINT(fmt, ...)
+#define DPRINT(fmt, ...) 
 
 const size_t DEFAULT_CHUNK_SIZE = 128;
 const size_t MAX_CHUNK_SIZE = (1024*64);
@@ -38,16 +38,20 @@ struct rust_obstack_alloc {
 
 void *
 rust_obstack_chunk::alloc(size_t len, type_desc *tydesc) {
-    alen = align_to(alen, DEFAULT_ALIGNMENT);
+    DPRINT("alloc(%lu) alen=%lu size=%lu\n", len, alen, size);
 
-    if (sizeof(rust_obstack_alloc) + len > size - alen) {
-        DPRINT("Not enough space, len=%lu!\n", len);
+    size_t aligned_alen = align_to(alen, DEFAULT_ALIGNMENT);
+    size_t end_alen = aligned_alen + sizeof(rust_obstack_alloc) + len;
+    if (end_alen > size) {
+        DPRINT("Not enough space, len=%lu alen=%lu aligned=%lu end=%lu!\n",
+               len, alen, aligned_alen, end_alen);
         return NULL;    // Not enough space.
     }
 
-    rust_obstack_alloc *a = new(data + alen) rust_obstack_alloc(len, tydesc);
-    alen += sizeof(*a) + len;
+    rust_obstack_alloc *a =
+      new(data + aligned_alen) rust_obstack_alloc(len, tydesc);
     memset(a->data, '\0', len); // FIXME: For GC.
+    alen = end_alen;
     return &a->data;
 }
 
@@ -63,7 +67,9 @@ rust_obstack_chunk::free(void *ptr) {
 
 void *
 rust_obstack_chunk::mark() {
-    return data + alen;
+    uint8_t *m = data + alen;
+    assert (m >= data && m <= data+size);
+    return m;
 }
 
 // Allocates the given number of bytes in a new chunk.
@@ -109,7 +115,9 @@ rust_obstack::free(void *ptr) {
 
     assert(chunk);
     while (!chunk->free(ptr)) {
-        DPRINT("deleting chunk at %p\n", chunk);
+        DPRINT("deleting chunk at %p (ptr=%p, data=%p-%p)\n",
+               chunk, ptr,
+               chunk->data, chunk->data + chunk->size);
         rust_obstack_chunk *prev = chunk->prev;
         task->free(chunk);
         chunk = prev;
@@ -119,7 +127,11 @@ rust_obstack::free(void *ptr) {
 
 void *
 rust_obstack::mark() {
-    return chunk ? chunk->mark() : NULL;
+    void *m = chunk ? chunk->mark() : NULL;
+    DPRINT("mark == %p, chunk == %p, data == %p-%p\n", m, chunk,
+           (chunk ? chunk->data : NULL),
+           (chunk ? chunk->data + chunk->size : NULL));
+    return m;
 }