mirror of
https://github.com/jemalloc/jemalloc.git
synced 2026-05-01 09:37:50 +03:00
74 lines
2.2 KiB
C
74 lines
2.2 KiB
C
#include "test/jemalloc_test.h"
|
|
|
|
#include "jemalloc/internal/jemalloc_init.h"
|
|
|
|
TEST_BEGIN(test_malloc_init_hard_idempotent) {
|
|
expect_false(malloc_init_hard(),
|
|
"malloc_init_hard should return false when already initialized");
|
|
expect_true(malloc_initialized(),
|
|
"malloc_initialized should still be true after re-calling "
|
|
"malloc_init_hard");
|
|
}
|
|
TEST_END
|
|
|
|
TEST_BEGIN(test_malloc_initializer_set_idempotent_on_main) {
|
|
test_skip_if(!malloc_is_initializer());
|
|
|
|
malloc_initializer_set();
|
|
|
|
expect_true(malloc_is_initializer(),
|
|
"malloc_is_initializer should still be true after re-setting "
|
|
"from the same thread");
|
|
expect_true(malloc_initializer_is_set(),
|
|
"malloc_initializer_is_set should still be true after re-setting");
|
|
}
|
|
TEST_END
|
|
|
|
#ifdef JEMALLOC_THREADED_INIT
|
|
static atomic_b_t initializer_worker_done;
|
|
static atomic_b_t initializer_worker_result;
|
|
|
|
static void *
|
|
is_initializer_worker(void *unused) {
|
|
atomic_store_b(&initializer_worker_result, malloc_is_initializer(),
|
|
ATOMIC_RELEASE);
|
|
atomic_store_b(&initializer_worker_done, true, ATOMIC_RELEASE);
|
|
return NULL;
|
|
}
|
|
#endif
|
|
|
|
TEST_BEGIN(test_malloc_is_initializer_false_in_worker) {
|
|
#ifndef JEMALLOC_THREADED_INIT
|
|
/*
|
|
* Without JEMALLOC_THREADED_INIT (e.g. macOS), malloc_initializer
|
|
* is a process-wide bool and malloc_is_initializer() returns true
|
|
* for every thread after init. The per-thread distinction this
|
|
* test exercises only exists in threaded-init builds.
|
|
*/
|
|
test_skip_if(true);
|
|
#else
|
|
test_skip_if(!malloc_is_initializer());
|
|
|
|
atomic_store_b(&initializer_worker_done, false, ATOMIC_RELEASE);
|
|
atomic_store_b(&initializer_worker_result, false, ATOMIC_RELEASE);
|
|
|
|
thd_t thd;
|
|
thd_create(&thd, is_initializer_worker, NULL);
|
|
thd_join(thd, NULL);
|
|
|
|
expect_true(
|
|
atomic_load_b(&initializer_worker_done, ATOMIC_ACQUIRE),
|
|
"Worker should have completed");
|
|
expect_false(
|
|
atomic_load_b(&initializer_worker_result, ATOMIC_ACQUIRE),
|
|
"malloc_is_initializer should be false in a non-init thread");
|
|
#endif
|
|
}
|
|
TEST_END
|
|
|
|
int
|
|
main(void) {
|
|
return test(test_malloc_init_hard_idempotent,
|
|
test_malloc_initializer_set_idempotent_on_main,
|
|
test_malloc_is_initializer_false_in_worker);
|
|
}
|