X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Flibsystemd-bus%2Ftest-bus-kernel-benchmark.c;h=d9ccdc223947fea8b3bd0df73f19dc37932a3ff8;hp=403af885c3d44c18c281438d4b714e210e170fa2;hb=83d97ce664806829b027c02708d648d469724362;hpb=8f155917bf5c11c8f156d7f25f242257d6086cb9

diff --git a/src/libsystemd-bus/test-bus-kernel-benchmark.c b/src/libsystemd-bus/test-bus-kernel-benchmark.c
index 403af885c..d9ccdc223 100644
--- a/src/libsystemd-bus/test-bus-kernel-benchmark.c
+++ b/src/libsystemd-bus/test-bus-kernel-benchmark.c
@@ -24,6 +24,7 @@
 
 #include "util.h"
 #include "log.h"
+#include "time-util.h"
 
 #include "sd-bus.h"
 #include "bus-message.h"
@@ -31,11 +32,11 @@
 #include "bus-kernel.h"
 #include "bus-internal.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 (;;) {
@@ -46,180 +47,256 @@ static void server(sd_bus *b, usec_t *result) {
 
                 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);
-
-                        x = now(CLOCK_MONOTONIC);
+                if (sd_bus_message_is_method_call(m, "benchmark.server", "Ping"))
+                        assert_se(sd_bus_reply_method_return(b, m, NULL) >= 0);
+                else if (sd_bus_message_is_method_call(m, "benchmark.server", "Work")) {
+                        const void *p;
+                        size_t sz;
 
-                        for (i = 0; i < sz; i++)
-                                q[i] = toupper(p[i]);
+                        /* Make sure the mmap is mapped */
+                        assert_se(sd_bus_message_read_array(m, 'y', &p, &sz) > 0);
 
-                        x = now(CLOCK_MONOTONIC) - x;
-
-                        assert_se(sd_bus_send(b, reply, NULL) >= 0);
+                        assert_se(sd_bus_reply_method_return(b, 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_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_copying * 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;
 
         log_set_max_level(LOG_DEBUG);
 
-        lsize = 1;
-        rsize = MAX_SIZE;
+        for (i = 1; i < argc; i++) {
+                if (streq(argv[i], "chart")) {
+                        mode = MODE_CHART;
+                        continue;
+                }
 
-        for (;;) {
-                usec_t copy = 0, memfd = 0;
-                unsigned i;
+                assert_se(parse_sec(argv[i], &arg_loop_usec) >= 0);
+        }
 
-                csize = (lsize + rsize) / 2;
+        assert_se(arg_loop_usec > 0);
 
-                log_info("Trying size=%zu", csize);
+        bus_ref = bus_kernel_create("deine-mutter", &bus_name);
+        if (bus_ref == -ENOENT)
+                exit(EXIT_TEST_SKIP);
 
-                if (csize <= lsize)
-                        break;
+        assert_se(bus_ref >= 0);
 
-                for (i = 0; i <  N_TRIES; i++) {
-                        usec_t t;
+        address = strappend("kernel:path=", bus_name);
+        assert_se(address);
 
-                        run_benchmark(csize, true, &t);
-                        copy += t;
-                }
+        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, false, &t);
-                        memfd += t;
-                }
+        sync();
+        setpriority(PRIO_PROCESS, 0, -19);
+
+        pid = fork();
+        assert_se(pid >= 0);
+
+        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;
 }