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