Add session rotation ongoing/completed notification commands
[lttng-tools.git] / src / bin / lttng-sessiond / notification-thread-events.c
CommitLineData
ab0ee2ca
JG
1/*
2 * Copyright (C) 2017 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
3 *
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License, version 2 only, as
6 * published by the Free Software Foundation.
7 *
8 * This program is distributed in the hope that it will be useful, but WITHOUT
9 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
11 * more details.
12 *
13 * You should have received a copy of the GNU General Public License along with
14 * this program; if not, write to the Free Software Foundation, Inc., 51
15 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
16 */
17
18#define _LGPL_SOURCE
19#include <urcu.h>
20#include <urcu/rculfhash.h>
21
ab0ee2ca
JG
22#include <common/defaults.h>
23#include <common/error.h>
24#include <common/futex.h>
25#include <common/unix.h>
26#include <common/dynamic-buffer.h>
27#include <common/hashtable/utils.h>
28#include <common/sessiond-comm/sessiond-comm.h>
29#include <common/macros.h>
30#include <lttng/condition/condition.h>
31#include <lttng/action/action.h>
32#include <lttng/notification/notification-internal.h>
33#include <lttng/condition/condition-internal.h>
34#include <lttng/condition/buffer-usage-internal.h>
e8360425 35#include <lttng/condition/session-consumed-size-internal.h>
ab0ee2ca 36#include <lttng/notification/channel-internal.h>
1da26331 37
ab0ee2ca
JG
38#include <time.h>
39#include <unistd.h>
40#include <assert.h>
41#include <inttypes.h>
d53ea4e4 42#include <fcntl.h>
ab0ee2ca 43
1da26331
JG
44#include "notification-thread.h"
45#include "notification-thread-events.h"
46#include "notification-thread-commands.h"
47#include "lttng-sessiond.h"
48#include "kernel.h"
49
ab0ee2ca
JG
50#define CLIENT_POLL_MASK_IN (LPOLLIN | LPOLLERR | LPOLLHUP | LPOLLRDHUP)
51#define CLIENT_POLL_MASK_IN_OUT (CLIENT_POLL_MASK_IN | LPOLLOUT)
52
53struct lttng_trigger_list_element {
54 struct lttng_trigger *trigger;
55 struct cds_list_head node;
56};
57
58struct lttng_channel_trigger_list {
59 struct channel_key channel_key;
60 struct cds_list_head list;
61 struct cds_lfht_node channel_triggers_ht_node;
62};
63
64struct lttng_trigger_ht_element {
65 struct lttng_trigger *trigger;
66 struct cds_lfht_node node;
67};
68
69struct lttng_condition_list_element {
70 struct lttng_condition *condition;
71 struct cds_list_head node;
72};
73
74struct notification_client_list_element {
75 struct notification_client *client;
76 struct cds_list_head node;
77};
78
79struct notification_client_list {
80 struct lttng_trigger *trigger;
81 struct cds_list_head list;
82 struct cds_lfht_node notification_trigger_ht_node;
83};
84
85struct notification_client {
86 int socket;
87 /* Client protocol version. */
88 uint8_t major, minor;
89 uid_t uid;
90 gid_t gid;
91 /*
5332364f 92 * Indicates if the credentials and versions of the client have been
ab0ee2ca
JG
93 * checked.
94 */
95 bool validated;
96 /*
97 * Conditions to which the client's notification channel is subscribed.
98 * List of struct lttng_condition_list_node. The condition member is
99 * owned by the client.
100 */
101 struct cds_list_head condition_list;
102 struct cds_lfht_node client_socket_ht_node;
103 struct {
104 struct {
14fa22f8
JG
105 /*
106 * During the reception of a message, the reception
107 * buffers' "size" is set to contain the current
108 * message's complete payload.
109 */
ab0ee2ca
JG
110 struct lttng_dynamic_buffer buffer;
111 /* Bytes left to receive for the current message. */
112 size_t bytes_to_receive;
113 /* Type of the message being received. */
114 enum lttng_notification_channel_message_type msg_type;
115 /*
116 * Indicates whether or not credentials are expected
117 * from the client.
118 */
01ea340e 119 bool expect_creds;
ab0ee2ca
JG
120 /*
121 * Indicates whether or not credentials were received
122 * from the client.
123 */
124 bool creds_received;
14fa22f8 125 /* Only used during credentials reception. */
ab0ee2ca
JG
126 lttng_sock_cred creds;
127 } inbound;
128 struct {
129 /*
130 * Indicates whether or not a notification addressed to
131 * this client was dropped because a command reply was
132 * already buffered.
133 *
134 * A notification is dropped whenever the buffer is not
135 * empty.
136 */
137 bool dropped_notification;
138 /*
139 * Indicates whether or not a command reply is already
140 * buffered. In this case, it means that the client is
141 * not consuming command replies before emitting a new
142 * one. This could be caused by a protocol error or a
143 * misbehaving/malicious client.
144 */
145 bool queued_command_reply;
146 struct lttng_dynamic_buffer buffer;
147 } outbound;
148 } communication;
149};
150
151struct channel_state_sample {
152 struct channel_key key;
153 struct cds_lfht_node channel_state_ht_node;
154 uint64_t highest_usage;
155 uint64_t lowest_usage;
e8360425 156 uint64_t channel_total_consumed;
ab0ee2ca
JG
157};
158
e4db5ace 159static unsigned long hash_channel_key(struct channel_key *key);
e8360425 160static int evaluate_condition(const struct lttng_condition *condition,
e4db5ace 161 struct lttng_evaluation **evaluation,
e8360425
JD
162 const struct notification_thread_state *state,
163 const struct channel_state_sample *previous_sample,
164 const struct channel_state_sample *latest_sample,
165 uint64_t previous_session_consumed_total,
166 uint64_t latest_session_consumed_total,
167 struct channel_info *channel_info);
e4db5ace
JR
168static
169int send_evaluation_to_clients(struct lttng_trigger *trigger,
170 struct lttng_evaluation *evaluation,
171 struct notification_client_list *client_list,
172 struct notification_thread_state *state,
173 uid_t channel_uid, gid_t channel_gid);
174
8abe313a
JG
175
176static
177void session_info_destroy(void *_data);
178static
179void session_info_get(struct session_info *session_info);
180static
181void session_info_put(struct session_info *session_info);
182static
183struct session_info *session_info_create(const char *name,
184 uid_t uid, gid_t gid);
185static
186void session_info_add_channel(struct session_info *session_info,
187 struct channel_info *channel_info);
188static
189void session_info_remove_channel(struct session_info *session_info,
190 struct channel_info *channel_info);
191
ab0ee2ca
JG
192static
193int match_client(struct cds_lfht_node *node, const void *key)
194{
195 /* This double-cast is intended to supress pointer-to-cast warning. */
196 int socket = (int) (intptr_t) key;
197 struct notification_client *client;
198
199 client = caa_container_of(node, struct notification_client,
200 client_socket_ht_node);
201
202 return !!(client->socket == socket);
203}
204
205static
206int match_channel_trigger_list(struct cds_lfht_node *node, const void *key)
207{
208 struct channel_key *channel_key = (struct channel_key *) key;
209 struct lttng_channel_trigger_list *trigger_list;
210
211 trigger_list = caa_container_of(node, struct lttng_channel_trigger_list,
212 channel_triggers_ht_node);
213
214 return !!((channel_key->key == trigger_list->channel_key.key) &&
215 (channel_key->domain == trigger_list->channel_key.domain));
216}
217
218static
219int match_channel_state_sample(struct cds_lfht_node *node, const void *key)
220{
221 struct channel_key *channel_key = (struct channel_key *) key;
222 struct channel_state_sample *sample;
223
224 sample = caa_container_of(node, struct channel_state_sample,
225 channel_state_ht_node);
226
227 return !!((channel_key->key == sample->key.key) &&
228 (channel_key->domain == sample->key.domain));
229}
230
231static
232int match_channel_info(struct cds_lfht_node *node, const void *key)
233{
234 struct channel_key *channel_key = (struct channel_key *) key;
235 struct channel_info *channel_info;
236
237 channel_info = caa_container_of(node, struct channel_info,
238 channels_ht_node);
239
240 return !!((channel_key->key == channel_info->key.key) &&
241 (channel_key->domain == channel_info->key.domain));
242}
243
244static
245int match_condition(struct cds_lfht_node *node, const void *key)
246{
247 struct lttng_condition *condition_key = (struct lttng_condition *) key;
248 struct lttng_trigger_ht_element *trigger;
249 struct lttng_condition *condition;
250
251 trigger = caa_container_of(node, struct lttng_trigger_ht_element,
252 node);
253 condition = lttng_trigger_get_condition(trigger->trigger);
254 assert(condition);
255
256 return !!lttng_condition_is_equal(condition_key, condition);
257}
258
259static
260int match_client_list(struct cds_lfht_node *node, const void *key)
261{
262 struct lttng_trigger *trigger_key = (struct lttng_trigger *) key;
263 struct notification_client_list *client_list;
264 struct lttng_condition *condition;
265 struct lttng_condition *condition_key = lttng_trigger_get_condition(
266 trigger_key);
267
268 assert(condition_key);
269
270 client_list = caa_container_of(node, struct notification_client_list,
271 notification_trigger_ht_node);
272 condition = lttng_trigger_get_condition(client_list->trigger);
273
274 return !!lttng_condition_is_equal(condition_key, condition);
275}
276
277static
278int match_client_list_condition(struct cds_lfht_node *node, const void *key)
279{
280 struct lttng_condition *condition_key = (struct lttng_condition *) key;
281 struct notification_client_list *client_list;
282 struct lttng_condition *condition;
283
284 assert(condition_key);
285
286 client_list = caa_container_of(node, struct notification_client_list,
287 notification_trigger_ht_node);
288 condition = lttng_trigger_get_condition(client_list->trigger);
289
290 return !!lttng_condition_is_equal(condition_key, condition);
291}
292
293static
294unsigned long lttng_condition_buffer_usage_hash(
295 struct lttng_condition *_condition)
296{
297 unsigned long hash = 0;
298 struct lttng_condition_buffer_usage *condition;
299
300 condition = container_of(_condition,
301 struct lttng_condition_buffer_usage, parent);
302
303 if (condition->session_name) {
304 hash ^= hash_key_str(condition->session_name, lttng_ht_seed);
305 }
306 if (condition->channel_name) {
8f56701f 307 hash ^= hash_key_str(condition->channel_name, lttng_ht_seed);
ab0ee2ca
JG
308 }
309 if (condition->domain.set) {
310 hash ^= hash_key_ulong(
311 (void *) condition->domain.type,
312 lttng_ht_seed);
313 }
314 if (condition->threshold_ratio.set) {
315 uint64_t val;
316
317 val = condition->threshold_ratio.value * (double) UINT32_MAX;
318 hash ^= hash_key_u64(&val, lttng_ht_seed);
6633b0dd 319 } else if (condition->threshold_bytes.set) {
ab0ee2ca
JG
320 uint64_t val;
321
322 val = condition->threshold_bytes.value;
323 hash ^= hash_key_u64(&val, lttng_ht_seed);
324 }
325 return hash;
326}
327
e8360425
JD
328static
329unsigned long lttng_condition_session_consumed_size_hash(
330 struct lttng_condition *_condition)
331{
332 unsigned long hash = 0;
333 struct lttng_condition_session_consumed_size *condition;
334 uint64_t val;
335
336 condition = container_of(_condition,
337 struct lttng_condition_session_consumed_size, parent);
338
339 if (condition->session_name) {
340 hash ^= hash_key_str(condition->session_name, lttng_ht_seed);
341 }
342 val = condition->consumed_threshold_bytes.value;
343 hash ^= hash_key_u64(&val, lttng_ht_seed);
344 return hash;
345}
346
ab0ee2ca
JG
347/*
348 * The lttng_condition hashing code is kept in this file (rather than
349 * condition.c) since it makes use of GPLv2 code (hashtable utils), which we
350 * don't want to link in liblttng-ctl.
351 */
352static
353unsigned long lttng_condition_hash(struct lttng_condition *condition)
354{
355 switch (condition->type) {
356 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_LOW:
357 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_HIGH:
358 return lttng_condition_buffer_usage_hash(condition);
e8360425
JD
359 case LTTNG_CONDITION_TYPE_SESSION_CONSUMED_SIZE:
360 return lttng_condition_session_consumed_size_hash(condition);
ab0ee2ca
JG
361 default:
362 ERR("[notification-thread] Unexpected condition type caught");
363 abort();
364 }
365}
366
e4db5ace
JR
367static
368unsigned long hash_channel_key(struct channel_key *key)
369{
370 unsigned long key_hash = hash_key_u64(&key->key, lttng_ht_seed);
371 unsigned long domain_hash = hash_key_ulong(
372 (void *) (unsigned long) key->domain, lttng_ht_seed);
373
374 return key_hash ^ domain_hash;
375}
376
ab0ee2ca
JG
377static
378void channel_info_destroy(struct channel_info *channel_info)
379{
380 if (!channel_info) {
381 return;
382 }
383
8abe313a
JG
384 if (channel_info->session_info) {
385 session_info_remove_channel(channel_info->session_info,
386 channel_info);
387 session_info_put(channel_info->session_info);
ab0ee2ca 388 }
8abe313a
JG
389 if (channel_info->name) {
390 free(channel_info->name);
ab0ee2ca
JG
391 }
392 free(channel_info);
393}
394
8abe313a 395/* Don't call directly, use the ref-counting mechanism. */
ab0ee2ca 396static
8abe313a 397void session_info_destroy(void *_data)
ab0ee2ca 398{
8abe313a 399 struct session_info *session_info = _data;
3b68d0a3 400 int ret;
ab0ee2ca 401
8abe313a
JG
402 assert(session_info);
403 if (session_info->channel_infos_ht) {
3b68d0a3
JR
404 ret = cds_lfht_destroy(session_info->channel_infos_ht, NULL);
405 if (ret) {
4c3f302b 406 ERR("[notification-thread] Failed to destroy channel information hash table");
3b68d0a3 407 }
8abe313a
JG
408 }
409 free(session_info->name);
410 free(session_info);
411}
ab0ee2ca 412
8abe313a
JG
413static
414void session_info_get(struct session_info *session_info)
415{
416 if (!session_info) {
417 return;
418 }
419 lttng_ref_get(&session_info->ref);
420}
421
422static
423void session_info_put(struct session_info *session_info)
424{
425 if (!session_info) {
426 return;
ab0ee2ca 427 }
8abe313a
JG
428 lttng_ref_put(&session_info->ref);
429}
430
431static
432struct session_info *session_info_create(const char *name, uid_t uid, gid_t gid)
433{
434 struct session_info *session_info;
ab0ee2ca 435
8abe313a 436 assert(name);
ab0ee2ca 437
8abe313a
JG
438 session_info = zmalloc(sizeof(*session_info));
439 if (!session_info) {
440 goto end;
441 }
442 lttng_ref_init(&session_info->ref, session_info_destroy);
443
444 session_info->channel_infos_ht = cds_lfht_new(DEFAULT_HT_SIZE,
445 1, 0, CDS_LFHT_AUTO_RESIZE | CDS_LFHT_ACCOUNTING, NULL);
446 if (!session_info->channel_infos_ht) {
ab0ee2ca
JG
447 goto error;
448 }
8abe313a
JG
449
450 cds_lfht_node_init(&session_info->sessions_ht_node);
451 session_info->name = strdup(name);
452 if (!session_info->name) {
ab0ee2ca
JG
453 goto error;
454 }
8abe313a
JG
455 session_info->uid = uid;
456 session_info->gid = gid;
457end:
458 return session_info;
459error:
460 session_info_put(session_info);
461 return NULL;
462}
463
464static
465void session_info_add_channel(struct session_info *session_info,
466 struct channel_info *channel_info)
467{
468 rcu_read_lock();
469 cds_lfht_add(session_info->channel_infos_ht,
470 hash_channel_key(&channel_info->key),
471 &channel_info->session_info_channels_ht_node);
472 rcu_read_unlock();
473}
474
475static
476void session_info_remove_channel(struct session_info *session_info,
477 struct channel_info *channel_info)
478{
479 rcu_read_lock();
480 cds_lfht_del(session_info->channel_infos_ht,
481 &channel_info->session_info_channels_ht_node);
482 rcu_read_unlock();
483}
484
485static
486struct channel_info *channel_info_create(const char *channel_name,
487 struct channel_key *channel_key, uint64_t channel_capacity,
488 struct session_info *session_info)
489{
490 struct channel_info *channel_info = zmalloc(sizeof(*channel_info));
491
492 if (!channel_info) {
493 goto end;
494 }
495
ab0ee2ca 496 cds_lfht_node_init(&channel_info->channels_ht_node);
8abe313a
JG
497 cds_lfht_node_init(&channel_info->session_info_channels_ht_node);
498 memcpy(&channel_info->key, channel_key, sizeof(*channel_key));
499 channel_info->capacity = channel_capacity;
500
501 channel_info->name = strdup(channel_name);
502 if (!channel_info->name) {
503 goto error;
504 }
505
506 /*
507 * Set the references between session and channel infos:
508 * - channel_info holds a strong reference to session_info
509 * - session_info holds a weak reference to channel_info
510 */
511 session_info_get(session_info);
512 session_info_add_channel(session_info, channel_info);
513 channel_info->session_info = session_info;
ab0ee2ca 514end:
8abe313a 515 return channel_info;
ab0ee2ca 516error:
8abe313a 517 channel_info_destroy(channel_info);
ab0ee2ca
JG
518 return NULL;
519}
520
e4db5ace
JR
521/* This function must be called with the RCU read lock held. */
522static
523int evaluate_condition_for_client(struct lttng_trigger *trigger,
524 struct lttng_condition *condition,
525 struct notification_client *client,
526 struct notification_thread_state *state)
527{
528 int ret;
529 struct cds_lfht_iter iter;
530 struct cds_lfht_node *node;
531 struct channel_info *channel_info = NULL;
532 struct channel_key *channel_key = NULL;
533 struct channel_state_sample *last_sample = NULL;
534 struct lttng_channel_trigger_list *channel_trigger_list = NULL;
535 struct lttng_evaluation *evaluation = NULL;
536 struct notification_client_list client_list = { 0 };
537 struct notification_client_list_element client_list_element = { 0 };
538
539 assert(trigger);
540 assert(condition);
541 assert(client);
542 assert(state);
543
544 /* Find the channel associated with the trigger. */
545 cds_lfht_for_each_entry(state->channel_triggers_ht, &iter,
546 channel_trigger_list , channel_triggers_ht_node) {
547 struct lttng_trigger_list_element *element;
548
549 cds_list_for_each_entry(element, &channel_trigger_list->list, node) {
550 struct lttng_condition *current_condition =
551 lttng_trigger_get_condition(
552 element->trigger);
553
554 assert(current_condition);
555 if (!lttng_condition_is_equal(condition,
2ae99f0b 556 current_condition)) {
e4db5ace
JR
557 continue;
558 }
559
560 /* Found the trigger, save the channel key. */
561 channel_key = &channel_trigger_list->channel_key;
562 break;
563 }
564 if (channel_key) {
565 /* The channel key was found stop iteration. */
566 break;
567 }
568 }
569
570 if (!channel_key){
571 /* No channel found; normal exit. */
572 DBG("[notification-thread] No channel associated with newly subscribed-to condition");
573 ret = 0;
574 goto end;
575 }
576
577 /* Fetch channel info for the matching channel. */
578 cds_lfht_lookup(state->channels_ht,
579 hash_channel_key(channel_key),
580 match_channel_info,
581 channel_key,
582 &iter);
583 node = cds_lfht_iter_get_node(&iter);
584 assert(node);
585 channel_info = caa_container_of(node, struct channel_info,
586 channels_ht_node);
587
588 /* Retrieve the channel's last sample, if it exists. */
589 cds_lfht_lookup(state->channel_state_ht,
590 hash_channel_key(channel_key),
591 match_channel_state_sample,
592 channel_key,
593 &iter);
594 node = cds_lfht_iter_get_node(&iter);
595 if (node) {
596 last_sample = caa_container_of(node,
597 struct channel_state_sample,
598 channel_state_ht_node);
599 } else {
600 /* Nothing to evaluate, no sample was ever taken. Normal exit */
601 DBG("[notification-thread] No channel sample associated with newly subscribed-to condition");
602 ret = 0;
603 goto end;
604 }
605
e8360425
JD
606 ret = evaluate_condition(condition, &evaluation, state,
607 NULL, last_sample,
608 0, channel_info->session_info->consumed_data_size,
609 channel_info);
e4db5ace 610 if (ret) {
066a3c82 611 WARN("[notification-thread] Fatal error occurred while evaluating a newly subscribed-to condition");
e4db5ace
JR
612 goto end;
613 }
614
615 if (!evaluation) {
616 /* Evaluation yielded nothing. Normal exit. */
617 DBG("[notification-thread] Newly subscribed-to condition evaluated to false, nothing to report to client");
618 ret = 0;
619 goto end;
620 }
621
622 /*
623 * Create a temporary client list with the client currently
624 * subscribing.
625 */
626 cds_lfht_node_init(&client_list.notification_trigger_ht_node);
627 CDS_INIT_LIST_HEAD(&client_list.list);
628 client_list.trigger = trigger;
629
630 CDS_INIT_LIST_HEAD(&client_list_element.node);
631 client_list_element.client = client;
632 cds_list_add(&client_list_element.node, &client_list.list);
633
634 /* Send evaluation result to the newly-subscribed client. */
635 DBG("[notification-thread] Newly subscribed-to condition evaluated to true, notifying client");
636 ret = send_evaluation_to_clients(trigger, evaluation, &client_list,
8abe313a
JG
637 state, channel_info->session_info->uid,
638 channel_info->session_info->gid);
e4db5ace
JR
639
640end:
641 return ret;
642}
643
ab0ee2ca
JG
644static
645int notification_thread_client_subscribe(struct notification_client *client,
646 struct lttng_condition *condition,
647 struct notification_thread_state *state,
648 enum lttng_notification_channel_status *_status)
649{
650 int ret = 0;
651 struct cds_lfht_iter iter;
652 struct cds_lfht_node *node;
653 struct notification_client_list *client_list;
654 struct lttng_condition_list_element *condition_list_element = NULL;
655 struct notification_client_list_element *client_list_element = NULL;
656 enum lttng_notification_channel_status status =
657 LTTNG_NOTIFICATION_CHANNEL_STATUS_OK;
658
659 /*
660 * Ensure that the client has not already subscribed to this condition
661 * before.
662 */
663 cds_list_for_each_entry(condition_list_element, &client->condition_list, node) {
664 if (lttng_condition_is_equal(condition_list_element->condition,
665 condition)) {
666 status = LTTNG_NOTIFICATION_CHANNEL_STATUS_ALREADY_SUBSCRIBED;
667 goto end;
668 }
669 }
670
671 condition_list_element = zmalloc(sizeof(*condition_list_element));
672 if (!condition_list_element) {
673 ret = -1;
674 goto error;
675 }
676 client_list_element = zmalloc(sizeof(*client_list_element));
677 if (!client_list_element) {
678 ret = -1;
679 goto error;
680 }
681
682 rcu_read_lock();
683
684 /*
685 * Add the newly-subscribed condition to the client's subscription list.
686 */
687 CDS_INIT_LIST_HEAD(&condition_list_element->node);
688 condition_list_element->condition = condition;
689 cds_list_add(&condition_list_element->node, &client->condition_list);
690
ab0ee2ca
JG
691 cds_lfht_lookup(state->notification_trigger_clients_ht,
692 lttng_condition_hash(condition),
693 match_client_list_condition,
694 condition,
695 &iter);
696 node = cds_lfht_iter_get_node(&iter);
697 if (!node) {
2ae99f0b
JG
698 /*
699 * No notification-emiting trigger registered with this
700 * condition. We don't evaluate the condition right away
701 * since this trigger is not registered yet.
702 */
4fb43b68 703 free(client_list_element);
ab0ee2ca
JG
704 goto end_unlock;
705 }
706
707 client_list = caa_container_of(node, struct notification_client_list,
708 notification_trigger_ht_node);
2ae99f0b
JG
709 /*
710 * The condition to which the client just subscribed is evaluated
711 * at this point so that conditions that are already TRUE result
712 * in a notification being sent out.
713 */
e4db5ace
JR
714 if (evaluate_condition_for_client(client_list->trigger, condition,
715 client, state)) {
716 WARN("[notification-thread] Evaluation of a condition on client subscription failed, aborting.");
717 ret = -1;
4bd5b4af 718 free(client_list_element);
e4db5ace
JR
719 goto end_unlock;
720 }
721
722 /*
723 * Add the client to the list of clients interested in a given trigger
724 * if a "notification" trigger with a corresponding condition was
725 * added prior.
726 */
ab0ee2ca
JG
727 client_list_element->client = client;
728 CDS_INIT_LIST_HEAD(&client_list_element->node);
729 cds_list_add(&client_list_element->node, &client_list->list);
730end_unlock:
731 rcu_read_unlock();
732end:
733 if (_status) {
734 *_status = status;
735 }
736 return ret;
737error:
738 free(condition_list_element);
739 free(client_list_element);
740 return ret;
741}
742
743static
744int notification_thread_client_unsubscribe(
745 struct notification_client *client,
746 struct lttng_condition *condition,
747 struct notification_thread_state *state,
748 enum lttng_notification_channel_status *_status)
749{
750 struct cds_lfht_iter iter;
751 struct cds_lfht_node *node;
752 struct notification_client_list *client_list;
753 struct lttng_condition_list_element *condition_list_element,
754 *condition_tmp;
755 struct notification_client_list_element *client_list_element,
756 *client_tmp;
757 bool condition_found = false;
758 enum lttng_notification_channel_status status =
759 LTTNG_NOTIFICATION_CHANNEL_STATUS_OK;
760
761 /* Remove the condition from the client's condition list. */
762 cds_list_for_each_entry_safe(condition_list_element, condition_tmp,
763 &client->condition_list, node) {
764 if (!lttng_condition_is_equal(condition_list_element->condition,
765 condition)) {
766 continue;
767 }
768
769 cds_list_del(&condition_list_element->node);
770 /*
771 * The caller may be iterating on the client's conditions to
772 * tear down a client's connection. In this case, the condition
773 * will be destroyed at the end.
774 */
775 if (condition != condition_list_element->condition) {
776 lttng_condition_destroy(
777 condition_list_element->condition);
778 }
779 free(condition_list_element);
780 condition_found = true;
781 break;
782 }
783
784 if (!condition_found) {
785 status = LTTNG_NOTIFICATION_CHANNEL_STATUS_UNKNOWN_CONDITION;
786 goto end;
787 }
788
789 /*
790 * Remove the client from the list of clients interested the trigger
791 * matching the condition.
792 */
793 rcu_read_lock();
794 cds_lfht_lookup(state->notification_trigger_clients_ht,
795 lttng_condition_hash(condition),
796 match_client_list_condition,
797 condition,
798 &iter);
799 node = cds_lfht_iter_get_node(&iter);
800 if (!node) {
801 goto end_unlock;
802 }
803
804 client_list = caa_container_of(node, struct notification_client_list,
805 notification_trigger_ht_node);
806 cds_list_for_each_entry_safe(client_list_element, client_tmp,
807 &client_list->list, node) {
808 if (client_list_element->client->socket != client->socket) {
809 continue;
810 }
811 cds_list_del(&client_list_element->node);
812 free(client_list_element);
813 break;
814 }
815end_unlock:
816 rcu_read_unlock();
817end:
818 lttng_condition_destroy(condition);
819 if (_status) {
820 *_status = status;
821 }
822 return 0;
823}
824
825static
826void notification_client_destroy(struct notification_client *client,
827 struct notification_thread_state *state)
828{
829 struct lttng_condition_list_element *condition_list_element, *tmp;
830
831 if (!client) {
832 return;
833 }
834
835 /* Release all conditions to which the client was subscribed. */
836 cds_list_for_each_entry_safe(condition_list_element, tmp,
837 &client->condition_list, node) {
838 (void) notification_thread_client_unsubscribe(client,
839 condition_list_element->condition, state, NULL);
840 }
841
842 if (client->socket >= 0) {
843 (void) lttcomm_close_unix_sock(client->socket);
844 }
845 lttng_dynamic_buffer_reset(&client->communication.inbound.buffer);
846 lttng_dynamic_buffer_reset(&client->communication.outbound.buffer);
847 free(client);
848}
849
850/*
851 * Call with rcu_read_lock held (and hold for the lifetime of the returned
852 * client pointer).
853 */
854static
855struct notification_client *get_client_from_socket(int socket,
856 struct notification_thread_state *state)
857{
858 struct cds_lfht_iter iter;
859 struct cds_lfht_node *node;
860 struct notification_client *client = NULL;
861
862 cds_lfht_lookup(state->client_socket_ht,
863 hash_key_ulong((void *) (unsigned long) socket, lttng_ht_seed),
864 match_client,
865 (void *) (unsigned long) socket,
866 &iter);
867 node = cds_lfht_iter_get_node(&iter);
868 if (!node) {
869 goto end;
870 }
871
872 client = caa_container_of(node, struct notification_client,
873 client_socket_ht_node);
874end:
875 return client;
876}
877
878static
e8360425
JD
879bool buffer_usage_condition_applies_to_channel(
880 struct lttng_condition *condition,
8abe313a 881 struct channel_info *channel_info)
ab0ee2ca
JG
882{
883 enum lttng_condition_status status;
e8360425
JD
884 enum lttng_domain_type condition_domain;
885 const char *condition_session_name = NULL;
886 const char *condition_channel_name = NULL;
ab0ee2ca 887
e8360425
JD
888 status = lttng_condition_buffer_usage_get_domain_type(condition,
889 &condition_domain);
890 assert(status == LTTNG_CONDITION_STATUS_OK);
891 if (channel_info->key.domain != condition_domain) {
ab0ee2ca
JG
892 goto fail;
893 }
894
e8360425
JD
895 status = lttng_condition_buffer_usage_get_session_name(
896 condition, &condition_session_name);
897 assert((status == LTTNG_CONDITION_STATUS_OK) && condition_session_name);
898
899 status = lttng_condition_buffer_usage_get_channel_name(
900 condition, &condition_channel_name);
901 assert((status == LTTNG_CONDITION_STATUS_OK) && condition_channel_name);
902
903 if (strcmp(channel_info->session_info->name, condition_session_name)) {
904 goto fail;
905 }
906 if (strcmp(channel_info->name, condition_channel_name)) {
ab0ee2ca
JG
907 goto fail;
908 }
909
e8360425
JD
910 return true;
911fail:
912 return false;
913}
914
915static
916bool session_consumed_size_condition_applies_to_channel(
917 struct lttng_condition *condition,
918 struct channel_info *channel_info)
919{
920 enum lttng_condition_status status;
921 const char *condition_session_name = NULL;
922
923 status = lttng_condition_session_consumed_size_get_session_name(
924 condition, &condition_session_name);
925 assert((status == LTTNG_CONDITION_STATUS_OK) && condition_session_name);
926
927 if (strcmp(channel_info->session_info->name, condition_session_name)) {
ab0ee2ca
JG
928 goto fail;
929 }
930
e8360425
JD
931 return true;
932fail:
933 return false;
934}
ab0ee2ca 935
e8360425
JD
936static
937bool trigger_applies_to_channel(struct lttng_trigger *trigger,
938 struct channel_info *channel_info)
939{
940 struct lttng_condition *condition;
941 bool trigger_applies;
ab0ee2ca 942
e8360425
JD
943 condition = lttng_trigger_get_condition(trigger);
944 if (!condition) {
ab0ee2ca
JG
945 goto fail;
946 }
e8360425
JD
947
948 switch (lttng_condition_get_type(condition)) {
949 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_LOW:
950 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_HIGH:
951 trigger_applies = buffer_usage_condition_applies_to_channel(
952 condition, channel_info);
953 break;
954 case LTTNG_CONDITION_TYPE_SESSION_CONSUMED_SIZE:
955 trigger_applies = session_consumed_size_condition_applies_to_channel(
956 condition, channel_info);
957 break;
958 default:
ab0ee2ca
JG
959 goto fail;
960 }
961
e8360425 962 return trigger_applies;
ab0ee2ca
JG
963fail:
964 return false;
965}
966
967static
968bool trigger_applies_to_client(struct lttng_trigger *trigger,
969 struct notification_client *client)
970{
971 bool applies = false;
972 struct lttng_condition_list_element *condition_list_element;
973
974 cds_list_for_each_entry(condition_list_element, &client->condition_list,
975 node) {
976 applies = lttng_condition_is_equal(
977 condition_list_element->condition,
978 lttng_trigger_get_condition(trigger));
979 if (applies) {
980 break;
981 }
982 }
983 return applies;
984}
985
8abe313a
JG
986static
987int match_session(struct cds_lfht_node *node, const void *key)
988{
989 const char *name = key;
990 struct session_info *session_info = caa_container_of(
991 node, struct session_info, sessions_ht_node);
992
993 return !strcmp(session_info->name, name);
994}
995
996static
997struct session_info *find_or_create_session_info(
998 struct notification_thread_state *state,
999 const char *name, uid_t uid, gid_t gid)
1000{
1001 struct session_info *session = NULL;
1002 struct cds_lfht_node *node;
1003 struct cds_lfht_iter iter;
1004
1005 rcu_read_lock();
1006 cds_lfht_lookup(state->sessions_ht,
1007 hash_key_str(name, lttng_ht_seed),
1008 match_session,
1009 name,
1010 &iter);
1011 node = cds_lfht_iter_get_node(&iter);
1012 if (node) {
1013 DBG("[notification-thread] Found session info of session \"%s\" (uid = %i, gid = %i)",
1014 name, uid, gid);
1015 session = caa_container_of(node, struct session_info,
1016 sessions_ht_node);
1017 assert(session->uid == uid);
1018 assert(session->gid == gid);
1019 goto end;
1020 }
1021
1022 session = session_info_create(name, uid, gid);
1023 if (!session) {
1024 ERR("[notification-thread] Failed to allocation session info for session \"%s\" (uid = %i, gid = %i)",
1025 name, uid, gid);
1026 goto end;
1027 }
1028end:
1029 rcu_read_unlock();
1030 return session;
1031}
1032
ab0ee2ca
JG
1033static
1034int handle_notification_thread_command_add_channel(
8abe313a
JG
1035 struct notification_thread_state *state,
1036 const char *session_name, uid_t session_uid, gid_t session_gid,
1037 const char *channel_name, enum lttng_domain_type channel_domain,
1038 uint64_t channel_key_int, uint64_t channel_capacity,
1039 enum lttng_error_code *cmd_result)
ab0ee2ca
JG
1040{
1041 struct cds_list_head trigger_list;
8abe313a
JG
1042 struct channel_info *new_channel_info = NULL;
1043 struct channel_key channel_key = {
1044 .key = channel_key_int,
1045 .domain = channel_domain,
1046 };
ab0ee2ca
JG
1047 struct lttng_channel_trigger_list *channel_trigger_list = NULL;
1048 struct lttng_trigger_ht_element *trigger_ht_element = NULL;
1049 int trigger_count = 0;
1050 struct cds_lfht_iter iter;
8abe313a 1051 struct session_info *session_info = NULL;
ab0ee2ca
JG
1052
1053 DBG("[notification-thread] Adding channel %s from session %s, channel key = %" PRIu64 " in %s domain",
8abe313a
JG
1054 channel_name, session_name, channel_key_int,
1055 channel_domain == LTTNG_DOMAIN_KERNEL ? "kernel" : "user space");
ab0ee2ca
JG
1056
1057 CDS_INIT_LIST_HEAD(&trigger_list);
1058
8abe313a
JG
1059 session_info = find_or_create_session_info(state, session_name,
1060 session_uid, session_gid);
1061 if (!session_info) {
1062 /* Allocation error or an internal error occured. */
ab0ee2ca
JG
1063 goto error;
1064 }
1065
8abe313a
JG
1066 new_channel_info = channel_info_create(channel_name, &channel_key,
1067 channel_capacity, session_info);
1068 if (!new_channel_info) {
1069 goto error;
1070 }
ab0ee2ca
JG
1071
1072 /* Build a list of all triggers applying to the new channel. */
1073 cds_lfht_for_each_entry(state->triggers_ht, &iter, trigger_ht_element,
1074 node) {
1075 struct lttng_trigger_list_element *new_element;
1076
1077 if (!trigger_applies_to_channel(trigger_ht_element->trigger,
8abe313a 1078 new_channel_info)) {
ab0ee2ca
JG
1079 continue;
1080 }
1081
1082 new_element = zmalloc(sizeof(*new_element));
1083 if (!new_element) {
1084 goto error;
1085 }
1086 CDS_INIT_LIST_HEAD(&new_element->node);
1087 new_element->trigger = trigger_ht_element->trigger;
1088 cds_list_add(&new_element->node, &trigger_list);
1089 trigger_count++;
1090 }
1091
1092 DBG("[notification-thread] Found %i triggers that apply to newly added channel",
1093 trigger_count);
1094 channel_trigger_list = zmalloc(sizeof(*channel_trigger_list));
1095 if (!channel_trigger_list) {
1096 goto error;
1097 }
8abe313a 1098 channel_trigger_list->channel_key = new_channel_info->key;
ab0ee2ca
JG
1099 CDS_INIT_LIST_HEAD(&channel_trigger_list->list);
1100 cds_lfht_node_init(&channel_trigger_list->channel_triggers_ht_node);
1101 cds_list_splice(&trigger_list, &channel_trigger_list->list);
1102
1103 rcu_read_lock();
1104 /* Add channel to the channel_ht which owns the channel_infos. */
1105 cds_lfht_add(state->channels_ht,
8abe313a 1106 hash_channel_key(&new_channel_info->key),
ab0ee2ca
JG
1107 &new_channel_info->channels_ht_node);
1108 /*
1109 * Add the list of triggers associated with this channel to the
1110 * channel_triggers_ht.
1111 */
1112 cds_lfht_add(state->channel_triggers_ht,
8abe313a 1113 hash_channel_key(&new_channel_info->key),
ab0ee2ca
JG
1114 &channel_trigger_list->channel_triggers_ht_node);
1115 rcu_read_unlock();
1116 *cmd_result = LTTNG_OK;
1117 return 0;
1118error:
ab0ee2ca 1119 channel_info_destroy(new_channel_info);
8abe313a 1120 session_info_put(session_info);
ab0ee2ca
JG
1121 return 1;
1122}
1123
1124static
1125int handle_notification_thread_command_remove_channel(
1126 struct notification_thread_state *state,
1127 uint64_t channel_key, enum lttng_domain_type domain,
1128 enum lttng_error_code *cmd_result)
1129{
1130 struct cds_lfht_node *node;
1131 struct cds_lfht_iter iter;
1132 struct lttng_channel_trigger_list *trigger_list;
1133 struct lttng_trigger_list_element *trigger_list_element, *tmp;
1134 struct channel_key key = { .key = channel_key, .domain = domain };
1135 struct channel_info *channel_info;
1136
1137 DBG("[notification-thread] Removing channel key = %" PRIu64 " in %s domain",
1138 channel_key, domain == LTTNG_DOMAIN_KERNEL ? "kernel" : "user space");
1139
1140 rcu_read_lock();
1141
1142 cds_lfht_lookup(state->channel_triggers_ht,
1143 hash_channel_key(&key),
1144 match_channel_trigger_list,
1145 &key,
1146 &iter);
1147 node = cds_lfht_iter_get_node(&iter);
1148 /*
1149 * There is a severe internal error if we are being asked to remove a
1150 * channel that doesn't exist.
1151 */
1152 if (!node) {
1153 ERR("[notification-thread] Channel being removed is unknown to the notification thread");
1154 goto end;
1155 }
1156
1157 /* Free the list of triggers associated with this channel. */
1158 trigger_list = caa_container_of(node, struct lttng_channel_trigger_list,
1159 channel_triggers_ht_node);
1160 cds_list_for_each_entry_safe(trigger_list_element, tmp,
1161 &trigger_list->list, node) {
1162 cds_list_del(&trigger_list_element->node);
1163 free(trigger_list_element);
1164 }
1165 cds_lfht_del(state->channel_triggers_ht, node);
1166 free(trigger_list);
1167
1168 /* Free sampled channel state. */
1169 cds_lfht_lookup(state->channel_state_ht,
1170 hash_channel_key(&key),
1171 match_channel_state_sample,
1172 &key,
1173 &iter);
1174 node = cds_lfht_iter_get_node(&iter);
1175 /*
1176 * This is expected to be NULL if the channel is destroyed before we
1177 * received a sample.
1178 */
1179 if (node) {
1180 struct channel_state_sample *sample = caa_container_of(node,
1181 struct channel_state_sample,
1182 channel_state_ht_node);
1183
1184 cds_lfht_del(state->channel_state_ht, node);
1185 free(sample);
1186 }
1187
1188 /* Remove the channel from the channels_ht and free it. */
1189 cds_lfht_lookup(state->channels_ht,
1190 hash_channel_key(&key),
1191 match_channel_info,
1192 &key,
1193 &iter);
1194 node = cds_lfht_iter_get_node(&iter);
1195 assert(node);
1196 channel_info = caa_container_of(node, struct channel_info,
1197 channels_ht_node);
1198 cds_lfht_del(state->channels_ht, node);
1199 channel_info_destroy(channel_info);
1200end:
1201 rcu_read_unlock();
1202 *cmd_result = LTTNG_OK;
1203 return 0;
1204}
1205
1da26331
JG
1206static
1207int condition_is_supported(struct lttng_condition *condition)
1208{
1209 int ret;
1210
1211 switch (lttng_condition_get_type(condition)) {
1212 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_LOW:
1213 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_HIGH:
1214 {
1215 enum lttng_domain_type domain;
1216
1217 ret = lttng_condition_buffer_usage_get_domain_type(condition,
1218 &domain);
1219 if (ret) {
1220 ret = -1;
1221 goto end;
1222 }
1223
1224 if (domain != LTTNG_DOMAIN_KERNEL) {
1225 ret = 1;
1226 goto end;
1227 }
1228
1229 /*
1230 * Older kernel tracers don't expose the API to monitor their
1231 * buffers. Therefore, we reject triggers that require that
1232 * mechanism to be available to be evaluated.
1233 */
1234 ret = kernel_supports_ring_buffer_snapshot_sample_positions(
1235 kernel_tracer_fd);
1236 break;
1237 }
1238 default:
1239 ret = 1;
1240 }
1241end:
1242 return ret;
1243}
1244
ab0ee2ca
JG
1245/*
1246 * FIXME A client's credentials are not checked when registering a trigger, nor
1247 * are they stored alongside with the trigger.
1248 *
1da26331 1249 * The effects of this are benign since:
ab0ee2ca
JG
1250 * - The client will succeed in registering the trigger, as it is valid,
1251 * - The trigger will, internally, be bound to the channel,
1252 * - The notifications will not be sent since the client's credentials
1253 * are checked against the channel at that moment.
1da26331
JG
1254 *
1255 * If this function returns a non-zero value, it means something is
50ca7858 1256 * fundamentally broken and the whole subsystem/thread will be torn down.
1da26331
JG
1257 *
1258 * If a non-fatal error occurs, just set the cmd_result to the appropriate
1259 * error code.
ab0ee2ca
JG
1260 */
1261static
1262int handle_notification_thread_command_register_trigger(
8abe313a
JG
1263 struct notification_thread_state *state,
1264 struct lttng_trigger *trigger,
1265 enum lttng_error_code *cmd_result)
ab0ee2ca
JG
1266{
1267 int ret = 0;
1268 struct lttng_condition *condition;
1269 struct notification_client *client;
1270 struct notification_client_list *client_list = NULL;
1271 struct lttng_trigger_ht_element *trigger_ht_element = NULL;
1272 struct notification_client_list_element *client_list_element, *tmp;
1273 struct cds_lfht_node *node;
1274 struct cds_lfht_iter iter;
1275 struct channel_info *channel;
1276 bool free_trigger = true;
1277
1278 rcu_read_lock();
1279
1280 condition = lttng_trigger_get_condition(trigger);
1da26331
JG
1281 assert(condition);
1282
1283 ret = condition_is_supported(condition);
1284 if (ret < 0) {
1285 goto error;
1286 } else if (ret == 0) {
1287 *cmd_result = LTTNG_ERR_NOT_SUPPORTED;
1288 goto error;
1289 } else {
1290 /* Feature is supported, continue. */
1291 ret = 0;
1292 }
1293
ab0ee2ca
JG
1294 trigger_ht_element = zmalloc(sizeof(*trigger_ht_element));
1295 if (!trigger_ht_element) {
1296 ret = -1;
1297 goto error;
1298 }
1299
1300 /* Add trigger to the trigger_ht. */
1301 cds_lfht_node_init(&trigger_ht_element->node);
1302 trigger_ht_element->trigger = trigger;
1303
1304 node = cds_lfht_add_unique(state->triggers_ht,
1305 lttng_condition_hash(condition),
1306 match_condition,
1307 condition,
1308 &trigger_ht_element->node);
1309 if (node != &trigger_ht_element->node) {
1310 /* Not a fatal error, simply report it to the client. */
1311 *cmd_result = LTTNG_ERR_TRIGGER_EXISTS;
1312 goto error_free_ht_element;
1313 }
1314
1315 /*
1316 * Ownership of the trigger and of its wrapper was transfered to
1317 * the triggers_ht.
1318 */
1319 trigger_ht_element = NULL;
1320 free_trigger = false;
1321
1322 /*
1323 * The rest only applies to triggers that have a "notify" action.
1324 * It is not skipped as this is the only action type currently
1325 * supported.
1326 */
1327 client_list = zmalloc(sizeof(*client_list));
1328 if (!client_list) {
1329 ret = -1;
1330 goto error_free_ht_element;
1331 }
1332 cds_lfht_node_init(&client_list->notification_trigger_ht_node);
1333 CDS_INIT_LIST_HEAD(&client_list->list);
1334 client_list->trigger = trigger;
1335
1336 /* Build a list of clients to which this new trigger applies. */
1337 cds_lfht_for_each_entry(state->client_socket_ht, &iter, client,
1338 client_socket_ht_node) {
1339 if (!trigger_applies_to_client(trigger, client)) {
1340 continue;
1341 }
1342
1343 client_list_element = zmalloc(sizeof(*client_list_element));
1344 if (!client_list_element) {
1345 ret = -1;
1346 goto error_free_client_list;
1347 }
1348 CDS_INIT_LIST_HEAD(&client_list_element->node);
1349 client_list_element->client = client;
1350 cds_list_add(&client_list_element->node, &client_list->list);
1351 }
1352
1353 cds_lfht_add(state->notification_trigger_clients_ht,
1354 lttng_condition_hash(condition),
1355 &client_list->notification_trigger_ht_node);
ab0ee2ca
JG
1356
1357 /*
1358 * Add the trigger to list of triggers bound to the channels currently
1359 * known.
1360 */
1361 cds_lfht_for_each_entry(state->channels_ht, &iter, channel,
1362 channels_ht_node) {
1363 struct lttng_trigger_list_element *trigger_list_element;
1364 struct lttng_channel_trigger_list *trigger_list;
1365
1366 if (!trigger_applies_to_channel(trigger, channel)) {
1367 continue;
1368 }
1369
1370 cds_lfht_lookup(state->channel_triggers_ht,
1371 hash_channel_key(&channel->key),
1372 match_channel_trigger_list,
1373 &channel->key,
1374 &iter);
1375 node = cds_lfht_iter_get_node(&iter);
1376 assert(node);
ab0ee2ca
JG
1377 trigger_list = caa_container_of(node,
1378 struct lttng_channel_trigger_list,
1379 channel_triggers_ht_node);
1380
1381 trigger_list_element = zmalloc(sizeof(*trigger_list_element));
1382 if (!trigger_list_element) {
1383 ret = -1;
1384 goto error_free_client_list;
1385 }
1386 CDS_INIT_LIST_HEAD(&trigger_list_element->node);
1387 trigger_list_element->trigger = trigger;
1388 cds_list_add(&trigger_list_element->node, &trigger_list->list);
ab0ee2ca
JG
1389 }
1390
2ae99f0b
JG
1391 /*
1392 * Since there is nothing preventing clients from subscribing to a
1393 * condition before the corresponding trigger is registered, we have
1394 * to evaluate this new condition right away.
1395 *
1396 * At some point, we were waiting for the next "evaluation" (e.g. on
1397 * reception of a channel sample) to evaluate this new condition, but
1398 * that was broken.
1399 *
1400 * The reason it was broken is that waiting for the next sample
1401 * does not allow us to properly handle transitions for edge-triggered
1402 * conditions.
1403 *
1404 * Consider this example: when we handle a new channel sample, we
1405 * evaluate each conditions twice: once with the previous state, and
1406 * again with the newest state. We then use those two results to
1407 * determine whether a state change happened: a condition was false and
1408 * became true. If a state change happened, we have to notify clients.
1409 *
1410 * Now, if a client subscribes to a given notification and registers
1411 * a trigger *after* that subscription, we have to make sure the
1412 * condition is evaluated at this point while considering only the
1413 * current state. Otherwise, the next evaluation cycle may only see
1414 * that the evaluations remain the same (true for samples n-1 and n) and
1415 * the client will never know that the condition has been met.
1416 */
1417 cds_list_for_each_entry_safe(client_list_element, tmp,
1418 &client_list->list, node) {
1419 ret = evaluate_condition_for_client(trigger, condition,
1420 client_list_element->client, state);
1421 if (ret) {
1422 goto error_free_client_list;
1423 }
1424 }
1425
1426 /*
1427 * Client list ownership transferred to the
1428 * notification_trigger_clients_ht.
1429 */
1430 client_list = NULL;
1431
ab0ee2ca
JG
1432 *cmd_result = LTTNG_OK;
1433error_free_client_list:
1434 if (client_list) {
1435 cds_list_for_each_entry_safe(client_list_element, tmp,
1436 &client_list->list, node) {
1437 free(client_list_element);
1438 }
1439 free(client_list);
1440 }
1441error_free_ht_element:
1442 free(trigger_ht_element);
1443error:
1444 if (free_trigger) {
1445 struct lttng_action *action = lttng_trigger_get_action(trigger);
1446
1447 lttng_condition_destroy(condition);
1448 lttng_action_destroy(action);
1449 lttng_trigger_destroy(trigger);
1450 }
1451 rcu_read_unlock();
1452 return ret;
1453}
1454
cc2295b5 1455static
ab0ee2ca
JG
1456int handle_notification_thread_command_unregister_trigger(
1457 struct notification_thread_state *state,
1458 struct lttng_trigger *trigger,
1459 enum lttng_error_code *_cmd_reply)
1460{
1461 struct cds_lfht_iter iter;
1462 struct cds_lfht_node *node, *triggers_ht_node;
1463 struct lttng_channel_trigger_list *trigger_list;
1464 struct notification_client_list *client_list;
1465 struct notification_client_list_element *client_list_element, *tmp;
1466 struct lttng_trigger_ht_element *trigger_ht_element = NULL;
1467 struct lttng_condition *condition = lttng_trigger_get_condition(
1468 trigger);
1469 struct lttng_action *action;
1470 enum lttng_error_code cmd_reply;
1471
1472 rcu_read_lock();
1473
1474 cds_lfht_lookup(state->triggers_ht,
1475 lttng_condition_hash(condition),
1476 match_condition,
1477 condition,
1478 &iter);
1479 triggers_ht_node = cds_lfht_iter_get_node(&iter);
1480 if (!triggers_ht_node) {
1481 cmd_reply = LTTNG_ERR_TRIGGER_NOT_FOUND;
1482 goto end;
1483 } else {
1484 cmd_reply = LTTNG_OK;
1485 }
1486
1487 /* Remove trigger from channel_triggers_ht. */
1488 cds_lfht_for_each_entry(state->channel_triggers_ht, &iter, trigger_list,
1489 channel_triggers_ht_node) {
1490 struct lttng_trigger_list_element *trigger_element, *tmp;
1491
1492 cds_list_for_each_entry_safe(trigger_element, tmp,
1493 &trigger_list->list, node) {
1494 struct lttng_condition *current_condition =
1495 lttng_trigger_get_condition(
1496 trigger_element->trigger);
1497
1498 assert(current_condition);
1499 if (!lttng_condition_is_equal(condition,
1500 current_condition)) {
1501 continue;
1502 }
1503
1504 DBG("[notification-thread] Removed trigger from channel_triggers_ht");
1505 cds_list_del(&trigger_element->node);
e4db5ace
JR
1506 /* A trigger can only appear once per channel */
1507 break;
ab0ee2ca
JG
1508 }
1509 }
1510
1511 /*
1512 * Remove and release the client list from
1513 * notification_trigger_clients_ht.
1514 */
1515 cds_lfht_lookup(state->notification_trigger_clients_ht,
1516 lttng_condition_hash(condition),
1517 match_client_list,
1518 trigger,
1519 &iter);
1520 node = cds_lfht_iter_get_node(&iter);
1521 assert(node);
1522 client_list = caa_container_of(node, struct notification_client_list,
1523 notification_trigger_ht_node);
1524 cds_list_for_each_entry_safe(client_list_element, tmp,
1525 &client_list->list, node) {
1526 free(client_list_element);
1527 }
1528 cds_lfht_del(state->notification_trigger_clients_ht, node);
1529 free(client_list);
1530
1531 /* Remove trigger from triggers_ht. */
1532 trigger_ht_element = caa_container_of(triggers_ht_node,
1533 struct lttng_trigger_ht_element, node);
1534 cds_lfht_del(state->triggers_ht, triggers_ht_node);
1535
1536 condition = lttng_trigger_get_condition(trigger_ht_element->trigger);
1537 lttng_condition_destroy(condition);
1538 action = lttng_trigger_get_action(trigger_ht_element->trigger);
1539 lttng_action_destroy(action);
1540 lttng_trigger_destroy(trigger_ht_element->trigger);
1541 free(trigger_ht_element);
1542end:
1543 rcu_read_unlock();
1544 if (_cmd_reply) {
1545 *_cmd_reply = cmd_reply;
1546 }
1547 return 0;
1548}
1549
1550/* Returns 0 on success, 1 on exit requested, negative value on error. */
1551int handle_notification_thread_command(
1552 struct notification_thread_handle *handle,
1553 struct notification_thread_state *state)
1554{
1555 int ret;
1556 uint64_t counter;
1557 struct notification_thread_command *cmd;
1558
8abe313a
JG
1559 /* Read the event pipe to put it back into a quiescent state. */
1560 ret = read(lttng_pipe_get_readfd(handle->cmd_queue.event_pipe), &counter,
1561 sizeof(counter));
ab0ee2ca
JG
1562 if (ret == -1) {
1563 goto error;
1564 }
1565
1566 pthread_mutex_lock(&handle->cmd_queue.lock);
1567 cmd = cds_list_first_entry(&handle->cmd_queue.list,
1568 struct notification_thread_command, cmd_list_node);
1569 switch (cmd->type) {
1570 case NOTIFICATION_COMMAND_TYPE_REGISTER_TRIGGER:
1571 DBG("[notification-thread] Received register trigger command");
1572 ret = handle_notification_thread_command_register_trigger(
1573 state, cmd->parameters.trigger,
1574 &cmd->reply_code);
1575 break;
1576 case NOTIFICATION_COMMAND_TYPE_UNREGISTER_TRIGGER:
1577 DBG("[notification-thread] Received unregister trigger command");
1578 ret = handle_notification_thread_command_unregister_trigger(
1579 state, cmd->parameters.trigger,
1580 &cmd->reply_code);
1581 break;
1582 case NOTIFICATION_COMMAND_TYPE_ADD_CHANNEL:
1583 DBG("[notification-thread] Received add channel command");
1584 ret = handle_notification_thread_command_add_channel(
8abe313a
JG
1585 state,
1586 cmd->parameters.add_channel.session.name,
1587 cmd->parameters.add_channel.session.uid,
1588 cmd->parameters.add_channel.session.gid,
1589 cmd->parameters.add_channel.channel.name,
1590 cmd->parameters.add_channel.channel.domain,
1591 cmd->parameters.add_channel.channel.key,
1592 cmd->parameters.add_channel.channel.capacity,
ab0ee2ca
JG
1593 &cmd->reply_code);
1594 break;
1595 case NOTIFICATION_COMMAND_TYPE_REMOVE_CHANNEL:
1596 DBG("[notification-thread] Received remove channel command");
1597 ret = handle_notification_thread_command_remove_channel(
1598 state, cmd->parameters.remove_channel.key,
1599 cmd->parameters.remove_channel.domain,
1600 &cmd->reply_code);
1601 break;
1602 case NOTIFICATION_COMMAND_TYPE_QUIT:
1603 DBG("[notification-thread] Received quit command");
1604 cmd->reply_code = LTTNG_OK;
1605 ret = 1;
1606 goto end;
1607 default:
1608 ERR("[notification-thread] Unknown internal command received");
1609 goto error_unlock;
1610 }
1611
1612 if (ret) {
1613 goto error_unlock;
1614 }
1615end:
1616 cds_list_del(&cmd->cmd_list_node);
8ada111f 1617 lttng_waiter_wake_up(&cmd->reply_waiter);
ab0ee2ca
JG
1618 pthread_mutex_unlock(&handle->cmd_queue.lock);
1619 return ret;
1620error_unlock:
1621 /* Wake-up and return a fatal error to the calling thread. */
8ada111f 1622 lttng_waiter_wake_up(&cmd->reply_waiter);
ab0ee2ca
JG
1623 pthread_mutex_unlock(&handle->cmd_queue.lock);
1624 cmd->reply_code = LTTNG_ERR_FATAL;
1625error:
1626 /* Indicate a fatal error to the caller. */
1627 return -1;
1628}
1629
1630static
1631unsigned long hash_client_socket(int socket)
1632{
1633 return hash_key_ulong((void *) (unsigned long) socket, lttng_ht_seed);
1634}
1635
1636static
1637int socket_set_non_blocking(int socket)
1638{
1639 int ret, flags;
1640
1641 /* Set the pipe as non-blocking. */
1642 ret = fcntl(socket, F_GETFL, 0);
1643 if (ret == -1) {
1644 PERROR("fcntl get socket flags");
1645 goto end;
1646 }
1647 flags = ret;
1648
1649 ret = fcntl(socket, F_SETFL, flags | O_NONBLOCK);
1650 if (ret == -1) {
1651 PERROR("fcntl set O_NONBLOCK socket flag");
1652 goto end;
1653 }
1654 DBG("Client socket (fd = %i) set as non-blocking", socket);
1655end:
1656 return ret;
1657}
1658
1659static
14fa22f8 1660int client_reset_inbound_state(struct notification_client *client)
ab0ee2ca
JG
1661{
1662 int ret;
1663
1664 ret = lttng_dynamic_buffer_set_size(
1665 &client->communication.inbound.buffer, 0);
1666 assert(!ret);
1667
1668 client->communication.inbound.bytes_to_receive =
1669 sizeof(struct lttng_notification_channel_message);
1670 client->communication.inbound.msg_type =
1671 LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_UNKNOWN;
ab0ee2ca
JG
1672 LTTNG_SOCK_SET_UID_CRED(&client->communication.inbound.creds, -1);
1673 LTTNG_SOCK_SET_GID_CRED(&client->communication.inbound.creds, -1);
14fa22f8
JG
1674 ret = lttng_dynamic_buffer_set_size(
1675 &client->communication.inbound.buffer,
1676 client->communication.inbound.bytes_to_receive);
1677 return ret;
ab0ee2ca
JG
1678}
1679
1680int handle_notification_thread_client_connect(
1681 struct notification_thread_state *state)
1682{
1683 int ret;
1684 struct notification_client *client;
1685
1686 DBG("[notification-thread] Handling new notification channel client connection");
1687
1688 client = zmalloc(sizeof(*client));
1689 if (!client) {
1690 /* Fatal error. */
1691 ret = -1;
1692 goto error;
1693 }
1694 CDS_INIT_LIST_HEAD(&client->condition_list);
1695 lttng_dynamic_buffer_init(&client->communication.inbound.buffer);
1696 lttng_dynamic_buffer_init(&client->communication.outbound.buffer);
01ea340e 1697 client->communication.inbound.expect_creds = true;
14fa22f8
JG
1698 ret = client_reset_inbound_state(client);
1699 if (ret) {
1700 ERR("[notification-thread] Failed to reset client communication's inbound state");
1701 ret = 0;
1702 goto error;
1703 }
ab0ee2ca
JG
1704
1705 ret = lttcomm_accept_unix_sock(state->notification_channel_socket);
1706 if (ret < 0) {
1707 ERR("[notification-thread] Failed to accept new notification channel client connection");
1708 ret = 0;
1709 goto error;
1710 }
1711
1712 client->socket = ret;
1713
1714 ret = socket_set_non_blocking(client->socket);
1715 if (ret) {
1716 ERR("[notification-thread] Failed to set new notification channel client connection socket as non-blocking");
1717 goto error;
1718 }
1719
1720 ret = lttcomm_setsockopt_creds_unix_sock(client->socket);
1721 if (ret < 0) {
1722 ERR("[notification-thread] Failed to set socket options on new notification channel client socket");
1723 ret = 0;
1724 goto error;
1725 }
1726
1727 ret = lttng_poll_add(&state->events, client->socket,
1728 LPOLLIN | LPOLLERR |
1729 LPOLLHUP | LPOLLRDHUP);
1730 if (ret < 0) {
1731 ERR("[notification-thread] Failed to add notification channel client socket to poll set");
1732 ret = 0;
1733 goto error;
1734 }
1735 DBG("[notification-thread] Added new notification channel client socket (%i) to poll set",
1736 client->socket);
1737
ab0ee2ca
JG
1738 rcu_read_lock();
1739 cds_lfht_add(state->client_socket_ht,
1740 hash_client_socket(client->socket),
1741 &client->client_socket_ht_node);
1742 rcu_read_unlock();
1743
1744 return ret;
1745error:
1746 notification_client_destroy(client, state);
1747 return ret;
1748}
1749
1750int handle_notification_thread_client_disconnect(
1751 int client_socket,
1752 struct notification_thread_state *state)
1753{
1754 int ret = 0;
1755 struct notification_client *client;
1756
1757 rcu_read_lock();
1758 DBG("[notification-thread] Closing client connection (socket fd = %i)",
1759 client_socket);
1760 client = get_client_from_socket(client_socket, state);
1761 if (!client) {
1762 /* Internal state corruption, fatal error. */
1763 ERR("[notification-thread] Unable to find client (socket fd = %i)",
1764 client_socket);
1765 ret = -1;
1766 goto end;
1767 }
1768
1769 ret = lttng_poll_del(&state->events, client_socket);
1770 if (ret) {
1771 ERR("[notification-thread] Failed to remove client socket from poll set");
1772 }
1773 cds_lfht_del(state->client_socket_ht,
1774 &client->client_socket_ht_node);
1775 notification_client_destroy(client, state);
1776end:
1777 rcu_read_unlock();
1778 return ret;
1779}
1780
1781int handle_notification_thread_client_disconnect_all(
1782 struct notification_thread_state *state)
1783{
1784 struct cds_lfht_iter iter;
1785 struct notification_client *client;
1786 bool error_encoutered = false;
1787
1788 rcu_read_lock();
1789 DBG("[notification-thread] Closing all client connections");
1790 cds_lfht_for_each_entry(state->client_socket_ht, &iter, client,
1791 client_socket_ht_node) {
1792 int ret;
1793
1794 ret = handle_notification_thread_client_disconnect(
1795 client->socket, state);
1796 if (ret) {
1797 error_encoutered = true;
1798 }
1799 }
1800 rcu_read_unlock();
1801 return error_encoutered ? 1 : 0;
1802}
1803
1804int handle_notification_thread_trigger_unregister_all(
1805 struct notification_thread_state *state)
1806{
4149ace8 1807 bool error_occurred = false;
ab0ee2ca
JG
1808 struct cds_lfht_iter iter;
1809 struct lttng_trigger_ht_element *trigger_ht_element;
1810
1811 cds_lfht_for_each_entry(state->triggers_ht, &iter, trigger_ht_element,
1812 node) {
1813 int ret = handle_notification_thread_command_unregister_trigger(
1814 state, trigger_ht_element->trigger, NULL);
1815 if (ret) {
4149ace8 1816 error_occurred = true;
ab0ee2ca
JG
1817 }
1818 }
4149ace8 1819 return error_occurred ? -1 : 0;
ab0ee2ca
JG
1820}
1821
1822static
1823int client_flush_outgoing_queue(struct notification_client *client,
1824 struct notification_thread_state *state)
1825{
1826 ssize_t ret;
1827 size_t to_send_count;
1828
1829 assert(client->communication.outbound.buffer.size != 0);
1830 to_send_count = client->communication.outbound.buffer.size;
1831 DBG("[notification-thread] Flushing client (socket fd = %i) outgoing queue",
1832 client->socket);
1833
1834 ret = lttcomm_send_unix_sock_non_block(client->socket,
1835 client->communication.outbound.buffer.data,
1836 to_send_count);
1837 if ((ret < 0 && (errno == EAGAIN || errno == EWOULDBLOCK)) ||
1838 (ret > 0 && ret < to_send_count)) {
1839 DBG("[notification-thread] Client (socket fd = %i) outgoing queue could not be completely flushed",
1840 client->socket);
1841 to_send_count -= max(ret, 0);
1842
1843 memcpy(client->communication.outbound.buffer.data,
1844 client->communication.outbound.buffer.data +
1845 client->communication.outbound.buffer.size - to_send_count,
1846 to_send_count);
1847 ret = lttng_dynamic_buffer_set_size(
1848 &client->communication.outbound.buffer,
1849 to_send_count);
1850 if (ret) {
1851 goto error;
1852 }
1853
1854 /*
1855 * We want to be notified whenever there is buffer space
1856 * available to send the rest of the payload.
1857 */
1858 ret = lttng_poll_mod(&state->events, client->socket,
1859 CLIENT_POLL_MASK_IN_OUT);
1860 if (ret) {
1861 goto error;
1862 }
1863 } else if (ret < 0) {
1864 /* Generic error, disconnect the client. */
1865 ERR("[notification-thread] Failed to send flush outgoing queue, disconnecting client (socket fd = %i)",
1866 client->socket);
1867 ret = handle_notification_thread_client_disconnect(
1868 client->socket, state);
1869 if (ret) {
1870 goto error;
1871 }
1872 } else {
1873 /* No error and flushed the queue completely. */
1874 ret = lttng_dynamic_buffer_set_size(
1875 &client->communication.outbound.buffer, 0);
1876 if (ret) {
1877 goto error;
1878 }
1879 ret = lttng_poll_mod(&state->events, client->socket,
1880 CLIENT_POLL_MASK_IN);
1881 if (ret) {
1882 goto error;
1883 }
1884
1885 client->communication.outbound.queued_command_reply = false;
1886 client->communication.outbound.dropped_notification = false;
1887 }
1888
1889 return 0;
1890error:
1891 return -1;
1892}
1893
1894static
1895int client_send_command_reply(struct notification_client *client,
1896 struct notification_thread_state *state,
1897 enum lttng_notification_channel_status status)
1898{
1899 int ret;
1900 struct lttng_notification_channel_command_reply reply = {
1901 .status = (int8_t) status,
1902 };
1903 struct lttng_notification_channel_message msg = {
1904 .type = (int8_t) LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_COMMAND_REPLY,
1905 .size = sizeof(reply),
1906 };
1907 char buffer[sizeof(msg) + sizeof(reply)];
1908
1909 if (client->communication.outbound.queued_command_reply) {
1910 /* Protocol error. */
1911 goto error;
1912 }
1913
1914 memcpy(buffer, &msg, sizeof(msg));
1915 memcpy(buffer + sizeof(msg), &reply, sizeof(reply));
1916 DBG("[notification-thread] Send command reply (%i)", (int) status);
1917
1918 /* Enqueue buffer to outgoing queue and flush it. */
1919 ret = lttng_dynamic_buffer_append(
1920 &client->communication.outbound.buffer,
1921 buffer, sizeof(buffer));
1922 if (ret) {
1923 goto error;
1924 }
1925
1926 ret = client_flush_outgoing_queue(client, state);
1927 if (ret) {
1928 goto error;
1929 }
1930
1931 if (client->communication.outbound.buffer.size != 0) {
1932 /* Queue could not be emptied. */
1933 client->communication.outbound.queued_command_reply = true;
1934 }
1935
1936 return 0;
1937error:
1938 return -1;
1939}
1940
1941static
1942int client_dispatch_message(struct notification_client *client,
1943 struct notification_thread_state *state)
1944{
1945 int ret = 0;
1946
1947 if (client->communication.inbound.msg_type !=
1948 LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_HANDSHAKE &&
1949 client->communication.inbound.msg_type !=
1950 LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_UNKNOWN &&
1951 !client->validated) {
1952 WARN("[notification-thread] client attempted a command before handshake");
1953 ret = -1;
1954 goto end;
1955 }
1956
1957 switch (client->communication.inbound.msg_type) {
1958 case LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_UNKNOWN:
1959 {
1960 /*
1961 * Receiving message header. The function will be called again
1962 * once the rest of the message as been received and can be
1963 * interpreted.
1964 */
1965 const struct lttng_notification_channel_message *msg;
1966
1967 assert(sizeof(*msg) ==
1968 client->communication.inbound.buffer.size);
1969 msg = (const struct lttng_notification_channel_message *)
1970 client->communication.inbound.buffer.data;
1971
1972 if (msg->size == 0 || msg->size > DEFAULT_MAX_NOTIFICATION_CLIENT_MESSAGE_PAYLOAD_SIZE) {
1973 ERR("[notification-thread] Invalid notification channel message: length = %u", msg->size);
1974 ret = -1;
1975 goto end;
1976 }
1977
1978 switch (msg->type) {
1979 case LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_SUBSCRIBE:
1980 case LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_UNSUBSCRIBE:
1981 case LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_HANDSHAKE:
1982 break;
1983 default:
1984 ret = -1;
1985 ERR("[notification-thread] Invalid notification channel message: unexpected message type");
1986 goto end;
1987 }
1988
1989 client->communication.inbound.bytes_to_receive = msg->size;
1990 client->communication.inbound.msg_type =
1991 (enum lttng_notification_channel_message_type) msg->type;
ab0ee2ca 1992 ret = lttng_dynamic_buffer_set_size(
14fa22f8 1993 &client->communication.inbound.buffer, msg->size);
ab0ee2ca
JG
1994 if (ret) {
1995 goto end;
1996 }
1997 break;
1998 }
1999 case LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_HANDSHAKE:
2000 {
2001 struct lttng_notification_channel_command_handshake *handshake_client;
2002 struct lttng_notification_channel_command_handshake handshake_reply = {
2003 .major = LTTNG_NOTIFICATION_CHANNEL_VERSION_MAJOR,
2004 .minor = LTTNG_NOTIFICATION_CHANNEL_VERSION_MINOR,
2005 };
2006 struct lttng_notification_channel_message msg_header = {
2007 .type = LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_HANDSHAKE,
2008 .size = sizeof(handshake_reply),
2009 };
2010 enum lttng_notification_channel_status status =
2011 LTTNG_NOTIFICATION_CHANNEL_STATUS_OK;
2012 char send_buffer[sizeof(msg_header) + sizeof(handshake_reply)];
2013
2014 memcpy(send_buffer, &msg_header, sizeof(msg_header));
2015 memcpy(send_buffer + sizeof(msg_header), &handshake_reply,
2016 sizeof(handshake_reply));
2017
2018 handshake_client =
2019 (struct lttng_notification_channel_command_handshake *)
2020 client->communication.inbound.buffer.data;
47a32869 2021 client->major = handshake_client->major;
ab0ee2ca
JG
2022 client->minor = handshake_client->minor;
2023 if (!client->communication.inbound.creds_received) {
2024 ERR("[notification-thread] No credentials received from client");
2025 ret = -1;
2026 goto end;
2027 }
2028
2029 client->uid = LTTNG_SOCK_GET_UID_CRED(
2030 &client->communication.inbound.creds);
2031 client->gid = LTTNG_SOCK_GET_GID_CRED(
2032 &client->communication.inbound.creds);
2033 DBG("[notification-thread] Received handshake from client (uid = %u, gid = %u) with version %i.%i",
2034 client->uid, client->gid, (int) client->major,
2035 (int) client->minor);
2036
2037 if (handshake_client->major != LTTNG_NOTIFICATION_CHANNEL_VERSION_MAJOR) {
2038 status = LTTNG_NOTIFICATION_CHANNEL_STATUS_UNSUPPORTED_VERSION;
2039 }
2040
2041 ret = lttng_dynamic_buffer_append(&client->communication.outbound.buffer,
2042 send_buffer, sizeof(send_buffer));
2043 if (ret) {
2044 ERR("[notification-thread] Failed to send protocol version to notification channel client");
2045 goto end;
2046 }
2047
2048 ret = client_flush_outgoing_queue(client, state);
2049 if (ret) {
2050 goto end;
2051 }
2052
2053 ret = client_send_command_reply(client, state, status);
2054 if (ret) {
2055 ERR("[notification-thread] Failed to send reply to notification channel client");
2056 goto end;
2057 }
2058
2059 /* Set reception state to receive the next message header. */
14fa22f8
JG
2060 ret = client_reset_inbound_state(client);
2061 if (ret) {
2062 ERR("[notification-thread] Failed to reset client communication's inbound state");
2063 goto end;
2064 }
ab0ee2ca
JG
2065 client->validated = true;
2066 break;
2067 }
2068 case LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_SUBSCRIBE:
2069 case LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_UNSUBSCRIBE:
2070 {
2071 struct lttng_condition *condition;
2072 enum lttng_notification_channel_status status =
2073 LTTNG_NOTIFICATION_CHANNEL_STATUS_OK;
2074 const struct lttng_buffer_view condition_view =
2075 lttng_buffer_view_from_dynamic_buffer(
2076 &client->communication.inbound.buffer,
2077 0, -1);
2078 size_t expected_condition_size =
2079 client->communication.inbound.buffer.size;
2080
2081 ret = lttng_condition_create_from_buffer(&condition_view,
2082 &condition);
2083 if (ret != expected_condition_size) {
2084 ERR("[notification-thread] Malformed condition received from client");
2085 goto end;
2086 }
2087
2088 if (client->communication.inbound.msg_type ==
2089 LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_SUBSCRIBE) {
ab0ee2ca
JG
2090 ret = notification_thread_client_subscribe(client,
2091 condition, state, &status);
2092 } else {
2093 ret = notification_thread_client_unsubscribe(client,
2094 condition, state, &status);
2095 }
2096 if (ret) {
2097 goto end;
2098 }
2099
2100 ret = client_send_command_reply(client, state, status);
2101 if (ret) {
2102 ERR("[notification-thread] Failed to send reply to notification channel client");
2103 goto end;
2104 }
2105
2106 /* Set reception state to receive the next message header. */
14fa22f8
JG
2107 ret = client_reset_inbound_state(client);
2108 if (ret) {
2109 ERR("[notification-thread] Failed to reset client communication's inbound state");
2110 goto end;
2111 }
ab0ee2ca
JG
2112 break;
2113 }
2114 default:
2115 abort();
2116 }
2117end:
2118 return ret;
2119}
2120
2121/* Incoming data from client. */
2122int handle_notification_thread_client_in(
2123 struct notification_thread_state *state, int socket)
2124{
14fa22f8 2125 int ret = 0;
ab0ee2ca
JG
2126 struct notification_client *client;
2127 ssize_t recv_ret;
2128 size_t offset;
2129
2130 client = get_client_from_socket(socket, state);
2131 if (!client) {
2132 /* Internal error, abort. */
2133 ret = -1;
2134 goto end;
2135 }
2136
14fa22f8
JG
2137 offset = client->communication.inbound.buffer.size -
2138 client->communication.inbound.bytes_to_receive;
01ea340e 2139 if (client->communication.inbound.expect_creds) {
ab0ee2ca
JG
2140 recv_ret = lttcomm_recv_creds_unix_sock(socket,
2141 client->communication.inbound.buffer.data + offset,
2142 client->communication.inbound.bytes_to_receive,
2143 &client->communication.inbound.creds);
2144 if (recv_ret > 0) {
01ea340e 2145 client->communication.inbound.expect_creds = false;
ab0ee2ca
JG
2146 client->communication.inbound.creds_received = true;
2147 }
2148 } else {
2149 recv_ret = lttcomm_recv_unix_sock_non_block(socket,
2150 client->communication.inbound.buffer.data + offset,
2151 client->communication.inbound.bytes_to_receive);
2152 }
2153 if (recv_ret < 0) {
2154 goto error_disconnect_client;
2155 }
2156
2157 client->communication.inbound.bytes_to_receive -= recv_ret;
ab0ee2ca
JG
2158 if (client->communication.inbound.bytes_to_receive == 0) {
2159 ret = client_dispatch_message(client, state);
2160 if (ret) {
2161 /*
2162 * Only returns an error if this client must be
2163 * disconnected.
2164 */
2165 goto error_disconnect_client;
2166 }
2167 } else {
2168 goto end;
2169 }
2170end:
2171 return ret;
2172error_disconnect_client:
2173 ret = handle_notification_thread_client_disconnect(socket, state);
2174 return ret;
2175}
2176
2177/* Client ready to receive outgoing data. */
2178int handle_notification_thread_client_out(
2179 struct notification_thread_state *state, int socket)
2180{
2181 int ret;
2182 struct notification_client *client;
2183
2184 client = get_client_from_socket(socket, state);
2185 if (!client) {
2186 /* Internal error, abort. */
2187 ret = -1;
2188 goto end;
2189 }
2190
2191 ret = client_flush_outgoing_queue(client, state);
2192 if (ret) {
2193 goto end;
2194 }
2195end:
2196 return ret;
2197}
2198
2199static
e8360425
JD
2200bool evaluate_buffer_usage_condition(const struct lttng_condition *condition,
2201 const struct channel_state_sample *sample,
2202 uint64_t buffer_capacity)
ab0ee2ca
JG
2203{
2204 bool result = false;
2205 uint64_t threshold;
2206 enum lttng_condition_type condition_type;
e8360425 2207 const struct lttng_condition_buffer_usage *use_condition = container_of(
ab0ee2ca
JG
2208 condition, struct lttng_condition_buffer_usage,
2209 parent);
2210
ab0ee2ca
JG
2211 if (use_condition->threshold_bytes.set) {
2212 threshold = use_condition->threshold_bytes.value;
2213 } else {
2214 /*
2215 * Threshold was expressed as a ratio.
2216 *
2217 * TODO the threshold (in bytes) of conditions expressed
2218 * as a ratio of total buffer size could be cached to
2219 * forego this double-multiplication or it could be performed
2220 * as fixed-point math.
2221 *
2222 * Note that caching should accomodate the case where the
2223 * condition applies to multiple channels (i.e. don't assume
2224 * that all channels matching my_chann* have the same size...)
2225 */
2226 threshold = (uint64_t) (use_condition->threshold_ratio.value *
2227 (double) buffer_capacity);
2228 }
2229
2230 condition_type = lttng_condition_get_type(condition);
2231 if (condition_type == LTTNG_CONDITION_TYPE_BUFFER_USAGE_LOW) {
2232 DBG("[notification-thread] Low buffer usage condition being evaluated: threshold = %" PRIu64 ", highest usage = %" PRIu64,
2233 threshold, sample->highest_usage);
2234
2235 /*
2236 * The low condition should only be triggered once _all_ of the
2237 * streams in a channel have gone below the "low" threshold.
2238 */
2239 if (sample->highest_usage <= threshold) {
2240 result = true;
2241 }
2242 } else {
2243 DBG("[notification-thread] High buffer usage condition being evaluated: threshold = %" PRIu64 ", highest usage = %" PRIu64,
2244 threshold, sample->highest_usage);
2245
2246 /*
2247 * For high buffer usage scenarios, we want to trigger whenever
2248 * _any_ of the streams has reached the "high" threshold.
2249 */
2250 if (sample->highest_usage >= threshold) {
2251 result = true;
2252 }
2253 }
e8360425 2254
ab0ee2ca
JG
2255 return result;
2256}
2257
2258static
e8360425
JD
2259bool evaluate_session_consumed_size_condition(
2260 const struct lttng_condition *condition,
2261 uint64_t session_consumed_size)
2262{
2263 uint64_t threshold;
2264 const struct lttng_condition_session_consumed_size *size_condition =
2265 container_of(condition,
2266 struct lttng_condition_session_consumed_size,
2267 parent);
2268
2269 threshold = size_condition->consumed_threshold_bytes.value;
2270 DBG("[notification-thread] Session consumed size condition being evaluated: threshold = %" PRIu64 ", current size = %" PRIu64,
2271 threshold, session_consumed_size);
2272 return session_consumed_size >= threshold;
2273}
2274
2275static
2276int evaluate_condition(const struct lttng_condition *condition,
ab0ee2ca 2277 struct lttng_evaluation **evaluation,
e8360425
JD
2278 const struct notification_thread_state *state,
2279 const struct channel_state_sample *previous_sample,
2280 const struct channel_state_sample *latest_sample,
2281 uint64_t previous_session_consumed_total,
2282 uint64_t latest_session_consumed_total,
2283 struct channel_info *channel_info)
ab0ee2ca
JG
2284{
2285 int ret = 0;
2286 enum lttng_condition_type condition_type;
e8360425
JD
2287 const bool previous_sample_available = !!previous_sample;
2288 bool previous_sample_result = false;
ab0ee2ca
JG
2289 bool latest_sample_result;
2290
2291 condition_type = lttng_condition_get_type(condition);
ab0ee2ca 2292
e8360425
JD
2293 switch (condition_type) {
2294 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_LOW:
2295 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_HIGH:
2296 if (caa_likely(previous_sample_available)) {
2297 previous_sample_result =
2298 evaluate_buffer_usage_condition(condition,
2299 previous_sample, channel_info->capacity);
2300 }
2301 latest_sample_result = evaluate_buffer_usage_condition(
2302 condition, latest_sample,
2303 channel_info->capacity);
2304 break;
2305 case LTTNG_CONDITION_TYPE_SESSION_CONSUMED_SIZE:
2306 if (caa_likely(previous_sample_available)) {
2307 previous_sample_result =
2308 evaluate_session_consumed_size_condition(
2309 condition,
2310 previous_session_consumed_total);
2311 }
2312 latest_sample_result =
2313 evaluate_session_consumed_size_condition(
2314 condition,
2315 latest_session_consumed_total);
2316 break;
2317 default:
2318 /* Unknown condition type; internal error. */
2319 abort();
2320 }
ab0ee2ca
JG
2321
2322 if (!latest_sample_result ||
2323 (previous_sample_result == latest_sample_result)) {
2324 /*
2325 * Only trigger on a condition evaluation transition.
2326 *
2327 * NOTE: This edge-triggered logic may not be appropriate for
2328 * future condition types.
2329 */
2330 goto end;
2331 }
2332
e8360425
JD
2333 if (!evaluation || !latest_sample_result) {
2334 goto end;
2335 }
2336
2337 switch (condition_type) {
2338 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_LOW:
2339 case LTTNG_CONDITION_TYPE_BUFFER_USAGE_HIGH:
ab0ee2ca
JG
2340 *evaluation = lttng_evaluation_buffer_usage_create(
2341 condition_type,
2342 latest_sample->highest_usage,
e8360425
JD
2343 channel_info->capacity);
2344 break;
2345 case LTTNG_CONDITION_TYPE_SESSION_CONSUMED_SIZE:
2346 *evaluation = lttng_evaluation_session_consumed_size_create(
e8360425
JD
2347 latest_session_consumed_total);
2348 break;
2349 default:
2350 abort();
2351 }
2352
2353 if (!*evaluation) {
2354 ret = -1;
2355 goto end;
ab0ee2ca
JG
2356 }
2357end:
2358 return ret;
2359}
2360
2361static
2362int client_enqueue_dropped_notification(struct notification_client *client,
2363 struct notification_thread_state *state)
2364{
2365 int ret;
2366 struct lttng_notification_channel_message msg = {
2367 .type = (int8_t) LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_NOTIFICATION_DROPPED,
2368 .size = 0,
2369 };
2370
2371 ret = lttng_dynamic_buffer_append(
2372 &client->communication.outbound.buffer, &msg,
2373 sizeof(msg));
2374 return ret;
2375}
2376
2377static
2378int send_evaluation_to_clients(struct lttng_trigger *trigger,
2379 struct lttng_evaluation *evaluation,
2380 struct notification_client_list* client_list,
2381 struct notification_thread_state *state,
2382 uid_t channel_uid, gid_t channel_gid)
2383{
2384 int ret = 0;
2385 struct lttng_dynamic_buffer msg_buffer;
2386 struct notification_client_list_element *client_list_element, *tmp;
2387 struct lttng_notification *notification;
2388 struct lttng_condition *condition;
3647288f
JG
2389 struct lttng_notification_channel_message msg_header = {
2390 .type = (int8_t) LTTNG_NOTIFICATION_CHANNEL_MESSAGE_TYPE_NOTIFICATION,
2391 };
ab0ee2ca
JG
2392
2393 lttng_dynamic_buffer_init(&msg_buffer);
2394
2395 condition = lttng_trigger_get_condition(trigger);
2396 assert(condition);
2397
2398 notification = lttng_notification_create(condition, evaluation);
2399 if (!notification) {
2400 ret = -1;
2401 goto end;
2402 }
2403
3647288f
JG
2404 ret = lttng_dynamic_buffer_append(&msg_buffer, &msg_header,
2405 sizeof(msg_header));
ab0ee2ca
JG
2406 if (ret) {
2407 goto end;
2408 }
2409
3647288f 2410 ret = lttng_notification_serialize(notification, &msg_buffer);
ab0ee2ca 2411 if (ret) {
ab0ee2ca
JG
2412 ERR("[notification-thread] Failed to serialize notification");
2413 ret = -1;
2414 goto end;
2415 }
2416
3647288f
JG
2417 /* Update payload size. */
2418 ((struct lttng_notification_channel_message * ) msg_buffer.data)->size =
2419 (uint32_t) (msg_buffer.size - sizeof(msg_header));
2420
ab0ee2ca
JG
2421 cds_list_for_each_entry_safe(client_list_element, tmp,
2422 &client_list->list, node) {
2423 struct notification_client *client =
2424 client_list_element->client;
2425
2426 if (client->uid != channel_uid && client->gid != channel_gid &&
2427 client->uid != 0) {
2428 /* Client is not allowed to monitor this channel. */
2429 DBG("[notification-thread] Skipping client at it does not have the permission to receive notification for this channel");
2430 continue;
2431 }
2432
2433 DBG("[notification-thread] Sending notification to client (fd = %i, %zu bytes)",
2434 client->socket, msg_buffer.size);
2435 if (client->communication.outbound.buffer.size) {
2436 /*
2437 * Outgoing data is already buffered for this client;
2438 * drop the notification and enqueue a "dropped
2439 * notification" message if this is the first dropped
2440 * notification since the socket spilled-over to the
2441 * queue.
2442 */
2443 DBG("[notification-thread] Dropping notification addressed to client (socket fd = %i)",
2444 client->socket);
2445 if (!client->communication.outbound.dropped_notification) {
2446 client->communication.outbound.dropped_notification = true;
2447 ret = client_enqueue_dropped_notification(
2448 client, state);
2449 if (ret) {
2450 goto end;
2451 }
2452 }
2453 continue;
2454 }
2455
2456 ret = lttng_dynamic_buffer_append_buffer(
2457 &client->communication.outbound.buffer,
2458 &msg_buffer);
2459 if (ret) {
2460 goto end;
2461 }
2462
2463 ret = client_flush_outgoing_queue(client, state);
2464 if (ret) {
2465 goto end;
2466 }
2467 }
2468 ret = 0;
2469end:
2470 lttng_notification_destroy(notification);
2471 lttng_dynamic_buffer_reset(&msg_buffer);
2472 return ret;
2473}
2474
2475int handle_notification_thread_channel_sample(
2476 struct notification_thread_state *state, int pipe,
2477 enum lttng_domain_type domain)
2478{
2479 int ret = 0;
2480 struct lttcomm_consumer_channel_monitor_msg sample_msg;
ab0ee2ca
JG
2481 struct channel_info *channel_info;
2482 struct cds_lfht_node *node;
2483 struct cds_lfht_iter iter;
2484 struct lttng_channel_trigger_list *trigger_list;
2485 struct lttng_trigger_list_element *trigger_list_element;
2486 bool previous_sample_available = false;
e8360425
JD
2487 struct channel_state_sample previous_sample, latest_sample;
2488 uint64_t previous_session_consumed_total, latest_session_consumed_total;
ab0ee2ca
JG
2489
2490 /*
2491 * The monitoring pipe only holds messages smaller than PIPE_BUF,
2492 * ensuring that read/write of sampling messages are atomic.
2493 */
7c2551ef 2494 ret = lttng_read(pipe, &sample_msg, sizeof(sample_msg));
ab0ee2ca
JG
2495 if (ret != sizeof(sample_msg)) {
2496 ERR("[notification-thread] Failed to read from monitoring pipe (fd = %i)",
2497 pipe);
2498 ret = -1;
2499 goto end;
2500 }
2501
2502 ret = 0;
2503 latest_sample.key.key = sample_msg.key;
2504 latest_sample.key.domain = domain;
2505 latest_sample.highest_usage = sample_msg.highest;
2506 latest_sample.lowest_usage = sample_msg.lowest;
e8360425 2507 latest_sample.channel_total_consumed = sample_msg.total_consumed;
ab0ee2ca
JG
2508
2509 rcu_read_lock();
2510
2511 /* Retrieve the channel's informations */
2512 cds_lfht_lookup(state->channels_ht,
2513 hash_channel_key(&latest_sample.key),
2514 match_channel_info,
2515 &latest_sample.key,
2516 &iter);
2517 node = cds_lfht_iter_get_node(&iter);
e0b5f87b 2518 if (caa_unlikely(!node)) {
ab0ee2ca
JG
2519 /*
2520 * Not an error since the consumer can push a sample to the pipe
2521 * and the rest of the session daemon could notify us of the
2522 * channel's destruction before we get a chance to process that
2523 * sample.
2524 */
2525 DBG("[notification-thread] Received a sample for an unknown channel from consumerd, key = %" PRIu64 " in %s domain",
2526 latest_sample.key.key,
2527 domain == LTTNG_DOMAIN_KERNEL ? "kernel" :
2528 "user space");
2529 goto end_unlock;
2530 }
2531 channel_info = caa_container_of(node, struct channel_info,
2532 channels_ht_node);
e8360425 2533 DBG("[notification-thread] Handling channel sample for channel %s (key = %" PRIu64 ") in session %s (highest usage = %" PRIu64 ", lowest usage = %" PRIu64", total consumed = %" PRIu64")",
8abe313a 2534 channel_info->name,
ab0ee2ca 2535 latest_sample.key.key,
8abe313a 2536 channel_info->session_info->name,
ab0ee2ca 2537 latest_sample.highest_usage,
e8360425
JD
2538 latest_sample.lowest_usage,
2539 latest_sample.channel_total_consumed);
2540
2541 previous_session_consumed_total =
2542 channel_info->session_info->consumed_data_size;
ab0ee2ca
JG
2543
2544 /* Retrieve the channel's last sample, if it exists, and update it. */
2545 cds_lfht_lookup(state->channel_state_ht,
2546 hash_channel_key(&latest_sample.key),
2547 match_channel_state_sample,
2548 &latest_sample.key,
2549 &iter);
2550 node = cds_lfht_iter_get_node(&iter);
e0b5f87b 2551 if (caa_likely(node)) {
ab0ee2ca
JG
2552 struct channel_state_sample *stored_sample;
2553
2554 /* Update the sample stored. */
2555 stored_sample = caa_container_of(node,
2556 struct channel_state_sample,
2557 channel_state_ht_node);
e8360425 2558
ab0ee2ca
JG
2559 memcpy(&previous_sample, stored_sample,
2560 sizeof(previous_sample));
2561 stored_sample->highest_usage = latest_sample.highest_usage;
2562 stored_sample->lowest_usage = latest_sample.lowest_usage;
0f0479d7 2563 stored_sample->channel_total_consumed = latest_sample.channel_total_consumed;
ab0ee2ca 2564 previous_sample_available = true;
e8360425
JD
2565
2566 latest_session_consumed_total =
2567 previous_session_consumed_total +
2568 (latest_sample.channel_total_consumed - previous_sample.channel_total_consumed);
ab0ee2ca
JG
2569 } else {
2570 /*
2571 * This is the channel's first sample, allocate space for and
2572 * store the new sample.
2573 */
2574 struct channel_state_sample *stored_sample;
2575
2576 stored_sample = zmalloc(sizeof(*stored_sample));
2577 if (!stored_sample) {
2578 ret = -1;
2579 goto end_unlock;
2580 }
2581
2582 memcpy(stored_sample, &latest_sample, sizeof(*stored_sample));
2583 cds_lfht_node_init(&stored_sample->channel_state_ht_node);
2584 cds_lfht_add(state->channel_state_ht,
2585 hash_channel_key(&stored_sample->key),
2586 &stored_sample->channel_state_ht_node);
e8360425
JD
2587
2588 latest_session_consumed_total =
2589 previous_session_consumed_total +
2590 latest_sample.channel_total_consumed;
ab0ee2ca
JG
2591 }
2592
e8360425
JD
2593 channel_info->session_info->consumed_data_size =
2594 latest_session_consumed_total;
2595
ab0ee2ca
JG
2596 /* Find triggers associated with this channel. */
2597 cds_lfht_lookup(state->channel_triggers_ht,
2598 hash_channel_key(&latest_sample.key),
2599 match_channel_trigger_list,
2600 &latest_sample.key,
2601 &iter);
2602 node = cds_lfht_iter_get_node(&iter);
e0b5f87b 2603 if (caa_likely(!node)) {
ab0ee2ca
JG
2604 goto end_unlock;
2605 }
2606
2607 trigger_list = caa_container_of(node, struct lttng_channel_trigger_list,
2608 channel_triggers_ht_node);
2609 cds_list_for_each_entry(trigger_list_element, &trigger_list->list,
2610 node) {
2611 struct lttng_condition *condition;
2612 struct lttng_action *action;
2613 struct lttng_trigger *trigger;
2614 struct notification_client_list *client_list;
2615 struct lttng_evaluation *evaluation = NULL;
2616
2617 trigger = trigger_list_element->trigger;
2618 condition = lttng_trigger_get_condition(trigger);
2619 assert(condition);
2620 action = lttng_trigger_get_action(trigger);
2621
2622 /* Notify actions are the only type currently supported. */
2623 assert(lttng_action_get_type(action) ==
2624 LTTNG_ACTION_TYPE_NOTIFY);
2625
2626 /*
2627 * Check if any client is subscribed to the result of this
2628 * evaluation.
2629 */
2630 cds_lfht_lookup(state->notification_trigger_clients_ht,
2631 lttng_condition_hash(condition),
2632 match_client_list,
2633 trigger,
2634 &iter);
2635 node = cds_lfht_iter_get_node(&iter);
2636 assert(node);
2637
2638 client_list = caa_container_of(node,
2639 struct notification_client_list,
2640 notification_trigger_ht_node);
2641 if (cds_list_empty(&client_list->list)) {
2642 /*
2643 * No clients interested in the evaluation's result,
2644 * skip it.
2645 */
2646 continue;
2647 }
2648
2649 ret = evaluate_condition(condition, &evaluation, state,
2650 previous_sample_available ? &previous_sample : NULL,
e8360425
JD
2651 &latest_sample,
2652 previous_session_consumed_total,
2653 latest_session_consumed_total,
2654 channel_info);
2655 if (caa_unlikely(ret)) {
ab0ee2ca
JG
2656 goto end_unlock;
2657 }
2658
e8360425 2659 if (caa_likely(!evaluation)) {
ab0ee2ca
JG
2660 continue;
2661 }
2662
2663 /* Dispatch evaluation result to all clients. */
2664 ret = send_evaluation_to_clients(trigger_list_element->trigger,
2665 evaluation, client_list, state,
8abe313a
JG
2666 channel_info->session_info->uid,
2667 channel_info->session_info->gid);
2668 lttng_evaluation_destroy(evaluation);
e0b5f87b 2669 if (caa_unlikely(ret)) {
ab0ee2ca
JG
2670 goto end_unlock;
2671 }
2672 }
2673end_unlock:
2674 rcu_read_unlock();
2675end:
2676 return ret;
2677}
This page took 0.147472 seconds and 4 git commands to generate.