common: compile libconsumer, libust-consumer, libkernel-consumer as C++

[lttng-tools.git] / tests / regression / tools / notification / consumer_testpoints.cpp

diff --git a/tests/regression/tools/notification/consumer_testpoints.cpp b/tests/regression/tools/notification/consumer_testpoints.cpp
new file mode 100644 (file)
index 0000000..07c855c
--- /dev/null
+++ b/tests/regression/tools/notification/consumer_testpoints.cpp
@@ -0,0 +1,139 @@
+/*
+ * Copyright (C) 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
+ *
+ * SPDX-License-Identifier: GPL-2.0-only
+ *
+ */
+
+#include <common/compat/getenv.h>
+#include <common/consumer/consumer.h>
+#include <common/pipe.h>
+#include <common/error.h>
+#include <unistd.h>
+#include <stdbool.h>
+#include <lttng/constant.h>
+#include <lttng/lttng-export.h>
+#include <fcntl.h>
+#include <dlfcn.h>
+#include <stdio.h>
+
+static char *pause_pipe_path;
+static struct lttng_pipe *pause_pipe;
+static int *data_consumption_state;
+static enum lttng_consumer_type (*lttng_consumer_get_type)(void);
+
+int lttng_opt_verbose;
+int lttng_opt_mi;
+int lttng_opt_quiet;
+
+static
+void __attribute__((destructor)) pause_pipe_fini(void)
+{
+       int ret;
+
+       if (pause_pipe_path) {
+               ret = unlink(pause_pipe_path);
+               if (ret) {
+                       PERROR("unlink pause pipe");
+               }
+       }
+
+       free(pause_pipe_path);
+       lttng_pipe_destroy(pause_pipe);
+}
+
+/*
+ * We use this testpoint, invoked at the start of the consumerd's data handling
+ * thread to create a named pipe/FIFO which a test application can use to either
+ * pause or resume the consumption of data.
+ */
+extern "C" LTTNG_EXPORT int __testpoint_consumerd_thread_data(void);
+int __testpoint_consumerd_thread_data(void)
+{
+       int ret = 0;
+       const char *pause_pipe_path_prefix, *domain;
+
+       pause_pipe_path_prefix = lttng_secure_getenv(
+                       "CONSUMER_PAUSE_PIPE_PATH");
+       if (!pause_pipe_path_prefix) {
+               ret = -1;
+               goto end;
+       }
+
+       /*
+        * These symbols are exclusive to the consumerd process, hence we can't
+        * rely on their presence in the sessiond. Not looking-up these symbols
+        * dynamically would not allow this shared object to be LD_PRELOAD-ed
+        * when launching the session daemon.
+        */
+       data_consumption_state = (int *) dlsym(NULL, "data_consumption_paused");
+       LTTNG_ASSERT(data_consumption_state);
+       lttng_consumer_get_type = (lttng_consumer_type (*)()) dlsym(NULL, "lttng_consumer_get_type");
+       LTTNG_ASSERT(lttng_consumer_get_type);
+
+       switch (lttng_consumer_get_type()) {
+       case LTTNG_CONSUMER_KERNEL:
+               domain = "kernel";
+               break;
+       case LTTNG_CONSUMER32_UST:
+               domain = "ust32";
+               break;
+       case LTTNG_CONSUMER64_UST:
+               domain = "ust64";
+               break;
+       default:
+               abort();
+       }
+
+       ret = asprintf(&pause_pipe_path, "%s-%s", pause_pipe_path_prefix,
+                       domain);
+       if (ret < 1) {
+               ERR("Failed to allocate pause pipe path");
+               goto end;
+       }
+
+       DBG("Creating pause pipe at %s", pause_pipe_path);
+       pause_pipe = lttng_pipe_named_open(pause_pipe_path,
+                       S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP, O_NONBLOCK);
+       if (!pause_pipe) {
+               ERR("Failed to create pause pipe at %s", pause_pipe_path);
+               ret = -1;
+               goto end;
+       }
+
+       /* Only the read end of the pipe is useful to us. */
+       ret = lttng_pipe_write_close(pause_pipe);
+end:
+       return ret;
+}
+
+extern "C" LTTNG_EXPORT int __testpoint_consumerd_thread_data_poll(void);
+int __testpoint_consumerd_thread_data_poll(void)
+{
+       int ret = 0;
+       uint8_t value;
+       bool value_read = false;
+
+       if (!pause_pipe) {
+               ret = -1;
+               goto end;
+       }
+
+       /* Purge pipe and only consider the freshest value. */
+       do {
+               ret = lttng_pipe_read(pause_pipe, &value, sizeof(value));
+               if (ret == sizeof(value)) {
+                       value_read = true;
+               }
+       } while (ret == sizeof(value));
+
+       ret = (errno == EAGAIN) ? 0 : -errno;
+
+       if (value_read) {
+               *data_consumption_state = !!value;
+               DBG("Message received on pause pipe: %s data consumption",
+                               *data_consumption_state ? "paused" : "resumed");
+       }
+end:
+       return ret;
+}