tests/regression/tools/notification/sessiond_testpoints.c

   1 /*
   2  * Copyright (C) 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
   3  * Copyright (C) 2020 Francis Deslauriers <francis.deslauriers@efficios.com>
   4  *
   5  * SPDX-License-Identifier: GPL-2.0-only
   6  *
   7  */
   8
   9 #include <common/compat/getenv.h>
  10 #include <common/consumer/consumer.h>
  11 #include <common/pipe.h>
  12 #include <common/error.h>
  13 #include <unistd.h>
  14 #include <stdbool.h>
  15 #include <lttng/constant.h>
  16 #include <fcntl.h>
  17 #include <dlfcn.h>
  18 #include <assert.h>
  19 #include <stdio.h>
  20
  21 static char *pause_pipe_path;
  22 static struct lttng_pipe *pause_pipe;
  23 static int *trigger_notif_consumption_state;;
  24
  25 int lttng_opt_verbose;
  26 int lttng_opt_mi;
  27 int lttng_opt_quiet;
  28
  29 static
  30 void __attribute__((destructor)) pause_pipe_fini(void)
  31 {
  32         int ret;
  33
  34         if (pause_pipe_path) {
  35                 ret = unlink(pause_pipe_path);
  36                 if (ret) {
  37                         PERROR("unlink pause pipe");
  38                 }
  39         }
  40
  41         free(pause_pipe_path);
  42         lttng_pipe_destroy(pause_pipe);
  43 }
  44
  45 /*
  46  */
  47 int __testpoint_sessiond_thread_notification(void);
  48 int __testpoint_sessiond_thread_notification(void)
  49 {
  50         int ret = 0;
  51         const char *pause_pipe_path_prefix;
  52
  53         pause_pipe_path_prefix = lttng_secure_getenv(
  54                         "TRIGGER_PAUSE_PIPE_PATH");
  55         if (!pause_pipe_path_prefix) {
  56                 ret = -1;
  57                 goto end;
  58         }
  59
  60         trigger_notif_consumption_state = dlsym(NULL, "trigger_consumption_paused");
  61         assert(trigger_notif_consumption_state);
  62
  63         ret = asprintf(&pause_pipe_path, "%s", pause_pipe_path_prefix);
  64         if (ret < 1) {
  65                 ERR("Failed to allocate pause pipe path");
  66                 goto end;
  67         }
  68
  69         DBG("Creating pause pipe at %s", pause_pipe_path);
  70         pause_pipe = lttng_pipe_named_open(pause_pipe_path,
  71                         S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP, O_NONBLOCK);
  72         if (!pause_pipe) {
  73                 ERR("Failed to create pause pipe at %s", pause_pipe_path);
  74                 ret = -1;
  75                 goto end;
  76         }
  77
  78         /* Only the read end of the pipe is useful to us. */
  79         ret = lttng_pipe_write_close(pause_pipe);
  80 end:
  81         return ret;
  82 }
  83
  84 int __testpoint_sessiond_handle_trigger_event_pipe(void);
  85 int __testpoint_sessiond_handle_trigger_event_pipe(void)
  86 {
  87         int ret = 0;
  88         uint8_t value;
  89         bool value_read = false;
  90
  91         if (!pause_pipe) {
  92                 ret = -1;
  93                 goto end;
  94         }
  95
  96         /* Purge pipe and only consider the freshest value. */
  97         do {
  98                 ret = lttng_pipe_read(pause_pipe, &value, sizeof(value));
  99                 if (ret == sizeof(value)) {
 100                         value_read = true;
 101                 }
 102         } while (ret == sizeof(value));
 103
 104         ret = (errno == EAGAIN) ? 0 : -errno;
 105
 106         if (value_read) {
 107                 *trigger_notif_consumption_state = !!value;
 108                 DBG("Message received on pause pipe: %s data consumption",
 109                                 *trigger_notif_consumption_state ? "paused" : "resumed");
 110         }
 111 end:
 112         return ret;
 113 }