ust-fd: Add close_range declaration

[lttng-ust.git] / tests / benchmark / bench.c

diff --git a/tests/benchmark/bench.c b/tests/benchmark/bench.c
index ce476f9bd996c4b3ac2859c3b669c084db7e7dec..fcd373ad973ac1edc3d8a8806b6a762a43f2442c 100644 (file)
--- a/tests/benchmark/bench.c
+++ b/tests/benchmark/bench.c
@@ -1,57 +1,89 @@
 /*
- * bench.c
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ *
+ * Copyright 2010 Douglas Santos <douglas.santos@polymtl.ca>
+ * Copyright 2021 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
  *
  * LTTng Userspace Tracer (UST) - benchmark tool
  */
 
-#define _GNU_SOURCE
 #include <stdio.h>
 #include <pthread.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <sched.h>
-#include <ust/marker.h>
 #include <time.h>
+#include <urcu/compiler.h>
 
-static int nr_cpus;
-static unsigned long nr_events;
+#ifdef TRACING
+#define LTTNG_UST_TRACEPOINT_DEFINE
+#include "ust_tests_benchmark.h"
+#endif
 
-void do_stuff(void)
-{
-       int v;
-       FILE *file;
+#define printf_verbose(fmt, args...)           \
+       do {                                    \
+               if (verbose_mode)               \
+                       printf(fmt, ## args);   \
+       } while (0)
+
+static int verbose_mode;
+
+struct thread_counter {
+       unsigned long long nr_loops;
+};
 
-       v = 1;
+static int nr_threads;
+static unsigned long duration;
 
-       file = fopen("/dev/null", "a");
-       fprintf(file, "%d", v);
-       fclose(file);
-       time(NULL);
+static volatile int test_go, test_stop;
 
-#ifdef MARKER
-       ust_marker(event, "event %d", v);
+static
+void do_stuff(void)
+{
+       int i;
+#ifdef TRACING
+       int v = 50;
 #endif
 
+       for (i = 0; i < 100; i++)
+               cmm_barrier();
+#ifdef TRACING
+       lttng_ust_tracepoint(ust_tests_benchmark, tpbench, v);
+#endif
 }
 
 
-void *function(void *arg)
+static
+void *function(void *arg __attribute__((unused)))
 {
-       unsigned long i;
+       unsigned long long nr_loops = 0;
+       struct thread_counter *thread_counter = arg;
+
+       while (!test_go)
+               cmm_barrier();
 
-       for(i = 0; i < nr_events; i++) {
+       for (;;) {
                do_stuff();
+               nr_loops++;
+               if (test_stop)
+                       break;
        }
+       thread_counter->nr_loops = nr_loops;
        return NULL;
 }
 
+static
 void usage(char **argv) {
-       printf("Usage: %s nr_cpus nr_events\n", argv[0]);
+       printf("Usage: %s nr_threads duration(s) <OPTIONS>\n", argv[0]);
+       printf("OPTIONS:\n");
+       printf("        [-v] (verbose output)\n");
+       printf("\n");
 }
 
-
 int main(int argc, char **argv)
 {
+       unsigned long long total_loops = 0;
+       unsigned long i_thr;
        void *retval;
        int i;
 
@@ -60,25 +92,53 @@ int main(int argc, char **argv)
                exit(1);
        }
 
-       nr_cpus = atoi(argv[1]);
-       printf("using %d processor(s)\n", nr_cpus);
+       nr_threads = atoi(argv[1]);
+       duration = atol(argv[2]);
+
+       for (i = 3; i < argc; i++) {
+               if (argv[i][0] != '-')
+                       continue;
+               switch (argv[i][1]) {
+               case 'v':
+                       verbose_mode = 1;
+                       break;
+               }
+       }
+
+       printf_verbose("using %d thread(s)\n", nr_threads);
+       printf_verbose("for a duration of %lds\n", duration);
 
-       nr_events = atol(argv[2]);
-       printf("using %ld events per cpu\n", nr_events);
+       pthread_t thread[nr_threads];
+       struct thread_counter thread_counter[nr_threads];
 
-       pthread_t thread[nr_cpus];
-       for (i = 0; i < nr_cpus; i++) {
-               if (pthread_create(&thread[i], NULL, function, NULL)) {
+       for (i = 0; i < nr_threads; i++) {
+               thread_counter[i].nr_loops = 0;
+               if (pthread_create(&thread[i], NULL, function, &thread_counter[i])) {
                        fprintf(stderr, "thread create %d failed\n", i);
                        exit(1);
                }
        }
 
-       for (i = 0; i < nr_cpus; i++) {
+       test_go = 1;
+
+       for (i_thr = 0; i_thr < duration; i_thr++) {
+               sleep(1);
+               if (verbose_mode) {
+                       fwrite(".", sizeof(char), 1, stdout);
+                       fflush(stdout);
+               }
+       }
+       printf_verbose("\n");
+
+       test_stop = 1;
+
+       for (i = 0; i < nr_threads; i++) {
                if (pthread_join(thread[i], &retval)) {
                        fprintf(stderr, "thread join %d failed\n", i);
                        exit(1);
                }
+               total_loops += thread_counter[i].nr_loops;
        }
+       printf("Number of loops: %llu\n", total_loops);
        return 0;
 }