radix tree test suite: add some more functionality

[mirror_ubuntu-bionic-kernel.git] / tools / testing / radix-tree / linux.c

diff --git a/tools/testing/radix-tree/linux.c b/tools/testing/radix-tree/linux.c
index 1f32a16a3848c7bc4f30eae6613191e06a0322d3..ff0452e8a0c4522181344cb9d2eb2c56105d481e 100644 (file)
--- a/tools/testing/radix-tree/linux.c
+++ b/tools/testing/radix-tree/linux.c
@@ -54,6 +54,21 @@ void kmem_cache_free(struct kmem_cache *cachep, void *objp)
        free(objp);
 }
 
+void *kmalloc(size_t size, gfp_t gfp)
+{
+       void *ret = malloc(size);
+       uatomic_inc(&nr_allocated);
+       return ret;
+}
+
+void kfree(void *p)
+{
+       if (!p)
+               return;
+       uatomic_dec(&nr_allocated);
+       free(p);
+}
+
 struct kmem_cache *
 kmem_cache_create(const char *name, size_t size, size_t offset,
        unsigned long flags, void (*ctor)(void *))