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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
#include "memory.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
void *safe_malloc(size_t size) {
for (size_t i = 0; i < MAX_ALLOC_ATTEMPTS; i++) {
void *p = malloc(size);
if (p)
return p;
if (i < MAX_ALLOC_ATTEMPTS - 1) {
fprintf(stderr, "Failed malloc. Retrying in 1s...\n");
sleep(1);
}
}
fprintf(stderr, "FATAL: Failed malloc %d times. Exiting...", MAX_ALLOC_ATTEMPTS);
exit(EXIT_FAILURE);
}
void *safe_realloc(void *ptr, size_t size) {
if (!ptr)
return safe_malloc(size);
for (size_t i = 0; i < MAX_ALLOC_ATTEMPTS; i++) {
void *p = realloc(ptr, size);
if (p)
return p;
if (i < MAX_ALLOC_ATTEMPTS - 1) {
fprintf(stderr, "Failed realloc. Retrying in 1s...\n");
sleep(1);
}
}
fprintf(stderr, "FATAL: Failed realloc %d times. Exiting...", MAX_ALLOC_ATTEMPTS);
exit(EXIT_FAILURE);
}
// expand an array
// returns pointer to the expanded array
void *array_expand(void *arr, size_t el_size, size_t *cap, size_t to_fit) {
while (*cap < to_fit)
*cap *= 2;
return safe_realloc(arr, *cap * el_size);
}
void *array_append(void *arr, size_t el_size, size_t *len, size_t *cap, void *val) {
if (*len >= *cap)
arr = array_expand(arr, el_size, cap, *len + 1);
memcpy((char *)arr + (*len)++ * el_size, val, el_size);
return arr;
}
void *array_concat(void *arr1, size_t el_size, size_t *len1, size_t *cap, void *arr2, size_t len2) {
if (*len1 + len2 > *cap)
arr1 = array_expand(arr1, el_size, cap, *len1 + len2);
memcpy((char *)arr1 + *len1 * el_size, arr2, len2 * el_size);
*len1 += len2;
return arr1;
}
|
