alloc.h

Go to the documentation of this file.

 
#ifndef __CWIST_CORE_MEM_ALLOC_H__
#define __CWIST_CORE_MEM_ALLOC_H__
 
#include <stddef.h>
#include <ttak/mem/owner.h>
#include <cwist/core/mem/arena.h>
 
ttak_owner_t *cwist_create_owner(void);
 
void *cwist_malloc(size_t size);
 
void *cwist_alloc(size_t size);
 
void *cwist_alloc_array(size_t count, size_t elem_size);
 
void *cwist_realloc(void *ptr, size_t new_size);
 
char *cwist_strdup(const char *src);
 
char *cwist_strndup(const char *src, size_t n);
 
void cwist_free(void *ptr);
 
#if defined(__GNUC__) || defined(__clang__)
#define CWIST_ATTRIBUTE_CLEANUP(func) __attribute__((cleanup(func)))
#else
#define CWIST_ATTRIBUTE_CLEANUP(func)
#endif
 
static inline void cwist_defer_free_cb(void *pp) {
    void **p = (void **)pp;
    if (*p) {
        cwist_free(*p);
    }
}
 
#define CWIST_DEFER_FREE CWIST_ATTRIBUTE_CLEANUP(cwist_defer_free_cb)
 
#define cwist_alloc_scoped(var, cast, size)                         \
    void *_cwist_scoped_##var CWIST_DEFER_FREE = cwist_alloc(size); \
    cast var = (cast)_cwist_scoped_##var
 
typedef struct {
    cwist_arena_t *arena;
    void *ptr;
    int heap; /* 1 if ptr came from cwist_alloc() instead of the arena */
} cwist_scratch_t;
 
static inline void *cwist_scratch_alloc(cwist_scratch_t *s, size_t size) {
    s->arena = cwist_arena_create(0);
    if (s->arena) {
        s->ptr = cwist_arena_alloc(s->arena, size);
        if (s->ptr) {
            s->heap = 0;
            return s->ptr;
        }
    }
    s->ptr = cwist_alloc(size);
    s->heap = 1;
    return s->ptr;
}
 
static inline void cwist_scratch_cleanup(void *sp) {
    cwist_scratch_t *s = (cwist_scratch_t *)sp;
    if (s->heap && s->ptr) {
        cwist_free(s->ptr);
    }
    if (s->arena) {
        cwist_arena_destroy(s->arena);
    }
}
 
#define CWIST_SCRATCH_DEFER CWIST_ATTRIBUTE_CLEANUP(cwist_scratch_cleanup)
 
#endif