X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=tests%2Fregression%2Ftools%2Fnotification%2Fnotification.c;h=d870a576fc85a5aff02f115bfffcdf37e60a643e;hp=3142e10c654b2dc60d03372abc62cec00c90e9e4;hb=acb988666ee674d3cf10d853cacf10b20f0e1624;hpb=a740a2427f8f8542db9ba02fe7bf2ae908e5f9af

diff --git a/tests/regression/tools/notification/notification.c b/tests/regression/tools/notification/notification.c
index 3142e10c6..d870a576f 100644
--- a/tests/regression/tools/notification/notification.c
+++ b/tests/regression/tools/notification/notification.c
@@ -32,12 +32,17 @@
 #include <lttng/condition/event-rule.h>
 #include <lttng/domain.h>
 #include <lttng/endpoint.h>
+#include <lttng/event-rule/kprobe.h>
+#include <lttng/event-rule/syscall.h>
 #include <lttng/event-rule/tracepoint.h>
+#include <lttng/event-rule/uprobe.h>
+#include <lttng/kernel-probe.h>
 #include <lttng/lttng-error.h>
 #include <lttng/lttng.h>
 #include <lttng/notification/channel.h>
 #include <lttng/notification/notification.h>
 #include <lttng/trigger/trigger.h>
+#include <lttng/userspace-probe.h>
 
 #include <tap/tap.h>
 
@@ -856,7 +861,7 @@ static void create_tracepoint_event_rule_trigger(const char *event_pattern,
 		const char *trigger_name,
 		const char *filter,
 		unsigned int exclusion_count,
-		const char **exclusions,
+		const char * const *exclusions,
 		enum lttng_domain_type domain_type,
 		struct lttng_condition **condition,
 		struct lttng_trigger **trigger)
@@ -1124,11 +1129,459 @@ static void test_tracepoint_event_rule_notification_filter(
 	lttng_condition_destroy(ctrl_condition);
 }
 
