Browse Source
This moves the reusable parts of the memory pool logic used by fast-import.c into its own file for use by other components. Signed-off-by: Jameson Miller <jamill@microsoft.com> Signed-off-by: Junio C Hamano <gitster@pobox.com>maint
Jameson Miller
7 years ago
committed by
Junio C Hamano
4 changed files with 91 additions and 69 deletions
@ -0,0 +1,55 @@ |
|||||||
|
/* |
||||||
|
* Memory Pool implementation logic. |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "cache.h" |
||||||
|
#include "mem-pool.h" |
||||||
|
|
||||||
|
static struct mp_block *mem_pool_alloc_block(struct mem_pool *mem_pool, size_t block_alloc) |
||||||
|
{ |
||||||
|
struct mp_block *p; |
||||||
|
|
||||||
|
mem_pool->pool_alloc += sizeof(struct mp_block) + block_alloc; |
||||||
|
p = xmalloc(st_add(sizeof(struct mp_block), block_alloc)); |
||||||
|
p->next_block = mem_pool->mp_block; |
||||||
|
p->next_free = (char *)p->space; |
||||||
|
p->end = p->next_free + block_alloc; |
||||||
|
mem_pool->mp_block = p; |
||||||
|
|
||||||
|
return p; |
||||||
|
} |
||||||
|
|
||||||
|
void *mem_pool_alloc(struct mem_pool *mem_pool, size_t len) |
||||||
|
{ |
||||||
|
struct mp_block *p; |
||||||
|
void *r; |
||||||
|
|
||||||
|
/* round up to a 'uintmax_t' alignment */ |
||||||
|
if (len & (sizeof(uintmax_t) - 1)) |
||||||
|
len += sizeof(uintmax_t) - (len & (sizeof(uintmax_t) - 1)); |
||||||
|
|
||||||
|
for (p = mem_pool->mp_block; p; p = p->next_block) |
||||||
|
if (p->end - p->next_free >= len) |
||||||
|
break; |
||||||
|
|
||||||
|
if (!p) { |
||||||
|
if (len >= (mem_pool->block_alloc / 2)) { |
||||||
|
mem_pool->pool_alloc += len; |
||||||
|
return xmalloc(len); |
||||||
|
} |
||||||
|
|
||||||
|
p = mem_pool_alloc_block(mem_pool, mem_pool->block_alloc); |
||||||
|
} |
||||||
|
|
||||||
|
r = p->next_free; |
||||||
|
p->next_free += len; |
||||||
|
return r; |
||||||
|
} |
||||||
|
|
||||||
|
void *mem_pool_calloc(struct mem_pool *mem_pool, size_t count, size_t size) |
||||||
|
{ |
||||||
|
size_t len = st_mult(count, size); |
||||||
|
void *r = mem_pool_alloc(mem_pool, len); |
||||||
|
memset(r, 0, len); |
||||||
|
return r; |
||||||
|
} |
@ -0,0 +1,34 @@ |
|||||||
|
#ifndef MEM_POOL_H |
||||||
|
#define MEM_POOL_H |
||||||
|
|
||||||
|
struct mp_block { |
||||||
|
struct mp_block *next_block; |
||||||
|
char *next_free; |
||||||
|
char *end; |
||||||
|
uintmax_t space[FLEX_ARRAY]; /* more */ |
||||||
|
}; |
||||||
|
|
||||||
|
struct mem_pool { |
||||||
|
struct mp_block *mp_block; |
||||||
|
|
||||||
|
/* |
||||||
|
* The amount of available memory to grow the pool by. |
||||||
|
* This size does not include the overhead for the mp_block. |
||||||
|
*/ |
||||||
|
size_t block_alloc; |
||||||
|
|
||||||
|
/* The total amount of memory allocated by the pool. */ |
||||||
|
size_t pool_alloc; |
||||||
|
}; |
||||||
|
|
||||||
|
/* |
||||||
|
* Alloc memory from the mem_pool. |
||||||
|
*/ |
||||||
|
void *mem_pool_alloc(struct mem_pool *pool, size_t len); |
||||||
|
|
||||||
|
/* |
||||||
|
* Allocate and zero memory from the memory pool. |
||||||
|
*/ |
||||||
|
void *mem_pool_calloc(struct mem_pool *pool, size_t count, size_t size); |
||||||
|
|
||||||
|
#endif |
Loading…
Reference in new issue