aboutsummaryrefslogtreecommitdiff
path: root/tests/test-aio-multithread.c
diff options
context:
space:
mode:
authorThomas Huth <thuth@redhat.com>2021-03-10 07:33:14 +0100
committerThomas Huth <thuth@redhat.com>2021-03-12 15:46:30 +0100
commitda668aa15b99150a8595c491aee00d5d2426aaf9 (patch)
tree0463b0a303e807bdab46460f6c702be611bd7179 /tests/test-aio-multithread.c
parent363fc963054d8e82cfd55fa9b9aa130692a8dbd7 (diff)
tests: Move unit tests into a separate directory
The main tests directory still looks very crowded, and it's not clear which files are part of a unit tests and which belong to a different test subsystem. Let's clean up the mess and move the unit tests to a separate directory. Message-Id: <20210310063314.1049838-1-thuth@redhat.com> Acked-by: Paolo Bonzini <pbonzini@redhat.com> Signed-off-by: Thomas Huth <thuth@redhat.com>
Diffstat (limited to 'tests/test-aio-multithread.c')
-rw-r--r--tests/test-aio-multithread.c460
1 files changed, 0 insertions, 460 deletions
diff --git a/tests/test-aio-multithread.c b/tests/test-aio-multithread.c
deleted file mode 100644
index a555cc8835..0000000000
--- a/tests/test-aio-multithread.c
+++ /dev/null
@@ -1,460 +0,0 @@
-/*
- * AioContext multithreading tests
- *
- * Copyright Red Hat, Inc. 2016
- *
- * Authors:
- * Paolo Bonzini <pbonzini@redhat.com>
- *
- * This work is licensed under the terms of the GNU LGPL, version 2 or later.
- * See the COPYING.LIB file in the top-level directory.
- */
-
-#include "qemu/osdep.h"
-#include "block/aio.h"
-#include "qemu/coroutine.h"
-#include "qemu/thread.h"
-#include "qemu/error-report.h"
-#include "iothread.h"
-
-/* AioContext management */
-
-#define NUM_CONTEXTS 5
-
-static IOThread *threads[NUM_CONTEXTS];
-static AioContext *ctx[NUM_CONTEXTS];
-static __thread int id = -1;
-
-static QemuEvent done_event;
-
-/* Run a function synchronously on a remote iothread. */
-
-typedef struct CtxRunData {
- QEMUBHFunc *cb;
- void *arg;
-} CtxRunData;
-
-static void ctx_run_bh_cb(void *opaque)
-{
- CtxRunData *data = opaque;
-
- data->cb(data->arg);
- qemu_event_set(&done_event);
-}
-
-static void ctx_run(int i, QEMUBHFunc *cb, void *opaque)
-{
- CtxRunData data = {
- .cb = cb,
- .arg = opaque
- };
-
- qemu_event_reset(&done_event);
- aio_bh_schedule_oneshot(ctx[i], ctx_run_bh_cb, &data);
- qemu_event_wait(&done_event);
-}
-
-/* Starting the iothreads. */
-
-static void set_id_cb(void *opaque)
-{
- int *i = opaque;
-
- id = *i;
-}
-
-static void create_aio_contexts(void)
-{
- int i;
-
- for (i = 0; i < NUM_CONTEXTS; i++) {
- threads[i] = iothread_new();
- ctx[i] = iothread_get_aio_context(threads[i]);
- }
-
- qemu_event_init(&done_event, false);
- for (i = 0; i < NUM_CONTEXTS; i++) {
- ctx_run(i, set_id_cb, &i);
- }
-}
-
-/* Stopping the iothreads. */
-
-static void join_aio_contexts(void)
-{
- int i;
-
- for (i = 0; i < NUM_CONTEXTS; i++) {
- aio_context_ref(ctx[i]);
- }
- for (i = 0; i < NUM_CONTEXTS; i++) {
- iothread_join(threads[i]);
- }
- for (i = 0; i < NUM_CONTEXTS; i++) {
- aio_context_unref(ctx[i]);
- }
- qemu_event_destroy(&done_event);
-}
-
-/* Basic test for the stuff above. */
-
-static void test_lifecycle(void)
-{
- create_aio_contexts();
- join_aio_contexts();
-}
-
-/* aio_co_schedule test. */
-
-static Coroutine *to_schedule[NUM_CONTEXTS];
-
-static bool now_stopping;
-
-static int count_retry;
-static int count_here;
-static int count_other;
-
-static bool schedule_next(int n)
-{
- Coroutine *co;
-
- co = qatomic_xchg(&to_schedule[n], NULL);
- if (!co) {
- qatomic_inc(&count_retry);
- return false;
- }
-
- if (n == id) {
- qatomic_inc(&count_here);
- } else {
- qatomic_inc(&count_other);
- }
-
- aio_co_schedule(ctx[n], co);
- return true;
-}
-
-static void finish_cb(void *opaque)
-{
- schedule_next(id);
-}
-
-static coroutine_fn void test_multi_co_schedule_entry(void *opaque)
-{
- g_assert(to_schedule[id] == NULL);
-
- while (!qatomic_mb_read(&now_stopping)) {
- int n;
-
- n = g_test_rand_int_range(0, NUM_CONTEXTS);
- schedule_next(n);
-
- qatomic_mb_set(&to_schedule[id], qemu_coroutine_self());
- qemu_coroutine_yield();
- g_assert(to_schedule[id] == NULL);
- }
-}
-
-
-static void test_multi_co_schedule(int seconds)
-{
- int i;
-
- count_here = count_other = count_retry = 0;
- now_stopping = false;
-
- create_aio_contexts();
- for (i = 0; i < NUM_CONTEXTS; i++) {
- Coroutine *co1 = qemu_coroutine_create(test_multi_co_schedule_entry, NULL);
- aio_co_schedule(ctx[i], co1);
- }
-
- g_usleep(seconds * 1000000);
-
- qatomic_mb_set(&now_stopping, true);
- for (i = 0; i < NUM_CONTEXTS; i++) {
- ctx_run(i, finish_cb, NULL);
- to_schedule[i] = NULL;
- }
-
- join_aio_contexts();
- g_test_message("scheduled %d, queued %d, retry %d, total %d",
- count_other, count_here, count_retry,
- count_here + count_other + count_retry);
-}
-
-static void test_multi_co_schedule_1(void)
-{
- test_multi_co_schedule(1);
-}
-
-static void test_multi_co_schedule_10(void)
-{
- test_multi_co_schedule(10);
-}
-
-/* CoMutex thread-safety. */
-
-static uint32_t atomic_counter;
-static uint32_t running;
-static uint32_t counter;
-static CoMutex comutex;
-
-static void coroutine_fn test_multi_co_mutex_entry(void *opaque)
-{
- while (!qatomic_mb_read(&now_stopping)) {
- qemu_co_mutex_lock(&comutex);
- counter++;
- qemu_co_mutex_unlock(&comutex);
-
- /* Increase atomic_counter *after* releasing the mutex. Otherwise
- * there is a chance (it happens about 1 in 3 runs) that the iothread
- * exits before the coroutine is woken up, causing a spurious
- * assertion failure.
- */
- qatomic_inc(&atomic_counter);
- }
- qatomic_dec(&running);
-}
-
-static void test_multi_co_mutex(int threads, int seconds)
-{
- int i;
-
- qemu_co_mutex_init(&comutex);
- counter = 0;
- atomic_counter = 0;
- now_stopping = false;
-
- create_aio_contexts();
- assert(threads <= NUM_CONTEXTS);
- running = threads;
- for (i = 0; i < threads; i++) {
- Coroutine *co1 = qemu_coroutine_create(test_multi_co_mutex_entry, NULL);
- aio_co_schedule(ctx[i], co1);
- }
-
- g_usleep(seconds * 1000000);
-
- qatomic_mb_set(&now_stopping, true);
- while (running > 0) {
- g_usleep(100000);
- }
-
- join_aio_contexts();
- g_test_message("%d iterations/second", counter / seconds);
- g_assert_cmpint(counter, ==, atomic_counter);
-}
-
-/* Testing with NUM_CONTEXTS threads focuses on the queue. The mutex however
- * is too contended (and the threads spend too much time in aio_poll)
- * to actually stress the handoff protocol.
- */
-static void test_multi_co_mutex_1(void)
-{
- test_multi_co_mutex(NUM_CONTEXTS, 1);
-}
-
-static void test_multi_co_mutex_10(void)
-{
- test_multi_co_mutex(NUM_CONTEXTS, 10);
-}
-
-/* Testing with fewer threads stresses the handoff protocol too. Still, the
- * case where the locker _can_ pick up a handoff is very rare, happening
- * about 10 times in 1 million, so increase the runtime a bit compared to
- * other "quick" testcases that only run for 1 second.
- */
-static void test_multi_co_mutex_2_3(void)
-{
- test_multi_co_mutex(2, 3);
-}
-
-static void test_multi_co_mutex_2_30(void)
-{
- test_multi_co_mutex(2, 30);
-}
-
-/* Same test with fair mutexes, for performance comparison. */
-
-#ifdef CONFIG_LINUX
-#include "qemu/futex.h"
-
-/* The nodes for the mutex reside in this structure (on which we try to avoid
- * false sharing). The head of the mutex is in the "mutex_head" variable.
- */
-static struct {
- int next, locked;
- int padding[14];
-} nodes[NUM_CONTEXTS] __attribute__((__aligned__(64)));
-
-static int mutex_head = -1;
-
-static void mcs_mutex_lock(void)
-{
- int prev;
-
- nodes[id].next = -1;
- nodes[id].locked = 1;
- prev = qatomic_xchg(&mutex_head, id);
- if (prev != -1) {
- qatomic_set(&nodes[prev].next, id);
- qemu_futex_wait(&nodes[id].locked, 1);
- }
-}
-
-static void mcs_mutex_unlock(void)
-{
- int next;
- if (qatomic_read(&nodes[id].next) == -1) {
- if (qatomic_read(&mutex_head) == id &&
- qatomic_cmpxchg(&mutex_head, id, -1) == id) {
- /* Last item in the list, exit. */
- return;
- }
- while (qatomic_read(&nodes[id].next) == -1) {
- /* mcs_mutex_lock did the xchg, but has not updated
- * nodes[prev].next yet.
- */
- }
- }
-
- /* Wake up the next in line. */
- next = qatomic_read(&nodes[id].next);
- nodes[next].locked = 0;
- qemu_futex_wake(&nodes[next].locked, 1);
-}
-
-static void test_multi_fair_mutex_entry(void *opaque)
-{
- while (!qatomic_mb_read(&now_stopping)) {
- mcs_mutex_lock();
- counter++;
- mcs_mutex_unlock();
- qatomic_inc(&atomic_counter);
- }
- qatomic_dec(&running);
-}
-
-static void test_multi_fair_mutex(int threads, int seconds)
-{
- int i;
-
- assert(mutex_head == -1);
- counter = 0;
- atomic_counter = 0;
- now_stopping = false;
-
- create_aio_contexts();
- assert(threads <= NUM_CONTEXTS);
- running = threads;
- for (i = 0; i < threads; i++) {
- Coroutine *co1 = qemu_coroutine_create(test_multi_fair_mutex_entry, NULL);
- aio_co_schedule(ctx[i], co1);
- }
-
- g_usleep(seconds * 1000000);
-
- qatomic_mb_set(&now_stopping, true);
- while (running > 0) {
- g_usleep(100000);
- }
-
- join_aio_contexts();
- g_test_message("%d iterations/second", counter / seconds);
- g_assert_cmpint(counter, ==, atomic_counter);
-}
-
-static void test_multi_fair_mutex_1(void)
-{
- test_multi_fair_mutex(NUM_CONTEXTS, 1);
-}
-
-static void test_multi_fair_mutex_10(void)
-{
- test_multi_fair_mutex(NUM_CONTEXTS, 10);
-}
-#endif
-
-/* Same test with pthread mutexes, for performance comparison and
- * portability. */
-
-static QemuMutex mutex;
-
-static void test_multi_mutex_entry(void *opaque)
-{
- while (!qatomic_mb_read(&now_stopping)) {
- qemu_mutex_lock(&mutex);
- counter++;
- qemu_mutex_unlock(&mutex);
- qatomic_inc(&atomic_counter);
- }
- qatomic_dec(&running);
-}
-
-static void test_multi_mutex(int threads, int seconds)
-{
- int i;
-
- qemu_mutex_init(&mutex);
- counter = 0;
- atomic_counter = 0;
- now_stopping = false;
-
- create_aio_contexts();
- assert(threads <= NUM_CONTEXTS);
- running = threads;
- for (i = 0; i < threads; i++) {
- Coroutine *co1 = qemu_coroutine_create(test_multi_mutex_entry, NULL);
- aio_co_schedule(ctx[i], co1);
- }
-
- g_usleep(seconds * 1000000);
-
- qatomic_mb_set(&now_stopping, true);
- while (running > 0) {
- g_usleep(100000);
- }
-
- join_aio_contexts();
- g_test_message("%d iterations/second", counter / seconds);
- g_assert_cmpint(counter, ==, atomic_counter);
-}
-
-static void test_multi_mutex_1(void)
-{
- test_multi_mutex(NUM_CONTEXTS, 1);
-}
-
-static void test_multi_mutex_10(void)
-{
- test_multi_mutex(NUM_CONTEXTS, 10);
-}
-
-/* End of tests. */
-
-int main(int argc, char **argv)
-{
- init_clocks(NULL);
-
- g_test_init(&argc, &argv, NULL);
- g_test_add_func("/aio/multi/lifecycle", test_lifecycle);
- if (g_test_quick()) {
- g_test_add_func("/aio/multi/schedule", test_multi_co_schedule_1);
- g_test_add_func("/aio/multi/mutex/contended", test_multi_co_mutex_1);
- g_test_add_func("/aio/multi/mutex/handoff", test_multi_co_mutex_2_3);
-#ifdef CONFIG_LINUX
- g_test_add_func("/aio/multi/mutex/mcs", test_multi_fair_mutex_1);
-#endif
- g_test_add_func("/aio/multi/mutex/pthread", test_multi_mutex_1);
- } else {
- g_test_add_func("/aio/multi/schedule", test_multi_co_schedule_10);
- g_test_add_func("/aio/multi/mutex/contended", test_multi_co_mutex_10);
- g_test_add_func("/aio/multi/mutex/handoff", test_multi_co_mutex_2_30);
-#ifdef CONFIG_LINUX
- g_test_add_func("/aio/multi/mutex/mcs", test_multi_fair_mutex_10);
-#endif
- g_test_add_func("/aio/multi/mutex/pthread", test_multi_mutex_10);
- }
- return g_test_run();
-}