#include "includes.h"
TALLOC_CTX *talloc_init(void)
{
TALLOC_CTX *t;
t = (TALLOC_CTX *)malloc(sizeof(*t));
if (!t) return NULL;
t->list = NULL;
t->total_alloc_size = 0;
return t;
}
void *talloc(TALLOC_CTX *t, size_t size)
{
void *p;
struct talloc_chunk *tc;
if (size == 0) return NULL;
p = malloc(size);
if (!p) return p;
tc = malloc(sizeof(*tc));
if (!tc) {
SAFE_FREE(p);
return NULL;
}
tc->ptr = p;
tc->size = size;
tc->next = t->list;
t->list = tc;
t->total_alloc_size += size;
return p;
}
void *talloc_realloc(TALLOC_CTX *t, void *ptr, size_t size)
{
struct talloc_chunk *tc;
void *new_ptr;
if (size == 0)
return NULL;
if (ptr == NULL)
return talloc(t, size);
for (tc=t->list; tc; tc=tc->next) {
if (tc->ptr == ptr) {
new_ptr = Realloc(ptr, size);
if (new_ptr) {
t->total_alloc_size += (size - tc->size);
tc->size = size;
tc->ptr = new_ptr;
}
return new_ptr;
}
}
return NULL;
}
void talloc_destroy_pool(TALLOC_CTX *t)
{
struct talloc_chunk *c;
if (!t)
return;
while (t->list) {
c = t->list->next;
SAFE_FREE(t->list->ptr);
SAFE_FREE(t->list);
t->list = c;
}
t->list = NULL;
t->total_alloc_size = 0;
}
void talloc_destroy(TALLOC_CTX *t)
{
if (!t)
return;
talloc_destroy_pool(t);
memset(t, 0, sizeof(*t));
SAFE_FREE(t);
}
size_t talloc_pool_size(TALLOC_CTX *t)
{
return t->total_alloc_size;
}
void *talloc_zero(TALLOC_CTX *t, size_t size)
{
void *p = talloc(t, size);
if (p)
memset(p, '\0', size);
return p;
}
void *talloc_memdup(TALLOC_CTX *t, void *p, size_t size)
{
void *newp = talloc(t,size);
if (!newp)
return 0;
memcpy(newp, p, size);
return newp;
}
char *talloc_strdup(TALLOC_CTX *t, char *p)
{
return talloc_memdup(t, p, strlen(p) + 1);
}