mem.c

Go to the documentation of this file.

#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);
}