Tests: triggers: Add kprobe event rule condition kernel tests

[lttng-tools.git] / tests / regression / tools / notification / notification.c

diff --git a/tests/regression/tools/notification/notification.c b/tests/regression/tools/notification/notification.c
index f5e44210ab7c2898e5815654fbaaba201793b4f5..6b70840c738091cdf88d29032836e219be0d8634 100644 (file)
--- a/tests/regression/tools/notification/notification.c
+++ b/tests/regression/tools/notification/notification.c
@@ -32,7 +32,9 @@
 #include <lttng/condition/event-rule.h>
 #include <lttng/domain.h>
 #include <lttng/endpoint.h>
 #include <lttng/condition/event-rule.h>
 #include <lttng/domain.h>
 #include <lttng/endpoint.h>

+#include <lttng/event-rule/kprobe.h>

 #include <lttng/event-rule/tracepoint.h>
 #include <lttng/event-rule/tracepoint.h>

+#include <lttng/kernel-probe.h>

 #include <lttng/lttng-error.h>
 #include <lttng/lttng.h>
 #include <lttng/notification/channel.h>
 #include <lttng/lttng-error.h>
 #include <lttng/lttng.h>
 #include <lttng/notification/channel.h>


@@ -1207,6 +1209,100 @@ static void test_tracepoint_event_rule_notification_exclusion(
        return;
 }
 
        return;
 }
 

+static void test_kprobe_event_rule_notification(
+               enum lttng_domain_type domain_type)
+{
+       int i, ret;
+       const int notification_count = 3;
+       enum lttng_notification_channel_status nc_status;
+       enum lttng_event_rule_status event_rule_status;
+       enum lttng_trigger_status trigger_status;
+       struct lttng_notification_channel *notification_channel = NULL;
+       struct lttng_condition *condition = NULL;
+       struct lttng_kernel_probe_location *location = NULL;
+       struct lttng_event_rule *event_rule = NULL;
+       struct lttng_action *action = NULL;
+       struct lttng_trigger *trigger = NULL;
+       const char * const trigger_name = "kprobe_trigger";
+       const char * const symbol_name = "lttng_test_filter_event_write";
+
+       action = lttng_action_notify_create();
+       if (!action) {
+               fail("Failed to create notify action");
+               goto end;
+       }
+
+       location = lttng_kernel_probe_location_symbol_create(symbol_name, 0);
+       if (!location) {
+               fail("Failed to create kernel probe location");
+               goto end;
+       }
+
+       notification_channel = lttng_notification_channel_create(
+                       lttng_session_daemon_notification_endpoint);
+       ok(notification_channel, "Notification channel object creation");
+
+       event_rule = lttng_event_rule_kprobe_create();
+       ok(event_rule, "kprobe event rule object creation");
+
+       event_rule_status = lttng_event_rule_kprobe_set_location(
+                       event_rule, location);
+       ok(event_rule_status == LTTNG_EVENT_RULE_STATUS_OK,
+                       "Setting kprobe event rule location: '%s'", symbol_name);
+
+       event_rule_status = lttng_event_rule_kprobe_set_name(
+                       event_rule, trigger_name);
+       ok(event_rule_status == LTTNG_EVENT_RULE_STATUS_OK,
+                       "Setting kprobe event rule name: '%s'", trigger_name);
+
+       condition = lttng_condition_event_rule_create(event_rule);
+       ok(condition, "Condition event rule object creation");
+
+       /* Register the trigger for condition. */
+       trigger = lttng_trigger_create(condition, action);
+       if (!trigger) {
+               fail("Failed to create trigger with kernel probe event rule condition and notify action");
+               goto end;
+       }
+
+       trigger_status = lttng_trigger_set_name(trigger, trigger_name);
+       ok(trigger_status == LTTNG_TRIGGER_STATUS_OK,
+                       "Setting trigger name to '%s'", trigger_name);
+
+       ret = lttng_register_trigger(trigger);
+       if (ret) {
+               fail("Failed to register trigger with kernel probe event rule condition and notify action");
+               goto end;
+       }
+
+       nc_status = lttng_notification_channel_subscribe(
+                       notification_channel, condition);
+       ok(nc_status == LTTNG_NOTIFICATION_CHANNEL_STATUS_OK,
+                       "Subscribe to tracepoint event rule condition");
+
+       resume_application();
+
+       for (i = 0; i < notification_count; i++) {
+               char *name = get_next_notification_trigger_name(
+                               notification_channel);
+
+               ok(strcmp(trigger_name, name) == 0,
+                               "Received notification for the expected trigger name: '%s' (%d/%d)",
+                               trigger_name, i + 1, notification_count);
+               free(name);
+       }
+
+end:
+       suspend_application();
+       lttng_notification_channel_destroy(notification_channel);
+       lttng_unregister_trigger(trigger);
+       lttng_trigger_destroy(trigger);
+       lttng_action_destroy(action);
+       lttng_condition_destroy(condition);
+       lttng_kernel_probe_location_destroy(location);
+       return;
+}
+

 int main(int argc, const char *argv[])
 {
        int test_scenario;
 int main(int argc, const char *argv[])
 {
        int test_scenario;


@@ -1322,6 +1418,19 @@ int main(int argc, const char *argv[])
 
                break;
        }
 
                break;
        }

+       case 4:
+       {
+               plan_tests(10);
+
+               /* Test cases that need the kernel tracer. */
+               assert(domain_type == LTTNG_DOMAIN_KERNEL);
+               diag("Test kprobe event rule notifications for domain %s",
+                               domain_type_string);
+
+               test_kprobe_event_rule_notification(domain_type);
+
+               break;
+       }

        default:
                abort();
        }
        default:
                abort();
        }