diff options
Diffstat (limited to 'src/alloc.h')
-rw-r--r-- | src/alloc.h | 122 |
1 files changed, 122 insertions, 0 deletions
diff --git a/src/alloc.h b/src/alloc.h new file mode 100644 index 0000000..7b3e93a --- /dev/null +++ b/src/alloc.h @@ -0,0 +1,122 @@ +/* + Copyright (c) 2012-2014, Matthias Schiffer <mschiffer@universe-factory.net> + All rights reserved. + + Redistribution and use in source and binary forms, with or without + modification, are permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, + this list of conditions and the following disclaimer. + 2. Redistributions in binary form must reproduce the above copyright notice, + this list of conditions and the following disclaimer in the documentation + and/or other materials provided with the distribution. + + THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE + FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER + CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +*/ + +/** + \file + + \em memory allocation functions + */ + + +#pragma once + +#include "log.h" + + +/** + Allocates a block of uninitialized memory on the heap + + Terminates the process on failure. +*/ +static inline void * fastd_alloc(size_t size) { + void *ret = malloc(size); + if (!ret) + exit_errno("malloc"); + + return ret; +} + +/** + Allocates a block of memory set to zero for an array on the heap + + Terminates the process on failure. +*/ +static inline void * fastd_alloc0_array(size_t members, size_t size) { + void *ret = calloc(members, size); + if (!ret) + exit_errno("calloc"); + + return ret; +} + +/** + Allocates a block of memory set to zero on the heap + + Terminates the process on failure. +*/ +static inline void * fastd_alloc0(size_t size) { + return fastd_alloc0_array(1, size); +} + +/** + Reallocates a block of memory on the heap + + Terminates the process on failure. +*/ +static inline void * fastd_realloc(void *ptr, size_t size) { + void *ret = realloc(ptr, size); + if (!ret) + exit_errno("realloc"); + + return ret; +} + + +/** Allocates a block of uninitialized memory in the size of a given type */ +#define fastd_new(type) ((type *)fastd_alloc(sizeof(type))) + +/** Allocates a block of memory set to zero in the size of a given type */ +#define fastd_new0(type) ((type *)fastd_alloc0(sizeof(type))) + +/** Allocates a block of undefined memory for an array of elements of a given type */ +#define fastd_new_array(members, type) ((type *)fastd_alloc(members * sizeof(type))) + +/** Allocates a block of memory set to zero for an array of elements of a given type */ +#define fastd_new0_array(members, type) ((type *)fastd_alloc0_array(members, sizeof(type))) + + +/** Duplicates a string (string may be NULL) */ +static inline char * fastd_strdup(const char *s) { + if (!s) + return NULL; + + char *ret = strdup(s); + if (!ret) + exit_errno("strdup"); + + return ret; +} + +/** Duplicates a string up to a maximum length (string may be NULL) */ +static inline char * fastd_strndup(const char *s, size_t n) { + if (!s) + return NULL; + + char *ret = strndup(s, n); + if (!ret) + exit_errno("strndup"); + + return ret; +} |