aboutsummaryrefslogtreecommitdiff
path: root/src/alloc.h
blob: 770adba36a1408da567a6ea6a27bf65854bcd39d (plain) (blame)
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
#ifndef ORYX_ALLOC_H
#define ORYX_ALLOC_H

#include <stdalign.h>
#include <stddef.h>

#include "common.h"

typedef struct _arena *arena;

/* Allocate a buffer of NMEMB elements of size SIZE.  If PTR is non-null then
   reallocate the buffer it points to.  Aborts on out-of-memory or overflow. */
void *bufalloc(void *ptr, size_t nmemb, size_t size)
	__attribute__((returns_nonnull, warn_unused_result, alloc_size(2, 3)));

/* Allocate a buffer of NMEMB elements of size SIZE with alignment ALIGN using
   the arena-allocator A. */
void *arena_alloc(arena *a, size_t nmemb, size_t size, size_t align)
	__attribute__((returns_nonnull, nonnull, warn_unused_result, malloc,
                   alloc_size(2, 3), alloc_align(4)));

void *_arena_grow(arena *a, void *ptr, size_t old_nmemb, size_t new_nmemb,
                  size_t size, size_t align)
	__attribute__((returns_nonnull, nonnull, warn_unused_result));

/* Deallocate all memory associated with the arena A. */
void arena_free(arena *a)
	__attribute__((nonnull));

/* Allocate a buffer of N elements of type T using the arena-allocator A. */
#define arena_new(a, T, n) ((T *)arena_alloc((a), (n), sizeof(T), alignof(T)))

#define arena_grow(a, p, T, on, nn)                                            \
	((T *)_arena_grow((a), (p), (on), (nn), sizeof(T), alignof(T)))

#endif /* !ORYX_ALLOC_H */