X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=tests%2Fbenchmark%2Fbench.c;h=fcd373ad973ac1edc3d8a8806b6a762a43f2442c;hb=cf914d169665d78854f6d0e1c8ff9d9aff6122d1;hp=1626fd2beeb7998db7a9d755cd942570689a1faf;hpb=c0c0989ab70574e09b2f7e8b48c2da6af664a849;p=lttng-ust.git

diff --git a/tests/benchmark/bench.c b/tests/benchmark/bench.c
index 1626fd2b..fcd373ad 100644
--- a/tests/benchmark/bench.c
+++ b/tests/benchmark/bench.c
@@ -2,6 +2,7 @@
  * 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
  */
@@ -12,51 +13,77 @@
 #include <unistd.h>
 #include <sched.h>
 #include <time.h>
+#include <urcu/compiler.h>
 
 #ifdef TRACING
-#define TRACEPOINT_DEFINE
+#define LTTNG_UST_TRACEPOINT_DEFINE
 #include "ust_tests_benchmark.h"
 #endif
 
-static int nr_cpus;
-static unsigned long nr_events;
+#define printf_verbose(fmt, args...)		\
+	do {					\
+		if (verbose_mode)		\
+			printf(fmt, ## args);	\
+	} while (0)
 
-void do_stuff(void)
-{
-	int v;
-	FILE *file;
+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;
 
+static
+void do_stuff(void)
+{
+	int i;
 #ifdef TRACING
-	tracepoint(ust_tests_benchmark, tpbench, v);
+	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;
 
-	for(i = 0; i < nr_events; i++) {
+	while (!test_go)
+		cmm_barrier();
+
+	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;
 
@@ -65,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]);
 
-	nr_events = atol(argv[2]);
-	printf("using %ld events per cpu\n", nr_events);
+	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);
 
-	pthread_t thread[nr_cpus];
-	for (i = 0; i < nr_cpus; i++) {
-		if (pthread_create(&thread[i], NULL, function, NULL)) {
+	pthread_t thread[nr_threads];
+	struct thread_counter thread_counter[nr_threads];
+
+	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;
 }