doubango/trunk/tinySAK/src/tsk_buffer.c

/*
* Copyright (C) 2009 Mamadou Diop.
*
* Contact: Mamadou Diop <diopmamadou@yahoo.fr>
*	
* This file is part of Open Source Doubango Framework.
*
* DOUBANGO is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*	
* DOUBANGO is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
* GNU Lesser General Public License for more details.
*	
* You should have received a copy of the GNU General Public License
* along with DOUBANGO.
*
*/

/**@file tsk_buffer.c
 * @brief Data buffer.
 *
 * @author Mamadou Diop <diopmamadou(at)yahoo.fr>
 *
 * @date Created: Sat Nov 8 16:54:58 2009 mdiop
 */
#include "tsk_buffer.h"
#include "tsk_memory.h"

#include <stdio.h>
#include <string.h>

#if defined(_MSC_VER) || TSK_UNDER_WINDOWS
#	define vsnprintf	_vsnprintf
#endif

int tsk_buffer_appendEx(tsk_buffer_t* self, const char* format, ...)
{
	/*
	 * I suppose that sizeof(char) = 1-byte
	 */
	int len = 0;
	va_list list;
	char *buffer;
	size_t oldsize;

	if(!self)
	{
		return -1;
	}

	oldsize = self->size;
	buffer = (char*)TSK_BUFFER_DATA(self);
	
	/* initialize variable arguments */
	va_start(list, format);
	
	/* compute destination len for windows mobile
	*/
#if defined(_WIN32_WCE)
	{
		int n;
		len = (strlen(format)*2);
		buffer = tsk_realloc(buffer, (oldsize+len));
		for(;;)
		{
			if( (n = vsnprintf((char*)(buffer + oldsize), len, format, list)) >= 0 && (n<=len) )
			{
				len = n;
				break;
			}
			else
			{
				len += 5;
				buffer = tsk_realloc(buffer, (oldsize+len));
			}
		}
	}
#else
    len = vsnprintf(0, 0, format, list);
    buffer = tsk_realloc(buffer, oldsize+len+1);
    vsnprintf((buffer + oldsize), len
#if !defined(_MSC_VER) || defined(__GNUC__)
		+1
#endif
		, format, list);
#endif
	
	/* reset variable arguments */
	va_end( list );

	self->data = buffer;
	self->size = (oldsize+len);
	
	return 0;
}


int tsk_buffer_append(tsk_buffer_t* self, const void* data, size_t size)
{
	if(self && size)
	{
		size_t oldsize = self->size;
		size_t newsize = oldsize + size;
		
		self->data = tsk_realloc(self->data, newsize);
		if(self->data)
		{
			memcpy((void*)(TSK_BUFFER_TO_U8(self) + oldsize), data, size);
			self->size = newsize;
			return 0;
		}
	}
	return -1;
}

int tsk_buffer_realloc(tsk_buffer_t* self, size_t size)
{
	if(self)
	{
		if(size == 0){
			return tsk_buffer_cleanup(self);
		}

		if(self->size == 0){
			self->data = tsk_calloc(size, sizeof(uint8_t));
		}
		else{
			self->data = tsk_realloc(self->data, size);
		}

		self->size = size;
		return 0;
	}
	return -1;
}

int tsk_buffer_remove(tsk_buffer_t* self, size_t position, size_t size)
{
	if(self)
	{
		if((position + size) > self->size)
		{
			size = self->size - position;
			memcpy(((uint8_t*)self->data) + position, ((uint8_t*)self->data) + position + size, 
				self->size-(position+size));
			return tsk_buffer_realloc(self, (self->size-size));
		}
	}
	return -1;
}

int tsk_buffer_insert(tsk_buffer_t* self, size_t position, const void*data, size_t size)
{
	if(self && size)
	{
		int ret;
		if(position > self->size){
			return -2;
		}

		if((ret = tsk_buffer_realloc(self, (self->size + size)))){
			return ret;
		}
		memmove(((uint8_t*)self->data) + position + size, ((uint8_t*)self->data) + position,
			self->size - (position + size));
		

		if(data){
			memcpy(((uint8_t*)self->data) + position, data, size);
		}
		else{
			memset(((uint8_t*)self->data) + position, 0, size);
		}

		return 0;
	}
	return -1;
}

int tsk_buffer_cleanup(tsk_buffer_t* self)
{
	if(self && self->data)
	{
		tsk_free(&(self->data));
		self->size = 0;
	}
	return 0;
}













//========================================================
//	Buffer object definition
//
static void* tsk_buffer_create(void * self, va_list * app)
{
	tsk_buffer_t *buffer = self;
	const void *data = va_arg(*app, const void *);
	size_t size = va_arg(*app, size_t);
	
	if(data && size)
	{
		buffer->data = tsk_calloc((size+1), sizeof(uint8_t));
		memcpy(buffer->data, data, size);
		buffer->size = size;
	}
	return self;
}

static void* tsk_buffer_destroy(void * self)
{ 
	tsk_buffer_t *buffer = self;
	if(buffer)
	{
		TSK_FREE(buffer->data);
		buffer->size = 0;
	}

	return self;
}

static int tsk_buffer_cmp(const void *obj1, const void *obj2)
{
	return -1;
}

static const tsk_object_def_t tsk_buffer_def_s = 
{
	sizeof(tsk_buffer_t),
	tsk_buffer_create, 
	tsk_buffer_destroy,
	tsk_buffer_cmp, 
};
const void *tsk_buffer_def_t = &tsk_buffer_def_s;