hashsum: cleanup at exit

sjaeckel · sjaeckel · commit eb75c894db2e · 2017-09-21T18:37:37.000+02:00
diff --git a/demos/hashsum.c b/demos/hashsum.c
@@ -38,6 +38,11 @@
 
 static char* hashsum;
 
+static void cleanup(void)
+{
+   free(hashsum);
+}
+
 static void die(int status)
 {
    unsigned long w, x;
@@ -53,7 +58,6 @@ static void die(int status)
       }
    }
    if (w != 0) fprintf(o, "\n");
-   free(hashsum);
    exit(status);
 }
 
@@ -173,6 +177,7 @@ int main(int argc, char **argv)
    unsigned char hash_buffer[MAXBLOCKSIZE];
 
    hashsum = strdup(basename(argv[0]));
+   atexit(cleanup);
 
    /* You need to register algorithms before using them */
    register_all_ciphers();

Original file line number	Diff line number	Diff line change
`@@ -38,6 +38,11 @@`
`38`	`38`
`39`	`39`	`static char* hashsum;`
`40`	`40`
	`41`	`+static void cleanup(void)`
	`42`	`+{`
	`43`	`+ free(hashsum);`
	`44`	`+}`
	`45`	`+`
`41`	`46`	`static void die(int status)`
`42`	`47`	`{`
`43`	`48`	`unsigned long w, x;`
`@@ -53,7 +58,6 @@ static void die(int status)`
`53`	`58`	`}`
`54`	`59`	`}`
`55`	`60`	`if (w != 0) fprintf(o, "\n");`
`56`		`- free(hashsum);`
`57`	`61`	`exit(status);`
`58`	`62`	`}`
`59`	`63`
`@@ -173,6 +177,7 @@ int main(int argc, char **argv)`
`173`	`177`	`unsigned char hash_buffer[MAXBLOCKSIZE];`
`174`	`178`
`175`	`179`	`hashsum = strdup(basename(argv[0]));`
	`180`	`+ atexit(cleanup);`
`176`	`181`
`177`	`182`	`/* You need to register algorithms before using them */`
`178`	`183`	`register_all_ciphers();`