From 22aad63775337f61ebed66e4f7c2696287c749fa Mon Sep 17 00:00:00 2001 From: Alexandre Julliard Date: Sat, 27 May 2006 13:23:24 +0200 Subject: [PATCH] wmc: Clean output files when aborting on an error or signal. --- tools/wmc/wmc.c | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) diff --git a/tools/wmc/wmc.c b/tools/wmc/wmc.c index f852f2bf8c8..9a19654aa71 100644 --- a/tools/wmc/wmc.c +++ b/tools/wmc/wmc.c @@ -116,6 +116,17 @@ time_t now; /* The time of start of wmc */ int getopt (int argc, char *const *argv, const char *optstring); static void segvhandler(int sig); +static void cleanup_files(void) +{ + if (output_name) unlink( output_name ); + if (header_name) unlink( header_name ); +} + +static void exit_on_signal( int sig ) +{ + exit(1); /* this will call the atexit functions */ +} + int main(int argc,char *argv[]) { extern char* optarg; @@ -126,7 +137,13 @@ int main(int argc,char *argv[]) int i; int cmdlen; + atexit( cleanup_files ); signal(SIGSEGV, segvhandler); + signal( SIGTERM, exit_on_signal ); + signal( SIGINT, exit_on_signal ); +#ifdef SIGHUP + signal( SIGHUP, exit_on_signal ); +#endif now = time(NULL); @@ -269,7 +286,8 @@ int main(int argc,char *argv[]) write_rc_file(output_name); if(!rcinline) write_bin_files(); - + output_name = NULL; + header_name = NULL; return 0; }