blob: 0003108e693385f51ba6882cb1eb0a4ac9df8b13 (
plain)
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
|
// Copyright © Tavian Barnes <tavianator@tavianator.com>
// SPDX-License-Identifier: 0BSD
#include "alloc.h"
#include "bit.h"
#include "diag.h"
#include <errno.h>
#include <stdlib.h>
#include <string.h>
/** Portable aligned_alloc()/posix_memalign(). */
static void *xmemalign(size_t align, size_t size) {
bfs_assert(has_single_bit(align));
bfs_assert(align >= sizeof(void *));
bfs_assert((size & (align - 1)) == 0);
#if __APPLE__
void *ptr = NULL;
errno = posix_memalign(&ptr, align, size);
return ptr;
#else
return aligned_alloc(align, size);
#endif
}
void *alloc(size_t align, size_t size) {
bfs_assert(has_single_bit(align));
bfs_assert((size & (align - 1)) == 0);
if (align <= alignof(max_align_t)) {
return malloc(size);
} else {
return xmemalign(align, size);
}
}
void *zalloc(size_t align, size_t size) {
bfs_assert(has_single_bit(align));
bfs_assert((size & (align - 1)) == 0);
if (align <= alignof(max_align_t)) {
return calloc(1, size);
}
void *ret = xmemalign(align, size);
if (ret) {
memset(ret, 0, size);
}
return ret;
}
|
