Add used list to mempool; fix mem iteration.

1 files changed, 13 insertions, 11 deletions

diff --git a/mem/include/mem.h b/mem/include/mem.h
index b3d9157..bcff39f 100644
--- a/mem/include/mem.h
+++ b/mem/include/mem.h
@@ -92,17 +92,19 @@
 /// The caller can use 'i' as the index of the current chunk.
 ///
 /// It is valid to mem_free() the chunk at each step of the iteration.
-#define mem_foreach(MEM, ITER, BODY)                                \
-  size_t i = 0;                                                     \
-  do {                                                              \
-    if ((MEM)->mem.chunks[i].used) {                                \
-      __typeof__((MEM)->object[0])* ITER =                          \
-          &(((__typeof__((MEM)->object[0])*)(MEM)->mem.blocks))[i]; \
-      (void)ITER;                                                   \
-      BODY;                                                         \
-    }                                                               \
-    i = (MEM)->chunks[i].next;                                      \
-  } while (i);
+#define mem_foreach(MEM, ITER, BODY)                                  \
+  {                                                                   \
+    size_t i = 0;                                                     \
+    do {                                                              \
+      if ((MEM)->mem.chunks[i].used) {                                \
+        __typeof__((MEM)->object[0])* ITER =                          \
+            &(((__typeof__((MEM)->object[0])*)(MEM)->mem.blocks))[i]; \
+        (void)ITER;                                                   \
+        BODY;                                                         \
+      }                                                               \
+      i = (MEM)->mem.chunks[i].next;                                  \
+    } while (i);                                                      \
+  }
 
 // -----------------------------------------------------------------------------