From 0acc93d13730caabd154e187276aab165c616004 Mon Sep 17 00:00:00 2001 From: Quinn Date: Thu, 19 Jun 2025 23:57:46 +0200 Subject: [PATCH] use an exit function, rather than manually freeing, since it's saver for when errors happen --- src/main.c | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/src/main.c b/src/main.c index 56b3493..94f4bb1 100644 --- a/src/main.c +++ b/src/main.c @@ -20,17 +20,22 @@ typedef struct dynrdat { ull dat[]; } dynrdat; +dynrdat* rdat = NULL; + static inline ull pow2_ceil(ull x) { x -= !!x; // if x=0, remains 0; else x -= 1 int lz = __builtin_clzll(x | 1); // get leading zeroes return (~0ULL >> lz) + 1; // bit-shift the maximum value by this amount of leading zeroes } +static void quit(void) { + free(rdat); +} + int main(int argc, char** argv) { // return the result if no input if (argc <= 1) return printf("%s\n", (clock() & 1) ? "heads" : "tails"); - - dynrdat* rdat = NULL; + atexit(quit); // loop through arguments for (unsigned i = 1; i < (unsigned)argc; ++i) { @@ -47,10 +52,7 @@ int main(int argc, char** argv) { size_t cap = pow2_ceil(c + !!mod); if (!rdat || rdat->cap < cap) { void* ptr = realloc(rdat, sizeof(dynrdat) + sizeof(ull) * cap); - if (!ptr) { - free(rdat); - error(1, "insufficient memory\n", ); - } + if (!ptr) error(1, "insufficient memory\n", ); rdat = ptr; rdat->cap = cap; @@ -84,7 +86,5 @@ int main(int argc, char** argv) { printf("results:\n heads: %llu\n tails: %llu\n", headsc, tailsc); } - free(rdat); - return 0; }