src/libelogind/sd-bus/sd-bus.c

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