21dfa4300d
```
grep -Irl assert_ test/ | xargs sed -i \
's/witness_assert/witness_do_not_replace/g';
grep -Irl assert_ test/ | xargs sed -i \
's/malloc_mutex_assert_owner/malloc_mutex_do_not_replace_owner/g';
grep -Ir assert_ test/ | grep -o "[_a-zA-Z]*assert_[_a-zA-Z]*" | \
grep -v "^assert_"; # confirm no output
grep -Irl assert_ test/ | xargs sed -i 's/assert_/expect_/g';
grep -Irl witness_do_not_replace test/ | xargs sed -i \
's/witness_do_not_replace/witness_assert/g';
grep -Irl malloc_mutex_do_not_replace_owner test/ | xargs sed -i \
's/malloc_mutex_do_not_replace_owner/malloc_mutex_assert_owner/g';
```
87 lines
1.8 KiB
C
87 lines
1.8 KiB
C
#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);
|
|
}
|