freewtp/src/common/capwap_array.c

#include "capwap.h"
#include "capwap_array.h"

/* */
struct capwap_array* capwap_array_create(unsigned short itemsize, unsigned long initcount) {
	struct capwap_array* array;

	ASSERT(itemsize > 0);

	array = (struct capwap_array*)capwap_alloc(sizeof(struct capwap_array));
	if (!array) {
		capwap_outofmemory();
	}

	memset(array, 0, sizeof(struct capwap_array));
	array->itemsize = itemsize;
	if (initcount > 0) {
		capwap_array_resize(array, initcount);
	}

	return array;
}

/* */
struct capwap_array* capwap_array_clone(struct capwap_array* array) {
	struct capwap_array* clone;

	ASSERT (array != NULL);

	/* Clone array e items */
	clone = capwap_array_create(array->itemsize, array->count);
	memcpy(clone->buffer, array->buffer, array->itemsize * array->count);
	clone->zeroed = array->zeroed;

	return clone;
}

/* */
void capwap_array_free(struct capwap_array* array) {
	ASSERT(array != NULL);

	if (array->buffer) {
		capwap_free(array->buffer);
	}

	capwap_free(array);
}

/* */
void* capwap_array_get_item_pointer(struct capwap_array* array, unsigned long pos) {
	ASSERT(array != NULL);
	ASSERT((array->count == 0) || (array->buffer != NULL));

	if (pos >= array->count) {
		capwap_array_resize(array, pos + 1);
	}

	return (void*)(((char*)array->buffer) + array->itemsize * pos);
}

/* */
void capwap_array_resize(struct capwap_array* array, unsigned long count) {
	void* newbuffer = NULL;

	ASSERT(array != NULL);
	ASSERT(array->itemsize > 0);

	if (array->count == count) {
		return;
	}

	if (count > 0) {
		newbuffer = capwap_alloc(array->itemsize * count);
		if (!newbuffer) {
			capwap_outofmemory();
		}
	}

	if (array->buffer) {
		if (newbuffer != NULL) {
			int newcount = min(array->count, count);
			memcpy(newbuffer, array->buffer, array->itemsize * newcount);

			/* Zeroed new items */
			if (array->zeroed && (count > newcount)) {
				memset(newbuffer + array->itemsize * newcount, 0, array->itemsize * (count - newcount));
			}
		}

		capwap_free(array->buffer);
	}

	array->buffer = newbuffer;
	array->count = count;
}