Heap Allocator Fix

This commit is contained in:
gingerBill 2015-11-29 18:04:33 +00:00
parent 0d09743505
commit 7c971f24fb
2 changed files with 109 additions and 240 deletions

View File

@ -5,7 +5,7 @@ gb single-file public domain libraries for C & C++
library | latest version | category | languages | description
----------------|----------------|----------|-----------|-------------
**gb_string.h** | 0.93 | strings | C, C++ | A better string library for C & C++
**gb.hpp** | 0.26 | misc | C++11 | (Experimental) A C++11 helper library without STL geared towards game development
**gb.hpp** | 0.26a | misc | C++11 | (Experimental) A C++11 helper library without STL geared towards game development
**gb_math.hpp** | 0.02b | math | C++11 | A C++11 math library geared towards game development
**gb_ini.h** | 0.91a | misc | C, C++ | A simple ini file loader library for C & C++

343
gb.hpp
View File

@ -1,4 +1,4 @@
// gb.hpp - v0.26 - public domain C++11 helper library - no warranty implied; use at your own risk
// gb.hpp - v0.26a - public domain C++11 helper library - no warranty implied; use at your own risk
// (Experimental) A C++11 helper library without STL geared towards game development
/*
@ -782,7 +782,6 @@ struct Atomic64 { u64 nonatomic; };
namespace atomic
{
// TODO(bill): Should these functions have suffixes or is the overloading fine?
u32 load(const volatile Atomic32* object);
void store(volatile Atomic32* object, u32 value);
u32 compare_exchange_strong(volatile Atomic32* object, u32 expected, u32 desired);
@ -860,7 +859,7 @@ struct Allocator
/// Allocates the specified amount of memory aligned to the specified alignment
void* (*alloc)(Allocator* a, usize size, usize align);
/// Deallocates/frees an allocation made with alloc()
void (*dealloc)(Allocator* a, const void* ptr);
void (*dealloc)(Allocator* a, void* ptr);
/// Returns the amount of usuable memory allocated at `ptr`.
///
/// If the allocator does not support tracking of the allocation size,
@ -879,19 +878,21 @@ struct Heap_Allocator : Allocator
usize size;
};
Mutex mutex;
bool32 use_mutex;
s64 total_allocated_count;
s64 allocation_count;
#if defined(GB_SYSTEM_WINDOWS)
HANDLE heap_handle;
HANDLE win32_heap_handle;
#endif
};
namespace heap_allocator
{
Heap_Allocator make();
Heap_Allocator make(bool use_mutex = true);
void destroy(Heap_Allocator* heap);
} // namespaceheap_allocator
} // namespace heap_allocator
// TODO(bill): Implement more Allocator types
@ -932,7 +933,7 @@ void* pointer_sub(void* ptr, usize bytes);
const void* pointer_add(const void* ptr, usize bytes);
const void* pointer_sub(const void* ptr, usize bytes);
void* set(void* ptr, u8 value, usize bytes);
void* set(void* ptr, usize bytes, u8 value);
void* zero(void* ptr, usize bytes);
void* copy(const void* src, usize bytes, void* dest);
@ -943,6 +944,9 @@ bool equals(const void* a, const void* b, usize bytes);
template <typename T>
T* zero_struct(T* ptr);
template <typename T>
T* zero_array(T* ptr, usize count);
template <typename T>
T* copy_array(const T* src_array, usize count, T* dest_array);
@ -956,7 +960,7 @@ void swap(T (& a)[N], T (& b)[N]);
} // namespace memory
void* alloc(Allocator* a, usize size, usize align = GB_DEFAULT_ALIGNMENT);
void dealloc(Allocator* a, const void* ptr); // TODO(bill): Should `ptr` be `const void*` or just `void*` ???
void dealloc(Allocator* a, void* ptr);
s64 allocated_size(Allocator* a, const void* ptr);
s64 total_allocated(Allocator* a);
@ -1508,7 +1512,6 @@ append(Array<T>* a, const T& item)
a->data[a->count++] = item;
}
template <typename T>
inline void
append(Array<T>* a, T&& item)
@ -2059,6 +2062,14 @@ zero_struct(T* ptr)
return static_cast<T*>(memory::zero(ptr, sizeof(T)));
}
template <typename T>
inline T*
zero_array(T* ptr, usize count)
{
return static_cast<T*>(memory::zero(ptr, count * sizeof(T)));
}
template <typename T>
inline T*
copy_array(const T* src_array, usize count, T* dest_array)
@ -2607,164 +2618,6 @@ current_id()
} // namespace thread
// NOTE(bill): This is just an idea I have been having
#if 0 || defined(GB_USE_C_STYLE_ALLOCATOR)
namespace heap_allocator
{
internal_linkage void*
alloc(Allocator* a, usize size, usize align)
{
Data* data = static_cast<Data*>(a->data);
mutex::lock(&data->mutex);
defer (mutex::unlock(&data->mutex));
usize total = size + align - (size % align);
void* ptr = malloc(total);
data->total_allocated_count += total;
data->allocation_count++;
return ptr;
}
internal_linkage void
dealloc(Allocator* a, const void* ptr)
{
if (!ptr)
return;
Data* data = static_cast<Data*>(a->data);
mutex::lock(&data->mutex);
defer (mutex::unlock(&data->mutex));
data->total_allocated_count -= allocated_size(a, ptr);
data->allocation_count--;
::free(const_cast<void*>(ptr));
}
internal_linkage s64
allocated_size(Allocator* a, const void* ptr)
{
Mutex* mutex = &static_cast<Data*>(a->data)->mutex;
mutex::lock(mutex);
defer (mutex::unlock(mutex));
#if defined(GB_SYSTEM_WINDOWS)
return static_cast<usize>(_msize(const_cast<void*>(ptr)));
#elif defined(GB_SYSTEM_OSX)
return static_cast<usize>(malloc_size(ptr));
#else
return static_cast<usize>(malloc_usable_size(const_cast<void*>(ptr)));
#endif
}
internal_linkage s64
total_allocated(Allocator* a)
{
return static_cast<Data*>(a->data)->total_allocated_count;
}
Allocator
make()
{
Allocator a = {};
a.data = malloc(sizeof(Data));
memory::zero(a.data, sizeof(Data));
a.alloc = heap_allocator::alloc;
a.dealloc = heap_allocator::dealloc;
a.allocated_size = heap_allocator::allocated_size;
a.total_allocated = heap_allocator::total_allocated;
return a;
}
void
destroy(Allocator* a)
{
::free(a->data);
*a = {};
}
} // namespace heap_allocator
namespace arena_allocator
{
internal_linkage void*
alloc(Allocator* a, usize size, usize align)
{
}
internal_linkage void
dealloc(Allocator* a, const void* ptr)
{
if (!ptr)
return;
}
internal_linkage s64
allocated_size(Allocator* a, const void* ptr)
{
}
internal_linkage s64
total_allocated(Allocator* a)
{
}
Allocator
make(Allocator* backing, usize size)
{
Allocator a = {};
physical_start =
return a;
}
Allocator
make(void* start, usize size)
{
}
void
destroy(Allocator* a)
{
}
void
clear(Allocator* a)
{
}
} // namespace arena_allocator
namespace temporary_arena_memory
{
Data
make(Allocator* arena)
{
}
void
free(Data* tmp)
{
}
} // namespace temporary_arena_memory
#else
namespace heap_allocator
{
@ -2775,12 +2628,16 @@ alloc(Allocator* a, usize size, usize align)
{
Heap_Allocator* heap = reinterpret_cast<Heap_Allocator*>(a);
if (heap->use_mutex)
mutex::lock(&heap->mutex);
usize total = size + align - (size % align);
#if defined (GB_SYSTEM_WINDOWS)
total += sizeof(Heap_Allocator::Header);
void* data = HeapAlloc(heap->heap_handle, 0, total);
void* data = HeapAlloc(heap->win32_heap_handle, 0, total);
Heap_Allocator::Header* h = static_cast<Heap_Allocator::Header*>(data);
h->size = total;
data = (h + 1);
@ -2793,36 +2650,53 @@ alloc(Allocator* a, usize size, usize align)
heap->total_allocated_count += total;
heap->allocation_count++;
if (heap->use_mutex)
mutex::unlock(&heap->mutex);
return data;
}
internal_linkage void
dealloc(Allocator* a, const void* ptr)
dealloc(Allocator* a, void* ptr)
{
if (!ptr)
return;
Heap_Allocator* heap = reinterpret_cast<Heap_Allocator*>(a);
if (heap->use_mutex)
mutex::lock(&heap->mutex);
heap->total_allocated_count -= allocated_size(heap, ptr);
heap->allocation_count--;
#if defined (GB_SYSTEM_WINDOWS)
ptr = static_cast<const Heap_Allocator::Header*>(ptr) + 1;
HeapFree(heap->heap_handle, 0, const_cast<void*>(ptr));
auto* header = static_cast<Heap_Allocator::Header*>(ptr) - 1;
HeapFree(heap->win32_heap_handle, 0, header);
#else
::free(const_cast<void*>(ptr));
::free(ptr);
#endif
if (heap->use_mutex)
mutex::unlock(&heap->mutex);
}
inline s64
allocated_size(Allocator* a, const void* ptr)
{
#if defined(GB_SYSTEM_WINDOWS)
auto* heap = reinterpret_cast<Heap_Allocator*>(a);
if (heap->use_mutex)
mutex::lock(&heap->mutex);
const auto* h = static_cast<const Heap_Allocator::Header*>(ptr) - 1;
return static_cast<usize>(h->size);
auto result = h->size;
if (heap->use_mutex)
mutex::unlock(&heap->mutex);
return static_cast<usize>(result);
#elif defined(GB_SYSTEM_OSX)
return static_cast<usize>(malloc_size(ptr));
@ -2838,16 +2712,30 @@ allocated_size(Allocator* a, const void* ptr)
inline s64
total_allocated(Allocator* a)
{
return reinterpret_cast<Heap_Allocator*>(a)->total_allocated_count;
auto* heap = reinterpret_cast<Heap_Allocator*>(a);
if (heap->use_mutex)
mutex::lock(&heap->mutex);
s64 result = heap->total_allocated_count;
if (heap->use_mutex)
mutex::unlock(&heap->mutex);
return result;
}
} // namespace functions
Heap_Allocator make()
Heap_Allocator
make(bool use_mutex)
{
Heap_Allocator heap = {};
heap.use_mutex = use_mutex;
if (use_mutex)
heap.mutex = mutex::make();
#if defined(GB_SYSTEM_WINDOWS)
heap.heap_handle = HeapCreate(0, 0, 0);
heap.win32_heap_handle = HeapCreate(0, 0, 0);
#endif
heap.alloc = functions::alloc;
@ -2860,8 +2748,11 @@ Heap_Allocator make()
void
destroy(Heap_Allocator* heap)
{
if (heap->use_mutex)
mutex::destroy(&heap->mutex);
#if defined (GB_SYSTEM_WINDOWS)
HeapDestroy(heap->heap_handle);
HeapDestroy(heap->win32_heap_handle);
#endif
}
} // namespace heap_allocator
@ -2890,7 +2781,7 @@ alloc(Allocator* a, usize size, usize align)
}
inline void
dealloc(Allocator* a, const void*) {}
dealloc(Allocator* a, void*) {}
inline s64 allocated_size(Allocator*, const void*) { return -1; }
@ -2988,8 +2879,6 @@ free(Temporary_Arena_Memory* tmp)
}
} // namespace temporary_arena_memory
#endif
////////////////////////////////
/// ///
/// Memory ///
@ -3036,7 +2925,7 @@ pointer_sub(const void* ptr, usize bytes)
}
GB_FORCE_INLINE void*
set(void* ptr, u8 value, usize bytes)
set(void* ptr, usize bytes, u8 value)
{
return memset(ptr, value, bytes);
}
@ -3044,7 +2933,7 @@ set(void* ptr, u8 value, usize bytes)
GB_FORCE_INLINE void*
zero(void* ptr, usize bytes)
{
return memory::set(ptr, 0, bytes);
return memory::set(ptr, bytes, 0);
}
GB_FORCE_INLINE void*
@ -3066,9 +2955,6 @@ equals(const void* a, const void* b, usize bytes)
}
} // namespace memory
// NOTE(bill): This is just an idea I have been having
#if 0 || defined(GB_USE_C_STYLE_ALLOCATOR)
inline void*
alloc(Allocator* a, usize size, usize align)
{
@ -3077,7 +2963,7 @@ alloc(Allocator* a, usize size, usize align)
}
inline void
dealloc(Allocator* a, const void* ptr)
dealloc(Allocator* a, void* ptr)
{
GB_ASSERT(a != nullptr);
if (ptr)
@ -3098,40 +2984,6 @@ total_allocated(Allocator* a)
return a->total_allocated(a);
}
#else
inline void*
alloc(Allocator* a, usize size, usize align)
{
GB_ASSERT(a != nullptr);
return a->alloc(a, size, align);
}
inline void
dealloc(Allocator* a, const void* ptr)
{
GB_ASSERT(a != nullptr);
if (ptr)
a->dealloc(a, ptr);
}
inline s64
allocated_size(Allocator* a, const void* ptr)
{
GB_ASSERT(a != nullptr);
return a->allocated_size(a, ptr);
}
inline s64
total_allocated(Allocator* a)
{
GB_ASSERT(a != nullptr);
return a->total_allocated(a);
}
#endif
////////////////////////////////
/// ///
/// String ///
@ -3171,7 +3023,7 @@ make(Allocator* a, const void* init_str, Size len)
return str;
}
void
inline void
free(String str)
{
if (str == nullptr)
@ -3387,8 +3239,6 @@ trim_space(String* str)
} // namespace string
namespace strconv
{
// NOTE(bill): Inspired by the golang strconv library but not exactly due to numerous reasons
@ -3464,22 +3314,31 @@ parse_f64(const char* str, f64* value)
bool
parse_int(const char* str, int base, s8* value)
{
// TODO(bill):
return false;
s64 v;
bool test = parse_int(str, base, &v);
if (test)
*value = v;
return test;
}
bool
parse_int(const char* str, int base, s16* value)
{
// TODO(bill):
return false;
s64 v;
bool test = parse_int(str, base, &v);
if (test)
*value = v;
return test;
}
bool
parse_int(const char* str, int base, s32* value)
{
// TODO(bill):
return false;
s64 v;
bool test = parse_int(str, base, &v);
if (test)
*value = v;
return test;
}
bool
@ -3492,22 +3351,31 @@ parse_int(const char* str, int base, s64* value)
bool
parse_uint(const char* str, int base, u8* value)
{
// TODO(bill):
return false;
u64 v;
bool test = parse_uint(str, base, &v);
if (test)
*value = v;
return test;
}
bool
parse_uint(const char* str, int base, u16* value)
{
// TODO(bill):
return false;
u64 v;
bool test = parse_uint(str, base, &v);
if (test)
*value = v;
return test;
}
bool
parse_uint(const char* str, int base, u32* value)
{
// TODO(bill):
return false;
u64 v;
bool test = parse_uint(str, base, &v);
if (test)
*value = v;
return test;
}
bool
@ -4263,6 +4131,7 @@ __GB_NAMESPACE_END
/*
Version History:
0.26a - Heap_Allocator Fix
0.26 - Better Allocation system
0.25a - Array bug fix
0.25 - Faster Heap_Allocator for Windows using HeapAlloc