1 files changed, 92 insertions, 0 deletions
diff --git a/tests/portable/reallocarray-t.c b/tests/portable/reallocarray-t.c
new file mode 100644
index 0000000..481da58
--- /dev/null
+++ b/tests/portable/reallocarray-t.c
@@ -0,0 +1,92 @@
+/*
+ * reallocarray test suite.
+ *
+ * This does some simple sanity checks and checks some of the overflow
+ * detection, but isn't particularly thorough.
+ *
+ * The canonical version of this file is maintained in the rra-c-util package,
+ * which can be found at <http://www.eyrie.org/~eagle/software/rra-c-util/>.
+ *
+ * Written by Russ Allbery <eagle@eyrie.org>
+ *
+ * The authors hereby relinquish any claim to any copyright that they may have
+ * in this work, whether granted under contract or by operation of law or
+ * international treaty, and hereby commit to the public, at large, that they
+ * shall not, at any time in the future, seek to enforce any copyright in this
+ * work against any person or entity, or prevent any person or entity from
+ * copying, publishing, distributing or creating derivative works of this
+ * work.
+ */
+
+#include <config.h>
+#include <portable/system.h>
+
+#include <errno.h>
+
+#include <tests/tap/basic.h>
+
+void *test_reallocarray(void *, size_t, size_t);
+
+
+int
+main(void)
+{
+    char *p, *base;
+    size_t sqrt_max;
+    int oerrno;
+
+    plan(15);
+
+    /* Test success cases and write to the memory for valgrind checks. */
+    p = test_reallocarray(NULL, 2, 5);
+    memcpy(p, "123456789", 10);
+    is_string("123456789", p, "reallocarray of NULL");
+    p = test_reallocarray(p, 4, 5);
+    is_string("123456789", p, "reallocarray after resize");
+    memcpy(p + 9, "0123456789", 11);
+    is_string("1234567890123456789", p, "write to larger memory segment");
+    free(p);
+
+    /*
+     * If nmemb or size are 0, we should either get NULL or a pointer we can
+     * free.  Make sure we don't get something weird, like division by zero.
+     */
+    p = test_reallocarray(NULL, 0, 100);
+    if (p != NULL)
+        free(p);
+    p = test_reallocarray(NULL, 100, 0);
+    if (p != NULL)
+        free(p);
+
+    /* Test the range-checking error cases. */
+    p = test_reallocarray(NULL, 2, SIZE_MAX / 2);
+    oerrno = errno;
+    ok(p == NULL, "reallocarray fails for 2, SIZE_MAX / 2");
+    is_int(ENOMEM, oerrno, "...with correct errno");
+    base = malloc(10);
+    p = test_reallocarray(base, 3, SIZE_MAX / 3);
+    oerrno = errno;
+    ok(p == NULL, "reallocarray fails for 3, SIZE_MAX / 3");
+    is_int(ENOMEM, oerrno, "...with correct errno");
+    sqrt_max = (1UL << (sizeof(size_t) * 4));
+    p = test_reallocarray(base, sqrt_max, sqrt_max);
+    oerrno = errno;
+    ok(p == NULL, "reallocarray fails for sqrt(SIZE_MAX), sqrt(SIZE_MAX)");
+    is_int(ENOMEM, oerrno, "...with correct errno");
+    p = test_reallocarray(base, 1, SIZE_MAX);
+    oerrno = errno;
+    ok(p == NULL, "reallocarray fails for 1, SIZE_MAX");
+    is_int(ENOMEM, oerrno, "...with correct errno");
+    p = test_reallocarray(base, SIZE_MAX, 1);
+    oerrno = errno;
+    ok(p == NULL, "reallocarray fails for SIZE_MAX, 1");
+    is_int(ENOMEM, oerrno, "...with correct errno");
+    p = test_reallocarray(base, 2, SIZE_MAX);
+    oerrno = errno;
+    ok(p == NULL, "reallocarray fails for 2, SIZE_MAX");
+    is_int(ENOMEM, oerrno, "...with correct errno");
+
+    /* Clean up and exit. */
+    free(base);
+    return 0;
+}