From 4f4466155b6d9b5259ccd956551a675ad693e773 Mon Sep 17 00:00:00 2001 From: Mosder Date: Fri, 31 Jul 2026 14:50:59 +0200 Subject: feat: created safe_malloc and safe_realloc --- src/mem/alloc.c | 44 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) (limited to 'src/mem/alloc.c') diff --git a/src/mem/alloc.c b/src/mem/alloc.c index e69de29..7acea02 100644 --- a/src/mem/alloc.c +++ b/src/mem/alloc.c @@ -0,0 +1,44 @@ +#include "moslib/mem/alloc.h" + +#include +#include +#include + +#define ERR_MESS(fl, ln, fn, msg) "%s:%d - error in function %s:\n%s\n", fl, ln, fn, msg + +void *safe_malloc_fn(size_t size, const char *file, int line, const char *func) { + for (size_t i = 0; i < MOSLIB_MAX_ALLOC_ATTEMPTS; i++) { + void *p = malloc(size); + if (p) + return p; + if (i < MOSLIB_MAX_ALLOC_ATTEMPTS - 1) { + fprintf(stderr, ERR_MESS(file, line, func, "malloc failed - retrying in 1s...")); + sleep(1); + } + } + + char errmes[64]; + sprintf(errmes, "FATAL: malloc failed %d times - exiting...", MOSLIB_MAX_ALLOC_ATTEMPTS); + fprintf(stderr, ERR_MESS(file, line, func, errmes)); + exit(EXIT_FAILURE); +} + +void *safe_realloc_fn(void *ptr, size_t size, const char *file, int line, const char *func) { + if (!ptr) + return safe_malloc_fn(size, file, line, func); + + for (size_t i = 0; i < MOSLIB_MAX_ALLOC_ATTEMPTS; i++) { + void *p = realloc(ptr, size); + if (p) + return p; + if (i < MOSLIB_MAX_ALLOC_ATTEMPTS - 1) { + fprintf(stderr, ERR_MESS(file, line, func, "realloc failed - retrying in 1s...")); + sleep(1); + } + } + + char errmes[64]; + sprintf(errmes, "FATAL: realloc failed %d times - exiting...", MOSLIB_MAX_ALLOC_ATTEMPTS); + fprintf(stderr, ERR_MESS(file, line, func, errmes)); + exit(EXIT_FAILURE); +} -- cgit v1.2.3