52 lines
1.5 KiB
C
52 lines
1.5 KiB
C
#include "alloc.h"
|
|
#include "me/mem/mem.h"
|
|
#include "api.h"
|
|
#include <stdlib.h>
|
|
|
|
static void *ts_malloc_default(size_t size)
|
|
{
|
|
void *result = malloc(size);
|
|
if (size > 0 && !result)
|
|
{
|
|
fprintf(stderr, "tree-sitter failed to allocate %zu bytes", size);
|
|
abort();
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static void *ts_calloc_default(size_t count, size_t size)
|
|
{
|
|
void *result = calloc(count, size);
|
|
if (count > 0 && !result)
|
|
{
|
|
fprintf(stderr, "tree-sitter failed to allocate %zu bytes", count * size);
|
|
abort();
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static void *ts_realloc_default(void *buffer, size_t size)
|
|
{
|
|
void *result = realloc(buffer, size);
|
|
if (size > 0 && !result)
|
|
{
|
|
fprintf(stderr, "tree-sitter failed to reallocate %zu bytes", size);
|
|
abort();
|
|
}
|
|
return result;
|
|
}
|
|
|
|
// Allow clients to override allocation functions dynamically
|
|
TS_PUBLIC void *(*ts_current_malloc)(size_t) = mem_alloc;
|
|
TS_PUBLIC void *(*ts_current_calloc)(size_t, size_t) = mem_alloc_array;
|
|
TS_PUBLIC void *(*ts_current_realloc)(void *, size_t) = mem_realloc;
|
|
TS_PUBLIC void (*ts_current_free)(void *) = mem_free;
|
|
|
|
void ts_set_allocator(void *(*new_malloc)(size_t size), void *(*new_calloc)(size_t count, size_t size),
|
|
void *(*new_realloc)(void *ptr, size_t size), void (*new_free)(void *ptr))
|
|
{
|
|
ts_current_malloc = new_malloc ? new_malloc : ts_malloc_default;
|
|
ts_current_calloc = new_calloc ? new_calloc : ts_calloc_default;
|
|
ts_current_realloc = new_realloc ? new_realloc : ts_realloc_default;
|
|
ts_current_free = new_free ? new_free : free;
|
|
}
|