90b35272b5c06f6f6db2a012e29b89b0a3f7d783
[deliverable/lttng-tools.git] / 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 }
This page took 0.034022 seconds and 5 git commands to generate.