bfe69aea09
* format all (clang format 13) * added clang-format check * re-add missing newline * cargo doc was missing * more brackets * fixed fmt workflow * clang format * shellcheck * install clang-format-13 * update ubuntu for maximum clang-formattability * yml whitespaces * fmt * shellcheck only for .sh * oops path * ignored shellcheck warning
54 lines
1.3 KiB
C
54 lines
1.3 KiB
C
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#define MAP_SIZE (16 * 1024)
|
|
|
|
#ifdef _WIN32
|
|
#define posix_memalign(p, a, s) \
|
|
(((*(p)) = _aligned_malloc((s), (a))), *(p) ? 0 : errno)
|
|
#define RETADDR (uintptr_t) _ReturnAddress()
|
|
#else
|
|
#define RETADDR (uintptr_t) __builtin_return_address(0)
|
|
#endif
|
|
|
|
#ifdef __GNUC__
|
|
#define MAX(a, b) \
|
|
({ \
|
|
__typeof__(a) _a = (a); \
|
|
__typeof__(b) _b = (b); \
|
|
_a > _b ? _a : _b; \
|
|
})
|
|
#else
|
|
#define MAX(a, b) (((a) > (b)) ? (a) : (b))
|
|
#endif
|
|
|
|
size_t libafl_alloc_map[MAP_SIZE];
|
|
|
|
void *malloc(size_t size) {
|
|
uintptr_t k = RETADDR;
|
|
k = (k >> 4) ^ (k << 8);
|
|
k &= MAP_SIZE - 1;
|
|
libafl_alloc_map[k] = MAX(libafl_alloc_map[k], size);
|
|
|
|
// We cannot malloc in malloc.
|
|
// Hence, even realloc(NULL, size) would loop in an optimized build.
|
|
// We fall back to a stricter allocation function. Fingers crossed.
|
|
void *ret = NULL;
|
|
if (posix_memalign(&ret, 1 << 6, size) != 0) { return NULL; }
|
|
return ret;
|
|
}
|
|
|
|
void *calloc(size_t nmemb, size_t size) {
|
|
size *= nmemb;
|
|
|
|
uintptr_t k = RETADDR;
|
|
k = (k >> 4) ^ (k << 8);
|
|
k &= MAP_SIZE - 1;
|
|
libafl_alloc_map[k] = MAX(libafl_alloc_map[k], size);
|
|
|
|
void *ret = NULL;
|
|
if (posix_memalign(&ret, 1 << 6, size) != 0) { return NULL; };
|
|
return ret;
|
|
}
|