1 files changed, 5 insertions, 0 deletions
diff --git a/include/rbtree.h b/include/rbtree.h
index b1ddcd2..aac175b 100644
--- a/include/rbtree.h
+++ b/include/rbtree.h
@@ -9,6 +9,7 @@
 
 #include "config.h"
 #include "sqfs/predef.h"
+#include "mempool.h"
 #include "compat.h"
 
 #include <stddef.h>
@@ -26,6 +27,10 @@ typedef struct rbtree_node_t {
 typedef struct rbtree_t {
 	rbtree_node_t *root;
 
+#ifndef NO_CUSTOM_ALLOC
+	mem_pool_t *pool;
+#endif
+
 	int (*key_compare)(const void *ctx,
 			   const void *lhs, const void *hrs);