diff --git a/src/rvprof_core.c b/src/rvprof_core.c index 46ddcba..bec919d 100644 --- a/src/rvprof_core.c +++ b/src/rvprof_core.c @@ -72,4 +72,17 @@ void rvprof_init(const char* output_file){ // register atexit handler to automate cleanup register_atexit_handler(); -} \ No newline at end of file +} + +void rvprof_set_program_name(const char* program_name){ + if (g_rvprof.config.program_name){ + rvprof_free(g_rvprof.config.program_name, strlen(g_rvprof.config.program_name)+1); + } + if (program_name) { + g_rvprof.config.program_name = rvprof_malloc(strlen(program_name)+1); + if (g_rvprof.config.program_name) { + strcpy(g_rvprof.config.program_name, program_name); + } + } +} +