006a931ab8
When debugging a test with 'script/test -D', the DOT-graph generation code was sometimes causing reallocations that were not captured by the allocation tracker, because we explicitly disable allocation-tracking for that method in order to reduce noise when debugging memory leaks. By growing the relevant array *prior* to turning off allocation tracking, we can ensure that it is not reallocated within that function, avoiding false positive memory leak errors. Fixes #302
81 lines
1.6 KiB
C
81 lines
1.6 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);
|
|
|
|
static inline void *ts_malloc(size_t size) {
|
|
return ts_record_malloc(size);
|
|
}
|
|
|
|
static inline void *ts_calloc(size_t count, size_t size) {
|
|
return ts_record_calloc(count, size);
|
|
}
|
|
|
|
static inline void *ts_realloc(void *buffer, size_t size) {
|
|
return ts_record_realloc(buffer, size);
|
|
}
|
|
|
|
static inline void ts_free(void *buffer) {
|
|
ts_record_free(buffer);
|
|
}
|
|
|
|
#else
|
|
|
|
#include <stdlib.h>
|
|
|
|
static inline bool ts_toggle_allocation_recording(bool value) {
|
|
return false;
|
|
}
|
|
|
|
static inline void *ts_malloc(size_t size) {
|
|
void *result = malloc(size);
|
|
if (size > 0 && !result) {
|
|
fprintf(stderr, "tree-sitter failed to allocate %lu bytes", size);
|
|
exit(1);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static inline void *ts_calloc(size_t count, size_t size) {
|
|
void *result = calloc(count, size);
|
|
if (count > 0 && !result) {
|
|
fprintf(stderr, "tree-sitter failed to allocate %lu bytes", count * size);
|
|
exit(1);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static inline void *ts_realloc(void *buffer, size_t size) {
|
|
void *result = realloc(buffer, size);
|
|
if (size > 0 && !result) {
|
|
fprintf(stderr, "tree-sitter failed to reallocate %lu bytes", size);
|
|
exit(1);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static inline void ts_free(void *buffer) {
|
|
free(buffer);
|
|
}
|
|
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // TREE_SITTER_ALLOC_H_
|