fish-shell/intern.cpp

/** \file intern.c

    Library for pooling common strings

*/
#include "config.h"


#include <stdlib.h>
#include <stdio.h>
#include <wchar.h>
#include <unistd.h>

#include "fallback.h"
#include "util.h"

#include "wutil.h"
#include "common.h"
#include "intern.h"

/**
   Table of interned strings
*/
static hash_table_t *intern_table=0;

/**
   Table of static interned strings
*/
static hash_table_t *intern_static_table=0;

const wchar_t *intern( const wchar_t *in )
{
    ASSERT_IS_MAIN_THREAD();
	const wchar_t *res=0;

//	debug( 0, L"intern %ls", in );
	
	if( !in )
		return 0;
	
	if( !intern_table )
	{
		intern_table = (hash_table_t *)malloc( sizeof( hash_table_t ) );
		if( !intern_table )
		{
			DIE_MEM();
		}
		hash_init( intern_table, &hash_wcs_func, &hash_wcs_cmp );
	}
	
	if( intern_static_table )
	{
		res = (const wchar_t *)hash_get( intern_static_table, in );
	}
	
	if( !res )
	{
		res = (const wchar_t *)hash_get( intern_table, in );
		
		if( !res )
		{
			res = wcsdup( in );
			if( !res )
			{
				DIE_MEM();
			}
			
			hash_put( intern_table, res, res );
		}
	}
	
	return res;
}

const wchar_t *intern_static( const wchar_t *in )
{
	const wchar_t *res=0;
	
	if( !in )
		return 0;
	
	if( !intern_static_table )
	{
		intern_static_table = (hash_table_t *)malloc( sizeof( hash_table_t ) );
		if( !intern_static_table )
		{
			DIE_MEM();			
		}
		hash_init( intern_static_table, &hash_wcs_func, &hash_wcs_cmp );
	}
	
	res = (const wchar_t *)hash_get( intern_static_table, in );
	
	if( !res )
	{
		res = in;
		hash_put( intern_static_table, res, res );
	}
	
	return res;
}

/**
   Free the specified key/value pair. Should only be called by intern_free_all at shutdown
*/
static void clear_value( void *key, void *data )
{
	debug( 3,  L"interned string: '%ls'", data );	
	free( (void *)data );
}

void intern_free_all()
{
	if( intern_table )
	{
		hash_foreach( intern_table, &clear_value );		
		hash_destroy( intern_table );		
		free( intern_table );
		intern_table=0;		
	}

	if( intern_static_table )
	{
		hash_destroy( intern_static_table );		
		free( intern_static_table );
		intern_static_table=0;		
	}
	
}