87 lines
1.8 KiB
C
87 lines
1.8 KiB
C
#ifndef TREE_SITTER_ALLOC_H_
|
|
#define TREE_SITTER_ALLOC_H_
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#include <stdlib.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
|
|
#if defined(TREE_SITTER_TEST)
|
|
|
|
void *ts_record_malloc(size_t);
|
|
void *ts_record_calloc(size_t, size_t);
|
|
void *ts_record_realloc(void *, size_t);
|
|
void ts_record_free(void *);
|
|
bool ts_toggle_allocation_recording(bool);
|
|
|
|
#define ts_malloc ts_record_malloc
|
|
#define ts_calloc ts_record_calloc
|
|
#define ts_realloc ts_record_realloc
|
|
#define ts_free ts_record_free
|
|
|
|
#else
|
|
|
|
// Allow clients to override allocation functions
|
|
|
|
#ifndef ts_malloc
|
|
#define ts_malloc ts_malloc_default
|
|
#endif
|
|
#ifndef ts_calloc
|
|
#define ts_calloc ts_calloc_default
|
|
#endif
|
|
#ifndef ts_realloc
|
|
#define ts_realloc ts_realloc_default
|
|
#endif
|
|
#ifndef ts_free
|
|
#define ts_free ts_free_default
|
|
#endif
|
|
|
|
#include <stdlib.h>
|
|
|
|
static inline bool ts_toggle_allocation_recording(bool value) {
|
|
(void)value;
|
|
return false;
|
|
}
|
|
|
|
|
|
static inline 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);
|
|
exit(1);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static inline 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);
|
|
exit(1);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static inline 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);
|
|
exit(1);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static inline void ts_free_default(void *buffer) {
|
|
free(buffer);
|
|
}
|
|
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // TREE_SITTER_ALLOC_H_
|