#include "util.h"
#include "log.h"
+#include "time-util.h"
#include "sd-bus.h"
#include "bus-message.h"
#include "bus-error.h"
#include "bus-kernel.h"
#include "bus-internal.h"
+#include "bus-util.h"
-#define N_TRIES 500
-#define MAX_SIZE (5*1024*1024)
+#define MAX_SIZE (4*1024*1024)
-static void server(sd_bus *b, usec_t *result) {
- usec_t x = 0;
+static usec_t arg_loop_usec = 100 * USEC_PER_MSEC;
+
+static void server(sd_bus *b, size_t *result) {
int r;
for (;;) {
if (r == 0)
assert_se(sd_bus_wait(b, (usec_t) -1) >= 0);
-
if (!m)
continue;
- /* log_error("huhu %s from %s", sd_bus_message_get_member(m), sd_bus_message_get_sender(m)); */
-
- if (sd_bus_message_is_method_call(m, "benchmark.server", "Work")) {
- _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
- size_t i, sz;
- char *q;
- const char *p;
-
- assert_se(sd_bus_message_read_array(m, 'y', (const void**) &p, &sz) > 0);
- assert_se(sd_bus_message_new_method_return(b, m, &reply) >= 0);
- assert_se(sd_bus_message_append_array_space(reply, 'y', sz, (void**) &q) >= 0);
+ if (sd_bus_message_is_method_call(m, "benchmark.server", "Ping"))
+ assert_se(sd_bus_reply_method_return(m, NULL) >= 0);
+ else if (sd_bus_message_is_method_call(m, "benchmark.server", "Work")) {
+ const void *p;
+ size_t sz;
- x = now(CLOCK_MONOTONIC);
+ /* Make sure the mmap is mapped */
+ assert_se(sd_bus_message_read_array(m, 'y', &p, &sz) > 0);
- for (i = 0; i < sz; i++)
- q[i] = toupper(p[i]);
-
- x = now(CLOCK_MONOTONIC) - x;
-
- assert_se(sd_bus_send(b, reply, NULL) >= 0);
+ assert_se(sd_bus_reply_method_return(m, NULL) >= 0);
} else if (sd_bus_message_is_method_call(m, "benchmark.server", "Exit")) {
- usec_t t;
+ uint64_t res;
+ assert_se(sd_bus_message_read(m, "t", &res) > 0);
- assert_se(sd_bus_message_read(m, "t", &t) > 0);
- assert_se(t >= x);
- *result = t - x;
+ *result = res;
return;
- } else if (sd_bus_message_is_method_call(m, "benchmark.server", "Ping")) {
- assert_se(sd_bus_reply_method_return(b, m, "y", 1) >= 0);
} else
assert_not_reached("Unknown method");
}
}
-static void client(sd_bus *b, size_t sz) {
+static void transaction(sd_bus *b, size_t sz) {
_cleanup_bus_message_unref_ sd_bus_message *m = NULL, *reply = NULL;
- char *p;
- const char *q;
- usec_t t;
- size_t l, i;
-
- assert_se(sd_bus_call_method(b, ":1.1", "/", "benchmark.server", "Ping", NULL, NULL, NULL) >= 0);
+ uint8_t *p;
assert_se(sd_bus_message_new_method_call(b, ":1.1", "/", "benchmark.server", "Work", &m) >= 0);
assert_se(sd_bus_message_append_array_space(m, 'y', sz, (void**) &p) >= 0);
- for (i = 0; i < sz; i++)
- p[i] = 'a' + (char) (i % 26);
+ memset(p, 0x80, sz);
- t = now(CLOCK_MONOTONIC);
- assert_se(sd_bus_send_with_reply_and_block(b, m, 0, NULL, &reply) >= 0);
- t = now(CLOCK_MONOTONIC) - t;
+ assert_se(sd_bus_call(b, m, 0, NULL, &reply) >= 0);
+}
- assert_se(sd_bus_message_read_array(reply, 'y', (const void**) &q, &l) > 0);
- assert_se(l == sz);
+static void client_bisect(const char *address) {
+ _cleanup_bus_message_unref_ sd_bus_message *x = NULL;
+ size_t lsize, rsize, csize;
+ sd_bus *b;
+ int r;
- for (i = 0; i < sz; i++) {
- assert_se(q[i] == 'A' + (char) (i % 26));
- }
+ r = sd_bus_new(&b);
+ assert_se(r >= 0);
- sd_bus_message_unref(m);
+ r = sd_bus_set_address(b, address);
+ assert_se(r >= 0);
- assert_se(sd_bus_message_new_method_call(b, ":1.1", "/", "benchmark.server", "Exit", &m) >= 0);
- assert_se(sd_bus_message_append(m, "t", t) >= 0);
- assert_se(sd_bus_send(b, m, NULL) >= 0);
-}
+ r = sd_bus_start(b);
+ assert_se(r >= 0);
-static void run_benchmark(size_t sz, bool force_copy, usec_t *result) {
+ assert_se(sd_bus_call_method(b, ":1.1", "/", "benchmark.server", "Ping", NULL, NULL, NULL) >= 0);
- _cleanup_close_ int bus_ref = -1;
- _cleanup_free_ char *bus_name = NULL, *address = NULL;
- sd_bus *b;
- int r;
- pid_t pid;
+ lsize = 1;
+ rsize = MAX_SIZE;
- bus_ref = bus_kernel_create("deine-mutter", &bus_name);
- if (bus_ref == -ENOENT)
- exit(EXIT_TEST_SKIP);
+ printf("SIZE\tCOPY\tMEMFD\n");
- assert_se(bus_ref >= 0);
+ for (;;) {
+ usec_t t;
+ unsigned n_copying, n_memfd;
- address = strappend("kernel:path=", bus_name);
- assert_se(address);
+ csize = (lsize + rsize) / 2;
+
+ if (csize <= lsize)
+ break;
+
+ if (csize <= 0)
+ break;
+
+ printf("%zu\t", csize);
+
+ b->use_memfd = 0;
+
+ t = now(CLOCK_MONOTONIC);
+ for (n_copying = 0;; n_copying++) {
+ transaction(b, csize);
+ if (now(CLOCK_MONOTONIC) >= t + arg_loop_usec)
+ break;
+ }
+ printf("%u\t", (unsigned) ((n_copying * USEC_PER_SEC) / arg_loop_usec));
+
+ b->use_memfd = -1;
+
+ t = now(CLOCK_MONOTONIC);
+ for (n_memfd = 0;; n_memfd++) {
+ transaction(b, csize);
+ if (now(CLOCK_MONOTONIC) >= t + arg_loop_usec)
+ break;
+ }
+ printf("%u\n", (unsigned) ((n_memfd * USEC_PER_SEC) / arg_loop_usec));
+
+ if (n_copying == n_memfd)
+ break;
+
+ if (n_copying > n_memfd)
+ lsize = csize;
+ else
+ rsize = csize;
+ }
+
+ b->use_memfd = 1;
+ assert_se(sd_bus_message_new_method_call(b, ":1.1", "/", "benchmark.server", "Exit", &x) >= 0);
+ assert_se(sd_bus_message_append(x, "t", csize) >= 0);
+ assert_se(sd_bus_send(b, x, NULL) >= 0);
+
+ sd_bus_unref(b);
+}
+
+static void client_chart(const char *address) {
+ _cleanup_bus_message_unref_ sd_bus_message *x = NULL;
+ size_t csize;
+ sd_bus *b;
+ int r;
r = sd_bus_new(&b);
assert_se(r >= 0);
- b->use_memfd = force_copy ? 0 : -1;
-
r = sd_bus_set_address(b, address);
assert_se(r >= 0);
r = sd_bus_start(b);
assert_se(r >= 0);
- pid = fork();
- assert_se(pid >= 0);
+ assert_se(sd_bus_call_method(b, ":1.1", "/", "benchmark.server", "Ping", NULL, NULL, NULL) >= 0);
- if (pid == 0) {
- close_nointr_nofail(bus_ref);
- sd_bus_unref(b);
+ printf("SIZE\tCOPY\tMEMFD\n");
- r = sd_bus_new(&b);
- assert_se(r >= 0);
+ for (csize = 1; csize <= MAX_SIZE; csize *= 2) {
+ usec_t t;
+ unsigned n_copying, n_memfd;
- b->use_memfd = force_copy ? 0 : -1;
+ printf("%zu\t", csize);
- r = sd_bus_set_address(b, address);
- assert_se(r >= 0);
+ b->use_memfd = 0;
- r = sd_bus_start(b);
- assert_se(r >= 0);
+ t = now(CLOCK_MONOTONIC);
+ for (n_copying = 0;; n_copying++) {
+ transaction(b, csize);
+ if (now(CLOCK_MONOTONIC) >= t + arg_loop_usec)
+ break;
+ }
- client(b, sz);
- _exit(0);
+ printf("%u\t", (unsigned) ((n_copying * USEC_PER_SEC) / arg_loop_usec));
+
+ b->use_memfd = -1;
+
+ t = now(CLOCK_MONOTONIC);
+ for (n_memfd = 0;; n_memfd++) {
+ transaction(b, csize);
+ if (now(CLOCK_MONOTONIC) >= t + arg_loop_usec)
+ break;
+ }
+
+ printf("%u\n", (unsigned) ((n_memfd * USEC_PER_SEC) / arg_loop_usec));
}
- server(b, result);
- sd_bus_unref(b);
+ b->use_memfd = 1;
+ assert_se(sd_bus_message_new_method_call(b, ":1.1", "/", "benchmark.server", "Exit", &x) >= 0);
+ assert_se(sd_bus_message_append(x, "t", csize) >= 0);
+ assert_se(sd_bus_send(b, x, NULL) >= 0);
- assert_se(waitpid(pid, NULL, 0) == pid);
+ sd_bus_unref(b);
}
int main(int argc, char *argv[]) {
- size_t lsize, rsize, csize;
+ enum {
+ MODE_BISECT,
+ MODE_CHART,
+ } mode = MODE_BISECT;
+ int i;
+ _cleanup_free_ char *bus_name = NULL, *address = NULL;
+ _cleanup_close_ int bus_ref = -1;
+ cpu_set_t cpuset;
+ size_t result;
+ sd_bus *b;
+ pid_t pid;
+ int r;
+
+ for (i = 1; i < argc; i++) {
+ if (streq(argv[i], "chart")) {
+ mode = MODE_CHART;
+ continue;
+ }
- log_set_max_level(LOG_DEBUG);
+ assert_se(parse_sec(argv[i], &arg_loop_usec) >= 0);
+ }
- lsize = 1;
- rsize = MAX_SIZE;
+ assert_se(arg_loop_usec > 0);
- for (;;) {
- usec_t copy = 0, memfd = 0;
- unsigned i;
+ bus_ref = bus_kernel_create_bus("deine-mutter", &bus_name);
+ if (bus_ref == -ENOENT)
+ exit(EXIT_TEST_SKIP);
- csize = (lsize + rsize) / 2;
+ assert_se(bus_ref >= 0);
- log_info("Trying size=%zu", csize);
+ address = strappend("kernel:path=", bus_name);
+ assert_se(address);
- if (csize <= lsize)
- break;
+ r = sd_bus_new(&b);
+ assert_se(r >= 0);
+
+ r = sd_bus_set_address(b, address);
+ assert_se(r >= 0);
- for (i = 0; i < N_TRIES; i++) {
- usec_t t;
+ r = sd_bus_start(b);
+ assert_se(r >= 0);
- run_benchmark(csize, true, &t);
- copy += t;
- }
+ sync();
+ setpriority(PRIO_PROCESS, 0, -19);
- for (i = 0; i < N_TRIES; i++) {
- usec_t t;
+ pid = fork();
+ assert_se(pid >= 0);
- run_benchmark(csize, false, &t);
- memfd += t;
- }
+ if (pid == 0) {
+ CPU_ZERO(&cpuset);
+ CPU_SET(0, &cpuset);
+ pthread_setaffinity_np(pthread_self(), sizeof(cpu_set_t), &cpuset);
- copy /= N_TRIES;
- memfd /= N_TRIES;
+ close_nointr_nofail(bus_ref);
+ sd_bus_unref(b);
- if (copy == memfd)
+ switch (mode) {
+ case MODE_BISECT:
+ client_bisect(address);
break;
- if (copy < memfd)
- lsize = csize;
- else
- rsize = csize;
+ case MODE_CHART:
+ client_chart(address);
+ break;
+ }
+
+ _exit(0);
}
- log_info("Copying/memfd are equally fast at %zu", csize);
+ CPU_ZERO(&cpuset);
+ CPU_SET(1, &cpuset);
+ pthread_setaffinity_np(pthread_self(), sizeof(cpu_set_t), &cpuset);
+
+ server(b, &result);
+
+ if (mode == MODE_BISECT)
+ printf("Copying/memfd are equally fast at %zu bytes\n", result);
+
+ assert_se(waitpid(pid, NULL, 0) == pid);
+
+ sd_bus_unref(b);
return 0;
}