1 files changed, 121 insertions, 0 deletions
diff --git a/src/vpp/mem/mem.c b/src/vpp/mem/mem.c
new file mode 100644
index 00000000000..d438c970447
--- /dev/null
+++ b/src/vpp/mem/mem.c
@@ -0,0 +1,121 @@
+#include <stdio.h>
+#include <vppinfra/mem.h>
+
+extern void * __libc_malloc (size_t);
+extern void __libc_free (void *);
+extern void * __libc_calloc (size_t, size_t);
+extern void * __libc_realloc (void *, size_t);
+extern void * __libc_valloc (size_t);
+extern void * __libc_memalign (size_t, size_t);
+extern void * __libc_pvalloc (size_t);
+
+__thread u64 vpp_mem_no_vpp_heap;
+
+static void no_heap (void)
+{
+  vpp_mem_no_vpp_heap++;
+
+  if (1 == vpp_mem_no_vpp_heap)
+    fprintf (stderr, "vpp mem: libc allocation requested but no vpp heap ready, defaulting to libc.\n");
+}
+
+static_always_inline int
+check_vpp_heap (void)
+{
+  if (PREDICT_TRUE (clib_mem_get_per_cpu_heap () != 0))
+    return 1;
+
+  no_heap ();
+  return 0;
+}
+
+void *
+malloc(size_t size)
+{
+  if (!check_vpp_heap ())
+    return __libc_malloc (size);
+
+  return clib_mem_alloc (size);
+}
+
+void
+free(void *p)
+{
+  if (!p)
+    return;
+
+  if (!check_vpp_heap ())
+    return __libc_free (p);
+
+  clib_mem_free (p);
+}
+
+void *
+calloc(size_t nmemb, size_t size)
+{
+  void * p;
+
+  if (!check_vpp_heap ())
+    return __libc_calloc (nmemb, size);
+
+  p = clib_mem_alloc (nmemb * size);
+  clib_memset (p, 0, nmemb * size);
+  return p;
+}
+
+void *
+realloc(void *p, size_t size)
+{
+  if (!check_vpp_heap ())
+    return __libc_realloc (p, size);
+
+  return clib_mem_realloc (p, size, clib_mem_size (p));
+}
+
+int
+posix_memalign(void **memptr, size_t alignment, size_t size)
+{
+  if (!check_vpp_heap ())
+    *memptr = __libc_memalign (alignment, size);
+  else
+    *memptr = clib_mem_alloc_aligned (size, alignment);
+  return 0;
+}
+
+void *
+aligned_alloc(size_t alignment, size_t size)
+{
+  if (!check_vpp_heap ())
+    return __libc_memalign (alignment, size);
+
+  return clib_mem_alloc_aligned (size, alignment);
+}
+
+void *
+valloc(size_t size)
+{
+  if (!check_vpp_heap ())
+    return __libc_valloc (size);
+
+  return clib_mem_alloc_aligned (size, clib_mem_get_page_size ());
+}
+
+void *memalign(size_t alignment, size_t size)
+{
+  if (!check_vpp_heap ())
+    return __libc_memalign (alignment, size);
+
+  return clib_mem_alloc_aligned (size, alignment);
+}
+
+void *
+pvalloc(size_t size)
+{
+  uword pagesz;
+
+  if (!check_vpp_heap ())
+    return __libc_pvalloc (size);
+
+  pagesz = clib_mem_get_page_size ();
+  return clib_mem_alloc_aligned (round_pow2 (size, pagesz), pagesz);
+}