]> git.ipfire.org Git - thirdparty/systemd.git/blob - src/libsystemd/sd-bus/test-bus-objects.c
tree-wide: use UINT64_MAX or friends
[thirdparty/systemd.git] / src / libsystemd / sd-bus / test-bus-objects.c
1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
2
3 #include <pthread.h>
4 #include <stdlib.h>
5
6 #include "sd-bus.h"
7
8 #include "alloc-util.h"
9 #include "bus-dump.h"
10 #include "bus-internal.h"
11 #include "bus-message.h"
12 #include "log.h"
13 #include "macro.h"
14 #include "strv.h"
15 #include "util.h"
16
17 struct context {
18 int fds[2];
19 bool quit;
20 char *something;
21 char *automatic_string_property;
22 uint32_t automatic_integer_property;
23 };
24
25 static int something_handler(sd_bus_message *m, void *userdata, sd_bus_error *error) {
26 struct context *c = userdata;
27 const char *s;
28 char *n = NULL;
29 int r;
30
31 r = sd_bus_message_read(m, "s", &s);
32 assert_se(r > 0);
33
34 n = strjoin("<<<", s, ">>>");
35 assert_se(n);
36
37 free(c->something);
38 c->something = n;
39
40 log_info("AlterSomething() called, got %s, returning %s", s, n);
41
42 /* This should fail, since the return type doesn't match */
43 assert_se(sd_bus_reply_method_return(m, "u", 4711) == -ENOMSG);
44
45 r = sd_bus_reply_method_return(m, "s", n);
46 assert_se(r >= 0);
47
48 return 1;
49 }
50
51 static int exit_handler(sd_bus_message *m, void *userdata, sd_bus_error *error) {
52 struct context *c = userdata;
53 int r;
54
55 c->quit = true;
56
57 log_info("Exit called");
58
59 r = sd_bus_reply_method_return(m, "");
60 assert_se(r >= 0);
61
62 return 1;
63 }
64
65 static int get_handler(sd_bus *bus, const char *path, const char *interface, const char *property, sd_bus_message *reply, void *userdata, sd_bus_error *error) {
66 struct context *c = userdata;
67 int r;
68
69 log_info("property get for %s called, returning \"%s\".", property, c->something);
70
71 r = sd_bus_message_append(reply, "s", c->something);
72 assert_se(r >= 0);
73
74 return 1;
75 }
76
77 static int set_handler(sd_bus *bus, const char *path, const char *interface, const char *property, sd_bus_message *value, void *userdata, sd_bus_error *error) {
78 struct context *c = userdata;
79 const char *s;
80 char *n;
81 int r;
82
83 log_info("property set for %s called", property);
84
85 r = sd_bus_message_read(value, "s", &s);
86 assert_se(r >= 0);
87
88 n = strdup(s);
89 assert_se(n);
90
91 free(c->something);
92 c->something = n;
93
94 return 1;
95 }
96
97 static int value_handler(sd_bus *bus, const char *path, const char *interface, const char *property, sd_bus_message *reply, void *userdata, sd_bus_error *error) {
98 _cleanup_free_ char *s = NULL;
99 const char *x;
100 int r;
101
102 assert_se(asprintf(&s, "object %p, path %s", userdata, path) >= 0);
103 r = sd_bus_message_append(reply, "s", s);
104 assert_se(r >= 0);
105
106 assert_se(x = startswith(path, "/value/"));
107
108 assert_se(PTR_TO_UINT(userdata) == 30);
109
110 return 1;
111 }
112
113 static int notify_test(sd_bus_message *m, void *userdata, sd_bus_error *error) {
114 int r;
115
116 assert_se(sd_bus_emit_properties_changed(sd_bus_message_get_bus(m), m->path, "org.freedesktop.systemd.ValueTest", "Value", NULL) >= 0);
117
118 r = sd_bus_reply_method_return(m, NULL);
119 assert_se(r >= 0);
120
121 return 1;
122 }
123
124 static int notify_test2(sd_bus_message *m, void *userdata, sd_bus_error *error) {
125 int r;
126
127 assert_se(sd_bus_emit_properties_changed_strv(sd_bus_message_get_bus(m), m->path, "org.freedesktop.systemd.ValueTest", NULL) >= 0);
128
129 r = sd_bus_reply_method_return(m, NULL);
130 assert_se(r >= 0);
131
132 return 1;
133 }
134
135 static int emit_interfaces_added(sd_bus_message *m, void *userdata, sd_bus_error *error) {
136 int r;
137
138 assert_se(sd_bus_emit_interfaces_added(sd_bus_message_get_bus(m), "/value/a/x", "org.freedesktop.systemd.ValueTest", NULL) >= 0);
139
140 r = sd_bus_reply_method_return(m, NULL);
141 assert_se(r >= 0);
142
143 return 1;
144 }
145
146 static int emit_interfaces_removed(sd_bus_message *m, void *userdata, sd_bus_error *error) {
147 int r;
148
149 assert_se(sd_bus_emit_interfaces_removed(sd_bus_message_get_bus(m), "/value/a/x", "org.freedesktop.systemd.ValueTest", NULL) >= 0);
150
151 r = sd_bus_reply_method_return(m, NULL);
152 assert_se(r >= 0);
153
154 return 1;
155 }
156
157 static int emit_object_added(sd_bus_message *m, void *userdata, sd_bus_error *error) {
158 int r;
159
160 assert_se(sd_bus_emit_object_added(sd_bus_message_get_bus(m), "/value/a/x") >= 0);
161
162 r = sd_bus_reply_method_return(m, NULL);
163 assert_se(r >= 0);
164
165 return 1;
166 }
167
168 static int emit_object_removed(sd_bus_message *m, void *userdata, sd_bus_error *error) {
169 int r;
170
171 assert_se(sd_bus_emit_object_removed(sd_bus_message_get_bus(m), "/value/a/x") >= 0);
172
173 r = sd_bus_reply_method_return(m, NULL);
174 assert_se(r >= 0);
175
176 return 1;
177 }
178
179 static const sd_bus_vtable vtable[] = {
180 SD_BUS_VTABLE_START(0),
181 SD_BUS_METHOD("AlterSomething", "s", "s", something_handler, 0),
182 SD_BUS_METHOD("Exit", "", "", exit_handler, 0),
183 SD_BUS_WRITABLE_PROPERTY("Something", "s", get_handler, set_handler, 0, 0),
184 SD_BUS_WRITABLE_PROPERTY("AutomaticStringProperty", "s", NULL, NULL, offsetof(struct context, automatic_string_property), 0),
185 SD_BUS_WRITABLE_PROPERTY("AutomaticIntegerProperty", "u", NULL, NULL, offsetof(struct context, automatic_integer_property), 0),
186 SD_BUS_METHOD("NoOperation", NULL, NULL, NULL, 0),
187 SD_BUS_METHOD("EmitInterfacesAdded", NULL, NULL, emit_interfaces_added, 0),
188 SD_BUS_METHOD("EmitInterfacesRemoved", NULL, NULL, emit_interfaces_removed, 0),
189 SD_BUS_METHOD("EmitObjectAdded", NULL, NULL, emit_object_added, 0),
190 SD_BUS_METHOD("EmitObjectRemoved", NULL, NULL, emit_object_removed, 0),
191 SD_BUS_VTABLE_END
192 };
193
194 static const sd_bus_vtable vtable2[] = {
195 SD_BUS_VTABLE_START(0),
196 SD_BUS_METHOD("NotifyTest", "", "", notify_test, 0),
197 SD_BUS_METHOD("NotifyTest2", "", "", notify_test2, 0),
198 SD_BUS_PROPERTY("Value", "s", value_handler, 10, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
199 SD_BUS_PROPERTY("Value2", "s", value_handler, 10, SD_BUS_VTABLE_PROPERTY_EMITS_INVALIDATION),
200 SD_BUS_PROPERTY("Value3", "s", value_handler, 10, SD_BUS_VTABLE_PROPERTY_CONST),
201 SD_BUS_PROPERTY("Value4", "s", value_handler, 10, 0),
202 SD_BUS_PROPERTY("AnExplicitProperty", "s", NULL, offsetof(struct context, something), SD_BUS_VTABLE_PROPERTY_EXPLICIT|SD_BUS_VTABLE_PROPERTY_EMITS_INVALIDATION),
203 SD_BUS_VTABLE_END
204 };
205
206 static int enumerator_callback(sd_bus *bus, const char *path, void *userdata, char ***nodes, sd_bus_error *error) {
207
208 if (object_path_startswith("/value", path))
209 assert_se(*nodes = strv_new("/value/a", "/value/b", "/value/c"));
210
211 return 1;
212 }
213
214 static int enumerator2_callback(sd_bus *bus, const char *path, void *userdata, char ***nodes, sd_bus_error *error) {
215
216 if (object_path_startswith("/value/a", path))
217 assert_se(*nodes = strv_new("/value/a/x", "/value/a/y", "/value/a/z"));
218
219 return 1;
220 }
221
222 static void *server(void *p) {
223 struct context *c = p;
224 sd_bus *bus = NULL;
225 sd_id128_t id;
226 int r;
227
228 c->quit = false;
229
230 assert_se(sd_id128_randomize(&id) >= 0);
231
232 assert_se(sd_bus_new(&bus) >= 0);
233 assert_se(sd_bus_set_fd(bus, c->fds[0], c->fds[0]) >= 0);
234 assert_se(sd_bus_set_server(bus, 1, id) >= 0);
235
236 assert_se(sd_bus_add_object_vtable(bus, NULL, "/foo", "org.freedesktop.systemd.test", vtable, c) >= 0);
237 assert_se(sd_bus_add_object_vtable(bus, NULL, "/foo", "org.freedesktop.systemd.test2", vtable, c) >= 0);
238 assert_se(sd_bus_add_fallback_vtable(bus, NULL, "/value", "org.freedesktop.systemd.ValueTest", vtable2, NULL, UINT_TO_PTR(20)) >= 0);
239 assert_se(sd_bus_add_node_enumerator(bus, NULL, "/value", enumerator_callback, NULL) >= 0);
240 assert_se(sd_bus_add_node_enumerator(bus, NULL, "/value/a", enumerator2_callback, NULL) >= 0);
241 assert_se(sd_bus_add_object_manager(bus, NULL, "/value") >= 0);
242 assert_se(sd_bus_add_object_manager(bus, NULL, "/value/a") >= 0);
243
244 assert_se(sd_bus_start(bus) >= 0);
245
246 log_error("Entering event loop on server");
247
248 while (!c->quit) {
249 log_error("Loop!");
250
251 r = sd_bus_process(bus, NULL);
252 if (r < 0) {
253 log_error_errno(r, "Failed to process requests: %m");
254 goto fail;
255 }
256
257 if (r == 0) {
258 r = sd_bus_wait(bus, UINT64_MAX);
259 if (r < 0) {
260 log_error_errno(r, "Failed to wait: %m");
261 goto fail;
262 }
263
264 continue;
265 }
266 }
267
268 r = 0;
269
270 fail:
271 if (bus) {
272 sd_bus_flush(bus);
273 sd_bus_unref(bus);
274 }
275
276 return INT_TO_PTR(r);
277 }
278
279 static int client(struct context *c) {
280 _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
281 _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL;
282 _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
283 const char *s;
284 int r;
285
286 assert_se(sd_bus_new(&bus) >= 0);
287 assert_se(sd_bus_set_fd(bus, c->fds[1], c->fds[1]) >= 0);
288 assert_se(sd_bus_start(bus) >= 0);
289
290 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "NoOperation", &error, NULL, NULL);
291 assert_se(r >= 0);
292
293 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "AlterSomething", &error, &reply, "s", "hallo");
294 assert_se(r >= 0);
295
296 r = sd_bus_message_read(reply, "s", &s);
297 assert_se(r >= 0);
298 assert_se(streq(s, "<<<hallo>>>"));
299
300 reply = sd_bus_message_unref(reply);
301
302 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Doesntexist", &error, &reply, "");
303 assert_se(r < 0);
304 assert_se(sd_bus_error_has_name(&error, SD_BUS_ERROR_UNKNOWN_METHOD));
305
306 sd_bus_error_free(&error);
307
308 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Doesntexist", &error, &reply, NULL); /* NULL and "" are equivalent */
309 assert_se(r < 0);
310 assert_se(sd_bus_error_has_name(&error, SD_BUS_ERROR_UNKNOWN_METHOD));
311
312 sd_bus_error_free(&error);
313
314 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "AlterSomething", &error, &reply, "as", 1, "hallo");
315 assert_se(r < 0);
316 assert_se(sd_bus_error_has_name(&error, SD_BUS_ERROR_INVALID_ARGS));
317
318 sd_bus_error_free(&error);
319
320 r = sd_bus_get_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Something", &error, &reply, "s");
321 assert_se(r >= 0);
322
323 r = sd_bus_message_read(reply, "s", &s);
324 assert_se(r >= 0);
325 assert_se(streq(s, "<<<hallo>>>"));
326
327 reply = sd_bus_message_unref(reply);
328
329 r = sd_bus_set_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Something", &error, "s", "test");
330 assert_se(r >= 0);
331
332 r = sd_bus_get_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Something", &error, &reply, "s");
333 assert_se(r >= 0);
334
335 r = sd_bus_message_read(reply, "s", &s);
336 assert_se(r >= 0);
337 assert_se(streq(s, "test"));
338
339 reply = sd_bus_message_unref(reply);
340
341 r = sd_bus_set_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "AutomaticIntegerProperty", &error, "u", 815);
342 assert_se(r >= 0);
343
344 assert_se(c->automatic_integer_property == 815);
345
346 r = sd_bus_set_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "AutomaticStringProperty", &error, "s", "Du Dödel, Du!");
347 assert_se(r >= 0);
348
349 assert_se(streq(c->automatic_string_property, "Du Dödel, Du!"));
350
351 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.DBus.Introspectable", "Introspect", &error, &reply, "");
352 assert_se(r >= 0);
353
354 r = sd_bus_message_read(reply, "s", &s);
355 assert_se(r >= 0);
356 fputs(s, stdout);
357
358 reply = sd_bus_message_unref(reply);
359
360 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.DBus.Introspectable", "Introspect", &error, &reply, NULL); /* NULL and "" are equivalent */
361 assert_se(r >= 0);
362
363 r = sd_bus_message_read(reply, "s", &s);
364 assert_se(r >= 0);
365 fputs(s, stdout);
366
367 reply = sd_bus_message_unref(reply);
368
369 r = sd_bus_get_property(bus, "org.freedesktop.systemd.test", "/value/xuzz", "org.freedesktop.systemd.ValueTest", "Value", &error, &reply, "s");
370 assert_se(r >= 0);
371
372 r = sd_bus_message_read(reply, "s", &s);
373 assert_se(r >= 0);
374 log_info("read %s", s);
375
376 reply = sd_bus_message_unref(reply);
377
378 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/", "org.freedesktop.DBus.Introspectable", "Introspect", &error, &reply, NULL);
379 assert_se(r >= 0);
380
381 r = sd_bus_message_read(reply, "s", &s);
382 assert_se(r >= 0);
383 fputs(s, stdout);
384
385 reply = sd_bus_message_unref(reply);
386
387 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value", "org.freedesktop.DBus.Introspectable", "Introspect", &error, &reply, NULL);
388 assert_se(r >= 0);
389
390 r = sd_bus_message_read(reply, "s", &s);
391 assert_se(r >= 0);
392 fputs(s, stdout);
393
394 reply = sd_bus_message_unref(reply);
395
396 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value/a", "org.freedesktop.DBus.Introspectable", "Introspect", &error, &reply, NULL);
397 assert_se(r >= 0);
398
399 r = sd_bus_message_read(reply, "s", &s);
400 assert_se(r >= 0);
401 fputs(s, stdout);
402
403 reply = sd_bus_message_unref(reply);
404
405 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.DBus.Properties", "GetAll", &error, &reply, "s", NULL);
406 assert_se(r >= 0);
407
408 sd_bus_message_dump(reply, stdout, SD_BUS_MESSAGE_DUMP_WITH_HEADER);
409
410 reply = sd_bus_message_unref(reply);
411
412 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value/a", "org.freedesktop.DBus.Properties", "GetAll", &error, &reply, "s", "org.freedesktop.systemd.ValueTest2");
413 assert_se(r < 0);
414 assert_se(sd_bus_error_has_name(&error, SD_BUS_ERROR_UNKNOWN_INTERFACE));
415 sd_bus_error_free(&error);
416
417 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.DBus.ObjectManager", "GetManagedObjects", &error, &reply, NULL);
418 assert_se(r < 0);
419 assert_se(sd_bus_error_has_name(&error, SD_BUS_ERROR_UNKNOWN_METHOD));
420 sd_bus_error_free(&error);
421
422 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value", "org.freedesktop.DBus.ObjectManager", "GetManagedObjects", &error, &reply, NULL);
423 assert_se(r >= 0);
424
425 sd_bus_message_dump(reply, stdout, SD_BUS_MESSAGE_DUMP_WITH_HEADER);
426
427 reply = sd_bus_message_unref(reply);
428
429 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value/a", "org.freedesktop.systemd.ValueTest", "NotifyTest", &error, NULL, NULL);
430 assert_se(r >= 0);
431
432 r = sd_bus_process(bus, &reply);
433 assert_se(r > 0);
434
435 assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.Properties", "PropertiesChanged"));
436 sd_bus_message_dump(reply, stdout, SD_BUS_MESSAGE_DUMP_WITH_HEADER);
437
438 reply = sd_bus_message_unref(reply);
439
440 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value/a", "org.freedesktop.systemd.ValueTest", "NotifyTest2", &error, NULL, NULL);
441 assert_se(r >= 0);
442
443 r = sd_bus_process(bus, &reply);
444 assert_se(r > 0);
445
446 assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.Properties", "PropertiesChanged"));
447 sd_bus_message_dump(reply, stdout, SD_BUS_MESSAGE_DUMP_WITH_HEADER);
448
449 reply = sd_bus_message_unref(reply);
450
451 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "EmitInterfacesAdded", &error, NULL, NULL);
452 assert_se(r >= 0);
453
454 r = sd_bus_process(bus, &reply);
455 assert_se(r > 0);
456
457 assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.ObjectManager", "InterfacesAdded"));
458 sd_bus_message_dump(reply, stdout, SD_BUS_MESSAGE_DUMP_WITH_HEADER);
459
460 reply = sd_bus_message_unref(reply);
461
462 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "EmitInterfacesRemoved", &error, NULL, NULL);
463 assert_se(r >= 0);
464
465 r = sd_bus_process(bus, &reply);
466 assert_se(r > 0);
467
468 assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.ObjectManager", "InterfacesRemoved"));
469 sd_bus_message_dump(reply, stdout, SD_BUS_MESSAGE_DUMP_WITH_HEADER);
470
471 reply = sd_bus_message_unref(reply);
472
473 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "EmitObjectAdded", &error, NULL, NULL);
474 assert_se(r >= 0);
475
476 r = sd_bus_process(bus, &reply);
477 assert_se(r > 0);
478
479 assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.ObjectManager", "InterfacesAdded"));
480 sd_bus_message_dump(reply, stdout, SD_BUS_MESSAGE_DUMP_WITH_HEADER);
481
482 reply = sd_bus_message_unref(reply);
483
484 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "EmitObjectRemoved", &error, NULL, NULL);
485 assert_se(r >= 0);
486
487 r = sd_bus_process(bus, &reply);
488 assert_se(r > 0);
489
490 assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.ObjectManager", "InterfacesRemoved"));
491 sd_bus_message_dump(reply, stdout, SD_BUS_MESSAGE_DUMP_WITH_HEADER);
492
493 reply = sd_bus_message_unref(reply);
494
495 r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Exit", &error, NULL, NULL);
496 assert_se(r >= 0);
497
498 sd_bus_flush(bus);
499
500 return 0;
501 }
502
503 int main(int argc, char *argv[]) {
504 struct context c = {};
505 pthread_t s;
506 void *p;
507 int r, q;
508
509 c.automatic_integer_property = 4711;
510 assert_se(c.automatic_string_property = strdup("dudeldu"));
511
512 assert_se(socketpair(AF_UNIX, SOCK_STREAM, 0, c.fds) >= 0);
513
514 r = pthread_create(&s, NULL, server, &c);
515 if (r != 0)
516 return -r;
517
518 r = client(&c);
519
520 q = pthread_join(s, &p);
521 if (q != 0)
522 return -q;
523
524 if (r < 0)
525 return r;
526
527 if (PTR_TO_INT(p) < 0)
528 return PTR_TO_INT(p);
529
530 free(c.something);
531 free(c.automatic_string_property);
532
533 return EXIT_SUCCESS;
534 }