project-base/server-skynet-source-3rd-jemalloc
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Split out locality checking in batch allocation tests

Browse Source
This commit is contained in:
Yinan Zhang 2020-10-20 11:00:09 -07:00
parent be5e49f4fa
commit ac480136d7
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
test/unit/batch_alloc.c
View File

@ -35,20 +35,28 @@ verify_stats(bin_stats_t *before, bin_stats_t *after, size_t batch,
} }
static void static void
verify_batch(tsd_t *tsd, void **ptrs, size_t batch, size_t usize, bool zero, verify_batch_basic(tsd_t *tsd, void **ptrs, size_t batch, size_t usize,
bool zero) {
for (size_t i = 0; i < batch; ++i) {
void *p = ptrs[i];
expect_zu_eq(isalloc(tsd_tsdn(tsd), p), usize, "");
if (zero) {
for (size_t k = 0; k < usize; ++k) {
expect_true(*((unsigned char *)p + k) == 0, "");
}
}
}
}
static void
verify_batch_locality(tsd_t *tsd, void **ptrs, size_t batch, size_t usize,
arena_t *arena, unsigned nregs) { arena_t *arena, unsigned nregs) {
for (size_t i = 0, j = 0; i < batch; ++i, ++j) { for (size_t i = 0, j = 0; i < batch; ++i, ++j) {
if (j == nregs) { if (j == nregs) {
j = 0; j = 0;
} }
void *p = ptrs[i]; void *p = ptrs[i];
expect_zu_eq(isalloc(tsd_tsdn(tsd), p), usize, "");
expect_ptr_eq(iaalloc(tsd_tsdn(tsd), p), arena, ""); expect_ptr_eq(iaalloc(tsd_tsdn(tsd), p), arena, "");
if (zero) {
for (size_t k = 0; k < usize; ++k) {
expect_true(*((unsigned char *)p + k) == 0, "");
}
}
if (j == 0) { if (j == 0) {
expect_true(PAGE_ALIGNED(p), ""); expect_true(PAGE_ALIGNED(p), "");
continue; continue;
@ -154,7 +162,8 @@ test_wrapper(size_t size, size_t alignment, bool zero, unsigned arena_flag) {
assert_zu_eq(filled, batch, ""); assert_zu_eq(filled, batch, "");
memcpy(&stats_after, &bin->stats, sizeof(bin_stats_t)); memcpy(&stats_after, &bin->stats, sizeof(bin_stats_t));
verify_stats(&stats_before, &stats_after, batch, nregs); verify_stats(&stats_before, &stats_after, batch, nregs);
verify_batch(tsd, ptrs, batch, usize, zero, arena, verify_batch_basic(tsd, ptrs, batch, usize, zero);
verify_batch_locality(tsd, ptrs, batch, usize, arena,
nregs); nregs);
release_batch(ptrs, batch, usize); release_batch(ptrs, batch, usize);
} }