|
Message-Id: <20200802225426.32002-1-ariadne@dereferenced.org> Date: Sun, 2 Aug 2020 16:54:26 -0600 From: Ariadne Conill <ariadne@...eferenced.org> To: musl@...ts.openwall.com Cc: Ariadne Conill <ariadne@...eferenced.org> Subject: [PATCH v4] implement recallocarray(3) This OpenBSD extension is similar to reallocarray(3), but zero-initializes the new memory area. This extension is placed in _BSD_SOURCE, like reallocarray(3). Changes from v3: - use calloc() instead of realloc() and always copy - explicitly zero old memory block - restore overflow checking for old size Changes from v2: - drop overflow checking for old size Changes from v1: - use realloc() instead of reallocarray() --- include/stdlib.h | 1 + src/malloc/recallocarray.c | 39 ++++++++++++++++++++++++++++++++++++++ 2 files changed, 40 insertions(+) create mode 100644 src/malloc/recallocarray.c diff --git a/include/stdlib.h b/include/stdlib.h index b54a051f..a0412ad4 100644 --- a/include/stdlib.h +++ b/include/stdlib.h @@ -146,6 +146,7 @@ int clearenv(void); #define WCOREDUMP(s) ((s) & 0x80) #define WIFCONTINUED(s) ((s) == 0xffff) void *reallocarray (void *, size_t, size_t); +void *recallocarray (void *, size_t, size_t, size_t); #endif #ifdef _GNU_SOURCE diff --git a/src/malloc/recallocarray.c b/src/malloc/recallocarray.c new file mode 100644 index 00000000..ce3adde4 --- /dev/null +++ b/src/malloc/recallocarray.c @@ -0,0 +1,39 @@ +#define _BSD_SOURCE +#include <errno.h> +#include <stdlib.h> +#include <string.h> + +void *recallocarray(void *ptr, size_t om, size_t m, size_t n) +{ + void *newptr; + size_t old_size, new_size; + + if (n && m > -1 / n) { + errno = ENOMEM; + return 0; + } + new_size = m * n; + + if (n && om > -1 / n) { + errno = EINVAL; + return 0; + } + old_size = om * n; + + newptr = calloc(m, n); + if (!newptr) + return ptr; + + if (new_size <= old_size) { + memcpy((char *) newptr, ptr, new_size); + } + else { + memcpy((char *) newptr, ptr, old_size); + memset((char *) newptr + old_size, 0, new_size - old_size); + } + + memset(ptr, 0, old_size); + free(ptr); + + return newptr; +} -- 2.28.0
Powered by blists - more mailing lists
Confused about mailing lists and their use? Read about mailing lists on Wikipedia and check out these guidelines on proper formatting of your messages.