From 27207d35d4b4bbd5d2b2c5f7e13a61ea43d04a4a Mon Sep 17 00:00:00 2001 From: Alan Coopersmith Date: Sat, 3 Aug 2019 16:13:21 -0700 Subject: Add reallocarray fallback if not provided by libc nor libbsd Implementation copied from the Xserver Signed-off-by: Alan Coopersmith --- src/util/reallocarray.c | 43 +++++++++++++++++++++++++++++++++++++++++++ src/util/replace.h | 10 ++++++++++ 2 files changed, 53 insertions(+) create mode 100644 src/util/reallocarray.c (limited to 'src') diff --git a/src/util/reallocarray.c b/src/util/reallocarray.c new file mode 100644 index 0000000..025e24e --- /dev/null +++ b/src/util/reallocarray.c @@ -0,0 +1,43 @@ +/* $OpenBSD: reallocarray.c,v 1.2 2014/12/08 03:45:00 bcook Exp $ */ +/* + * Copyright (c) 2008 Otto Moerbeek + * + * Permission to use, copy, modify, and 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. + */ + +#ifdef HAVE_CONFIG_H +#include +#endif + +#include +#include +#include +#include +#include "src/util/replace.h" + +/* + * This is sqrt(SIZE_MAX+1), as s1*s2 <= SIZE_MAX + * if both s1 < MUL_NO_OVERFLOW and s2 < MUL_NO_OVERFLOW + */ +#define MUL_NO_OVERFLOW ((size_t)1 << (sizeof(size_t) * 4)) + +void * +reallocarray(void *optr, size_t nmemb, size_t size) +{ + if ((nmemb >= MUL_NO_OVERFLOW || size >= MUL_NO_OVERFLOW) && + nmemb > 0 && SIZE_MAX / nmemb < size) { + errno = ENOMEM; + return NULL; + } + return realloc(optr, size * nmemb); +} diff --git a/src/util/replace.h b/src/util/replace.h index 96ad001..16be544 100644 --- a/src/util/replace.h +++ b/src/util/replace.h @@ -31,6 +31,16 @@ #include +#include +#if defined(HAVE_LIBBSD) && defined(HAVE_REALLOCARRAY) +#include /* for reallocarray */ +#endif + +#ifndef HAVE_REALLOCARRAY +extern _X_HIDDEN void * +reallocarray(void *optr, size_t nmemb, size_t size); +#endif + #include #if defined(HAVE_LIBBSD) && defined(HAVE_STRLCPY) #include /* for strlcpy, strlcat */ -- cgit v1.2.3