^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0-only
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) * Copyright (C) 2016 Red Hat, Inc.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) * Author: Michael S. Tsirkin <mst@redhat.com>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) * Command line processing and common functions for ring benchmarking.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) #define _GNU_SOURCE
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) #include <getopt.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) #include <pthread.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) #include <assert.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) #include <sched.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) #include "main.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) #include <sys/eventfd.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) #include <stdlib.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) #include <stdio.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) #include <unistd.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) #include <limits.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20) int runcycles = 10000000;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) int max_outstanding = INT_MAX;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) int batch = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) int param = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) bool do_sleep = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) bool do_relax = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) bool do_exit = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) unsigned ring_size = 256;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) static int kickfd = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) static int callfd = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) void notify(int fd)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36) unsigned long long v = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) int r;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) vmexit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) r = write(fd, &v, sizeof v);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41) assert(r == sizeof v);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) vmentry();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) void wait_for_notify(int fd)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47) unsigned long long v = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48) int r;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) vmexit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51) r = read(fd, &v, sizeof v);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52) assert(r == sizeof v);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) vmentry();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56) void kick(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58) notify(kickfd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 61) void wait_for_kick(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 62) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 63) wait_for_notify(kickfd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 64) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 65)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 66) void call(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 67) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 68) notify(callfd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 69) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 70)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 71) void wait_for_call(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 72) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 73) wait_for_notify(callfd);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 74) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 75)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 76) void set_affinity(const char *arg)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 77) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 78) cpu_set_t cpuset;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 79) int ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 80) pthread_t self;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 81) long int cpu;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 82) char *endptr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 83)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 84) if (!arg)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 85) return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 86)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 87) cpu = strtol(arg, &endptr, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 88) assert(!*endptr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 89)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 90) assert(cpu >= 0 && cpu < CPU_SETSIZE);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 91)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 92) self = pthread_self();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 93) CPU_ZERO(&cpuset);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 94) CPU_SET(cpu, &cpuset);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 95)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 96) ret = pthread_setaffinity_np(self, sizeof(cpu_set_t), &cpuset);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 97) assert(!ret);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 98) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 99)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 100) void poll_used(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 101) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 102) while (used_empty())
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 103) busy_wait();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 104) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 105)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 106) static void __attribute__((__flatten__)) run_guest(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 107) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 108) int completed_before;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 109) int completed = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 110) int started = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 111) int bufs = runcycles;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 112) int spurious = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 113) int r;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 114) unsigned len;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 115) void *buf;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 116) int tokick = batch;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 117)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 118) for (;;) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 119) if (do_sleep)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 120) disable_call();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 121) completed_before = completed;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 122) do {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 123) if (started < bufs &&
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 124) started - completed < max_outstanding) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 125) r = add_inbuf(0, "Buffer\n", "Hello, world!");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 126) if (__builtin_expect(r == 0, true)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 127) ++started;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 128) if (!--tokick) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 129) tokick = batch;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 130) if (do_sleep)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 131) kick_available();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 132) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 133)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 134) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 135) } else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 136) r = -1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 137)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 138) /* Flush out completed bufs if any */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 139) if (get_buf(&len, &buf)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 140) ++completed;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 141) if (__builtin_expect(completed == bufs, false))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 142) return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 143) r = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 144) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 145) } while (r == 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 146) if (completed == completed_before)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 147) ++spurious;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 148) assert(completed <= bufs);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 149) assert(started <= bufs);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 150) if (do_sleep) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 151) if (used_empty() && enable_call())
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 152) wait_for_call();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 153) } else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 154) poll_used();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 155) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 156) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 157) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 158)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 159) void poll_avail(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 160) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 161) while (avail_empty())
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 162) busy_wait();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 163) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 164)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 165) static void __attribute__((__flatten__)) run_host(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 166) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 167) int completed_before;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 168) int completed = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 169) int spurious = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 170) int bufs = runcycles;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 171) unsigned len;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 172) void *buf;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 173)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 174) for (;;) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 175) if (do_sleep) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 176) if (avail_empty() && enable_kick())
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 177) wait_for_kick();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 178) } else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 179) poll_avail();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 180) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 181) if (do_sleep)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 182) disable_kick();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 183) completed_before = completed;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 184) while (__builtin_expect(use_buf(&len, &buf), true)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 185) if (do_sleep)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 186) call_used();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 187) ++completed;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 188) if (__builtin_expect(completed == bufs, false))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 189) return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 190) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 191) if (completed == completed_before)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 192) ++spurious;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 193) assert(completed <= bufs);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 194) if (completed == bufs)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 195) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 196) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 197) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 198)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 199) void *start_guest(void *arg)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 200) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 201) set_affinity(arg);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 202) run_guest();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 203) pthread_exit(NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 204) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 205)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 206) void *start_host(void *arg)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 207) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 208) set_affinity(arg);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 209) run_host();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 210) pthread_exit(NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 211) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 212)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 213) static const char optstring[] = "";
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 214) static const struct option longopts[] = {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 215) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 216) .name = "help",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 217) .has_arg = no_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 218) .val = 'h',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 219) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 220) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 221) .name = "host-affinity",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 222) .has_arg = required_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 223) .val = 'H',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 224) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 225) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 226) .name = "guest-affinity",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 227) .has_arg = required_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 228) .val = 'G',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 229) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 230) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 231) .name = "ring-size",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 232) .has_arg = required_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 233) .val = 'R',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 234) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 235) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 236) .name = "run-cycles",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 237) .has_arg = required_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 238) .val = 'C',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 239) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 240) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 241) .name = "outstanding",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 242) .has_arg = required_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 243) .val = 'o',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 244) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 245) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 246) .name = "batch",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 247) .has_arg = required_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 248) .val = 'b',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 249) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 250) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 251) .name = "param",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 252) .has_arg = required_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 253) .val = 'p',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 254) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 255) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 256) .name = "sleep",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 257) .has_arg = no_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 258) .val = 's',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 259) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 260) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 261) .name = "relax",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 262) .has_arg = no_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 263) .val = 'x',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 264) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 265) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 266) .name = "exit",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 267) .has_arg = no_argument,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 268) .val = 'e',
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 269) },
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 270) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 271) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 272) };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 273)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 274) static void help(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 275) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 276) fprintf(stderr, "Usage: <test> [--help]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 277) " [--host-affinity H]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 278) " [--guest-affinity G]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 279) " [--ring-size R (default: %d)]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 280) " [--run-cycles C (default: %d)]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 281) " [--batch b]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 282) " [--outstanding o]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 283) " [--param p]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 284) " [--sleep]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 285) " [--relax]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 286) " [--exit]"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 287) "\n",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 288) ring_size,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 289) runcycles);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 290) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 291)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 292) int main(int argc, char **argv)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 293) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 294) int ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 295) pthread_t host, guest;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 296) void *tret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 297) char *host_arg = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 298) char *guest_arg = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 299) char *endptr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 300) long int c;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 301)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 302) kickfd = eventfd(0, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 303) assert(kickfd >= 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 304) callfd = eventfd(0, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 305) assert(callfd >= 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 306)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 307) for (;;) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 308) int o = getopt_long(argc, argv, optstring, longopts, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 309) switch (o) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 310) case -1:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 311) goto done;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 312) case '?':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 313) help();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 314) exit(2);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 315) case 'H':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 316) host_arg = optarg;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 317) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 318) case 'G':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 319) guest_arg = optarg;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 320) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 321) case 'R':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 322) ring_size = strtol(optarg, &endptr, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 323) assert(ring_size && !(ring_size & (ring_size - 1)));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 324) assert(!*endptr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 325) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 326) case 'C':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 327) c = strtol(optarg, &endptr, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 328) assert(!*endptr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 329) assert(c > 0 && c < INT_MAX);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 330) runcycles = c;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 331) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 332) case 'o':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 333) c = strtol(optarg, &endptr, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 334) assert(!*endptr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 335) assert(c > 0 && c < INT_MAX);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 336) max_outstanding = c;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 337) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 338) case 'p':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 339) c = strtol(optarg, &endptr, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 340) assert(!*endptr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 341) assert(c > 0 && c < INT_MAX);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 342) param = c;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 343) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 344) case 'b':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 345) c = strtol(optarg, &endptr, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 346) assert(!*endptr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 347) assert(c > 0 && c < INT_MAX);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 348) batch = c;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 349) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 350) case 's':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 351) do_sleep = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 352) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 353) case 'x':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 354) do_relax = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 355) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 356) case 'e':
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 357) do_exit = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 358) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 359) default:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 360) help();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 361) exit(4);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 362) break;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 363) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 364) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 365)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 366) /* does nothing here, used to make sure all smp APIs compile */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 367) smp_acquire();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 368) smp_release();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 369) smp_mb();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 370) done:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 371)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 372) if (batch > max_outstanding)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 373) batch = max_outstanding;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 374)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 375) if (optind < argc) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 376) help();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 377) exit(4);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 378) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 379) alloc_ring();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 380)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 381) ret = pthread_create(&host, NULL, start_host, host_arg);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 382) assert(!ret);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 383) ret = pthread_create(&guest, NULL, start_guest, guest_arg);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 384) assert(!ret);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 385)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 386) ret = pthread_join(guest, &tret);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 387) assert(!ret);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 388) ret = pthread_join(host, &tret);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 389) assert(!ret);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 390) return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 391) }