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