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