You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
1.8 KiB
87 lines
1.8 KiB
#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);
|
|
assert_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);
|
|
}
|
|
assert_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);
|
|
assert_ptr_not_null(p, "Error in malloc()");
|
|
|
|
unsigned arena_ind, old_arena_ind;
|
|
size_t sz = sizeof(unsigned);
|
|
assert_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);
|
|
assert_zd_eq(join_ret, 0, "Unexpected thread join error");
|
|
}
|
|
free(p);
|
|
}
|
|
TEST_END
|
|
|
|
int
|
|
main(void) {
|
|
return test(
|
|
test_thread_arena);
|
|
}
|