1 files changed, 86 insertions, 0 deletions

diff --git a/examples/redis-unstable/deps/jemalloc/test/integration/thread_arena.c b/examples/redis-unstable/deps/jemalloc/test/integration/thread_arena.c
new file mode 100644
index 0000000..4a6abf6
--- /dev/null
+++ b/examples/redis-unstable/deps/jemalloc/test/integration/thread_arena.c
@@ -0,0 +1,86 @@
+#include "test/jemalloc_test.h"
+
+#define NTHREADS 10
+
+void *
+thd_start(void *arg) {
+        unsigned main_arena_ind = *(unsigned *)arg;
+        void *p;
+        unsigned arena_ind;
+        size_t size;
+        int err;
+
+        p = malloc(1);
+        expect_ptr_not_null(p, "Error in malloc()");
+        free(p);
+
+        size = sizeof(arena_ind);
+        if ((err = mallctl("thread.arena", (void *)&arena_ind, &size,
+            (void *)&main_arena_ind, sizeof(main_arena_ind)))) {
+                char buf[BUFERROR_BUF];
+
+                buferror(err, buf, sizeof(buf));
+                test_fail("Error in mallctl(): %s", buf);
+        }
+
+        size = sizeof(arena_ind);
+        if ((err = mallctl("thread.arena", (void *)&arena_ind, &size, NULL,
+            0))) {
+                char buf[BUFERROR_BUF];
+
+                buferror(err, buf, sizeof(buf));
+                test_fail("Error in mallctl(): %s", buf);
+        }
+        expect_u_eq(arena_ind, main_arena_ind,
+            "Arena index should be same as for main thread");
+
+        return NULL;
+}
+
+static void
+mallctl_failure(int err) {
+        char buf[BUFERROR_BUF];
+
+        buferror(err, buf, sizeof(buf));
+        test_fail("Error in mallctl(): %s", buf);
+}
+
+TEST_BEGIN(test_thread_arena) {
+        void *p;
+        int err;
+        thd_t thds[NTHREADS];
+        unsigned i;
+
+        p = malloc(1);
+        expect_ptr_not_null(p, "Error in malloc()");
+
+        unsigned arena_ind, old_arena_ind;
+        size_t sz = sizeof(unsigned);
+        expect_d_eq(mallctl("arenas.create", (void *)&arena_ind, &sz, NULL, 0),
+            0, "Arena creation failure");
+
+        size_t size = sizeof(arena_ind);
+        if ((err = mallctl("thread.arena", (void *)&old_arena_ind, &size,
+            (void *)&arena_ind, sizeof(arena_ind))) != 0) {
+                mallctl_failure(err);
+        }
+
+        for (i = 0; i < NTHREADS; i++) {
+                thd_create(&thds[i], thd_start,
+                    (void *)&arena_ind);
+        }
+
+        for (i = 0; i < NTHREADS; i++) {
+                intptr_t join_ret;
+                thd_join(thds[i], (void *)&join_ret);
+                expect_zd_eq(join_ret, 0, "Unexpected thread join error");
+        }
+        free(p);
+}
+TEST_END
+
+int
+main(void) {
+        return test(
+            test_thread_arena);
+}