X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=tests%2Fregression%2Ftools%2Fnotification%2Fnotification.c;h=6b70840c738091cdf88d29032836e219be0d8634;hp=f5e44210ab7c2898e5815654fbaaba201793b4f5;hb=51704c8eb4792d8ff748ffdbc2118471a9a67e6c;hpb=3e4ee3b37bab181a2da93b09e4714af5dbbc29c0

diff --git a/tests/regression/tools/notification/notification.c b/tests/regression/tools/notification/notification.c
index f5e44210a..6b70840c7 100644
--- 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/event-rule/kprobe.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>
@@ -1207,6 +1209,100 @@ static void test_tracepoint_event_rule_notification_exclusion(
 	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;
@@ -1322,6 +1418,19 @@ int main(int argc, const char *argv[])
 
 		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();
 	}