Fix: tests: missing _GNU_SOURCE for F_GETPIPE_SZ
[lttng-tools.git] / src / bin / lttng-sessiond / rotate.cpp
1 /*
2 * Copyright (C) 2017 Julien Desfossez <jdesfossez@efficios.com>
3 * Copyright (C) 2018 Jérémie Galarneau <jeremie.galarneau@efficios.com>
4 *
5 * SPDX-License-Identifier: GPL-2.0-only
6 *
7 */
8
9 #define _LGPL_SOURCE
10 #include <lttng/trigger/trigger.h>
11 #include <common/error.hpp>
12 #include <common/config/session-config.hpp>
13 #include <common/defaults.hpp>
14 #include <common/utils.hpp>
15 #include <common/futex.hpp>
16 #include <common/align.hpp>
17 #include <common/time.hpp>
18 #include <common/hashtable/utils.hpp>
19 #include <common/kernel-ctl/kernel-ctl.hpp>
20 #include <common/credentials.hpp>
21 #include <sys/stat.h>
22 #include <time.h>
23 #include <signal.h>
24 #include <inttypes.h>
25
26 #include <lttng/notification/channel-internal.hpp>
27 #include <lttng/rotate-internal.hpp>
28
29 #include "session.hpp"
30 #include "rotate.hpp"
31 #include "rotation-thread.hpp"
32 #include "lttng-sessiond.hpp"
33 #include "health-sessiond.hpp"
34 #include "cmd.hpp"
35 #include "utils.hpp"
36 #include "notification-thread-commands.hpp"
37
38 #include <urcu.h>
39 #include <urcu/list.h>
40 #include <urcu/rculfhash.h>
41
42 int subscribe_session_consumed_size_rotation(struct ltt_session *session, uint64_t size,
43 struct notification_thread_handle *notification_thread_handle)
44 {
45 int ret;
46 enum lttng_condition_status condition_status;
47 enum lttng_notification_channel_status nc_status;
48 struct lttng_action *action;
49 const struct lttng_credentials session_creds = {
50 .uid = LTTNG_OPTIONAL_INIT_VALUE(session->uid),
51 .gid = LTTNG_OPTIONAL_INIT_VALUE(session->gid),
52 };
53
54 session->rotate_condition = lttng_condition_session_consumed_size_create();
55 if (!session->rotate_condition) {
56 ERR("Failed to create session consumed size condition object");
57 ret = -1;
58 goto end;
59 }
60
61 condition_status = lttng_condition_session_consumed_size_set_threshold(
62 session->rotate_condition, size);
63 if (condition_status != LTTNG_CONDITION_STATUS_OK) {
64 ERR("Could not set session consumed size condition threshold (size = %" PRIu64 ")",
65 size);
66 ret = -1;
67 goto end;
68 }
69
70 condition_status =
71 lttng_condition_session_consumed_size_set_session_name(
72 session->rotate_condition, session->name);
73 if (condition_status != LTTNG_CONDITION_STATUS_OK) {
74 ERR("Could not set session consumed size condition session name (name = %s)",
75 session->name);
76 ret = -1;
77 goto end;
78 }
79
80 action = lttng_action_notify_create();
81 if (!action) {
82 ERR("Could not create notify action");
83 ret = -1;
84 goto end;
85 }
86
87 session->rotate_trigger = lttng_trigger_create(session->rotate_condition,
88 action);
89 if (!session->rotate_trigger) {
90 ERR("Could not create size-based rotation trigger");
91 ret = -1;
92 goto end;
93 }
94
95 /* Ensure this trigger is not visible to external users. */
96 lttng_trigger_set_hidden(session->rotate_trigger);
97 lttng_trigger_set_credentials(
98 session->rotate_trigger, &session_creds);
99
100 nc_status = lttng_notification_channel_subscribe(
101 rotate_notification_channel, session->rotate_condition);
102 if (nc_status != LTTNG_NOTIFICATION_CHANNEL_STATUS_OK) {
103 ERR("Could not subscribe to session consumed size notification");
104 ret = -1;
105 goto end;
106 }
107
108 ret = notification_thread_command_register_trigger(
109 notification_thread_handle, session->rotate_trigger,
110 true);
111 if (ret < 0 && ret != -LTTNG_ERR_TRIGGER_EXISTS) {
112 ERR("Register trigger, %s", lttng_strerror(ret));
113 ret = -1;
114 goto end;
115 }
116
117 ret = 0;
118
119 end:
120 return ret;
121 }
122
123 int unsubscribe_session_consumed_size_rotation(struct ltt_session *session,
124 struct notification_thread_handle *notification_thread_handle)
125 {
126 int ret = 0;
127 enum lttng_notification_channel_status status;
128
129 status = lttng_notification_channel_unsubscribe(
130 rotate_notification_channel,
131 session->rotate_condition);
132 if (status != LTTNG_NOTIFICATION_CHANNEL_STATUS_OK) {
133 ERR("Session unsubscribe error: %d", (int) status);
134 ret = -1;
135 goto end;
136 }
137
138 ret = notification_thread_command_unregister_trigger(
139 notification_thread_handle, session->rotate_trigger);
140 if (ret != LTTNG_OK) {
141 ERR("Session unregister trigger error: %d", ret);
142 goto end;
143 }
144
145 ret = 0;
146 end:
147 return ret;
148 }
This page took 0.0335760000000001 seconds and 4 git commands to generate.