diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/Makemodule.am | 2 | ||||
-rw-r--r-- | lib/util/alloc.c | 54 |
2 files changed, 55 insertions, 1 deletions
diff --git a/lib/Makemodule.am b/lib/Makemodule.am index 6301435..fb9d613 100644 --- a/lib/Makemodule.am +++ b/lib/Makemodule.am @@ -47,7 +47,7 @@ libutil_a_SOURCES += lib/util/read_data.c include/util.h libutil_a_SOURCES += lib/util/print_version.c lib/util/mkdir_p.c libutil_a_SOURCES += lib/util/str_table.c include/str_table.h libutil_a_SOURCES += lib/util/dirstack.c lib/util/padd_file.c -libutil_a_SOURCES += lib/util/read_data_at.c +libutil_a_SOURCES += lib/util/read_data_at.c lib/util/alloc.c libutil_a_SOURCES += lib/util/source_date_epoch.c libutil_a_CFLAGS = $(AM_CFLAGS) libutil_a_CPPFLAGS = $(AM_CPPFLAGS) diff --git a/lib/util/alloc.c b/lib/util/alloc.c new file mode 100644 index 0000000..2d748a7 --- /dev/null +++ b/lib/util/alloc.c @@ -0,0 +1,54 @@ +/* SPDX-License-Identifier: GPL-3.0-or-later */ +/* + * alloc.c + * + * Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at> + */ +#include "config.h" + +#include "util.h" + +#include <stddef.h> +#include <stdlib.h> +#include <errno.h> + +void *alloc_flex(size_t base_size, size_t item_size, size_t nmemb) +{ + size_t size; + + if (SZ_MUL_OV(nmemb, item_size, &size)) { + errno = EOVERFLOW; + return NULL; + } + + if (SZ_ADD_OV(base_size, size, &size)) { + errno = EOVERFLOW; + return NULL; + } + + return calloc(1, size); +} + +void *alloc_array(size_t item_size, size_t nmemb) +{ + size_t size; + + if (SZ_MUL_OV(nmemb, item_size, &size)) { + errno = EOVERFLOW; + return NULL; + } + + return calloc(1, size); +} + +void *alloc_string(size_t len) +{ + size_t size; + + if (SZ_ADD_OV(len, 1, &size)) { + errno = EOVERFLOW; + return NULL; + } + + return calloc(1, len); +} |