81 lines
1.7 KiB
C
81 lines
1.7 KiB
C
#include "test/jemalloc_test.h"
|
|
|
|
static const uint64_t interval = 1 << 20;
|
|
|
|
TEST_BEGIN(test_counter_accum) {
|
|
uint64_t increment = interval >> 4;
|
|
unsigned n = interval / increment;
|
|
uint64_t accum = 0;
|
|
|
|
counter_accum_t c;
|
|
counter_accum_init(&c, interval);
|
|
|
|
tsd_t *tsd = tsd_fetch();
|
|
bool trigger;
|
|
for (unsigned i = 0; i < n; i++) {
|
|
trigger = counter_accum(tsd_tsdn(tsd), &c, increment);
|
|
accum += increment;
|
|
if (accum < interval) {
|
|
expect_b_eq(trigger, false, "Should not trigger");
|
|
} else {
|
|
expect_b_eq(trigger, true, "Should have triggered");
|
|
}
|
|
}
|
|
expect_b_eq(trigger, true, "Should have triggered");
|
|
}
|
|
TEST_END
|
|
|
|
void
|
|
expect_counter_value(counter_accum_t *c, uint64_t v) {
|
|
uint64_t accum = locked_read_u64_unsynchronized(&c->accumbytes);
|
|
expect_u64_eq(accum, v, "Counter value mismatch");
|
|
}
|
|
|
|
#define N_THDS (16)
|
|
#define N_ITER_THD (1 << 12)
|
|
#define ITER_INCREMENT (interval >> 4)
|
|
|
|
static void *
|
|
thd_start(void *varg) {
|
|
counter_accum_t *c = (counter_accum_t *)varg;
|
|
|
|
tsd_t *tsd = tsd_fetch();
|
|
bool trigger;
|
|
uintptr_t n_triggered = 0;
|
|
for (unsigned i = 0; i < N_ITER_THD; i++) {
|
|
trigger = counter_accum(tsd_tsdn(tsd), c, ITER_INCREMENT);
|
|
n_triggered += trigger ? 1 : 0;
|
|
}
|
|
|
|
return (void *)n_triggered;
|
|
}
|
|
|
|
|
|
TEST_BEGIN(test_counter_mt) {
|
|
counter_accum_t shared_c;
|
|
counter_accum_init(&shared_c, interval);
|
|
|
|
thd_t thds[N_THDS];
|
|
unsigned i;
|
|
for (i = 0; i < N_THDS; i++) {
|
|
thd_create(&thds[i], thd_start, (void *)&shared_c);
|
|
}
|
|
|
|
uint64_t sum = 0;
|
|
for (i = 0; i < N_THDS; i++) {
|
|
void *ret;
|
|
thd_join(thds[i], &ret);
|
|
sum += (uintptr_t)ret;
|
|
}
|
|
expect_u64_eq(sum, N_THDS * N_ITER_THD / (interval / ITER_INCREMENT),
|
|
"Incorrect number of triggers");
|
|
}
|
|
TEST_END
|
|
|
|
int
|
|
main(void) {
|
|
return test(
|
|
test_counter_accum,
|
|
test_counter_mt);
|
|
}
|