2019-08-10 13:12:47 +08:00
|
|
|
#include "test/jemalloc_test.h"
|
|
|
|
|
2020-03-03 06:14:08 +08:00
|
|
|
static void
|
|
|
|
do_fill_test(cache_bin_t *bin, cache_bin_info_t *info, void **ptrs,
|
|
|
|
cache_bin_sz_t ncached_max, cache_bin_sz_t nfill_attempt,
|
|
|
|
cache_bin_sz_t nfill_succeed) {
|
|
|
|
bool success;
|
|
|
|
void *ptr;
|
2021-01-08 05:22:08 +08:00
|
|
|
assert_true(cache_bin_ncached_get_local(bin, info) == 0, "");
|
2020-03-03 06:14:08 +08:00
|
|
|
CACHE_BIN_PTR_ARRAY_DECLARE(arr, nfill_attempt);
|
|
|
|
cache_bin_init_ptr_array_for_fill(bin, info, &arr, nfill_attempt);
|
|
|
|
for (cache_bin_sz_t i = 0; i < nfill_succeed; i++) {
|
|
|
|
arr.ptr[i] = &ptrs[i];
|
|
|
|
}
|
|
|
|
cache_bin_finish_fill(bin, info, &arr, nfill_succeed);
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(bin, info) == nfill_succeed,
|
|
|
|
"");
|
2020-03-03 06:14:08 +08:00
|
|
|
cache_bin_low_water_set(bin);
|
2019-08-10 13:12:47 +08:00
|
|
|
|
2020-03-03 06:14:08 +08:00
|
|
|
for (cache_bin_sz_t i = 0; i < nfill_succeed; i++) {
|
2020-03-05 00:58:42 +08:00
|
|
|
ptr = cache_bin_alloc(bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_true(success, "");
|
|
|
|
expect_ptr_eq(ptr, (void *)&ptrs[i],
|
|
|
|
"Should pop in order filled");
|
|
|
|
expect_true(cache_bin_low_water_get(bin, info)
|
|
|
|
== nfill_succeed - i - 1, "");
|
|
|
|
}
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(bin, info) == 0, "");
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_true(cache_bin_low_water_get(bin, info) == 0, "");
|
|
|
|
}
|
2019-08-10 13:12:47 +08:00
|
|
|
|
2020-03-03 06:14:08 +08:00
|
|
|
static void
|
|
|
|
do_flush_test(cache_bin_t *bin, cache_bin_info_t *info, void **ptrs,
|
|
|
|
cache_bin_sz_t nfill, cache_bin_sz_t nflush) {
|
2019-08-10 13:12:47 +08:00
|
|
|
bool success;
|
2021-01-08 05:22:08 +08:00
|
|
|
assert_true(cache_bin_ncached_get_local(bin, info) == 0, "");
|
2020-03-03 06:14:08 +08:00
|
|
|
|
|
|
|
for (cache_bin_sz_t i = 0; i < nfill; i++) {
|
|
|
|
success = cache_bin_dalloc_easy(bin, &ptrs[i]);
|
|
|
|
expect_true(success, "");
|
|
|
|
}
|
|
|
|
|
|
|
|
CACHE_BIN_PTR_ARRAY_DECLARE(arr, nflush);
|
|
|
|
cache_bin_init_ptr_array_for_flush(bin, info, &arr, nflush);
|
|
|
|
for (cache_bin_sz_t i = 0; i < nflush; i++) {
|
2021-01-30 05:10:44 +08:00
|
|
|
expect_ptr_eq(arr.ptr[i], &ptrs[nflush - i - 1], "");
|
2019-08-10 13:12:47 +08:00
|
|
|
}
|
2020-03-03 06:14:08 +08:00
|
|
|
cache_bin_finish_flush(bin, info, &arr, nflush);
|
|
|
|
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(bin, info) == nfill - nflush,
|
|
|
|
"");
|
|
|
|
while (cache_bin_ncached_get_local(bin, info) > 0) {
|
2020-03-05 00:58:42 +08:00
|
|
|
cache_bin_alloc(bin, &success);
|
2019-08-10 13:12:47 +08:00
|
|
|
}
|
2020-03-03 06:14:08 +08:00
|
|
|
}
|
|
|
|
|
2020-10-23 07:07:25 +08:00
|
|
|
static void
|
|
|
|
do_batch_alloc_test(cache_bin_t *bin, cache_bin_info_t *info, void **ptrs,
|
|
|
|
cache_bin_sz_t nfill, size_t batch) {
|
2021-01-08 05:22:08 +08:00
|
|
|
assert_true(cache_bin_ncached_get_local(bin, info) == 0, "");
|
2020-10-23 07:07:25 +08:00
|
|
|
CACHE_BIN_PTR_ARRAY_DECLARE(arr, nfill);
|
|
|
|
cache_bin_init_ptr_array_for_fill(bin, info, &arr, nfill);
|
|
|
|
for (cache_bin_sz_t i = 0; i < nfill; i++) {
|
|
|
|
arr.ptr[i] = &ptrs[i];
|
|
|
|
}
|
|
|
|
cache_bin_finish_fill(bin, info, &arr, nfill);
|
2021-01-08 05:22:08 +08:00
|
|
|
assert_true(cache_bin_ncached_get_local(bin, info) == nfill, "");
|
2020-10-23 07:07:25 +08:00
|
|
|
cache_bin_low_water_set(bin);
|
|
|
|
|
|
|
|
void **out = malloc((batch + 1) * sizeof(void *));
|
|
|
|
size_t n = cache_bin_alloc_batch(bin, batch, out);
|
|
|
|
assert_true(n == ((size_t)nfill < batch ? (size_t)nfill : batch), "");
|
|
|
|
for (cache_bin_sz_t i = 0; i < (cache_bin_sz_t)n; i++) {
|
|
|
|
expect_ptr_eq(out[i], &ptrs[i], "");
|
|
|
|
}
|
|
|
|
expect_true(cache_bin_low_water_get(bin, info) == nfill -
|
|
|
|
(cache_bin_sz_t)n, "");
|
2021-01-08 05:22:08 +08:00
|
|
|
while (cache_bin_ncached_get_local(bin, info) > 0) {
|
2020-10-23 07:07:25 +08:00
|
|
|
bool success;
|
|
|
|
cache_bin_alloc(bin, &success);
|
|
|
|
}
|
|
|
|
free(out);
|
|
|
|
}
|
|
|
|
|
2020-03-03 06:14:08 +08:00
|
|
|
TEST_BEGIN(test_cache_bin) {
|
2020-05-12 06:03:06 +08:00
|
|
|
const int ncached_max = 100;
|
2020-03-03 06:14:08 +08:00
|
|
|
bool success;
|
|
|
|
void *ptr;
|
|
|
|
|
|
|
|
cache_bin_t bin;
|
|
|
|
cache_bin_info_t info;
|
2020-05-12 06:03:06 +08:00
|
|
|
cache_bin_info_init(&info, ncached_max);
|
2020-03-03 06:14:08 +08:00
|
|
|
|
|
|
|
size_t size;
|
|
|
|
size_t alignment;
|
|
|
|
cache_bin_info_compute_alloc(&info, 1, &size, &alignment);
|
|
|
|
void *mem = mallocx(size, MALLOCX_ALIGN(alignment));
|
|
|
|
assert_ptr_not_null(mem, "Unexpected mallocx failure");
|
|
|
|
|
|
|
|
size_t cur_offset = 0;
|
|
|
|
cache_bin_preincrement(&info, 1, mem, &cur_offset);
|
|
|
|
cache_bin_init(&bin, &info, mem, &cur_offset);
|
|
|
|
cache_bin_postincrement(&info, 1, mem, &cur_offset);
|
|
|
|
|
|
|
|
assert_zu_eq(cur_offset, size, "Should use all requested memory");
|
|
|
|
|
|
|
|
/* Initialize to empty; should then have 0 elements. */
|
2020-05-12 06:03:06 +08:00
|
|
|
expect_d_eq(ncached_max, cache_bin_info_ncached_max(&info), "");
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(&bin, &info) == 0, "");
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_true(cache_bin_low_water_get(&bin, &info) == 0, "");
|
|
|
|
|
2020-03-05 00:58:42 +08:00
|
|
|
ptr = cache_bin_alloc_easy(&bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_false(success, "Shouldn't successfully allocate when empty");
|
|
|
|
expect_ptr_null(ptr, "Shouldn't get a non-null pointer on failure");
|
|
|
|
|
2020-03-05 00:58:42 +08:00
|
|
|
ptr = cache_bin_alloc(&bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_false(success, "Shouldn't successfully allocate when empty");
|
|
|
|
expect_ptr_null(ptr, "Shouldn't get a non-null pointer on failure");
|
|
|
|
|
|
|
|
/*
|
|
|
|
* We allocate one more item than ncached_max, so we can test cache bin
|
|
|
|
* exhaustion.
|
|
|
|
*/
|
|
|
|
void **ptrs = mallocx(sizeof(void *) * (ncached_max + 1), 0);
|
|
|
|
assert_ptr_not_null(ptrs, "Unexpected mallocx failure");
|
|
|
|
for (cache_bin_sz_t i = 0; i < ncached_max; i++) {
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(&bin, &info) == i, "");
|
2020-03-03 06:14:08 +08:00
|
|
|
success = cache_bin_dalloc_easy(&bin, &ptrs[i]);
|
|
|
|
expect_true(success,
|
|
|
|
"Should be able to dalloc into a non-full cache bin.");
|
|
|
|
expect_true(cache_bin_low_water_get(&bin, &info) == 0,
|
|
|
|
"Pushes and pops shouldn't change low water of zero.");
|
|
|
|
}
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(&bin, &info) == ncached_max,
|
|
|
|
"");
|
2020-03-03 06:14:08 +08:00
|
|
|
success = cache_bin_dalloc_easy(&bin, &ptrs[ncached_max]);
|
|
|
|
expect_false(success, "Shouldn't be able to dalloc into a full bin.");
|
|
|
|
|
|
|
|
cache_bin_low_water_set(&bin);
|
|
|
|
|
|
|
|
for (cache_bin_sz_t i = 0; i < ncached_max; i++) {
|
|
|
|
expect_true(cache_bin_low_water_get(&bin, &info)
|
|
|
|
== ncached_max - i, "");
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(&bin, &info)
|
2020-03-03 06:14:08 +08:00
|
|
|
== ncached_max - i, "");
|
|
|
|
/*
|
2020-03-05 00:58:42 +08:00
|
|
|
* This should fail -- the easy variant can't change the low
|
|
|
|
* water mark.
|
2020-03-03 06:14:08 +08:00
|
|
|
*/
|
2020-03-05 00:58:42 +08:00
|
|
|
ptr = cache_bin_alloc_easy(&bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_ptr_null(ptr, "");
|
|
|
|
expect_false(success, "");
|
|
|
|
expect_true(cache_bin_low_water_get(&bin, &info)
|
|
|
|
== ncached_max - i, "");
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(&bin, &info)
|
2020-03-03 06:14:08 +08:00
|
|
|
== ncached_max - i, "");
|
|
|
|
|
|
|
|
/* This should succeed, though. */
|
2020-03-05 00:58:42 +08:00
|
|
|
ptr = cache_bin_alloc(&bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_true(success, "");
|
|
|
|
expect_ptr_eq(ptr, &ptrs[ncached_max - i - 1],
|
|
|
|
"Alloc should pop in stack order");
|
|
|
|
expect_true(cache_bin_low_water_get(&bin, &info)
|
|
|
|
== ncached_max - i - 1, "");
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(&bin, &info)
|
2020-03-03 06:14:08 +08:00
|
|
|
== ncached_max - i - 1, "");
|
|
|
|
}
|
|
|
|
/* Now we're empty -- all alloc attempts should fail. */
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(&bin, &info) == 0, "");
|
2020-03-05 00:58:42 +08:00
|
|
|
ptr = cache_bin_alloc_easy(&bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_ptr_null(ptr, "");
|
|
|
|
expect_false(success, "");
|
2020-03-05 00:58:42 +08:00
|
|
|
ptr = cache_bin_alloc(&bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_ptr_null(ptr, "");
|
|
|
|
expect_false(success, "");
|
|
|
|
|
|
|
|
for (cache_bin_sz_t i = 0; i < ncached_max / 2; i++) {
|
|
|
|
cache_bin_dalloc_easy(&bin, &ptrs[i]);
|
|
|
|
}
|
|
|
|
cache_bin_low_water_set(&bin);
|
|
|
|
|
|
|
|
for (cache_bin_sz_t i = ncached_max / 2; i < ncached_max; i++) {
|
|
|
|
cache_bin_dalloc_easy(&bin, &ptrs[i]);
|
|
|
|
}
|
2021-01-08 05:22:08 +08:00
|
|
|
expect_true(cache_bin_ncached_get_local(&bin, &info) == ncached_max,
|
|
|
|
"");
|
2020-03-03 06:14:08 +08:00
|
|
|
for (cache_bin_sz_t i = ncached_max - 1; i >= ncached_max / 2; i--) {
|
|
|
|
/*
|
|
|
|
* Size is bigger than low water -- the reduced version should
|
|
|
|
* succeed.
|
|
|
|
*/
|
2020-03-05 00:58:42 +08:00
|
|
|
ptr = cache_bin_alloc_easy(&bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_true(success, "");
|
|
|
|
expect_ptr_eq(ptr, &ptrs[i], "");
|
|
|
|
}
|
|
|
|
/* But now, we've hit low-water. */
|
2020-03-05 00:58:42 +08:00
|
|
|
ptr = cache_bin_alloc_easy(&bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_false(success, "");
|
|
|
|
expect_ptr_null(ptr, "");
|
|
|
|
|
|
|
|
/* We're going to test filling -- we must be empty to start. */
|
2021-01-08 05:22:08 +08:00
|
|
|
while (cache_bin_ncached_get_local(&bin, &info)) {
|
2020-03-05 00:58:42 +08:00
|
|
|
cache_bin_alloc(&bin, &success);
|
2020-03-03 06:14:08 +08:00
|
|
|
expect_true(success, "");
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Test fill. */
|
|
|
|
/* Try to fill all, succeed fully. */
|
|
|
|
do_fill_test(&bin, &info, ptrs, ncached_max, ncached_max, ncached_max);
|
|
|
|
/* Try to fill all, succeed partially. */
|
|
|
|
do_fill_test(&bin, &info, ptrs, ncached_max, ncached_max,
|
|
|
|
ncached_max / 2);
|
|
|
|
/* Try to fill all, fail completely. */
|
|
|
|
do_fill_test(&bin, &info, ptrs, ncached_max, ncached_max, 0);
|
|
|
|
|
|
|
|
/* Try to fill some, succeed fully. */
|
|
|
|
do_fill_test(&bin, &info, ptrs, ncached_max, ncached_max / 2,
|
|
|
|
ncached_max / 2);
|
|
|
|
/* Try to fill some, succeed partially. */
|
|
|
|
do_fill_test(&bin, &info, ptrs, ncached_max, ncached_max / 2,
|
2020-10-23 05:56:15 +08:00
|
|
|
ncached_max / 4);
|
2020-03-03 06:14:08 +08:00
|
|
|
/* Try to fill some, fail completely. */
|
|
|
|
do_fill_test(&bin, &info, ptrs, ncached_max, ncached_max / 2, 0);
|
2019-08-10 13:12:47 +08:00
|
|
|
|
2020-03-03 06:14:08 +08:00
|
|
|
do_flush_test(&bin, &info, ptrs, ncached_max, ncached_max);
|
|
|
|
do_flush_test(&bin, &info, ptrs, ncached_max, ncached_max / 2);
|
|
|
|
do_flush_test(&bin, &info, ptrs, ncached_max, 0);
|
|
|
|
do_flush_test(&bin, &info, ptrs, ncached_max / 2, ncached_max / 2);
|
|
|
|
do_flush_test(&bin, &info, ptrs, ncached_max / 2, ncached_max / 4);
|
|
|
|
do_flush_test(&bin, &info, ptrs, ncached_max / 2, 0);
|
2020-10-23 05:56:15 +08:00
|
|
|
|
2020-10-23 07:07:25 +08:00
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max, ncached_max);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max, ncached_max * 2);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max, ncached_max / 2);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max, 2);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max, 1);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max, 0);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max / 2,
|
|
|
|
ncached_max / 2);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max / 2, ncached_max);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max / 2,
|
|
|
|
ncached_max / 4);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max / 2, 2);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max / 2, 1);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, ncached_max / 2, 0);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 2, ncached_max);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 2, 2);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 2, 1);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 2, 0);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 1, 2);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 1, 1);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 1, 0);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 0, 2);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 0, 1);
|
|
|
|
do_batch_alloc_test(&bin, &info, ptrs, 0, 0);
|
|
|
|
|
2020-10-23 05:56:15 +08:00
|
|
|
free(ptrs);
|
2019-08-10 13:12:47 +08:00
|
|
|
}
|
|
|
|
TEST_END
|
|
|
|
|
|
|
|
int
|
|
|
|
main(void) {
|
|
|
|
return test(test_cache_bin);
|
|
|
|
}
|