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(bus->state == BUS_HELLO || bus->state == 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 (*a != ';' && *a != ',' && *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 (**p != 0 && **p != ';') {
 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 (**p != 0 && **p != ';') {
 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 (**p != 0 && **p != ';') {
 681                 r = parse_address_key(p, "guid", guid);
 682                 if (r < 0)
 683                         goto fail;
 684                 else if (r > 0)
 685                         continue;
 686
 687                 r = parse_address_key(p, "path", &path);
 688                 if (r < 0)
 689                         goto fail;
 690                 else if (r > 0)
 691                         continue;
 692
 693                 if (startswith(*p, "argv")) {
 694                         unsigned ul;
 695
 696                         errno = 0;
 697                         ul = strtoul(*p + 4, (char**) p, 10);
 698                         if (errno > 0 || **p != '=' || ul > 256) {
 699                                 r = -EINVAL;
 700                                 goto fail;
 701                         }
 702
 703                         (*p)++;
 704
 705                         if (ul >= n_argv) {
 706                                 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 (**p != 0 && **p != ';') {
 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(",argv4=--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=", e, ",argv3=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 (bus->state != BUS_OPENING &&
1345             bus->state != BUS_AUTHENTICATING &&
1346             bus->state != BUS_HELLO &&
1347             bus->state != BUS_RUNNING)
1348                 return;
1349
1350         bus->state = BUS_CLOSING;
1351 }
1352
1353 _public_ sd_bus *sd_bus_ref(sd_bus *bus) {
1354
1355         if (!bus)
1356                 return NULL;
1357
1358         assert_se(REFCNT_INC(bus->n_ref) >= 2);
1359
1360         return bus;
1361 }
1362
1363 _public_ sd_bus *sd_bus_unref(sd_bus *bus) {
1364         unsigned i;
1365
1366         if (!bus)
1367                 return NULL;
1368
1369         i = REFCNT_DEC(bus->n_ref);
1370         if (i > 0)
1371                 return NULL;
1372
1373         bus_free(bus);
1374         return NULL;
1375 }
1376
1377 #if 0 /// UNNEEDED by elogind
1378 _public_ int sd_bus_is_open(sd_bus *bus) {
1379
1380         assert_return(bus, -EINVAL);
1381         assert_return(!bus_pid_changed(bus), -ECHILD);
1382
1383         return BUS_IS_OPEN(bus->state);
1384 }
1385 #endif // 0
1386
1387 _public_ int sd_bus_can_send(sd_bus *bus, char type) {
1388         int r;
1389
1390         assert_return(bus, -EINVAL);
1391         assert_return(bus->state != BUS_UNSET, -ENOTCONN);
1392         assert_return(!bus_pid_changed(bus), -ECHILD);
1393
1394         if (bus->hello_flags & KDBUS_HELLO_MONITOR)
1395                 return 0;
1396
1397         if (type == SD_BUS_TYPE_UNIX_FD) {
1398                 if (!(bus->hello_flags & KDBUS_HELLO_ACCEPT_FD))
1399                         return 0;
1400
1401                 r = bus_ensure_running(bus);
1402                 if (r < 0)
1403                         return r;
1404
1405                 return bus->can_fds;
1406         }
1407
1408         return bus_type_is_valid(type);
1409 }
1410
1411 #if 0 /// UNNEEDED by elogind
1412 _public_ int sd_bus_get_bus_id(sd_bus *bus, sd_id128_t *id) {
1413         int r;
1414
1415         assert_return(bus, -EINVAL);
1416         assert_return(id, -EINVAL);
1417         assert_return(!bus_pid_changed(bus), -ECHILD);
1418
1419         r = bus_ensure_running(bus);
1420         if (r < 0)
1421                 return r;
1422
1423         *id = bus->server_id;
1424         return 0;
1425 }
1426 #endif // 0
1427
1428 static int bus_seal_message(sd_bus *b, sd_bus_message *m, usec_t timeout) {
1429         assert(b);
1430         assert(m);
1431
1432         if (m->sealed) {
1433                 /* If we copy the same message to multiple
1434                  * destinations, avoid using the same cookie
1435                  * numbers. */
1436                 b->cookie = MAX(b->cookie, BUS_MESSAGE_COOKIE(m));
1437                 return 0;
1438         }
1439
1440         if (timeout == 0)
1441                 timeout = BUS_DEFAULT_TIMEOUT;
1442
1443         return bus_message_seal(m, ++b->cookie, timeout);
1444 }
1445
1446 static int bus_remarshal_message(sd_bus *b, sd_bus_message **m) {
1447         bool remarshal = false;
1448
1449         assert(b);
1450
1451         /* wrong packet version */
1452         if (b->message_version != 0 && b->message_version != (*m)->header->version)
1453                 remarshal = true;
1454
1455         /* wrong packet endianness */
1456         if (b->message_endian != 0 && b->message_endian != (*m)->header->endian)
1457                 remarshal = true;
1458
1459         return remarshal ? bus_message_remarshal(b, m) : 0;
1460 }
1461
1462 int bus_seal_synthetic_message(sd_bus *b, sd_bus_message *m) {
1463         assert(b);
1464         assert(m);
1465
1466         /* Fake some timestamps, if they were requested, and not
1467          * already initialized */
1468         if (b->attach_flags & KDBUS_ATTACH_TIMESTAMP) {
1469                 if (m->realtime <= 0)
1470                         m->realtime = now(CLOCK_REALTIME);
1471
1472                 if (m->monotonic <= 0)
1473                         m->monotonic = now(CLOCK_MONOTONIC);
1474         }
1475
1476         /* The bus specification says the serial number cannot be 0,
1477          * hence let's fill something in for synthetic messages. Since
1478          * synthetic messages might have a fake sender and we don't
1479          * want to interfere with the real sender's serial numbers we
1480          * pick a fixed, artificial one. We use (uint32_t) -1 rather
1481          * than (uint64_t) -1 since dbus1 only had 32bit identifiers,
1482          * even though kdbus can do 64bit. */
1483         return bus_message_seal(m, 0xFFFFFFFFULL, 0);
1484 }
1485
1486 static int bus_write_message(sd_bus *bus, sd_bus_message *m, bool hint_sync_call, size_t *idx) {
1487         int r;
1488
1489         assert(bus);
1490         assert(m);
1491
1492         r = bus_socket_write_message(bus, m, idx);
1493         if (r <= 0)
1494                 return r;
1495
1496         if (*idx >= BUS_MESSAGE_SIZE(m))
1497                 log_debug("Sent message type=%s sender=%s destination=%s object=%s interface=%s member=%s cookie=%" PRIu64 " reply_cookie=%" PRIu64 " error=%s",
1498                           bus_message_type_to_string(m->header->type),
1499                           strna(sd_bus_message_get_sender(m)),
1500                           strna(sd_bus_message_get_destination(m)),
1501                           strna(sd_bus_message_get_path(m)),
1502                           strna(sd_bus_message_get_interface(m)),
1503                           strna(sd_bus_message_get_member(m)),
1504                           BUS_MESSAGE_COOKIE(m),
1505                           m->reply_cookie,
1506                           strna(m->error.message));
1507
1508         return r;
1509 }
1510
1511 static int dispatch_wqueue(sd_bus *bus) {
1512         int r, ret = 0;
1513
1514         assert(bus);
1515         assert(bus->state == BUS_RUNNING || bus->state == BUS_HELLO);
1516
1517         while (bus->wqueue_size > 0) {
1518
1519                 r = bus_write_message(bus, bus->wqueue[0], false, &bus->windex);
1520                 if (r < 0)
1521                         return r;
1522                 else if (r == 0)
1523                         /* Didn't do anything this time */
1524                         return ret;
1525                 else if (bus->windex >= BUS_MESSAGE_SIZE(bus->wqueue[0])) {
1526                         /* Fully written. Let's drop the entry from
1527                          * the queue.
1528                          *
1529                          * This isn't particularly optimized, but
1530                          * well, this is supposed to be our worst-case
1531                          * buffer only, and the socket buffer is
1532                          * supposed to be our primary buffer, and if
1533                          * it got full, then all bets are off
1534                          * anyway. */
1535
1536                         bus->wqueue_size--;
1537                         sd_bus_message_unref(bus->wqueue[0]);
1538                         memmove(bus->wqueue, bus->wqueue + 1, sizeof(sd_bus_message*) * bus->wqueue_size);
1539                         bus->windex = 0;
1540
1541                         ret = 1;
1542                 }
1543         }
1544
1545         return ret;
1546 }
1547
1548 static int bus_read_message(sd_bus *bus, bool hint_priority, int64_t priority) {
1549         assert(bus);
1550
1551         return bus_socket_read_message(bus);
1552 }
1553
1554 int bus_rqueue_make_room(sd_bus *bus) {
1555         assert(bus);
1556
1557         if (bus->rqueue_size >= BUS_RQUEUE_MAX)
1558                 return -ENOBUFS;
1559
1560         if (!GREEDY_REALLOC(bus->rqueue, bus->rqueue_allocated, bus->rqueue_size + 1))
1561                 return -ENOMEM;
1562
1563         return 0;
1564 }
1565
1566 static int dispatch_rqueue(sd_bus *bus, bool hint_priority, int64_t priority, sd_bus_message **m) {
1567         int r, ret = 0;
1568
1569         assert(bus);
1570         assert(m);
1571         assert(bus->state == BUS_RUNNING || bus->state == BUS_HELLO);
1572
1573         /* Note that the priority logic is only available on kdbus,
1574          * where the rqueue is unused. We check the rqueue here
1575          * anyway, because it's simple... */
1576
1577         for (;;) {
1578                 if (bus->rqueue_size > 0) {
1579                         /* Dispatch a queued message */
1580
1581                         *m = bus->rqueue[0];
1582                         bus->rqueue_size--;
1583                         memmove(bus->rqueue, bus->rqueue + 1, sizeof(sd_bus_message*) * bus->rqueue_size);
1584                         return 1;
1585                 }
1586
1587                 /* Try to read a new message */
1588                 r = bus_read_message(bus, hint_priority, priority);
1589                 if (r < 0)
1590                         return r;
1591                 if (r == 0)
1592                         return ret;
1593
1594                 ret = 1;
1595         }
1596 }
1597
1598 static int bus_send_internal(sd_bus *bus, sd_bus_message *_m, uint64_t *cookie, bool hint_sync_call) {
1599         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = sd_bus_message_ref(_m);
1600         int r;
1601
1602         assert_return(m, -EINVAL);
1603
1604         if (!bus)
1605                 bus = m->bus;
1606
1607         assert_return(!bus_pid_changed(bus), -ECHILD);
1608
1609         if (!BUS_IS_OPEN(bus->state))
1610                 return -ENOTCONN;
1611
1612         if (m->n_fds > 0) {
1613                 r = sd_bus_can_send(bus, SD_BUS_TYPE_UNIX_FD);
1614                 if (r < 0)
1615                         return r;
1616                 if (r == 0)
1617                         return -EOPNOTSUPP;
1618         }
1619
1620         /* If the cookie number isn't kept, then we know that no reply
1621          * is expected */
1622         if (!cookie && !m->sealed)
1623                 m->header->flags |= BUS_MESSAGE_NO_REPLY_EXPECTED;
1624
1625         r = bus_seal_message(bus, m, 0);
1626         if (r < 0)
1627                 return r;
1628
1629         /* Remarshall if we have to. This will possibly unref the
1630          * message and place a replacement in m */
1631         r = bus_remarshal_message(bus, &m);
1632         if (r < 0)
1633                 return r;
1634
1635         /* If this is a reply and no reply was requested, then let's
1636          * suppress this, if we can */
1637         if (m->dont_send)
1638                 goto finish;
1639
1640         if ((bus->state == BUS_RUNNING || bus->state == BUS_HELLO) && bus->wqueue_size <= 0) {
1641                 size_t idx = 0;
1642
1643                 r = bus_write_message(bus, m, hint_sync_call, &idx);
1644                 if (r < 0) {
1645                         if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
1646                                 bus_enter_closing(bus);
1647                                 return -ECONNRESET;
1648                         }
1649
1650                         return r;
1651                 }
1652
1653                 if (idx < BUS_MESSAGE_SIZE(m))  {
1654                         /* Wasn't fully written. So let's remember how
1655                          * much was written. Note that the first entry
1656                          * of the wqueue array is always allocated so
1657                          * that we always can remember how much was
1658                          * written. */
1659                         bus->wqueue[0] = sd_bus_message_ref(m);
1660                         bus->wqueue_size = 1;
1661                         bus->windex = idx;
1662                 }
1663
1664         } else {
1665                 /* Just append it to the queue. */
1666
1667                 if (bus->wqueue_size >= BUS_WQUEUE_MAX)
1668                         return -ENOBUFS;
1669
1670                 if (!GREEDY_REALLOC(bus->wqueue, bus->wqueue_allocated, bus->wqueue_size + 1))
1671                         return -ENOMEM;
1672
1673                 bus->wqueue[bus->wqueue_size++] = sd_bus_message_ref(m);
1674         }
1675
1676 finish:
1677         if (cookie)
1678                 *cookie = BUS_MESSAGE_COOKIE(m);
1679
1680         return 1;
1681 }
1682
1683 _public_ int sd_bus_send(sd_bus *bus, sd_bus_message *m, uint64_t *cookie) {
1684         return bus_send_internal(bus, m, cookie, false);
1685 }
1686
1687 #if 0 /// UNNEEDED by elogind
1688 _public_ int sd_bus_send_to(sd_bus *bus, sd_bus_message *m, const char *destination, uint64_t *cookie) {
1689         int r;
1690
1691         assert_return(m, -EINVAL);
1692
1693         if (!bus)
1694                 bus = m->bus;
1695
1696         assert_return(!bus_pid_changed(bus), -ECHILD);
1697
1698         if (!BUS_IS_OPEN(bus->state))
1699                 return -ENOTCONN;
1700
1701         if (!streq_ptr(m->destination, destination)) {
1702
1703                 if (!destination)
1704                         return -EEXIST;
1705
1706                 r = sd_bus_message_set_destination(m, destination);
1707                 if (r < 0)
1708                         return r;
1709         }
1710
1711         return sd_bus_send(bus, m, cookie);
1712 }
1713 #endif // 0
1714
1715 static usec_t calc_elapse(uint64_t usec) {
1716         if (usec == (uint64_t) -1)
1717                 return 0;
1718
1719         return now(CLOCK_MONOTONIC) + usec;
1720 }
1721
1722 static int timeout_compare(const void *a, const void *b) {
1723         const struct reply_callback *x = a, *y = b;
1724
1725         if (x->timeout != 0 && y->timeout == 0)
1726                 return -1;
1727
1728         if (x->timeout == 0 && y->timeout != 0)
1729                 return 1;
1730
1731         if (x->timeout < y->timeout)
1732                 return -1;
1733
1734         if (x->timeout > y->timeout)
1735                 return 1;
1736
1737         return 0;
1738 }
1739
1740 _public_ int sd_bus_call_async(
1741                 sd_bus *bus,
1742                 sd_bus_slot **slot,
1743                 sd_bus_message *_m,
1744                 sd_bus_message_handler_t callback,
1745                 void *userdata,
1746                 uint64_t usec) {
1747
1748         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = sd_bus_message_ref(_m);
1749         _cleanup_(sd_bus_slot_unrefp) sd_bus_slot *s = NULL;
1750         int r;
1751
1752         assert_return(m, -EINVAL);
1753         assert_return(m->header->type == SD_BUS_MESSAGE_METHOD_CALL, -EINVAL);
1754         assert_return(!(m->header->flags & BUS_MESSAGE_NO_REPLY_EXPECTED), -EINVAL);
1755         assert_return(callback, -EINVAL);
1756
1757         if (!bus)
1758                 bus = m->bus;
1759
1760         assert_return(!bus_pid_changed(bus), -ECHILD);
1761
1762         if (!BUS_IS_OPEN(bus->state))
1763                 return -ENOTCONN;
1764
1765         r = ordered_hashmap_ensure_allocated(&bus->reply_callbacks, &uint64_hash_ops);
1766         if (r < 0)
1767                 return r;
1768
1769         r = prioq_ensure_allocated(&bus->reply_callbacks_prioq, timeout_compare);
1770         if (r < 0)
1771                 return r;
1772
1773         r = bus_seal_message(bus, m, usec);
1774         if (r < 0)
1775                 return r;
1776
1777         r = bus_remarshal_message(bus, &m);
1778         if (r < 0)
1779                 return r;
1780
1781         s = bus_slot_allocate(bus, !slot, BUS_REPLY_CALLBACK, sizeof(struct reply_callback), userdata);
1782         if (!s)
1783                 return -ENOMEM;
1784
1785         s->reply_callback.callback = callback;
1786
1787         s->reply_callback.cookie = BUS_MESSAGE_COOKIE(m);
1788         r = ordered_hashmap_put(bus->reply_callbacks, &s->reply_callback.cookie, &s->reply_callback);
1789         if (r < 0) {
1790                 s->reply_callback.cookie = 0;
1791                 return r;
1792         }
1793
1794         s->reply_callback.timeout = calc_elapse(m->timeout);
1795         if (s->reply_callback.timeout != 0) {
1796                 r = prioq_put(bus->reply_callbacks_prioq, &s->reply_callback, &s->reply_callback.prioq_idx);
1797                 if (r < 0) {
1798                         s->reply_callback.timeout = 0;
1799                         return r;
1800                 }
1801         }
1802
1803         r = sd_bus_send(bus, m, &s->reply_callback.cookie);
1804         if (r < 0)
1805                 return r;
1806
1807         if (slot)
1808                 *slot = s;
1809         s = NULL;
1810
1811         return r;
1812 }
1813
1814 int bus_ensure_running(sd_bus *bus) {
1815         int r;
1816
1817         assert(bus);
1818
1819         if (bus->state == BUS_UNSET || bus->state == BUS_CLOSED || bus->state == BUS_CLOSING)
1820                 return -ENOTCONN;
1821         if (bus->state == BUS_RUNNING)
1822                 return 1;
1823
1824         for (;;) {
1825                 r = sd_bus_process(bus, NULL);
1826                 if (r < 0)
1827                         return r;
1828                 if (bus->state == BUS_RUNNING)
1829                         return 1;
1830                 if (r > 0)
1831                         continue;
1832
1833                 r = sd_bus_wait(bus, (uint64_t) -1);
1834                 if (r < 0)
1835                         return r;
1836         }
1837 }
1838
1839 _public_ int sd_bus_call(
1840                 sd_bus *bus,
1841                 sd_bus_message *_m,
1842                 uint64_t usec,
1843                 sd_bus_error *error,
1844                 sd_bus_message **reply) {
1845
1846         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = sd_bus_message_ref(_m);
1847         usec_t timeout;
1848         uint64_t cookie;
1849         unsigned i;
1850         int r;
1851
1852         bus_assert_return(m, -EINVAL, error);
1853         bus_assert_return(m->header->type == SD_BUS_MESSAGE_METHOD_CALL, -EINVAL, error);
1854         bus_assert_return(!(m->header->flags & BUS_MESSAGE_NO_REPLY_EXPECTED), -EINVAL, error);
1855         bus_assert_return(!bus_error_is_dirty(error), -EINVAL, error);
1856
1857         if (!bus)
1858                 bus = m->bus;
1859
1860         bus_assert_return(!bus_pid_changed(bus), -ECHILD, error);
1861
1862         if (!BUS_IS_OPEN(bus->state)) {
1863                 r = -ENOTCONN;
1864                 goto fail;
1865         }
1866
1867         r = bus_ensure_running(bus);
1868         if (r < 0)
1869                 goto fail;
1870
1871         i = bus->rqueue_size;
1872
1873         r = bus_seal_message(bus, m, usec);
1874         if (r < 0)
1875                 goto fail;
1876
1877         r = bus_remarshal_message(bus, &m);
1878         if (r < 0)
1879                 goto fail;
1880
1881         r = bus_send_internal(bus, m, &cookie, true);
1882         if (r < 0)
1883                 goto fail;
1884
1885         timeout = calc_elapse(m->timeout);
1886
1887         for (;;) {
1888                 usec_t left;
1889
1890                 while (i < bus->rqueue_size) {
1891                         sd_bus_message *incoming = NULL;
1892
1893                         incoming = bus->rqueue[i];
1894
1895                         if (incoming->reply_cookie == cookie) {
1896                                 /* Found a match! */
1897
1898                                 memmove(bus->rqueue + i, bus->rqueue + i + 1, sizeof(sd_bus_message*) * (bus->rqueue_size - i - 1));
1899                                 bus->rqueue_size--;
1900                                 log_debug_bus_message(incoming);
1901
1902                                 if (incoming->header->type == SD_BUS_MESSAGE_METHOD_RETURN) {
1903
1904                                         if (incoming->n_fds <= 0 || (bus->hello_flags & KDBUS_HELLO_ACCEPT_FD)) {
1905                                                 if (reply)
1906                                                         *reply = incoming;
1907                                                 else
1908                                                         sd_bus_message_unref(incoming);
1909
1910                                                 return 1;
1911                                         }
1912
1913                                         r = sd_bus_error_setf(error, SD_BUS_ERROR_INCONSISTENT_MESSAGE, "Reply message contained file descriptors which I couldn't accept. Sorry.");
1914                                         sd_bus_message_unref(incoming);
1915                                         return r;
1916
1917                                 } else if (incoming->header->type == SD_BUS_MESSAGE_METHOD_ERROR) {
1918                                         r = sd_bus_error_copy(error, &incoming->error);
1919                                         sd_bus_message_unref(incoming);
1920                                         return r;
1921                                 } else {
1922                                         r = -EIO;
1923                                         goto fail;
1924                                 }
1925
1926                         } else if (BUS_MESSAGE_COOKIE(incoming) == cookie &&
1927                                    bus->unique_name &&
1928                                    incoming->sender &&
1929                                    streq(bus->unique_name, incoming->sender)) {
1930
1931                                 memmove(bus->rqueue + i, bus->rqueue + i + 1, sizeof(sd_bus_message*) * (bus->rqueue_size - i - 1));
1932                                 bus->rqueue_size--;
1933
1934                                 /* Our own message? Somebody is trying
1935                                  * to send its own client a message,
1936                                  * let's not dead-lock, let's fail
1937                                  * immediately. */
1938
1939                                 sd_bus_message_unref(incoming);
1940                                 r = -ELOOP;
1941                                 goto fail;
1942                         }
1943
1944                         /* Try to read more, right-away */
1945                         i++;
1946                 }
1947
1948                 r = bus_read_message(bus, false, 0);
1949                 if (r < 0) {
1950                         if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
1951                                 bus_enter_closing(bus);
1952                                 r = -ECONNRESET;
1953                         }
1954
1955                         goto fail;
1956                 }
1957                 if (r > 0)
1958                         continue;
1959
1960                 if (timeout > 0) {
1961                         usec_t n;
1962
1963                         n = now(CLOCK_MONOTONIC);
1964                         if (n >= timeout) {
1965                                 r = -ETIMEDOUT;
1966                                 goto fail;
1967                         }
1968
1969                         left = timeout - n;
1970                 } else
1971                         left = (uint64_t) -1;
1972
1973                 r = bus_poll(bus, true, left);
1974                 if (r < 0)
1975                         goto fail;
1976                 if (r == 0) {
1977                         r = -ETIMEDOUT;
1978                         goto fail;
1979                 }
1980
1981                 r = dispatch_wqueue(bus);
1982                 if (r < 0) {
1983                         if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
1984                                 bus_enter_closing(bus);
1985                                 r = -ECONNRESET;
1986                         }
1987
1988                         goto fail;
1989                 }
1990         }
1991
1992 fail:
1993         return sd_bus_error_set_errno(error, r);
1994 }
1995
1996 #if 0 /// UNNEEDED by elogind
1997 _public_ int sd_bus_get_fd(sd_bus *bus) {
1998
1999         assert_return(bus, -EINVAL);
2000         assert_return(bus->input_fd == bus->output_fd, -EPERM);
2001         assert_return(!bus_pid_changed(bus), -ECHILD);
2002
2003         return bus->input_fd;
2004 }
2005 #endif // 0
2006
2007 _public_ int sd_bus_get_events(sd_bus *bus) {
2008         int flags = 0;
2009
2010         assert_return(bus, -EINVAL);
2011         assert_return(!bus_pid_changed(bus), -ECHILD);
2012
2013         if (!BUS_IS_OPEN(bus->state) && bus->state != BUS_CLOSING)
2014                 return -ENOTCONN;
2015
2016         if (bus->state == BUS_OPENING)
2017                 flags |= POLLOUT;
2018         else if (bus->state == BUS_AUTHENTICATING) {
2019
2020                 if (bus_socket_auth_needs_write(bus))
2021                         flags |= POLLOUT;
2022
2023                 flags |= POLLIN;
2024
2025         } else if (bus->state == BUS_RUNNING || bus->state == BUS_HELLO) {
2026                 if (bus->rqueue_size <= 0)
2027                         flags |= POLLIN;
2028                 if (bus->wqueue_size > 0)
2029                         flags |= POLLOUT;
2030         }
2031
2032         return flags;
2033 }
2034
2035 _public_ int sd_bus_get_timeout(sd_bus *bus, uint64_t *timeout_usec) {
2036         struct reply_callback *c;
2037
2038         assert_return(bus, -EINVAL);
2039         assert_return(timeout_usec, -EINVAL);
2040         assert_return(!bus_pid_changed(bus), -ECHILD);
2041
2042         if (!BUS_IS_OPEN(bus->state) && bus->state != BUS_CLOSING)
2043                 return -ENOTCONN;
2044
2045         if (bus->track_queue) {
2046                 *timeout_usec = 0;
2047                 return 1;
2048         }
2049
2050         if (bus->state == BUS_CLOSING) {
2051                 *timeout_usec = 0;
2052                 return 1;
2053         }
2054
2055         if (bus->state == BUS_AUTHENTICATING) {
2056                 *timeout_usec = bus->auth_timeout;
2057                 return 1;
2058         }
2059
2060         if (bus->state != BUS_RUNNING && bus->state != BUS_HELLO) {
2061                 *timeout_usec = (uint64_t) -1;
2062                 return 0;
2063         }
2064
2065         if (bus->rqueue_size > 0) {
2066                 *timeout_usec = 0;
2067                 return 1;
2068         }
2069
2070         c = prioq_peek(bus->reply_callbacks_prioq);
2071         if (!c) {
2072                 *timeout_usec = (uint64_t) -1;
2073                 return 0;
2074         }
2075
2076         if (c->timeout == 0) {
2077                 *timeout_usec = (uint64_t) -1;
2078                 return 0;
2079         }
2080
2081         *timeout_usec = c->timeout;
2082         return 1;
2083 }
2084
2085 static int process_timeout(sd_bus *bus) {
2086         _cleanup_(sd_bus_error_free) sd_bus_error error_buffer = SD_BUS_ERROR_NULL;
2087         _cleanup_(sd_bus_message_unrefp) sd_bus_message* m = NULL;
2088         struct reply_callback *c;
2089         sd_bus_slot *slot;
2090         usec_t n;
2091         int r;
2092
2093         assert(bus);
2094
2095         c = prioq_peek(bus->reply_callbacks_prioq);
2096         if (!c)
2097                 return 0;
2098
2099         n = now(CLOCK_MONOTONIC);
2100         if (c->timeout > n)
2101                 return 0;
2102
2103         r = bus_message_new_synthetic_error(
2104                         bus,
2105                         c->cookie,
2106                         &SD_BUS_ERROR_MAKE_CONST(SD_BUS_ERROR_NO_REPLY, "Method call timed out"),
2107                         &m);
2108         if (r < 0)
2109                 return r;
2110
2111         r = bus_seal_synthetic_message(bus, m);
2112         if (r < 0)
2113                 return r;
2114
2115         assert_se(prioq_pop(bus->reply_callbacks_prioq) == c);
2116         c->timeout = 0;
2117
2118         ordered_hashmap_remove(bus->reply_callbacks, &c->cookie);
2119         c->cookie = 0;
2120
2121         slot = container_of(c, sd_bus_slot, reply_callback);
2122
2123         bus->iteration_counter++;
2124
2125         bus->current_message = m;
2126         bus->current_slot = sd_bus_slot_ref(slot);
2127         bus->current_handler = c->callback;
2128         bus->current_userdata = slot->userdata;
2129         r = c->callback(m, slot->userdata, &error_buffer);
2130         bus->current_userdata = NULL;
2131         bus->current_handler = NULL;
2132         bus->current_slot = NULL;
2133         bus->current_message = NULL;
2134
2135         if (slot->floating) {
2136                 bus_slot_disconnect(slot);
2137                 sd_bus_slot_unref(slot);
2138         }
2139
2140         sd_bus_slot_unref(slot);
2141
2142         return bus_maybe_reply_error(m, r, &error_buffer);
2143 }
2144
2145 static int process_hello(sd_bus *bus, sd_bus_message *m) {
2146         assert(bus);
2147         assert(m);
2148
2149         if (bus->state != BUS_HELLO)
2150                 return 0;
2151
2152         /* Let's make sure the first message on the bus is the HELLO
2153          * reply. But note that we don't actually parse the message
2154          * here (we leave that to the usual handling), we just verify
2155          * we don't let any earlier msg through. */
2156
2157         if (m->header->type != SD_BUS_MESSAGE_METHOD_RETURN &&
2158             m->header->type != SD_BUS_MESSAGE_METHOD_ERROR)
2159                 return -EIO;
2160
2161         if (m->reply_cookie != 1)
2162                 return -EIO;
2163
2164         return 0;
2165 }
2166
2167 static int process_reply(sd_bus *bus, sd_bus_message *m) {
2168         _cleanup_(sd_bus_message_unrefp) sd_bus_message *synthetic_reply = NULL;
2169         _cleanup_(sd_bus_error_free) sd_bus_error error_buffer = SD_BUS_ERROR_NULL;
2170         struct reply_callback *c;
2171         sd_bus_slot *slot;
2172         int r;
2173
2174         assert(bus);
2175         assert(m);
2176
2177         if (m->header->type != SD_BUS_MESSAGE_METHOD_RETURN &&
2178             m->header->type != SD_BUS_MESSAGE_METHOD_ERROR)
2179                 return 0;
2180
2181         if (m->destination && bus->unique_name && !streq_ptr(m->destination, bus->unique_name))
2182                 return 0;
2183
2184         c = ordered_hashmap_remove(bus->reply_callbacks, &m->reply_cookie);
2185         if (!c)
2186                 return 0;
2187
2188         c->cookie = 0;
2189
2190         slot = container_of(c, sd_bus_slot, reply_callback);
2191
2192         if (m->n_fds > 0 && !(bus->hello_flags & KDBUS_HELLO_ACCEPT_FD)) {
2193
2194                 /* If the reply contained a file descriptor which we
2195                  * didn't want we pass an error instead. */
2196
2197                 r = bus_message_new_synthetic_error(
2198                                 bus,
2199                                 m->reply_cookie,
2200                                 &SD_BUS_ERROR_MAKE_CONST(SD_BUS_ERROR_INCONSISTENT_MESSAGE, "Reply message contained file descriptor"),
2201                                 &synthetic_reply);
2202                 if (r < 0)
2203                         return r;
2204
2205                 /* Copy over original timestamp */
2206                 synthetic_reply->realtime = m->realtime;
2207                 synthetic_reply->monotonic = m->monotonic;
2208                 synthetic_reply->seqnum = m->seqnum;
2209
2210                 r = bus_seal_synthetic_message(bus, synthetic_reply);
2211                 if (r < 0)
2212                         return r;
2213
2214                 m = synthetic_reply;
2215         } else {
2216                 r = sd_bus_message_rewind(m, true);
2217                 if (r < 0)
2218                         return r;
2219         }
2220
2221         if (c->timeout != 0) {
2222                 prioq_remove(bus->reply_callbacks_prioq, c, &c->prioq_idx);
2223                 c->timeout = 0;
2224         }
2225
2226         bus->current_slot = sd_bus_slot_ref(slot);
2227         bus->current_handler = c->callback;
2228         bus->current_userdata = slot->userdata;
2229         r = c->callback(m, slot->userdata, &error_buffer);
2230         bus->current_userdata = NULL;
2231         bus->current_handler = NULL;
2232         bus->current_slot = NULL;
2233
2234         if (slot->floating) {
2235                 bus_slot_disconnect(slot);
2236                 sd_bus_slot_unref(slot);
2237         }
2238
2239         sd_bus_slot_unref(slot);
2240
2241         return bus_maybe_reply_error(m, r, &error_buffer);
2242 }
2243
2244 static int process_filter(sd_bus *bus, sd_bus_message *m) {
2245         _cleanup_(sd_bus_error_free) sd_bus_error error_buffer = SD_BUS_ERROR_NULL;
2246         struct filter_callback *l;
2247         int r;
2248
2249         assert(bus);
2250         assert(m);
2251
2252         do {
2253                 bus->filter_callbacks_modified = false;
2254
2255                 LIST_FOREACH(callbacks, l, bus->filter_callbacks) {
2256                         sd_bus_slot *slot;
2257
2258                         if (bus->filter_callbacks_modified)
2259                                 break;
2260
2261                         /* Don't run this more than once per iteration */
2262                         if (l->last_iteration == bus->iteration_counter)
2263                                 continue;
2264
2265                         l->last_iteration = bus->iteration_counter;
2266
2267                         r = sd_bus_message_rewind(m, true);
2268                         if (r < 0)
2269                                 return r;
2270
2271                         slot = container_of(l, sd_bus_slot, filter_callback);
2272
2273                         bus->current_slot = sd_bus_slot_ref(slot);
2274                         bus->current_handler = l->callback;
2275                         bus->current_userdata = slot->userdata;
2276                         r = l->callback(m, slot->userdata, &error_buffer);
2277                         bus->current_userdata = NULL;
2278                         bus->current_handler = NULL;
2279                         bus->current_slot = sd_bus_slot_unref(slot);
2280
2281                         r = bus_maybe_reply_error(m, r, &error_buffer);
2282                         if (r != 0)
2283                                 return r;
2284
2285                 }
2286
2287         } while (bus->filter_callbacks_modified);
2288
2289         return 0;
2290 }
2291
2292 static int process_match(sd_bus *bus, sd_bus_message *m) {
2293         int r;
2294
2295         assert(bus);
2296         assert(m);
2297
2298         do {
2299                 bus->match_callbacks_modified = false;
2300
2301                 r = bus_match_run(bus, &bus->match_callbacks, m);
2302                 if (r != 0)
2303                         return r;
2304
2305         } while (bus->match_callbacks_modified);
2306
2307         return 0;
2308 }
2309
2310 static int process_builtin(sd_bus *bus, sd_bus_message *m) {
2311         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
2312         int r;
2313
2314         assert(bus);
2315         assert(m);
2316
2317         if (bus->hello_flags & KDBUS_HELLO_MONITOR)
2318                 return 0;
2319
2320         if (bus->manual_peer_interface)
2321                 return 0;
2322
2323         if (m->header->type != SD_BUS_MESSAGE_METHOD_CALL)
2324                 return 0;
2325
2326         if (!streq_ptr(m->interface, "org.freedesktop.DBus.Peer"))
2327                 return 0;
2328
2329         if (m->header->flags & BUS_MESSAGE_NO_REPLY_EXPECTED)
2330                 return 1;
2331
2332         if (streq_ptr(m->member, "Ping"))
2333                 r = sd_bus_message_new_method_return(m, &reply);
2334         else if (streq_ptr(m->member, "GetMachineId")) {
2335                 sd_id128_t id;
2336                 char sid[33];
2337
2338                 r = sd_id128_get_machine(&id);
2339                 if (r < 0)
2340                         return r;
2341
2342                 r = sd_bus_message_new_method_return(m, &reply);
2343                 if (r < 0)
2344                         return r;
2345
2346                 r = sd_bus_message_append(reply, "s", sd_id128_to_string(id, sid));
2347         } else {
2348                 r = sd_bus_message_new_method_errorf(
2349                                 m, &reply,
2350                                 SD_BUS_ERROR_UNKNOWN_METHOD,
2351                                  "Unknown method '%s' on interface '%s'.", m->member, m->interface);
2352         }
2353
2354         if (r < 0)
2355                 return r;
2356
2357         r = sd_bus_send(bus, reply, NULL);
2358         if (r < 0)
2359                 return r;
2360
2361         return 1;
2362 }
2363
2364 static int process_fd_check(sd_bus *bus, sd_bus_message *m) {
2365         assert(bus);
2366         assert(m);
2367
2368         /* If we got a message with a file descriptor which we didn't
2369          * want to accept, then let's drop it. How can this even
2370          * happen? For example, when the kernel queues a message into
2371          * an activatable names's queue which allows fds, and then is
2372          * delivered to us later even though we ourselves did not
2373          * negotiate it. */
2374
2375         if (bus->hello_flags & KDBUS_HELLO_MONITOR)
2376                 return 0;
2377
2378         if (m->n_fds <= 0)
2379                 return 0;
2380
2381         if (bus->hello_flags & KDBUS_HELLO_ACCEPT_FD)
2382                 return 0;
2383
2384         if (m->header->type != SD_BUS_MESSAGE_METHOD_CALL)
2385                 return 1; /* just eat it up */
2386
2387         return sd_bus_reply_method_errorf(m, SD_BUS_ERROR_INCONSISTENT_MESSAGE, "Message contains file descriptors, which I cannot accept. Sorry.");
2388 }
2389
2390 static int process_message(sd_bus *bus, sd_bus_message *m) {
2391         int r;
2392
2393         assert(bus);
2394         assert(m);
2395
2396         bus->current_message = m;
2397         bus->iteration_counter++;
2398
2399         log_debug_bus_message(m);
2400
2401         r = process_hello(bus, m);
2402         if (r != 0)
2403                 goto finish;
2404
2405         r = process_reply(bus, m);
2406         if (r != 0)
2407                 goto finish;
2408
2409         r = process_fd_check(bus, m);
2410         if (r != 0)
2411                 goto finish;
2412
2413         r = process_filter(bus, m);
2414         if (r != 0)
2415                 goto finish;
2416
2417         r = process_match(bus, m);
2418         if (r != 0)
2419                 goto finish;
2420
2421         r = process_builtin(bus, m);
2422         if (r != 0)
2423                 goto finish;
2424
2425         r = bus_process_object(bus, m);
2426
2427 finish:
2428         bus->current_message = NULL;
2429         return r;
2430 }
2431
2432 static int dispatch_track(sd_bus *bus) {
2433         assert(bus);
2434
2435         if (!bus->track_queue)
2436                 return 0;
2437
2438         bus_track_dispatch(bus->track_queue);
2439         return 1;
2440 }
2441
2442 static int process_running(sd_bus *bus, bool hint_priority, int64_t priority, sd_bus_message **ret) {
2443         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
2444         int r;
2445
2446         assert(bus);
2447         assert(bus->state == BUS_RUNNING || bus->state == BUS_HELLO);
2448
2449         r = process_timeout(bus);
2450         if (r != 0)
2451                 goto null_message;
2452
2453         r = dispatch_wqueue(bus);
2454         if (r != 0)
2455                 goto null_message;
2456
2457         r = dispatch_track(bus);
2458         if (r != 0)
2459                 goto null_message;
2460
2461         r = dispatch_rqueue(bus, hint_priority, priority, &m);
2462         if (r < 0)
2463                 return r;
2464         if (!m)
2465                 goto null_message;
2466
2467         r = process_message(bus, m);
2468         if (r != 0)
2469                 goto null_message;
2470
2471         if (ret) {
2472                 r = sd_bus_message_rewind(m, true);
2473                 if (r < 0)
2474                         return r;
2475
2476                 *ret = m;
2477                 m = NULL;
2478                 return 1;
2479         }
2480
2481         if (m->header->type == SD_BUS_MESSAGE_METHOD_CALL) {
2482
2483                 log_debug("Unprocessed message call sender=%s object=%s interface=%s member=%s",
2484                           strna(sd_bus_message_get_sender(m)),
2485                           strna(sd_bus_message_get_path(m)),
2486                           strna(sd_bus_message_get_interface(m)),
2487                           strna(sd_bus_message_get_member(m)));
2488
2489                 r = sd_bus_reply_method_errorf(
2490                                 m,
2491                                 SD_BUS_ERROR_UNKNOWN_OBJECT,
2492                                 "Unknown object '%s'.", m->path);
2493                 if (r < 0)
2494                         return r;
2495         }
2496
2497         return 1;
2498
2499 null_message:
2500         if (r >= 0 && ret)
2501                 *ret = NULL;
2502
2503         return r;
2504 }
2505
2506 static int bus_exit_now(sd_bus *bus) {
2507         assert(bus);
2508
2509         /* Exit due to close, if this is requested. If this is bus object is attached to an event source, invokes
2510          * sd_event_exit(), otherwise invokes libc exit(). */
2511
2512         if (bus->exited) /* did we already exit? */
2513                 return 0;
2514         if (!bus->exit_triggered) /* was the exit condition triggered? */
2515                 return 0;
2516         if (!bus->exit_on_disconnect) /* Shall we actually exit on disconnection? */
2517                 return 0;
2518
2519         bus->exited = true; /* never exit more than once */
2520
2521         log_debug("Bus connection disconnected, exiting.");
2522
2523         if (bus->event)
2524                 return sd_event_exit(bus->event, EXIT_FAILURE);
2525         else
2526                 exit(EXIT_FAILURE);
2527
2528         assert_not_reached("exit() didn't exit?");
2529 }
2530
2531 static int process_closing_reply_callback(sd_bus *bus, struct reply_callback *c) {
2532         _cleanup_(sd_bus_error_free) sd_bus_error error_buffer = SD_BUS_ERROR_NULL;
2533         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
2534         sd_bus_slot *slot;
2535         int r;
2536
2537         assert(bus);
2538         assert(c);
2539
2540         r = bus_message_new_synthetic_error(
2541                         bus,
2542                         c->cookie,
2543                         &SD_BUS_ERROR_MAKE_CONST(SD_BUS_ERROR_NO_REPLY, "Connection terminated"),
2544                         &m);
2545         if (r < 0)
2546                 return r;
2547
2548         r = bus_seal_synthetic_message(bus, m);
2549         if (r < 0)
2550                 return r;
2551
2552         if (c->timeout != 0) {
2553                 prioq_remove(bus->reply_callbacks_prioq, c, &c->prioq_idx);
2554                 c->timeout = 0;
2555         }
2556
2557         ordered_hashmap_remove(bus->reply_callbacks, &c->cookie);
2558         c->cookie = 0;
2559
2560         slot = container_of(c, sd_bus_slot, reply_callback);
2561
2562         bus->iteration_counter++;
2563
2564         bus->current_message = m;
2565         bus->current_slot = sd_bus_slot_ref(slot);
2566         bus->current_handler = c->callback;
2567         bus->current_userdata = slot->userdata;
2568         r = c->callback(m, slot->userdata, &error_buffer);
2569         bus->current_userdata = NULL;
2570         bus->current_handler = NULL;
2571         bus->current_slot = NULL;
2572         bus->current_message = NULL;
2573
2574         if (slot->floating) {
2575                 bus_slot_disconnect(slot);
2576                 sd_bus_slot_unref(slot);
2577         }
2578
2579         sd_bus_slot_unref(slot);
2580
2581         return bus_maybe_reply_error(m, r, &error_buffer);
2582 }
2583
2584 static int process_closing(sd_bus *bus, sd_bus_message **ret) {
2585         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
2586         struct reply_callback *c;
2587         int r;
2588
2589         assert(bus);
2590         assert(bus->state == BUS_CLOSING);
2591
2592         /* First, fail all outstanding method calls */
2593         c = ordered_hashmap_first(bus->reply_callbacks);
2594         if (c)
2595                 return process_closing_reply_callback(bus, c);
2596
2597         /* Then, fake-drop all remaining bus tracking references */
2598         if (bus->tracks) {
2599                 bus_track_close(bus->tracks);
2600                 return 1;
2601         }
2602
2603         /* Then, synthesize a Disconnected message */
2604         r = sd_bus_message_new_signal(
2605                         bus,
2606                         &m,
2607                         "/org/freedesktop/DBus/Local",
2608                         "org.freedesktop.DBus.Local",
2609                         "Disconnected");
2610         if (r < 0)
2611                 return r;
2612
2613         bus_message_set_sender_local(bus, m);
2614
2615         r = bus_seal_synthetic_message(bus, m);
2616         if (r < 0)
2617                 return r;
2618
2619         sd_bus_close(bus);
2620
2621         bus->current_message = m;
2622         bus->iteration_counter++;
2623
2624         r = process_filter(bus, m);
2625         if (r != 0)
2626                 goto finish;
2627
2628         r = process_match(bus, m);
2629         if (r != 0)
2630                 goto finish;
2631
2632         /* Nothing else to do, exit now, if the condition holds */
2633         bus->exit_triggered = true;
2634         (void) bus_exit_now(bus);
2635
2636         if (ret) {
2637                 *ret = m;
2638                 m = NULL;
2639         }
2640
2641         r = 1;
2642
2643 finish:
2644         bus->current_message = NULL;
2645
2646         return r;
2647 }
2648
2649 static int bus_process_internal(sd_bus *bus, bool hint_priority, int64_t priority, sd_bus_message **ret) {
2650         BUS_DONT_DESTROY(bus);
2651         int r;
2652
2653         /* Returns 0 when we didn't do anything. This should cause the
2654          * caller to invoke sd_bus_wait() before returning the next
2655          * time. Returns > 0 when we did something, which possibly
2656          * means *ret is filled in with an unprocessed message. */
2657
2658         assert_return(bus, -EINVAL);
2659         assert_return(!bus_pid_changed(bus), -ECHILD);
2660
2661         /* We don't allow recursively invoking sd_bus_process(). */
2662         assert_return(!bus->current_message, -EBUSY);
2663         assert(!bus->current_slot);
2664
2665         switch (bus->state) {
2666
2667         case BUS_UNSET:
2668                 return -ENOTCONN;
2669
2670         case BUS_CLOSED:
2671                 return -ECONNRESET;
2672
2673         case BUS_OPENING:
2674                 r = bus_socket_process_opening(bus);
2675                 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2676                         bus_enter_closing(bus);
2677                         r = 1;
2678                 } else if (r < 0)
2679                         return r;
2680                 if (ret)
2681                         *ret = NULL;
2682                 return r;
2683
2684         case BUS_AUTHENTICATING:
2685                 r = bus_socket_process_authenticating(bus);
2686                 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2687                         bus_enter_closing(bus);
2688                         r = 1;
2689                 } else if (r < 0)
2690                         return r;
2691
2692                 if (ret)
2693                         *ret = NULL;
2694
2695                 return r;
2696
2697         case BUS_RUNNING:
2698         case BUS_HELLO:
2699                 r = process_running(bus, hint_priority, priority, ret);
2700                 if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2701                         bus_enter_closing(bus);
2702                         r = 1;
2703
2704                         if (ret)
2705                                 *ret = NULL;
2706                 }
2707
2708                 return r;
2709
2710         case BUS_CLOSING:
2711                 return process_closing(bus, ret);
2712         }
2713
2714         assert_not_reached("Unknown state");
2715 }
2716
2717 _public_ int sd_bus_process(sd_bus *bus, sd_bus_message **ret) {
2718         return bus_process_internal(bus, false, 0, ret);
2719 }
2720
2721 #if 0 /// UNNEEDED by elogind
2722 _public_ int sd_bus_process_priority(sd_bus *bus, int64_t priority, sd_bus_message **ret) {
2723         return bus_process_internal(bus, true, priority, ret);
2724 }
2725 #endif // 0
2726
2727 static int bus_poll(sd_bus *bus, bool need_more, uint64_t timeout_usec) {
2728         struct pollfd p[2] = {};
2729         int r, e, n;
2730         struct timespec ts;
2731         usec_t m = USEC_INFINITY;
2732
2733         assert(bus);
2734
2735         if (bus->state == BUS_CLOSING)
2736                 return 1;
2737
2738         if (!BUS_IS_OPEN(bus->state))
2739                 return -ENOTCONN;
2740
2741         e = sd_bus_get_events(bus);
2742         if (e < 0)
2743                 return e;
2744
2745         if (need_more)
2746                 /* The caller really needs some more data, he doesn't
2747                  * care about what's already read, or any timeouts
2748                  * except its own. */
2749                 e |= POLLIN;
2750         else {
2751                 usec_t until;
2752                 /* The caller wants to process if there's something to
2753                  * process, but doesn't care otherwise */
2754
2755                 r = sd_bus_get_timeout(bus, &until);
2756                 if (r < 0)
2757                         return r;
2758                 if (r > 0) {
2759                         usec_t nw;
2760                         nw = now(CLOCK_MONOTONIC);
2761                         m = until > nw ? until - nw : 0;
2762                 }
2763         }
2764
2765         if (timeout_usec != (uint64_t) -1 && (m == (uint64_t) -1 || timeout_usec < m))
2766                 m = timeout_usec;
2767
2768         p[0].fd = bus->input_fd;
2769         if (bus->output_fd == bus->input_fd) {
2770                 p[0].events = e;
2771                 n = 1;
2772         } else {
2773                 p[0].events = e & POLLIN;
2774                 p[1].fd = bus->output_fd;
2775                 p[1].events = e & POLLOUT;
2776                 n = 2;
2777         }
2778
2779         r = ppoll(p, n, m == (uint64_t) -1 ? NULL : timespec_store(&ts, m), NULL);
2780         if (r < 0)
2781                 return -errno;
2782
2783         return r > 0 ? 1 : 0;
2784 }
2785
2786 _public_ int sd_bus_wait(sd_bus *bus, uint64_t timeout_usec) {
2787
2788         assert_return(bus, -EINVAL);
2789         assert_return(!bus_pid_changed(bus), -ECHILD);
2790
2791         if (bus->state == BUS_CLOSING)
2792                 return 0;
2793
2794         if (!BUS_IS_OPEN(bus->state))
2795                 return -ENOTCONN;
2796
2797         if (bus->rqueue_size > 0)
2798                 return 0;
2799
2800         return bus_poll(bus, false, timeout_usec);
2801 }
2802
2803 _public_ int sd_bus_flush(sd_bus *bus) {
2804         int r;
2805
2806         assert_return(bus, -EINVAL);
2807         assert_return(!bus_pid_changed(bus), -ECHILD);
2808
2809         if (bus->state == BUS_CLOSING)
2810                 return 0;
2811
2812         if (!BUS_IS_OPEN(bus->state))
2813                 return -ENOTCONN;
2814
2815         r = bus_ensure_running(bus);
2816         if (r < 0)
2817                 return r;
2818
2819         if (bus->wqueue_size <= 0)
2820                 return 0;
2821
2822         for (;;) {
2823                 r = dispatch_wqueue(bus);
2824                 if (r < 0) {
2825                         if (IN_SET(r, -ENOTCONN, -ECONNRESET, -EPIPE, -ESHUTDOWN)) {
2826                                 bus_enter_closing(bus);
2827                                 return -ECONNRESET;
2828                         }
2829
2830                         return r;
2831                 }
2832
2833                 if (bus->wqueue_size <= 0)
2834                         return 0;
2835
2836                 r = bus_poll(bus, false, (uint64_t) -1);
2837                 if (r < 0)
2838                         return r;
2839         }
2840 }
2841
2842 #if 0 /// UNNEEDED by elogind
2843 _public_ int sd_bus_add_filter(
2844                 sd_bus *bus,
2845                 sd_bus_slot **slot,
2846                 sd_bus_message_handler_t callback,
2847                 void *userdata) {
2848
2849         sd_bus_slot *s;
2850
2851         assert_return(bus, -EINVAL);
2852         assert_return(callback, -EINVAL);
2853         assert_return(!bus_pid_changed(bus), -ECHILD);
2854
2855         s = bus_slot_allocate(bus, !slot, BUS_FILTER_CALLBACK, sizeof(struct filter_callback), userdata);
2856         if (!s)
2857                 return -ENOMEM;
2858
2859         s->filter_callback.callback = callback;
2860
2861         bus->filter_callbacks_modified = true;
2862         LIST_PREPEND(callbacks, bus->filter_callbacks, &s->filter_callback);
2863
2864         if (slot)
2865                 *slot = s;
2866
2867         return 0;
2868 }
2869 #endif // 0
2870
2871 _public_ int sd_bus_add_match(
2872                 sd_bus *bus,
2873                 sd_bus_slot **slot,
2874                 const char *match,
2875                 sd_bus_message_handler_t callback,
2876                 void *userdata) {
2877
2878         struct bus_match_component *components = NULL;
2879         unsigned n_components = 0;
2880         sd_bus_slot *s = NULL;
2881         int r = 0;
2882
2883         assert_return(bus, -EINVAL);
2884         assert_return(match, -EINVAL);
2885         assert_return(!bus_pid_changed(bus), -ECHILD);
2886
2887         r = bus_match_parse(match, &components, &n_components);
2888         if (r < 0)
2889                 goto finish;
2890
2891         s = bus_slot_allocate(bus, !slot, BUS_MATCH_CALLBACK, sizeof(struct match_callback), userdata);
2892         if (!s) {
2893                 r = -ENOMEM;
2894                 goto finish;
2895         }
2896
2897         s->match_callback.callback = callback;
2898         s->match_callback.cookie = ++bus->match_cookie;
2899
2900         if (bus->bus_client) {
2901                 enum bus_match_scope scope;
2902
2903                 scope = bus_match_get_scope(components, n_components);
2904
2905                 /* Do not install server-side matches for matches
2906                  * against the local service, interface or bus path. */
2907                 if (scope != BUS_MATCH_LOCAL) {
2908
2909                         /* We store the original match string, so that
2910                          * we can use it to remove the match again. */
2911
2912                         s->match_callback.match_string = strdup(match);
2913                         if (!s->match_callback.match_string) {
2914                                 r = -ENOMEM;
2915                                 goto finish;
2916                         }
2917
2918                         r = bus_add_match_internal(bus, s->match_callback.match_string, components, n_components, s->match_callback.cookie);
2919                         if (r < 0)
2920                                 goto finish;
2921
2922                         s->match_added = true;
2923                 }
2924         }
2925
2926         bus->match_callbacks_modified = true;
2927         r = bus_match_add(&bus->match_callbacks, components, n_components, &s->match_callback);
2928         if (r < 0)
2929                 goto finish;
2930
2931         if (slot)
2932                 *slot = s;
2933         s = NULL;
2934
2935 finish:
2936         bus_match_parse_free(components, n_components);
2937         sd_bus_slot_unref(s);
2938
2939         return r;
2940 }
2941
2942 #if 0 /// UNNEEDED by elogind
2943 int bus_remove_match_by_string(
2944                 sd_bus *bus,
2945                 const char *match,
2946                 sd_bus_message_handler_t callback,
2947                 void *userdata) {
2948
2949         struct bus_match_component *components = NULL;
2950         unsigned n_components = 0;
2951         struct match_callback *c;
2952         int r = 0;
2953
2954         assert_return(bus, -EINVAL);
2955         assert_return(match, -EINVAL);
2956         assert_return(!bus_pid_changed(bus), -ECHILD);
2957
2958         r = bus_match_parse(match, &components, &n_components);
2959         if (r < 0)
2960                 goto finish;
2961
2962         r = bus_match_find(&bus->match_callbacks, components, n_components, NULL, NULL, &c);
2963         if (r <= 0)
2964                 goto finish;
2965
2966         sd_bus_slot_unref(container_of(c, sd_bus_slot, match_callback));
2967
2968 finish:
2969         bus_match_parse_free(components, n_components);
2970
2971         return r;
2972 }
2973 #endif // 0
2974
2975 bool bus_pid_changed(sd_bus *bus) {
2976         assert(bus);
2977
2978         /* We don't support people creating a bus connection and
2979          * keeping it around over a fork(). Let's complain. */
2980
2981         return bus->original_pid != getpid_cached();
2982 }
2983
2984 static int io_callback(sd_event_source *s, int fd, uint32_t revents, void *userdata) {
2985         sd_bus *bus = userdata;
2986         int r;
2987
2988         assert(bus);
2989
2990         r = sd_bus_process(bus, NULL);
2991         if (r < 0)
2992                 return r;
2993
2994         return 1;
2995 }
2996
2997 static int time_callback(sd_event_source *s, uint64_t usec, void *userdata) {
2998         sd_bus *bus = userdata;
2999         int r;
3000
3001         assert(bus);
3002
3003         r = sd_bus_process(bus, NULL);
3004         if (r < 0)
3005                 return r;
3006
3007         return 1;
3008 }
3009
3010 static int prepare_callback(sd_event_source *s, void *userdata) {
3011         sd_bus *bus = userdata;
3012         int r, e;
3013         usec_t until;
3014
3015         assert(s);
3016         assert(bus);
3017
3018         e = sd_bus_get_events(bus);
3019         if (e < 0)
3020                 return e;
3021
3022         if (bus->output_fd != bus->input_fd) {
3023
3024                 r = sd_event_source_set_io_events(bus->input_io_event_source, e & POLLIN);
3025                 if (r < 0)
3026                         return r;
3027
3028                 r = sd_event_source_set_io_events(bus->output_io_event_source, e & POLLOUT);
3029                 if (r < 0)
3030                         return r;
3031         } else {
3032                 r = sd_event_source_set_io_events(bus->input_io_event_source, e);
3033                 if (r < 0)
3034                         return r;
3035         }
3036
3037         r = sd_bus_get_timeout(bus, &until);
3038         if (r < 0)
3039                 return r;
3040         if (r > 0) {
3041                 int j;
3042
3043                 j = sd_event_source_set_time(bus->time_event_source, until);
3044                 if (j < 0)
3045                         return j;
3046         }
3047
3048         r = sd_event_source_set_enabled(bus->time_event_source, r > 0);
3049         if (r < 0)
3050                 return r;
3051
3052         return 1;
3053 }
3054
3055 static int quit_callback(sd_event_source *event, void *userdata) {
3056         sd_bus *bus = userdata;
3057
3058         assert(event);
3059
3060         sd_bus_flush(bus);
3061         sd_bus_close(bus);
3062
3063         return 1;
3064 }
3065
3066 static int attach_io_events(sd_bus *bus) {
3067         int r;
3068
3069         assert(bus);
3070
3071         if (bus->input_fd < 0)
3072                 return 0;
3073
3074         if (!bus->event)
3075                 return 0;
3076
3077         if (!bus->input_io_event_source) {
3078                 r = sd_event_add_io(bus->event, &bus->input_io_event_source, bus->input_fd, 0, io_callback, bus);
3079                 if (r < 0)
3080                         return r;
3081
3082                 r = sd_event_source_set_prepare(bus->input_io_event_source, prepare_callback);
3083                 if (r < 0)
3084                         return r;
3085
3086                 r = sd_event_source_set_priority(bus->input_io_event_source, bus->event_priority);
3087                 if (r < 0)
3088                         return r;
3089
3090                 r = sd_event_source_set_description(bus->input_io_event_source, "bus-input");
3091         } else
3092                 r = sd_event_source_set_io_fd(bus->input_io_event_source, bus->input_fd);
3093
3094         if (r < 0)
3095                 return r;
3096
3097         if (bus->output_fd != bus->input_fd) {
3098                 assert(bus->output_fd >= 0);
3099
3100                 if (!bus->output_io_event_source) {
3101                         r = sd_event_add_io(bus->event, &bus->output_io_event_source, bus->output_fd, 0, io_callback, bus);
3102                         if (r < 0)
3103                                 return r;
3104
3105                         r = sd_event_source_set_priority(bus->output_io_event_source, bus->event_priority);
3106                         if (r < 0)
3107                                 return r;
3108
3109                         r = sd_event_source_set_description(bus->input_io_event_source, "bus-output");
3110                 } else
3111                         r = sd_event_source_set_io_fd(bus->output_io_event_source, bus->output_fd);
3112
3113                 if (r < 0)
3114                         return r;
3115         }
3116
3117         return 0;
3118 }
3119
3120 static void detach_io_events(sd_bus *bus) {
3121         assert(bus);
3122
3123         if (bus->input_io_event_source) {
3124                 sd_event_source_set_enabled(bus->input_io_event_source, SD_EVENT_OFF);
3125                 bus->input_io_event_source = sd_event_source_unref(bus->input_io_event_source);
3126         }
3127
3128         if (bus->output_io_event_source) {
3129                 sd_event_source_set_enabled(bus->output_io_event_source, SD_EVENT_OFF);
3130                 bus->output_io_event_source = sd_event_source_unref(bus->output_io_event_source);
3131         }
3132 }
3133
3134 _public_ int sd_bus_attach_event(sd_bus *bus, sd_event *event, int priority) {
3135         int r;
3136
3137         assert_return(bus, -EINVAL);
3138         assert_return(!bus->event, -EBUSY);
3139
3140         assert(!bus->input_io_event_source);
3141         assert(!bus->output_io_event_source);
3142         assert(!bus->time_event_source);
3143
3144         if (event)
3145                 bus->event = sd_event_ref(event);
3146         else  {
3147                 r = sd_event_default(&bus->event);
3148                 if (r < 0)
3149                         return r;
3150         }
3151
3152         bus->event_priority = priority;
3153
3154         r = sd_event_add_time(bus->event, &bus->time_event_source, CLOCK_MONOTONIC, 0, 0, time_callback, bus);
3155         if (r < 0)
3156                 goto fail;
3157
3158         r = sd_event_source_set_priority(bus->time_event_source, priority);
3159         if (r < 0)
3160                 goto fail;
3161
3162         r = sd_event_source_set_description(bus->time_event_source, "bus-time");
3163         if (r < 0)
3164                 goto fail;
3165
3166         r = sd_event_add_exit(bus->event, &bus->quit_event_source, quit_callback, bus);
3167         if (r < 0)
3168                 goto fail;
3169
3170         r = sd_event_source_set_description(bus->quit_event_source, "bus-exit");
3171         if (r < 0)
3172                 goto fail;
3173
3174         r = attach_io_events(bus);
3175         if (r < 0)
3176                 goto fail;
3177
3178         return 0;
3179
3180 fail:
3181         sd_bus_detach_event(bus);
3182         return r;
3183 }
3184
3185 _public_ int sd_bus_detach_event(sd_bus *bus) {
3186         assert_return(bus, -EINVAL);
3187
3188         if (!bus->event)
3189                 return 0;
3190
3191         detach_io_events(bus);
3192
3193         if (bus->time_event_source) {
3194                 sd_event_source_set_enabled(bus->time_event_source, SD_EVENT_OFF);
3195                 bus->time_event_source = sd_event_source_unref(bus->time_event_source);
3196         }
3197
3198         if (bus->quit_event_source) {
3199                 sd_event_source_set_enabled(bus->quit_event_source, SD_EVENT_OFF);
3200                 bus->quit_event_source = sd_event_source_unref(bus->quit_event_source);
3201         }
3202
3203         bus->event = sd_event_unref(bus->event);
3204         return 1;
3205 }
3206
3207 _public_ sd_event* sd_bus_get_event(sd_bus *bus) {
3208         assert_return(bus, NULL);
3209
3210         return bus->event;
3211 }
3212
3213 _public_ sd_bus_message* sd_bus_get_current_message(sd_bus *bus) {
3214         assert_return(bus, NULL);
3215
3216         return bus->current_message;
3217 }
3218
3219 #if 0 /// UNNEEDED by elogind
3220 _public_ sd_bus_slot* sd_bus_get_current_slot(sd_bus *bus) {
3221         assert_return(bus, NULL);
3222
3223         return bus->current_slot;
3224 }
3225 #endif // 0
3226
3227 _public_ sd_bus_message_handler_t sd_bus_get_current_handler(sd_bus *bus) {
3228         assert_return(bus, NULL);
3229
3230         return bus->current_handler;
3231 }
3232
3233 _public_ void* sd_bus_get_current_userdata(sd_bus *bus) {
3234         assert_return(bus, NULL);
3235
3236         return bus->current_userdata;
3237 }
3238
3239 static int bus_default(int (*bus_open)(sd_bus **), sd_bus **default_bus, sd_bus **ret) {
3240         sd_bus *b = NULL;
3241         int r;
3242
3243         assert(bus_open);
3244         assert(default_bus);
3245
3246         if (!ret)
3247                 return !!*default_bus;
3248
3249         if (*default_bus) {
3250                 *ret = sd_bus_ref(*default_bus);
3251                 return 0;
3252         }
3253
3254         r = bus_open(&b);
3255         if (r < 0)
3256                 return r;
3257
3258         b->default_bus_ptr = default_bus;
3259         b->tid = gettid();
3260         *default_bus = b;
3261
3262         *ret = b;
3263         return 1;
3264 }
3265
3266 _public_ int sd_bus_default_system(sd_bus **ret) {
3267         return bus_default(sd_bus_open_system, &default_system_bus, ret);
3268 }
3269
3270
3271 _public_ int sd_bus_default_user(sd_bus **ret) {
3272 #if 0 /// elogind does not support user buses
3273         return bus_default(sd_bus_open_user, &default_user_bus, ret);
3274 #else
3275         return sd_bus_default_system(ret);
3276 #endif // 0
3277 }
3278
3279 _public_ int sd_bus_default(sd_bus **ret) {
3280
3281         const char *e;
3282
3283         /* Let's try our best to reuse another cached connection. If
3284          * the starter bus type is set, connect via our normal
3285          * connection logic, ignoring $DBUS_STARTER_ADDRESS, so that
3286          * we can share the connection with the user/system default
3287          * bus. */
3288
3289         e = secure_getenv("DBUS_STARTER_BUS_TYPE");
3290         if (e) {
3291                 if (streq(e, "system"))
3292                         return sd_bus_default_system(ret);
3293 #if 0 /// elogind does not support systemd units
3294                 else if (STR_IN_SET(e, "user", "session"))
3295                         return sd_bus_default_user(ret);
3296 #endif // 0
3297         }
3298
3299         /* No type is specified, so we have not other option than to
3300          * use the starter address if it is set. */
3301
3302         e = secure_getenv("DBUS_STARTER_ADDRESS");
3303         if (e) {
3304
3305                 return bus_default(sd_bus_open, &default_starter_bus, ret);
3306         }
3307
3308         /* Finally, if nothing is set use the cached connection for
3309          * the right scope */
3310
3311 #if 0 /// elogind does not support systemd units
3312         if (cg_pid_get_owner_uid(0, NULL) >= 0)
3313                 return sd_bus_default_user(ret);
3314         else
3315 #endif // 0
3316                 return sd_bus_default_system(ret);
3317 }
3318
3319 #if 0 /// UNNEEDED by elogind
3320 _public_ int sd_bus_get_tid(sd_bus *b, pid_t *tid) {
3321         assert_return(b, -EINVAL);
3322         assert_return(tid, -EINVAL);
3323         assert_return(!bus_pid_changed(b), -ECHILD);
3324
3325         if (b->tid != 0) {
3326                 *tid = b->tid;
3327                 return 0;
3328         }
3329
3330         if (b->event)
3331                 return sd_event_get_tid(b->event, tid);
3332
3333         return -ENXIO;
3334 }
3335
3336 _public_ int sd_bus_path_encode(const char *prefix, const char *external_id, char **ret_path) {
3337         _cleanup_free_ char *e = NULL;
3338         char *ret;
3339
3340         assert_return(object_path_is_valid(prefix), -EINVAL);
3341         assert_return(external_id, -EINVAL);
3342         assert_return(ret_path, -EINVAL);
3343
3344         e = bus_label_escape(external_id);
3345         if (!e)
3346                 return -ENOMEM;
3347
3348         ret = strjoin(prefix, "/", e);
3349         if (!ret)
3350                 return -ENOMEM;
3351
3352         *ret_path = ret;
3353         return 0;
3354 }
3355
3356 _public_ int sd_bus_path_decode(const char *path, const char *prefix, char **external_id) {
3357         const char *e;
3358         char *ret;
3359
3360         assert_return(object_path_is_valid(path), -EINVAL);
3361         assert_return(object_path_is_valid(prefix), -EINVAL);
3362         assert_return(external_id, -EINVAL);
3363
3364         e = object_path_startswith(path, prefix);
3365         if (!e) {
3366                 *external_id = NULL;
3367                 return 0;
3368         }
3369
3370         ret = bus_label_unescape(e);
3371         if (!ret)
3372                 return -ENOMEM;
3373
3374         *external_id = ret;
3375         return 1;
3376 }
3377
3378 _public_ int sd_bus_path_encode_many(char **out, const char *path_template, ...) {
3379         _cleanup_strv_free_ char **labels = NULL;
3380         char *path, *path_pos, **label_pos;
3381         const char *sep, *template_pos;
3382         size_t path_length;
3383         va_list list;
3384         int r;
3385
3386         assert_return(out, -EINVAL);
3387         assert_return(path_template, -EINVAL);
3388
3389         path_length = strlen(path_template);
3390
3391         va_start(list, path_template);
3392         for (sep = strchr(path_template, '%'); sep; sep = strchr(sep + 1, '%')) {
3393                 const char *arg;
3394                 char *label;
3395
3396                 arg = va_arg(list, const char *);
3397                 if (!arg) {
3398                         va_end(list);
3399                         return -EINVAL;
3400                 }
3401
3402                 label = bus_label_escape(arg);
3403                 if (!label) {
3404                         va_end(list);
3405                         return -ENOMEM;
3406                 }
3407
3408                 r = strv_consume(&labels, label);
3409                 if (r < 0) {
3410                         va_end(list);
3411                         return r;
3412                 }
3413
3414                 /* add label length, but account for the format character */
3415                 path_length += strlen(label) - 1;
3416         }
3417         va_end(list);
3418
3419         path = malloc(path_length + 1);
3420         if (!path)
3421                 return -ENOMEM;
3422
3423         path_pos = path;
3424         label_pos = labels;
3425
3426         for (template_pos = path_template; *template_pos; ) {
3427                 sep = strchrnul(template_pos, '%');
3428                 path_pos = mempcpy(path_pos, template_pos, sep - template_pos);
3429                 if (!*sep)
3430                         break;
3431
3432                 path_pos = stpcpy(path_pos, *label_pos++);
3433                 template_pos = sep + 1;
3434         }
3435
3436         *path_pos = 0;
3437         *out = path;
3438         return 0;
3439 }
3440
3441 _public_ int sd_bus_path_decode_many(const char *path, const char *path_template, ...) {
3442         _cleanup_strv_free_ char **labels = NULL;
3443         const char *template_pos, *path_pos;
3444         char **label_pos;
3445         va_list list;
3446         int r;
3447
3448         /*
3449          * This decodes an object-path based on a template argument. The
3450          * template consists of a verbatim path, optionally including special
3451          * directives:
3452          *
3453          *   - Each occurrence of '%' in the template matches an arbitrary
3454          *     substring of a label in the given path. At most one such
3455          *     directive is allowed per label. For each such directive, the
3456          *     caller must provide an output parameter (char **) via va_arg. If
3457          *     NULL is passed, the given label is verified, but not returned.
3458          *     For each matched label, the *decoded* label is stored in the
3459          *     passed output argument, and the caller is responsible to free
3460          *     it. Note that the output arguments are only modified if the
3461          *     actualy path matched the template. Otherwise, they're left
3462          *     untouched.
3463          *
3464          * This function returns <0 on error, 0 if the path does not match the
3465          * template, 1 if it matched.
3466          */
3467
3468         assert_return(path, -EINVAL);
3469         assert_return(path_template, -EINVAL);
3470
3471         path_pos = path;
3472
3473         for (template_pos = path_template; *template_pos; ) {
3474                 const char *sep;
3475                 size_t length;
3476                 char *label;
3477
3478                 /* verify everything until the next '%' matches verbatim */
3479                 sep = strchrnul(template_pos, '%');
3480                 length = sep - template_pos;
3481                 if (strncmp(path_pos, template_pos, length))
3482                         return 0;
3483
3484                 path_pos += length;
3485                 template_pos += length;
3486
3487                 if (!*template_pos)
3488                         break;
3489
3490                 /* We found the next '%' character. Everything up until here
3491                  * matched. We now skip ahead to the end of this label and make
3492                  * sure it matches the tail of the label in the path. Then we
3493                  * decode the string in-between and save it for later use. */
3494
3495                 ++template_pos; /* skip over '%' */
3496
3497                 sep = strchrnul(template_pos, '/');
3498                 length = sep - template_pos; /* length of suffix to match verbatim */
3499
3500                 /* verify the suffixes match */
3501                 sep = strchrnul(path_pos, '/');
3502                 if (sep - path_pos < (ssize_t)length ||
3503                     strncmp(sep - length, template_pos, length))
3504                         return 0;
3505
3506                 template_pos += length; /* skip over matched label */
3507                 length = sep - path_pos - length; /* length of sub-label to decode */
3508
3509                 /* store unescaped label for later use */
3510                 label = bus_label_unescape_n(path_pos, length);
3511                 if (!label)
3512                         return -ENOMEM;
3513
3514                 r = strv_consume(&labels, label);
3515                 if (r < 0)
3516                         return r;
3517
3518                 path_pos = sep; /* skip decoded label and suffix */
3519         }
3520
3521         /* end of template must match end of path */
3522         if (*path_pos)
3523                 return 0;
3524
3525         /* copy the labels over to the caller */
3526         va_start(list, path_template);
3527         for (label_pos = labels; label_pos && *label_pos; ++label_pos) {
3528                 char **arg;
3529
3530                 arg = va_arg(list, char **);
3531                 if (arg)
3532                         *arg = *label_pos;
3533                 else
3534                         free(*label_pos);
3535         }
3536         va_end(list);
3537
3538         free(labels);
3539         labels = NULL;
3540         return 1;
3541 }
3542
3543 _public_ int sd_bus_try_close(sd_bus *bus) {
3544         assert_return(bus, -EINVAL);
3545         assert_return(!bus_pid_changed(bus), -ECHILD);
3546
3547         return -EOPNOTSUPP;
3548 }
3549
3550 _public_ int sd_bus_get_description(sd_bus *bus, const char **description) {
3551         assert_return(bus, -EINVAL);
3552         assert_return(description, -EINVAL);
3553         assert_return(bus->description, -ENXIO);
3554         assert_return(!bus_pid_changed(bus), -ECHILD);
3555
3556         *description = bus->description;
3557         return 0;
3558 }
3559 #endif // 0
3560
3561 int bus_get_root_path(sd_bus *bus) {
3562         int r;
3563
3564         if (bus->cgroup_root)
3565                 return 0;
3566
3567         r = cg_get_root_path(&bus->cgroup_root);
3568         if (r == -ENOENT) {
3569                 bus->cgroup_root = strdup("/");
3570                 if (!bus->cgroup_root)
3571                         return -ENOMEM;
3572
3573                 r = 0;
3574         }
3575
3576         return r;
3577 }
3578
3579 #if 0 /// UNNEEDED by elogind
3580 _public_ int sd_bus_get_scope(sd_bus *bus, const char **scope) {
3581         assert_return(bus, -EINVAL);
3582         assert_return(scope, -EINVAL);
3583         assert_return(!bus_pid_changed(bus), -ECHILD);
3584
3585         if (bus->is_user) {
3586                 *scope = "user";
3587                 return 0;
3588         }
3589
3590         if (bus->is_system) {
3591                 *scope = "system";
3592                 return 0;
3593         }
3594
3595         return -ENODATA;
3596 }
3597
3598 _public_ int sd_bus_get_address(sd_bus *bus, const char **address) {
3599
3600         assert_return(bus, -EINVAL);
3601         assert_return(address, -EINVAL);
3602         assert_return(!bus_pid_changed(bus), -ECHILD);
3603
3604         if (bus->address) {
3605                 *address = bus->address;
3606                 return 0;
3607         }
3608
3609         return -ENODATA;
3610 }
3611
3612 _public_ int sd_bus_get_creds_mask(sd_bus *bus, uint64_t *mask) {
3613         assert_return(bus, -EINVAL);
3614         assert_return(mask, -EINVAL);
3615         assert_return(!bus_pid_changed(bus), -ECHILD);
3616
3617         *mask = bus->creds_mask;
3618         return 0;
3619 }
3620
3621 _public_ int sd_bus_is_bus_client(sd_bus *bus) {
3622         assert_return(bus, -EINVAL);
3623         assert_return(!bus_pid_changed(bus), -ECHILD);
3624
3625         return bus->bus_client;
3626 }
3627
3628 _public_ int sd_bus_is_server(sd_bus *bus) {
3629         assert_return(bus, -EINVAL);
3630         assert_return(!bus_pid_changed(bus), -ECHILD);
3631
3632         return bus->is_server;
3633 }
3634
3635 _public_ int sd_bus_is_anonymous(sd_bus *bus) {
3636         assert_return(bus, -EINVAL);
3637         assert_return(!bus_pid_changed(bus), -ECHILD);
3638
3639         return bus->anonymous_auth;
3640 }
3641
3642 _public_ int sd_bus_is_trusted(sd_bus *bus) {
3643         assert_return(bus, -EINVAL);
3644         assert_return(!bus_pid_changed(bus), -ECHILD);
3645
3646         return bus->trusted;
3647 }
3648
3649 _public_ int sd_bus_is_monitor(sd_bus *bus) {
3650         assert_return(bus, -EINVAL);
3651         assert_return(!bus_pid_changed(bus), -ECHILD);
3652
3653         return !!(bus->hello_flags & KDBUS_HELLO_MONITOR);
3654 }
3655
3656 static void flush_close(sd_bus *bus) {
3657         if (!bus)
3658                 return;
3659
3660         /* Flushes and closes the specified bus. We take a ref before,
3661          * to ensure the flushing does not cause the bus to be
3662          * unreferenced. */
3663
3664         sd_bus_flush_close_unref(sd_bus_ref(bus));
3665 }
3666
3667 _public_ void sd_bus_default_flush_close(void) {
3668         flush_close(default_starter_bus);
3669         flush_close(default_user_bus);
3670         flush_close(default_system_bus);
3671 }
3672 #endif // 0
3673
3674 _public_ int sd_bus_set_exit_on_disconnect(sd_bus *bus, int b) {
3675         assert_return(bus, -EINVAL);
3676
3677         /* Turns on exit-on-disconnect, and triggers it immediately if the bus connection was already
3678          * disconnected. Note that this is triggered exclusively on disconnections triggered by the server side, never
3679          * from the client side. */
3680         bus->exit_on_disconnect = b;
3681
3682         /* If the exit condition was triggered already, exit immediately. */
3683         return bus_exit_now(bus);
3684 }
3685
3686 #if 0 /// UNNEEDED by elogind
3687 _public_ int sd_bus_get_exit_on_disconnect(sd_bus *bus) {
3688         assert_return(bus, -EINVAL);
3689
3690         return bus->exit_on_disconnect;
3691 }
3692 #endif // 0