1 files changed, 66 insertions, 0 deletions

diff --git a/examples/redis-unstable/deps/jemalloc/test/integration/MALLOCX_ARENA.c b/examples/redis-unstable/deps/jemalloc/test/integration/MALLOCX_ARENA.c
new file mode 100644
index 0000000..7e61df0
--- /dev/null
+++ b/examples/redis-unstable/deps/jemalloc/test/integration/MALLOCX_ARENA.c
@@ -0,0 +1,66 @@
+#include "test/jemalloc_test.h"
+
+#define NTHREADS 10
+
+static bool have_dss =
+#ifdef JEMALLOC_DSS
+    true
+#else
+    false
+#endif
+    ;
+
+void *
+thd_start(void *arg) {
+        unsigned thread_ind = (unsigned)(uintptr_t)arg;
+        unsigned arena_ind;
+        void *p;
+        size_t sz;
+
+        sz = sizeof(arena_ind);
+        expect_d_eq(mallctl("arenas.create", (void *)&arena_ind, &sz, NULL, 0),
+            0, "Error in arenas.create");
+
+        if (thread_ind % 4 != 3) {
+                size_t mib[3];
+                size_t miblen = sizeof(mib) / sizeof(size_t);
+                const char *dss_precs[] = {"disabled", "primary", "secondary"};
+                unsigned prec_ind = thread_ind %
+                    (sizeof(dss_precs)/sizeof(char*));
+                const char *dss = dss_precs[prec_ind];
+                int expected_err = (have_dss || prec_ind == 0) ? 0 : EFAULT;
+                expect_d_eq(mallctlnametomib("arena.0.dss", mib, &miblen), 0,
+                    "Error in mallctlnametomib()");
+                mib[1] = arena_ind;
+                expect_d_eq(mallctlbymib(mib, miblen, NULL, NULL, (void *)&dss,
+                    sizeof(const char *)), expected_err,
+                    "Error in mallctlbymib()");
+        }
+
+        p = mallocx(1, MALLOCX_ARENA(arena_ind));
+        expect_ptr_not_null(p, "Unexpected mallocx() error");
+        dallocx(p, 0);
+
+        return NULL;
+}
+
+TEST_BEGIN(test_MALLOCX_ARENA) {
+        thd_t thds[NTHREADS];
+        unsigned i;
+
+        for (i = 0; i < NTHREADS; i++) {
+                thd_create(&thds[i], thd_start,
+                    (void *)(uintptr_t)i);
+        }
+
+        for (i = 0; i < NTHREADS; i++) {
+                thd_join(thds[i], NULL);
+        }
+}
+TEST_END
+
+int
+main(void) {
+        return test(
+            test_MALLOCX_ARENA);
+}