pkmem: add wrapper functions for if bkt is unknown

1 files changed, 16 insertions, 0 deletions

diff --git a/pkmem.h b/pkmem.h
index 4687638..4473241 100644
--- a/pkmem.h
+++ b/pkmem.h
@@ -27,8 +27,10 @@ bool pk_memory_is_in_bucket(const void* ptr, const struct pk_membucket* bkt);
 
 void* pk_new_base(size_t sz, size_t alignment);
 void* pk_new_bkt(size_t sz, size_t alignment, struct pk_membucket* bkt);
+void* pk_new(size_t sz, size_t alignment, struct pk_membucket* bkt);
 void pk_delete_base(const void* ptr, size_t sz);
 void pk_delete_bkt(const void* ptr, size_t sz, struct pk_membucket* bkt);
+void pk_delete(const void* ptr, size_t sz, struct pk_membucket* bkt);
 
 #if defined(__cplusplus)
 
@@ -488,6 +490,13 @@ pk_new_base(size_t sz, size_t alignment)
 	return pk_new_bkt(sz, alignment, bkt);
 }
 
+void*
+pk_new(size_t sz, size_t alignment, struct pk_membucket* bkt)
+{
+	if (bkt != NULL) return pk_new_bkt(sz, alignment, bkt);
+	return pk_new_base(sz, alignment);
+}
+
 void
 pk_delete_bkt(const void* ptr, size_t sz, struct pk_membucket* bkt)
 {
@@ -595,4 +604,11 @@ pk_delete_base(const void* ptr, size_t sz)
 	pk_delete_bkt(ptr, sz, bkt);
 }
 
+void
+pk_delete(const void* ptr, size_t sz, struct pk_membucket* bkt)
+{
+	if (bkt != NULL) return pk_delete_bkt(ptr, sz, bkt);
+	return pk_delete_base(ptr, sz);
+}
+
 #endif /* PK_IMPL_MEM */