From 4bae9413f75ab9d12dd5fedfccc633e7efce7e42 Mon Sep 17 00:00:00 2001 From: Peter Eisentraut Date: Mon, 2 Dec 2024 10:35:37 +0100 Subject: [PATCH v1 05/19] seg: Use palloc() instead of malloc() for flex --- contrib/seg/segscan.l | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/contrib/seg/segscan.l b/contrib/seg/segscan.l index 35c3a11ac14..004ac07fae2 100644 --- a/contrib/seg/segscan.l +++ b/contrib/seg/segscan.l @@ -38,6 +38,9 @@ static char *scanbuf; // FIXME %option noinput %option nounput %option noyywrap +%option noyyalloc +%option noyyrealloc +%option noyyfree %option warn %option prefix="seg_yy" @@ -124,3 +127,30 @@ seg_scanner_finish(yyscan_t yyscanner) yylex_destroy(yyscanner); pfree(scanbuf); } + +/* + * Interface functions to make flex use palloc() instead of malloc(). + * It'd be better to make these static, but flex insists otherwise. + */ + +void * +yyalloc(yy_size_t size, yyscan_t yyscanner) +{ + return palloc(size); +} + +void * +yyrealloc(void *ptr, yy_size_t size, yyscan_t yyscanner) +{ + if (ptr) + return repalloc(ptr, size); + else + return palloc(size); +} + +void +yyfree(void *ptr, yyscan_t yyscanner) +{ + if (ptr) + pfree(ptr); +} -- 2.47.1