Define functions to get the number of used blocks.

1 files changed, 16 insertions, 1 deletions

diff --git a/mempool/src/mempool.c b/mempool/src/mempool.c
index b09038b..46f1053 100644
--- a/mempool/src/mempool.c
+++ b/mempool/src/mempool.c
@@ -24,6 +24,7 @@ bool mempool_make_(
 
   pool->block_size_bytes = block_size_bytes;
   pool->num_blocks       = num_blocks;
+  pool->num_used_blocks  = 0;
   pool->head             = 0;
   pool->used             = 0;
   pool->trap             = true;
@@ -91,6 +92,8 @@ void* mempool_alloc_(mempool* pool) {
   pool->head      = head->next_free;
   head->next_free = 0;
 
+  pool->num_used_blocks++;
+
   return block;
 }
 
@@ -119,6 +122,8 @@ void mempool_free_(mempool* pool, void** block_ptr) {
     pool->used = 0;
   }
 
+  pool->num_used_blocks--;
+
   *block_ptr = 0;
 }
 
@@ -136,9 +141,19 @@ size_t mempool_get_block_index_(const mempool* pool, const void* block) {
   return block_byte_index / pool->block_size_bytes;
 }
 
+size_t mempool_block_size_bytes_(const mempool* pool) {
+  assert(pool);
+  return pool->block_size_bytes;
+}
+
 size_t mempool_capacity_(const mempool* pool) {
   assert(pool);
-  return pool->num_blocks * pool->block_size_bytes;
+  return pool->num_blocks;
+}
+
+size_t mempool_size_(const mempool* pool) {
+  assert(pool);
+  return pool->num_used_blocks;
 }
 
 void mempool_enable_traps_(mempool* pool, bool enable) {