mirror of
https://github.com/fish-shell/fish-shell
synced 2024-09-21 15:01:53 +00:00
100 lines
2 KiB
C++
100 lines
2 KiB
C++
/** \file halloc_util.c
|
|
|
|
A hierarchical memory allocation system. Works just like talloc
|
|
used in Samba, except that an arbitrary block allocated with
|
|
malloc() can be registered to be freed by halloc_free.
|
|
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
|
|
#include "fallback.h"
|
|
#include "util.h"
|
|
|
|
#include "common.h"
|
|
#include "halloc.h"
|
|
|
|
void *global_context=0;
|
|
|
|
void halloc_util_init()
|
|
{
|
|
global_context = halloc( 0, 0 );
|
|
}
|
|
|
|
void halloc_util_destroy()
|
|
{
|
|
halloc_free( global_context );
|
|
}
|
|
|
|
array_list_t *al_halloc( void *context )
|
|
{
|
|
array_list_t *res = (array_list_t *)halloc( context, sizeof( array_list_t ) );
|
|
if( !res )
|
|
DIE_MEM();
|
|
al_init( res );
|
|
halloc_register_function( context?context:res, (void (*)(void *)) &al_destroy, res );
|
|
return res;
|
|
}
|
|
|
|
string_buffer_t *sb_halloc( void *context )
|
|
{
|
|
string_buffer_t *res = (string_buffer_t *)halloc( context, sizeof( string_buffer_t ) );
|
|
if( !res )
|
|
DIE_MEM();
|
|
sb_init( res );
|
|
halloc_register_function( context?context:res, (void (*)(void *)) &sb_destroy, res );
|
|
return res;
|
|
}
|
|
|
|
/**
|
|
A function that takes a single parameter, which is a function pointer, and calls it.
|
|
*/
|
|
static void halloc_passthrough( void *f )
|
|
{
|
|
void (*func)() = (void (*)() )f;
|
|
func();
|
|
}
|
|
|
|
void halloc_register_function_void( void *context, void (*func)() )
|
|
{
|
|
halloc_register_function( context, &halloc_passthrough, (void *)func );
|
|
}
|
|
|
|
void *halloc_register( void *context, void *data )
|
|
{
|
|
if( !data )
|
|
return 0;
|
|
|
|
halloc_register_function( context, &free, data );
|
|
return data;
|
|
}
|
|
|
|
wchar_t *halloc_wcsdup( void *context, const wchar_t *in )
|
|
{
|
|
size_t len=wcslen(in);
|
|
wchar_t *out = (wchar_t *)halloc( context, sizeof( wchar_t)*(len+1));
|
|
|
|
if( out == 0 )
|
|
{
|
|
DIE_MEM();
|
|
}
|
|
memcpy( out, in, sizeof( wchar_t)*(len+1));
|
|
return out;
|
|
}
|
|
|
|
wchar_t *halloc_wcsndup( void * context, const wchar_t *in, int c )
|
|
{
|
|
wchar_t *res = (wchar_t *)halloc( context, sizeof(wchar_t)*(c+1) );
|
|
if( res == 0 )
|
|
{
|
|
DIE_MEM();
|
|
}
|
|
wcslcpy( res, in, c+1 );
|
|
res[c] = L'\0';
|
|
return res;
|
|
}
|