1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
#include "moslib/mem/alloc.h"
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#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);
}
|
