2 * Copyright (C) 2019 Simon Marchi <simon.marchi@efficios.com>
4 * SPDX-License-Identifier: LGPL-2.1-only
9 #include <common/error.h>
10 #include <common/macros.h>
11 #include <common/payload-view.h>
12 #include <common/payload.h>
13 #include <common/snapshot.h>
15 #include <lttng/action/action-internal.h>
16 #include <lttng/action/rate-policy-internal.h>
17 #include <lttng/action/rate-policy.h>
18 #include <lttng/action/snapshot-session-internal.h>
19 #include <lttng/action/snapshot-session.h>
20 #include <lttng/snapshot-internal.h>
21 #include <lttng/snapshot.h>
23 #define IS_SNAPSHOT_SESSION_ACTION(action) \
24 (lttng_action_get_type(action) == LTTNG_ACTION_TYPE_SNAPSHOT_SESSION)
26 struct lttng_action_snapshot_session
{
27 struct lttng_action parent
;
33 * When non-NULL, use this custom output when taking the snapshot,
34 * rather than the session's registered snapshot output.
38 struct lttng_snapshot_output
*output
;
39 struct lttng_rate_policy
*policy
;
42 struct lttng_action_snapshot_session_comm
{
43 /* All string lengths include the trailing \0. */
44 uint32_t session_name_len
;
45 uint32_t snapshot_output_len
;
46 uint32_t rate_policy_len
;
49 * Variable data (all strings are null-terminated):
51 * - session name string
52 * - snapshot output object
58 static const struct lttng_rate_policy
*
59 lttng_action_snapshot_session_internal_get_rate_policy(
60 const struct lttng_action
*action
);
62 static struct lttng_action_snapshot_session
*
63 action_snapshot_session_from_action(struct lttng_action
*action
)
68 action
, struct lttng_action_snapshot_session
, parent
);
71 static const struct lttng_action_snapshot_session
*
72 action_snapshot_session_from_action_const(const struct lttng_action
*action
)
77 action
, struct lttng_action_snapshot_session
, parent
);
80 static bool lttng_action_snapshot_session_validate(struct lttng_action
*action
)
83 struct lttng_action_snapshot_session
*action_snapshot_session
;
89 action_snapshot_session
= action_snapshot_session_from_action(action
);
91 /* A non-empty session name is mandatory. */
92 if (!action_snapshot_session
->session_name
||
93 strlen(action_snapshot_session
->session_name
) == 0) {
97 if (action_snapshot_session
->output
&&
98 !lttng_snapshot_output_validate(action_snapshot_session
->output
)) {
107 static bool lttng_action_snapshot_session_is_equal(
108 const struct lttng_action
*_a
, const struct lttng_action
*_b
)
110 bool is_equal
= false;
111 const struct lttng_action_snapshot_session
*a
, *b
;
113 a
= action_snapshot_session_from_action_const(_a
);
114 b
= action_snapshot_session_from_action_const(_b
);
116 /* Action is not valid if this is not true. */
117 assert(a
->session_name
);
118 assert(b
->session_name
);
119 if (strcmp(a
->session_name
, b
->session_name
)) {
123 if (a
->output
&& b
->output
&&
124 !lttng_snapshot_output_is_equal(a
->output
, b
->output
)) {
126 } else if (!!a
->output
!= !!b
->output
) {
130 is_equal
= lttng_rate_policy_is_equal(a
->policy
, b
->policy
);
135 static size_t serialize_strlen(const char *str
)
137 return str
? strlen(str
) + 1 : 0;
140 static int lttng_action_snapshot_session_serialize(
141 struct lttng_action
*action
, struct lttng_payload
*payload
)
143 struct lttng_action_snapshot_session
*action_snapshot_session
;
144 struct lttng_action_snapshot_session_comm comm
= {};
146 size_t size_before_comm
;
151 size_before_comm
= payload
->buffer
.size
;
153 action_snapshot_session
= action_snapshot_session_from_action(action
);
154 comm
.session_name_len
=
155 serialize_strlen(action_snapshot_session
->session_name
);
158 ret
= lttng_dynamic_buffer_append(
159 &payload
->buffer
, &comm
, sizeof(comm
));
164 assert(action_snapshot_session
->session_name
);
165 DBG("Serializing snapshot session action: session-name: %s",
166 action_snapshot_session
->session_name
);
168 /* Add session name. */
169 ret
= lttng_dynamic_buffer_append(&payload
->buffer
,
170 action_snapshot_session
->session_name
,
171 comm
.session_name_len
);
176 /* Serialize the snapshot output object, if any. */
177 if (action_snapshot_session
->output
) {
178 const size_t size_before_output
= payload
->buffer
.size
;
179 struct lttng_action_snapshot_session_comm
*comm_in_payload
;
181 ret
= lttng_snapshot_output_serialize(
182 action_snapshot_session
->output
,
188 comm_in_payload
= (typeof(comm_in_payload
))(
189 payload
->buffer
.data
+ size_before_comm
);
190 /* Adjust action length in header. */
191 comm_in_payload
->snapshot_output_len
=
192 payload
->buffer
.size
- size_before_output
;
195 /* Serialize the rate policy. */
197 const size_t size_before_output
= payload
->buffer
.size
;
198 struct lttng_action_snapshot_session_comm
*comm_in_payload
;
200 ret
= lttng_rate_policy_serialize(
201 action_snapshot_session
->policy
, payload
);
207 comm_in_payload
= (typeof(comm_in_payload
))(
208 payload
->buffer
.data
+ size_before_comm
);
209 /* Adjust rate policy length in header. */
210 comm_in_payload
->rate_policy_len
=
211 payload
->buffer
.size
- size_before_output
;
218 static void lttng_action_snapshot_session_destroy(struct lttng_action
*action
)
220 struct lttng_action_snapshot_session
*action_snapshot_session
;
226 action_snapshot_session
= action_snapshot_session_from_action(action
);
228 free(action_snapshot_session
->session_name
);
229 lttng_snapshot_output_destroy(action_snapshot_session
->output
);
230 lttng_rate_policy_destroy(action_snapshot_session
->policy
);
231 free(action_snapshot_session
);
237 ssize_t
lttng_action_snapshot_session_create_from_payload(
238 struct lttng_payload_view
*view
,
239 struct lttng_action
**p_action
)
241 ssize_t consumed_len
;
242 const char *variable_data
;
243 struct lttng_action
*action
;
244 enum lttng_action_status status
;
245 struct lttng_snapshot_output
*snapshot_output
= NULL
;
246 struct lttng_rate_policy
*policy
= NULL
;
247 const struct lttng_action_snapshot_session_comm
*comm
;
248 const struct lttng_payload_view snapshot_session_comm_view
=
249 lttng_payload_view_from_view(
250 view
, 0, sizeof(*comm
));
252 action
= lttng_action_snapshot_session_create();
257 if (!lttng_payload_view_is_valid(&snapshot_session_comm_view
)) {
258 /* Payload not large enough to contain the header. */
262 comm
= (typeof(comm
)) snapshot_session_comm_view
.buffer
.data
;
263 variable_data
= (const char *) &comm
->data
;
265 consumed_len
= sizeof(struct lttng_action_snapshot_session_comm
);
267 if (!lttng_buffer_view_contains_string(
268 &view
->buffer
, variable_data
, comm
->session_name_len
)) {
272 status
= lttng_action_snapshot_session_set_session_name(
273 action
, variable_data
);
274 if (status
!= LTTNG_ACTION_STATUS_OK
) {
278 variable_data
+= comm
->session_name_len
;
279 consumed_len
+= comm
->session_name_len
;
281 /* If there is a snapshot output object, deserialize it. */
282 if (comm
->snapshot_output_len
> 0) {
283 ssize_t snapshot_output_consumed_len
;
284 enum lttng_action_status action_status
;
285 struct lttng_payload_view snapshot_output_buffer_view
=
286 lttng_payload_view_from_view(view
, consumed_len
,
287 comm
->snapshot_output_len
);
289 if (!lttng_payload_view_is_valid(&snapshot_output_buffer_view
)) {
290 ERR("Failed to create buffer view for snapshot output.");
294 snapshot_output_consumed_len
=
295 lttng_snapshot_output_create_from_payload(
296 &snapshot_output_buffer_view
,
298 if (snapshot_output_consumed_len
!= comm
->snapshot_output_len
) {
299 ERR("Failed to deserialize snapshot output object: "
300 "consumed-len: %zd, expected-len: %" PRIu32
,
301 snapshot_output_consumed_len
,
302 comm
->snapshot_output_len
);
306 action_status
= lttng_action_snapshot_session_set_output(
307 action
, snapshot_output
);
308 if (action_status
!= LTTNG_ACTION_STATUS_OK
) {
312 /* Ownership has been transferred to the action. */
313 snapshot_output
= NULL
;
316 variable_data
+= comm
->snapshot_output_len
;
317 consumed_len
+= comm
->snapshot_output_len
;
320 if (comm
->rate_policy_len
<= 0) {
321 ERR("Rate policy should be present.");
325 ssize_t rate_policy_consumed_len
;
326 struct lttng_payload_view policy_view
=
327 lttng_payload_view_from_view(view
, consumed_len
,
328 comm
->rate_policy_len
);
330 if (!lttng_payload_view_is_valid(&policy_view
)) {
331 ERR("Failed to create buffer view for rate policy.");
335 rate_policy_consumed_len
=
336 lttng_rate_policy_create_from_payload(
337 &policy_view
, &policy
);
338 if (rate_policy_consumed_len
< 0) {
342 if (rate_policy_consumed_len
!= comm
->rate_policy_len
) {
343 ERR("Failed to deserialize rate policy object: "
344 "consumed-len: %zd, expected-len: %" PRIu32
,
345 rate_policy_consumed_len
,
346 comm
->rate_policy_len
);
350 status
= lttng_action_snapshot_session_set_rate_policy(
352 if (status
!= LTTNG_ACTION_STATUS_OK
) {
357 variable_data
+= comm
->rate_policy_len
;
358 consumed_len
+= comm
->rate_policy_len
;
369 lttng_rate_policy_destroy(policy
);
370 lttng_action_snapshot_session_destroy(action
);
371 lttng_snapshot_output_destroy(snapshot_output
);
376 struct lttng_action
*lttng_action_snapshot_session_create(void)
378 struct lttng_action
*action
= NULL
;
379 struct lttng_rate_policy
*policy
= NULL
;
380 enum lttng_action_status status
;
382 /* Create a every N = 1 rate policy. */
383 policy
= lttng_rate_policy_every_n_create(1);
388 action
= zmalloc(sizeof(struct lttng_action_snapshot_session
));
393 lttng_action_init(action
, LTTNG_ACTION_TYPE_SNAPSHOT_SESSION
,
394 lttng_action_snapshot_session_validate
,
395 lttng_action_snapshot_session_serialize
,
396 lttng_action_snapshot_session_is_equal
,
397 lttng_action_snapshot_session_destroy
,
398 lttng_action_snapshot_session_internal_get_rate_policy
);
400 status
= lttng_action_snapshot_session_set_rate_policy(action
, policy
);
401 if (status
!= LTTNG_ACTION_STATUS_OK
) {
408 lttng_rate_policy_destroy(policy
);
412 enum lttng_action_status
lttng_action_snapshot_session_set_session_name(
413 struct lttng_action
*action
, const char *session_name
)
415 struct lttng_action_snapshot_session
*action_snapshot_session
;
416 enum lttng_action_status status
;
418 if (!action
|| !IS_SNAPSHOT_SESSION_ACTION(action
) || !session_name
||
419 strlen(session_name
) == 0) {
420 status
= LTTNG_ACTION_STATUS_INVALID
;
424 action_snapshot_session
= action_snapshot_session_from_action(action
);
426 free(action_snapshot_session
->session_name
);
428 action_snapshot_session
->session_name
= strdup(session_name
);
429 if (!action_snapshot_session
->session_name
) {
430 status
= LTTNG_ACTION_STATUS_ERROR
;
434 status
= LTTNG_ACTION_STATUS_OK
;
439 enum lttng_action_status
lttng_action_snapshot_session_get_session_name(
440 const struct lttng_action
*action
, const char **session_name
)
442 const struct lttng_action_snapshot_session
*action_snapshot_session
;
443 enum lttng_action_status status
;
445 if (!action
|| !IS_SNAPSHOT_SESSION_ACTION(action
) || !session_name
) {
446 status
= LTTNG_ACTION_STATUS_INVALID
;
450 action_snapshot_session
= action_snapshot_session_from_action_const(action
);
452 if (action_snapshot_session
->session_name
) {
453 *session_name
= action_snapshot_session
->session_name
;
454 status
= LTTNG_ACTION_STATUS_OK
;
456 status
= LTTNG_ACTION_STATUS_UNSET
;
464 enum lttng_action_status
lttng_action_snapshot_session_set_output(
465 struct lttng_action
*action
,
466 struct lttng_snapshot_output
*output
)
468 struct lttng_action_snapshot_session
*action_snapshot_session
;
469 enum lttng_action_status status
;
471 if (!action
|| !IS_SNAPSHOT_SESSION_ACTION(action
) || !output
) {
472 status
= LTTNG_ACTION_STATUS_INVALID
;
476 action_snapshot_session
= action_snapshot_session_from_action(action
);
478 lttng_snapshot_output_destroy(action_snapshot_session
->output
);
479 action_snapshot_session
->output
= output
;
481 status
= LTTNG_ACTION_STATUS_OK
;
487 enum lttng_action_status
lttng_action_snapshot_session_get_output(
488 const struct lttng_action
*action
,
489 const struct lttng_snapshot_output
**output
)
491 const struct lttng_action_snapshot_session
*action_snapshot_session
;
492 enum lttng_action_status status
;
494 if (!action
|| !IS_SNAPSHOT_SESSION_ACTION(action
)|| !output
) {
495 status
= LTTNG_ACTION_STATUS_INVALID
;
499 action_snapshot_session
= action_snapshot_session_from_action_const(action
);
501 if (action_snapshot_session
->output
) {
502 *output
= action_snapshot_session
->output
;
503 status
= LTTNG_ACTION_STATUS_OK
;
505 status
= LTTNG_ACTION_STATUS_UNSET
;
512 enum lttng_action_status
lttng_action_snapshot_session_set_rate_policy(
513 struct lttng_action
*action
,
514 const struct lttng_rate_policy
*policy
)
516 enum lttng_action_status status
;
517 struct lttng_action_snapshot_session
*snapshot_session_action
;
518 struct lttng_rate_policy
*copy
= NULL
;
520 if (!action
|| !policy
|| !IS_SNAPSHOT_SESSION_ACTION(action
)) {
521 status
= LTTNG_ACTION_STATUS_INVALID
;
525 copy
= lttng_rate_policy_copy(policy
);
527 status
= LTTNG_ACTION_STATUS_ERROR
;
531 snapshot_session_action
= action_snapshot_session_from_action(action
);
533 /* Free the previous rate policy .*/
534 lttng_rate_policy_destroy(snapshot_session_action
->policy
);
536 /* Assign the policy. */
537 snapshot_session_action
->policy
= copy
;
538 status
= LTTNG_ACTION_STATUS_OK
;
542 lttng_rate_policy_destroy(copy
);
546 enum lttng_action_status
lttng_action_snapshot_session_get_rate_policy(
547 const struct lttng_action
*action
,
548 const struct lttng_rate_policy
**policy
)
550 enum lttng_action_status status
;
551 const struct lttng_action_snapshot_session
*snapshot_session_action
;
553 if (!action
|| !policy
|| !IS_SNAPSHOT_SESSION_ACTION(action
)) {
554 status
= LTTNG_ACTION_STATUS_INVALID
;
558 snapshot_session_action
=
559 action_snapshot_session_from_action_const(action
);
561 *policy
= snapshot_session_action
->policy
;
562 status
= LTTNG_ACTION_STATUS_OK
;
567 static const struct lttng_rate_policy
*
568 lttng_action_snapshot_session_internal_get_rate_policy(
569 const struct lttng_action
*action
)
571 const struct lttng_action_snapshot_session
*_action
;
572 _action
= action_snapshot_session_from_action_const(action
);
574 return _action
->policy
;