bendns' repos / helix
Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'vendor/tree-sitter/src/alloc.c')
-rw-r--r--vendor/tree-sitter/src/alloc.c48
1 files changed, 48 insertions, 0 deletions
diff --git a/vendor/tree-sitter/src/alloc.c b/vendor/tree-sitter/src/alloc.c
new file mode 100644
index 00000000..e5cb1d5e
--- /dev/null
+++ b/vendor/tree-sitter/src/alloc.c
@@ -0,0 +1,48 @@
+#include "alloc.h"
+#include "tree_sitter/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) = ts_malloc_default;
+TS_PUBLIC void *(*ts_current_calloc)(size_t, size_t) = ts_calloc_default;
+TS_PUBLIC void *(*ts_current_realloc)(void *, size_t) = ts_realloc_default;
+TS_PUBLIC void (*ts_current_free)(void *) = 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;
+}