+static void test_tracepoint_event_rule_notification_exclusion(
+		enum lttng_domain_type domain_type)
+{
+	enum lttng_notification_channel_status nc_status;
+	struct lttng_condition *ctrl_condition = NULL, *condition = NULL;
+	struct lttng_notification_channel *notification_channel = NULL;
+	struct lttng_trigger *ctrl_trigger = NULL, *trigger = NULL;
+	int ctrl_count = 0, count = 0, i;
+	const char * const ctrl_trigger_name = "control_exclusion_trigger";
+	const char * const trigger_name = "exclusion_trigger";
+	const char * const pattern = "tp:tptest*";
+	const char * const exclusions[] = {
+		"tp:tptest2",
+		"tp:tptest3",
+		"tp:tptest4",
+		"tp:tptest5"
+	};
+
+	notification_channel = lttng_notification_channel_create(
+			lttng_session_daemon_notification_endpoint);
+	ok(notification_channel, "Notification channel object creation");
+
+	create_tracepoint_event_rule_trigger(pattern, ctrl_trigger_name, NULL,
+			0, NULL, domain_type, &ctrl_condition, &ctrl_trigger);
+
+	nc_status = lttng_notification_channel_subscribe(
+			notification_channel, ctrl_condition);
+	ok(nc_status == LTTNG_NOTIFICATION_CHANNEL_STATUS_OK,
+			"Subscribe to tracepoint event rule condition");
+
+	create_tracepoint_event_rule_trigger(pattern, trigger_name, NULL, 4,
+			exclusions, domain_type, &condition, &trigger);
+
+	nc_status = lttng_notification_channel_subscribe(
+			notification_channel, condition);
+	ok(nc_status == LTTNG_NOTIFICATION_CHANNEL_STATUS_OK,
+			"Subscribe to tracepoint event rule condition");
+
+	/*
+	 * We registered 2 notifications triggers, one with an exclusion and
+	 * one without (control).
+	 * - The trigger with an exclusion will fire once every iteration.
+	 * - The trigger without an exclusion will fire 5 times every
+	 *   iteration.
+	 *
+	 *   We should get 5 times as many notifications from the control
+	 *   trigger.
+	 */
+	resume_application();
+
+	/*
+	 * Get 6 notifications. We should get 1 for the regular trigger (with
+	 * the exclusion) and 5 from the control trigger. This works whatever
+	 * the order we receive the notifications.
+	 */
+	for (i = 0; i < 6; i++) {
+		char *name = get_next_notification_trigger_name(
+				notification_channel);
+
+		if (strcmp(ctrl_trigger_name, name) == 0) {
+			ctrl_count++;
+		} else if (strcmp(trigger_name, name) == 0) {
+			count++;
+		}
+
+		free(name);
+	}
+
+	ok(ctrl_count / 5 == count,
+			"Got 5 times as many control notif as of regular notif");
+
+	suspend_application();
+
+	lttng_unregister_trigger(trigger);
+	lttng_unregister_trigger(ctrl_trigger);
+	lttng_notification_channel_destroy(notification_channel);
+	lttng_trigger_destroy(trigger);
+	lttng_trigger_destroy(ctrl_trigger);
+	lttng_condition_destroy(condition);
+	lttng_condition_destroy(ctrl_condition);
+	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_event_rule_destroy(event_rule);
+	lttng_condition_destroy(condition);
+	lttng_kernel_probe_location_destroy(location);
+	return;
+}
+
+static void test_uprobe_event_rule_notification(
+		enum lttng_domain_type domain_type,
+		const char *testapp_path,
+		const char *test_symbol_name)
+{
+	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_userspace_probe_location *probe_location = NULL;
+	struct lttng_userspace_probe_location_lookup_method *lookup_method =
+			NULL;
+	struct lttng_condition *condition = NULL;
+	struct lttng_event_rule *event_rule = NULL;
+	struct lttng_action *action = NULL;
+	struct lttng_trigger *trigger = NULL;
+	const char * const trigger_name = "uprobe_trigger";
+
+	action = lttng_action_notify_create();
+	if (!action) {
+		fail("Failed to create notify action");
+		goto end;
+	}
+
+	lookup_method = lttng_userspace_probe_location_lookup_method_function_elf_create();
+	if (!lookup_method) {
+		fail("Setup error on userspace probe lookup method creation");
+		goto end;
+	}
+
+	probe_location = lttng_userspace_probe_location_function_create(
+			testapp_path, test_symbol_name, lookup_method);
+	if (!probe_location) {
+		fail("Failed to create userspace 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_uprobe_create();
+	ok(event_rule, "kprobe event rule object creation");
+
+	event_rule_status = lttng_event_rule_uprobe_set_location(
+			event_rule, probe_location);
+	ok(event_rule_status == LTTNG_EVENT_RULE_STATUS_OK,
+			"Setting uprobe event rule location");
+
+	event_rule_status = lttng_event_rule_uprobe_set_name(
+			event_rule, trigger_name);
+	ok(event_rule_status == LTTNG_EVENT_RULE_STATUS_OK,
+			"Setting uprobe 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 userspace 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 name to trigger '%s'", trigger_name);
+
+	ret = lttng_register_trigger(trigger);
+	if (ret) {
+		fail("Failed to register trigger with userspace 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_event_rule_destroy(event_rule);
+	lttng_condition_destroy(condition);
+	return;
+}
+
+static void test_syscall_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_event_rule *event_rule = NULL;
+	struct lttng_action *action = NULL;
+	struct lttng_trigger *trigger = NULL;
+	const char * const trigger_name = "syscall_trigger";
+	const char * const syscall_name = "openat";
+
+	action = lttng_action_notify_create();
+	if (!action) {
+		fail("Failed to create notify action");
+		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_syscall_create();
+	ok(event_rule, "syscall event rule object creation");
+
+	event_rule_status = lttng_event_rule_syscall_set_pattern(
+			event_rule, syscall_name);
+	ok(event_rule_status == LTTNG_EVENT_RULE_STATUS_OK,
+			"Setting syscall event rule pattern: '%s'", syscall_name);
+
+	condition = lttng_condition_event_rule_create(event_rule);
+	ok(condition, "Condition syscall event rule object creation");
+
+	/* Register the trigger for condition. */
+	trigger = lttng_trigger_create(condition, action);
+	if (!trigger) {
+		fail("Failed to create trigger with syscall 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 name to trigger '%s'", trigger_name);
+
+	ret = lttng_register_trigger(trigger);
+	if (ret) {
+		fail("Failed to register trigger with syscall 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);
+	return;
+}
+
+static void test_syscall_event_rule_notification_filter(
+		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_event_rule *event_rule = NULL;
+	struct lttng_action *action = NULL;
+	struct lttng_trigger *trigger = NULL;
+	const char * const trigger_name = "syscall_trigger";
+	const char * const syscall_name = "openat";
+	const char * const filter_pattern = "filename == \"/proc/cpuinfo\"";
+
+	action = lttng_action_notify_create();
+	if (!action) {
+		fail("Failed to create notify action");
+		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_syscall_create();
+	ok(event_rule, "syscall event rule object creation");
+
+	event_rule_status = lttng_event_rule_syscall_set_pattern(
+			event_rule, syscall_name);
+	ok(event_rule_status == LTTNG_EVENT_RULE_STATUS_OK,
+			"Setting syscall event rule pattern: '%s'", syscall_name);
+
+	event_rule_status = lttng_event_rule_syscall_set_filter(
+			event_rule, filter_pattern);
+	ok(event_rule_status == LTTNG_EVENT_RULE_STATUS_OK,
+			"Setting filter: '%s'", filter_pattern);
+
+	condition = lttng_condition_event_rule_create(event_rule);
+	ok(condition, "Condition event rule object creation");
+
+	/* Register the triggers for condition */
+	trigger = lttng_trigger_create(condition, action);
+	if (!trigger) {
+		fail("Failed to create trigger with syscall filtering 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 name to trigger '%s'", trigger_name);
+
+	ret = lttng_register_trigger(trigger);
+	if (ret) {
+		fail("Failed to register trigger with syscall filtering 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 < 3; 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_unregister_trigger(trigger);
+	lttng_notification_channel_destroy(notification_channel);
+	lttng_trigger_destroy(trigger);
+	lttng_event_rule_destroy(event_rule);
+	lttng_condition_destroy(condition);
+	return;
+}
+
 int main(int argc, const char *argv[])
 {
 	int test_scenario;
-	const char *session_name = NULL;
-	const char *channel_name = NULL;
 	const char *domain_type_string = NULL;
 	enum lttng_domain_type domain_type = LTTNG_DOMAIN_NONE;
 
@@ -1180,6 +1633,7 @@ int main(int argc, const char *argv[])
 	}
 	case 2:
 	{
+		const char *session_name, *channel_name;
 		/* Test cases that need a tracing session enabled. */
 		plan_tests(99);
 
@@ -1220,6 +1674,81 @@ int main(int argc, const char *argv[])
 				domain_type, argv);
 		break;
 	}
+	case 3:
+	{
+		/*
+		 * Test cases that need a test app with more than one event
+		 * type.
+		 */
+		plan_tests(19);
+
+		/*
+		 * At the moment, the only test case of this scenario is
+		 * exclusion which is only supported by UST.
+		 */
+		assert(domain_type == LTTNG_DOMAIN_UST);
+		diag("Test tracepoint event rule notifications with exclusion for domain %s",
+				domain_type_string);
+		test_tracepoint_event_rule_notification_exclusion(domain_type);
+
+		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;
+	}
+	case 5:
+	{
+		plan_tests(19);
+		/* Test cases that need the kernel tracer. */
+		assert(domain_type == LTTNG_DOMAIN_KERNEL);
+
+		diag("Test syscall event rule notifications for domain %s",
+				domain_type_string);
+
+		test_syscall_event_rule_notification(domain_type);
+
+		diag("Test syscall filtering event rule notifications for domain %s",
+				domain_type_string);
+
+		test_syscall_event_rule_notification_filter(domain_type);
+
+		break;
+	}
+	case 6:
+	{
+		const char *testapp_path, *test_symbol_name;
+
+		plan_tests(10);
+
+		if (argc < 7) {
+			fail("Missing parameter for tests to run %d", argc);
+			goto error;
+		}
+
+		testapp_path = argv[5];
+		test_symbol_name = argv[6];
+		/* Test cases that need the kernel tracer. */
+		assert(domain_type == LTTNG_DOMAIN_KERNEL);
+
+		diag("Test userspace-probe event rule notifications for domain %s",
+				domain_type_string);
+
+		test_uprobe_event_rule_notification(
+				domain_type, testapp_path, test_symbol_name);
+
+		break;
+	}
 	default:
 		abort();
 	}