manage file name allocations and allow them to be exported

File names used in locations need to live beyond a program delete, but
allocating them when exporting locations is too costly. Instead manage them in
the colm program and allow them to be exported so they can live beyond the colm
program. Will do something similar with the token text.

1 files changed, 20 insertions, 0 deletions

diff --git a/src/program.c b/src/program.c
index d2f9b200..d941cb31 100644
--- a/src/program.c
+++ b/src/program.c
@@ -195,6 +195,9 @@ program_t *colm_new_program( struct colm_sections *rtd )
 	vm_init( prg );
 
 	rtd->init_need();
+
+	prg->stream_fns = malloc( sizeof(char*) * 1 );
+	prg->stream_fns[0] = 0;
 	return prg;
 }
 
@@ -233,6 +236,13 @@ void colm_set_reduce_ctx( struct colm_program *prg, void *ctx )
 	prg->red_ctx = ctx;
 }
 
+const char **colm_extract_fns( struct colm_program *prg )
+{
+	const char **fns = prg->stream_fns;
+	prg->stream_fns = 0;
+	return fns;
+}
+
 int colm_delete_program( program_t *prg )
 {
 	tree_t **sp = prg->stack_root;
@@ -281,6 +291,16 @@ int colm_delete_program( program_t *prg )
 
 	vm_clear( prg );
 
+	if ( prg->stream_fns ) {
+		char **ptr = prg->stream_fns;
+		while ( *ptr != 0 ) {
+			free( *ptr );
+			ptr += 1;
+		}
+
+		free( prg->stream_fns );
+	}
+
 	free( prg );
 
 	return exit_status;