chiark / gitweb /
bus: don't require that if a child object node exists its parent node must too
[elogind.git] / src / libsystemd-bus / sd-bus.c
1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
2
3 /***
4   This file is part of systemd.
5
6   Copyright 2013 Lennart Poettering
7
8   systemd is free software; you can redistribute it and/or modify it
9   under the terms of the GNU Lesser General Public License as published by
10   the Free Software Foundation; either version 2.1 of the License, or
11   (at your option) any later version.
12
13   systemd is distributed in the hope that it will be useful, but
14   WITHOUT ANY WARRANTY; without even the implied warranty of
15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16   Lesser General Public License for more details.
17
18   You should have received a copy of the GNU Lesser General Public License
19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
20 ***/
21
22 #include <endian.h>
23 #include <assert.h>
24 #include <stdlib.h>
25 #include <unistd.h>
26 #include <netdb.h>
27 #include <sys/poll.h>
28 #include <byteswap.h>
29 #include <sys/mman.h>
30 #include <pthread.h>
31
32 #include "util.h"
33 #include "macro.h"
34 #include "strv.h"
35 #include "set.h"
36 #include "missing.h"
37
38 #include "sd-bus.h"
39 #include "bus-internal.h"
40 #include "bus-message.h"
41 #include "bus-type.h"
42 #include "bus-socket.h"
43 #include "bus-kernel.h"
44 #include "bus-control.h"
45 #include "bus-introspect.h"
46 #include "bus-signature.h"
47
48 static int bus_poll(sd_bus *bus, bool need_more, uint64_t timeout_usec);
49
50 static void bus_close_fds(sd_bus *b) {
51         assert(b);
52
53         if (b->input_fd >= 0)
54                 close_nointr_nofail(b->input_fd);
55
56         if (b->output_fd >= 0 && b->output_fd != b->input_fd)
57                 close_nointr_nofail(b->output_fd);
58
59         b->input_fd = b->output_fd = -1;
60 }
61
62 static void bus_node_destroy(sd_bus *b, struct node *n) {
63         struct node_callback *c;
64         struct node_vtable *v;
65         struct node_enumerator *e;
66
67         assert(b);
68
69         if (!n)
70                 return;
71
72         while (n->child)
73                 bus_node_destroy(b, n->child);
74
75         while ((c = n->callbacks)) {
76                 LIST_REMOVE(struct node_callback, callbacks, n->callbacks, c);
77                 free(c);
78         }
79
80         while ((v = n->vtables)) {
81                 LIST_REMOVE(struct node_vtable, vtables, n->vtables, v);
82                 free(v->interface);
83                 free(v);
84         }
85
86         while ((e = n->enumerators)) {
87                 LIST_REMOVE(struct node_enumerator, enumerators, n->enumerators, e);
88                 free(e);
89         }
90
91         if (n->parent)
92                 LIST_REMOVE(struct node, siblings, n->parent->child, n);
93
94         assert_se(hashmap_remove(b->nodes, n->path) == n);
95         free(n->path);
96         free(n);
97 }
98
99 static void bus_free(sd_bus *b) {
100         struct filter_callback *f;
101         struct node *n;
102         unsigned i;
103
104         assert(b);
105
106         bus_close_fds(b);
107
108         if (b->kdbus_buffer)
109                 munmap(b->kdbus_buffer, KDBUS_POOL_SIZE);
110
111         free(b->rbuffer);
112         free(b->unique_name);
113         free(b->auth_buffer);
114         free(b->address);
115         free(b->kernel);
116
117         free(b->exec_path);
118         strv_free(b->exec_argv);
119
120         close_many(b->fds, b->n_fds);
121         free(b->fds);
122
123         for (i = 0; i < b->rqueue_size; i++)
124                 sd_bus_message_unref(b->rqueue[i]);
125         free(b->rqueue);
126
127         for (i = 0; i < b->wqueue_size; i++)
128                 sd_bus_message_unref(b->wqueue[i]);
129         free(b->wqueue);
130
131         hashmap_free_free(b->reply_callbacks);
132         prioq_free(b->reply_callbacks_prioq);
133
134         while ((f = b->filter_callbacks)) {
135                 LIST_REMOVE(struct filter_callback, callbacks, b->filter_callbacks, f);
136                 free(f);
137         }
138
139         bus_match_free(&b->match_callbacks);
140
141         hashmap_free_free(b->vtable_methods);
142         hashmap_free_free(b->vtable_properties);
143
144         while ((n = hashmap_first(b->nodes)))
145                 bus_node_destroy(b, n);
146
147         hashmap_free(b->nodes);
148
149         bus_kernel_flush_memfd(b);
150
151         assert_se(pthread_mutex_destroy(&b->memfd_cache_mutex) == 0);
152
153         free(b);
154 }
155
156 int sd_bus_new(sd_bus **ret) {
157         sd_bus *r;
158
159         if (!ret)
160                 return -EINVAL;
161
162         r = new0(sd_bus, 1);
163         if (!r)
164                 return -ENOMEM;
165
166         r->n_ref = REFCNT_INIT;
167         r->input_fd = r->output_fd = -1;
168         r->message_version = 1;
169         r->hello_flags |= KDBUS_HELLO_ACCEPT_FD;
170         r->original_pid = getpid();
171
172         assert_se(pthread_mutex_init(&r->memfd_cache_mutex, NULL) == 0);
173
174         /* We guarantee that wqueue always has space for at least one
175          * entry */
176         r->wqueue = new(sd_bus_message*, 1);
177         if (!r->wqueue) {
178                 free(r);
179                 return -ENOMEM;
180         }
181
182         *ret = r;
183         return 0;
184 }
185
186 int sd_bus_set_address(sd_bus *bus, const char *address) {
187         char *a;
188
189         if (!bus)
190                 return -EINVAL;
191         if (bus->state != BUS_UNSET)
192                 return -EPERM;
193         if (!address)
194                 return -EINVAL;
195         if (bus_pid_changed(bus))
196                 return -ECHILD;
197
198         a = strdup(address);
199         if (!a)
200                 return -ENOMEM;
201
202         free(bus->address);
203         bus->address = a;
204
205         return 0;
206 }
207
208 int sd_bus_set_fd(sd_bus *bus, int input_fd, int output_fd) {
209         if (!bus)
210                 return -EINVAL;
211         if (bus->state != BUS_UNSET)
212                 return -EPERM;
213         if (input_fd < 0)
214                 return -EINVAL;
215         if (output_fd < 0)
216                 return -EINVAL;
217         if (bus_pid_changed(bus))
218                 return -ECHILD;
219
220         bus->input_fd = input_fd;
221         bus->output_fd = output_fd;
222         return 0;
223 }
224
225 int sd_bus_set_exec(sd_bus *bus, const char *path, char *const argv[]) {
226         char *p, **a;
227
228         if (!bus)
229                 return -EINVAL;
230         if (bus->state != BUS_UNSET)
231                 return -EPERM;
232         if (!path)
233                 return -EINVAL;
234         if (strv_isempty(argv))
235                 return -EINVAL;
236         if (bus_pid_changed(bus))
237                 return -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 int sd_bus_set_bus_client(sd_bus *bus, int b) {
259         if (!bus)
260                 return -EINVAL;
261         if (bus->state != BUS_UNSET)
262                 return -EPERM;
263         if (bus_pid_changed(bus))
264                 return -ECHILD;
265
266         bus->bus_client = !!b;
267         return 0;
268 }
269
270 int sd_bus_negotiate_fds(sd_bus *bus, int b) {
271         if (!bus)
272                 return -EINVAL;
273         if (bus->state != BUS_UNSET)
274                 return -EPERM;
275         if (bus_pid_changed(bus))
276                 return -ECHILD;
277
278         SET_FLAG(bus->hello_flags, KDBUS_HELLO_ACCEPT_FD, b);
279         return 0;
280 }
281
282 int sd_bus_negotiate_attach_comm(sd_bus *bus, int b) {
283         if (!bus)
284                 return -EINVAL;
285         if (bus->state != BUS_UNSET)
286                 return -EPERM;
287         if (bus_pid_changed(bus))
288                 return -ECHILD;
289
290         SET_FLAG(bus->hello_flags, KDBUS_HELLO_ATTACH_COMM, b);
291         return 0;
292 }
293
294 int sd_bus_negotiate_attach_exe(sd_bus *bus, int b) {
295         if (!bus)
296                 return -EINVAL;
297         if (bus->state != BUS_UNSET)
298                 return -EPERM;
299         if (bus_pid_changed(bus))
300                 return -ECHILD;
301
302         SET_FLAG(bus->hello_flags, KDBUS_HELLO_ATTACH_EXE, b);
303         return 0;
304 }
305
306 int sd_bus_negotiate_attach_cmdline(sd_bus *bus, int b) {
307         if (!bus)
308                 return -EINVAL;
309         if (bus->state != BUS_UNSET)
310                 return -EPERM;
311         if (bus_pid_changed(bus))
312                 return -ECHILD;
313
314         SET_FLAG(bus->hello_flags, KDBUS_HELLO_ATTACH_CMDLINE, b);
315         return 0;
316 }
317
318 int sd_bus_negotiate_attach_cgroup(sd_bus *bus, int b) {
319         if (!bus)
320                 return -EINVAL;
321         if (bus->state != BUS_UNSET)
322                 return -EPERM;
323         if (bus_pid_changed(bus))
324                 return -ECHILD;
325
326         SET_FLAG(bus->hello_flags, KDBUS_HELLO_ATTACH_CGROUP, b);
327         return 0;
328 }
329
330 int sd_bus_negotiate_attach_caps(sd_bus *bus, int b) {
331         if (!bus)
332                 return -EINVAL;
333         if (bus->state != BUS_UNSET)
334                 return -EPERM;
335         if (bus_pid_changed(bus))
336                 return -ECHILD;
337
338         SET_FLAG(bus->hello_flags, KDBUS_HELLO_ATTACH_CAPS, b);
339         return 0;
340 }
341
342 int sd_bus_negotiate_attach_selinux_context(sd_bus *bus, int b) {
343         if (!bus)
344                 return -EINVAL;
345         if (bus->state != BUS_UNSET)
346                 return -EPERM;
347         if (bus_pid_changed(bus))
348                 return -ECHILD;
349
350         SET_FLAG(bus->hello_flags, KDBUS_HELLO_ATTACH_SECLABEL, b);
351         return 0;
352 }
353
354 int sd_bus_negotiate_attach_audit(sd_bus *bus, int b) {
355         if (!bus)
356                 return -EINVAL;
357         if (bus->state != BUS_UNSET)
358                 return -EPERM;
359         if (bus_pid_changed(bus))
360                 return -ECHILD;
361
362         SET_FLAG(bus->hello_flags, KDBUS_HELLO_ATTACH_AUDIT, b);
363         return 0;
364 }
365
366 int sd_bus_set_server(sd_bus *bus, int b, sd_id128_t server_id) {
367         if (!bus)
368                 return -EINVAL;
369         if (!b && !sd_id128_equal(server_id, SD_ID128_NULL))
370                 return -EINVAL;
371         if (bus->state != BUS_UNSET)
372                 return -EPERM;
373         if (bus_pid_changed(bus))
374                 return -ECHILD;
375
376         bus->is_server = !!b;
377         bus->server_id = server_id;
378         return 0;
379 }
380
381 int sd_bus_set_anonymous(sd_bus *bus, int b) {
382         if (!bus)
383                 return -EINVAL;
384         if (bus->state != BUS_UNSET)
385                 return -EPERM;
386         if (bus_pid_changed(bus))
387                 return -ECHILD;
388
389         bus->anonymous_auth = !!b;
390         return 0;
391 }
392
393 static int hello_callback(sd_bus *bus, sd_bus_message *reply, void *userdata) {
394         const char *s;
395         int r;
396
397         assert(bus);
398         assert(bus->state == BUS_HELLO);
399         assert(reply);
400
401         r = bus_message_to_errno(reply);
402         if (r < 0)
403                 return r;
404
405         r = sd_bus_message_read(reply, "s", &s);
406         if (r < 0)
407                 return r;
408
409         if (!service_name_is_valid(s) || s[0] != ':')
410                 return -EBADMSG;
411
412         bus->unique_name = strdup(s);
413         if (!bus->unique_name)
414                 return -ENOMEM;
415
416         bus->state = BUS_RUNNING;
417
418         return 1;
419 }
420
421 static int bus_send_hello(sd_bus *bus) {
422         _cleanup_bus_message_unref_ sd_bus_message *m = NULL;
423         int r;
424
425         assert(bus);
426
427         if (!bus->bus_client || bus->is_kernel)
428                 return 0;
429
430         r = sd_bus_message_new_method_call(
431                         bus,
432                         "org.freedesktop.DBus",
433                         "/",
434                         "org.freedesktop.DBus",
435                         "Hello",
436                         &m);
437         if (r < 0)
438                 return r;
439
440         return sd_bus_send_with_reply(bus, m, hello_callback, NULL, 0, &bus->hello_serial);
441 }
442
443 int bus_start_running(sd_bus *bus) {
444         assert(bus);
445
446         if (bus->bus_client && !bus->is_kernel) {
447                 bus->state = BUS_HELLO;
448                 return 1;
449         }
450
451         bus->state = BUS_RUNNING;
452         return 1;
453 }
454
455 static int parse_address_key(const char **p, const char *key, char **value) {
456         size_t l, n = 0;
457         const char *a;
458         char *r = NULL;
459
460         assert(p);
461         assert(*p);
462         assert(value);
463
464         if (key) {
465                 l = strlen(key);
466                 if (strncmp(*p, key, l) != 0)
467                         return 0;
468
469                 if ((*p)[l] != '=')
470                         return 0;
471
472                 if (*value)
473                         return -EINVAL;
474
475                 a = *p + l + 1;
476         } else
477                 a = *p;
478
479         while (*a != ';' && *a != ',' && *a != 0) {
480                 char c, *t;
481
482                 if (*a == '%') {
483                         int x, y;
484
485                         x = unhexchar(a[1]);
486                         if (x < 0) {
487                                 free(r);
488                                 return x;
489                         }
490
491                         y = unhexchar(a[2]);
492                         if (y < 0) {
493                                 free(r);
494                                 return y;
495                         }
496
497                         c = (char) ((x << 4) | y);
498                         a += 3;
499                 } else {
500                         c = *a;
501                         a++;
502                 }
503
504                 t = realloc(r, n + 2);
505                 if (!t) {
506                         free(r);
507                         return -ENOMEM;
508                 }
509
510                 r = t;
511                 r[n++] = c;
512         }
513
514         if (!r) {
515                 r = strdup("");
516                 if (!r)
517                         return -ENOMEM;
518         } else
519                 r[n] = 0;
520
521         if (*a == ',')
522                 a++;
523
524         *p = a;
525
526         free(*value);
527         *value = r;
528
529         return 1;
530 }
531
532 static void skip_address_key(const char **p) {
533         assert(p);
534         assert(*p);
535
536         *p += strcspn(*p, ",");
537
538         if (**p == ',')
539                 (*p) ++;
540 }
541
542 static int parse_unix_address(sd_bus *b, const char **p, char **guid) {
543         _cleanup_free_ char *path = NULL, *abstract = NULL;
544         size_t l;
545         int r;
546
547         assert(b);
548         assert(p);
549         assert(*p);
550         assert(guid);
551
552         while (**p != 0 && **p != ';') {
553                 r = parse_address_key(p, "guid", guid);
554                 if (r < 0)
555                         return r;
556                 else if (r > 0)
557                         continue;
558
559                 r = parse_address_key(p, "path", &path);
560                 if (r < 0)
561                         return r;
562                 else if (r > 0)
563                         continue;
564
565                 r = parse_address_key(p, "abstract", &abstract);
566                 if (r < 0)
567                         return r;
568                 else if (r > 0)
569                         continue;
570
571                 skip_address_key(p);
572         }
573
574         if (!path && !abstract)
575                 return -EINVAL;
576
577         if (path && abstract)
578                 return -EINVAL;
579
580         if (path) {
581                 l = strlen(path);
582                 if (l > sizeof(b->sockaddr.un.sun_path))
583                         return -E2BIG;
584
585                 b->sockaddr.un.sun_family = AF_UNIX;
586                 strncpy(b->sockaddr.un.sun_path, path, sizeof(b->sockaddr.un.sun_path));
587                 b->sockaddr_size = offsetof(struct sockaddr_un, sun_path) + l;
588         } else if (abstract) {
589                 l = strlen(abstract);
590                 if (l > sizeof(b->sockaddr.un.sun_path) - 1)
591                         return -E2BIG;
592
593                 b->sockaddr.un.sun_family = AF_UNIX;
594                 b->sockaddr.un.sun_path[0] = 0;
595                 strncpy(b->sockaddr.un.sun_path+1, abstract, sizeof(b->sockaddr.un.sun_path)-1);
596                 b->sockaddr_size = offsetof(struct sockaddr_un, sun_path) + 1 + l;
597         }
598
599         return 0;
600 }
601
602 static int parse_tcp_address(sd_bus *b, const char **p, char **guid) {
603         _cleanup_free_ char *host = NULL, *port = NULL, *family = NULL;
604         int r;
605         struct addrinfo *result, hints = {
606                 .ai_socktype = SOCK_STREAM,
607                 .ai_flags = AI_ADDRCONFIG,
608         };
609
610         assert(b);
611         assert(p);
612         assert(*p);
613         assert(guid);
614
615         while (**p != 0 && **p != ';') {
616                 r = parse_address_key(p, "guid", guid);
617                 if (r < 0)
618                         return r;
619                 else if (r > 0)
620                         continue;
621
622                 r = parse_address_key(p, "host", &host);
623                 if (r < 0)
624                         return r;
625                 else if (r > 0)
626                         continue;
627
628                 r = parse_address_key(p, "port", &port);
629                 if (r < 0)
630                         return r;
631                 else if (r > 0)
632                         continue;
633
634                 r = parse_address_key(p, "family", &family);
635                 if (r < 0)
636                         return r;
637                 else if (r > 0)
638                         continue;
639
640                 skip_address_key(p);
641         }
642
643         if (!host || !port)
644                 return -EINVAL;
645
646         if (family) {
647                 if (streq(family, "ipv4"))
648                         hints.ai_family = AF_INET;
649                 else if (streq(family, "ipv6"))
650                         hints.ai_family = AF_INET6;
651                 else
652                         return -EINVAL;
653         }
654
655         r = getaddrinfo(host, port, &hints, &result);
656         if (r == EAI_SYSTEM)
657                 return -errno;
658         else if (r != 0)
659                 return -EADDRNOTAVAIL;
660
661         memcpy(&b->sockaddr, result->ai_addr, result->ai_addrlen);
662         b->sockaddr_size = result->ai_addrlen;
663
664         freeaddrinfo(result);
665
666         return 0;
667 }
668
669 static int parse_exec_address(sd_bus *b, const char **p, char **guid) {
670         char *path = NULL;
671         unsigned n_argv = 0, j;
672         char **argv = NULL;
673         int r;
674
675         assert(b);
676         assert(p);
677         assert(*p);
678         assert(guid);
679
680         while (**p != 0 && **p != ';') {
681                 r = parse_address_key(p, "guid", guid);
682                 if (r < 0)
683                         goto fail;
684                 else if (r > 0)
685                         continue;
686
687                 r = parse_address_key(p, "path", &path);
688                 if (r < 0)
689                         goto fail;
690                 else if (r > 0)
691                         continue;
692
693                 if (startswith(*p, "argv")) {
694                         unsigned ul;
695
696                         errno = 0;
697                         ul = strtoul(*p + 4, (char**) p, 10);
698                         if (errno > 0 || **p != '=' || ul > 256) {
699                                 r = -EINVAL;
700                                 goto fail;
701                         }
702
703                         (*p) ++;
704
705                         if (ul >= n_argv) {
706                                 char **x;
707
708                                 x = realloc(argv, sizeof(char*) * (ul + 2));
709                                 if (!x) {
710                                         r = -ENOMEM;
711                                         goto fail;
712                                 }
713
714                                 memset(x + n_argv, 0, sizeof(char*) * (ul - n_argv + 2));
715
716                                 argv = x;
717                                 n_argv = ul + 1;
718                         }
719
720                         r = parse_address_key(p, NULL, argv + ul);
721                         if (r < 0)
722                                 goto fail;
723
724                         continue;
725                 }
726
727                 skip_address_key(p);
728         }
729
730         if (!path) {
731                 r = -EINVAL;
732                 goto fail;
733         }
734
735         /* Make sure there are no holes in the array, with the
736          * exception of argv[0] */
737         for (j = 1; j < n_argv; j++)
738                 if (!argv[j]) {
739                         r = -EINVAL;
740                         goto fail;
741                 }
742
743         if (argv && argv[0] == NULL) {
744                 argv[0] = strdup(path);
745                 if (!argv[0]) {
746                         r = -ENOMEM;
747                         goto fail;
748                 }
749         }
750
751         b->exec_path = path;
752         b->exec_argv = argv;
753         return 0;
754
755 fail:
756         for (j = 0; j < n_argv; j++)
757                 free(argv[j]);
758
759         free(argv);
760         free(path);
761         return r;
762 }
763
764 static int parse_kernel_address(sd_bus *b, const char **p, char **guid) {
765         _cleanup_free_ char *path = NULL;
766         int r;
767
768         assert(b);
769         assert(p);
770         assert(*p);
771         assert(guid);
772
773         while (**p != 0 && **p != ';') {
774                 r = parse_address_key(p, "guid", guid);
775                 if (r < 0)
776                         return r;
777                 else if (r > 0)
778                         continue;
779
780                 r = parse_address_key(p, "path", &path);
781                 if (r < 0)
782                         return r;
783                 else if (r > 0)
784                         continue;
785
786                 skip_address_key(p);
787         }
788
789         if (!path)
790                 return -EINVAL;
791
792         free(b->kernel);
793         b->kernel = path;
794         path = NULL;
795
796         return 0;
797 }
798
799 static void bus_reset_parsed_address(sd_bus *b) {
800         assert(b);
801
802         zero(b->sockaddr);
803         b->sockaddr_size = 0;
804         strv_free(b->exec_argv);
805         free(b->exec_path);
806         b->exec_path = NULL;
807         b->exec_argv = NULL;
808         b->server_id = SD_ID128_NULL;
809         free(b->kernel);
810         b->kernel = NULL;
811 }
812
813 static int bus_parse_next_address(sd_bus *b) {
814         _cleanup_free_ char *guid = NULL;
815         const char *a;
816         int r;
817
818         assert(b);
819
820         if (!b->address)
821                 return 0;
822         if (b->address[b->address_index] == 0)
823                 return 0;
824
825         bus_reset_parsed_address(b);
826
827         a = b->address + b->address_index;
828
829         while (*a != 0) {
830
831                 if (*a == ';') {
832                         a++;
833                         continue;
834                 }
835
836                 if (startswith(a, "unix:")) {
837                         a += 5;
838
839                         r = parse_unix_address(b, &a, &guid);
840                         if (r < 0)
841                                 return r;
842                         break;
843
844                 } else if (startswith(a, "tcp:")) {
845
846                         a += 4;
847                         r = parse_tcp_address(b, &a, &guid);
848                         if (r < 0)
849                                 return r;
850
851                         break;
852
853                 } else if (startswith(a, "unixexec:")) {
854
855                         a += 9;
856                         r = parse_exec_address(b, &a, &guid);
857                         if (r < 0)
858                                 return r;
859
860                         break;
861
862                 } else if (startswith(a, "kernel:")) {
863
864                         a += 7;
865                         r = parse_kernel_address(b, &a, &guid);
866                         if (r < 0)
867                                 return r;
868
869                         break;
870                 }
871
872                 a = strchr(a, ';');
873                 if (!a)
874                         return 0;
875         }
876
877         if (guid) {
878                 r = sd_id128_from_string(guid, &b->server_id);
879                 if (r < 0)
880                         return r;
881         }
882
883         b->address_index = a - b->address;
884         return 1;
885 }
886
887 static int bus_start_address(sd_bus *b) {
888         int r;
889
890         assert(b);
891
892         for (;;) {
893                 sd_bus_close(b);
894
895                 if (b->sockaddr.sa.sa_family != AF_UNSPEC) {
896
897                         r = bus_socket_connect(b);
898                         if (r >= 0)
899                                 return r;
900
901                         b->last_connect_error = -r;
902
903                 } else if (b->exec_path) {
904
905                         r = bus_socket_exec(b);
906                         if (r >= 0)
907                                 return r;
908
909                         b->last_connect_error = -r;
910                 } else if (b->kernel) {
911
912                         r = bus_kernel_connect(b);
913                         if (r >= 0)
914                                 return r;
915
916                         b->last_connect_error = -r;
917                 }
918
919                 r = bus_parse_next_address(b);
920                 if (r < 0)
921                         return r;
922                 if (r == 0)
923                         return b->last_connect_error ? -b->last_connect_error : -ECONNREFUSED;
924         }
925 }
926
927 int bus_next_address(sd_bus *b) {
928         assert(b);
929
930         bus_reset_parsed_address(b);
931         return bus_start_address(b);
932 }
933
934 static int bus_start_fd(sd_bus *b) {
935         struct stat st;
936         int r;
937
938         assert(b);
939         assert(b->input_fd >= 0);
940         assert(b->output_fd >= 0);
941
942         r = fd_nonblock(b->input_fd, true);
943         if (r < 0)
944                 return r;
945
946         r = fd_cloexec(b->input_fd, true);
947         if (r < 0)
948                 return r;
949
950         if (b->input_fd != b->output_fd) {
951                 r = fd_nonblock(b->output_fd, true);
952                 if (r < 0)
953                         return r;
954
955                 r = fd_cloexec(b->output_fd, true);
956                 if (r < 0)
957                         return r;
958         }
959
960         if (fstat(b->input_fd, &st) < 0)
961                 return -errno;
962
963         if (S_ISCHR(b->input_fd))
964                 return bus_kernel_take_fd(b);
965         else
966                 return bus_socket_take_fd(b);
967 }
968
969 int sd_bus_start(sd_bus *bus) {
970         int r;
971
972         if (!bus)
973                 return -EINVAL;
974         if (bus->state != BUS_UNSET)
975                 return -EPERM;
976         if (bus_pid_changed(bus))
977                 return -ECHILD;
978
979         bus->state = BUS_OPENING;
980
981         if (bus->is_server && bus->bus_client)
982                 return -EINVAL;
983
984         if (bus->input_fd >= 0)
985                 r = bus_start_fd(bus);
986         else if (bus->address || bus->sockaddr.sa.sa_family != AF_UNSPEC || bus->exec_path || bus->kernel)
987                 r = bus_start_address(bus);
988         else
989                 return -EINVAL;
990
991         if (r < 0)
992                 return r;
993
994         return bus_send_hello(bus);
995 }
996
997 int sd_bus_open_system(sd_bus **ret) {
998         const char *e;
999         sd_bus *b;
1000         int r;
1001
1002         if (!ret)
1003                 return -EINVAL;
1004
1005         r = sd_bus_new(&b);
1006         if (r < 0)
1007                 return r;
1008
1009         e = secure_getenv("DBUS_SYSTEM_BUS_ADDRESS");
1010         if (e) {
1011                 r = sd_bus_set_address(b, e);
1012                 if (r < 0)
1013                         goto fail;
1014         } else {
1015                 b->sockaddr.un.sun_family = AF_UNIX;
1016                 strncpy(b->sockaddr.un.sun_path, "/run/dbus/system_bus_socket", sizeof(b->sockaddr.un.sun_path));
1017                 b->sockaddr_size = offsetof(struct sockaddr_un, sun_path) + sizeof("/run/dbus/system_bus_socket") - 1;
1018         }
1019
1020         b->bus_client = true;
1021
1022         r = sd_bus_start(b);
1023         if (r < 0)
1024                 goto fail;
1025
1026         *ret = b;
1027         return 0;
1028
1029 fail:
1030         bus_free(b);
1031         return r;
1032 }
1033
1034 int sd_bus_open_user(sd_bus **ret) {
1035         const char *e;
1036         sd_bus *b;
1037         size_t l;
1038         int r;
1039
1040         if (!ret)
1041                 return -EINVAL;
1042
1043         r = sd_bus_new(&b);
1044         if (r < 0)
1045                 return r;
1046
1047         e = secure_getenv("DBUS_SESSION_BUS_ADDRESS");
1048         if (e) {
1049                 r = sd_bus_set_address(b, e);
1050                 if (r < 0)
1051                         goto fail;
1052         } else {
1053                 e = secure_getenv("XDG_RUNTIME_DIR");
1054                 if (!e) {
1055                         r = -ENOENT;
1056                         goto fail;
1057                 }
1058
1059                 l = strlen(e);
1060                 if (l + 4 > sizeof(b->sockaddr.un.sun_path)) {
1061                         r = -E2BIG;
1062                         goto fail;
1063                 }
1064
1065                 b->sockaddr.un.sun_family = AF_UNIX;
1066                 memcpy(mempcpy(b->sockaddr.un.sun_path, e, l), "/bus", 4);
1067                 b->sockaddr_size = offsetof(struct sockaddr_un, sun_path) + l + 4;
1068         }
1069
1070         b->bus_client = true;
1071
1072         r = sd_bus_start(b);
1073         if (r < 0)
1074                 goto fail;
1075
1076         *ret = b;
1077         return 0;
1078
1079 fail:
1080         bus_free(b);
1081         return r;
1082 }
1083
1084 void sd_bus_close(sd_bus *bus) {
1085         if (!bus)
1086                 return;
1087         if (bus->state == BUS_CLOSED)
1088                 return;
1089         if (bus_pid_changed(bus))
1090                 return;
1091
1092         bus->state = BUS_CLOSED;
1093
1094         if (!bus->is_kernel)
1095                 bus_close_fds(bus);
1096
1097         /* We'll leave the fd open in case this is a kernel bus, since
1098          * there might still be memblocks around that reference this
1099          * bus, and they might need to invoke the
1100          * KDBUS_CMD_MSG_RELEASE ioctl on the fd when they are
1101          * freed. */
1102 }
1103
1104 sd_bus *sd_bus_ref(sd_bus *bus) {
1105         if (!bus)
1106                 return NULL;
1107
1108         assert_se(REFCNT_INC(bus->n_ref) >= 2);
1109
1110         return bus;
1111 }
1112
1113 sd_bus *sd_bus_unref(sd_bus *bus) {
1114         if (!bus)
1115                 return NULL;
1116
1117         if (REFCNT_DEC(bus->n_ref) <= 0)
1118                 bus_free(bus);
1119
1120         return NULL;
1121 }
1122
1123 int sd_bus_is_open(sd_bus *bus) {
1124         if (!bus)
1125                 return -EINVAL;
1126         if (bus_pid_changed(bus))
1127                 return -ECHILD;
1128
1129         return BUS_IS_OPEN(bus->state);
1130 }
1131
1132 int sd_bus_can_send(sd_bus *bus, char type) {
1133         int r;
1134
1135         if (!bus)
1136                 return -EINVAL;
1137         if (bus->state == BUS_UNSET)
1138                 return -ENOTCONN;
1139         if (bus_pid_changed(bus))
1140                 return -ECHILD;
1141
1142         if (type == SD_BUS_TYPE_UNIX_FD) {
1143                 if (!(bus->hello_flags & KDBUS_HELLO_ACCEPT_FD))
1144                         return 0;
1145
1146                 r = bus_ensure_running(bus);
1147                 if (r < 0)
1148                         return r;
1149
1150                 return bus->can_fds;
1151         }
1152
1153         return bus_type_is_valid(type);
1154 }
1155
1156 int sd_bus_get_server_id(sd_bus *bus, sd_id128_t *server_id) {
1157         int r;
1158
1159         if (!bus)
1160                 return -EINVAL;
1161         if (!server_id)
1162                 return -EINVAL;
1163         if (bus_pid_changed(bus))
1164                 return -ECHILD;
1165
1166         r = bus_ensure_running(bus);
1167         if (r < 0)
1168                 return r;
1169
1170         *server_id = bus->server_id;
1171         return 0;
1172 }
1173
1174 static int bus_seal_message(sd_bus *b, sd_bus_message *m) {
1175         assert(m);
1176
1177         if (m->header->version > b->message_version)
1178                 return -EPERM;
1179
1180         if (m->sealed)
1181                 return 0;
1182
1183         return bus_message_seal(m, ++b->serial);
1184 }
1185
1186 static int dispatch_wqueue(sd_bus *bus) {
1187         int r, ret = 0;
1188
1189         assert(bus);
1190         assert(bus->state == BUS_RUNNING || bus->state == BUS_HELLO);
1191
1192         while (bus->wqueue_size > 0) {
1193
1194                 if (bus->is_kernel)
1195                         r = bus_kernel_write_message(bus, bus->wqueue[0]);
1196                 else
1197                         r = bus_socket_write_message(bus, bus->wqueue[0], &bus->windex);
1198
1199                 if (r < 0) {
1200                         sd_bus_close(bus);
1201                         return r;
1202                 } else if (r == 0)
1203                         /* Didn't do anything this time */
1204                         return ret;
1205                 else if (bus->is_kernel || bus->windex >= BUS_MESSAGE_SIZE(bus->wqueue[0])) {
1206                         /* Fully written. Let's drop the entry from
1207                          * the queue.
1208                          *
1209                          * This isn't particularly optimized, but
1210                          * well, this is supposed to be our worst-case
1211                          * buffer only, and the socket buffer is
1212                          * supposed to be our primary buffer, and if
1213                          * it got full, then all bets are off
1214                          * anyway. */
1215
1216                         sd_bus_message_unref(bus->wqueue[0]);
1217                         bus->wqueue_size --;
1218                         memmove(bus->wqueue, bus->wqueue + 1, sizeof(sd_bus_message*) * bus->wqueue_size);
1219                         bus->windex = 0;
1220
1221                         ret = 1;
1222                 }
1223         }
1224
1225         return ret;
1226 }
1227
1228 static int dispatch_rqueue(sd_bus *bus, sd_bus_message **m) {
1229         sd_bus_message *z = NULL;
1230         int r, ret = 0;
1231
1232         assert(bus);
1233         assert(m);
1234         assert(bus->state == BUS_RUNNING || bus->state == BUS_HELLO);
1235
1236         if (bus->rqueue_size > 0) {
1237                 /* Dispatch a queued message */
1238
1239                 *m = bus->rqueue[0];
1240                 bus->rqueue_size --;
1241                 memmove(bus->rqueue, bus->rqueue + 1, sizeof(sd_bus_message*) * bus->rqueue_size);
1242                 return 1;
1243         }
1244
1245         /* Try to read a new message */
1246         do {
1247                 if (bus->is_kernel)
1248                         r = bus_kernel_read_message(bus, &z);
1249                 else
1250                         r = bus_socket_read_message(bus, &z);
1251
1252                 if (r < 0) {
1253                         sd_bus_close(bus);
1254                         return r;
1255                 }
1256                 if (r == 0)
1257                         return ret;
1258
1259                 ret = 1;
1260         } while (!z);
1261
1262         *m = z;
1263         return ret;
1264 }
1265
1266 int sd_bus_send(sd_bus *bus, sd_bus_message *m, uint64_t *serial) {
1267         int r;
1268
1269         if (!bus)
1270                 return -EINVAL;
1271         if (!BUS_IS_OPEN(bus->state))
1272                 return -ENOTCONN;
1273         if (!m)
1274                 return -EINVAL;
1275         if (bus_pid_changed(bus))
1276                 return -ECHILD;
1277
1278         if (m->n_fds > 0) {
1279                 r = sd_bus_can_send(bus, SD_BUS_TYPE_UNIX_FD);
1280                 if (r < 0)
1281                         return r;
1282                 if (r == 0)
1283                         return -ENOTSUP;
1284         }
1285
1286         /* If the serial number isn't kept, then we know that no reply
1287          * is expected */
1288         if (!serial && !m->sealed)
1289                 m->header->flags |= SD_BUS_MESSAGE_NO_REPLY_EXPECTED;
1290
1291         r = bus_seal_message(bus, m);
1292         if (r < 0)
1293                 return r;
1294
1295         /* If this is a reply and no reply was requested, then let's
1296          * suppress this, if we can */
1297         if (m->dont_send && !serial)
1298                 return 0;
1299
1300         if ((bus->state == BUS_RUNNING || bus->state == BUS_HELLO) && bus->wqueue_size <= 0) {
1301                 size_t idx = 0;
1302
1303                 if (bus->is_kernel)
1304                         r = bus_kernel_write_message(bus, m);
1305                 else
1306                         r = bus_socket_write_message(bus, m, &idx);
1307
1308                 if (r < 0) {
1309                         sd_bus_close(bus);
1310                         return r;
1311                 } else if (!bus->is_kernel && idx < BUS_MESSAGE_SIZE(m))  {
1312                         /* Wasn't fully written. So let's remember how
1313                          * much was written. Note that the first entry
1314                          * of the wqueue array is always allocated so
1315                          * that we always can remember how much was
1316                          * written. */
1317                         bus->wqueue[0] = sd_bus_message_ref(m);
1318                         bus->wqueue_size = 1;
1319                         bus->windex = idx;
1320                 }
1321         } else {
1322                 sd_bus_message **q;
1323
1324                 /* Just append it to the queue. */
1325
1326                 if (bus->wqueue_size >= BUS_WQUEUE_MAX)
1327                         return -ENOBUFS;
1328
1329                 q = realloc(bus->wqueue, sizeof(sd_bus_message*) * (bus->wqueue_size + 1));
1330                 if (!q)
1331                         return -ENOMEM;
1332
1333                 bus->wqueue = q;
1334                 q[bus->wqueue_size ++] = sd_bus_message_ref(m);
1335         }
1336
1337         if (serial)
1338                 *serial = BUS_MESSAGE_SERIAL(m);
1339
1340         return 0;
1341 }
1342
1343 static usec_t calc_elapse(uint64_t usec) {
1344         if (usec == (uint64_t) -1)
1345                 return 0;
1346
1347         if (usec == 0)
1348                 usec = BUS_DEFAULT_TIMEOUT;
1349
1350         return now(CLOCK_MONOTONIC) + usec;
1351 }
1352
1353 static int timeout_compare(const void *a, const void *b) {
1354         const struct reply_callback *x = a, *y = b;
1355
1356         if (x->timeout != 0 && y->timeout == 0)
1357                 return -1;
1358
1359         if (x->timeout == 0 && y->timeout != 0)
1360                 return 1;
1361
1362         if (x->timeout < y->timeout)
1363                 return -1;
1364
1365         if (x->timeout > y->timeout)
1366                 return 1;
1367
1368         return 0;
1369 }
1370
1371 int sd_bus_send_with_reply(
1372                 sd_bus *bus,
1373                 sd_bus_message *m,
1374                 sd_bus_message_handler_t callback,
1375                 void *userdata,
1376                 uint64_t usec,
1377                 uint64_t *serial) {
1378
1379         struct reply_callback *c;
1380         int r;
1381
1382         if (!bus)
1383                 return -EINVAL;
1384         if (!BUS_IS_OPEN(bus->state))
1385                 return -ENOTCONN;
1386         if (!m)
1387                 return -EINVAL;
1388         if (!callback)
1389                 return -EINVAL;
1390         if (m->header->type != SD_BUS_MESSAGE_TYPE_METHOD_CALL)
1391                 return -EINVAL;
1392         if (m->header->flags & SD_BUS_MESSAGE_NO_REPLY_EXPECTED)
1393                 return -EINVAL;
1394         if (bus_pid_changed(bus))
1395                 return -ECHILD;
1396
1397         r = hashmap_ensure_allocated(&bus->reply_callbacks, uint64_hash_func, uint64_compare_func);
1398         if (r < 0)
1399                 return r;
1400
1401         if (usec != (uint64_t) -1) {
1402                 r = prioq_ensure_allocated(&bus->reply_callbacks_prioq, timeout_compare);
1403                 if (r < 0)
1404                         return r;
1405         }
1406
1407         r = bus_seal_message(bus, m);
1408         if (r < 0)
1409                 return r;
1410
1411         c = new0(struct reply_callback, 1);
1412         if (!c)
1413                 return -ENOMEM;
1414
1415         c->callback = callback;
1416         c->userdata = userdata;
1417         c->serial = BUS_MESSAGE_SERIAL(m);
1418         c->timeout = calc_elapse(usec);
1419
1420         r = hashmap_put(bus->reply_callbacks, &c->serial, c);
1421         if (r < 0) {
1422                 free(c);
1423                 return r;
1424         }
1425
1426         if (c->timeout != 0) {
1427                 r = prioq_put(bus->reply_callbacks_prioq, c, &c->prioq_idx);
1428                 if (r < 0) {
1429                         c->timeout = 0;
1430                         sd_bus_send_with_reply_cancel(bus, c->serial);
1431                         return r;
1432                 }
1433         }
1434
1435         r = sd_bus_send(bus, m, serial);
1436         if (r < 0) {
1437                 sd_bus_send_with_reply_cancel(bus, c->serial);
1438                 return r;
1439         }
1440
1441         return r;
1442 }
1443
1444 int sd_bus_send_with_reply_cancel(sd_bus *bus, uint64_t serial) {
1445         struct reply_callback *c;
1446
1447         if (!bus)
1448                 return -EINVAL;
1449         if (serial == 0)
1450                 return -EINVAL;
1451         if (bus_pid_changed(bus))
1452                 return -ECHILD;
1453
1454         c = hashmap_remove(bus->reply_callbacks, &serial);
1455         if (!c)
1456                 return 0;
1457
1458         if (c->timeout != 0)
1459                 prioq_remove(bus->reply_callbacks_prioq, c, &c->prioq_idx);
1460
1461         free(c);
1462         return 1;
1463 }
1464
1465 int bus_ensure_running(sd_bus *bus) {
1466         int r;
1467
1468         assert(bus);
1469
1470         if (bus->state == BUS_UNSET || bus->state == BUS_CLOSED)
1471                 return -ENOTCONN;
1472         if (bus->state == BUS_RUNNING)
1473                 return 1;
1474
1475         for (;;) {
1476                 r = sd_bus_process(bus, NULL);
1477                 if (r < 0)
1478                         return r;
1479                 if (bus->state == BUS_RUNNING)
1480                         return 1;
1481                 if (r > 0)
1482                         continue;
1483
1484                 r = sd_bus_wait(bus, (uint64_t) -1);
1485                 if (r < 0)
1486                         return r;
1487         }
1488 }
1489
1490 int sd_bus_send_with_reply_and_block(
1491                 sd_bus *bus,
1492                 sd_bus_message *m,
1493                 uint64_t usec,
1494                 sd_bus_error *error,
1495                 sd_bus_message **reply) {
1496
1497         int r;
1498         usec_t timeout;
1499         uint64_t serial;
1500         bool room = false;
1501
1502         if (!bus)
1503                 return -EINVAL;
1504         if (!BUS_IS_OPEN(bus->state))
1505                 return -ENOTCONN;
1506         if (!m)
1507                 return -EINVAL;
1508         if (m->header->type != SD_BUS_MESSAGE_TYPE_METHOD_CALL)
1509                 return -EINVAL;
1510         if (m->header->flags & SD_BUS_MESSAGE_NO_REPLY_EXPECTED)
1511                 return -EINVAL;
1512         if (bus_error_is_dirty(error))
1513                 return -EINVAL;
1514         if (bus_pid_changed(bus))
1515                 return -ECHILD;
1516
1517         r = bus_ensure_running(bus);
1518         if (r < 0)
1519                 return r;
1520
1521         r = sd_bus_send(bus, m, &serial);
1522         if (r < 0)
1523                 return r;
1524
1525         timeout = calc_elapse(usec);
1526
1527         for (;;) {
1528                 usec_t left;
1529                 sd_bus_message *incoming = NULL;
1530
1531                 if (!room) {
1532                         sd_bus_message **q;
1533
1534                         if (bus->rqueue_size >= BUS_RQUEUE_MAX)
1535                                 return -ENOBUFS;
1536
1537                         /* Make sure there's room for queuing this
1538                          * locally, before we read the message */
1539
1540                         q = realloc(bus->rqueue, (bus->rqueue_size + 1) * sizeof(sd_bus_message*));
1541                         if (!q)
1542                                 return -ENOMEM;
1543
1544                         bus->rqueue = q;
1545                         room = true;
1546                 }
1547
1548                 if (bus->is_kernel)
1549                         r = bus_kernel_read_message(bus, &incoming);
1550                 else
1551                         r = bus_socket_read_message(bus, &incoming);
1552                 if (r < 0)
1553                         return r;
1554                 if (incoming) {
1555
1556                         if (incoming->reply_serial == serial) {
1557                                 /* Found a match! */
1558
1559                                 if (incoming->header->type == SD_BUS_MESSAGE_TYPE_METHOD_RETURN) {
1560
1561                                         if (reply)
1562                                                 *reply = incoming;
1563                                         else
1564                                                 sd_bus_message_unref(incoming);
1565
1566                                         return 0;
1567                                 }
1568
1569                                 if (incoming->header->type == SD_BUS_MESSAGE_TYPE_METHOD_ERROR) {
1570                                         int k;
1571
1572                                         r = sd_bus_error_copy(error, &incoming->error);
1573                                         if (r < 0) {
1574                                                 sd_bus_message_unref(incoming);
1575                                                 return r;
1576                                         }
1577
1578                                         k = bus_error_to_errno(&incoming->error);
1579                                         sd_bus_message_unref(incoming);
1580                                         return k;
1581                                 }
1582
1583                                 sd_bus_message_unref(incoming);
1584                                 return -EIO;
1585                         }
1586
1587                         /* There's already guaranteed to be room for
1588                          * this, so need to resize things here */
1589                         bus->rqueue[bus->rqueue_size ++] = incoming;
1590                         room = false;
1591
1592                         /* Try to read more, right-away */
1593                         continue;
1594                 }
1595                 if (r != 0)
1596                         continue;
1597
1598                 if (timeout > 0) {
1599                         usec_t n;
1600
1601                         n = now(CLOCK_MONOTONIC);
1602                         if (n >= timeout)
1603                                 return -ETIMEDOUT;
1604
1605                         left = timeout - n;
1606                 } else
1607                         left = (uint64_t) -1;
1608
1609                 r = bus_poll(bus, true, left);
1610                 if (r < 0)
1611                         return r;
1612
1613                 r = dispatch_wqueue(bus);
1614                 if (r < 0)
1615                         return r;
1616         }
1617 }
1618
1619 int sd_bus_get_fd(sd_bus *bus) {
1620         if (!bus)
1621                 return -EINVAL;
1622         if (!BUS_IS_OPEN(bus->state))
1623                 return -ENOTCONN;
1624         if (bus->input_fd != bus->output_fd)
1625                 return -EPERM;
1626         if (bus_pid_changed(bus))
1627                 return -ECHILD;
1628
1629         return bus->input_fd;
1630 }
1631
1632 int sd_bus_get_events(sd_bus *bus) {
1633         int flags = 0;
1634
1635         if (!bus)
1636                 return -EINVAL;
1637         if (!BUS_IS_OPEN(bus->state))
1638                 return -ENOTCONN;
1639         if (bus_pid_changed(bus))
1640                 return -ECHILD;
1641
1642         if (bus->state == BUS_OPENING)
1643                 flags |= POLLOUT;
1644         else if (bus->state == BUS_AUTHENTICATING) {
1645
1646                 if (bus_socket_auth_needs_write(bus))
1647                         flags |= POLLOUT;
1648
1649                 flags |= POLLIN;
1650
1651         } else if (bus->state == BUS_RUNNING || bus->state == BUS_HELLO) {
1652                 if (bus->rqueue_size <= 0)
1653                         flags |= POLLIN;
1654                 if (bus->wqueue_size > 0)
1655                         flags |= POLLOUT;
1656         }
1657
1658         return flags;
1659 }
1660
1661 int sd_bus_get_timeout(sd_bus *bus, uint64_t *timeout_usec) {
1662         struct reply_callback *c;
1663
1664         if (!bus)
1665                 return -EINVAL;
1666         if (!timeout_usec)
1667                 return -EINVAL;
1668         if (!BUS_IS_OPEN(bus->state))
1669                 return -ENOTCONN;
1670         if (bus_pid_changed(bus))
1671                 return -ECHILD;
1672
1673         if (bus->state == BUS_AUTHENTICATING) {
1674                 *timeout_usec = bus->auth_timeout;
1675                 return 1;
1676         }
1677
1678         if (bus->state != BUS_RUNNING && bus->state != BUS_HELLO) {
1679                 *timeout_usec = (uint64_t) -1;
1680                 return 0;
1681         }
1682
1683         c = prioq_peek(bus->reply_callbacks_prioq);
1684         if (!c) {
1685                 *timeout_usec = (uint64_t) -1;
1686                 return 0;
1687         }
1688
1689         *timeout_usec = c->timeout;
1690         return 1;
1691 }
1692
1693 static int process_timeout(sd_bus *bus) {
1694         _cleanup_bus_message_unref_ sd_bus_message* m = NULL;
1695         struct reply_callback *c;
1696         usec_t n;
1697         int r;
1698
1699         assert(bus);
1700
1701         c = prioq_peek(bus->reply_callbacks_prioq);
1702         if (!c)
1703                 return 0;
1704
1705         n = now(CLOCK_MONOTONIC);
1706         if (c->timeout > n)
1707                 return 0;
1708
1709         r = bus_message_new_synthetic_error(
1710                         bus,
1711                         c->serial,
1712                         &SD_BUS_ERROR_MAKE("org.freedesktop.DBus.Error.Timeout", "Timed out"),
1713                         &m);
1714         if (r < 0)
1715                 return r;
1716
1717         assert_se(prioq_pop(bus->reply_callbacks_prioq) == c);
1718         hashmap_remove(bus->reply_callbacks, &c->serial);
1719
1720         r = c->callback(bus, m, c->userdata);
1721         free(c);
1722
1723         return r < 0 ? r : 1;
1724 }
1725
1726 static int process_hello(sd_bus *bus, sd_bus_message *m) {
1727         assert(bus);
1728         assert(m);
1729
1730         if (bus->state != BUS_HELLO)
1731                 return 0;
1732
1733         /* Let's make sure the first message on the bus is the HELLO
1734          * reply. But note that we don't actually parse the message
1735          * here (we leave that to the usual handling), we just verify
1736          * we don't let any earlier msg through. */
1737
1738         if (m->header->type != SD_BUS_MESSAGE_TYPE_METHOD_RETURN &&
1739             m->header->type != SD_BUS_MESSAGE_TYPE_METHOD_ERROR)
1740                 return -EIO;
1741
1742         if (m->reply_serial != bus->hello_serial)
1743                 return -EIO;
1744
1745         return 0;
1746 }
1747
1748 static int process_reply(sd_bus *bus, sd_bus_message *m) {
1749         struct reply_callback *c;
1750         int r;
1751
1752         assert(bus);
1753         assert(m);
1754
1755         if (m->header->type != SD_BUS_MESSAGE_TYPE_METHOD_RETURN &&
1756             m->header->type != SD_BUS_MESSAGE_TYPE_METHOD_ERROR)
1757                 return 0;
1758
1759         c = hashmap_remove(bus->reply_callbacks, &m->reply_serial);
1760         if (!c)
1761                 return 0;
1762
1763         if (c->timeout != 0)
1764                 prioq_remove(bus->reply_callbacks_prioq, c, &c->prioq_idx);
1765
1766         r = sd_bus_message_rewind(m, true);
1767         if (r < 0)
1768                 return r;
1769
1770         r = c->callback(bus, m, c->userdata);
1771         free(c);
1772
1773         return r;
1774 }
1775
1776 static int process_filter(sd_bus *bus, sd_bus_message *m) {
1777         struct filter_callback *l;
1778         int r;
1779
1780         assert(bus);
1781         assert(m);
1782
1783         do {
1784                 bus->filter_callbacks_modified = false;
1785
1786                 LIST_FOREACH(callbacks, l, bus->filter_callbacks) {
1787
1788                         if (bus->filter_callbacks_modified)
1789                                 break;
1790
1791                         /* Don't run this more than once per iteration */
1792                         if (l->last_iteration == bus->iteration_counter)
1793                                 continue;
1794
1795                         l->last_iteration = bus->iteration_counter;
1796
1797                         r = sd_bus_message_rewind(m, true);
1798                         if (r < 0)
1799                                 return r;
1800
1801                         r = l->callback(bus, m, l->userdata);
1802                         if (r != 0)
1803                                 return r;
1804
1805                 }
1806
1807         } while (bus->filter_callbacks_modified);
1808
1809         return 0;
1810 }
1811
1812 static int process_match(sd_bus *bus, sd_bus_message *m) {
1813         int r;
1814
1815         assert(bus);
1816         assert(m);
1817
1818         do {
1819                 bus->match_callbacks_modified = false;
1820
1821                 r = bus_match_run(bus, &bus->match_callbacks, m);
1822                 if (r != 0)
1823                         return r;
1824
1825         } while (bus->match_callbacks_modified);
1826
1827         return 0;
1828 }
1829
1830 static int process_builtin(sd_bus *bus, sd_bus_message *m) {
1831         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
1832         int r;
1833
1834         assert(bus);
1835         assert(m);
1836
1837         if (m->header->type != SD_BUS_MESSAGE_TYPE_METHOD_CALL)
1838                 return 0;
1839
1840         if (!streq_ptr(m->interface, "org.freedesktop.DBus.Peer"))
1841                 return 0;
1842
1843         if (m->header->flags & SD_BUS_MESSAGE_NO_REPLY_EXPECTED)
1844                 return 1;
1845
1846         if (streq_ptr(m->member, "Ping"))
1847                 r = sd_bus_message_new_method_return(bus, m, &reply);
1848         else if (streq_ptr(m->member, "GetMachineId")) {
1849                 sd_id128_t id;
1850                 char sid[33];
1851
1852                 r = sd_id128_get_machine(&id);
1853                 if (r < 0)
1854                         return r;
1855
1856                 r = sd_bus_message_new_method_return(bus, m, &reply);
1857                 if (r < 0)
1858                         return r;
1859
1860                 r = sd_bus_message_append(reply, "s", sd_id128_to_string(id, sid));
1861         } else {
1862                 r = sd_bus_message_new_method_errorf(
1863                                 bus, m, &reply,
1864                                 "org.freedesktop.DBus.Error.UnknownMethod",
1865                                  "Unknown method '%s' on interface '%s'.", m->member, m->interface);
1866         }
1867
1868         if (r < 0)
1869                 return r;
1870
1871         r = sd_bus_send(bus, reply, NULL);
1872         if (r < 0)
1873                 return r;
1874
1875         return 1;
1876 }
1877
1878 static int node_vtable_get_userdata(
1879                 sd_bus *bus,
1880                 const char *path,
1881                 struct node_vtable *c,
1882                 void **userdata) {
1883
1884         void *u;
1885         int r;
1886
1887         assert(bus);
1888         assert(path);
1889         assert(c);
1890
1891         u = c->userdata;
1892         if (c->find) {
1893                 r = c->find(bus, path, c->interface, &u, u);
1894                 if (r <= 0)
1895                         return r;
1896         }
1897
1898         if (userdata)
1899                 *userdata = u;
1900
1901         return 1;
1902 }
1903
1904 static void *vtable_property_convert_userdata(const sd_bus_vtable *p, void *u) {
1905         assert(p);
1906
1907         return (uint8_t*) u + p->property.offset;
1908 }
1909
1910 static int vtable_property_get_userdata(
1911                 sd_bus *bus,
1912                 const char *path,
1913                 struct vtable_member *p,
1914                 void **userdata) {
1915
1916         void *u;
1917         int r;
1918
1919         assert(bus);
1920         assert(path);
1921         assert(p);
1922         assert(userdata);
1923
1924         r = node_vtable_get_userdata(bus, path, p->parent, &u);
1925         if (r <= 0)
1926                 return r;
1927
1928         *userdata = vtable_property_convert_userdata(p->vtable, u);
1929         return 1;
1930 }
1931
1932 static int add_enumerated_to_set(sd_bus *bus, const char *prefix, struct node_enumerator *first, Set *s) {
1933         struct node_enumerator *c;
1934         int r;
1935
1936         assert(bus);
1937         assert(prefix);
1938         assert(s);
1939
1940         LIST_FOREACH(enumerators, c, first) {
1941                 char **children = NULL, **k;
1942
1943                 r = c->callback(bus, prefix, &children, c->userdata);
1944                 if (r < 0)
1945                         return r;
1946
1947                 STRV_FOREACH(k, children) {
1948                         if (r < 0) {
1949                                 free(*k);
1950                                 continue;
1951                         }
1952
1953                         if (!object_path_is_valid(*k) && object_path_startswith(*k, prefix)) {
1954                                 free(*k);
1955                                 r = -EINVAL;
1956                                 continue;
1957                         }
1958
1959                         r = set_consume(s, *k);
1960                 }
1961
1962                 free(children);
1963                 if (r < 0)
1964                         return r;
1965         }
1966
1967         return 0;
1968 }
1969
1970 static int add_subtree_to_set(sd_bus *bus, const char *prefix, struct node *n, Set *s) {
1971         struct node *i;
1972         int r;
1973
1974         assert(bus);
1975         assert(prefix);
1976         assert(n);
1977         assert(s);
1978
1979         r = add_enumerated_to_set(bus, prefix, n->enumerators, s);
1980         if (r < 0)
1981                 return r;
1982
1983         LIST_FOREACH(siblings, i, n->child) {
1984                 char *t;
1985
1986                 t = strdup(i->path);
1987                 if (!t)
1988                         return -ENOMEM;
1989
1990                 r = set_consume(s, t);
1991                 if (r < 0 && r != -EEXIST)
1992                         return r;
1993
1994                 r = add_subtree_to_set(bus, prefix, i, s);
1995                 if (r < 0)
1996                         return r;
1997         }
1998
1999         return 0;
2000 }
2001
2002 static int get_child_nodes(sd_bus *bus, const char *prefix, struct node *n, Set **_s) {
2003         Set *s = NULL;
2004         int r;
2005
2006         assert(bus);
2007         assert(n);
2008         assert(_s);
2009
2010         s = set_new(string_hash_func, string_compare_func);
2011         if (!s)
2012                 return -ENOMEM;
2013
2014         r = add_subtree_to_set(bus, prefix, n, s);
2015         if (r < 0) {
2016                 set_free_free(s);
2017                 return r;
2018         }
2019
2020         *_s = s;
2021         return 0;
2022 }
2023
2024 static int node_callbacks_run(
2025                 sd_bus *bus,
2026                 sd_bus_message *m,
2027                 struct node_callback *first,
2028                 bool require_fallback,
2029                 bool *found_object) {
2030
2031         struct node_callback *c;
2032         int r;
2033
2034         assert(bus);
2035         assert(m);
2036         assert(found_object);
2037
2038         LIST_FOREACH(callbacks, c, first) {
2039                 if (require_fallback && !c->is_fallback)
2040                         continue;
2041
2042                 *found_object = true;
2043
2044                 if (c->last_iteration == bus->iteration_counter)
2045                         continue;
2046
2047                 r = sd_bus_message_rewind(m, true);
2048                 if (r < 0)
2049                         return r;
2050
2051                 r = c->callback(bus, m, c->userdata);
2052                 if (r != 0)
2053                         return r;
2054         }
2055
2056         return 0;
2057 }
2058
2059 static int method_callbacks_run(
2060                 sd_bus *bus,
2061                 sd_bus_message *m,
2062                 struct vtable_member *c,
2063                 bool require_fallback,
2064                 bool *found_object) {
2065
2066         const char *signature;
2067         void *u;
2068         int r;
2069
2070         assert(bus);
2071         assert(m);
2072         assert(c);
2073         assert(found_object);
2074
2075         if (require_fallback && !c->parent->is_fallback)
2076                 return 0;
2077
2078         r = node_vtable_get_userdata(bus, m->path, c->parent, &u);
2079         if (r <= 0)
2080                 return r;
2081
2082         *found_object = true;
2083
2084         r = sd_bus_message_rewind(m, true);
2085         if (r < 0)
2086                 return r;
2087
2088         r = sd_bus_message_get_signature(m, true, &signature);
2089         if (r < 0)
2090                 return r;
2091
2092         if (!streq(c->vtable->method.signature, signature)) {
2093                 r = sd_bus_reply_method_errorf(bus, m,
2094                                                "org.freedesktop.DBus.Error.InvalidArgs",
2095                                                "Invalid arguments '%s' to call %s:%s, expecting '%s'.",
2096                                                signature, c->interface, c->member, c->vtable->method.signature);
2097                 if (r < 0)
2098                         return r;
2099
2100                 return 1;
2101         }
2102
2103         return c->vtable->method.handler(bus, m, u);
2104 }
2105
2106 static int property_get_set_callbacks_run(
2107                 sd_bus *bus,
2108                 sd_bus_message *m,
2109                 struct vtable_member *c,
2110                 bool require_fallback,
2111                 bool is_get,
2112                 bool *found_object) {
2113
2114         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
2115         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
2116         void *u;
2117         int r;
2118
2119         assert(bus);
2120         assert(m);
2121         assert(found_object);
2122
2123         if (require_fallback && !c->parent->is_fallback)
2124                 return 0;
2125
2126         r = vtable_property_get_userdata(bus, m->path, c, &u);
2127         if (r <= 0)
2128                 return r;
2129
2130         *found_object = true;
2131
2132         r = sd_bus_message_new_method_return(bus, m, &reply);
2133         if (r < 0)
2134                 return r;
2135
2136         c->last_iteration = bus->iteration_counter;
2137
2138         if (is_get) {
2139                 r = sd_bus_message_open_container(reply, 'v', c->vtable->property.signature);
2140                 if (r < 0)
2141                         return r;
2142
2143                 if (c->vtable->property.get) {
2144                         r = c->vtable->property.get(bus, m->path, c->interface, c->member, reply, &error, u);
2145                         if (r < 0)
2146                                 return r;
2147                 } else
2148                         assert_not_reached("automatic properties not supported yet");
2149
2150                 if (sd_bus_error_is_set(&error)) {
2151                         r = sd_bus_reply_method_error(bus, m, &error);
2152                         if (r < 0)
2153                                 return r;
2154
2155                         return 1;
2156                 }
2157
2158                 r = sd_bus_message_close_container(reply);
2159                 if (r < 0)
2160                         return r;
2161
2162         } else {
2163                 if (c->vtable->type != _SD_BUS_VTABLE_WRITABLE_PROPERTY)
2164                         sd_bus_error_setf(&error, "org.freedesktop.DBus.Error.PropertyReadOnly", "Property '%s' is not writable.", c->member);
2165                 else  {
2166                         r = sd_bus_message_enter_container(m, 'v', c->vtable->property.signature);
2167                         if (r < 0)
2168                                 return r;
2169
2170                         if (c->vtable->property.set) {
2171                                 r = c->vtable->property.set(bus, m->path, c->interface, c->member, m, &error, u);
2172                                 if (r < 0)
2173                                         return r;
2174                         } else
2175                                 assert_not_reached("automatic properties not supported yet");
2176                 }
2177
2178                 if (sd_bus_error_is_set(&error)) {
2179                         r = sd_bus_reply_method_error(bus, m, &error);
2180                         if (r < 0)
2181                                 return r;
2182
2183                         return 1;
2184                 }
2185
2186                 r = sd_bus_message_exit_container(m);
2187                 if (r < 0)
2188                         return r;
2189         }
2190
2191         r = sd_bus_send(bus, reply, NULL);
2192         if (r < 0)
2193                 return r;
2194
2195         return 1;
2196 }
2197
2198 static int vtable_append_all_properties(
2199                 sd_bus *bus,
2200                 sd_bus_message *reply,
2201                 const char *path,
2202                 struct node_vtable *c,
2203                 void *userdata,
2204                 sd_bus_error *error) {
2205
2206         const sd_bus_vtable *v;
2207         int r;
2208
2209         assert(bus);
2210         assert(reply);
2211         assert(c);
2212
2213         for (v = c->vtable+1; v->type != _SD_BUS_VTABLE_END; v++) {
2214                 if (v->type != _SD_BUS_VTABLE_PROPERTY && v->type != _SD_BUS_VTABLE_WRITABLE_PROPERTY)
2215                         continue;
2216
2217                 r = sd_bus_message_open_container(reply, 'e', "sv");
2218                 if (r < 0)
2219                         return r;
2220
2221                 r = sd_bus_message_append(reply, "s", c->interface);
2222                 if (r < 0)
2223                         return r;
2224
2225                 r = sd_bus_message_open_container(reply, 'v', v->property.signature);
2226                 if (r < 0)
2227                         return r;
2228
2229                 r = v->property.get(bus, path, c->interface, v->property.member, reply, error, vtable_property_convert_userdata(v, userdata));
2230                 if (r < 0)
2231                         return r;
2232
2233                 if (sd_bus_error_is_set(error))
2234                         return 0;
2235
2236                 r = sd_bus_message_close_container(reply);
2237                 if (r < 0)
2238                         return r;
2239
2240                 r = sd_bus_message_close_container(reply);
2241                 if (r < 0)
2242                         return r;
2243         }
2244
2245         return 1;
2246 }
2247
2248 static int property_get_all_callbacks_run(
2249                 sd_bus *bus,
2250                 sd_bus_message *m,
2251                 struct node_vtable *first,
2252                 bool require_fallback,
2253                 const char *iface,
2254                 bool *found_object) {
2255
2256         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
2257         struct node_vtable *c;
2258         bool found_interface = false;
2259         int r;
2260
2261         assert(bus);
2262         assert(m);
2263         assert(found_object);
2264
2265         r = sd_bus_message_new_method_return(bus, m, &reply);
2266         if (r < 0)
2267                 return r;
2268
2269         r = sd_bus_message_open_container(reply, 'a', "{sv}");
2270         if (r < 0)
2271                 return r;
2272
2273         LIST_FOREACH(vtables, c, first) {
2274                 _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
2275                 void *u;
2276
2277                 if (require_fallback && !c->is_fallback)
2278                         continue;
2279
2280                 r = node_vtable_get_userdata(bus, m->path, c, &u);
2281                 if (r < 0)
2282                         return r;
2283                 if (r == 0)
2284                         continue;
2285
2286                 *found_object = true;
2287
2288                 if (iface && !streq(c->interface, iface))
2289                         continue;
2290                 found_interface = true;
2291
2292                 c->last_iteration = bus->iteration_counter;
2293
2294                 r = vtable_append_all_properties(bus, reply, m->path, c, u, &error);
2295                 if (r < 0)
2296                         return r;
2297
2298                 if (sd_bus_error_is_set(&error)) {
2299                         r = sd_bus_reply_method_error(bus, m, &error);
2300                         if (r < 0)
2301                                 return r;
2302
2303                         return 1;
2304                 }
2305         }
2306
2307         if (!found_interface) {
2308                 r = sd_bus_reply_method_errorf(
2309                                 bus, m,
2310                                 "org.freedesktop.DBus.Error.UnknownInterface",
2311                                 "Unknown interface '%s'.", iface);
2312                 if (r < 0)
2313                         return r;
2314
2315                 return 1;
2316         }
2317
2318         r = sd_bus_message_close_container(reply);
2319         if (r < 0)
2320                 return r;
2321
2322         r = sd_bus_send(bus, reply, NULL);
2323         if (r < 0)
2324                 return r;
2325
2326         return 1;
2327 }
2328
2329 static bool bus_node_with_object_manager(sd_bus *bus, struct node *n) {
2330         assert(bus);
2331
2332         if (n->object_manager)
2333                 return true;
2334
2335         if (n->parent)
2336                 return bus_node_with_object_manager(bus, n->parent);
2337
2338         return false;
2339 }
2340
2341 static bool bus_node_exists(sd_bus *bus, struct node *n, const char *path, bool require_fallback) {
2342         struct node_vtable *c;
2343         struct node_callback *k;
2344
2345         assert(bus);
2346         assert(n);
2347
2348         /* Tests if there's anything attached directly to this node
2349          * for the specified path */
2350
2351         LIST_FOREACH(callbacks, k, n->callbacks) {
2352                 if (require_fallback && !k->is_fallback)
2353                         continue;
2354
2355                 return true;
2356         }
2357
2358         LIST_FOREACH(vtables, c, n->vtables) {
2359
2360                 if (require_fallback && !c->is_fallback)
2361                         continue;
2362
2363                 if (node_vtable_get_userdata(bus, path, c, NULL) > 0)
2364                         return true;
2365         }
2366
2367         return !require_fallback && (n->enumerators || n->object_manager);
2368 }
2369
2370 static int process_introspect(
2371                 sd_bus *bus,
2372                 sd_bus_message *m,
2373                 struct node *n,
2374                 bool require_fallback,
2375                 bool *found_object) {
2376
2377         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
2378         _cleanup_set_free_free_ Set *s = NULL;
2379         struct introspect intro;
2380         struct node_vtable *c;
2381         bool empty;
2382         int r;
2383
2384         assert(bus);
2385         assert(m);
2386         assert(n);
2387         assert(found_object);
2388
2389         r = get_child_nodes(bus, m->path, n, &s);
2390         if (r < 0)
2391                 return r;
2392
2393         r = introspect_begin(&intro);
2394         if (r < 0)
2395                 return r;
2396
2397         r = introspect_write_default_interfaces(&intro, bus_node_with_object_manager(bus, n));
2398         if (r < 0)
2399                 return r;
2400
2401         empty = set_isempty(s);
2402
2403         LIST_FOREACH(vtables, c, n->vtables) {
2404                 if (require_fallback && !c->is_fallback)
2405                         continue;
2406
2407                 r = node_vtable_get_userdata(bus, m->path, c, NULL);
2408                 if (r < 0)
2409                         return r;
2410                 if (r == 0)
2411                         continue;
2412
2413                 empty = false;
2414
2415                 r = introspect_write_interface(&intro, c->interface, c->vtable);
2416                 if (r < 0)
2417                         goto finish;
2418         }
2419
2420         if (empty) {
2421                 /* Nothing?, let's see if we exist at all, and if not
2422                  * refuse to do anything */
2423                 r = bus_node_exists(bus, n, m->path, require_fallback);
2424                 if (r < 0)
2425                         return r;
2426
2427                 if (r == 0)
2428                         goto finish;
2429         }
2430
2431         *found_object = true;
2432
2433         r = introspect_write_child_nodes(&intro, s, m->path);
2434         if (r < 0)
2435                 goto finish;
2436
2437         r = introspect_finish(&intro, bus, m, &reply);
2438         if (r < 0)
2439                 goto finish;
2440
2441         r = sd_bus_send(bus, reply, NULL);
2442         if (r < 0)
2443                 goto finish;
2444
2445         r = 1;
2446
2447 finish:
2448         introspect_free(&intro);
2449         return r;
2450 }
2451
2452 static int object_manager_serialize_vtable(
2453                 sd_bus *bus,
2454                 sd_bus_message *reply,
2455                 const char *path,
2456                 struct node_vtable *c,
2457                 sd_bus_error *error) {
2458
2459         void *u;
2460         int r;
2461
2462         assert(bus);
2463         assert(reply);
2464         assert(path);
2465         assert(c);
2466         assert(error);
2467
2468         r = node_vtable_get_userdata(bus, path, c, &u);
2469         if (r <= 0)
2470                 return r;
2471
2472         r = sd_bus_message_open_container(reply, 'e', "sa{sv}");
2473         if (r < 0)
2474                 return r;
2475
2476         r = sd_bus_message_append(reply, "s", c->interface);
2477         if (r < 0)
2478                 return r;
2479
2480         r = sd_bus_message_open_container(reply, 'a', "{sv}");
2481         if (r < 0)
2482                 return r;
2483
2484         r = vtable_append_all_properties(bus, reply, path, c, u, error);
2485         if (r < 0)
2486                 return r;
2487
2488         r = sd_bus_message_close_container(reply);
2489         if (r < 0)
2490                 return r;
2491
2492         r = sd_bus_message_close_container(reply);
2493         if (r < 0)
2494                 return r;
2495
2496         return 0;
2497 }
2498
2499 static int object_manager_serialize_path(
2500                 sd_bus *bus,
2501                 sd_bus_message *reply,
2502                 const char *path,
2503                 bool require_fallback,
2504                 sd_bus_error *error) {
2505
2506         struct node_vtable *i;
2507         struct node *n;
2508         int r;
2509
2510         assert(bus);
2511         assert(reply);
2512         assert(path);
2513         assert(error);
2514
2515         n = hashmap_get(bus->nodes, path);
2516         if (!n)
2517                 return 0;
2518
2519         r = sd_bus_message_open_container(reply, 'e', "oa{sa{sv}}");
2520         if (r < 0)
2521                 return r;
2522
2523         r = sd_bus_message_append(reply, "o", path);
2524         if (r < 0)
2525                 return r;
2526
2527         r = sd_bus_message_open_container(reply, 'a', "{sa{sv}}");
2528         if (r < 0)
2529                 return r;
2530
2531         LIST_FOREACH(vtables, i, n->vtables) {
2532
2533                 if (require_fallback && !i->is_fallback)
2534                         continue;
2535
2536                 r = object_manager_serialize_vtable(bus, reply, path, i, error);
2537                 if (r < 0)
2538                         return r;
2539                 if (sd_bus_error_is_set(error))
2540                         return 0;
2541         }
2542
2543         r = sd_bus_message_close_container(reply);
2544         if (r < 0)
2545                 return r;
2546
2547         r = sd_bus_message_close_container(reply);
2548         if (r < 0)
2549                 return r;
2550
2551         return 1;
2552 }
2553
2554 static int object_manager_serialize_path_and_fallbacks(
2555                 sd_bus *bus,
2556                 sd_bus_message *reply,
2557                 const char *path,
2558                 sd_bus_error *error) {
2559
2560         size_t pl;
2561         int r;
2562
2563         assert(bus);
2564         assert(reply);
2565         assert(path);
2566         assert(error);
2567
2568         /* First, add all vtables registered for this path */
2569         r = object_manager_serialize_path(bus, reply, path, false, error);
2570         if (r < 0)
2571                 return r;
2572         if (sd_bus_error_is_set(error))
2573                 return 0;
2574
2575         /* Second, add fallback vtables registered for any of the prefixes */
2576         pl = strlen(path);
2577         if (pl > 1) {
2578                 char p[pl + 1];
2579                 strcpy(p, path);
2580
2581                 for (;;) {
2582                         char *e;
2583
2584                         e = strrchr(p, '/');
2585                         if (e == p || !e)
2586                                 break;
2587
2588                         *e = 0;
2589
2590                         r = object_manager_serialize_path(bus, reply, p, true, error);
2591                         if (r < 0)
2592                                 return r;
2593
2594                         if (sd_bus_error_is_set(error))
2595                                 return 0;
2596                 }
2597         }
2598
2599         return 0;
2600 }
2601
2602 static int process_get_managed_objects(
2603                 sd_bus *bus,
2604                 sd_bus_message *m,
2605                 struct node *n,
2606                 bool require_fallback,
2607                 bool *found_object) {
2608
2609         _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
2610         _cleanup_set_free_free_ Set *s = NULL;
2611         bool empty;
2612         int r;
2613
2614         assert(bus);
2615         assert(m);
2616         assert(n);
2617         assert(found_object);
2618
2619         if (!bus_node_with_object_manager(bus, n))
2620                 return 0;
2621
2622         r = get_child_nodes(bus, m->path, n, &s);
2623         if (r < 0)
2624                 return r;
2625
2626         r = sd_bus_message_new_method_return(bus, m, &reply);
2627         if (r < 0)
2628                 return r;
2629
2630         r = sd_bus_message_open_container(reply, 'a', "{oa{sa{sv}}}");
2631         if (r < 0)
2632                 return r;
2633
2634         empty = set_isempty(s);
2635         if (empty) {
2636                 struct node_vtable *c;
2637
2638                 /* Hmm, so we have no children? Then let's check
2639                  * whether we exist at all, i.e. whether at least one
2640                  * vtable exists. */
2641
2642                 LIST_FOREACH(vtables, c, n->vtables) {
2643
2644                         if (require_fallback && !c->is_fallback)
2645                                 continue;
2646
2647                         if (r < 0)
2648                                 return r;
2649                         if (r == 0)
2650                                 continue;
2651
2652                         empty = false;
2653                         break;
2654                 }
2655
2656                 if (empty)
2657                         return 0;
2658         } else {
2659                 Iterator i;
2660                 char *path;
2661
2662                 SET_FOREACH(path, s, i) {
2663                         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
2664
2665                         r = object_manager_serialize_path_and_fallbacks(bus, reply, path, &error);
2666                         if (r < 0)
2667                                 return -ENOMEM;
2668
2669                         if (sd_bus_error_is_set(&error)) {
2670                                 r = sd_bus_reply_method_error(bus, m, &error);
2671                                 if (r < 0)
2672                                         return r;
2673
2674                                 return 1;
2675                         }
2676                 }
2677         }
2678
2679         r = sd_bus_message_close_container(reply);
2680         if (r < 0)
2681                 return r;
2682
2683         r = sd_bus_send(bus, reply, NULL);
2684         if (r < 0)
2685                 return r;
2686
2687         return 1;
2688 }
2689
2690 static int object_find_and_run(sd_bus *bus, sd_bus_message *m, const char *p, bool require_fallback, bool *found_object) {
2691         struct node *n;
2692         struct vtable_member vtable_key, *v;
2693         int r;
2694
2695         assert(bus);
2696         assert(m);
2697         assert(p);
2698         assert(found_object);
2699
2700         n = hashmap_get(bus->nodes, p);
2701         if (!n)
2702                 return 0;
2703
2704         /* First, try object callbacks */
2705         r = node_callbacks_run(bus, m, n->callbacks, require_fallback, found_object);
2706         if (r != 0)
2707                 return r;
2708
2709         if (!m->interface || !m->member)
2710                 return 0;
2711
2712         /* Then, look for a known method */
2713         vtable_key.path = (char*) p;
2714         vtable_key.interface = m->interface;
2715         vtable_key.member = m->member;
2716
2717         v = hashmap_get(bus->vtable_methods, &vtable_key);
2718         if (v) {
2719                 r = method_callbacks_run(bus, m, v, require_fallback, found_object);
2720                 if (r != 0)
2721                         return r;
2722         }
2723
2724         /* Then, look for a known property */
2725         if (streq(m->interface, "org.freedesktop.DBus.Properties")) {
2726                 bool get = false;
2727
2728                 get = streq(m->member, "Get");
2729
2730                 if (get || streq(m->member, "Set")) {
2731
2732                         r = sd_bus_message_rewind(m, true);
2733                         if (r < 0)
2734                                 return r;
2735
2736                         vtable_key.path = (char*) p;
2737
2738                         r = sd_bus_message_read(m, "ss", &vtable_key.interface, &vtable_key.member);
2739                         if (r < 0)
2740                                 return r;
2741
2742                         v = hashmap_get(bus->vtable_properties, &vtable_key);
2743                         if (v) {
2744                                 r = property_get_set_callbacks_run(bus, m, v, require_fallback, get, found_object);
2745                                 if (r != 0)
2746                                         return r;
2747                         }
2748
2749                 } else if (streq(m->member, "GetAll")) {
2750                         const char *iface;
2751
2752                         r = sd_bus_message_rewind(m, true);
2753                         if (r < 0)
2754                                 return r;
2755
2756                         r = sd_bus_message_read(m, "s", &iface);
2757                         if (r < 0)
2758                                 return r;
2759
2760                         if (iface[0] == 0)
2761                                 iface = NULL;
2762
2763                         r = property_get_all_callbacks_run(bus, m, n->vtables, require_fallback, iface, found_object);
2764                         if (r != 0)
2765                                 return r;
2766                 }
2767
2768         } else if (sd_bus_message_is_method_call(m, "org.freedesktop.DBus.Introspectable", "Introspect")) {
2769
2770                 r = process_introspect(bus, m, n, require_fallback, found_object);
2771                 if (r != 0)
2772                         return r;
2773
2774         } else if (sd_bus_message_is_method_call(m, "org.freedesktop.DBus.ObjectManager", "GetManagedObjects")) {
2775
2776                 r = process_get_managed_objects(bus, m, n, require_fallback, found_object);
2777                 if (r != 0)
2778                         return r;
2779         }
2780
2781         if (!*found_object) {
2782                 r = bus_node_exists(bus, n, m->path, require_fallback);
2783                 if (r < 0)
2784                         return r;
2785
2786                 if (r > 0)
2787                         *found_object = true;
2788         }
2789
2790         return 0;
2791 }
2792
2793 static int process_object(sd_bus *bus, sd_bus_message *m) {
2794         int r;
2795         size_t pl;
2796         bool found_object = false;
2797
2798         assert(bus);
2799         assert(m);
2800
2801         if (m->header->type != SD_BUS_MESSAGE_TYPE_METHOD_CALL)
2802                 return 0;
2803
2804         if (!m->path)
2805                 return 0;
2806
2807         if (hashmap_isempty(bus->nodes))
2808                 return 0;
2809
2810         pl = strlen(m->path);
2811         do {
2812                 char p[pl+1];
2813
2814                 bus->nodes_modified = false;
2815
2816                 r = object_find_and_run(bus, m, m->path, false, &found_object);
2817                 if (r != 0)
2818                         return r;
2819
2820                 /* Look for fallback prefixes */
2821                 strcpy(p, m->path);
2822                 for (;;) {
2823                         char *e;
2824
2825                         if (streq(p, "/"))
2826                                 break;
2827
2828                         if (bus->nodes_modified)
2829                                 break;
2830
2831                         e = strrchr(p, '/');
2832                         assert(e);
2833                         if (e == p)
2834                                 *(e+1) = 0;
2835                         else
2836                                 *e = 0;
2837
2838                         r = object_find_and_run(bus, m, p, true, &found_object);
2839                         if (r != 0)
2840                                 return r;
2841                 }
2842
2843         } while (bus->nodes_modified);
2844
2845         if (!found_object)
2846                 return 0;
2847
2848         if (sd_bus_message_is_method_call(m, "org.freedesktop.DBus.Properties", "Get") ||
2849             sd_bus_message_is_method_call(m, "org.freedesktop.DBus.Properties", "Set"))
2850                 r = sd_bus_reply_method_errorf(
2851                                 bus, m,
2852                                 "org.freedesktop.DBus.Error.UnknownProperty",
2853                                 "Unknown property or interface.");
2854         else
2855                 r = sd_bus_reply_method_errorf(
2856                                 bus, m,
2857                                 "org.freedesktop.DBus.Error.UnknownMethod",
2858                                 "Unknown method '%s' or interface '%s'.", m->member, m->interface);
2859
2860         if (r < 0)
2861                 return r;
2862
2863         return 1;
2864 }
2865
2866 static int process_message(sd_bus *bus, sd_bus_message *m) {
2867         int r;
2868
2869         assert(bus);
2870         assert(m);
2871
2872         bus->iteration_counter++;
2873
2874         r = process_hello(bus, m);
2875         if (r != 0)
2876                 return r;
2877
2878         r = process_reply(bus, m);
2879         if (r != 0)
2880                 return r;
2881
2882         r = process_filter(bus, m);
2883         if (r != 0)
2884                 return r;
2885
2886         r = process_match(bus, m);
2887         if (r != 0)
2888                 return r;
2889
2890         r = process_builtin(bus, m);
2891         if (r != 0)
2892                 return r;
2893
2894         return process_object(bus, m);
2895 }
2896
2897 static int process_running(sd_bus *bus, sd_bus_message **ret) {
2898         _cleanup_bus_message_unref_ sd_bus_message *m = NULL;
2899         int r;
2900
2901         assert(bus);
2902         assert(bus->state == BUS_RUNNING || bus->state == BUS_HELLO);
2903
2904         r = process_timeout(bus);
2905         if (r != 0)
2906                 goto null_message;
2907
2908         r = dispatch_wqueue(bus);
2909         if (r != 0)
2910                 goto null_message;
2911
2912         r = dispatch_rqueue(bus, &m);
2913         if (r < 0)
2914                 return r;
2915         if (!m)
2916                 goto null_message;
2917
2918         r = process_message(bus, m);
2919         if (r != 0)
2920                 goto null_message;
2921
2922         if (ret) {
2923                 r = sd_bus_message_rewind(m, true);
2924                 if (r < 0)
2925                         return r;
2926
2927                 *ret = m;
2928                 m = NULL;
2929                 return 1;
2930         }
2931
2932         if (m->header->type == SD_BUS_MESSAGE_TYPE_METHOD_CALL) {
2933
2934                 r = sd_bus_reply_method_errorf(
2935                                 bus, m,
2936                                 "org.freedesktop.DBus.Error.UnknownObject",
2937                                 "Unknown object '%s'.", m->path);
2938                 if (r < 0)
2939                         return r;
2940         }
2941
2942         return 1;
2943
2944 null_message:
2945         if (r >= 0 && ret)
2946                 *ret = NULL;
2947
2948         return r;
2949 }
2950
2951 int sd_bus_process(sd_bus *bus, sd_bus_message **ret) {
2952         int r;
2953
2954         /* Returns 0 when we didn't do anything. This should cause the
2955          * caller to invoke sd_bus_wait() before returning the next
2956          * time. Returns > 0 when we did something, which possibly
2957          * means *ret is filled in with an unprocessed message. */
2958
2959         if (!bus)
2960                 return -EINVAL;
2961         if (bus_pid_changed(bus))
2962                 return -ECHILD;
2963
2964         /* We don't allow recursively invoking sd_bus_process(). */
2965         if (bus->processing)
2966                 return -EBUSY;
2967
2968         switch (bus->state) {
2969
2970         case BUS_UNSET:
2971         case BUS_CLOSED:
2972                 return -ENOTCONN;
2973
2974         case BUS_OPENING:
2975                 r = bus_socket_process_opening(bus);
2976                 if (r < 0)
2977                         return r;
2978                 if (ret)
2979                         *ret = NULL;
2980                 return r;
2981
2982         case BUS_AUTHENTICATING:
2983
2984                 r = bus_socket_process_authenticating(bus);
2985                 if (r < 0)
2986                         return r;
2987                 if (ret)
2988                         *ret = NULL;
2989                 return r;
2990
2991         case BUS_RUNNING:
2992         case BUS_HELLO:
2993
2994                 bus->processing = true;
2995                 r = process_running(bus, ret);
2996                 bus->processing = false;
2997
2998                 return r;
2999         }
3000
3001         assert_not_reached("Unknown state");
3002 }
3003
3004 static int bus_poll(sd_bus *bus, bool need_more, uint64_t timeout_usec) {
3005         struct pollfd p[2] = {};
3006         int r, e, n;
3007         struct timespec ts;
3008         usec_t until, m;
3009
3010         assert(bus);
3011
3012         if (!BUS_IS_OPEN(bus->state))
3013                 return -ENOTCONN;
3014
3015         e = sd_bus_get_events(bus);
3016         if (e < 0)
3017                 return e;
3018
3019         if (need_more)
3020                 e |= POLLIN;
3021
3022         r = sd_bus_get_timeout(bus, &until);
3023         if (r < 0)
3024                 return r;
3025         if (r == 0)
3026                 m = (uint64_t) -1;
3027         else {
3028                 usec_t nw;
3029                 nw = now(CLOCK_MONOTONIC);
3030                 m = until > nw ? until - nw : 0;
3031         }
3032
3033         if (timeout_usec != (uint64_t) -1 && (m == (uint64_t) -1 || timeout_usec < m))
3034                 m = timeout_usec;
3035
3036         p[0].fd = bus->input_fd;
3037         if (bus->output_fd == bus->input_fd) {
3038                 p[0].events = e;
3039                 n = 1;
3040         } else {
3041                 p[0].events = e & POLLIN;
3042                 p[1].fd = bus->output_fd;
3043                 p[1].events = e & POLLOUT;
3044                 n = 2;
3045         }
3046
3047         r = ppoll(p, n, m == (uint64_t) -1 ? NULL : timespec_store(&ts, m), NULL);
3048         if (r < 0)
3049                 return -errno;
3050
3051         return r > 0 ? 1 : 0;
3052 }
3053
3054 int sd_bus_wait(sd_bus *bus, uint64_t timeout_usec) {
3055
3056         if (!bus)
3057                 return -EINVAL;
3058         if (!BUS_IS_OPEN(bus->state))
3059                 return -ENOTCONN;
3060         if (bus_pid_changed(bus))
3061                 return -ECHILD;
3062
3063         if (bus->rqueue_size > 0)
3064                 return 0;
3065
3066         return bus_poll(bus, false, timeout_usec);
3067 }
3068
3069 int sd_bus_flush(sd_bus *bus) {
3070         int r;
3071
3072         if (!bus)
3073                 return -EINVAL;
3074         if (!BUS_IS_OPEN(bus->state))
3075                 return -ENOTCONN;
3076         if (bus_pid_changed(bus))
3077                 return -ECHILD;
3078
3079         r = bus_ensure_running(bus);
3080         if (r < 0)
3081                 return r;
3082
3083         if (bus->wqueue_size <= 0)
3084                 return 0;
3085
3086         for (;;) {
3087                 r = dispatch_wqueue(bus);
3088                 if (r < 0)
3089                         return r;
3090
3091                 if (bus->wqueue_size <= 0)
3092                         return 0;
3093
3094                 r = bus_poll(bus, false, (uint64_t) -1);
3095                 if (r < 0)
3096                         return r;
3097         }
3098 }
3099
3100 int sd_bus_add_filter(sd_bus *bus, sd_bus_message_handler_t callback, void *userdata) {
3101         struct filter_callback *f;
3102
3103         if (!bus)
3104                 return -EINVAL;
3105         if (!callback)
3106                 return -EINVAL;
3107         if (bus_pid_changed(bus))
3108                 return -ECHILD;
3109
3110         f = new0(struct filter_callback, 1);
3111         if (!f)
3112                 return -ENOMEM;
3113         f->callback = callback;
3114         f->userdata = userdata;
3115
3116         bus->filter_callbacks_modified = true;
3117         LIST_PREPEND(struct filter_callback, callbacks, bus->filter_callbacks, f);
3118         return 0;
3119 }
3120
3121 int sd_bus_remove_filter(sd_bus *bus, sd_bus_message_handler_t callback, void *userdata) {
3122         struct filter_callback *f;
3123
3124         if (!bus)
3125                 return -EINVAL;
3126         if (!callback)
3127                 return -EINVAL;
3128         if (bus_pid_changed(bus))
3129                 return -ECHILD;
3130
3131         LIST_FOREACH(callbacks, f, bus->filter_callbacks) {
3132                 if (f->callback == callback && f->userdata == userdata) {
3133                         bus->filter_callbacks_modified = true;
3134                         LIST_REMOVE(struct filter_callback, callbacks, bus->filter_callbacks, f);
3135                         free(f);
3136                         return 1;
3137                 }
3138         }
3139
3140         return 0;
3141 }
3142
3143 static struct node *bus_node_allocate(sd_bus *bus, const char *path) {
3144         struct node *n, *parent;
3145         const char *e;
3146         char *s, *p;
3147         int r;
3148
3149         assert(bus);
3150         assert(path);
3151         assert(path[0] == '/');
3152
3153         n = hashmap_get(bus->nodes, path);
3154         if (n)
3155                 return n;
3156
3157         r = hashmap_ensure_allocated(&bus->nodes, string_hash_func, string_compare_func);
3158         if (r < 0)
3159                 return NULL;
3160
3161         s = strdup(path);
3162         if (!s)
3163                 return NULL;
3164
3165         if (streq(path, "/"))
3166                 parent = NULL;
3167         else {
3168                 e = strrchr(path, '/');
3169                 assert(e);
3170
3171                 p = strndupa(path, MAX(1, path - e));
3172
3173                 parent = bus_node_allocate(bus, p);
3174                 if (!parent) {
3175                         free(s);
3176                         return NULL;
3177                 }
3178         }
3179
3180         n = new0(struct node, 1);
3181         if (!n)
3182                 return NULL;
3183
3184         n->parent = parent;
3185         n->path = s;
3186
3187         r = hashmap_put(bus->nodes, s, n);
3188         if (r < 0) {
3189                 free(s);
3190                 free(n);
3191                 return NULL;
3192         }
3193
3194         if (parent)
3195                 LIST_PREPEND(struct node, siblings, parent->child, n);
3196
3197         return n;
3198 }
3199
3200 static void bus_node_gc(sd_bus *b, struct node *n) {
3201         assert(b);
3202
3203         if (!n)
3204                 return;
3205
3206         if (n->child ||
3207             n->callbacks ||
3208             n->vtables ||
3209             n->enumerators ||
3210             n->object_manager)
3211                 return;
3212
3213         assert(hashmap_remove(b->nodes, n->path) == n);
3214
3215         if (n->parent)
3216                 LIST_REMOVE(struct node, siblings, n->parent->child, n);
3217
3218         free(n->path);
3219         bus_node_gc(b, n->parent);
3220         free(n);
3221 }
3222
3223 static int bus_add_object(
3224                 sd_bus *b,
3225                 bool fallback,
3226                 const char *path,
3227                 sd_bus_message_handler_t callback,
3228                 void *userdata) {
3229
3230         struct node_callback *c;
3231         struct node *n;
3232         int r;
3233
3234         if (!b)
3235                 return -EINVAL;
3236         if (!object_path_is_valid(path))
3237                 return -EINVAL;
3238         if (!callback)
3239                 return -EINVAL;
3240         if (bus_pid_changed(b))
3241                 return -ECHILD;
3242
3243         n = bus_node_allocate(b, path);
3244         if (!n)
3245                 return -ENOMEM;
3246
3247         c = new0(struct node_callback, 1);
3248         if (!c) {
3249                 r = -ENOMEM;
3250                 goto fail;
3251         }
3252
3253         c->node = n;
3254         c->callback = callback;
3255         c->userdata = userdata;
3256         c->is_fallback = fallback;
3257
3258         LIST_PREPEND(struct node_callback, callbacks, n->callbacks, c);
3259         return 0;
3260
3261 fail:
3262         free(c);
3263         bus_node_gc(b, n);
3264         return r;
3265 }
3266
3267 static int bus_remove_object(
3268                 sd_bus *bus,
3269                 bool fallback,
3270                 const char *path,
3271                 sd_bus_message_handler_t callback,
3272                 void *userdata) {
3273
3274         struct node_callback *c;
3275         struct node *n;
3276
3277         if (!bus)
3278                 return -EINVAL;
3279         if (!object_path_is_valid(path))
3280                 return -EINVAL;
3281         if (!callback)
3282                 return -EINVAL;
3283         if (bus_pid_changed(bus))
3284                 return -ECHILD;
3285
3286         n = hashmap_get(bus->nodes, path);
3287         if (!n)
3288                 return 0;
3289