Mercurial > code
diff array.c @ 62:d10ab6bc555d
HG self failure
author | David Demelier <markand@malikania.fr> |
---|---|
date | Wed, 09 Nov 2011 19:26:09 +0100 |
parents | |
children | 9cc5d6d0563e |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/array.c Wed Nov 09 19:26:09 2011 +0100 @@ -0,0 +1,314 @@ +/* + * array.c -- manipulate dynamic arrays + * + * Copyright (c) 2011, David Demelier <markand@malikania.fr> + * + * Permission to use, copy, modify, and/or distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + * + * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES + * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR + * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES + * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN + * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF + * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + */ + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include "array.h" + +#define OFFSET(x) (arr->unit * (x)) + +static int array_grow(struct array *); + +struct array * +array_new(enum array_type type, size_t unit, int length) +{ + struct array *arr; + + if (unit == 0 || (arr = malloc(sizeof (struct array))) == NULL) + return NULL; + + memset(arr, 0, sizeof (struct array)); + arr->type = type; + arr->bsize = (length == 0) ? ARRAY_DEFAULT_BSIZE : length; + arr->unit = unit; + arr->size = OFFSET(arr->bsize); + + if ((arr->data = malloc(arr->size)) == NULL) { + free(arr); + return NULL; + } + + return arr; +} + +/* + * Add to the head of array. NOTE: this may be very slow when adding a lot + * of object (about 100000). If you need to add a lot of data please consider + * using linked list instead. + */ + +int +array_push(struct array *arr, const void *data) +{ + if (array_grow(arr) < 0) + return -1; + + memmove((char *) arr->data + arr->unit, arr->data, OFFSET(arr->length++)); + memcpy((char *) arr->data, data, arr->unit); + + return 0; +} + +/* + * Insert the data at the specified index. The function returns -1 on + * allocation failure or when the index is outof bounds otherwise 0 is returned. + */ + +int +array_insert(struct array *arr, const void *data, int index) +{ + if (index > arr->length - 1 || index < 0 || array_grow(arr) < 0) + return -1; + + memmove((char *) arr->data + OFFSET(index + 1), + (char *) arr->data + OFFSET(index), OFFSET(arr->length++ - index)); + memcpy((char *) arr->data + OFFSET(index), data, arr->unit); + + return 0; +} + +/* + * Append the data to the end of array. + */ + +int +array_append(struct array *arr, const void *data) +{ + if (array_grow(arr) < 0) + return -1; + + memcpy((char *) arr->data + OFFSET(arr->length++), data, arr->unit); + + return 0; +} + +/* + * Remove the array's head. + */ + +void +array_pop(struct array *arr) +{ + if (arr->length > 0) { + memmove((char *) arr->data, (char *) arr->data + OFFSET(1), + OFFSET(--arr->length)); + memset((char *) arr->data + OFFSET(arr->length), 0, arr->unit); + } +} + +/* + * Remove the array's tail. + */ + +void +array_unqueue(struct array *arr) +{ + if (arr->length > 0) + memset((char *) arr->data + OFFSET(--arr->length), 0, arr->unit); +} + +/* + * Remove the data at the specified index. Bounds are checked. + */ + +void +array_remove(struct array *arr, int index) +{ + if (arr->length > 0 && index >= 0 && index < arr->length) { + memmove((char *) arr->data + OFFSET(index), + (char *) arr->data + OFFSET(index + 1), + OFFSET(arr->length - index - 1)); + memset((char *) arr->data + OFFSET(--arr->length), 0, arr->unit); + } +} + +/* + * Remove the object referenced by the `data' argument. Useful when you + * don't know the index. + */ + +void +array_unref(struct array *arr, const void *data) +{ + void *elm; + int i; + + for (i = 0; i < arr->length; ++i) { + elm = (char *) arr->data + OFFSET(i); + + if (memcmp(elm, data, arr->unit) == 0) + array_remove(arr, i); + } +} + +/* + * Swap the two elements referenced by index `i1' and `i2'. This function needs + * to allocate data to swap elements thus if the functions fails it returns -1 + * otherwise 0 is returned. + */ + +int +array_iswap(struct array *arr, int i1, int i2) +{ + void *tmp; + + /* Out of bounds */ + if (i1 >= arr->length || i1 < 0 || i2 >= arr->length || i2 < 0) + return -1; + + /* + * Only allocate at this time, the user may do not want to use this + * function. + */ + + if ((tmp = malloc(arr->unit)) == NULL) + return -1; + + memcpy((char *) tmp, (char *) arr->data + OFFSET(i1), arr->unit); + memcpy((char *) arr->data + OFFSET(i1), (char *) arr->data + OFFSET(i2), + arr->unit); + memcpy((char *) arr->data + OFFSET(i2), (char *) tmp, arr->unit); + + /* + * Clear bytes for safety you probably don't want a password or + * secure data to be left somewhere in the memory. + */ + + memset(tmp, 0, arr->unit); + free(tmp); + + return 0; +} + +/* + * Swap the two elements referenced by data `o1' and `o2'. This function + * may be slow on large arrays since it must travel all the object + * to find the indexes. + */ + +int +array_pswap(struct array *arr, const void *o1, const void *o2) +{ + int found, i1, i2; + + for (i1 = found = 0; !found && i1 < arr->length; ++i1) + found = memcmp(arr->data + OFFSET(i1), o1, arr->unit) == 0; + + if (!found) + return -1; + + for (i2 = found = 0; !found && i2 < arr->length; ++i2) + found = memcmp(arr->data + OFFSET(i2), o2, arr->unit) == 0; + + if (!found) + return -1; + + return array_iswap(arr, --i1, --i2); +} + +/* + * Apply the function `fn' on each object and give the optional `udata' + * argument to the function too. + */ + +void +array_map(const struct array *arr, void (*fn)(void *, void *), void *udata) +{ + int i; + + for (i = 0; i < arr->length; ++i) + fn((char *) arr->data + OFFSET(i), udata); +} + +/* + * Compare each object with the user supplied function. If the `fn' function + * returns 1 then the data is returned. Optional idx argument can be set to + * indicate the data position. If the data was not found the function returns + * NULL. + */ + +void * +array_find(const struct array *arr, int (*fn)(void *, void *), int *ix, void *u) +{ + int st, i; + void *data; + + for (i = st = 0; i < arr->length && st != 1; ++i) + st = fn((char *) arr->data + OFFSET(i), u); + + if (st) { + data = (char *) arr->data + OFFSET(--i); + if (ix) + *ix = i; + } else + data = NULL; + + return data; +} + +/* + * Erase every bytes and set the length to 0. + */ + +void +array_clear(struct array *arr) +{ + memset(arr->data, 0, arr->size); + arr->length = 0; +} + +/* + * Same as array_clear except it also free the array object. + */ + +void +array_free(struct array *arr) +{ + array_clear(arr); + + if (arr->data) + free(arr->data); + + free(arr); +} + +/* + * Increate the array storage when it is full. If the buffer is fixed size + * it returns -1 on full buffer otherwise 0 is returned if allocation + * succeeded. + */ + +static int +array_grow(struct array *arr) +{ + if ((arr->size / arr->unit) > (size_t) arr->length) + return 0; + + if (arr->type == ARRAY_AUTO) { + if ((arr->data = realloc(arr->data, arr->size + + OFFSET(arr->bsize))) == NULL) + return -1; + + arr->size += OFFSET(arr->bsize); + } else + return -1; + + return 0; +}