#define VG_MAX_ALLOCATIONS 64
#define VG_FUZZ_ALLOCATIONS
-#define VG_DEBUG_ALLOCATIONS
typedef struct vg_linear_allocator vg_linear_allocator;
typedef struct vg_allocation_meta vg_allocation_meta;
*scratch;
int use_libc_malloc;
+ u32 quota;
}
static vg_mem;
{
data = malloc( size );
-#ifdef VG_DEBUG_ALLOCATIONS
- vg_info( "malloc: %p[%u]\n", data, size );
-#endif
-
vg_allocation_meta *meta = &alloc->alloc_table[ alloc->allocation_count ];
meta->type = k_allocation_type_block;
meta->data = data;
alloc->alloc_table[ alloc->allocation_count-1 ].data = data;
alloc->last_alloc = data;
-
-#ifdef VG_DEBUG_ALLOCATIONS
- vg_info( "realloc: %p (->%u)\n", data, newsize );
-#endif
return data;
}
else
return alloc->cur;
}
-/* get the capacity of allocator. TODO, should return free space? */
+/* get the capacity of allocator. */
VG_STATIC u32 vg_linear_get_capacity( void *buffer )
{
vg_linear_allocator *alloc = vg_linear_header( buffer );
return alloc->size;
}
+/* get the remaining size of the allocator */
+VG_STATIC u32 vg_linear_remaining( void *buffer )
+{
+ vg_linear_allocator *alloc = vg_linear_header( buffer );
+ return alloc->size - alloc->cur;
+}
+
/* yeet all memory from linear allocator */
VG_STATIC void vg_linear_clear( void *buffer )
{
vg_linear_allocator *alloc = vg_linear_header( buffer );
-#ifdef VG_DEBUG_ALLOCATIONS
- if( alloc->flags & VG_MEMORY_SYSTEM )
- vg_info( "linear_clear on %p\n", alloc );
-#endif
-
/* libc mode we recursively free any allocations made */
if( vg_mem.use_libc_malloc && (alloc->flags & VG_MEMORY_SYSTEM) )
{
if( meta->type == k_allocation_type_block )
{
free( meta->data );
-
-#ifdef VG_DEBUG_ALLOCATIONS
- vg_info( "free(%p) [%d]\n", meta->data, meta->type );
-#endif
}
else
{
vg_linear_clear( meta->data );
vg_linear_allocator *sub = vg_linear_header( meta->data );
-#ifdef VG_DEBUG_ALLOCATIONS
- vg_info( "free(%p) [alloc_table]\n", sub->alloc_table );
- vg_info( "free(%p) [%d]\n", sub, meta->type );
-#endif
-
free( sub->alloc_table );
free( sub );
}
else
header = malloc( sizeof(vg_linear_allocator) );
-#ifdef VG_DEBUG_ALLOCATIONS
- vg_info( "linear_create(sub) malloc(%p)\n", header );
-#endif
-
meta->data = header+1;
meta->type = k_allocation_type_linear;
}
header = malloc( sizeof(vg_linear_allocator) );
else
header = malloc( block_size );
-
-#ifdef VG_DEBUG_ALLOCATIONS
- vg_info( "linear_create malloc(%p)\n", header );
-#endif
}
header->allocation_count = 0;
{
u32 table_size = sizeof(vg_allocation_meta)*VG_MAX_ALLOCATIONS;
header->alloc_table = malloc( table_size );
-#ifdef VG_DEBUG_ALLOCATIONS
- vg_info( "linear_create(%p) [alloc_table]\n", header->alloc_table );
-#endif
}
else
header->alloc_table = NULL;
}
/* request all the memory we need in advance */
-VG_STATIC void vg_prealloc_quota( u32 size )
+VG_STATIC void vg_set_mem_quota( u32 size )
+{
+ vg_mem.quota = size;
+}
+
+VG_STATIC void vg_alloc_quota(void)
{
u32 size_scratch = 10*1024*1024;
- size = VG_MAX( size, size_scratch );
+ u32 size = VG_MAX( vg_mem.quota, size_scratch );
vg_mem.rtmemory = vg_create_linear_allocator( NULL, size, VG_MEMORY_SYSTEM );
vg_mem.scratch = vg_create_linear_allocator( vg_mem.rtmemory,