flex: Overriding The Default Memory Management
21.2 Overriding The Default Memory Management
=============================================
Flex calls the functions 'yyalloc', 'yyrealloc', and 'yyfree' when it
needs to allocate or free memory. By default, these functions are
wrappers around the standard C functions, 'malloc', 'realloc', and
'free', respectively. You can override the default implementations by
telling flex that you will provide your own implementations.
To override the default implementations, you must do two things:
1. Suppress the default implementations by specifying one or more of
the following options:
* '%option noyyalloc'
* '%option noyyrealloc'
* '%option noyyfree'.
2. Provide your own implementation of the following functions: (1)
// For a non-reentrant scanner
void * yyalloc (size_t bytes);
void * yyrealloc (void * ptr, size_t bytes);
void yyfree (void * ptr);
// For a reentrant scanner
void * yyalloc (size_t bytes, void * yyscanner);
void * yyrealloc (void * ptr, size_t bytes, void * yyscanner);
void yyfree (void * ptr, void * yyscanner);
In the following example, we will override all three memory routines.
We assume that there is a custom allocator with garbage collection. In
order to make this example interesting, we will use a reentrant scanner,
passing a pointer to the custom allocator through 'yyextra'.
%{
#include "some_allocator.h"
%}
/* Suppress the default implementations. */
%option noyyalloc noyyrealloc noyyfree
%option reentrant
/* Initialize the allocator. */
%{
#define YY_EXTRA_TYPE struct allocator*
#define YY_USER_INIT yyextra = allocator_create();
%}
%%
.|\n ;
%%
/* Provide our own implementations. */
void * yyalloc (size_t bytes, void* yyscanner) {
return allocator_alloc (yyextra, bytes);
}
void * yyrealloc (void * ptr, size_t bytes, void* yyscanner) {
return allocator_realloc (yyextra, bytes);
}
void yyfree (void * ptr, void * yyscanner) {
/* Do nothing -- we leave it to the garbage collector. */
}
---------- Footnotes ----------
(1) It is not necessary to override all (or any) of the memory
management routines. You may, for example, override 'yyrealloc', but
not 'yyfree' or 'yyalloc'.