Implements `lttng_event_notifier_notification_{create,destroy}()`
[lttng-tools.git] / src / bin / lttng-sessiond / notification-thread-events.c
index a878c0dbef96b57c3555672b1324d2b9ae6bd9fe..823f2a7eb3b2d53a4642844738ebd808445e2f26 100644 (file)
@@ -125,19 +125,6 @@ struct lttng_condition_list_element {
        struct cds_list_head node;
 };
 
-/*
- * Facilities to carry the different notifications type in the action processing
- * code path.
- */
-struct lttng_event_notifier_notification {
-       union {
-               struct lttng_ust_event_notifier_notification *ust;
-               struct lttng_kernel_event_notifier_notification *kernel;
-       } notification;
-       uint64_t token;
-       enum lttng_domain_type type;
-};
-
 struct channel_state_sample {
        struct channel_key key;
        struct cds_lfht_node channel_state_ht_node;
@@ -1654,7 +1641,7 @@ int handle_notification_thread_command_add_channel(
 
        DBG("[notification-thread] Adding channel %s from session %s, channel key = %" PRIu64 " in %s domain",
                        channel_name, session_name, channel_key_int,
-                       channel_domain == LTTNG_DOMAIN_KERNEL ? "kernel" : "user space");
+                       lttng_domain_type_str(channel_domain));
 
        CDS_INIT_LIST_HEAD(&trigger_list);
 
@@ -1755,7 +1742,7 @@ int handle_notification_thread_command_remove_channel(
        struct channel_info *channel_info;
 
        DBG("[notification-thread] Removing channel key = %" PRIu64 " in %s domain",
-                       channel_key, domain == LTTNG_DOMAIN_KERNEL ? "kernel" : "user space");
+                       channel_key, lttng_domain_type_str(domain));
 
        rcu_read_lock();
 
@@ -2610,11 +2597,15 @@ int handle_notification_thread_command_register_trigger(
                ret = evaluate_session_condition_for_client(condition, state,
                                &evaluation, &object_uid,
                                &object_gid);
+               LTTNG_OPTIONAL_SET(&object_creds.uid, object_uid);
+               LTTNG_OPTIONAL_SET(&object_creds.gid, object_gid);
                break;
        case LTTNG_OBJECT_TYPE_CHANNEL:
                ret = evaluate_channel_condition_for_client(condition, state,
                                &evaluation, &object_uid,
                                &object_gid);
+               LTTNG_OPTIONAL_SET(&object_creds.uid, object_uid);
+               LTTNG_OPTIONAL_SET(&object_creds.gid, object_gid);
                break;
        case LTTNG_OBJECT_TYPE_NONE:
                ret = 0;
@@ -2630,9 +2621,6 @@ int handle_notification_thread_command_register_trigger(
                goto error_put_client_list;
        }
 
-       LTTNG_OPTIONAL_SET(&object_creds.uid, object_uid);
-       LTTNG_OPTIONAL_SET(&object_creds.gid, object_gid);
-
        DBG("Newly registered trigger's condition evaluated to %s",
                        evaluation ? "true" : "false");
        if (!evaluation) {
@@ -4173,34 +4161,27 @@ end:
        return ret;
 }
 
-int handle_notification_thread_event_notification(struct notification_thread_state *state,
-               int notification_pipe_read_fd,
-               enum lttng_domain_type domain)
+static struct lttng_event_notifier_notification *receive_notification(
+               int notification_pipe_read_fd, enum lttng_domain_type domain)
 {
        int ret;
-       struct lttng_ust_event_notifier_notification ust_notification;
-       struct lttng_kernel_event_notifier_notification kernel_notification;
-       struct cds_lfht_node *node;
-       struct cds_lfht_iter iter;
-       struct notification_trigger_tokens_ht_element *element;
-       enum lttng_action_type action_type;
-       const struct lttng_action *action;
-       struct lttng_event_notifier_notification notification;
+       uint64_t token;
+       struct lttng_event_notifier_notification *notification = NULL;
        void *reception_buffer;
        size_t reception_size;
 
-       notification.type = domain;
+       struct lttng_ust_event_notifier_notification ust_notification;
+       struct lttng_kernel_event_notifier_notification kernel_notification;
 
+       /* Init lttng_event_notifier_notification */
        switch(domain) {
        case LTTNG_DOMAIN_UST:
                reception_buffer = (void *) &ust_notification;
                reception_size = sizeof(ust_notification);
-               notification.notification.ust = &ust_notification;
                break;
        case LTTNG_DOMAIN_KERNEL:
                reception_buffer = (void *) &kernel_notification;
                reception_size = sizeof(kernel_notification);
-               notification.notification.kernel = &kernel_notification;
                break;
        default:
                abort();
@@ -4221,26 +4202,56 @@ int handle_notification_thread_event_notification(struct notification_thread_sta
 
        switch(domain) {
        case LTTNG_DOMAIN_UST:
-               notification.token = ust_notification.token;
+               token = ust_notification.token;
                break;
        case LTTNG_DOMAIN_KERNEL:
-               notification.token = kernel_notification.token;
+               token = kernel_notification.token;
                break;
        default:
                abort();
        }
 
+       notification = lttng_event_notifier_notification_create(
+                       token, domain);
+end:
+       return notification;
+}
+
+int handle_notification_thread_event_notification(struct notification_thread_state *state,
+               int pipe,
+               enum lttng_domain_type domain)
+{
+       int ret;
+       enum lttng_trigger_status trigger_status;
+       struct cds_lfht_node *node;
+       struct cds_lfht_iter iter;
+       struct notification_trigger_tokens_ht_element *element;
+       struct lttng_evaluation *evaluation = NULL;
+       struct lttng_event_notifier_notification *notification = NULL;
+       enum action_executor_status executor_status;
+       struct notification_client_list *client_list = NULL;
+       const char *trigger_name;
+
+       notification = receive_notification(pipe, domain);
+       if (notification == NULL) {
+               ERR("[notification-thread] Error receiving notification from tracer (fd = %i, domain = %s)",
+                               pipe, lttng_domain_type_str(domain));
+               ret = -1;
+               goto end;
+       }
+
        /* Find triggers associated with this token. */
        rcu_read_lock();
        cds_lfht_lookup(state->trigger_tokens_ht,
-                       hash_key_u64(&notification.token, lttng_ht_seed),
-                       match_trigger_token, &notification.token, &iter);
+                       hash_key_u64(&notification->tracer_token, lttng_ht_seed),
+                       match_trigger_token, &notification->tracer_token, &iter);
        node = cds_lfht_iter_get_node(&iter);
-       if (caa_likely(!node)) {
+       if (caa_unlikely(!node)) {
                /*
-                * This is not an error, slow consumption of the pipe can lead
-                * to situations where a trigger is removed but we still get
-                * tracer notification matching to a previous trigger.
+                * This is not an error, slow consumption of the tracer
+                * notifications can lead to situations where a trigger is
+                * removed but we still get tracer notifications matching a
+                * trigger that no longer exists.
                 */
                ret = 0;
                goto end_unlock;
@@ -4250,17 +4261,94 @@ int handle_notification_thread_event_notification(struct notification_thread_sta
                        struct notification_trigger_tokens_ht_element,
                        node);
 
-       action = lttng_trigger_get_const_action(element->trigger);
-       action_type = lttng_action_get_type(action);
-       DBG("Received message from tracer event source: event source fd = %d, token = %" PRIu64 ", action type = '%s'",
-                       notification_pipe_read_fd, notification.token,
-                       lttng_action_type_string(action_type));
+       if (!lttng_trigger_should_fire(element->trigger)) {
+               ret = 0;
+               goto end_unlock;
+       }
 
-       /* TODO: Perform actions */
+       lttng_trigger_fire(element->trigger);
 
-       ret = 0;
+       trigger_status = lttng_trigger_get_name(element->trigger, &trigger_name);
+       assert(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+       evaluation = lttng_evaluation_event_rule_create(
+                       trigger_name);
+       if (evaluation == NULL) {
+               ERR("[notification-thread] Failed to create event rule hit evaluation while creating and enqueuing action executor job");
+               ret = -1;
+               goto end_unlock;
+       }
+
+       client_list = get_client_list_from_condition(state,
+                       lttng_trigger_get_const_condition(element->trigger));
+       executor_status = action_executor_enqueue(state->executor,
+                       element->trigger, evaluation, NULL, client_list);
+       switch (executor_status) {
+       case ACTION_EXECUTOR_STATUS_OK:
+               ret = 0;
+               break;
+       case ACTION_EXECUTOR_STATUS_OVERFLOW:
+       {
+               struct notification_client_list_element *client_list_element,
+                               *tmp;
+
+               /*
+                * Not a fatal error; this is expected and simply means the
+                * executor has too much work queued already.
+                */
+               ret = 0;
+
+               /* No clients subscribed to notifications for this trigger. */
+               if (!client_list) {
+                       break;
+               }
+
+               /* Warn clients that a notification (or more) was dropped. */
+               pthread_mutex_lock(&client_list->lock);
+               cds_list_for_each_entry_safe(client_list_element, tmp,
+                               &client_list->list, node) {
+                       enum client_transmission_status transmission_status;
+                       struct notification_client *client =
+                                       client_list_element->client;
+
+                       pthread_mutex_lock(&client->lock);
+                       ret = client_notification_overflow(client);
+                       if (ret) {
+                               /* Fatal error. */
+                               goto next_client;
+                       }
+
+                       transmission_status =
+                                       client_flush_outgoing_queue(client);
+                       ret = client_handle_transmission_status(
+                                       client, transmission_status, state);
+                       if (ret) {
+                               /* Fatal error. */
+                               goto next_client;
+                       }
+next_client:
+                       pthread_mutex_unlock(&client->lock);
+                       if (ret) {
+                               break;
+                       }
+               }
+
+               pthread_mutex_unlock(&client_list->lock);
+               break;
+       }
+       case ACTION_EXECUTOR_STATUS_ERROR:
+               /* Fatal error, shut down everything. */
+               ERR("Fatal error encoutered while enqueuing action to the action executor");
+               ret = -1;
+               goto end_unlock;
+       default:
+               /* Unhandled error. */
+               abort();
+       }
 
 end_unlock:
+       lttng_event_notifier_notification_destroy(notification);
+       notification_client_list_put(client_list);
        rcu_read_unlock();
 end:
        return ret;
@@ -4319,8 +4407,7 @@ int handle_notification_thread_channel_sample(
                 */
                DBG("[notification-thread] Received a sample for an unknown channel from consumerd, key = %" PRIu64 " in %s domain",
                                latest_sample.key.key,
-                               domain == LTTNG_DOMAIN_KERNEL ? "kernel" :
-                                       "user space");
+                               lttng_domain_type_str(domain));
                goto end_unlock;
        }
        channel_info = caa_container_of(node, struct channel_info,
This page took 0.026626 seconds and 4 git commands to generate.