Update listpool's interface to be on par with mempool's.

1 files changed, 26 insertions, 6 deletions

diff --git a/listpool/include/listpool.h b/listpool/include/listpool.h
index 1711449..85a3b27 100644
--- a/listpool/include/listpool.h
+++ b/listpool/include/listpool.h
@@ -45,13 +45,26 @@
     });                             \
   }
 
+/// Return the ith block.
+/// The block must have been allocated.
+#define listpool_get_block(POOL, INDEX) \
+  ((typeof((POOL)->blocks[0])*)listpool_get_block_(&(POOL)->pool, INDEX))
+
+/// Get the index to the given block.
+#define listpool_get_block_index(POOL, BLOCK_PTR) \
+  listpool_get_block_index_(&(POOL)->pool, BLOCK_PTR)
+
 /// Iterate over the used items of the pool.
-#define listpool_foreach(POOL, ITER, BODY)                    \
-  for (list* it_ = (POOL)->pool.used; it_; it_ = it_->next) { \
-    typeof((POOL)->blocks[0])* ITER =                         \
-        &(POOL)->blocks[it_ - (POOL)->pool.nodes];            \
-    (void)ITER;                                               \
-    BODY;                                                     \
+///
+/// The caller can use 'i' as the index of the current block.
+///
+/// It is valid to mempool_free() the object at each step of the iteration.
+#define listpool_foreach(POOL, ITER, BODY)                      \
+  for (list* it_ = (POOL)->pool.used; it_; it_ = it_->next) {   \
+    const size_t               i    = it_ - (POOL)->pool.nodes; \
+    typeof((POOL)->blocks[0])* ITER = &(POOL)->blocks[i];       \
+    (void)ITER;                                                 \
+    BODY;                                                       \
   }
 
 typedef struct listpool {
@@ -78,3 +91,10 @@ void* listpool_alloc_(listpool* pool);
 /// Free the block.
 /// The block pointer is conveniently set to 0.
 void listpool_free_(listpool* pool, void** block_ptr);
+
+/// Return the ith block.
+/// The block must have been allocated.
+void* listpool_get_block_(const listpool*, size_t block_index);
+
+/// Get the index to the given block.
+size_t listpool_get_block_index_(const listpool*, const void* block);