memory: generic allocator interface

This commit is contained in:
Andrei Alexeyev 2023-02-09 15:08:43 +01:00
parent 8547f443b7
commit e555126854
No known key found for this signature in database
GPG key ID: 72D26128040B9690
3 changed files with 142 additions and 0 deletions

82
src/memory/allocator.c Normal file
View file

@ -0,0 +1,82 @@
/*
* This software is licensed under the terms of the MIT License.
* See COPYING for further information.
* ---
* Copyright (c) 2011-2019, Lukas Weber <laochailan@web.de>.
* Copyright (c) 2012-2019, Andrei Alexeyev <akari@taisei-project.org>.
*/
#include "taisei.h"
#include "allocator.h"
#include "util.h"
#include "../util.h"
static void *allocator_default_alloc(Allocator *alloc, size_t size) {
return mem_alloc(size);
}
static void *allocator_default_alloc_aligned(Allocator *alloc, size_t size, size_t align) {
return mem_alloc_aligned(size, align);
}
static void allocator_default_free(Allocator *alloc, void *mem) {
mem_free(mem);
}
Allocator default_allocator = {
.procs = {
.alloc = allocator_default_alloc,
.alloc_aligned = allocator_default_alloc_aligned,
.free = allocator_default_free,
}
};
void *allocator_alloc(Allocator *alloc, size_t size) {
return NOT_NULL(alloc->procs.alloc)(alloc, size);
}
void *allocator_alloc_array(Allocator *alloc, size_t num_members, size_t size) {
return allocator_alloc(alloc, mem_util_calc_array_size(num_members, size));
}
void *allocator_alloc_aligned(Allocator *alloc, size_t size, size_t alignment) {
if(UNLIKELY(alloc->procs.alloc_aligned == NULL)) {
log_fatal("alloc_aligned not implemented by allocator");
}
return alloc->procs.alloc_aligned(alloc, size, alignment);
}
void allocator_free(Allocator *alloc, void *mem) {
return NOT_NULL(alloc->procs.free)(alloc, mem);
}
void allocator_deinit(Allocator *alloc) {
if(alloc->procs.deinit) {
alloc->procs.deinit(alloc);
}
}
static void *allocator_arena_alloc(Allocator *alloc, size_t size) {
return marena_alloc(NOT_NULL(alloc->userdata), size);
}
static void *allocator_arena_alloc_aligned(Allocator *alloc, size_t size, size_t alignment) {
return marena_alloc_aligned(NOT_NULL(alloc->userdata), size, alignment);
}
static void allocator_arena_free(Allocator *alloc, void *mem) {
// arena allocations can't be freed
}
void allocator_init_from_arena(Allocator *alloc, MemArena *arena) {
*alloc = (Allocator) {
.userdata = arena,
.procs = {
.alloc = allocator_arena_alloc,
.alloc_aligned = allocator_arena_alloc_aligned,
.free = allocator_arena_free,
}
};
}

59
src/memory/allocator.h Normal file
View file

@ -0,0 +1,59 @@
/*
* This software is licensed under the terms of the MIT License.
* See COPYING for further information.
* ---
* Copyright (c) 2011-2019, Lukas Weber <laochailan@web.de>.
* Copyright (c) 2012-2019, Andrei Alexeyev <akari@taisei-project.org>.
*/
#pragma once
#include "taisei.h"
#include "arena.h"
/*
* WARNING: Unlike the memory.h API, the allocator API is currently not guaranteed to return
* zero-initialized memory.
*/
typedef struct Allocator Allocator;
struct Allocator {
void *userdata;
struct {
void *(*alloc)(Allocator *alloc, size_t size);
void *(*alloc_aligned)(Allocator *alloc, size_t size, size_t align);
void (*free)(Allocator *alloc, void *mem);
void (*deinit)(Allocator *alloc);
} procs;
};
extern Allocator default_allocator;
void *allocator_alloc(Allocator *alloc, size_t size)
attr_alloc_size(2)
attr_malloc
attr_returns_allocated
attr_nonnull_all;
void *allocator_alloc_array(Allocator *alloc, size_t num_members, size_t size)
attr_alloc_size(2, 3)
attr_malloc
attr_returns_allocated
attr_nonnull_all;
void *allocator_alloc_aligned(Allocator *alloc, size_t size, size_t alignment)
attr_alloc_size(2)
attr_alloc_align(3)
attr_malloc
attr_returns_allocated
attr_nonnull_all;
void allocator_free(Allocator *alloc, void *mem)
attr_nonnull(1);
void allocator_deinit(Allocator *alloc)
attr_nonnull_all;
void allocator_init_from_arena(Allocator *alloc, MemArena *arena)
attr_nonnull_all;

View file

@ -1,5 +1,6 @@
memory_src = files(
'allocator.c',
'arena.c',
'memory.c',
)