Replaced arena
This commit is contained in:
@@ -23,16 +23,7 @@ SOFTWARE.
|
|||||||
#ifndef CI2_ARENA_H
|
#ifndef CI2_ARENA_H
|
||||||
#define CI2_ARENA_H
|
#define CI2_ARENA_H
|
||||||
|
|
||||||
#include "../ci2_platform.h"
|
#include <stddef.h>
|
||||||
#include "../error/ci2_exception.h"
|
|
||||||
|
|
||||||
enum
|
|
||||||
{
|
|
||||||
DEFAULT = 0, /* default: fail-handler (throws) + zero */
|
|
||||||
SOFT_FAIL = 1u << 0, /* on failure return NULL */
|
|
||||||
HARD_FAIL = 1u << 1, /* on failure call abort() */
|
|
||||||
NO_ZERO = 1u << 2, /* do not zero memory on success */
|
|
||||||
};
|
|
||||||
|
|
||||||
struct Arena
|
struct Arena
|
||||||
{
|
{
|
||||||
@@ -41,27 +32,24 @@ struct Arena
|
|||||||
};
|
};
|
||||||
typedef struct Arena Arena;
|
typedef struct Arena Arena;
|
||||||
|
|
||||||
/* Forward definition of an exception */
|
/* Initialize an Arena with an existing buffer */
|
||||||
CI2_API struct Exception arena_oom;
|
extern void
|
||||||
|
arena_back(struct Arena* a, char* buf, ptrdiff_t len);
|
||||||
|
|
||||||
/* Initialize an Arena from a buffer. */
|
/* Allocate a new Arena */
|
||||||
CI2_API void
|
extern struct Arena
|
||||||
arena_init(struct Arena* a, void* buf, ptrdiff_t len);
|
|
||||||
|
|
||||||
/* Dynamically allocate an Arena */
|
|
||||||
CI2_API struct Arena
|
|
||||||
arena_new(ptrdiff_t cap);
|
arena_new(ptrdiff_t cap);
|
||||||
|
|
||||||
/* Get the remaining capicity of an arena. */
|
/* Get the current capacity of an Arena */
|
||||||
CI2_API ptrdiff_t
|
extern ptrdiff_t
|
||||||
arena_size(struct Arena a);
|
arena_capacity(struct Arena a);
|
||||||
|
|
||||||
/* Make an allocation from an Arena. */
|
/* Make an allocation from an Arena and do not zero.*/
|
||||||
CI2_API void*
|
extern void*
|
||||||
arena_alloc(struct Arena* a,
|
arena_malloc(struct Arena* a, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count);
|
||||||
int flags,
|
|
||||||
ptrdiff_t size,
|
/* Make an allocation from an Arena and zero.*/
|
||||||
ptrdiff_t align,
|
extern void*
|
||||||
ptrdiff_t count);
|
arena_calloc(struct Arena* a, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count);
|
||||||
|
|
||||||
#endif // ci2_arena.h
|
#endif // ci2_arena.h
|
||||||
|
|||||||
+31
-26
@@ -21,21 +21,22 @@ SOFTWARE.
|
|||||||
|
|
||||||
* --------------------------------------------------------------------------*/
|
* --------------------------------------------------------------------------*/
|
||||||
#include "../include/platform/mem/ci2_arena.h"
|
#include "../include/platform/mem/ci2_arena.h"
|
||||||
#include <string.h>
|
#include <assert.h>
|
||||||
|
#include <stdint.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
struct Exception arena_oom = {"Arena is out of memory!"};
|
#include <string.h>
|
||||||
|
|
||||||
/* Initialize an Arena from a buffer. */
|
/* Initialize an Arena with an existing buffer */
|
||||||
void
|
void
|
||||||
arena_init(struct Arena* a, void* buf, ptrdiff_t len)
|
arena_back(struct Arena* a, char* buf, ptrdiff_t len)
|
||||||
{
|
{
|
||||||
CI2_ASSERT(buf != NULL);
|
assert(a != NULL);
|
||||||
CI2_ASSERT(len > 0);
|
assert(buf != NULL);
|
||||||
a->beg = (char*)buf;
|
a->beg = buf;
|
||||||
a->end = (char*)buf + len;
|
a->end = buf + len;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Dynamically allocate an Arena */
|
/* Allocate a new Arena */
|
||||||
struct Arena
|
struct Arena
|
||||||
arena_new(ptrdiff_t cap)
|
arena_new(ptrdiff_t cap)
|
||||||
{
|
{
|
||||||
@@ -45,36 +46,40 @@ arena_new(ptrdiff_t cap)
|
|||||||
return a;
|
return a;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Get the remaining capicity of an arena. */
|
/* Get the current capacity of an Arena */
|
||||||
ptrdiff_t
|
ptrdiff_t
|
||||||
arena_size(struct Arena a)
|
arena_capacity(struct Arena a)
|
||||||
{
|
{
|
||||||
ptrdiff_t res = a.end - a.beg;
|
ptrdiff_t res = a.end - a.beg;
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Make an allocation from an Arena. */
|
/* Make an allocation from an Arena and do not zero.*/
|
||||||
void*
|
void*
|
||||||
arena_alloc(struct Arena* a,
|
arena_calloc(struct Arena* a, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count)
|
||||||
int flags,
|
|
||||||
ptrdiff_t size,
|
|
||||||
ptrdiff_t align,
|
|
||||||
ptrdiff_t count)
|
|
||||||
{
|
{
|
||||||
ptrdiff_t padding = -(uintptr_t)a->beg & (align - 1);
|
ptrdiff_t padding = -(uintptr_t)a->beg & (align - 1);
|
||||||
ptrdiff_t available = a->end - a->beg - padding;
|
ptrdiff_t available = a->end - a->beg - padding;
|
||||||
if (available < 0 || count > available / size) {
|
if (available < 0 || count > available / size) {
|
||||||
if (flags & SOFT_FAIL)
|
return NULL;
|
||||||
return 0;
|
|
||||||
|
|
||||||
if (flags & HARD_FAIL)
|
|
||||||
CI2_DEBUG_TRAP();
|
|
||||||
|
|
||||||
CI2_RAISE(arena_oom);
|
|
||||||
}
|
}
|
||||||
void* p = a->beg + padding;
|
|
||||||
ptrdiff_t total = padding + count * size;
|
ptrdiff_t total = padding + count * size;
|
||||||
|
void* p = a->beg + total;
|
||||||
a->beg += total;
|
a->beg += total;
|
||||||
return flags & NO_ZERO ? p : memset(p, 0, total);
|
return memset(p, 0, total);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* Make an allocation from an Arena and zero.*/
|
||||||
|
void*
|
||||||
|
arena_malloc(struct Arena* a, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count)
|
||||||
|
{
|
||||||
|
ptrdiff_t padding = -(uintptr_t)a->beg & (align - 1);
|
||||||
|
ptrdiff_t available = a->end - a->beg - padding;
|
||||||
|
if (available < 0 || count > available / size) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
ptrdiff_t total = padding + count * size;
|
||||||
|
void* p = a->beg + total;
|
||||||
|
a->beg += total;
|
||||||
|
return p;
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user