]> git.ipfire.org Git - thirdparty/systemd.git/blob - src/libsystemd/sd-bus/sd-bus.c
tree-wide: make use of getpid_cached() wherever we can
[thirdparty/systemd.git] / src / libsystemd / sd-bus / sd-bus.c
1 /***
2 This file is part of systemd.
3
4 Copyright 2013 Lennart Poettering
5
6 systemd is free software; you can redistribute it and/or modify it
7 under the terms of the GNU Lesser General Public License as published by
8 the Free Software Foundation; either version 2.1 of the License, or
9 (at your option) any later version.
10
11 systemd is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 Lesser General Public License for more details.
15
16 You should have received a copy of the GNU Lesser General Public License
17 along with systemd; If not, see <http://www.gnu.org/licenses/>.
18 ***/
19
20 #include <endian.h>
21 #include <netdb.h>
22 #include <poll.h>
23 #include <pthread.h>
24 #include <stdlib.h>
25 #include <sys/mman.h>
26 #include <unistd.h>
27
28 #include "sd-bus.h"
29
30 #include "alloc-util.h"
31 #include "bus-container.h"
32 #include "bus-control.h"
33 #include "bus-internal.h"
34 #include "bus-kernel.h"
35 #include "bus-label.h"
36 #include "bus-message.h"
37 #include "bus-objects.h"
38 #include "bus-protocol.h"
39 #include "bus-slot.h"
40 #include "bus-socket.h"
41 #include "bus-track.h"
42 #include "bus-type.h"
43 #include "bus-util.h"
44 #include "cgroup-util.h"
45 #include "def.h"
46 #include "fd-util.h"
47 #include "hexdecoct.h"
48 #include "hostname-util.h"
49 #include "macro.h"
50 #include "missing.h"
51 #include "parse-util.h"
52 #include "string-util.h"
53 #include "strv.h"
54 #include "util.h"
55
56 #define log_debug_bus_message(m) \
57 do { \
58 sd_bus_message *_mm = (m); \
59 log_debug("Got message type=%s sender=%s destination=%s object=%s interface=%s member=%s cookie=%" PRIu64 " reply_cookie=%" PRIu64 " error=%s", \
60 bus_message_type_to_string(_mm->header->type), \
61 strna(sd_bus_message_get_sender(_mm)), \
62 strna(sd_bus_message_get_destination(_mm)), \
63 strna(sd_bus_message_get_path(_mm)), \
64 strna(sd_bus_message_get_interface(_mm)), \
65 strna(sd_bus_message_get_member(_mm)), \
66 BUS_MESSAGE_COOKIE(_mm), \
67 _mm->reply_cookie, \
68 strna(_mm->error.message)); \
69 } while (false)
70
71 static int bus_poll(sd_bus *bus, bool need_more, uint64_t timeout_usec);
72 static int attach_io_events(sd_bus *b);
73 static void detach_io_events(sd_bus *b);
74
75 static thread_local sd_bus *default_system_bus = NULL;
76 static thread_local sd_bus *default_user_bus = NULL;
77 static thread_local sd_bus *default_starter_bus = NULL;
78
79 static void bus_close_fds(sd_bus *b) {
80 assert(b);
81
82 detach_io_events(b);
83
84 if (b->input_fd != b->output_fd)
85 safe_close(b->output_fd);
86 b->output_fd = b->input_fd = safe_close(b->input_fd);
87 }
88
89 static void bus_reset_queues(sd_bus *b) {
90 assert(b);
91
92 while (b->rqueue_size > 0)
93 sd_bus_message_unref(b->rqueue[--b->rqueue_size]);
94
95 b->rqueue = mfree(b->rqueue);
96 b->rqueue_allocated = 0;
97
98 while (b->wqueue_size > 0)
99 sd_bus_message_unref(b->wqueue[--b->wqueue_size]);
100
101 b->wqueue = mfree(b->wqueue);
102 b->wqueue_allocated = 0;
103 }
104
105 static void bus_free(sd_bus *b) {
106 sd_bus_slot *s;
107
108 assert(b);
109 assert(!b->track_queue);
110 assert(!b->tracks);
111
112 b->state = BUS_CLOSED;
113
114 sd_bus_detach_event(b);
115
116 while ((s = b->slots)) {
117 /* At this point only floating slots can still be
118 * around, because the non-floating ones keep a
119 * reference to the bus, and we thus couldn't be
120 * destructing right now... We forcibly disconnect the
121 * slots here, so that they still can be referenced by
122 * apps, but are dead. */
123
124 assert(s->floating);
125 bus_slot_disconnect(s);
126 sd_bus_slot_unref(s);
127 }
128
129 if (b->default_bus_ptr)
130 *b->default_bus_ptr = NULL;
131
132 bus_close_fds(b);
133
134 if (b->kdbus_buffer)
135 munmap(b->kdbus_buffer, KDBUS_POOL_SIZE);
136
137 free(b->label);
138 free(b->rbuffer);
139 free(b->unique_name);
140 free(b->auth_buffer);
141 free(b->address);
142 free(b->kernel);
143 free(b->machine);
144 free(b->fake_label);
145 free(b->cgroup_root);
146 free(b->description);
147
148 free(b->exec_path);
149 strv_free(b->exec_argv);
150
151 close_many(b->fds, b->n_fds);
152 free(b->fds);
153
154 bus_reset_queues(b);
155
156 ordered_hashmap_free_free(b->reply_callbacks);
157 prioq_free(b->reply_callbacks_prioq);
158
159 assert(b->match_callbacks.type == BUS_MATCH_ROOT);
160 bus_match_free(&b->match_callbacks);
161
162 hashmap_free_free(b->vtable_methods);
163 hashmap_free_free(b->vtable_properties);
164
165 assert(hashmap_isempty(b->nodes));
166 hashmap_free(b->nodes);
167
168 bus_kernel_flush_memfd(b);
169
170 assert_se(pthread_mutex_destroy(&b->memfd_cache_mutex) == 0);
171
172 free(b);
173 }
174
175 _public_ int sd_bus_new(sd_bus **ret) {
176 sd_bus *r;
177
178 assert_return(ret, -EINVAL);
179
180 r = new0(sd_bus, 1);
181 if (!r)
182 return -ENOMEM;
183
184 r->n_ref = REFCNT_INIT;
185 r->input_fd = r->output_fd = -1;
186 r->message_version = 1;
187 r->creds_mask |= SD_BUS_CREDS_WELL_KNOWN_NAMES|SD_BUS_CREDS_UNIQUE_NAME;
188 r->hello_flags |= KDBUS_HELLO_ACCEPT_FD;
189 r->attach_flags |= KDBUS_ATTACH_NAMES;
190 r->original_pid = getpid_cached();
191
192 assert_se(pthread_mutex_init(&r->memfd_cache_mutex, NULL) == 0);
193
194 /* We guarantee that wqueue always has space for at least one
195 * entry */
196 if (!GREEDY_REALLOC(r->wqueue, r->wqueue_allocated, 1)) {
197 free(r);
198 return -ENOMEM;
199 }
200
201 *ret = r;
202 return 0;
203 }
204
205 _public_ int sd_bus_set_address(sd_bus *bus, const char *address) {
206 char *a;
207
208 assert_return(bus, -EINVAL);
209 assert_return(bus->state == BUS_UNSET, -EPERM);
210 assert_return(address, -EINVAL);
211 assert_return(!bus_pid_changed(bus), -ECHILD);
212
213 a = strdup(address);
214 if (!a)
215 return -ENOMEM;
216
217 free(bus->address);
218 bus->address = a;
219
220 return 0;
221 }
222
223 _public_ int sd_bus_set_fd(sd_bus *bus, int input_fd, int output_fd) {
224 assert_return(bus, -EINVAL);
225 assert_return(bus->state == BUS_UNSET, -EPERM);
226 assert_return(input_fd >= 0, -EBADF);
227 assert_return(output_fd >= 0, -EBADF);
228 assert_return(!bus_pid_changed(bus), -ECHILD);
229
230 bus->input_fd = input_fd;
231 bus->output_fd = output_fd;
232 return 0;
233 }
234
235 _public_ int sd_bus_set_exec(sd_bus *bus, const char *path, char *const argv[]) {
236 char *p, **a;
237
238 assert_return(bus, -EINVAL);
239 assert_return(bus->state == BUS_UNSET, -EPERM);
240 assert_return(path, -EINVAL);
241 assert_return(!strv_isempty(argv), -EINVAL);
242 assert_return(!bus_pid_changed(bus), -ECHILD);
243
244 p = strdup(path);
245 if (!p)
246 return -ENOMEM;
247
248 a = strv_copy(argv);
249 if (!a) {
250 free(p);
251 return -ENOMEM;
252 }
253
254 free(bus->exec_path);
255 strv_free(bus->exec_argv);
256
257 bus->exec_path = p;
258 bus->exec_argv = a;
259
260 return 0;
261 }
262
263 _public_ int sd_bus_set_bus_client(sd_bus *bus, int b) {
264 assert_return(bus, -EINVAL);
265 assert_return(bus->state == BUS_UNSET, -EPERM);
266 assert_return(!bus_pid_changed(bus), -ECHILD);
267
268 bus->bus_client = !!b;
269 return 0;
270 }
271
272 _public_ int sd_bus_set_monitor(sd_bus *bus, int b) {
273 assert_return(bus, -EINVAL);
274 assert_return(bus->state == BUS_UNSET, -EPERM);
275 assert_return(!bus_pid_changed(bus), -ECHILD);
276
277 SET_FLAG(bus->hello_flags, KDBUS_HELLO_MONITOR, b);
278 return 0;
279 }
280
281 _public_ int sd_bus_negotiate_fds(sd_bus *bus, int b) {
282 assert_return(bus, -EINVAL);
283 assert_return(bus->state == BUS_UNSET, -EPERM);
284 assert_return(!bus_pid_changed(bus), -ECHILD);
285
286 SET_FLAG(bus->hello_flags, KDBUS_HELLO_ACCEPT_FD, b);
287 return 0;
288 }
289
290 _public_ int sd_bus_negotiate_timestamp(sd_bus *bus, int b) {
291 uint64_t new_flags;
292 assert_return(bus, -EINVAL);
293 assert_return(!IN_SET(bus->state, BUS_CLOSING, BUS_CLOSED), -EPERM);
294 assert_return(!bus_pid_changed(bus), -ECHILD);
295
296 new_flags = bus->attach_flags;
297 SET_FLAG(new_flags, KDBUS_ATTACH_TIMESTAMP, b);
298
299 if (bus->attach_flags == new_flags)
300 return 0;
301
302 bus->attach_flags = new_flags;
303 if (bus->state != BUS_UNSET && bus->is_kernel)
304 bus_kernel_realize_attach_flags(bus);
305
306 return 0;
307 }
308
309 _public_ int sd_bus_negotiate_creds(sd_bus *bus, int b, uint64_t mask) {
310 uint64_t new_flags;
311
312 assert_return(bus, -EINVAL);
313 assert_return(mask <= _SD_BUS_CREDS_ALL, -EINVAL);
314 assert_return(!IN_SET(bus->state, BUS_CLOSING, BUS_CLOSED), -EPERM);
315 assert_return(!bus_pid_changed(bus), -ECHILD);
316
317 SET_FLAG(bus->creds_mask, mask, b);
318
319 /* The well knowns we need unconditionally, so that matches can work */
320 bus->creds_mask |= SD_BUS_CREDS_WELL_KNOWN_NAMES|SD_BUS_CREDS_UNIQUE_NAME;
321
322 /* Make sure we don't lose the timestamp flag */
323 new_flags = (bus->attach_flags & KDBUS_ATTACH_TIMESTAMP) | attach_flags_to_kdbus(bus->creds_mask);
324 if (bus->attach_flags == new_flags)
325 return 0;
326
327 bus->attach_flags = new_flags;
328 if (bus->state != BUS_UNSET && bus->is_kernel)
329 bus_kernel_realize_attach_flags(bus);
330
331 return 0;
332 }
333
334 _public_ int sd_bus_set_server(sd_bus *bus, int b, sd_id128_t server_id) {
335 assert_return(bus, -EINVAL);
336 assert_return(b || sd_id128_equal(server_id, SD_ID128_NULL), -EINVAL);
337 assert_return(bus->state == BUS_UNSET, -EPERM);
338 assert_return(!bus_pid_changed(bus), -ECHILD);
339
340 bus->is_server = !!b;
341 bus->server_id = server_id;
342 return 0;
343 }
344
345 _public_ int sd_bus_set_anonymous(sd_bus *bus, int b) {
346 assert_return(bus, -EINVAL);
347 assert_return(bus->state == BUS_UNSET, -EPERM);
348 assert_return(!bus_pid_changed(bus), -ECHILD);
349
350 bus->anonymous_auth = !!b;
351 return 0;
352 }
353
354 _public_ int sd_bus_set_trusted(sd_bus *bus, int b) {
355 assert_return(bus, -EINVAL);
356 assert_return(bus->state == BUS_UNSET, -EPERM);
357 assert_return(!bus_pid_changed(bus), -ECHILD);
358
359 bus->trusted = !!b;
360 return 0;
361 }
362
363 _public_ int sd_bus_set_description(sd_bus *bus, const char *description) {
364 assert_return(bus, -EINVAL);
365 assert_return(bus->state == BUS_UNSET, -EPERM);
366 assert_return(!bus_pid_changed(bus), -ECHILD);
367
368 return free_and_strdup(&bus->description, description);
369 }
370
371 _public_ int sd_bus_set_allow_interactive_authorization(sd_bus *bus, int b) {
372 assert_return(bus, -EINVAL);
373 assert_return(!bus_pid_changed(bus), -ECHILD);
374
375 bus->allow_interactive_authorization = !!b;
376 return 0;
377 }
378
379 _public_ int sd_bus_get_allow_interactive_authorization(sd_bus *bus) {
380 assert_return(bus, -EINVAL);
381 assert_return(!bus_pid_changed(bus), -ECHILD);
382
383 return bus->allow_interactive_authorization;
384 }
385
386 static int hello_callback(sd_bus_message *reply, void *userdata, sd_bus_error *error) {
387 const char *s;
388 sd_bus *bus;
389 int r;
390
391 assert(reply);
392 bus = reply->bus;
393 assert(bus);
394 assert(bus->state == BUS_HELLO || bus->state == BUS_CLOSING);
395
396 r = sd_bus_message_get_errno(reply);
397 if (r > 0)
398 return -r;
399
400 r = sd_bus_message_read(reply, "s", &s);
401 if (r < 0)
402 return r;
403
404 if (!service_name_is_valid(s) || s[0] != ':')
405 return -EBADMSG;
406
407 bus->unique_name = strdup(s);
408 if (!bus->unique_name)
409 return -ENOMEM;
410
411 if (bus->state == BUS_HELLO)
412 bus->state = BUS_RUNNING;
413
414 return 1;
415 }
416
417 static int bus_send_hello(sd_bus *bus) {
418 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
419 int r;
420
421 assert(bus);
422
423 if (!bus->bus_client || bus->is_kernel)
424 return 0;
425
426 r = sd_bus_message_new_method_call(
427 bus,
428 &m,
429 "org.freedesktop.DBus",
430 "/org/freedesktop/DBus",
431 "org.freedesktop.DBus",
432 "Hello");
433 if (r < 0)
434 return r;
435
436 return sd_bus_call_async(bus, NULL, m, hello_callback, NULL, 0);
437 }
438
439 int bus_start_running(sd_bus *bus) {
440 assert(bus);
441
442 if (bus->bus_client && !bus->is_kernel) {
443 bus->state = BUS_HELLO;
444 return 1;
445 }
446
447 bus->state = BUS_RUNNING;
448 return 1;
449 }
450
451 static int parse_address_key(const char **p, const char *key, char **value) {
452 size_t l, n = 0, allocated = 0;
453 const char *a;
454 char *r = NULL;
455
456 assert(p);
457 assert(*p);
458 assert(value);
459
460 if (key) {
461 l = strlen(key);
462 if (strncmp(*p, key, l) != 0)
463 return 0;
464
465 if ((*p)[l] != '=')
466 return 0;
467
468 if (*value)
469 return -EINVAL;
470
471 a = *p + l + 1;
472 } else
473 a = *p;
474
475 while (*a != ';' && *a != ',' && *a != 0) {
476 char c;
477
478 if (*a == '%') {
479 int x, y;
480
481 x = unhexchar(a[1]);
482 if (x < 0) {
483 free(r);
484 return x;
485 }
486
487 y = unhexchar(a[2]);
488 if (y < 0) {
489 free(r);
490 return y;
491 }
492
493 c = (char) ((x << 4) | y);
494 a += 3;
495 } else {
496 c = *a;
497 a++;
498 }
499
500 if (!GREEDY_REALLOC(r, allocated, n + 2))
501 return -ENOMEM;
502
503 r[n++] = c;
504 }
505
506 if (!r) {
507 r = strdup("");
508 if (!r)
509 return -ENOMEM;
510 } else
511 r[n] = 0;
512
513 if (*a == ',')
514 a++;
515
516 *p = a;
517
518 free(*value);
519 *value = r;
520
521 return 1;
522 }
523
524 static void skip_address_key(const char **p) {
525 assert(p);
526 assert(*p);
527
528 *p += strcspn(*p, ",");
529
530 if (**p == ',')
531 (*p)++;
532 }
533
534 static int parse_unix_address(sd_bus *b, const char **p, char **guid) {
535 _cleanup_free_ char *path = NULL, *abstract = NULL;
536 size_t l;
537 int r;
538
539 assert(b);
540 assert(p);
541 assert(*p);
542 assert(guid);
543
544 while (**p != 0 && **p != ';') {
545 r = parse_address_key(p, "guid", guid);
546 if (r < 0)
547 return r;
548 else if (r > 0)
549 continue;
550
551 r = parse_address_key(p, "path", &path);
552 if (r < 0)
553 return r;
554 else if (r > 0)
555 continue;
556
557 r = parse_address_key(p, "abstract", &abstract);
558 if (r < 0)
559 return r;
560 else if (r > 0)
561 continue;
562
563 skip_address_key(p);
564 }
565
566 if (!path && !abstract)
567 return -EINVAL;
568
569 if (path && abstract)
570 return -EINVAL;
571
572 if (path) {
573 l = strlen(path);
574 if (l > sizeof(b->sockaddr.un.sun_path))
575 return -E2BIG;
576
577 b->sockaddr.un.sun_family = AF_UNIX;
578 strncpy(b->sockaddr.un.sun_path, path, sizeof(b->sockaddr.un.sun_path));
579 b->sockaddr_size = offsetof(struct sockaddr_un, sun_path) + l;
580 } else if (abstract) {
581 l = strlen(abstract);
582 if (l > sizeof(b->sockaddr.un.sun_path) - 1)
583 return -E2BIG;
584
585 b->sockaddr.un.sun_family = AF_UNIX;
586 b->sockaddr.un.sun_path[0] = 0;
587 strncpy(b->sockaddr.un.sun_path+1, abstract, sizeof(b->sockaddr.un.sun_path)-1);
588 b->sockaddr_size = offsetof(struct sockaddr_un, sun_path) + 1 + l;
589 }
590
591 b->is_local = true;
592
593 return 0;
594 }
595
596 static int parse_tcp_address(sd_bus *b, const char **p, char **guid) {
597 _cleanup_free_ char *host = NULL, *port = NULL, *family = NULL;
598 int r;
599 struct addrinfo *result, hints = {
600 .ai_socktype = SOCK_STREAM,
601 .ai_flags = AI_ADDRCONFIG,
602 };
603
604 assert(b);
605 assert(p);
606 assert(*p);
607 assert(guid);
608
609 while (**p != 0 && **p != ';') {
610 r = parse_address_key(p, "guid", guid);
611 if (r < 0)
612 return r;
613 else if (r > 0)
614 continue;
615
616 r = parse_address_key(p, "host", &host);
617 if (r < 0)
618 return r;
619 else if (r > 0)
620 continue;
621
622 r = parse_address_key(p, "port", &port);
623 if (r < 0)
624 return r;
625 else if (r > 0)
626 continue;
627
628 r = parse_address_key(p, "family", &family);
629 if (r < 0)
630 return r;
631 else if (r > 0)
632 continue;
633
634 skip_address_key(p);
635 }
636
637 if (!host || !port)
638 return -EINVAL;
639
640 if (family) {
641 if (streq(family, "ipv4"))
642 hints.ai_family = AF_INET;
643 else if (streq(family, "ipv6"))
644 hints.ai_family = AF_INET6;
645 else
646 return -EINVAL;
647 }
648
649 r = getaddrinfo(host, port, &hints, &result);
650 if (r == EAI_SYSTEM)
651 return -errno;
652 else if (r != 0)
653 return -EADDRNOTAVAIL;
654
655 memcpy(&b->sockaddr, result->ai_addr, result->ai_addrlen);
656 b->sockaddr_size = result->ai_addrlen;
657
658 freeaddrinfo(result);
659
660 b->is_local = false;
661
662 return 0;
663 }
664
665 static int parse_exec_address(sd_bus *b, const char **p, char **guid) {
666 char *path = NULL;
667 unsigned n_argv = 0, j;
668 char **argv = NULL;
669 size_t allocated = 0;
670 int r;
671
672 assert(b);
673 assert(p);
674 assert(*p);
675 assert(guid);
676
677 while (**p != 0 && **p != ';') {
678 r = parse_address_key(p, "guid", guid);
679 if (r < 0)
680 goto fail;
681 else if (r > 0)
682 continue;
683
684 r = parse_address_key(p, "path", &path);
685 if (r < 0)
686 goto fail;
687 else if (r > 0)
688 continue;
689
690 if (startswith(*p, "argv")) {
691 unsigned ul;
692
693 errno = 0;
694 ul = strtoul(*p + 4, (char**) p, 10);
695 if (errno > 0 || **p != '=' || ul > 256) {
696 r = -EINVAL;
697 goto fail;
698 }
699
700 (*p)++;
701
702 if (ul >= n_argv) {
703 if (!GREEDY_REALLOC0(argv, allocated, ul + 2)) {
704 r = -ENOMEM;
705 goto fail;
706 }
707
708 n_argv = ul + 1;
709 }
710
711 r = parse_address_key(p, NULL, argv + ul);
712 if (r < 0)
713 goto fail;
714
715 continue;
716 }
717
718 skip_address_key(p);
719 }
720
721 if (!path) {
722 r = -EINVAL;
723 goto fail;
724 }
725
726 /* Make sure there are no holes in the array, with the
727 * exception of argv[0] */
728 for (j = 1; j < n_argv; j++)
729 if (!argv[j]) {
730 r = -EINVAL;
731 goto fail;
732 }
733
734 if (argv && argv[0] == NULL) {
735 argv[0] = strdup(path);
736 if (!argv[0]) {
737 r = -ENOMEM;
738 goto fail;
739 }
740 }
741
742 b->exec_path = path;
743 b->exec_argv = argv;
744
745 b->is_local = false;
746
747 return 0;
748
749 fail:
750 for (j = 0; j < n_argv; j++)
751 free(argv[j]);
752
753 free(argv);
754 free(path);
755 return r;
756 }
757
758 static int parse_kernel_address(sd_bus *b, const char **p, char **guid) {
759 _cleanup_free_ char *path = NULL;
760 int r;
761
762 assert(b);
763 assert(p);
764 assert(*p);
765 assert(guid);
766
767 while (**p != 0 && **p != ';') {
768 r = parse_address_key(p, "guid", guid);
769 if (r < 0)
770 return r;
771 else if (r > 0)
772 continue;
773
774 r = parse_address_key(p, "path", &path);
775 if (r < 0)
776 return r;
777 else if (r > 0)
778 continue;
779
780 skip_address_key(p);
781 }
782
783 if (!path)
784 return -EINVAL;
785
786 free(b->kernel);
787 b->kernel = path;
788 path = NULL;
789
790 b->is_local = true;
791
792 return 0;
793 }
794
795 static int parse_container_unix_address(sd_bus *b, const char **p, char **guid) {
796 _cleanup_free_ char *machine = NULL, *pid = NULL;
797 int r;
798
799 assert(b);
800 assert(p);
801 assert(*p);
802 assert(guid);
803
804 while (**p != 0 && **p != ';') {
805 r = parse_address_key(p, "guid", guid);
806 if (r < 0)
807 return r;
808 else if (r > 0)
809 continue;
810
811 r = parse_address_key(p, "machine", &machine);
812 if (r < 0)
813 return r;
814 else if (r > 0)
815 continue;
816
817 r = parse_address_key(p, "pid", &pid);
818 if (r < 0)
819 return r;
820 else if (r > 0)
821 continue;
822
823 skip_address_key(p);
824 }
825
826 if (!machine == !pid)
827 return -EINVAL;
828
829 if (machine) {
830 if (!machine_name_is_valid(machine))
831 return -EINVAL;
832
833 free(b->machine);
834 b->machine = machine;
835 machine = NULL;
836 } else {
837 b->machine = mfree(b->machine);
838 }
839
840 if (pid) {
841 r = parse_pid(pid, &b->nspid);
842 if (r < 0)
843 return r;
844 } else
845 b->nspid = 0;
846
847 b->sockaddr.un.sun_family = AF_UNIX;
848 strncpy(b->sockaddr.un.sun_path, "/var/run/dbus/system_bus_socket", sizeof(b->sockaddr.un.sun_path));
849 b->sockaddr_size = SOCKADDR_UN_LEN(b->sockaddr.un);
850 b->is_local = false;
851
852 return 0;
853 }
854
855 static int parse_container_kernel_address(sd_bus *b, const char **p, char **guid) {
856 _cleanup_free_ char *machine = NULL, *pid = NULL;
857 int r;
858
859 assert(b);
860 assert(p);
861 assert(*p);
862 assert(guid);
863
864 while (**p != 0 && **p != ';') {
865 r = parse_address_key(p, "guid", guid);
866 if (r < 0)
867 return r;
868 else if (r > 0)
869 continue;
870
871 r = parse_address_key(p, "machine", &machine);
872 if (r < 0)
873 return r;
874 else if (r > 0)
875 continue;
876
877 r = parse_address_key(p, "pid", &pid);
878 if (r < 0)
879 return r;
880 else if (r > 0)
881 continue;
882
883 skip_address_key(p);
884 }
885
886 if (!machine == !pid)
887 return -EINVAL;
888
889 if (machine) {
890 if (!machine_name_is_valid(machine))
891 return -EINVAL;
892
893 free(b->machine);
894 b->machine = machine;
895 machine = NULL;
896 } else {
897 b->machine = mfree(b->machine);
898 }
899
900 if (pid) {
901 r = parse_pid(pid, &b->nspid);
902 if (r < 0)
903 return r;
904 } else
905 b->nspid = 0;
906
907 r = free_and_strdup(&b->kernel, "/sys/fs/kdbus/0-system/bus");
908 if (r < 0)
909 return r;
910
911 b->is_local = false;
912
913 return 0;
914 }
915
916 static void bus_reset_parsed_address(sd_bus *b) {
917 assert(b);
918
919 zero(b->sockaddr);
920 b->sockaddr_size = 0;
921 b->exec_argv = strv_free(b->exec_argv);
922 b->exec_path = mfree(b->exec_path);
923 b->server_id = SD_ID128_NULL;
924 b->kernel = mfree(b->kernel);
925 b->machine = mfree(b->machine);
926 b->nspid = 0;
927 }
928
929 static int bus_parse_next_address(sd_bus *b) {
930 _cleanup_free_ char *guid = NULL;
931 const char *a;
932 int r;
933
934 assert(b);
935
936 if (!b->address)
937 return 0;
938 if (b->address[b->address_index] == 0)
939 return 0;
940
941 bus_reset_parsed_address(b);
942
943 a = b->address + b->address_index;
944
945 while (*a != 0) {
946
947 if (*a == ';') {
948 a++;
949 continue;
950 }
951
952 if (startswith(a, "unix:")) {
953 a += 5;
954
955 r = parse_unix_address(b, &a, &guid);
956 if (r < 0)
957 return r;
958 break;
959
960 } else if (startswith(a, "tcp:")) {
961
962 a += 4;
963 r = parse_tcp_address(b, &a, &guid);
964 if (r < 0)
965 return r;
966
967 break;
968
969 } else if (startswith(a, "unixexec:")) {
970
971 a += 9;
972 r = parse_exec_address(b, &a, &guid);
973 if (r < 0)
974 return r;
975
976 break;
977
978 } else if (startswith(a, "kernel:")) {
979
980 a += 7;
981 r = parse_kernel_address(b, &a, &guid);
982 if (r < 0)
983 return r;
984
985 break;
986 } else if (startswith(a, "x-machine-unix:")) {
987
988 a += 15;
989 r = parse_container_unix_address(b, &a, &guid);
990 if (r < 0)
991 return r;
992
993 break;
994 } else if (startswith(a, "x-machine-kernel:")) {
995
996 a += 17;
997 r = parse_container_kernel_address(b, &a, &guid);
998 if (r < 0)
999 return r;
1000
1001 break;
1002 }
1003
1004 a = strchr(a, ';');
1005 if (!a)
1006 return 0;
1007 }
1008
1009 if (guid) {
1010 r = sd_id128_from_string(guid, &b->server_id);
1011 if (r < 0)
1012 return r;
1013 }
1014
1015 b->address_index = a - b->address;
1016 return 1;
1017 }
1018
1019 static int bus_start_address(sd_bus *b) {
1020 bool container_kdbus_available = false;
1021 bool kdbus_available = false;
1022 int r;
1023
1024 assert(b);
1025
1026 for (;;) {
1027 bool skipped = false;
1028
1029 bus_close_fds(b);
1030
1031 /*
1032 * Usually, if you provide multiple different bus-addresses, we
1033 * try all of them in order. We use the first one that
1034 * succeeds. However, if you mix kernel and unix addresses, we
1035 * never try unix-addresses if a previous kernel address was
1036 * tried and kdbus was available. This is required to prevent
1037 * clients to fallback to the bus-proxy if kdbus is available
1038 * but failed (eg., too many connections).
1039 */
1040
1041 if (b->exec_path)
1042 r = bus_socket_exec(b);
1043 else if ((b->nspid > 0 || b->machine) && b->kernel) {
1044 r = bus_container_connect_kernel(b);
1045 if (r < 0 && !IN_SET(r, -ENOENT, -ESOCKTNOSUPPORT))
1046 container_kdbus_available = true;
1047
1048 } else if ((b->nspid > 0 || b->machine) && b->sockaddr.sa.sa_family != AF_UNSPEC) {
1049 if (!container_kdbus_available)
1050 r = bus_container_connect_socket(b);
1051 else
1052 skipped = true;
1053
1054 } else if (b->kernel) {
1055 r = bus_kernel_connect(b);
1056 if (r < 0 && !IN_SET(r, -ENOENT, -ESOCKTNOSUPPORT))
1057 kdbus_available = true;
1058
1059 } else if (b->sockaddr.sa.sa_family != AF_UNSPEC) {
1060 if (!kdbus_available)
1061 r = bus_socket_connect(b);
1062 else
1063 skipped = true;
1064 } else
1065 skipped = true;
1066
1067 if (!skipped) {
1068 if (r >= 0) {
1069 r = attach_io_events(b);
1070 if (r >= 0)
1071 return r;
1072 }
1073
1074 b->last_connect_error = -r;
1075 }
1076
1077 r = bus_parse_next_address(b);
1078 if (r < 0)
1079 return r;
1080 if (r == 0)
1081 return b->last_connect_error ? -b->last_connect_error : -ECONNREFUSED;
1082 }
1083 }
1084
1085 int bus_next_address(sd_bus *b) {
1086 assert(b);
1087
1088 bus_reset_parsed_address(b);
1089 return bus_start_address(b);
1090 }
1091
1092 static int bus_start_fd(sd_bus *b) {
1093 struct stat st;
1094 int r;
1095
1096 assert(b);
1097 assert(b->input_fd >= 0);
1098 assert(b->output_fd >= 0);
1099
1100 r = fd_nonblock(b->input_fd, true);
1101 if (r < 0)
1102 return r;
1103
1104 r = fd_cloexec(b->input_fd, true);
1105 if (r < 0)
1106 return r;
1107
1108 if (b->input_fd != b->output_fd) {
1109 r = fd_nonblock(b->output_fd, true);
1110 if (r < 0)
1111 return r;
1112
1113 r = fd_cloexec(b->output_fd, true);
1114 if (r < 0)
1115 return r;
1116 }
1117
1118 if (fstat(b->input_fd, &st) < 0)
1119 return -errno;
1120
1121 if (S_ISCHR(b->input_fd))
1122 return bus_kernel_take_fd(b);
1123 else
1124 return bus_socket_take_fd(b);
1125 }
1126
1127 _public_ int sd_bus_start(sd_bus *bus) {
1128 int r;
1129
1130 assert_return(bus, -EINVAL);
1131 assert_return(bus->state == BUS_UNSET, -EPERM);
1132 assert_return(!bus_pid_changed(bus), -ECHILD);
1133
1134 bus->state = BUS_OPENING;
1135
1136 if (bus->is_server && bus->bus_client)
1137 return -EINVAL;
1138
1139 if (bus->input_fd >= 0)
1140 r = bus_start_fd(bus);
1141 else if (bus->address || bus->sockaddr.sa.sa_family != AF_UNSPEC || bus->exec_path || bus->kernel || bus->machine)
1142 r = bus_start_address(bus);
1143 else
1144 return -EINVAL;
1145
1146 if (r < 0) {
1147 sd_bus_close(bus);
1148 return r;
1149 }
1150
1151 return bus_send_hello(bus);
1152 }
1153
1154 _public_ int sd_bus_open(sd_bus **ret) {
1155 const char *e;
1156 sd_bus *b;
1157 int r;
1158
1159 assert_return(ret, -EINVAL);
1160
1161 /* Let's connect to the starter bus if it is set, and
1162 * otherwise to the bus that is appropropriate for the scope
1163 * we are running in */
1164
1165 e = secure_getenv("DBUS_STARTER_BUS_TYPE");
1166 if (e) {
1167 if (streq(e, "system"))
1168 return sd_bus_open_system(ret);
1169 else if (STR_IN_SET(e, "session", "user"))
1170 return sd_bus_open_user(ret);
1171 }
1172
1173 e = secure_getenv("DBUS_STARTER_ADDRESS");
1174 if (!e) {
1175 if (cg_pid_get_owner_uid(0, NULL) >= 0)
1176 return sd_bus_open_user(ret);
1177 else
1178 return sd_bus_open_system(ret);
1179 }
1180
1181 r = sd_bus_new(&b);
1182 if (r < 0)
1183 return r;
1184
1185 r = sd_bus_set_address(b, e);
1186 if (r < 0)
1187 goto fail;
1188
1189 b->bus_client = true;
1190
1191 /* We don't know whether the bus is trusted or not, so better
1192 * be safe, and authenticate everything */
1193 b->trusted = false;
1194 b->is_local = false;
1195 b->attach_flags |= KDBUS_ATTACH_CAPS | KDBUS_ATTACH_CREDS;
1196 b->creds_mask |= SD_BUS_CREDS_UID | SD_BUS_CREDS_EUID | SD_BUS_CREDS_EFFECTIVE_CAPS;
1197
1198 r = sd_bus_start(b);
1199 if (r < 0)
1200 goto fail;
1201
1202 *ret = b;
1203 return 0;
1204
1205 fail:
1206 bus_free(b);
1207 return r;
1208 }
1209
1210 int bus_set_address_system(sd_bus *b) {
1211 const char *e;
1212 assert(b);
1213
1214 e = secure_getenv("DBUS_SYSTEM_BUS_ADDRESS");
1215 if (e)
1216 return sd_bus_set_address(b, e);
1217
1218 return sd_bus_set_address(b, DEFAULT_SYSTEM_BUS_ADDRESS);
1219 }
1220
1221 _public_ int sd_bus_open_system(sd_bus **ret) {
1222 sd_bus *b;
1223 int r;
1224
1225 assert_return(ret, -EINVAL);
1226
1227 r = sd_bus_new(&b);
1228 if (r < 0)
1229 return r;
1230
1231 r = bus_set_address_system(b);
1232 if (r < 0)
1233 goto fail;
1234
1235 b->bus_client = true;
1236 b->is_system = true;
1237
1238 /* Let's do per-method access control on the system bus. We
1239 * need the caller's UID and capability set for that. */
1240 b->trusted = false;
1241 b->attach_flags |= KDBUS_ATTACH_CAPS | KDBUS_ATTACH_CREDS;
1242 b->creds_mask |= SD_BUS_CREDS_UID | SD_BUS_CREDS_EUID | SD_BUS_CREDS_EFFECTIVE_CAPS;
1243 b->is_local = true;
1244
1245 r = sd_bus_start(b);
1246 if (r < 0)
1247 goto fail;
1248
1249 *ret = b;
1250 return 0;
1251
1252 fail:
1253 bus_free(b);
1254 return r;
1255 }
1256
1257 int bus_set_address_user(sd_bus *b) {
1258 const char *e;
1259 uid_t uid;
1260 int r;
1261
1262 assert(b);
1263
1264 e = secure_getenv("DBUS_SESSION_BUS_ADDRESS");
1265 if (e)
1266 return sd_bus_set_address(b, e);
1267
1268 r = cg_pid_get_owner_uid(0, &uid);
1269 if (r < 0)
1270 uid = getuid();
1271
1272 e = secure_getenv("XDG_RUNTIME_DIR");
1273 if (e) {
1274 _cleanup_free_ char *ee = NULL;
1275
1276 ee = bus_address_escape(e);
1277 if (!ee)
1278 return -ENOMEM;
1279
1280 (void) asprintf(&b->address, KERNEL_USER_BUS_ADDRESS_FMT ";" UNIX_USER_BUS_ADDRESS_FMT, uid, ee);
1281 } else
1282 (void) asprintf(&b->address, KERNEL_USER_BUS_ADDRESS_FMT, uid);
1283
1284 if (!b->address)
1285 return -ENOMEM;
1286
1287 return 0;
1288 }
1289
1290 _public_ int sd_bus_open_user(sd_bus **ret) {
1291 sd_bus *b;
1292 int r;
1293
1294 assert_return(ret, -EINVAL);
1295
1296 r = sd_bus_new(&b);
1297 if (r < 0)
1298 return r;
1299
1300 r = bus_set_address_user(b);
1301 if (r < 0)
1302 return r;
1303
1304 b->bus_client = true;
1305 b->is_user = true;
1306
1307 /* We don't do any per-method access control on the user
1308 * bus. */
1309 b->trusted = true;
1310 b->is_local = true;
1311
1312 r = sd_bus_start(b);
1313 if (r < 0)
1314 goto fail;
1315
1316 *ret = b;
1317 return 0;
1318
1319 fail:
1320 bus_free(b);
1321 return r;
1322 }
1323
1324 int bus_set_address_system_remote(sd_bus *b, const char *host) {
1325 _cleanup_free_ char *e = NULL;
1326 char *m = NULL, *c = NULL;
1327
1328 assert(b);
1329 assert(host);
1330
1331 /* Let's see if we shall enter some container */
1332 m = strchr(host, ':');
1333 if (m) {
1334 m++;
1335
1336 /* Let's make sure this is not a port of some kind,
1337 * and is a valid machine name. */
1338 if (!in_charset(m, "0123456789") && machine_name_is_valid(m)) {
1339 char *t;
1340
1341 /* Cut out the host part */
1342 t = strndupa(host, m - host - 1);
1343 e = bus_address_escape(t);
1344 if (!e)
1345 return -ENOMEM;
1346
1347 c = strjoina(",argv4=--machine=", m);
1348 }
1349 }
1350
1351 if (!e) {
1352 e = bus_address_escape(host);
1353 if (!e)
1354 return -ENOMEM;
1355 }
1356
1357 b->address = strjoin("unixexec:path=ssh,argv1=-xT,argv2=", e, ",argv3=systemd-stdio-bridge", c);
1358 if (!b->address)
1359 return -ENOMEM;
1360
1361 return 0;
1362 }
1363
1364 _public_ int sd_bus_open_system_remote(sd_bus **ret, const char *host) {
1365 sd_bus *bus;
1366 int r;
1367
1368 assert_return(host, -EINVAL);
1369 assert_return(ret, -EINVAL);
1370
1371 r = sd_bus_new(&bus);
1372 if (r < 0)
1373 return r;
1374
1375 r = bus_set_address_system_remote(bus, host);
1376 if (r < 0)
1377 goto fail;
1378
1379 bus->bus_client = true;
1380 bus->trusted = false;
1381 bus->is_system = true;
1382 bus->is_local = false;
1383
1384 r = sd_bus_start(bus);
1385 if (r < 0)
1386 goto fail;
1387
1388 *ret = bus;
1389 return 0;
1390
1391 fail:
1392 bus_free(bus);
1393 return r;
1394 }
1395
1396 int bus_set_address_system_machine(sd_bus *b, const char *machine) {
1397 _cleanup_free_ char *e = NULL;
1398
1399 assert(b);
1400 assert(machine);
1401
1402 e = bus_address_escape(machine);
1403 if (!e)
1404 return -ENOMEM;
1405
1406 b->address = strjoin("x-machine-kernel:machine=", e, ";x-machine-unix:machine=", e);
1407 if (!b->address)
1408 return -ENOMEM;
1409
1410 return 0;
1411 }
1412
1413 _public_ int sd_bus_open_system_machine(sd_bus **ret, const char *machine) {
1414 sd_bus *bus;
1415 int r;
1416
1417 assert_return(machine, -EINVAL);
1418 assert_return(ret, -EINVAL);
1419 assert_return(machine_name_is_valid(machine), -EINVAL);
1420
1421 r = sd_bus_new(&bus);
1422 if (r < 0)
1423 return r;
1424
1425 r = bus_set_address_system_machine(bus, machine);
1426 if (r < 0)
1427 goto fail;
1428
1429 bus->bus_client = true;
1430 bus->trusted = false;
1431 bus->is_system = true;
1432 bus->is_local = false;
1433
1434 r = sd_bus_start(bus);
1435 if (r < 0)
1436 goto fail;
1437
1438 *ret = bus;
1439 return 0;
1440
1441 fail:
1442 bus_free(bus);
1443 return r;
1444 }
1445
1446 _public_ void sd_bus_close(sd_bus *bus) {
1447
1448 if (!bus)
1449 return;
1450 if (bus->state == BUS_CLOSED)
1451 return;
1452 if (bus_pid_changed(bus))
1453 return;
1454
1455 bus->state = BUS_CLOSED;
1456
1457 sd_bus_detach_event(bus);
1458
1459 /* Drop all queued messages so that they drop references to
1460 * the bus object and the bus may be freed */
1461 bus_reset_queues(bus);
1462
1463 if (!bus->is_kernel)
1464 bus_close_fds(bus);
1465
1466 /* We'll leave the fd open in case this is a kernel bus, since
1467 * there might still be memblocks around that reference this
1468 * bus, and they might need to invoke the KDBUS_CMD_FREE
1469 * ioctl on the fd when they are freed. */
1470 }
1471
1472 _public_ sd_bus* sd_bus_flush_close_unref(sd_bus *bus) {
1473
1474 if (!bus)
1475 return NULL;
1476
1477 sd_bus_flush(bus);
1478 sd_bus_close(bus);
1479
1480 return sd_bus_unref(bus);
1481 }
1482
1483 static void bus_enter_closing(sd_bus *bus) {
1484 assert(bus);
1485
1486 if (bus->state != BUS_OPENING &&
1487 bus->state != BUS_AUTHENTICATING &&
1488 bus->state != BUS_HELLO &&
1489 bus->state != BUS_RUNNING)
1490 return;
1491
1492 bus->state = BUS_CLOSING;
1493 }
1494
1495 _public_ sd_bus *sd_bus_ref(sd_bus *bus) {
1496
1497 if (!bus)
1498 return NULL;
1499
1500 assert_se(REFCNT_INC(bus->n_ref) >= 2);
1501
1502 return bus;
1503 }
1504
1505 _public_ sd_bus *sd_bus_unref(sd_bus *bus) {
1506 unsigned i;
1507
1508 if (!bus)
1509 return NULL;
1510
1511 i = REFCNT_DEC(bus->n_ref);
1512 if (i > 0)
1513 return NULL;
1514
1515 bus_free(bus);
1516 return NULL;
1517 }
1518
1519 _public_ int sd_bus_is_open(sd_bus *bus) {
1520
1521 assert_return(bus, -EINVAL);
1522 assert_return(!bus_pid_changed(bus), -ECHILD);
1523
1524 return BUS_IS_OPEN(bus->state);
1525 }
1526
1527 _public_ int sd_bus_can_send(sd_bus *bus, char type) {
1528 int r;
1529
1530 assert_return(bus, -EINVAL);
1531 assert_return(bus->state != BUS_UNSET, -ENOTCONN);
1532 assert_return(!bus_pid_changed(bus), -ECHILD);
1533
1534 if (bus->hello_flags & KDBUS_HELLO_MONITOR)
1535 return 0;
1536
1537 if (type == SD_BUS_TYPE_UNIX_FD) {
1538 if (!(bus->hello_flags & KDBUS_HELLO_ACCEPT_FD))
1539 return 0;
1540
1541 r = bus_ensure_running(bus);
1542 if (r < 0)
1543 return r;
1544
1545 return bus->can_fds;
1546 }
1547
1548 return bus_type_is_valid(type);
1549 }
1550
1551 _public_ int sd_bus_get_bus_id(sd_bus *bus, sd_id128_t *id) {
1552 int r;
1553
1554 assert_return(bus, -EINVAL);
1555 assert_return(id, -EINVAL);
1556 assert_return(!bus_pid_changed(bus), -ECHILD);
1557
1558 r = bus_ensure_running(bus);
1559 if (r < 0)
1560 return r;
1561
1562 *id = bus->server_id;
1563 return 0;
1564 }
1565
1566 static int bus_seal_message(sd_bus *b, sd_bus_message *m, usec_t timeout) {
1567 assert(b);
1568 assert(m);
1569
1570 if (m->sealed) {
1571 /* If we copy the same message to multiple
1572 * destinations, avoid using the same cookie
1573 * numbers. */
1574 b->cookie = MAX(b->cookie, BUS_MESSAGE_COOKIE(m));
1575 return 0;
1576 }
1577
1578 if (timeout == 0)
1579 timeout = BUS_DEFAULT_TIMEOUT;
1580
1581 return bus_message_seal(m, ++b->cookie, timeout);
1582 }
1583
1584 static int bus_remarshal_message(sd_bus *b, sd_bus_message **m) {
1585 bool remarshal = false;
1586
1587 assert(b);
1588
1589 /* wrong packet version */
1590 if (b->message_version != 0 && b->message_version != (*m)->header->version)
1591 remarshal = true;
1592
1593 /* wrong packet endianness */
1594 if (b->message_endian != 0 && b->message_endian != (*m)->header->endian)
1595 remarshal = true;
1596
1597 /* TODO: kdbus-messages received from the kernel contain data which is
1598 * not allowed to be passed to KDBUS_CMD_SEND. Therefore, we have to
1599 * force remarshaling of the message. Technically, we could just
1600 * recreate the kdbus message, but that is non-trivial as other parts of
1601 * the message refer to m->kdbus already. This should be fixed! */
1602 if ((*m)->kdbus && (*m)->release_kdbus)
1603 remarshal = true;
1604
1605 return remarshal ? bus_message_remarshal(b, m) : 0;
1606 }
1607
1608 int bus_seal_synthetic_message(sd_bus *b, sd_bus_message *m) {
1609 assert(b);
1610 assert(m);
1611
1612 /* Fake some timestamps, if they were requested, and not
1613 * already initialized */
1614 if (b->attach_flags & KDBUS_ATTACH_TIMESTAMP) {
1615 if (m->realtime <= 0)
1616 m->realtime = now(CLOCK_REALTIME);
1617
1618 if (m->monotonic <= 0)
1619 m->monotonic = now(CLOCK_MONOTONIC);
1620 }
1621
1622 /* The bus specification says the serial number cannot be 0,
1623 * hence let's fill something in for synthetic messages. Since
1624 * synthetic messages might have a fake sender and we don't
1625 * want to interfere with the real sender's serial numbers we
1626 * pick a fixed, artificial one. We use (uint32_t) -1 rather
1627 * than (uint64_t) -1 since dbus1 only had 32bit identifiers,
1628 * even though kdbus can do 64bit. */
1629 return bus_message_seal(m, 0xFFFFFFFFULL, 0);
1630 }
1631
1632 static int bus_write_message(sd_bus *bus, sd_bus_message *m, bool hint_sync_call, size_t *idx) {
1633 int r;
1634
1635 assert(bus);
1636 assert(m);
1637
1638 if (bus->is_kernel)
1639 r = bus_kernel_write_message(bus, m, hint_sync_call);
1640 else
1641 r = bus_socket_write_message(bus, m, idx);
1642
1643 if (r <= 0)
1644 return r;
1645
1646 if (bus->is_kernel || *idx >= BUS_MESSAGE_SIZE(m))
1647 log_debug("Sent message type=%s sender=%s destination=%s object=%s interface=%s member=%s cookie=%" PRIu64 " reply_cookie=%" PRIu64 " error=%s",
1648 bus_message_type_to_string(m->header->type),
1649 strna(sd_bus_message_get_sender(m)),
1650 strna(sd_bus_message_get_destination(m)),
1651 strna(sd_bus_message_get_path(m)),
1652 strna(sd_bus_message_get_interface(m)),
1653 strna(sd_bus_message_get_member(m)),
1654 BUS_MESSAGE_COOKIE(m),
1655 m->reply_cookie,
1656 strna(m->error.message));
1657
1658 return r;
1659 }
1660
1661 static int dispatch_wqueue(sd_bus *bus) {
1662 int r, ret = 0;
1663
1664 assert(bus);
1665 assert(bus->state == BUS_RUNNING || bus->state == BUS_HELLO);
1666
1667 while (bus->wqueue_size > 0) {
1668
1669 r = bus_write_message(bus, bus->wqueue[0], false, &bus->windex);
1670 if (r < 0)
1671 return r;
1672 else if (r == 0)
1673 /* Didn't do anything this time */
1674 return ret;
1675 else if (bus->is_kernel || bus->windex >= BUS_MESSAGE_SIZE(bus->wqueue[0])) {
1676 /* Fully written. Let's drop the entry from
1677 * the queue.
1678 *
1679 * This isn't particularly optimized, but
1680 * well, this is supposed to be our worst-case
1681 * buffer only, and the socket buffer is
1682 * supposed to be our primary buffer, and if
1683 * it got full, then all bets are off
1684 * anyway. */
1685
1686 bus->wqueue_size--;
1687 sd_bus_message_unref(bus->wqueue[0]);
1688 memmove(bus->wqueue, bus->wqueue + 1, sizeof(sd_bus_message*) * bus->wqueue_size);
1689 bus->windex = 0;
1690
1691 ret = 1;
1692 }
1693 }
1694
1695 return ret;
1696 }
1697
1698 static int bus_read_message(sd_bus *bus, bool hint_priority, int64_t priority) {
1699 assert(bus);
1700
1701 if (bus->is_kernel)
1702 return bus_kernel_read_message(bus, hint_priority, priority);
1703 else
1704 return bus_socket_read_message(bus);
1705 }
1706
1707 int bus_rqueue_make_room(sd_bus *bus) {
1708 assert(bus);
1709
1710 if (bus->rqueue_size >= BUS_RQUEUE_MAX)
1711 return -ENOBUFS;
1712
1713 if (!GREEDY_REALLOC(bus->rqueue, bus->rqueue_allocated, bus->rqueue_size + 1))
1714 return -ENOMEM;
1715
1716 return 0;
1717 }
1718
1719 static int dispatch_rqueue(sd_bus *bus, bool hint_priority, int64_t priority, sd_bus_message **m) {
1720 int r, ret = 0;
1721
1722 assert(bus);
1723 assert(m);
1724 assert(bus->state == BUS_RUNNING || bus->state == BUS_HELLO);
1725
1726 /* Note that the priority logic is only available on kdbus,
1727 * where the rqueue is unused. We check the rqueue here
1728 * anyway, because it's simple... */
1729
1730 for (;;) {
1731 if (bus->rqueue_size > 0) {
1732 /* Dispatch a queued message */
1733
1734 *m = bus->rqueue[0];
1735 bus->rqueue_size--;
1736 memmove(bus->rqueue, bus->rqueue + 1, sizeof(sd_bus_message*) * bus->rqueue_size);
1737 return 1;
1738 }
1739
1740 /* Try to read a new message */
1741 r = bus_read_message(bus, hint_priority, priority);
1742 if (r < 0)
1743 return r;
1744 if (r == 0)
1745 return ret;
1746
1747 ret = 1;
1748 }
1749 }
1750
1751 static int bus_send_internal(sd_bus *bus, sd_bus_message *_m, uint64_t *cookie, bool hint_sync_call) {
1752 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = sd_bus_message_ref(_m);
1753 int r;
1754
1755 assert_return(m, -EINVAL);
1756
1757 if (!bus)
1758 bus = m->bus;
1759
1760 assert_return(!bus_pid_changed(bus), -ECHILD);
1761 assert_return(!bus->is_kernel || !(bus->hello_flags & KDBUS_HELLO_MONITOR), -EROFS);
1762
1763 if (!BUS_IS_OPEN(bus->state))
1764 return -ENOTCONN;
1765
1766 if (m->n_fds > 0) {
1767 r = sd_bus_can_send(bus, SD_BUS_TYPE_UNIX_FD);
1768 if (r < 0)
1769 return r;
1770 if (r == 0)
1771 return -EOPNOTSUPP;
1772 }
1773
1774 /* If the cookie number isn't kept, then we know that no reply
1775 * is expected */
1776 if (!cookie && !m->sealed)
1777 m->header->flags |= BUS_MESSAGE_NO_REPLY_EXPECTED;
1778
1779 r = bus_seal_message(bus, m, 0);
1780 if (r < 0)
1781 return r;
1782
1783 /* Remarshall if we have to. This will possibly unref the
1784 * message and place a replacement in m */
1785 r = bus_remarshal_message(bus, &m);
1786 if (r < 0)
1787 return r;
1788
1789 /* If this is a reply and no reply was requested, then let's
1790 * suppress this, if we can */
1791 if (m->dont_send)
1792 goto finish;
1793
1794 if ((bus->state == BUS_RUNNING || bus->state == BUS_HELLO) && bus->wqueue_size <= 0) {
1795 size_t idx = 0;
1796
1797 r = bus_write_message(bus, m, hint_sync_call, &idx);
1798 if (r < 0) {
1799 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
1800 bus_enter_closing(bus);
1801 return -ECONNRESET;
1802 }
1803
1804 return r;
1805 }
1806
1807 if (!bus->is_kernel && idx < BUS_MESSAGE_SIZE(m)) {
1808 /* Wasn't fully written. So let's remember how
1809 * much was written. Note that the first entry
1810 * of the wqueue array is always allocated so
1811 * that we always can remember how much was
1812 * written. */
1813 bus->wqueue[0] = sd_bus_message_ref(m);
1814 bus->wqueue_size = 1;
1815 bus->windex = idx;
1816 }
1817
1818 } else {
1819 /* Just append it to the queue. */
1820
1821 if (bus->wqueue_size >= BUS_WQUEUE_MAX)
1822 return -ENOBUFS;
1823
1824 if (!GREEDY_REALLOC(bus->wqueue, bus->wqueue_allocated, bus->wqueue_size + 1))
1825 return -ENOMEM;
1826
1827 bus->wqueue[bus->wqueue_size++] = sd_bus_message_ref(m);
1828 }
1829
1830 finish:
1831 if (cookie)
1832 *cookie = BUS_MESSAGE_COOKIE(m);
1833
1834 return 1;
1835 }
1836
1837 _public_ int sd_bus_send(sd_bus *bus, sd_bus_message *m, uint64_t *cookie) {
1838 return bus_send_internal(bus, m, cookie, false);
1839 }
1840
1841 _public_ int sd_bus_send_to(sd_bus *bus, sd_bus_message *m, const char *destination, uint64_t *cookie) {
1842 int r;
1843
1844 assert_return(m, -EINVAL);
1845
1846 if (!bus)
1847 bus = m->bus;
1848
1849 assert_return(!bus_pid_changed(bus), -ECHILD);
1850
1851 if (!BUS_IS_OPEN(bus->state))
1852 return -ENOTCONN;
1853
1854 if (!streq_ptr(m->destination, destination)) {
1855
1856 if (!destination)
1857 return -EEXIST;
1858
1859 r = sd_bus_message_set_destination(m, destination);
1860 if (r < 0)
1861 return r;
1862 }
1863
1864 return sd_bus_send(bus, m, cookie);
1865 }
1866
1867 static usec_t calc_elapse(uint64_t usec) {
1868 if (usec == (uint64_t) -1)
1869 return 0;
1870
1871 return now(CLOCK_MONOTONIC) + usec;
1872 }
1873
1874 static int timeout_compare(const void *a, const void *b) {
1875 const struct reply_callback *x = a, *y = b;
1876
1877 if (x->timeout != 0 && y->timeout == 0)
1878 return -1;
1879
1880 if (x->timeout == 0 && y->timeout != 0)
1881 return 1;
1882
1883 if (x->timeout < y->timeout)
1884 return -1;
1885
1886 if (x->timeout > y->timeout)
1887 return 1;
1888
1889 return 0;
1890 }
1891
1892 _public_ int sd_bus_call_async(
1893 sd_bus *bus,
1894 sd_bus_slot **slot,
1895 sd_bus_message *_m,
1896 sd_bus_message_handler_t callback,
1897 void *userdata,
1898 uint64_t usec) {
1899
1900 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = sd_bus_message_ref(_m);
1901 _cleanup_(sd_bus_slot_unrefp) sd_bus_slot *s = NULL;
1902 int r;
1903
1904 assert_return(m, -EINVAL);
1905 assert_return(m->header->type == SD_BUS_MESSAGE_METHOD_CALL, -EINVAL);
1906 assert_return(!(m->header->flags & BUS_MESSAGE_NO_REPLY_EXPECTED), -EINVAL);
1907 assert_return(callback, -EINVAL);
1908
1909 if (!bus)
1910 bus = m->bus;
1911
1912 assert_return(!bus_pid_changed(bus), -ECHILD);
1913 assert_return(!bus->is_kernel || !(bus->hello_flags & KDBUS_HELLO_MONITOR), -EROFS);
1914
1915 if (!BUS_IS_OPEN(bus->state))
1916 return -ENOTCONN;
1917
1918 r = ordered_hashmap_ensure_allocated(&bus->reply_callbacks, &uint64_hash_ops);
1919 if (r < 0)
1920 return r;
1921
1922 r = prioq_ensure_allocated(&bus->reply_callbacks_prioq, timeout_compare);
1923 if (r < 0)
1924 return r;
1925
1926 r = bus_seal_message(bus, m, usec);
1927 if (r < 0)
1928 return r;
1929
1930 r = bus_remarshal_message(bus, &m);
1931 if (r < 0)
1932 return r;
1933
1934 s = bus_slot_allocate(bus, !slot, BUS_REPLY_CALLBACK, sizeof(struct reply_callback), userdata);
1935 if (!s)
1936 return -ENOMEM;
1937
1938 s->reply_callback.callback = callback;
1939
1940 s->reply_callback.cookie = BUS_MESSAGE_COOKIE(m);
1941 r = ordered_hashmap_put(bus->reply_callbacks, &s->reply_callback.cookie, &s->reply_callback);
1942 if (r < 0) {
1943 s->reply_callback.cookie = 0;
1944 return r;
1945 }
1946
1947 s->reply_callback.timeout = calc_elapse(m->timeout);
1948 if (s->reply_callback.timeout != 0) {
1949 r = prioq_put(bus->reply_callbacks_prioq, &s->reply_callback, &s->reply_callback.prioq_idx);
1950 if (r < 0) {
1951 s->reply_callback.timeout = 0;
1952 return r;
1953 }
1954 }
1955
1956 r = sd_bus_send(bus, m, &s->reply_callback.cookie);
1957 if (r < 0)
1958 return r;
1959
1960 if (slot)
1961 *slot = s;
1962 s = NULL;
1963
1964 return r;
1965 }
1966
1967 int bus_ensure_running(sd_bus *bus) {
1968 int r;
1969
1970 assert(bus);
1971
1972 if (bus->state == BUS_UNSET || bus->state == BUS_CLOSED || bus->state == BUS_CLOSING)
1973 return -ENOTCONN;
1974 if (bus->state == BUS_RUNNING)
1975 return 1;
1976
1977 for (;;) {
1978 r = sd_bus_process(bus, NULL);
1979 if (r < 0)
1980 return r;
1981 if (bus->state == BUS_RUNNING)
1982 return 1;
1983 if (r > 0)
1984 continue;
1985
1986 r = sd_bus_wait(bus, (uint64_t) -1);
1987 if (r < 0)
1988 return r;
1989 }
1990 }
1991
1992 _public_ int sd_bus_call(
1993 sd_bus *bus,
1994 sd_bus_message *_m,
1995 uint64_t usec,
1996 sd_bus_error *error,
1997 sd_bus_message **reply) {
1998
1999 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = sd_bus_message_ref(_m);
2000 usec_t timeout;
2001 uint64_t cookie;
2002 unsigned i;
2003 int r;
2004
2005 bus_assert_return(m, -EINVAL, error);
2006 bus_assert_return(m->header->type == SD_BUS_MESSAGE_METHOD_CALL, -EINVAL, error);
2007 bus_assert_return(!(m->header->flags & BUS_MESSAGE_NO_REPLY_EXPECTED), -EINVAL, error);
2008 bus_assert_return(!bus_error_is_dirty(error), -EINVAL, error);
2009
2010 if (!bus)
2011 bus = m->bus;
2012
2013 bus_assert_return(!bus_pid_changed(bus), -ECHILD, error);
2014 bus_assert_return(!bus->is_kernel || !(bus->hello_flags & KDBUS_HELLO_MONITOR), -EROFS, error);
2015
2016 if (!BUS_IS_OPEN(bus->state)) {
2017 r = -ENOTCONN;
2018 goto fail;
2019 }
2020
2021 r = bus_ensure_running(bus);
2022 if (r < 0)
2023 goto fail;
2024
2025 i = bus->rqueue_size;
2026
2027 r = bus_seal_message(bus, m, usec);
2028 if (r < 0)
2029 goto fail;
2030
2031 r = bus_remarshal_message(bus, &m);
2032 if (r < 0)
2033 goto fail;
2034
2035 r = bus_send_internal(bus, m, &cookie, true);
2036 if (r < 0)
2037 goto fail;
2038
2039 timeout = calc_elapse(m->timeout);
2040
2041 for (;;) {
2042 usec_t left;
2043
2044 while (i < bus->rqueue_size) {
2045 sd_bus_message *incoming = NULL;
2046
2047 incoming = bus->rqueue[i];
2048
2049 if (incoming->reply_cookie == cookie) {
2050 /* Found a match! */
2051
2052 memmove(bus->rqueue + i, bus->rqueue + i + 1, sizeof(sd_bus_message*) * (bus->rqueue_size - i - 1));
2053 bus->rqueue_size--;
2054 log_debug_bus_message(incoming);
2055
2056 if (incoming->header->type == SD_BUS_MESSAGE_METHOD_RETURN) {
2057
2058 if (incoming->n_fds <= 0 || (bus->hello_flags & KDBUS_HELLO_ACCEPT_FD)) {
2059 if (reply)
2060 *reply = incoming;
2061 else
2062 sd_bus_message_unref(incoming);
2063
2064 return 1;
2065 }
2066
2067 r = sd_bus_error_setf(error, SD_BUS_ERROR_INCONSISTENT_MESSAGE, "Reply message contained file descriptors which I couldn't accept. Sorry.");
2068 sd_bus_message_unref(incoming);
2069 return r;
2070
2071 } else if (incoming->header->type == SD_BUS_MESSAGE_METHOD_ERROR) {
2072 r = sd_bus_error_copy(error, &incoming->error);
2073 sd_bus_message_unref(incoming);
2074 return r;
2075 } else {
2076 r = -EIO;
2077 goto fail;
2078 }
2079
2080 } else if (BUS_MESSAGE_COOKIE(incoming) == cookie &&
2081 bus->unique_name &&
2082 incoming->sender &&
2083 streq(bus->unique_name, incoming->sender)) {
2084
2085 memmove(bus->rqueue + i, bus->rqueue + i + 1, sizeof(sd_bus_message*) * (bus->rqueue_size - i - 1));
2086 bus->rqueue_size--;
2087
2088 /* Our own message? Somebody is trying
2089 * to send its own client a message,
2090 * let's not dead-lock, let's fail
2091 * immediately. */
2092
2093 sd_bus_message_unref(incoming);
2094 r = -ELOOP;
2095 goto fail;
2096 }
2097
2098 /* Try to read more, right-away */
2099 i++;
2100 }
2101
2102 r = bus_read_message(bus, false, 0);
2103 if (r < 0) {
2104 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2105 bus_enter_closing(bus);
2106 r = -ECONNRESET;
2107 }
2108
2109 goto fail;
2110 }
2111 if (r > 0)
2112 continue;
2113
2114 if (timeout > 0) {
2115 usec_t n;
2116
2117 n = now(CLOCK_MONOTONIC);
2118 if (n >= timeout) {
2119 r = -ETIMEDOUT;
2120 goto fail;
2121 }
2122
2123 left = timeout - n;
2124 } else
2125 left = (uint64_t) -1;
2126
2127 r = bus_poll(bus, true, left);
2128 if (r < 0)
2129 goto fail;
2130 if (r == 0) {
2131 r = -ETIMEDOUT;
2132 goto fail;
2133 }
2134
2135 r = dispatch_wqueue(bus);
2136 if (r < 0) {
2137 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2138 bus_enter_closing(bus);
2139 r = -ECONNRESET;
2140 }
2141
2142 goto fail;
2143 }
2144 }
2145
2146 fail:
2147 return sd_bus_error_set_errno(error, r);
2148 }
2149
2150 _public_ int sd_bus_get_fd(sd_bus *bus) {
2151
2152 assert_return(bus, -EINVAL);
2153 assert_return(bus->input_fd == bus->output_fd, -EPERM);
2154 assert_return(!bus_pid_changed(bus), -ECHILD);
2155
2156 return bus->input_fd;
2157 }
2158
2159 _public_ int sd_bus_get_events(sd_bus *bus) {
2160 int flags = 0;
2161
2162 assert_return(bus, -EINVAL);
2163 assert_return(!bus_pid_changed(bus), -ECHILD);
2164
2165 if (!BUS_IS_OPEN(bus->state) && bus->state != BUS_CLOSING)
2166 return -ENOTCONN;
2167
2168 if (bus->state == BUS_OPENING)
2169 flags |= POLLOUT;
2170 else if (bus->state == BUS_AUTHENTICATING) {
2171
2172 if (bus_socket_auth_needs_write(bus))
2173 flags |= POLLOUT;
2174
2175 flags |= POLLIN;
2176
2177 } else if (bus->state == BUS_RUNNING || bus->state == BUS_HELLO) {
2178 if (bus->rqueue_size <= 0)
2179 flags |= POLLIN;
2180 if (bus->wqueue_size > 0)
2181 flags |= POLLOUT;
2182 }
2183
2184 return flags;
2185 }
2186
2187 _public_ int sd_bus_get_timeout(sd_bus *bus, uint64_t *timeout_usec) {
2188 struct reply_callback *c;
2189
2190 assert_return(bus, -EINVAL);
2191 assert_return(timeout_usec, -EINVAL);
2192 assert_return(!bus_pid_changed(bus), -ECHILD);
2193
2194 if (!BUS_IS_OPEN(bus->state) && bus->state != BUS_CLOSING)
2195 return -ENOTCONN;
2196
2197 if (bus->track_queue) {
2198 *timeout_usec = 0;
2199 return 1;
2200 }
2201
2202 if (bus->state == BUS_CLOSING) {
2203 *timeout_usec = 0;
2204 return 1;
2205 }
2206
2207 if (bus->state == BUS_AUTHENTICATING) {
2208 *timeout_usec = bus->auth_timeout;
2209 return 1;
2210 }
2211
2212 if (bus->state != BUS_RUNNING && bus->state != BUS_HELLO) {
2213 *timeout_usec = (uint64_t) -1;
2214 return 0;
2215 }
2216
2217 if (bus->rqueue_size > 0) {
2218 *timeout_usec = 0;
2219 return 1;
2220 }
2221
2222 c = prioq_peek(bus->reply_callbacks_prioq);
2223 if (!c) {
2224 *timeout_usec = (uint64_t) -1;
2225 return 0;
2226 }
2227
2228 if (c->timeout == 0) {
2229 *timeout_usec = (uint64_t) -1;
2230 return 0;
2231 }
2232
2233 *timeout_usec = c->timeout;
2234 return 1;
2235 }
2236
2237 static int process_timeout(sd_bus *bus) {
2238 _cleanup_(sd_bus_error_free) sd_bus_error error_buffer = SD_BUS_ERROR_NULL;
2239 _cleanup_(sd_bus_message_unrefp) sd_bus_message* m = NULL;
2240 struct reply_callback *c;
2241 sd_bus_slot *slot;
2242 usec_t n;
2243 int r;
2244
2245 assert(bus);
2246
2247 c = prioq_peek(bus->reply_callbacks_prioq);
2248 if (!c)
2249 return 0;
2250
2251 n = now(CLOCK_MONOTONIC);
2252 if (c->timeout > n)
2253 return 0;
2254
2255 r = bus_message_new_synthetic_error(
2256 bus,
2257 c->cookie,
2258 &SD_BUS_ERROR_MAKE_CONST(SD_BUS_ERROR_NO_REPLY, "Method call timed out"),
2259 &m);
2260 if (r < 0)
2261 return r;
2262
2263 r = bus_seal_synthetic_message(bus, m);
2264 if (r < 0)
2265 return r;
2266
2267 assert_se(prioq_pop(bus->reply_callbacks_prioq) == c);
2268 c->timeout = 0;
2269
2270 ordered_hashmap_remove(bus->reply_callbacks, &c->cookie);
2271 c->cookie = 0;
2272
2273 slot = container_of(c, sd_bus_slot, reply_callback);
2274
2275 bus->iteration_counter++;
2276
2277 bus->current_message = m;
2278 bus->current_slot = sd_bus_slot_ref(slot);
2279 bus->current_handler = c->callback;
2280 bus->current_userdata = slot->userdata;
2281 r = c->callback(m, slot->userdata, &error_buffer);
2282 bus->current_userdata = NULL;
2283 bus->current_handler = NULL;
2284 bus->current_slot = NULL;
2285 bus->current_message = NULL;
2286
2287 if (slot->floating) {
2288 bus_slot_disconnect(slot);
2289 sd_bus_slot_unref(slot);
2290 }
2291
2292 sd_bus_slot_unref(slot);
2293
2294 return bus_maybe_reply_error(m, r, &error_buffer);
2295 }
2296
2297 static int process_hello(sd_bus *bus, sd_bus_message *m) {
2298 assert(bus);
2299 assert(m);
2300
2301 if (bus->state != BUS_HELLO)
2302 return 0;
2303
2304 /* Let's make sure the first message on the bus is the HELLO
2305 * reply. But note that we don't actually parse the message
2306 * here (we leave that to the usual handling), we just verify
2307 * we don't let any earlier msg through. */
2308
2309 if (m->header->type != SD_BUS_MESSAGE_METHOD_RETURN &&
2310 m->header->type != SD_BUS_MESSAGE_METHOD_ERROR)
2311 return -EIO;
2312
2313 if (m->reply_cookie != 1)
2314 return -EIO;
2315
2316 return 0;
2317 }
2318
2319 static int process_reply(sd_bus *bus, sd_bus_message *m) {
2320 _cleanup_(sd_bus_message_unrefp) sd_bus_message *synthetic_reply = NULL;
2321 _cleanup_(sd_bus_error_free) sd_bus_error error_buffer = SD_BUS_ERROR_NULL;
2322 struct reply_callback *c;
2323 sd_bus_slot *slot;
2324 int r;
2325
2326 assert(bus);
2327 assert(m);
2328
2329 if (m->header->type != SD_BUS_MESSAGE_METHOD_RETURN &&
2330 m->header->type != SD_BUS_MESSAGE_METHOD_ERROR)
2331 return 0;
2332
2333 if (bus->is_kernel && (bus->hello_flags & KDBUS_HELLO_MONITOR))
2334 return 0;
2335
2336 if (m->destination && bus->unique_name && !streq_ptr(m->destination, bus->unique_name))
2337 return 0;
2338
2339 c = ordered_hashmap_remove(bus->reply_callbacks, &m->reply_cookie);
2340 if (!c)
2341 return 0;
2342
2343 c->cookie = 0;
2344
2345 slot = container_of(c, sd_bus_slot, reply_callback);
2346
2347 if (m->n_fds > 0 && !(bus->hello_flags & KDBUS_HELLO_ACCEPT_FD)) {
2348
2349 /* If the reply contained a file descriptor which we
2350 * didn't want we pass an error instead. */
2351
2352 r = bus_message_new_synthetic_error(
2353 bus,
2354 m->reply_cookie,
2355 &SD_BUS_ERROR_MAKE_CONST(SD_BUS_ERROR_INCONSISTENT_MESSAGE, "Reply message contained file descriptor"),
2356 &synthetic_reply);
2357 if (r < 0)
2358 return r;
2359
2360 /* Copy over original timestamp */
2361 synthetic_reply->realtime = m->realtime;
2362 synthetic_reply->monotonic = m->monotonic;
2363 synthetic_reply->seqnum = m->seqnum;
2364
2365 r = bus_seal_synthetic_message(bus, synthetic_reply);
2366 if (r < 0)
2367 return r;
2368
2369 m = synthetic_reply;
2370 } else {
2371 r = sd_bus_message_rewind(m, true);
2372 if (r < 0)
2373 return r;
2374 }
2375
2376 if (c->timeout != 0) {
2377 prioq_remove(bus->reply_callbacks_prioq, c, &c->prioq_idx);
2378 c->timeout = 0;
2379 }
2380
2381 bus->current_slot = sd_bus_slot_ref(slot);
2382 bus->current_handler = c->callback;
2383 bus->current_userdata = slot->userdata;
2384 r = c->callback(m, slot->userdata, &error_buffer);
2385 bus->current_userdata = NULL;
2386 bus->current_handler = NULL;
2387 bus->current_slot = NULL;
2388
2389 if (slot->floating) {
2390 bus_slot_disconnect(slot);
2391 sd_bus_slot_unref(slot);
2392 }
2393
2394 sd_bus_slot_unref(slot);
2395
2396 return bus_maybe_reply_error(m, r, &error_buffer);
2397 }
2398
2399 static int process_filter(sd_bus *bus, sd_bus_message *m) {
2400 _cleanup_(sd_bus_error_free) sd_bus_error error_buffer = SD_BUS_ERROR_NULL;
2401 struct filter_callback *l;
2402 int r;
2403
2404 assert(bus);
2405 assert(m);
2406
2407 do {
2408 bus->filter_callbacks_modified = false;
2409
2410 LIST_FOREACH(callbacks, l, bus->filter_callbacks) {
2411 sd_bus_slot *slot;
2412
2413 if (bus->filter_callbacks_modified)
2414 break;
2415
2416 /* Don't run this more than once per iteration */
2417 if (l->last_iteration == bus->iteration_counter)
2418 continue;
2419
2420 l->last_iteration = bus->iteration_counter;
2421
2422 r = sd_bus_message_rewind(m, true);
2423 if (r < 0)
2424 return r;
2425
2426 slot = container_of(l, sd_bus_slot, filter_callback);
2427
2428 bus->current_slot = sd_bus_slot_ref(slot);
2429 bus->current_handler = l->callback;
2430 bus->current_userdata = slot->userdata;
2431 r = l->callback(m, slot->userdata, &error_buffer);
2432 bus->current_userdata = NULL;
2433 bus->current_handler = NULL;
2434 bus->current_slot = sd_bus_slot_unref(slot);
2435
2436 r = bus_maybe_reply_error(m, r, &error_buffer);
2437 if (r != 0)
2438 return r;
2439
2440 }
2441
2442 } while (bus->filter_callbacks_modified);
2443
2444 return 0;
2445 }
2446
2447 static int process_match(sd_bus *bus, sd_bus_message *m) {
2448 int r;
2449
2450 assert(bus);
2451 assert(m);
2452
2453 do {
2454 bus->match_callbacks_modified = false;
2455
2456 r = bus_match_run(bus, &bus->match_callbacks, m);
2457 if (r != 0)
2458 return r;
2459
2460 } while (bus->match_callbacks_modified);
2461
2462 return 0;
2463 }
2464
2465 static int process_builtin(sd_bus *bus, sd_bus_message *m) {
2466 _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
2467 int r;
2468
2469 assert(bus);
2470 assert(m);
2471
2472 if (bus->hello_flags & KDBUS_HELLO_MONITOR)
2473 return 0;
2474
2475 if (bus->manual_peer_interface)
2476 return 0;
2477
2478 if (m->header->type != SD_BUS_MESSAGE_METHOD_CALL)
2479 return 0;
2480
2481 if (!streq_ptr(m->interface, "org.freedesktop.DBus.Peer"))
2482 return 0;
2483
2484 if (m->header->flags & BUS_MESSAGE_NO_REPLY_EXPECTED)
2485 return 1;
2486
2487 if (streq_ptr(m->member, "Ping"))
2488 r = sd_bus_message_new_method_return(m, &reply);
2489 else if (streq_ptr(m->member, "GetMachineId")) {
2490 sd_id128_t id;
2491 char sid[33];
2492
2493 r = sd_id128_get_machine(&id);
2494 if (r < 0)
2495 return r;
2496
2497 r = sd_bus_message_new_method_return(m, &reply);
2498 if (r < 0)
2499 return r;
2500
2501 r = sd_bus_message_append(reply, "s", sd_id128_to_string(id, sid));
2502 } else {
2503 r = sd_bus_message_new_method_errorf(
2504 m, &reply,
2505 SD_BUS_ERROR_UNKNOWN_METHOD,
2506 "Unknown method '%s' on interface '%s'.", m->member, m->interface);
2507 }
2508
2509 if (r < 0)
2510 return r;
2511
2512 r = sd_bus_send(bus, reply, NULL);
2513 if (r < 0)
2514 return r;
2515
2516 return 1;
2517 }
2518
2519 static int process_fd_check(sd_bus *bus, sd_bus_message *m) {
2520 assert(bus);
2521 assert(m);
2522
2523 /* If we got a message with a file descriptor which we didn't
2524 * want to accept, then let's drop it. How can this even
2525 * happen? For example, when the kernel queues a message into
2526 * an activatable names's queue which allows fds, and then is
2527 * delivered to us later even though we ourselves did not
2528 * negotiate it. */
2529
2530 if (bus->hello_flags & KDBUS_HELLO_MONITOR)
2531 return 0;
2532
2533 if (m->n_fds <= 0)
2534 return 0;
2535
2536 if (bus->hello_flags & KDBUS_HELLO_ACCEPT_FD)
2537 return 0;
2538
2539 if (m->header->type != SD_BUS_MESSAGE_METHOD_CALL)
2540 return 1; /* just eat it up */
2541
2542 return sd_bus_reply_method_errorf(m, SD_BUS_ERROR_INCONSISTENT_MESSAGE, "Message contains file descriptors, which I cannot accept. Sorry.");
2543 }
2544
2545 static int process_message(sd_bus *bus, sd_bus_message *m) {
2546 int r;
2547
2548 assert(bus);
2549 assert(m);
2550
2551 bus->current_message = m;
2552 bus->iteration_counter++;
2553
2554 log_debug_bus_message(m);
2555
2556 r = process_hello(bus, m);
2557 if (r != 0)
2558 goto finish;
2559
2560 r = process_reply(bus, m);
2561 if (r != 0)
2562 goto finish;
2563
2564 r = process_fd_check(bus, m);
2565 if (r != 0)
2566 goto finish;
2567
2568 r = process_filter(bus, m);
2569 if (r != 0)
2570 goto finish;
2571
2572 r = process_match(bus, m);
2573 if (r != 0)
2574 goto finish;
2575
2576 r = process_builtin(bus, m);
2577 if (r != 0)
2578 goto finish;
2579
2580 r = bus_process_object(bus, m);
2581
2582 finish:
2583 bus->current_message = NULL;
2584 return r;
2585 }
2586
2587 static int dispatch_track(sd_bus *bus) {
2588 assert(bus);
2589
2590 if (!bus->track_queue)
2591 return 0;
2592
2593 bus_track_dispatch(bus->track_queue);
2594 return 1;
2595 }
2596
2597 static int process_running(sd_bus *bus, bool hint_priority, int64_t priority, sd_bus_message **ret) {
2598 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
2599 int r;
2600
2601 assert(bus);
2602 assert(bus->state == BUS_RUNNING || bus->state == BUS_HELLO);
2603
2604 r = process_timeout(bus);
2605 if (r != 0)
2606 goto null_message;
2607
2608 r = dispatch_wqueue(bus);
2609 if (r != 0)
2610 goto null_message;
2611
2612 r = dispatch_track(bus);
2613 if (r != 0)
2614 goto null_message;
2615
2616 r = dispatch_rqueue(bus, hint_priority, priority, &m);
2617 if (r < 0)
2618 return r;
2619 if (!m)
2620 goto null_message;
2621
2622 r = process_message(bus, m);
2623 if (r != 0)
2624 goto null_message;
2625
2626 if (ret) {
2627 r = sd_bus_message_rewind(m, true);
2628 if (r < 0)
2629 return r;
2630
2631 *ret = m;
2632 m = NULL;
2633 return 1;
2634 }
2635
2636 if (m->header->type == SD_BUS_MESSAGE_METHOD_CALL) {
2637
2638 log_debug("Unprocessed message call sender=%s object=%s interface=%s member=%s",
2639 strna(sd_bus_message_get_sender(m)),
2640 strna(sd_bus_message_get_path(m)),
2641 strna(sd_bus_message_get_interface(m)),
2642 strna(sd_bus_message_get_member(m)));
2643
2644 r = sd_bus_reply_method_errorf(
2645 m,
2646 SD_BUS_ERROR_UNKNOWN_OBJECT,
2647 "Unknown object '%s'.", m->path);
2648 if (r < 0)
2649 return r;
2650 }
2651
2652 return 1;
2653
2654 null_message:
2655 if (r >= 0 && ret)
2656 *ret = NULL;
2657
2658 return r;
2659 }
2660
2661 static int bus_exit_now(sd_bus *bus) {
2662 assert(bus);
2663
2664 /* Exit due to close, if this is requested. If this is bus object is attached to an event source, invokes
2665 * sd_event_exit(), otherwise invokes libc exit(). */
2666
2667 if (bus->exited) /* did we already exit? */
2668 return 0;
2669 if (!bus->exit_triggered) /* was the exit condition triggered? */
2670 return 0;
2671 if (!bus->exit_on_disconnect) /* Shall we actually exit on disconnection? */
2672 return 0;
2673
2674 bus->exited = true; /* never exit more than once */
2675
2676 log_debug("Bus connection disconnected, exiting.");
2677
2678 if (bus->event)
2679 return sd_event_exit(bus->event, EXIT_FAILURE);
2680 else
2681 exit(EXIT_FAILURE);
2682
2683 assert_not_reached("exit() didn't exit?");
2684 }
2685
2686 static int process_closing_reply_callback(sd_bus *bus, struct reply_callback *c) {
2687 _cleanup_(sd_bus_error_free) sd_bus_error error_buffer = SD_BUS_ERROR_NULL;
2688 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
2689 sd_bus_slot *slot;
2690 int r;
2691
2692 assert(bus);
2693 assert(c);
2694
2695 r = bus_message_new_synthetic_error(
2696 bus,
2697 c->cookie,
2698 &SD_BUS_ERROR_MAKE_CONST(SD_BUS_ERROR_NO_REPLY, "Connection terminated"),
2699 &m);
2700 if (r < 0)
2701 return r;
2702
2703 r = bus_seal_synthetic_message(bus, m);
2704 if (r < 0)
2705 return r;
2706
2707 if (c->timeout != 0) {
2708 prioq_remove(bus->reply_callbacks_prioq, c, &c->prioq_idx);
2709 c->timeout = 0;
2710 }
2711
2712 ordered_hashmap_remove(bus->reply_callbacks, &c->cookie);
2713 c->cookie = 0;
2714
2715 slot = container_of(c, sd_bus_slot, reply_callback);
2716
2717 bus->iteration_counter++;
2718
2719 bus->current_message = m;
2720 bus->current_slot = sd_bus_slot_ref(slot);
2721 bus->current_handler = c->callback;
2722 bus->current_userdata = slot->userdata;
2723 r = c->callback(m, slot->userdata, &error_buffer);
2724 bus->current_userdata = NULL;
2725 bus->current_handler = NULL;
2726 bus->current_slot = NULL;
2727 bus->current_message = NULL;
2728
2729 if (slot->floating) {
2730 bus_slot_disconnect(slot);
2731 sd_bus_slot_unref(slot);
2732 }
2733
2734 sd_bus_slot_unref(slot);
2735
2736 return bus_maybe_reply_error(m, r, &error_buffer);
2737 }
2738
2739 static int process_closing(sd_bus *bus, sd_bus_message **ret) {
2740 _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
2741 struct reply_callback *c;
2742 int r;
2743
2744 assert(bus);
2745 assert(bus->state == BUS_CLOSING);
2746
2747 /* First, fail all outstanding method calls */
2748 c = ordered_hashmap_first(bus->reply_callbacks);
2749 if (c)
2750 return process_closing_reply_callback(bus, c);
2751
2752 /* Then, fake-drop all remaining bus tracking references */
2753 if (bus->tracks) {
2754 bus_track_close(bus->tracks);
2755 return 1;
2756 }
2757
2758 /* Then, synthesize a Disconnected message */
2759 r = sd_bus_message_new_signal(
2760 bus,
2761 &m,
2762 "/org/freedesktop/DBus/Local",
2763 "org.freedesktop.DBus.Local",
2764 "Disconnected");
2765 if (r < 0)
2766 return r;
2767
2768 bus_message_set_sender_local(bus, m);
2769
2770 r = bus_seal_synthetic_message(bus, m);
2771 if (r < 0)
2772 return r;
2773
2774 sd_bus_close(bus);
2775
2776 bus->current_message = m;
2777 bus->iteration_counter++;
2778
2779 r = process_filter(bus, m);
2780 if (r != 0)
2781 goto finish;
2782
2783 r = process_match(bus, m);
2784 if (r != 0)
2785 goto finish;
2786
2787 /* Nothing else to do, exit now, if the condition holds */
2788 bus->exit_triggered = true;
2789 (void) bus_exit_now(bus);
2790
2791 if (ret) {
2792 *ret = m;
2793 m = NULL;
2794 }
2795
2796 r = 1;
2797
2798 finish:
2799 bus->current_message = NULL;
2800
2801 return r;
2802 }
2803
2804 static int bus_process_internal(sd_bus *bus, bool hint_priority, int64_t priority, sd_bus_message **ret) {
2805 BUS_DONT_DESTROY(bus);
2806 int r;
2807
2808 /* Returns 0 when we didn't do anything. This should cause the
2809 * caller to invoke sd_bus_wait() before returning the next
2810 * time. Returns > 0 when we did something, which possibly
2811 * means *ret is filled in with an unprocessed message. */
2812
2813 assert_return(bus, -EINVAL);
2814 assert_return(!bus_pid_changed(bus), -ECHILD);
2815
2816 /* We don't allow recursively invoking sd_bus_process(). */
2817 assert_return(!bus->current_message, -EBUSY);
2818 assert(!bus->current_slot);
2819
2820 switch (bus->state) {
2821
2822 case BUS_UNSET:
2823 return -ENOTCONN;
2824
2825 case BUS_CLOSED:
2826 return -ECONNRESET;
2827
2828 case BUS_OPENING:
2829 r = bus_socket_process_opening(bus);
2830 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2831 bus_enter_closing(bus);
2832 r = 1;
2833 } else if (r < 0)
2834 return r;
2835 if (ret)
2836 *ret = NULL;
2837 return r;
2838
2839 case BUS_AUTHENTICATING:
2840 r = bus_socket_process_authenticating(bus);
2841 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2842 bus_enter_closing(bus);
2843 r = 1;
2844 } else if (r < 0)
2845 return r;
2846
2847 if (ret)
2848 *ret = NULL;
2849
2850 return r;
2851
2852 case BUS_RUNNING:
2853 case BUS_HELLO:
2854 r = process_running(bus, hint_priority, priority, ret);
2855 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2856 bus_enter_closing(bus);
2857 r = 1;
2858
2859 if (ret)
2860 *ret = NULL;
2861 }
2862
2863 return r;
2864
2865 case BUS_CLOSING:
2866 return process_closing(bus, ret);
2867 }
2868
2869 assert_not_reached("Unknown state");
2870 }
2871
2872 _public_ int sd_bus_process(sd_bus *bus, sd_bus_message **ret) {
2873 return bus_process_internal(bus, false, 0, ret);
2874 }
2875
2876 _public_ int sd_bus_process_priority(sd_bus *bus, int64_t priority, sd_bus_message **ret) {
2877 return bus_process_internal(bus, true, priority, ret);
2878 }
2879
2880 static int bus_poll(sd_bus *bus, bool need_more, uint64_t timeout_usec) {
2881 struct pollfd p[2] = {};
2882 int r, e, n;
2883 struct timespec ts;
2884 usec_t m = USEC_INFINITY;
2885
2886 assert(bus);
2887
2888 if (bus->state == BUS_CLOSING)
2889 return 1;
2890
2891 if (!BUS_IS_OPEN(bus->state))
2892 return -ENOTCONN;
2893
2894 e = sd_bus_get_events(bus);
2895 if (e < 0)
2896 return e;
2897
2898 if (need_more)
2899 /* The caller really needs some more data, he doesn't
2900 * care about what's already read, or any timeouts
2901 * except its own. */
2902 e |= POLLIN;
2903 else {
2904 usec_t until;
2905 /* The caller wants to process if there's something to
2906 * process, but doesn't care otherwise */
2907
2908 r = sd_bus_get_timeout(bus, &until);
2909 if (r < 0)
2910 return r;
2911 if (r > 0) {
2912 usec_t nw;
2913 nw = now(CLOCK_MONOTONIC);
2914 m = until > nw ? until - nw : 0;
2915 }
2916 }
2917
2918 if (timeout_usec != (uint64_t) -1 && (m == (uint64_t) -1 || timeout_usec < m))
2919 m = timeout_usec;
2920
2921 p[0].fd = bus->input_fd;
2922 if (bus->output_fd == bus->input_fd) {
2923 p[0].events = e;
2924 n = 1;
2925 } else {
2926 p[0].events = e & POLLIN;
2927 p[1].fd = bus->output_fd;
2928 p[1].events = e & POLLOUT;
2929 n = 2;
2930 }
2931
2932 r = ppoll(p, n, m == (uint64_t) -1 ? NULL : timespec_store(&ts, m), NULL);
2933 if (r < 0)
2934 return -errno;
2935
2936 return r > 0 ? 1 : 0;
2937 }
2938
2939 _public_ int sd_bus_wait(sd_bus *bus, uint64_t timeout_usec) {
2940
2941 assert_return(bus, -EINVAL);
2942 assert_return(!bus_pid_changed(bus), -ECHILD);
2943
2944 if (bus->state == BUS_CLOSING)
2945 return 0;
2946
2947 if (!BUS_IS_OPEN(bus->state))
2948 return -ENOTCONN;
2949
2950 if (bus->rqueue_size > 0)
2951 return 0;
2952
2953 return bus_poll(bus, false, timeout_usec);
2954 }
2955
2956 _public_ int sd_bus_flush(sd_bus *bus) {
2957 int r;
2958
2959 assert_return(bus, -EINVAL);
2960 assert_return(!bus_pid_changed(bus), -ECHILD);
2961
2962 if (bus->state == BUS_CLOSING)
2963 return 0;
2964
2965 if (!BUS_IS_OPEN(bus->state))
2966 return -ENOTCONN;
2967
2968 r = bus_ensure_running(bus);
2969 if (r < 0)
2970 return r;
2971
2972 if (bus->wqueue_size <= 0)
2973 return 0;
2974
2975 for (;;) {
2976 r = dispatch_wqueue(bus);
2977 if (r < 0) {
2978 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2979 bus_enter_closing(bus);
2980 return -ECONNRESET;
2981 }
2982
2983 return r;
2984 }
2985
2986 if (bus->wqueue_size <= 0)
2987 return 0;
2988
2989 r = bus_poll(bus, false, (uint64_t) -1);
2990 if (r < 0)
2991 return r;
2992 }
2993 }
2994
2995 _public_ int sd_bus_add_filter(
2996 sd_bus *bus,
2997 sd_bus_slot **slot,
2998 sd_bus_message_handler_t callback,
2999 void *userdata) {
3000
3001 sd_bus_slot *s;
3002
3003 assert_return(bus, -EINVAL);
3004 assert_return(callback, -EINVAL);
3005 assert_return(!bus_pid_changed(bus), -ECHILD);
3006
3007 s = bus_slot_allocate(bus, !slot, BUS_FILTER_CALLBACK, sizeof(struct filter_callback), userdata);
3008 if (!s)
3009 return -ENOMEM;
3010
3011 s->filter_callback.callback = callback;
3012
3013 bus->filter_callbacks_modified = true;
3014 LIST_PREPEND(callbacks, bus->filter_callbacks, &s->filter_callback);
3015
3016 if (slot)
3017 *slot = s;
3018
3019 return 0;
3020 }
3021
3022 _public_ int sd_bus_add_match(
3023 sd_bus *bus,
3024 sd_bus_slot **slot,
3025 const char *match,
3026 sd_bus_message_handler_t callback,
3027 void *userdata) {
3028
3029 struct bus_match_component *components = NULL;
3030 unsigned n_components = 0;
3031 sd_bus_slot *s = NULL;
3032 int r = 0;
3033
3034 assert_return(bus, -EINVAL);
3035 assert_return(match, -EINVAL);
3036 assert_return(!bus_pid_changed(bus), -ECHILD);
3037
3038 r = bus_match_parse(match, &components, &n_components);
3039 if (r < 0)
3040 goto finish;
3041
3042 s = bus_slot_allocate(bus, !slot, BUS_MATCH_CALLBACK, sizeof(struct match_callback), userdata);
3043 if (!s) {
3044 r = -ENOMEM;
3045 goto finish;
3046 }
3047
3048 s->match_callback.callback = callback;
3049 s->match_callback.cookie = ++bus->match_cookie;
3050
3051 if (bus->bus_client) {
3052 enum bus_match_scope scope;
3053
3054 scope = bus_match_get_scope(components, n_components);
3055
3056 /* Do not install server-side matches for matches
3057 * against the local service, interface or bus
3058 * path. */
3059 if (scope != BUS_MATCH_LOCAL) {
3060
3061 if (!bus->is_kernel) {
3062 /* When this is not a kernel transport, we
3063 * store the original match string, so that we
3064 * can use it to remove the match again */
3065
3066 s->match_callback.match_string = strdup(match);
3067 if (!s->match_callback.match_string) {
3068 r = -ENOMEM;
3069 goto finish;
3070 }
3071 }
3072
3073 r = bus_add_match_internal(bus, s->match_callback.match_string, components, n_components, s->match_callback.cookie);
3074 if (r < 0)
3075 goto finish;
3076
3077 s->match_added = true;
3078 }
3079 }
3080
3081 bus->match_callbacks_modified = true;
3082 r = bus_match_add(&bus->match_callbacks, components, n_components, &s->match_callback);
3083 if (r < 0)
3084 goto finish;
3085
3086 if (slot)
3087 *slot = s;
3088 s = NULL;
3089
3090 finish:
3091 bus_match_parse_free(components, n_components);
3092 sd_bus_slot_unref(s);
3093
3094 return r;
3095 }
3096
3097 int bus_remove_match_by_string(
3098 sd_bus *bus,
3099 const char *match,
3100 sd_bus_message_handler_t callback,
3101 void *userdata) {
3102
3103 struct bus_match_component *components = NULL;
3104 unsigned n_components = 0;
3105 struct match_callback *c;
3106 int r = 0;
3107
3108 assert_return(bus, -EINVAL);
3109 assert_return(match, -EINVAL);
3110 assert_return(!bus_pid_changed(bus), -ECHILD);
3111
3112 r = bus_match_parse(match, &components, &n_components);
3113 if (r < 0)
3114 goto finish;
3115
3116 r = bus_match_find(&bus->match_callbacks, components, n_components, NULL, NULL, &c);
3117 if (r <= 0)
3118 goto finish;
3119
3120 sd_bus_slot_unref(container_of(c, sd_bus_slot, match_callback));
3121
3122 finish:
3123 bus_match_parse_free(components, n_components);
3124
3125 return r;
3126 }
3127
3128 bool bus_pid_changed(sd_bus *bus) {
3129 assert(bus);
3130
3131 /* We don't support people creating a bus connection and
3132 * keeping it around over a fork(). Let's complain. */
3133
3134 return bus->original_pid != getpid_cached();
3135 }
3136
3137 static int io_callback(sd_event_source *s, int fd, uint32_t revents, void *userdata) {
3138 sd_bus *bus = userdata;
3139 int r;
3140
3141 assert(bus);
3142
3143 r = sd_bus_process(bus, NULL);
3144 if (r < 0)
3145 return r;
3146
3147 return 1;
3148 }
3149
3150 static int time_callback(sd_event_source *s, uint64_t usec, void *userdata) {
3151 sd_bus *bus = userdata;
3152 int r;
3153
3154 assert(bus);
3155
3156 r = sd_bus_process(bus, NULL);
3157 if (r < 0)
3158 return r;
3159
3160 return 1;
3161 }
3162
3163 static int prepare_callback(sd_event_source *s, void *userdata) {
3164 sd_bus *bus = userdata;
3165 int r, e;
3166 usec_t until;
3167
3168 assert(s);
3169 assert(bus);
3170
3171 e = sd_bus_get_events(bus);
3172 if (e < 0)
3173 return e;
3174
3175 if (bus->output_fd != bus->input_fd) {
3176
3177 r = sd_event_source_set_io_events(bus->input_io_event_source, e & POLLIN);
3178 if (r < 0)
3179 return r;
3180
3181 r = sd_event_source_set_io_events(bus->output_io_event_source, e & POLLOUT);
3182 if (r < 0)
3183 return r;
3184 } else {
3185 r = sd_event_source_set_io_events(bus->input_io_event_source, e);
3186 if (r < 0)
3187 return r;
3188 }
3189
3190 r = sd_bus_get_timeout(bus, &until);
3191 if (r < 0)
3192 return r;
3193 if (r > 0) {
3194 int j;
3195
3196 j = sd_event_source_set_time(bus->time_event_source, until);
3197 if (j < 0)
3198 return j;
3199 }
3200
3201 r = sd_event_source_set_enabled(bus->time_event_source, r > 0);
3202 if (r < 0)
3203 return r;
3204
3205 return 1;
3206 }
3207
3208 static int quit_callback(sd_event_source *event, void *userdata) {
3209 sd_bus *bus = userdata;
3210
3211 assert(event);
3212
3213 sd_bus_flush(bus);
3214 sd_bus_close(bus);
3215
3216 return 1;
3217 }
3218
3219 static int attach_io_events(sd_bus *bus) {
3220 int r;
3221
3222 assert(bus);
3223
3224 if (bus->input_fd < 0)
3225 return 0;
3226
3227 if (!bus->event)
3228 return 0;
3229
3230 if (!bus->input_io_event_source) {
3231 r = sd_event_add_io(bus->event, &bus->input_io_event_source, bus->input_fd, 0, io_callback, bus);
3232 if (r < 0)
3233 return r;
3234
3235 r = sd_event_source_set_prepare(bus->input_io_event_source, prepare_callback);
3236 if (r < 0)
3237 return r;
3238
3239 r = sd_event_source_set_priority(bus->input_io_event_source, bus->event_priority);
3240 if (r < 0)
3241 return r;
3242
3243 r = sd_event_source_set_description(bus->input_io_event_source, "bus-input");
3244 } else
3245 r = sd_event_source_set_io_fd(bus->input_io_event_source, bus->input_fd);
3246
3247 if (r < 0)
3248 return r;
3249
3250 if (bus->output_fd != bus->input_fd) {
3251 assert(bus->output_fd >= 0);
3252
3253 if (!bus->output_io_event_source) {
3254 r = sd_event_add_io(bus->event, &bus->output_io_event_source, bus->output_fd, 0, io_callback, bus);
3255 if (r < 0)
3256 return r;
3257
3258 r = sd_event_source_set_priority(bus->output_io_event_source, bus->event_priority);
3259 if (r < 0)
3260 return r;
3261
3262 r = sd_event_source_set_description(bus->input_io_event_source, "bus-output");
3263 } else
3264 r = sd_event_source_set_io_fd(bus->output_io_event_source, bus->output_fd);
3265
3266 if (r < 0)
3267 return r;
3268 }
3269
3270 return 0;
3271 }
3272
3273 static void detach_io_events(sd_bus *bus) {
3274 assert(bus);
3275
3276 if (bus->input_io_event_source) {
3277 sd_event_source_set_enabled(bus->input_io_event_source, SD_EVENT_OFF);
3278 bus->input_io_event_source = sd_event_source_unref(bus->input_io_event_source);
3279 }
3280
3281 if (bus->output_io_event_source) {
3282 sd_event_source_set_enabled(bus->output_io_event_source, SD_EVENT_OFF);
3283 bus->output_io_event_source = sd_event_source_unref(bus->output_io_event_source);
3284 }
3285 }
3286
3287 _public_ int sd_bus_attach_event(sd_bus *bus, sd_event *event, int priority) {
3288 int r;
3289
3290 assert_return(bus, -EINVAL);
3291 assert_return(!bus->event, -EBUSY);
3292
3293 assert(!bus->input_io_event_source);
3294 assert(!bus->output_io_event_source);
3295 assert(!bus->time_event_source);
3296
3297 if (event)
3298 bus->event = sd_event_ref(event);
3299 else {
3300 r = sd_event_default(&bus->event);
3301 if (r < 0)
3302 return r;
3303 }
3304
3305 bus->event_priority = priority;
3306
3307 r = sd_event_add_time(bus->event, &bus->time_event_source, CLOCK_MONOTONIC, 0, 0, time_callback, bus);
3308 if (r < 0)
3309 goto fail;
3310
3311 r = sd_event_source_set_priority(bus->time_event_source, priority);
3312 if (r < 0)
3313 goto fail;
3314
3315 r = sd_event_source_set_description(bus->time_event_source, "bus-time");
3316 if (r < 0)
3317 goto fail;
3318
3319 r = sd_event_add_exit(bus->event, &bus->quit_event_source, quit_callback, bus);
3320 if (r < 0)
3321 goto fail;
3322
3323 r = sd_event_source_set_description(bus->quit_event_source, "bus-exit");
3324 if (r < 0)
3325 goto fail;
3326
3327 r = attach_io_events(bus);
3328 if (r < 0)
3329 goto fail;
3330
3331 return 0;
3332
3333 fail:
3334 sd_bus_detach_event(bus);
3335 return r;
3336 }
3337
3338 _public_ int sd_bus_detach_event(sd_bus *bus) {
3339 assert_return(bus, -EINVAL);
3340
3341 if (!bus->event)
3342 return 0;
3343
3344 detach_io_events(bus);
3345
3346 if (bus->time_event_source) {
3347 sd_event_source_set_enabled(bus->time_event_source, SD_EVENT_OFF);
3348 bus->time_event_source = sd_event_source_unref(bus->time_event_source);
3349 }
3350
3351 if (bus->quit_event_source) {
3352 sd_event_source_set_enabled(bus->quit_event_source, SD_EVENT_OFF);
3353 bus->quit_event_source = sd_event_source_unref(bus->quit_event_source);
3354 }
3355
3356 bus->event = sd_event_unref(bus->event);
3357 return 1;
3358 }
3359
3360 _public_ sd_event* sd_bus_get_event(sd_bus *bus) {
3361 assert_return(bus, NULL);
3362
3363 return bus->event;
3364 }
3365
3366 _public_ sd_bus_message* sd_bus_get_current_message(sd_bus *bus) {
3367 assert_return(bus, NULL);
3368
3369 return bus->current_message;
3370 }
3371
3372 _public_ sd_bus_slot* sd_bus_get_current_slot(sd_bus *bus) {
3373 assert_return(bus, NULL);
3374
3375 return bus->current_slot;
3376 }
3377
3378 _public_ sd_bus_message_handler_t sd_bus_get_current_handler(sd_bus *bus) {
3379 assert_return(bus, NULL);
3380
3381 return bus->current_handler;
3382 }
3383
3384 _public_ void* sd_bus_get_current_userdata(sd_bus *bus) {
3385 assert_return(bus, NULL);
3386
3387 return bus->current_userdata;
3388 }
3389
3390 static int bus_default(int (*bus_open)(sd_bus **), sd_bus **default_bus, sd_bus **ret) {
3391 sd_bus *b = NULL;
3392 int r;
3393
3394 assert(bus_open);
3395 assert(default_bus);
3396
3397 if (!ret)
3398 return !!*default_bus;
3399
3400 if (*default_bus) {
3401 *ret = sd_bus_ref(*default_bus);
3402 return 0;
3403 }
3404
3405 r = bus_open(&b);
3406 if (r < 0)
3407 return r;
3408
3409 b->default_bus_ptr = default_bus;
3410 b->tid = gettid();
3411 *default_bus = b;
3412
3413 *ret = b;
3414 return 1;
3415 }
3416
3417 _public_ int sd_bus_default_system(sd_bus **ret) {
3418 return bus_default(sd_bus_open_system, &default_system_bus, ret);
3419 }
3420
3421
3422 _public_ int sd_bus_default_user(sd_bus **ret) {
3423 return bus_default(sd_bus_open_user, &default_user_bus, ret);
3424 }
3425
3426 _public_ int sd_bus_default(sd_bus **ret) {
3427
3428 const char *e;
3429
3430 /* Let's try our best to reuse another cached connection. If
3431 * the starter bus type is set, connect via our normal
3432 * connection logic, ignoring $DBUS_STARTER_ADDRESS, so that
3433 * we can share the connection with the user/system default
3434 * bus. */
3435
3436 e = secure_getenv("DBUS_STARTER_BUS_TYPE");
3437 if (e) {
3438 if (streq(e, "system"))
3439 return sd_bus_default_system(ret);
3440 else if (STR_IN_SET(e, "user", "session"))
3441 return sd_bus_default_user(ret);
3442 }
3443
3444 /* No type is specified, so we have not other option than to
3445 * use the starter address if it is set. */
3446
3447 e = secure_getenv("DBUS_STARTER_ADDRESS");
3448 if (e) {
3449
3450 return bus_default(sd_bus_open, &default_starter_bus, ret);
3451 }
3452
3453 /* Finally, if nothing is set use the cached connection for
3454 * the right scope */
3455
3456 if (cg_pid_get_owner_uid(0, NULL) >= 0)
3457 return sd_bus_default_user(ret);
3458 else
3459 return sd_bus_default_system(ret);
3460 }
3461
3462 _public_ int sd_bus_get_tid(sd_bus *b, pid_t *tid) {
3463 assert_return(b, -EINVAL);
3464 assert_return(tid, -EINVAL);
3465 assert_return(!bus_pid_changed(b), -ECHILD);
3466
3467 if (b->tid != 0) {
3468 *tid = b->tid;
3469 return 0;
3470 }
3471
3472 if (b->event)
3473 return sd_event_get_tid(b->event, tid);
3474
3475 return -ENXIO;
3476 }
3477
3478 _public_ int sd_bus_path_encode(const char *prefix, const char *external_id, char **ret_path) {
3479 _cleanup_free_ char *e = NULL;
3480 char *ret;
3481
3482 assert_return(object_path_is_valid(prefix), -EINVAL);
3483 assert_return(external_id, -EINVAL);
3484 assert_return(ret_path, -EINVAL);
3485
3486 e = bus_label_escape(external_id);
3487 if (!e)
3488 return -ENOMEM;
3489
3490 ret = strjoin(prefix, "/", e);
3491 if (!ret)
3492 return -ENOMEM;
3493
3494 *ret_path = ret;
3495 return 0;
3496 }
3497
3498 _public_ int sd_bus_path_decode(const char *path, const char *prefix, char **external_id) {
3499 const char *e;
3500 char *ret;
3501
3502 assert_return(object_path_is_valid(path), -EINVAL);
3503 assert_return(object_path_is_valid(prefix), -EINVAL);
3504 assert_return(external_id, -EINVAL);
3505
3506 e = object_path_startswith(path, prefix);
3507 if (!e) {
3508 *external_id = NULL;
3509 return 0;
3510 }
3511
3512 ret = bus_label_unescape(e);
3513 if (!ret)
3514 return -ENOMEM;
3515
3516 *external_id = ret;
3517 return 1;
3518 }
3519
3520 _public_ int sd_bus_path_encode_many(char **out, const char *path_template, ...) {
3521 _cleanup_strv_free_ char **labels = NULL;
3522 char *path, *path_pos, **label_pos;
3523 const char *sep, *template_pos;
3524 size_t path_length;
3525 va_list list;
3526 int r;
3527
3528 assert_return(out, -EINVAL);
3529 assert_return(path_template, -EINVAL);
3530
3531 path_length = strlen(path_template);
3532
3533 va_start(list, path_template);
3534 for (sep = strchr(path_template, '%'); sep; sep = strchr(sep + 1, '%')) {
3535 const char *arg;
3536 char *label;
3537
3538 arg = va_arg(list, const char *);
3539 if (!arg) {
3540 va_end(list);
3541 return -EINVAL;
3542 }
3543
3544 label = bus_label_escape(arg);
3545 if (!label) {
3546 va_end(list);
3547 return -ENOMEM;
3548 }
3549
3550 r = strv_consume(&labels, label);
3551 if (r < 0) {
3552 va_end(list);
3553 return r;
3554 }
3555
3556 /* add label length, but account for the format character */
3557 path_length += strlen(label) - 1;
3558 }
3559 va_end(list);
3560
3561 path = malloc(path_length + 1);
3562 if (!path)
3563 return -ENOMEM;
3564
3565 path_pos = path;
3566 label_pos = labels;
3567
3568 for (template_pos = path_template; *template_pos; ) {
3569 sep = strchrnul(template_pos, '%');
3570 path_pos = mempcpy(path_pos, template_pos, sep - template_pos);
3571 if (!*sep)
3572 break;
3573
3574 path_pos = stpcpy(path_pos, *label_pos++);
3575 template_pos = sep + 1;
3576 }
3577
3578 *path_pos = 0;
3579 *out = path;
3580 return 0;
3581 }
3582
3583 _public_ int sd_bus_path_decode_many(const char *path, const char *path_template, ...) {
3584 _cleanup_strv_free_ char **labels = NULL;
3585 const char *template_pos, *path_pos;
3586 char **label_pos;
3587 va_list list;
3588 int r;
3589
3590 /*
3591 * This decodes an object-path based on a template argument. The
3592 * template consists of a verbatim path, optionally including special
3593 * directives:
3594 *
3595 * - Each occurrence of '%' in the template matches an arbitrary
3596 * substring of a label in the given path. At most one such
3597 * directive is allowed per label. For each such directive, the
3598 * caller must provide an output parameter (char **) via va_arg. If
3599 * NULL is passed, the given label is verified, but not returned.
3600 * For each matched label, the *decoded* label is stored in the
3601 * passed output argument, and the caller is responsible to free
3602 * it. Note that the output arguments are only modified if the
3603 * actualy path matched the template. Otherwise, they're left
3604 * untouched.
3605 *
3606 * This function returns <0 on error, 0 if the path does not match the
3607 * template, 1 if it matched.
3608 */
3609
3610 assert_return(path, -EINVAL);
3611 assert_return(path_template, -EINVAL);
3612
3613 path_pos = path;
3614
3615 for (template_pos = path_template; *template_pos; ) {
3616 const char *sep;
3617 size_t length;
3618 char *label;
3619
3620 /* verify everything until the next '%' matches verbatim */
3621 sep = strchrnul(template_pos, '%');
3622 length = sep - template_pos;
3623 if (strncmp(path_pos, template_pos, length))
3624 return 0;
3625
3626 path_pos += length;
3627 template_pos += length;
3628
3629 if (!*template_pos)
3630 break;
3631
3632 /* We found the next '%' character. Everything up until here
3633 * matched. We now skip ahead to the end of this label and make
3634 * sure it matches the tail of the label in the path. Then we
3635 * decode the string in-between and save it for later use. */
3636
3637 ++template_pos; /* skip over '%' */
3638
3639 sep = strchrnul(template_pos, '/');
3640 length = sep - template_pos; /* length of suffix to match verbatim */
3641
3642 /* verify the suffixes match */
3643 sep = strchrnul(path_pos, '/');
3644 if (sep - path_pos < (ssize_t)length ||
3645 strncmp(sep - length, template_pos, length))
3646 return 0;
3647
3648 template_pos += length; /* skip over matched label */
3649 length = sep - path_pos - length; /* length of sub-label to decode */
3650
3651 /* store unescaped label for later use */
3652 label = bus_label_unescape_n(path_pos, length);
3653 if (!label)
3654 return -ENOMEM;
3655
3656 r = strv_consume(&labels, label);
3657 if (r < 0)
3658 return r;
3659
3660 path_pos = sep; /* skip decoded label and suffix */
3661 }
3662
3663 /* end of template must match end of path */
3664 if (*path_pos)
3665 return 0;
3666
3667 /* copy the labels over to the caller */
3668 va_start(list, path_template);
3669 for (label_pos = labels; label_pos && *label_pos; ++label_pos) {
3670 char **arg;
3671
3672 arg = va_arg(list, char **);
3673 if (arg)
3674 *arg = *label_pos;
3675 else
3676 free(*label_pos);
3677 }
3678 va_end(list);
3679
3680 free(labels);
3681 labels = NULL;
3682 return 1;
3683 }
3684
3685 _public_ int sd_bus_try_close(sd_bus *bus) {
3686 int r;
3687
3688 assert_return(bus, -EINVAL);
3689 assert_return(!bus_pid_changed(bus), -ECHILD);
3690
3691 if (!bus->is_kernel)
3692 return -EOPNOTSUPP;
3693
3694 if (!BUS_IS_OPEN(bus->state))
3695 return -ENOTCONN;
3696
3697 if (bus->rqueue_size > 0)
3698 return -EBUSY;
3699
3700 if (bus->wqueue_size > 0)
3701 return -EBUSY;
3702
3703 r = bus_kernel_try_close(bus);
3704 if (r < 0)
3705 return r;
3706
3707 sd_bus_close(bus);
3708 return 0;
3709 }
3710
3711 _public_ int sd_bus_get_description(sd_bus *bus, const char **description) {
3712 assert_return(bus, -EINVAL);
3713 assert_return(description, -EINVAL);
3714 assert_return(bus->description, -ENXIO);
3715 assert_return(!bus_pid_changed(bus), -ECHILD);
3716
3717 *description = bus->description;
3718 return 0;
3719 }
3720
3721 int bus_get_root_path(sd_bus *bus) {
3722 int r;
3723
3724 if (bus->cgroup_root)
3725 return 0;
3726
3727 r = cg_get_root_path(&bus->cgroup_root);
3728 if (r == -ENOENT) {
3729 bus->cgroup_root = strdup("/");
3730 if (!bus->cgroup_root)
3731 return -ENOMEM;
3732
3733 r = 0;
3734 }
3735
3736 return r;
3737 }
3738
3739 _public_ int sd_bus_get_scope(sd_bus *bus, const char **scope) {
3740 int r;
3741
3742 assert_return(bus, -EINVAL);
3743 assert_return(scope, -EINVAL);
3744 assert_return(!bus_pid_changed(bus), -ECHILD);
3745
3746 if (bus->is_kernel) {
3747 _cleanup_free_ char *n = NULL;
3748 const char *dash;
3749
3750 r = bus_kernel_get_bus_name(bus, &n);
3751 if (r < 0)
3752 return r;
3753
3754 if (streq(n, "0-system")) {
3755 *scope = "system";
3756 return 0;
3757 }
3758
3759 dash = strchr(n, '-');
3760 if (streq_ptr(dash, "-user")) {
3761 *scope = "user";
3762 return 0;
3763 }
3764 }
3765
3766 if (bus->is_user) {
3767 *scope = "user";
3768 return 0;
3769 }
3770
3771 if (bus->is_system) {
3772 *scope = "system";
3773 return 0;
3774 }
3775
3776 return -ENODATA;
3777 }
3778
3779 _public_ int sd_bus_get_address(sd_bus *bus, const char **address) {
3780
3781 assert_return(bus, -EINVAL);
3782 assert_return(address, -EINVAL);
3783 assert_return(!bus_pid_changed(bus), -ECHILD);
3784
3785 if (bus->address) {
3786 *address = bus->address;
3787 return 0;
3788 }
3789
3790 return -ENODATA;
3791 }
3792
3793 _public_ int sd_bus_get_creds_mask(sd_bus *bus, uint64_t *mask) {
3794 assert_return(bus, -EINVAL);
3795 assert_return(mask, -EINVAL);
3796 assert_return(!bus_pid_changed(bus), -ECHILD);
3797
3798 *mask = bus->creds_mask;
3799 return 0;
3800 }
3801
3802 _public_ int sd_bus_is_bus_client(sd_bus *bus) {
3803 assert_return(bus, -EINVAL);
3804 assert_return(!bus_pid_changed(bus), -ECHILD);
3805
3806 return bus->bus_client;
3807 }
3808
3809 _public_ int sd_bus_is_server(sd_bus *bus) {
3810 assert_return(bus, -EINVAL);
3811 assert_return(!bus_pid_changed(bus), -ECHILD);
3812
3813 return bus->is_server;
3814 }
3815
3816 _public_ int sd_bus_is_anonymous(sd_bus *bus) {
3817 assert_return(bus, -EINVAL);
3818 assert_return(!bus_pid_changed(bus), -ECHILD);
3819
3820 return bus->anonymous_auth;
3821 }
3822
3823 _public_ int sd_bus_is_trusted(sd_bus *bus) {
3824 assert_return(bus, -EINVAL);
3825 assert_return(!bus_pid_changed(bus), -ECHILD);
3826
3827 return bus->trusted;
3828 }
3829
3830 _public_ int sd_bus_is_monitor(sd_bus *bus) {
3831 assert_return(bus, -EINVAL);
3832 assert_return(!bus_pid_changed(bus), -ECHILD);
3833
3834 return !!(bus->hello_flags & KDBUS_HELLO_MONITOR);
3835 }
3836
3837 static void flush_close(sd_bus *bus) {
3838 if (!bus)
3839 return;
3840
3841 /* Flushes and closes the specified bus. We take a ref before,
3842 * to ensure the flushing does not cause the bus to be
3843 * unreferenced. */
3844
3845 sd_bus_flush_close_unref(sd_bus_ref(bus));
3846 }
3847
3848 _public_ void sd_bus_default_flush_close(void) {
3849 flush_close(default_starter_bus);
3850 flush_close(default_user_bus);
3851 flush_close(default_system_bus);
3852 }
3853
3854 _public_ int sd_bus_set_exit_on_disconnect(sd_bus *bus, int b) {
3855 assert_return(bus, -EINVAL);
3856
3857 /* Turns on exit-on-disconnect, and triggers it immediately if the bus connection was already
3858 * disconnected. Note that this is triggered exclusively on disconnections triggered by the server side, never
3859 * from the client side. */
3860 bus->exit_on_disconnect = b;
3861
3862 /* If the exit condition was triggered already, exit immediately. */
3863 return bus_exit_now(bus);
3864 }
3865
3866 _public_ int sd_bus_get_exit_on_disconnect(sd_bus *bus) {
3867 assert_return(bus, -EINVAL);
3868
3869 return bus->exit_on_disconnect;
3870 }