8bb3198f72
Abstract arenas access to use arena_get() (or a0get() where appropriate) rather than directly reading e.g. arenas[ind]. Prior to the addition of the arenas.extend mallctl, the worst possible outcome of directly accessing arenas was a stale read, but arenas.extend may allocate and assign a new array to arenas. Add a tsd-based arenas_cache, which amortizes arenas reads. This introduces some subtle bootstrapping issues, with tsd_boot() now being split into tsd_boot[01]() to support tsd wrapper allocation bootstrapping, as well as an arenas_cache_bypass tsd variable which dynamically terminates allocation of arenas_cache itself. Promote a0malloc(), a0calloc(), and a0free() to be generally useful for internal allocation, and use them in several places (more may be appropriate). Abstract arena->nthreads management and fix a missing decrement during thread destruction (recent tsd refactoring left arenas_cleanup() unused). Change arena_choose() to propagate OOM, and handle OOM in all callers. This is important for providing consistent allocation behavior when the MALLOCX_ARENA() flag is being used. Prior to this fix, it was possible for an OOM to result in allocation silently allocating from a different arena than the one specified.
102 lines
1.9 KiB
C
102 lines
1.9 KiB
C
#include "test/jemalloc_test.h"
|
|
|
|
#define THREAD_DATA 0x72b65c10
|
|
|
|
typedef unsigned int data_t;
|
|
|
|
static bool data_cleanup_executed;
|
|
|
|
malloc_tsd_types(data_, data_t)
|
|
malloc_tsd_protos(, data_, data_t)
|
|
|
|
void
|
|
data_cleanup(void *arg)
|
|
{
|
|
data_t *data = (data_t *)arg;
|
|
|
|
if (!data_cleanup_executed) {
|
|
assert_x_eq(*data, THREAD_DATA,
|
|
"Argument passed into cleanup function should match tsd "
|
|
"value");
|
|
}
|
|
data_cleanup_executed = true;
|
|
|
|
/*
|
|
* Allocate during cleanup for two rounds, in order to assure that
|
|
* jemalloc's internal tsd reinitialization happens.
|
|
*/
|
|
switch (*data) {
|
|
case THREAD_DATA:
|
|
*data = 1;
|
|
data_tsd_set(data);
|
|
break;
|
|
case 1:
|
|
*data = 2;
|
|
data_tsd_set(data);
|
|
break;
|
|
case 2:
|
|
return;
|
|
default:
|
|
not_reached();
|
|
}
|
|
|
|
{
|
|
void *p = mallocx(1, 0);
|
|
assert_ptr_not_null(p, "Unexpeced mallocx() failure");
|
|
dallocx(p, 0);
|
|
}
|
|
}
|
|
|
|
malloc_tsd_externs(data_, data_t)
|
|
#define DATA_INIT 0x12345678
|
|
malloc_tsd_data(, data_, data_t, DATA_INIT)
|
|
malloc_tsd_funcs(, data_, data_t, DATA_INIT, data_cleanup)
|
|
|
|
static void *
|
|
thd_start(void *arg)
|
|
{
|
|
data_t d = (data_t)(uintptr_t)arg;
|
|
assert_x_eq(*data_tsd_get(), DATA_INIT,
|
|
"Initial tsd get should return initialization value");
|
|
|
|
data_tsd_set(&d);
|
|
assert_x_eq(*data_tsd_get(), d,
|
|
"After tsd set, tsd get should return value that was set");
|
|
|
|
d = 0;
|
|
assert_x_eq(*data_tsd_get(), (data_t)(uintptr_t)arg,
|
|
"Resetting local data should have no effect on tsd");
|
|
|
|
return (NULL);
|
|
}
|
|
|
|
TEST_BEGIN(test_tsd_main_thread)
|
|
{
|
|
|
|
thd_start((void *) 0xa5f3e329);
|
|
}
|
|
TEST_END
|
|
|
|
TEST_BEGIN(test_tsd_sub_thread)
|
|
{
|
|
thd_t thd;
|
|
|
|
data_cleanup_executed = false;
|
|
thd_create(&thd, thd_start, (void *)THREAD_DATA);
|
|
thd_join(thd, NULL);
|
|
assert_true(data_cleanup_executed,
|
|
"Cleanup function should have executed");
|
|
}
|
|
TEST_END
|
|
|
|
int
|
|
main(void)
|
|
{
|
|
|
|
data_tsd_boot();
|
|
|
|
return (test(
|
|
test_tsd_main_thread,
|
|
test_tsd_sub_thread));
|
|
}
|