]>
Commit | Line | Data |
---|---|---|
52ad194e | 1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
663996b3 | 2 | |
663996b3 MS |
3 | #include "bus-match.h" |
4 | #include "bus-message.h" | |
60f067b4 | 5 | #include "bus-slot.h" |
db2df898 MP |
6 | #include "bus-util.h" |
7 | #include "log.h" | |
8 | #include "macro.h" | |
bb4f798a | 9 | #include "memory-util.h" |
6e866b33 | 10 | #include "tests.h" |
663996b3 MS |
11 | |
12 | static bool mask[32]; | |
13 | ||
e3bff60a | 14 | static int filter(sd_bus_message *m, void *userdata, sd_bus_error *ret_error) { |
f47781d8 | 15 | log_info("Ran %u", PTR_TO_UINT(userdata)); |
e735f4d4 | 16 | assert_se(PTR_TO_UINT(userdata) < ELEMENTSOF(mask)); |
f47781d8 | 17 | mask[PTR_TO_UINT(userdata)] = true; |
663996b3 MS |
18 | return 0; |
19 | } | |
20 | ||
21 | static bool mask_contains(unsigned a[], unsigned n) { | |
22 | unsigned i, j; | |
23 | ||
24 | for (i = 0; i < ELEMENTSOF(mask); i++) { | |
25 | bool found = false; | |
26 | ||
27 | for (j = 0; j < n; j++) | |
28 | if (a[j] == i) { | |
29 | found = true; | |
30 | break; | |
31 | } | |
32 | ||
33 | if (found != mask[i]) | |
34 | return false; | |
35 | } | |
36 | ||
37 | return true; | |
38 | } | |
39 | ||
60f067b4 | 40 | static int match_add(sd_bus_slot *slots, struct bus_match_node *root, const char *match, int value) { |
14228c0d MB |
41 | struct bus_match_component *components = NULL; |
42 | unsigned n_components = 0; | |
60f067b4 | 43 | sd_bus_slot *s; |
14228c0d MB |
44 | int r; |
45 | ||
60f067b4 JS |
46 | s = slots + value; |
47 | zero(*s); | |
14228c0d MB |
48 | |
49 | r = bus_match_parse(match, &components, &n_components); | |
50 | if (r < 0) | |
51 | return r; | |
52 | ||
60f067b4 JS |
53 | s->userdata = INT_TO_PTR(value); |
54 | s->match_callback.callback = filter; | |
55 | ||
56 | r = bus_match_add(root, components, n_components, &s->match_callback); | |
14228c0d MB |
57 | bus_match_parse_free(components, n_components); |
58 | ||
59 | return r; | |
60 | } | |
61 | ||
86f210e9 MP |
62 | static void test_match_scope(const char *match, enum bus_match_scope scope) { |
63 | struct bus_match_component *components = NULL; | |
64 | unsigned n_components = 0; | |
65 | ||
66 | assert_se(bus_match_parse(match, &components, &n_components) >= 0); | |
67 | assert_se(bus_match_get_scope(components, n_components) == scope); | |
68 | bus_match_parse_free(components, n_components); | |
69 | } | |
70 | ||
663996b3 | 71 | int main(int argc, char *argv[]) { |
60f067b4 JS |
72 | struct bus_match_node root = { |
73 | .type = BUS_MATCH_ROOT, | |
74 | }; | |
75 | ||
4c89c718 MP |
76 | _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL; |
77 | _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL; | |
663996b3 | 78 | enum bus_match_node_type i; |
f47781d8 | 79 | sd_bus_slot slots[19]; |
60f067b4 | 80 | int r; |
663996b3 | 81 | |
6e866b33 MB |
82 | test_setup_logging(LOG_INFO); |
83 | ||
f5e65279 | 84 | r = sd_bus_open_user(&bus); |
60f067b4 | 85 | if (r < 0) |
6e866b33 MB |
86 | r = sd_bus_open_system(&bus); |
87 | if (r < 0) | |
88 | return log_tests_skipped("Failed to connect to bus"); | |
60f067b4 JS |
89 | |
90 | assert_se(match_add(slots, &root, "arg2='wal\\'do',sender='foo',type='signal',interface='bar.x',", 1) >= 0); | |
91 | assert_se(match_add(slots, &root, "arg2='wal\\'do2',sender='foo',type='signal',interface='bar.x',", 2) >= 0); | |
92 | assert_se(match_add(slots, &root, "arg3='test',sender='foo',type='signal',interface='bar.x',", 3) >= 0); | |
93 | assert_se(match_add(slots, &root, "arg3='test',sender='foo',type='method_call',interface='bar.x',", 4) >= 0); | |
94 | assert_se(match_add(slots, &root, "", 5) >= 0); | |
95 | assert_se(match_add(slots, &root, "interface='quux.x'", 6) >= 0); | |
96 | assert_se(match_add(slots, &root, "interface='bar.x'", 7) >= 0); | |
97 | assert_se(match_add(slots, &root, "member='waldo',path='/foo/bar'", 8) >= 0); | |
98 | assert_se(match_add(slots, &root, "path='/foo/bar'", 9) >= 0); | |
99 | assert_se(match_add(slots, &root, "path_namespace='/foo'", 10) >= 0); | |
100 | assert_se(match_add(slots, &root, "path_namespace='/foo/quux'", 11) >= 0); | |
101 | assert_se(match_add(slots, &root, "arg1='two'", 12) >= 0); | |
102 | assert_se(match_add(slots, &root, "member='waldo',arg2path='/prefix/'", 13) >= 0); | |
103 | assert_se(match_add(slots, &root, "member=waldo,path='/foo/bar',arg3namespace='prefix'", 14) >= 0); | |
13d276d0 MP |
104 | assert_se(match_add(slots, &root, "arg4has='pi'", 15) >= 0); |
105 | assert_se(match_add(slots, &root, "arg4has='pa'", 16) >= 0); | |
106 | assert_se(match_add(slots, &root, "arg4has='po'", 17) >= 0); | |
107 | assert_se(match_add(slots, &root, "arg4='pi'", 18) >= 0); | |
663996b3 MS |
108 | |
109 | bus_match_dump(&root, 0); | |
110 | ||
60f067b4 | 111 | assert_se(sd_bus_message_new_signal(bus, &m, "/foo/bar", "bar.x", "waldo") >= 0); |
f47781d8 | 112 | assert_se(sd_bus_message_append(m, "ssssas", "one", "two", "/prefix/three", "prefix.four", 3, "pi", "pa", "po") >= 0); |
52ad194e | 113 | assert_se(sd_bus_message_seal(m, 1, 0) >= 0); |
663996b3 MS |
114 | |
115 | zero(mask); | |
14228c0d | 116 | assert_se(bus_match_run(NULL, &root, m) == 0); |
f47781d8 | 117 | assert_se(mask_contains((unsigned[]) { 9, 8, 7, 5, 10, 12, 13, 14, 15, 16, 17 }, 11)); |
663996b3 | 118 | |
60f067b4 JS |
119 | assert_se(bus_match_remove(&root, &slots[8].match_callback) >= 0); |
120 | assert_se(bus_match_remove(&root, &slots[13].match_callback) >= 0); | |
663996b3 MS |
121 | |
122 | bus_match_dump(&root, 0); | |
123 | ||
124 | zero(mask); | |
14228c0d | 125 | assert_se(bus_match_run(NULL, &root, m) == 0); |
f47781d8 | 126 | assert_se(mask_contains((unsigned[]) { 9, 5, 10, 12, 14, 7, 15, 16, 17 }, 9)); |
663996b3 MS |
127 | |
128 | for (i = 0; i < _BUS_MATCH_NODE_TYPE_MAX; i++) { | |
129 | char buf[32]; | |
130 | const char *x; | |
131 | ||
132 | assert_se(x = bus_match_node_type_to_string(i, buf, sizeof(buf))); | |
133 | ||
134 | if (i >= BUS_MATCH_MESSAGE_TYPE) | |
135 | assert_se(bus_match_node_type_from_string(x, strlen(x)) == i); | |
136 | } | |
137 | ||
138 | bus_match_free(&root); | |
139 | ||
86f210e9 MP |
140 | test_match_scope("interface='foobar'", BUS_MATCH_GENERIC); |
141 | test_match_scope("", BUS_MATCH_GENERIC); | |
142 | test_match_scope("interface='org.freedesktop.DBus.Local'", BUS_MATCH_LOCAL); | |
143 | test_match_scope("sender='org.freedesktop.DBus.Local'", BUS_MATCH_LOCAL); | |
144 | test_match_scope("member='gurke',path='/org/freedesktop/DBus/Local'", BUS_MATCH_LOCAL); | |
145 | test_match_scope("arg2='piep',sender='org.freedesktop.DBus',member='waldo'", BUS_MATCH_DRIVER); | |
146 | ||
663996b3 MS |
147 | return 0; |
148 | } |