3fc56a0dc3
Rationale: "Basic allocation" refers to things outside of parsing proper, mostly initialization. If such allocations fail, the system is globally emory-starved from which it will likely not recover by returning failure. In this case, terminating the process is in fact the most robust strategy as it may mean the difference between a permanent hang and a temporary crash.
70 lines
2.2 KiB
C
70 lines
2.2 KiB
C
/* Arena allocator for Hammer.
|
|
* Copyright (C) 2012 Meredith L. Patterson, Dan "TQ" Hirsch
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation, version 2.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#ifndef HAMMER_ALLOCATOR__H__
|
|
#define HAMMER_ALLOCATOR__H__
|
|
#include <sys/types.h>
|
|
#include <setjmp.h>
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#if defined __llvm__
|
|
# if __has_attribute(malloc)
|
|
# define ATTR_MALLOC(n) __attribute__((malloc))
|
|
# else
|
|
# define ATTR_MALLOC(n)
|
|
# endif
|
|
#elif defined SWIG
|
|
# define ATTR_MALLOC(n)
|
|
#elif defined __GNUC__
|
|
# define ATTR_MALLOC(n) __attribute__((malloc, alloc_size(2)))
|
|
#else
|
|
# define ATTR_MALLOC(n)
|
|
#endif
|
|
|
|
// TODO(thequux): Turn this into an "HAllocatorVtable", and add a wrapper that also takes an environment pointer.
|
|
typedef struct HAllocator_ {
|
|
void* (*alloc)(struct HAllocator_* allocator, size_t size);
|
|
void* (*realloc)(struct HAllocator_* allocator, void* ptr, size_t size);
|
|
void (*free)(struct HAllocator_* allocator, void* ptr);
|
|
} HAllocator;
|
|
|
|
void* h_alloc(HAllocator* allocator, size_t size) ATTR_MALLOC(2);
|
|
|
|
typedef struct HArena_ HArena ; // hidden implementation
|
|
|
|
HArena *h_new_arena(HAllocator* allocator, size_t block_size); // pass 0 for default...
|
|
|
|
void* h_arena_malloc(HArena *arena, size_t count) ATTR_MALLOC(2);
|
|
void h_arena_free(HArena *arena, void* ptr); // For future expansion, with alternate memory managers.
|
|
void h_delete_arena(HArena *arena);
|
|
void h_arena_set_except(HArena *arena, jmp_buf *except);
|
|
|
|
typedef struct {
|
|
size_t used;
|
|
size_t wasted;
|
|
} HArenaStats;
|
|
|
|
void h_allocator_stats(HArena *arena, HArenaStats *stats);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // #ifndef LIB_ALLOCATOR__H__
|