traxys/tree-sitter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Avoid allocator from being switched more than once

Browse source
This commit is contained in:
Mingkai Dong 2021-12-18 16:45:18 +08:00
parent b9b051e933
commit 486ea2569d
2 changed files with 17 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
lib/include/tree_sitter/api.h
View file

@ -148,10 +148,15 @@ typedef struct {
/**
* Switch to a new allocator.
*
* Returns the old allocator. The caller needs to take care of any allocated
* memory managed by the old allocator.
* Returns true iff the switch successes.
*
* This function can only be invoked *once*, before tree-sitter has allocated
* any memory via malloc/calloc/realloc. Otherwise, memory bugs could occur
* since some memory is allocated by the old allocator, but freed by the new
* one.
*
*/
TSAllocator *ts_set_allocator(TSAllocator *new_alloc);
bool ts_set_allocator(TSAllocator *new_alloc);
/**
* Create a new parser.

13
lib/src/alloc.c
View file

@ -70,10 +70,15 @@ TSAllocator *ts_allocator = &ts_default_allocator;
#endif // defined(TREE_SITTER_ALLOCATION_TRACKING)
TSAllocator *ts_set_allocator(TSAllocator *new_alloc)
bool ts_set_allocator(TSAllocator *new_alloc)
{
TSAllocator *old = ts_allocator;
ts_allocator = new_alloc;
return old;
static bool using_default_allocator = true;
if (!using_default_allocator) {
fprintf(stderr, "tree-sitter's allocator can only be set once!");
return false;
}
using_default_allocator = false;
ts_allocator = new_alloc;
return true;
}