src/login/logind-dbus.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2011 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 <errno.h>
  21 #include <pwd.h>
  22 #include <string.h>
  23 #include <unistd.h>
  24
  25 #include "sd-messages.h"
  26
  27 #include "alloc-util.h"
  28 #include "audit-util.h"
  29 #include "bus-common-errors.h"
  30 #include "bus-error.h"
  31 #include "bus-util.h"
  32 #include "dirent-util.h"
  33 #include "efivars.h"
  34 #include "escape.h"
  35 #include "fd-util.h"
  36 #include "fileio-label.h"
  37 #include "format-util.h"
  38 #include "fs-util.h"
  39 #include "logind.h"
  40 #include "mkdir.h"
  41 #include "path-util.h"
  42 #include "process-util.h"
  43 #include "selinux-util.h"
  44 #include "sleep-config.h"
  45 #include "special.h"
  46 #include "strv.h"
  47 #include "terminal-util.h"
  48 #include "udev-util.h"
  49 #include "unit-name.h"
  50 #include "user-util.h"
  51 #include "utmp-wtmp.h"
  52
  53 int manager_get_session_from_creds(Manager *m, sd_bus_message *message, const char *name, sd_bus_error *error, Session **ret) {
  54         _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL;
  55         Session *session;
  56         int r;
  57
  58         assert(m);
  59         assert(message);
  60         assert(ret);
  61
  62         if (isempty(name)) {
  63                 r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_SESSION|SD_BUS_CREDS_AUGMENT, &creds);
  64                 if (r < 0)
  65                         return r;
  66
  67                 r = sd_bus_creds_get_session(creds, &name);
  68                 if (r == -ENXIO)
  69                         return sd_bus_error_setf(error, BUS_ERROR_NO_SESSION_FOR_PID,
  70                                                  "Caller does not belong to any known session");
  71                 if (r < 0)
  72                         return r;
  73         }
  74
  75         session = hashmap_get(m->sessions, name);
  76         if (!session)
  77                 return sd_bus_error_setf(error, BUS_ERROR_NO_SUCH_SESSION, "No session '%s' known", name);
  78
  79         *ret = session;
  80         return 0;
  81 }
  82
  83 int manager_get_user_from_creds(Manager *m, sd_bus_message *message, uid_t uid, sd_bus_error *error, User **ret) {
  84         User *user;
  85         int r;
  86
  87         assert(m);
  88         assert(message);
  89         assert(ret);
  90
  91         if (!uid_is_valid(uid)) {
  92                 _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL;
  93
  94                 /* Note that we get the owner UID of the session, not the actual client UID here! */
  95                 r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_OWNER_UID|SD_BUS_CREDS_AUGMENT, &creds);
  96                 if (r < 0)
  97                         return r;
  98
  99                 r = sd_bus_creds_get_owner_uid(creds, &uid);
 100                 if (r < 0)
 101                         return r;
 102         }
 103
 104         user = hashmap_get(m->users, UID_TO_PTR(uid));
 105         if (!user)
 106                 return sd_bus_error_setf(error, BUS_ERROR_NO_SUCH_USER, "No user "UID_FMT" known or logged in", uid);
 107
 108         *ret = user;
 109         return 0;
 110 }
 111
 112 int manager_get_seat_from_creds(Manager *m, sd_bus_message *message, const char *name, sd_bus_error *error, Seat **ret) {
 113         Seat *seat;
 114         int r;
 115
 116         assert(m);
 117         assert(message);
 118         assert(ret);
 119
 120         if (isempty(name)) {
 121                 Session *session;
 122
 123                 r = manager_get_session_from_creds(m, message, NULL, error, &session);
 124                 if (r < 0)
 125                         return r;
 126
 127                 seat = session->seat;
 128                 if (!seat)
 129                         return sd_bus_error_setf(error, BUS_ERROR_NO_SUCH_SEAT, "Session has no seat.");
 130         } else {
 131                 seat = hashmap_get(m->seats, name);
 132                 if (!seat)
 133                         return sd_bus_error_setf(error, BUS_ERROR_NO_SUCH_SEAT, "No seat '%s' known", name);
 134         }
 135
 136         *ret = seat;
 137         return 0;
 138 }
 139
 140 static int property_get_idle_hint(
 141                 sd_bus *bus,
 142                 const char *path,
 143                 const char *interface,
 144                 const char *property,
 145                 sd_bus_message *reply,
 146                 void *userdata,
 147                 sd_bus_error *error) {
 148
 149         Manager *m = userdata;
 150
 151         assert(bus);
 152         assert(reply);
 153         assert(m);
 154
 155         return sd_bus_message_append(reply, "b", manager_get_idle_hint(m, NULL) > 0);
 156 }
 157
 158 static int property_get_idle_since_hint(
 159                 sd_bus *bus,
 160                 const char *path,
 161                 const char *interface,
 162                 const char *property,
 163                 sd_bus_message *reply,
 164                 void *userdata,
 165                 sd_bus_error *error) {
 166
 167         Manager *m = userdata;
 168         dual_timestamp t = DUAL_TIMESTAMP_NULL;
 169
 170         assert(bus);
 171         assert(reply);
 172         assert(m);
 173
 174         manager_get_idle_hint(m, &t);
 175
 176         return sd_bus_message_append(reply, "t", streq(property, "IdleSinceHint") ? t.realtime : t.monotonic);
 177 }
 178
 179 static int property_get_inhibited(
 180                 sd_bus *bus,
 181                 const char *path,
 182                 const char *interface,
 183                 const char *property,
 184                 sd_bus_message *reply,
 185                 void *userdata,
 186                 sd_bus_error *error) {
 187
 188         Manager *m = userdata;
 189         InhibitWhat w;
 190
 191         assert(bus);
 192         assert(reply);
 193         assert(m);
 194
 195         w = manager_inhibit_what(m, streq(property, "BlockInhibited") ? INHIBIT_BLOCK : INHIBIT_DELAY);
 196
 197         return sd_bus_message_append(reply, "s", inhibit_what_to_string(w));
 198 }
 199
 200 static int property_get_preparing(
 201                 sd_bus *bus,
 202                 const char *path,
 203                 const char *interface,
 204                 const char *property,
 205                 sd_bus_message *reply,
 206                 void *userdata,
 207                 sd_bus_error *error) {
 208
 209         Manager *m = userdata;
 210         bool b;
 211
 212         assert(bus);
 213         assert(reply);
 214         assert(m);
 215
 216         if (streq(property, "PreparingForShutdown"))
 217                 b = !!(m->action_what & INHIBIT_SHUTDOWN);
 218         else
 219                 b = !!(m->action_what & INHIBIT_SLEEP);
 220
 221         return sd_bus_message_append(reply, "b", b);
 222 }
 223
 224 static int property_get_scheduled_shutdown(
 225                 sd_bus *bus,
 226                 const char *path,
 227                 const char *interface,
 228                 const char *property,
 229                 sd_bus_message *reply,
 230                 void *userdata,
 231                 sd_bus_error *error) {
 232
 233         Manager *m = userdata;
 234         int r;
 235
 236         assert(bus);
 237         assert(reply);
 238         assert(m);
 239
 240         r = sd_bus_message_open_container(reply, 'r', "st");
 241         if (r < 0)
 242                 return r;
 243
 244         r = sd_bus_message_append(reply, "st", m->scheduled_shutdown_type, m->scheduled_shutdown_timeout);
 245         if (r < 0)
 246                 return r;
 247
 248         return sd_bus_message_close_container(reply);
 249 }
 250
 251 static BUS_DEFINE_PROPERTY_GET_ENUM(property_get_handle_action, handle_action, HandleAction);
 252
 253 static int property_get_docked(
 254                 sd_bus *bus,
 255                 const char *path,
 256                 const char *interface,
 257                 const char *property,
 258                 sd_bus_message *reply,
 259                 void *userdata,
 260                 sd_bus_error *error) {
 261
 262         Manager *m = userdata;
 263
 264         assert(bus);
 265         assert(reply);
 266         assert(m);
 267
 268         return sd_bus_message_append(reply, "b", manager_is_docked_or_external_displays(m));
 269 }
 270
 271 static int property_get_current_sessions(
 272                 sd_bus *bus,
 273                 const char *path,
 274                 const char *interface,
 275                 const char *property,
 276                 sd_bus_message *reply,
 277                 void *userdata,
 278                 sd_bus_error *error) {
 279
 280         Manager *m = userdata;
 281
 282         assert(bus);
 283         assert(reply);
 284         assert(m);
 285
 286         return sd_bus_message_append(reply, "t", (uint64_t) hashmap_size(m->sessions));
 287 }
 288
 289 static int property_get_current_inhibitors(
 290                 sd_bus *bus,
 291                 const char *path,
 292                 const char *interface,
 293                 const char *property,
 294                 sd_bus_message *reply,
 295                 void *userdata,
 296                 sd_bus_error *error) {
 297
 298         Manager *m = userdata;
 299
 300         assert(bus);
 301         assert(reply);
 302         assert(m);
 303
 304         return sd_bus_message_append(reply, "t", (uint64_t) hashmap_size(m->inhibitors));
 305 }
 306
 307 static int method_get_session(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 308         _cleanup_free_ char *p = NULL;
 309         Manager *m = userdata;
 310         const char *name;
 311         Session *session;
 312         int r;
 313
 314         assert(message);
 315         assert(m);
 316
 317         r = sd_bus_message_read(message, "s", &name);
 318         if (r < 0)
 319                 return r;
 320
 321         r = manager_get_session_from_creds(m, message, name, error, &session);
 322         if (r < 0)
 323                 return r;
 324
 325         p = session_bus_path(session);
 326         if (!p)
 327                 return -ENOMEM;
 328
 329         return sd_bus_reply_method_return(message, "o", p);
 330 }
 331
 332 static int method_get_session_by_pid(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 333         _cleanup_free_ char *p = NULL;
 334         Session *session = NULL;
 335         Manager *m = userdata;
 336         pid_t pid;
 337         int r;
 338
 339         assert(message);
 340         assert(m);
 341
 342         assert_cc(sizeof(pid_t) == sizeof(uint32_t));
 343
 344         r = sd_bus_message_read(message, "u", &pid);
 345         if (r < 0)
 346                 return r;
 347         if (pid < 0)
 348                 return -EINVAL;
 349
 350         if (pid == 0) {
 351                 r = manager_get_session_from_creds(m, message, NULL, error, &session);
 352                 if (r < 0)
 353                         return r;
 354         } else {
 355                 r = manager_get_session_by_pid(m, pid, &session);
 356                 if (r < 0)
 357                         return r;
 358
 359                 if (!session)
 360                         return sd_bus_error_setf(error, BUS_ERROR_NO_SESSION_FOR_PID, "PID "PID_FMT" does not belong to any known session", pid);
 361         }
 362
 363         p = session_bus_path(session);
 364         if (!p)
 365                 return -ENOMEM;
 366
 367         return sd_bus_reply_method_return(message, "o", p);
 368 }
 369
 370 static int method_get_user(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 371         _cleanup_free_ char *p = NULL;
 372         Manager *m = userdata;
 373         uint32_t uid;
 374         User *user;
 375         int r;
 376
 377         assert(message);
 378         assert(m);
 379
 380         r = sd_bus_message_read(message, "u", &uid);
 381         if (r < 0)
 382                 return r;
 383
 384         r = manager_get_user_from_creds(m, message, uid, error, &user);
 385         if (r < 0)
 386                 return r;
 387
 388         p = user_bus_path(user);
 389         if (!p)
 390                 return -ENOMEM;
 391
 392         return sd_bus_reply_method_return(message, "o", p);
 393 }
 394
 395 static int method_get_user_by_pid(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 396         _cleanup_free_ char *p = NULL;
 397         Manager *m = userdata;
 398         User *user = NULL;
 399         pid_t pid;
 400         int r;
 401
 402         assert(message);
 403         assert(m);
 404
 405         assert_cc(sizeof(pid_t) == sizeof(uint32_t));
 406
 407         r = sd_bus_message_read(message, "u", &pid);
 408         if (r < 0)
 409                 return r;
 410         if (pid < 0)
 411                 return -EINVAL;
 412
 413         if (pid == 0) {
 414                 r = manager_get_user_from_creds(m, message, UID_INVALID, error, &user);
 415                 if (r < 0)
 416                         return r;
 417         } else {
 418                 r = manager_get_user_by_pid(m, pid, &user);
 419                 if (r < 0)
 420                         return r;
 421                 if (!user)
 422                         return sd_bus_error_setf(error, BUS_ERROR_NO_USER_FOR_PID, "PID "PID_FMT" does not belong to any known or logged in user", pid);
 423         }
 424
 425         p = user_bus_path(user);
 426         if (!p)
 427                 return -ENOMEM;
 428
 429         return sd_bus_reply_method_return(message, "o", p);
 430 }
 431
 432 static int method_get_seat(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 433         _cleanup_free_ char *p = NULL;
 434         Manager *m = userdata;
 435         const char *name;
 436         Seat *seat;
 437         int r;
 438
 439         assert(message);
 440         assert(m);
 441
 442         r = sd_bus_message_read(message, "s", &name);
 443         if (r < 0)
 444                 return r;
 445
 446         r = manager_get_seat_from_creds(m, message, name, error, &seat);
 447         if (r < 0)
 448                 return r;
 449
 450         p = seat_bus_path(seat);
 451         if (!p)
 452                 return -ENOMEM;
 453
 454         return sd_bus_reply_method_return(message, "o", p);
 455 }
 456
 457 static int method_list_sessions(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 458         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
 459         Manager *m = userdata;
 460         Session *session;
 461         Iterator i;
 462         int r;
 463
 464         assert(message);
 465         assert(m);
 466
 467         r = sd_bus_message_new_method_return(message, &reply);
 468         if (r < 0)
 469                 return r;
 470
 471         r = sd_bus_message_open_container(reply, 'a', "(susso)");
 472         if (r < 0)
 473                 return r;
 474
 475         HASHMAP_FOREACH(session, m->sessions, i) {
 476                 _cleanup_free_ char *p = NULL;
 477
 478                 p = session_bus_path(session);
 479                 if (!p)
 480                         return -ENOMEM;
 481
 482                 r = sd_bus_message_append(reply, "(susso)",
 483                                           session->id,
 484                                           (uint32_t) session->user->uid,
 485                                           session->user->name,
 486                                           session->seat ? session->seat->id : "",
 487                                           p);
 488                 if (r < 0)
 489                         return r;
 490         }
 491
 492         r = sd_bus_message_close_container(reply);
 493         if (r < 0)
 494                 return r;
 495
 496         return sd_bus_send(NULL, reply, NULL);
 497 }
 498
 499 static int method_list_users(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 500         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
 501         Manager *m = userdata;
 502         User *user;
 503         Iterator i;
 504         int r;
 505
 506         assert(message);
 507         assert(m);
 508
 509         r = sd_bus_message_new_method_return(message, &reply);
 510         if (r < 0)
 511                 return r;
 512
 513         r = sd_bus_message_open_container(reply, 'a', "(uso)");
 514         if (r < 0)
 515                 return r;
 516
 517         HASHMAP_FOREACH(user, m->users, i) {
 518                 _cleanup_free_ char *p = NULL;
 519
 520                 p = user_bus_path(user);
 521                 if (!p)
 522                         return -ENOMEM;
 523
 524                 r = sd_bus_message_append(reply, "(uso)",
 525                                           (uint32_t) user->uid,
 526                                           user->name,
 527                                           p);
 528                 if (r < 0)
 529                         return r;
 530         }
 531
 532         r = sd_bus_message_close_container(reply);
 533         if (r < 0)
 534                 return r;
 535
 536         return sd_bus_send(NULL, reply, NULL);
 537 }
 538
 539 static int method_list_seats(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 540         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
 541         Manager *m = userdata;
 542         Seat *seat;
 543         Iterator i;
 544         int r;
 545
 546         assert(message);
 547         assert(m);
 548
 549         r = sd_bus_message_new_method_return(message, &reply);
 550         if (r < 0)
 551                 return r;
 552
 553         r = sd_bus_message_open_container(reply, 'a', "(so)");
 554         if (r < 0)
 555                 return r;
 556
 557         HASHMAP_FOREACH(seat, m->seats, i) {
 558                 _cleanup_free_ char *p = NULL;
 559
 560                 p = seat_bus_path(seat);
 561                 if (!p)
 562                         return -ENOMEM;
 563
 564                 r = sd_bus_message_append(reply, "(so)", seat->id, p);
 565                 if (r < 0)
 566                         return r;
 567         }
 568
 569         r = sd_bus_message_close_container(reply);
 570         if (r < 0)
 571                 return r;
 572
 573         return sd_bus_send(NULL, reply, NULL);
 574 }
 575
 576 static int method_list_inhibitors(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 577         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
 578         Manager *m = userdata;
 579         Inhibitor *inhibitor;
 580         Iterator i;
 581         int r;
 582
 583         assert(message);
 584         assert(m);
 585
 586         r = sd_bus_message_new_method_return(message, &reply);
 587         if (r < 0)
 588                 return r;
 589
 590         r = sd_bus_message_open_container(reply, 'a', "(ssssuu)");
 591         if (r < 0)
 592                 return r;
 593
 594         HASHMAP_FOREACH(inhibitor, m->inhibitors, i) {
 595
 596                 r = sd_bus_message_append(reply, "(ssssuu)",
 597                                           strempty(inhibit_what_to_string(inhibitor->what)),
 598                                           strempty(inhibitor->who),
 599                                           strempty(inhibitor->why),
 600                                           strempty(inhibit_mode_to_string(inhibitor->mode)),
 601                                           (uint32_t) inhibitor->uid,
 602                                           (uint32_t) inhibitor->pid);
 603                 if (r < 0)
 604                         return r;
 605         }
 606
 607         r = sd_bus_message_close_container(reply);
 608         if (r < 0)
 609                 return r;
 610
 611         return sd_bus_send(NULL, reply, NULL);
 612 }
 613
 614 static int method_create_session(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 615         const char *service, *type, *class, *cseat, *tty, *display, *remote_user, *remote_host, *desktop;
 616         uint32_t audit_id = 0;
 617         _cleanup_free_ char *id = NULL;
 618         Session *session = NULL;
 619         Manager *m = userdata;
 620         User *user = NULL;
 621         Seat *seat = NULL;
 622         pid_t leader;
 623         uid_t uid;
 624         int remote;
 625         uint32_t vtnr = 0;
 626         SessionType t;
 627         SessionClass c;
 628         int r;
 629
 630         assert(message);
 631         assert(m);
 632
 633         assert_cc(sizeof(pid_t) == sizeof(uint32_t));
 634         assert_cc(sizeof(uid_t) == sizeof(uint32_t));
 635
 636         r = sd_bus_message_read(message, "uusssssussbss", &uid, &leader, &service, &type, &class, &desktop, &cseat, &vtnr, &tty, &display, &remote, &remote_user, &remote_host);
 637         if (r < 0)
 638                 return r;
 639
 640         if (!uid_is_valid(uid))
 641                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid UID");
 642         if (leader < 0 || leader == 1)
 643                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid leader PID");
 644
 645         if (isempty(type))
 646                 t = _SESSION_TYPE_INVALID;
 647         else {
 648                 t = session_type_from_string(type);
 649                 if (t < 0)
 650                         return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid session type %s", type);
 651         }
 652
 653         if (isempty(class))
 654                 c = _SESSION_CLASS_INVALID;
 655         else {
 656                 c = session_class_from_string(class);
 657                 if (c < 0)
 658                         return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid session class %s", class);
 659         }
 660
 661         if (isempty(desktop))
 662                 desktop = NULL;
 663         else {
 664                 if (!string_is_safe(desktop))
 665                         return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid desktop string %s", desktop);
 666         }
 667
 668         if (isempty(cseat))
 669                 seat = NULL;
 670         else {
 671                 seat = hashmap_get(m->seats, cseat);
 672                 if (!seat)
 673                         return sd_bus_error_setf(error, BUS_ERROR_NO_SUCH_SEAT, "No seat '%s' known", cseat);
 674         }
 675
 676         if (tty_is_vc(tty)) {
 677                 int v;
 678
 679                 if (!seat)
 680                         seat = m->seat0;
 681                 else if (seat != m->seat0)
 682                         return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "TTY %s is virtual console but seat %s is not seat0", tty, seat->id);
 683
 684                 v = vtnr_from_tty(tty);
 685                 if (v <= 0)
 686                         return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Cannot determine VT number from virtual console TTY %s", tty);
 687
 688                 if (!vtnr)
 689                         vtnr = (uint32_t) v;
 690                 else if (vtnr != (uint32_t) v)
 691                         return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Specified TTY and VT number do not match");
 692
 693         } else if (tty_is_console(tty)) {
 694
 695                 if (!seat)
 696                         seat = m->seat0;
 697                 else if (seat != m->seat0)
 698                         return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Console TTY specified but seat is not seat0");
 699
 700                 if (vtnr != 0)
 701                         return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Console TTY specified but VT number is not 0");
 702         }
 703
 704         if (seat) {
 705                 if (seat_has_vts(seat)) {
 706                         if (!vtnr || vtnr > 63)
 707                                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "VT number out of range");
 708                 } else {
 709                         if (vtnr != 0)
 710                                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Seat has no VTs but VT number not 0");
 711                 }
 712         }
 713
 714         r = sd_bus_message_enter_container(message, 'a', "(sv)");
 715         if (r < 0)
 716                 return r;
 717
 718         if (t == _SESSION_TYPE_INVALID) {
 719                 if (!isempty(display))
 720                         t = SESSION_X11;
 721                 else if (!isempty(tty))
 722                         t = SESSION_TTY;
 723                 else
 724                         t = SESSION_UNSPECIFIED;
 725         }
 726
 727         if (c == _SESSION_CLASS_INVALID) {
 728                 if (t == SESSION_UNSPECIFIED)
 729                         c = SESSION_BACKGROUND;
 730                 else
 731                         c = SESSION_USER;
 732         }
 733
 734         if (leader == 0) {
 735                 _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL;
 736
 737                 r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_PID, &creds);
 738                 if (r < 0)
 739                         return r;
 740
 741                 r = sd_bus_creds_get_pid(creds, (pid_t*) &leader);
 742                 if (r < 0)
 743                         return r;
 744         }
 745
 746         r = manager_get_session_by_pid(m, leader, NULL);
 747         if (r > 0)
 748                 return sd_bus_error_setf(error, BUS_ERROR_SESSION_BUSY, "Already running in a session");
 749
 750         /*
 751          * Old gdm and lightdm start the user-session on the same VT as
 752          * the greeter session. But they destroy the greeter session
 753          * after the user-session and want the user-session to take
 754          * over the VT. We need to support this for
 755          * backwards-compatibility, so make sure we allow new sessions
 756          * on a VT that a greeter is running on. Furthermore, to allow
 757          * re-logins, we have to allow a greeter to take over a used VT for
 758          * the exact same reasons.
 759          */
 760         if (c != SESSION_GREETER &&
 761             vtnr > 0 &&
 762             vtnr < m->seat0->position_count &&
 763             m->seat0->positions[vtnr] &&
 764             m->seat0->positions[vtnr]->class != SESSION_GREETER)
 765                 return sd_bus_error_setf(error, BUS_ERROR_SESSION_BUSY, "Already occupied by a session");
 766
 767         if (hashmap_size(m->sessions) >= m->sessions_max)
 768                 return sd_bus_error_setf(error, SD_BUS_ERROR_LIMITS_EXCEEDED, "Maximum number of sessions (%" PRIu64 ") reached, refusing further sessions.", m->sessions_max);
 769
 770         (void) audit_session_from_pid(leader, &audit_id);
 771         if (audit_session_is_valid(audit_id)) {
 772                 /* Keep our session IDs and the audit session IDs in sync */
 773
 774                 if (asprintf(&id, "%"PRIu32, audit_id) < 0)
 775                         return -ENOMEM;
 776
 777                 /* Wut? There's already a session by this name and we
 778                  * didn't find it above? Weird, then let's not trust
 779                  * the audit data and let's better register a new
 780                  * ID */
 781                 if (hashmap_get(m->sessions, id)) {
 782                         log_warning("Existing logind session ID %s used by new audit session, ignoring", id);
 783                         audit_id = AUDIT_SESSION_INVALID;
 784
 785                         id = mfree(id);
 786                 }
 787         }
 788
 789         if (!id) {
 790                 do {
 791                         id = mfree(id);
 792
 793                         if (asprintf(&id, "c%lu", ++m->session_counter) < 0)
 794                                 return -ENOMEM;
 795
 796                 } while (hashmap_get(m->sessions, id));
 797         }
 798
 799         r = manager_add_user_by_uid(m, uid, &user);
 800         if (r < 0)
 801                 goto fail;
 802
 803         r = manager_add_session(m, id, &session);
 804         if (r < 0)
 805                 goto fail;
 806
 807         session_set_user(session, user);
 808
 809         session->leader = leader;
 810         session->audit_id = audit_id;
 811         session->type = t;
 812         session->class = c;
 813         session->remote = remote;
 814         session->vtnr = vtnr;
 815
 816         if (!isempty(tty)) {
 817                 session->tty = strdup(tty);
 818                 if (!session->tty) {
 819                         r = -ENOMEM;
 820                         goto fail;
 821                 }
 822         }
 823
 824         if (!isempty(display)) {
 825                 session->display = strdup(display);
 826                 if (!session->display) {
 827                         r = -ENOMEM;
 828                         goto fail;
 829                 }
 830         }
 831
 832         if (!isempty(remote_user)) {
 833                 session->remote_user = strdup(remote_user);
 834                 if (!session->remote_user) {
 835                         r = -ENOMEM;
 836                         goto fail;
 837                 }
 838         }
 839
 840         if (!isempty(remote_host)) {
 841                 session->remote_host = strdup(remote_host);
 842                 if (!session->remote_host) {
 843                         r = -ENOMEM;
 844                         goto fail;
 845                 }
 846         }
 847
 848         if (!isempty(service)) {
 849                 session->service = strdup(service);
 850                 if (!session->service) {
 851                         r = -ENOMEM;
 852                         goto fail;
 853                 }
 854         }
 855
 856         if (!isempty(desktop)) {
 857                 session->desktop = strdup(desktop);
 858                 if (!session->desktop) {
 859                         r = -ENOMEM;
 860                         goto fail;
 861                 }
 862         }
 863
 864         if (seat) {
 865                 r = seat_attach_session(seat, session);
 866                 if (r < 0)
 867                         goto fail;
 868         }
 869
 870         r = session_start(session);
 871         if (r < 0)
 872                 goto fail;
 873
 874         session->create_message = sd_bus_message_ref(message);
 875
 876         /* Now, let's wait until the slice unit and stuff got
 877          * created. We send the reply back from
 878          * session_send_create_reply(). */
 879
 880         return 1;
 881
 882 fail:
 883         if (session)
 884                 session_add_to_gc_queue(session);
 885
 886         if (user)
 887                 user_add_to_gc_queue(user);
 888
 889         return r;
 890 }
 891
 892 static int method_release_session(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 893         Manager *m = userdata;
 894         Session *session;
 895         const char *name;
 896         int r;
 897
 898         assert(message);
 899         assert(m);
 900
 901         r = sd_bus_message_read(message, "s", &name);
 902         if (r < 0)
 903                 return r;
 904
 905         r = manager_get_session_from_creds(m, message, name, error, &session);
 906         if (r < 0)
 907                 return r;
 908
 909         r = session_release(session);
 910         if (r < 0)
 911                 return r;
 912
 913         return sd_bus_reply_method_return(message, NULL);
 914 }
 915
 916 static int method_activate_session(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 917         Manager *m = userdata;
 918         Session *session;
 919         const char *name;
 920         int r;
 921
 922         assert(message);
 923         assert(m);
 924
 925         r = sd_bus_message_read(message, "s", &name);
 926         if (r < 0)
 927                 return r;
 928
 929         r = manager_get_session_from_creds(m, message, name, error, &session);
 930         if (r < 0)
 931                 return r;
 932
 933         return bus_session_method_activate(message, session, error);
 934 }
 935
 936 static int method_activate_session_on_seat(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 937         const char *session_name, *seat_name;
 938         Manager *m = userdata;
 939         Session *session;
 940         Seat *seat;
 941         int r;
 942
 943         assert(message);
 944         assert(m);
 945
 946         /* Same as ActivateSession() but refuses to work if
 947          * the seat doesn't match */
 948
 949         r = sd_bus_message_read(message, "ss", &session_name, &seat_name);
 950         if (r < 0)
 951                 return r;
 952
 953         r = manager_get_session_from_creds(m, message, session_name, error, &session);
 954         if (r < 0)
 955                 return r;
 956
 957         r = manager_get_seat_from_creds(m, message, seat_name, error, &seat);
 958         if (r < 0)
 959                 return r;
 960
 961         if (session->seat != seat)
 962                 return sd_bus_error_setf(error, BUS_ERROR_SESSION_NOT_ON_SEAT, "Session %s not on seat %s", session_name, seat_name);
 963
 964         r = session_activate(session);
 965         if (r < 0)
 966                 return r;
 967
 968         return sd_bus_reply_method_return(message, NULL);
 969 }
 970
 971 static int method_lock_session(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 972         Manager *m = userdata;
 973         Session *session;
 974         const char *name;
 975         int r;
 976
 977         assert(message);
 978         assert(m);
 979
 980         r = sd_bus_message_read(message, "s", &name);
 981         if (r < 0)
 982                 return r;
 983
 984         r = manager_get_session_from_creds(m, message, name, error, &session);
 985         if (r < 0)
 986                 return r;
 987
 988         return bus_session_method_lock(message, session, error);
 989 }
 990
 991 static int method_lock_sessions(sd_bus_message *message, void *userdata, sd_bus_error *error) {
 992         Manager *m = userdata;
 993         int r;
 994
 995         assert(message);
 996         assert(m);
 997
 998         r = bus_verify_polkit_async(
 999                         message,
1000                         CAP_SYS_ADMIN,
1001                         "org.freedesktop.login1.lock-sessions",
1002                         NULL,
1003                         false,
1004                         UID_INVALID,
1005                         &m->polkit_registry,
1006                         error);
1007         if (r < 0)
1008                 return r;
1009         if (r == 0)
1010                 return 1; /* Will call us back */
1011
1012         r = session_send_lock_all(m, streq(sd_bus_message_get_member(message), "LockSessions"));
1013         if (r < 0)
1014                 return r;
1015
1016         return sd_bus_reply_method_return(message, NULL);
1017 }
1018
1019 static int method_kill_session(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1020         const char *name;
1021         Manager *m = userdata;
1022         Session *session;
1023         int r;
1024
1025         assert(message);
1026         assert(m);
1027
1028         r = sd_bus_message_read(message, "s", &name);
1029         if (r < 0)
1030                 return r;
1031
1032         r = manager_get_session_from_creds(m, message, name, error, &session);
1033         if (r < 0)
1034                 return r;
1035
1036         return bus_session_method_kill(message, session, error);
1037 }
1038
1039 static int method_kill_user(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1040         Manager *m = userdata;
1041         uint32_t uid;
1042         User *user;
1043         int r;
1044
1045         assert(message);
1046         assert(m);
1047
1048         r = sd_bus_message_read(message, "u", &uid);
1049         if (r < 0)
1050                 return r;
1051
1052         r = manager_get_user_from_creds(m, message, uid, error, &user);
1053         if (r < 0)
1054                 return r;
1055
1056         return bus_user_method_kill(message, user, error);
1057 }
1058
1059 static int method_terminate_session(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1060         Manager *m = userdata;
1061         const char *name;
1062         Session *session;
1063         int r;
1064
1065         assert(message);
1066         assert(m);
1067
1068         r = sd_bus_message_read(message, "s", &name);
1069         if (r < 0)
1070                 return r;
1071
1072         r = manager_get_session_from_creds(m, message, name, error, &session);
1073         if (r < 0)
1074                 return r;
1075
1076         return bus_session_method_terminate(message, session, error);
1077 }
1078
1079 static int method_terminate_user(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1080         Manager *m = userdata;
1081         uint32_t uid;
1082         User *user;
1083         int r;
1084
1085         assert(message);
1086         assert(m);
1087
1088         r = sd_bus_message_read(message, "u", &uid);
1089         if (r < 0)
1090                 return r;
1091
1092         r = manager_get_user_from_creds(m, message, uid, error, &user);
1093         if (r < 0)
1094                 return r;
1095
1096         return bus_user_method_terminate(message, user, error);
1097 }
1098
1099 static int method_terminate_seat(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1100         Manager *m = userdata;
1101         const char *name;
1102         Seat *seat;
1103         int r;
1104
1105         assert(message);
1106         assert(m);
1107
1108         r = sd_bus_message_read(message, "s", &name);
1109         if (r < 0)
1110                 return r;
1111
1112         r = manager_get_seat_from_creds(m, message, name, error, &seat);
1113         if (r < 0)
1114                 return r;
1115
1116         return bus_seat_method_terminate(message, seat, error);
1117 }
1118
1119 static int method_set_user_linger(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1120         _cleanup_free_ char *cc = NULL;
1121         Manager *m = userdata;
1122         int r, b, interactive;
1123         struct passwd *pw;
1124         const char *path;
1125         uint32_t uid;
1126         bool self = false;
1127
1128         assert(message);
1129         assert(m);
1130
1131         r = sd_bus_message_read(message, "ubb", &uid, &b, &interactive);
1132         if (r < 0)
1133                 return r;
1134
1135         if (!uid_is_valid(uid)) {
1136                 _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL;
1137
1138                 /* Note that we get the owner UID of the session, not the actual client UID here! */
1139                 r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_OWNER_UID|SD_BUS_CREDS_AUGMENT, &creds);
1140                 if (r < 0)
1141                         return r;
1142
1143                 r = sd_bus_creds_get_owner_uid(creds, &uid);
1144                 if (r < 0)
1145                         return r;
1146
1147                 self = true;
1148
1149         } else if (!uid_is_valid(uid))
1150                 return -EINVAL;
1151
1152         errno = 0;
1153         pw = getpwuid(uid);
1154         if (!pw)
1155                 return errno > 0 ? -errno : -ENOENT;
1156
1157         r = bus_verify_polkit_async(
1158                         message,
1159                         CAP_SYS_ADMIN,
1160                         self ? "org.freedesktop.login1.set-self-linger" : "org.freedesktop.login1.set-user-linger",
1161                         NULL,
1162                         interactive,
1163                         UID_INVALID,
1164                         &m->polkit_registry,
1165                         error);
1166         if (r < 0)
1167                 return r;
1168         if (r == 0)
1169                 return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
1170
1171         mkdir_p_label("/var/lib/systemd", 0755);
1172
1173         r = mkdir_safe_label("/var/lib/systemd/linger", 0755, 0, 0);
1174         if (r < 0)
1175                 return r;
1176
1177         cc = cescape(pw->pw_name);
1178         if (!cc)
1179                 return -ENOMEM;
1180
1181         path = strjoina("/var/lib/systemd/linger/", cc);
1182         if (b) {
1183                 User *u;
1184
1185                 r = touch(path);
1186                 if (r < 0)
1187                         return r;
1188
1189                 if (manager_add_user_by_uid(m, uid, &u) >= 0)
1190                         user_start(u);
1191
1192         } else {
1193                 User *u;
1194
1195                 r = unlink(path);
1196                 if (r < 0 && errno != ENOENT)
1197                         return -errno;
1198
1199                 u = hashmap_get(m->users, UID_TO_PTR(uid));
1200                 if (u)
1201                         user_add_to_gc_queue(u);
1202         }
1203
1204         return sd_bus_reply_method_return(message, NULL);
1205 }
1206
1207 static int trigger_device(Manager *m, struct udev_device *d) {
1208         _cleanup_udev_enumerate_unref_ struct udev_enumerate *e = NULL;
1209         struct udev_list_entry *first, *item;
1210         int r;
1211
1212         assert(m);
1213
1214         e = udev_enumerate_new(m->udev);
1215         if (!e)
1216                 return -ENOMEM;
1217
1218         if (d) {
1219                 r = udev_enumerate_add_match_parent(e, d);
1220                 if (r < 0)
1221                         return r;
1222         }
1223
1224         r = udev_enumerate_scan_devices(e);
1225         if (r < 0)
1226                 return r;
1227
1228         first = udev_enumerate_get_list_entry(e);
1229         udev_list_entry_foreach(item, first) {
1230                 _cleanup_free_ char *t = NULL;
1231                 const char *p;
1232
1233                 p = udev_list_entry_get_name(item);
1234
1235                 t = strappend(p, "/uevent");
1236                 if (!t)
1237                         return -ENOMEM;
1238
1239                 write_string_file(t, "change", WRITE_STRING_FILE_CREATE);
1240         }
1241
1242         return 0;
1243 }
1244
1245 static int attach_device(Manager *m, const char *seat, const char *sysfs) {
1246         _cleanup_udev_device_unref_ struct udev_device *d = NULL;
1247         _cleanup_free_ char *rule = NULL, *file = NULL;
1248         const char *id_for_seat;
1249         int r;
1250
1251         assert(m);
1252         assert(seat);
1253         assert(sysfs);
1254
1255         d = udev_device_new_from_syspath(m->udev, sysfs);
1256         if (!d)
1257                 return -ENODEV;
1258
1259         if (!udev_device_has_tag(d, "seat"))
1260                 return -ENODEV;
1261
1262         id_for_seat = udev_device_get_property_value(d, "ID_FOR_SEAT");
1263         if (!id_for_seat)
1264                 return -ENODEV;
1265
1266         if (asprintf(&file, "/etc/udev/rules.d/72-seat-%s.rules", id_for_seat) < 0)
1267                 return -ENOMEM;
1268
1269         if (asprintf(&rule, "TAG==\"seat\", ENV{ID_FOR_SEAT}==\"%s\", ENV{ID_SEAT}=\"%s\"", id_for_seat, seat) < 0)
1270                 return -ENOMEM;
1271
1272         mkdir_p_label("/etc/udev/rules.d", 0755);
1273         r = write_string_file_atomic_label(file, rule);
1274         if (r < 0)
1275                 return r;
1276
1277         return trigger_device(m, d);
1278 }
1279
1280 static int flush_devices(Manager *m) {
1281         _cleanup_closedir_ DIR *d;
1282
1283         assert(m);
1284
1285         d = opendir("/etc/udev/rules.d");
1286         if (!d) {
1287                 if (errno != ENOENT)
1288                         log_warning_errno(errno, "Failed to open /etc/udev/rules.d: %m");
1289         } else {
1290                 struct dirent *de;
1291
1292                 FOREACH_DIRENT_ALL(de, d, break) {
1293                         if (!dirent_is_file(de))
1294                                 continue;
1295
1296                         if (!startswith(de->d_name, "72-seat-"))
1297                                 continue;
1298
1299                         if (!endswith(de->d_name, ".rules"))
1300                                 continue;
1301
1302                         if (unlinkat(dirfd(d), de->d_name, 0) < 0)
1303                                 log_warning_errno(errno, "Failed to unlink %s: %m", de->d_name);
1304                 }
1305         }
1306
1307         return trigger_device(m, NULL);
1308 }
1309
1310 static int method_attach_device(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1311         const char *sysfs, *seat;
1312         Manager *m = userdata;
1313         int interactive, r;
1314
1315         assert(message);
1316         assert(m);
1317
1318         r = sd_bus_message_read(message, "ssb", &seat, &sysfs, &interactive);
1319         if (r < 0)
1320                 return r;
1321
1322         if (!path_startswith(sysfs, "/sys"))
1323                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Path %s is not in /sys", sysfs);
1324
1325         if (!seat_name_is_valid(seat))
1326                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Seat %s is not valid", seat);
1327
1328         r = bus_verify_polkit_async(
1329                         message,
1330                         CAP_SYS_ADMIN,
1331                         "org.freedesktop.login1.attach-device",
1332                         NULL,
1333                         interactive,
1334                         UID_INVALID,
1335                         &m->polkit_registry,
1336                         error);
1337         if (r < 0)
1338                 return r;
1339         if (r == 0)
1340                 return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
1341
1342         r = attach_device(m, seat, sysfs);
1343         if (r < 0)
1344                 return r;
1345
1346         return sd_bus_reply_method_return(message, NULL);
1347 }
1348
1349 static int method_flush_devices(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1350         Manager *m = userdata;
1351         int interactive, r;
1352
1353         assert(message);
1354         assert(m);
1355
1356         r = sd_bus_message_read(message, "b", &interactive);
1357         if (r < 0)
1358                 return r;
1359
1360         r = bus_verify_polkit_async(
1361                         message,
1362                         CAP_SYS_ADMIN,
1363                         "org.freedesktop.login1.flush-devices",
1364                         NULL,
1365                         interactive,
1366                         UID_INVALID,
1367                         &m->polkit_registry,
1368                         error);
1369         if (r < 0)
1370                 return r;
1371         if (r == 0)
1372                 return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
1373
1374         r = flush_devices(m);
1375         if (r < 0)
1376                 return r;
1377
1378         return sd_bus_reply_method_return(message, NULL);
1379 }
1380
1381 static int have_multiple_sessions(
1382                 Manager *m,
1383                 uid_t uid) {
1384
1385         Session *session;
1386         Iterator i;
1387
1388         assert(m);
1389
1390         /* Check for other users' sessions. Greeter sessions do not
1391          * count, and non-login sessions do not count either. */
1392         HASHMAP_FOREACH(session, m->sessions, i)
1393                 if (session->class == SESSION_USER &&
1394                     session->user->uid != uid)
1395                         return true;
1396
1397         return false;
1398 }
1399
1400 static int bus_manager_log_shutdown(
1401                 Manager *m,
1402                 InhibitWhat w,
1403                 const char *unit_name) {
1404
1405         const char *p, *q;
1406
1407         assert(m);
1408         assert(unit_name);
1409
1410         if (w != INHIBIT_SHUTDOWN)
1411                 return 0;
1412
1413         if (streq(unit_name, SPECIAL_POWEROFF_TARGET)) {
1414                 p = "MESSAGE=System is powering down";
1415                 q = "SHUTDOWN=power-off";
1416         } else if (streq(unit_name, SPECIAL_HALT_TARGET)) {
1417                 p = "MESSAGE=System is halting";
1418                 q = "SHUTDOWN=halt";
1419         } else if (streq(unit_name, SPECIAL_REBOOT_TARGET)) {
1420                 p = "MESSAGE=System is rebooting";
1421                 q = "SHUTDOWN=reboot";
1422         } else if (streq(unit_name, SPECIAL_KEXEC_TARGET)) {
1423                 p = "MESSAGE=System is rebooting with kexec";
1424                 q = "SHUTDOWN=kexec";
1425         } else {
1426                 p = "MESSAGE=System is shutting down";
1427                 q = NULL;
1428         }
1429
1430         if (isempty(m->wall_message))
1431                 p = strjoina(p, ".");
1432         else
1433                 p = strjoina(p, " (", m->wall_message, ").");
1434
1435         return log_struct(LOG_NOTICE,
1436                           "MESSAGE_ID=" SD_MESSAGE_SHUTDOWN_STR,
1437                           p,
1438                           q,
1439                           NULL);
1440 }
1441
1442 static int lid_switch_ignore_handler(sd_event_source *e, uint64_t usec, void *userdata) {
1443         Manager *m = userdata;
1444
1445         assert(e);
1446         assert(m);
1447
1448         m->lid_switch_ignore_event_source = sd_event_source_unref(m->lid_switch_ignore_event_source);
1449         return 0;
1450 }
1451
1452 int manager_set_lid_switch_ignore(Manager *m, usec_t until) {
1453         int r;
1454
1455         assert(m);
1456
1457         if (until <= now(CLOCK_MONOTONIC))
1458                 return 0;
1459
1460         /* We want to ignore the lid switch for a while after each
1461          * suspend, and after boot-up. Hence let's install a timer for
1462          * this. As long as the event source exists we ignore the lid
1463          * switch. */
1464
1465         if (m->lid_switch_ignore_event_source) {
1466                 usec_t u;
1467
1468                 r = sd_event_source_get_time(m->lid_switch_ignore_event_source, &u);
1469                 if (r < 0)
1470                         return r;
1471
1472                 if (until <= u)
1473                         return 0;
1474
1475                 r = sd_event_source_set_time(m->lid_switch_ignore_event_source, until);
1476         } else
1477                 r = sd_event_add_time(
1478                                 m->event,
1479                                 &m->lid_switch_ignore_event_source,
1480                                 CLOCK_MONOTONIC,
1481                                 until, 0,
1482                                 lid_switch_ignore_handler, m);
1483
1484         return r;
1485 }
1486
1487 static void reset_scheduled_shutdown(Manager *m) {
1488         m->scheduled_shutdown_timeout_source = sd_event_source_unref(m->scheduled_shutdown_timeout_source);
1489         m->wall_message_timeout_source = sd_event_source_unref(m->wall_message_timeout_source);
1490         m->nologin_timeout_source = sd_event_source_unref(m->nologin_timeout_source);
1491         m->scheduled_shutdown_type = mfree(m->scheduled_shutdown_type);
1492         m->scheduled_shutdown_timeout = 0;
1493         m->shutdown_dry_run = false;
1494
1495         if (m->unlink_nologin) {
1496                 (void) unlink("/run/nologin");
1497                 m->unlink_nologin = false;
1498         }
1499         (void) unlink("/run/systemd/shutdown/scheduled");
1500 }
1501
1502 static int execute_shutdown_or_sleep(
1503                 Manager *m,
1504                 InhibitWhat w,
1505                 const char *unit_name,
1506                 sd_bus_error *error) {
1507
1508         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
1509         char *c = NULL;
1510         const char *p;
1511         int r;
1512
1513         assert(m);
1514         assert(w >= 0);
1515         assert(w < _INHIBIT_WHAT_MAX);
1516         assert(unit_name);
1517
1518         bus_manager_log_shutdown(m, w, unit_name);
1519
1520         if (m->shutdown_dry_run) {
1521                 log_info("Running in dry run, suppressing action.");
1522                 reset_scheduled_shutdown(m);
1523         } else {
1524                 r = sd_bus_call_method(
1525                                 m->bus,
1526                                 "org.freedesktop.systemd1",
1527                                 "/org/freedesktop/systemd1",
1528                                 "org.freedesktop.systemd1.Manager",
1529                                 "StartUnit",
1530                                 error,
1531                                 &reply,
1532                                 "ss", unit_name, "replace-irreversibly");
1533                 if (r < 0)
1534                         return r;
1535
1536                 r = sd_bus_message_read(reply, "o", &p);
1537                 if (r < 0)
1538                         return r;
1539
1540                 c = strdup(p);
1541                 if (!c)
1542                         return -ENOMEM;
1543         }
1544
1545         m->action_unit = unit_name;
1546         free(m->action_job);
1547         m->action_job = c;
1548         m->action_what = w;
1549
1550         /* Make sure the lid switch is ignored for a while */
1551         manager_set_lid_switch_ignore(m, now(CLOCK_MONOTONIC) + m->holdoff_timeout_usec);
1552
1553         return 0;
1554 }
1555
1556 int manager_dispatch_delayed(Manager *manager, bool timeout) {
1557
1558         _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
1559         Inhibitor *offending = NULL;
1560         int r;
1561
1562         assert(manager);
1563
1564         if (manager->action_what == 0 || manager->action_job)
1565                 return 0;
1566
1567         if (manager_is_inhibited(manager, manager->action_what, INHIBIT_DELAY, NULL, false, false, 0, &offending)) {
1568                 _cleanup_free_ char *comm = NULL, *u = NULL;
1569
1570                 if (!timeout)
1571                         return 0;
1572
1573                 (void) get_process_comm(offending->pid, &comm);
1574                 u = uid_to_name(offending->uid);
1575
1576                 log_notice("Delay lock is active (UID "UID_FMT"/%s, PID "PID_FMT"/%s) but inhibitor timeout is reached.",
1577                            offending->uid, strna(u),
1578                            offending->pid, strna(comm));
1579         }
1580
1581         /* Actually do the operation */
1582         r = execute_shutdown_or_sleep(manager, manager->action_what, manager->action_unit, &error);
1583         if (r < 0) {
1584                 log_warning("Failed to send delayed message: %s", bus_error_message(&error, r));
1585
1586                 manager->action_unit = NULL;
1587                 manager->action_what = 0;
1588                 return r;
1589         }
1590
1591         return 1;
1592 }
1593
1594 static int manager_inhibit_timeout_handler(
1595                         sd_event_source *s,
1596                         uint64_t usec,
1597                         void *userdata) {
1598
1599         Manager *manager = userdata;
1600         int r;
1601
1602         assert(manager);
1603         assert(manager->inhibit_timeout_source == s);
1604
1605         r = manager_dispatch_delayed(manager, true);
1606         return (r < 0) ? r : 0;
1607 }
1608
1609 static int delay_shutdown_or_sleep(
1610                 Manager *m,
1611                 InhibitWhat w,
1612                 const char *unit_name) {
1613
1614         int r;
1615         usec_t timeout_val;
1616
1617         assert(m);
1618         assert(w >= 0);
1619         assert(w < _INHIBIT_WHAT_MAX);
1620         assert(unit_name);
1621
1622         timeout_val = now(CLOCK_MONOTONIC) + m->inhibit_delay_max;
1623
1624         if (m->inhibit_timeout_source) {
1625                 r = sd_event_source_set_time(m->inhibit_timeout_source, timeout_val);
1626                 if (r < 0)
1627                         return log_error_errno(r, "sd_event_source_set_time() failed: %m");
1628
1629                 r = sd_event_source_set_enabled(m->inhibit_timeout_source, SD_EVENT_ONESHOT);
1630                 if (r < 0)
1631                         return log_error_errno(r, "sd_event_source_set_enabled() failed: %m");
1632         } else {
1633                 r = sd_event_add_time(m->event, &m->inhibit_timeout_source, CLOCK_MONOTONIC,
1634                                       timeout_val, 0, manager_inhibit_timeout_handler, m);
1635                 if (r < 0)
1636                         return r;
1637         }
1638
1639         m->action_unit = unit_name;
1640         m->action_what = w;
1641
1642         return 0;
1643 }
1644
1645 static int send_prepare_for(Manager *m, InhibitWhat w, bool _active) {
1646
1647         static const char * const signal_name[_INHIBIT_WHAT_MAX] = {
1648                 [INHIBIT_SHUTDOWN] = "PrepareForShutdown",
1649                 [INHIBIT_SLEEP] = "PrepareForSleep"
1650         };
1651
1652         int active = _active;
1653
1654         assert(m);
1655         assert(w >= 0);
1656         assert(w < _INHIBIT_WHAT_MAX);
1657         assert(signal_name[w]);
1658
1659         return sd_bus_emit_signal(m->bus,
1660                                   "/org/freedesktop/login1",
1661                                   "org.freedesktop.login1.Manager",
1662                                   signal_name[w],
1663                                   "b",
1664                                   active);
1665 }
1666
1667 int bus_manager_shutdown_or_sleep_now_or_later(
1668                 Manager *m,
1669                 const char *unit_name,
1670                 InhibitWhat w,
1671                 sd_bus_error *error) {
1672
1673         bool delayed;
1674         int r;
1675
1676         assert(m);
1677         assert(unit_name);
1678         assert(w >= 0);
1679         assert(w <= _INHIBIT_WHAT_MAX);
1680         assert(!m->action_job);
1681
1682         /* Tell everybody to prepare for shutdown/sleep */
1683         send_prepare_for(m, w, true);
1684
1685         delayed =
1686                 m->inhibit_delay_max > 0 &&
1687                 manager_is_inhibited(m, w, INHIBIT_DELAY, NULL, false, false, 0, NULL);
1688
1689         if (delayed)
1690                 /* Shutdown is delayed, keep in mind what we
1691                  * want to do, and start a timeout */
1692                 r = delay_shutdown_or_sleep(m, w, unit_name);
1693         else
1694                 /* Shutdown is not delayed, execute it
1695                  * immediately */
1696                 r = execute_shutdown_or_sleep(m, w, unit_name, error);
1697
1698         return r;
1699 }
1700
1701 static int verify_shutdown_creds(
1702                 Manager *m,
1703                 sd_bus_message *message,
1704                 InhibitWhat w,
1705                 bool interactive,
1706                 const char *action,
1707                 const char *action_multiple_sessions,
1708                 const char *action_ignore_inhibit,
1709                 sd_bus_error *error) {
1710
1711         _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL;
1712         bool multiple_sessions, blocked;
1713         uid_t uid;
1714         int r;
1715
1716         assert(m);
1717         assert(message);
1718         assert(w >= 0);
1719         assert(w <= _INHIBIT_WHAT_MAX);
1720
1721         r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_EUID, &creds);
1722         if (r < 0)
1723                 return r;
1724
1725         r = sd_bus_creds_get_euid(creds, &uid);
1726         if (r < 0)
1727                 return r;
1728
1729         r = have_multiple_sessions(m, uid);
1730         if (r < 0)
1731                 return r;
1732
1733         multiple_sessions = r > 0;
1734         blocked = manager_is_inhibited(m, w, INHIBIT_BLOCK, NULL, false, true, uid, NULL);
1735
1736         if (multiple_sessions && action_multiple_sessions) {
1737                 r = bus_verify_polkit_async(message, CAP_SYS_BOOT, action_multiple_sessions, NULL, interactive, UID_INVALID, &m->polkit_registry, error);
1738                 if (r < 0)
1739                         return r;
1740                 if (r == 0)
1741                         return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
1742         }
1743
1744         if (blocked && action_ignore_inhibit) {
1745                 r = bus_verify_polkit_async(message, CAP_SYS_BOOT, action_ignore_inhibit, NULL, interactive, UID_INVALID, &m->polkit_registry, error);
1746                 if (r < 0)
1747                         return r;
1748                 if (r == 0)
1749                         return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
1750         }
1751
1752         if (!multiple_sessions && !blocked && action) {
1753                 r = bus_verify_polkit_async(message, CAP_SYS_BOOT, action, NULL, interactive, UID_INVALID, &m->polkit_registry, error);
1754                 if (r < 0)
1755                         return r;
1756                 if (r == 0)
1757                         return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
1758         }
1759
1760         return 0;
1761 }
1762
1763 static int method_do_shutdown_or_sleep(
1764                 Manager *m,
1765                 sd_bus_message *message,
1766                 const char *unit_name,
1767                 InhibitWhat w,
1768                 const char *action,
1769                 const char *action_multiple_sessions,
1770                 const char *action_ignore_inhibit,
1771                 const char *sleep_verb,
1772                 sd_bus_error *error) {
1773
1774         int interactive, r;
1775
1776         assert(m);
1777         assert(message);
1778         assert(unit_name);
1779         assert(w >= 0);
1780         assert(w <= _INHIBIT_WHAT_MAX);
1781
1782         r = sd_bus_message_read(message, "b", &interactive);
1783         if (r < 0)
1784                 return r;
1785
1786         /* Don't allow multiple jobs being executed at the same time */
1787         if (m->action_what)
1788                 return sd_bus_error_setf(error, BUS_ERROR_OPERATION_IN_PROGRESS, "There's already a shutdown or sleep operation in progress");
1789
1790         if (sleep_verb) {
1791                 r = can_sleep(sleep_verb);
1792                 if (r < 0)
1793                         return r;
1794
1795                 if (r == 0)
1796                         return sd_bus_error_setf(error, BUS_ERROR_SLEEP_VERB_NOT_SUPPORTED, "Sleep verb not supported");
1797         }
1798
1799         r = verify_shutdown_creds(m, message, w, interactive, action, action_multiple_sessions,
1800                                   action_ignore_inhibit, error);
1801         if (r != 0)
1802                 return r;
1803
1804         r = bus_manager_shutdown_or_sleep_now_or_later(m, unit_name, w, error);
1805         if (r < 0)
1806                 return r;
1807
1808         return sd_bus_reply_method_return(message, NULL);
1809 }
1810
1811 static int method_poweroff(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1812         Manager *m = userdata;
1813
1814         return method_do_shutdown_or_sleep(
1815                         m, message,
1816                         SPECIAL_POWEROFF_TARGET,
1817                         INHIBIT_SHUTDOWN,
1818                         "org.freedesktop.login1.power-off",
1819                         "org.freedesktop.login1.power-off-multiple-sessions",
1820                         "org.freedesktop.login1.power-off-ignore-inhibit",
1821                         NULL,
1822                         error);
1823 }
1824
1825 static int method_reboot(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1826         Manager *m = userdata;
1827
1828         return method_do_shutdown_or_sleep(
1829                         m, message,
1830                         SPECIAL_REBOOT_TARGET,
1831                         INHIBIT_SHUTDOWN,
1832                         "org.freedesktop.login1.reboot",
1833                         "org.freedesktop.login1.reboot-multiple-sessions",
1834                         "org.freedesktop.login1.reboot-ignore-inhibit",
1835                         NULL,
1836                         error);
1837 }
1838
1839 static int method_suspend(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1840         Manager *m = userdata;
1841
1842         return method_do_shutdown_or_sleep(
1843                         m, message,
1844                         SPECIAL_SUSPEND_TARGET,
1845                         INHIBIT_SLEEP,
1846                         "org.freedesktop.login1.suspend",
1847                         "org.freedesktop.login1.suspend-multiple-sessions",
1848                         "org.freedesktop.login1.suspend-ignore-inhibit",
1849                         "suspend",
1850                         error);
1851 }
1852
1853 static int nologin_timeout_handler(
1854                         sd_event_source *s,
1855                         uint64_t usec,
1856                         void *userdata) {
1857
1858         Manager *m = userdata;
1859         int r;
1860
1861         log_info("Creating /run/nologin, blocking further logins...");
1862
1863         r = write_string_file_atomic_label("/run/nologin", "System is going down.");
1864         if (r < 0)
1865                 log_error_errno(r, "Failed to create /run/nologin: %m");
1866         else
1867                 m->unlink_nologin = true;
1868
1869         return 0;
1870 }
1871
1872 static int update_schedule_file(Manager *m) {
1873         _cleanup_free_ char *temp_path = NULL;
1874         _cleanup_fclose_ FILE *f = NULL;
1875         int r;
1876
1877         assert(m);
1878
1879         r = mkdir_safe_label("/run/systemd/shutdown", 0755, 0, 0);
1880         if (r < 0)
1881                 return log_error_errno(r, "Failed to create shutdown subdirectory: %m");
1882
1883         r = fopen_temporary("/run/systemd/shutdown/scheduled", &f, &temp_path);
1884         if (r < 0)
1885                 return log_error_errno(r, "Failed to save information about scheduled shutdowns: %m");
1886
1887         (void) fchmod(fileno(f), 0644);
1888
1889         fprintf(f,
1890                 "USEC="USEC_FMT"\n"
1891                 "WARN_WALL=%i\n"
1892                 "MODE=%s\n",
1893                 m->scheduled_shutdown_timeout,
1894                 m->enable_wall_messages,
1895                 m->scheduled_shutdown_type);
1896
1897         if (!isempty(m->wall_message)) {
1898                 _cleanup_free_ char *t;
1899
1900                 t = cescape(m->wall_message);
1901                 if (!t) {
1902                         r = -ENOMEM;
1903                         goto fail;
1904                 }
1905
1906                 fprintf(f, "WALL_MESSAGE=%s\n", t);
1907         }
1908
1909         r = fflush_and_check(f);
1910         if (r < 0)
1911                 goto fail;
1912
1913         if (rename(temp_path, "/run/systemd/shutdown/scheduled") < 0) {
1914                 r = -errno;
1915                 goto fail;
1916         }
1917
1918         return 0;
1919
1920 fail:
1921         (void) unlink(temp_path);
1922         (void) unlink("/run/systemd/shutdown/scheduled");
1923
1924         return log_error_errno(r, "Failed to write information about scheduled shutdowns: %m");
1925 }
1926
1927 static int manager_scheduled_shutdown_handler(
1928                         sd_event_source *s,
1929                         uint64_t usec,
1930                         void *userdata) {
1931
1932         _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
1933         Manager *m = userdata;
1934         const char *target;
1935         int r;
1936
1937         assert(m);
1938
1939         if (isempty(m->scheduled_shutdown_type))
1940                 return 0;
1941
1942         if (streq(m->scheduled_shutdown_type, "halt"))
1943                 target = SPECIAL_HALT_TARGET;
1944         else if (streq(m->scheduled_shutdown_type, "poweroff"))
1945                 target = SPECIAL_POWEROFF_TARGET;
1946         else
1947                 target = SPECIAL_REBOOT_TARGET;
1948
1949         r = execute_shutdown_or_sleep(m, 0, target, &error);
1950         if (r < 0)
1951                 return log_error_errno(r, "Unable to execute transition to %s: %m", target);
1952
1953         return 0;
1954 }
1955
1956 static int method_schedule_shutdown(sd_bus_message *message, void *userdata, sd_bus_error *error) {
1957         Manager *m = userdata;
1958         _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL;
1959         const char *action_multiple_sessions = NULL;
1960         const char *action_ignore_inhibit = NULL;
1961         const char *action = NULL;
1962         uint64_t elapse;
1963         char *type;
1964         int r;
1965
1966         assert(m);
1967         assert(message);
1968
1969         r = sd_bus_message_read(message, "st", &type, &elapse);
1970         if (r < 0)
1971                 return r;
1972
1973         if (startswith(type, "dry-")) {
1974                 type += 4;
1975                 m->shutdown_dry_run = true;
1976         }
1977
1978         if (streq(type, "reboot")) {
1979                 action = "org.freedesktop.login1.reboot";
1980                 action_multiple_sessions = "org.freedesktop.login1.reboot-multiple-sessions";
1981                 action_ignore_inhibit = "org.freedesktop.login1.reboot-ignore-inhibit";
1982         } else if (streq(type, "halt")) {
1983                 action = "org.freedesktop.login1.halt";
1984                 action_multiple_sessions = "org.freedesktop.login1.halt-multiple-sessions";
1985                 action_ignore_inhibit = "org.freedesktop.login1.halt-ignore-inhibit";
1986         } else if (streq(type, "poweroff")) {
1987                 action = "org.freedesktop.login1.power-off";
1988                 action_multiple_sessions = "org.freedesktop.login1.power-off-multiple-sessions";
1989                 action_ignore_inhibit = "org.freedesktop.login1.power-off-ignore-inhibit";
1990         } else
1991                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Unsupported shutdown type");
1992
1993         r = verify_shutdown_creds(m, message, INHIBIT_SHUTDOWN, false,
1994                                   action, action_multiple_sessions, action_ignore_inhibit, error);
1995         if (r != 0)
1996                 return r;
1997
1998         if (m->scheduled_shutdown_timeout_source) {
1999                 r = sd_event_source_set_time(m->scheduled_shutdown_timeout_source, elapse);
2000                 if (r < 0)
2001                         return log_error_errno(r, "sd_event_source_set_time() failed: %m");
2002
2003                 r = sd_event_source_set_enabled(m->scheduled_shutdown_timeout_source, SD_EVENT_ONESHOT);
2004                 if (r < 0)
2005                         return log_error_errno(r, "sd_event_source_set_enabled() failed: %m");
2006         } else {
2007                 r = sd_event_add_time(m->event, &m->scheduled_shutdown_timeout_source,
2008                                       CLOCK_REALTIME, elapse, 0, manager_scheduled_shutdown_handler, m);
2009                 if (r < 0)
2010                         return log_error_errno(r, "sd_event_add_time() failed: %m");
2011         }
2012
2013         r = free_and_strdup(&m->scheduled_shutdown_type, type);
2014         if (r < 0) {
2015                 m->scheduled_shutdown_timeout_source = sd_event_source_unref(m->scheduled_shutdown_timeout_source);
2016                 return log_oom();
2017         }
2018
2019         if (m->nologin_timeout_source) {
2020                 r = sd_event_source_set_time(m->nologin_timeout_source, elapse);
2021                 if (r < 0)
2022                         return log_error_errno(r, "sd_event_source_set_time() failed: %m");
2023
2024                 r = sd_event_source_set_enabled(m->nologin_timeout_source, SD_EVENT_ONESHOT);
2025                 if (r < 0)
2026                         return log_error_errno(r, "sd_event_source_set_enabled() failed: %m");
2027         } else {
2028                 r = sd_event_add_time(m->event, &m->nologin_timeout_source,
2029                                       CLOCK_REALTIME, elapse - 5 * USEC_PER_MINUTE, 0, nologin_timeout_handler, m);
2030                 if (r < 0)
2031                         return log_error_errno(r, "sd_event_add_time() failed: %m");
2032         }
2033
2034         m->scheduled_shutdown_timeout = elapse;
2035
2036         r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_AUGMENT|SD_BUS_CREDS_TTY|SD_BUS_CREDS_UID, &creds);
2037         if (r >= 0) {
2038                 const char *tty = NULL;
2039
2040                 (void) sd_bus_creds_get_uid(creds, &m->scheduled_shutdown_uid);
2041                 (void) sd_bus_creds_get_tty(creds, &tty);
2042
2043                 r = free_and_strdup(&m->scheduled_shutdown_tty, tty);
2044                 if (r < 0) {
2045                         m->scheduled_shutdown_timeout_source = sd_event_source_unref(m->scheduled_shutdown_timeout_source);
2046                         return log_oom();
2047                 }
2048         }
2049
2050         r = manager_setup_wall_message_timer(m);
2051         if (r < 0)
2052                 return r;
2053
2054         if (!isempty(type)) {
2055                 r = update_schedule_file(m);
2056                 if (r < 0)
2057                         return r;
2058         } else
2059                 (void) unlink("/run/systemd/shutdown/scheduled");
2060
2061         return sd_bus_reply_method_return(message, NULL);
2062 }
2063
2064 static int method_cancel_scheduled_shutdown(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2065         Manager *m = userdata;
2066         bool cancelled;
2067
2068         assert(m);
2069         assert(message);
2070
2071         cancelled = m->scheduled_shutdown_type != NULL;
2072         reset_scheduled_shutdown(m);
2073
2074         if (cancelled) {
2075                 _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL;
2076                 const char *tty = NULL;
2077                 uid_t uid = 0;
2078                 int r;
2079
2080                 r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_AUGMENT|SD_BUS_CREDS_TTY|SD_BUS_CREDS_UID, &creds);
2081                 if (r >= 0) {
2082                         (void) sd_bus_creds_get_uid(creds, &uid);
2083                         (void) sd_bus_creds_get_tty(creds, &tty);
2084                 }
2085
2086                 utmp_wall("The system shutdown has been cancelled",
2087                           uid_to_name(uid), tty, logind_wall_tty_filter, m);
2088         }
2089
2090         return sd_bus_reply_method_return(message, "b", cancelled);
2091 }
2092
2093 static int method_hibernate(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2094         Manager *m = userdata;
2095
2096         return method_do_shutdown_or_sleep(
2097                         m, message,
2098                         SPECIAL_HIBERNATE_TARGET,
2099                         INHIBIT_SLEEP,
2100                         "org.freedesktop.login1.hibernate",
2101                         "org.freedesktop.login1.hibernate-multiple-sessions",
2102                         "org.freedesktop.login1.hibernate-ignore-inhibit",
2103                         "hibernate",
2104                         error);
2105 }
2106
2107 static int method_hybrid_sleep(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2108         Manager *m = userdata;
2109
2110         return method_do_shutdown_or_sleep(
2111                         m, message,
2112                         SPECIAL_HYBRID_SLEEP_TARGET,
2113                         INHIBIT_SLEEP,
2114                         "org.freedesktop.login1.hibernate",
2115                         "org.freedesktop.login1.hibernate-multiple-sessions",
2116                         "org.freedesktop.login1.hibernate-ignore-inhibit",
2117                         "hybrid-sleep",
2118                         error);
2119 }
2120
2121 static int method_can_shutdown_or_sleep(
2122                 Manager *m,
2123                 sd_bus_message *message,
2124                 InhibitWhat w,
2125                 const char *action,
2126                 const char *action_multiple_sessions,
2127                 const char *action_ignore_inhibit,
2128                 const char *sleep_verb,
2129                 sd_bus_error *error) {
2130
2131         _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL;
2132         bool multiple_sessions, challenge, blocked;
2133         const char *result = NULL;
2134         uid_t uid;
2135         int r;
2136
2137         assert(m);
2138         assert(message);
2139         assert(w >= 0);
2140         assert(w <= _INHIBIT_WHAT_MAX);
2141         assert(action);
2142         assert(action_multiple_sessions);
2143         assert(action_ignore_inhibit);
2144
2145         if (sleep_verb) {
2146                 r = can_sleep(sleep_verb);
2147                 if (r < 0)
2148                         return r;
2149                 if (r == 0)
2150                         return sd_bus_reply_method_return(message, "s", "na");
2151         }
2152
2153         r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_EUID, &creds);
2154         if (r < 0)
2155                 return r;
2156
2157         r = sd_bus_creds_get_euid(creds, &uid);
2158         if (r < 0)
2159                 return r;
2160
2161         r = have_multiple_sessions(m, uid);
2162         if (r < 0)
2163                 return r;
2164
2165         multiple_sessions = r > 0;
2166         blocked = manager_is_inhibited(m, w, INHIBIT_BLOCK, NULL, false, true, uid, NULL);
2167
2168         if (multiple_sessions) {
2169                 r = bus_test_polkit(message, CAP_SYS_BOOT, action_multiple_sessions, NULL, UID_INVALID, &challenge, error);
2170                 if (r < 0)
2171                         return r;
2172
2173                 if (r > 0)
2174                         result = "yes";
2175                 else if (challenge)
2176                         result = "challenge";
2177                 else
2178                         result = "no";
2179         }
2180
2181         if (blocked) {
2182                 r = bus_test_polkit(message, CAP_SYS_BOOT, action_ignore_inhibit, NULL, UID_INVALID, &challenge, error);
2183                 if (r < 0)
2184                         return r;
2185
2186                 if (r > 0 && !result)
2187                         result = "yes";
2188                 else if (challenge && (!result || streq(result, "yes")))
2189                         result = "challenge";
2190                 else
2191                         result = "no";
2192         }
2193
2194         if (!multiple_sessions && !blocked) {
2195                 /* If neither inhibit nor multiple sessions
2196                  * apply then just check the normal policy */
2197
2198                 r = bus_test_polkit(message, CAP_SYS_BOOT, action, NULL, UID_INVALID, &challenge, error);
2199                 if (r < 0)
2200                         return r;
2201
2202                 if (r > 0)
2203                         result = "yes";
2204                 else if (challenge)
2205                         result = "challenge";
2206                 else
2207                         result = "no";
2208         }
2209
2210         return sd_bus_reply_method_return(message, "s", result);
2211 }
2212
2213 static int method_can_poweroff(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2214         Manager *m = userdata;
2215
2216         return method_can_shutdown_or_sleep(
2217                         m, message,
2218                         INHIBIT_SHUTDOWN,
2219                         "org.freedesktop.login1.power-off",
2220                         "org.freedesktop.login1.power-off-multiple-sessions",
2221                         "org.freedesktop.login1.power-off-ignore-inhibit",
2222                         NULL,
2223                         error);
2224 }
2225
2226 static int method_can_reboot(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2227         Manager *m = userdata;
2228
2229         return method_can_shutdown_or_sleep(
2230                         m, message,
2231                         INHIBIT_SHUTDOWN,
2232                         "org.freedesktop.login1.reboot",
2233                         "org.freedesktop.login1.reboot-multiple-sessions",
2234                         "org.freedesktop.login1.reboot-ignore-inhibit",
2235                         NULL,
2236                         error);
2237 }
2238
2239 static int method_can_suspend(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2240         Manager *m = userdata;
2241
2242         return method_can_shutdown_or_sleep(
2243                         m, message,
2244                         INHIBIT_SLEEP,
2245                         "org.freedesktop.login1.suspend",
2246                         "org.freedesktop.login1.suspend-multiple-sessions",
2247                         "org.freedesktop.login1.suspend-ignore-inhibit",
2248                         "suspend",
2249                         error);
2250 }
2251
2252 static int method_can_hibernate(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2253         Manager *m = userdata;
2254
2255         return method_can_shutdown_or_sleep(
2256                         m, message,
2257                         INHIBIT_SLEEP,
2258                         "org.freedesktop.login1.hibernate",
2259                         "org.freedesktop.login1.hibernate-multiple-sessions",
2260                         "org.freedesktop.login1.hibernate-ignore-inhibit",
2261                         "hibernate",
2262                         error);
2263 }
2264
2265 static int method_can_hybrid_sleep(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2266         Manager *m = userdata;
2267
2268         return method_can_shutdown_or_sleep(
2269                         m, message,
2270                         INHIBIT_SLEEP,
2271                         "org.freedesktop.login1.hibernate",
2272                         "org.freedesktop.login1.hibernate-multiple-sessions",
2273                         "org.freedesktop.login1.hibernate-ignore-inhibit",
2274                         "hybrid-sleep",
2275                         error);
2276 }
2277
2278 static int property_get_reboot_to_firmware_setup(
2279                 sd_bus *bus,
2280                 const char *path,
2281                 const char *interface,
2282                 const char *property,
2283                 sd_bus_message *reply,
2284                 void *userdata,
2285                 sd_bus_error *error) {
2286         int r;
2287
2288         assert(bus);
2289         assert(reply);
2290         assert(userdata);
2291
2292         r = efi_get_reboot_to_firmware();
2293         if (r < 0 && r != -EOPNOTSUPP)
2294                 return r;
2295
2296         return sd_bus_message_append(reply, "b", r > 0);
2297 }
2298
2299 static int method_set_reboot_to_firmware_setup(
2300                 sd_bus_message *message,
2301                 void *userdata,
2302                 sd_bus_error *error) {
2303
2304         int b, r;
2305         Manager *m = userdata;
2306
2307         assert(message);
2308         assert(m);
2309
2310         r = sd_bus_message_read(message, "b", &b);
2311         if (r < 0)
2312                 return r;
2313
2314         r = bus_verify_polkit_async(message,
2315                                     CAP_SYS_ADMIN,
2316                                     "org.freedesktop.login1.set-reboot-to-firmware-setup",
2317                                     NULL,
2318                                     false,
2319                                     UID_INVALID,
2320                                     &m->polkit_registry,
2321                                     error);
2322         if (r < 0)
2323                 return r;
2324         if (r == 0)
2325                 return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
2326
2327         r = efi_set_reboot_to_firmware(b);
2328         if (r < 0)
2329                 return r;
2330
2331         return sd_bus_reply_method_return(message, NULL);
2332 }
2333
2334 static int method_can_reboot_to_firmware_setup(
2335                 sd_bus_message *message,
2336                 void *userdata,
2337                 sd_bus_error *error) {
2338
2339         int r;
2340         bool challenge;
2341         const char *result;
2342         Manager *m = userdata;
2343
2344         assert(message);
2345         assert(m);
2346
2347         r = efi_reboot_to_firmware_supported();
2348         if (r == -EOPNOTSUPP)
2349                 return sd_bus_reply_method_return(message, "s", "na");
2350         else if (r < 0)
2351                 return r;
2352
2353         r = bus_test_polkit(message,
2354                             CAP_SYS_ADMIN,
2355                             "org.freedesktop.login1.set-reboot-to-firmware-setup",
2356                             NULL,
2357                             UID_INVALID,
2358                             &challenge,
2359                             error);
2360         if (r < 0)
2361                 return r;
2362
2363         if (r > 0)
2364                 result = "yes";
2365         else if (challenge)
2366                 result = "challenge";
2367         else
2368                 result = "no";
2369
2370         return sd_bus_reply_method_return(message, "s", result);
2371 }
2372
2373 static int method_set_wall_message(
2374                 sd_bus_message *message,
2375                 void *userdata,
2376                 sd_bus_error *error) {
2377
2378         int r;
2379         Manager *m = userdata;
2380         char *wall_message;
2381         int enable_wall_messages;
2382
2383         assert(message);
2384         assert(m);
2385
2386         r = sd_bus_message_read(message, "sb", &wall_message, &enable_wall_messages);
2387         if (r < 0)
2388                 return r;
2389
2390         r = bus_verify_polkit_async(message,
2391                                     CAP_SYS_ADMIN,
2392                                     "org.freedesktop.login1.set-wall-message",
2393                                     NULL,
2394                                     false,
2395                                     UID_INVALID,
2396                                     &m->polkit_registry,
2397                                     error);
2398         if (r < 0)
2399                 return r;
2400         if (r == 0)
2401                 return 1; /* Will call us back */
2402
2403         r = free_and_strdup(&m->wall_message, empty_to_null(wall_message));
2404         if (r < 0)
2405                 return log_oom();
2406
2407         m->enable_wall_messages = enable_wall_messages;
2408
2409         return sd_bus_reply_method_return(message, NULL);
2410 }
2411
2412 static int method_inhibit(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2413         _cleanup_(sd_bus_creds_unrefp) sd_bus_creds *creds = NULL;
2414         const char *who, *why, *what, *mode;
2415         _cleanup_free_ char *id = NULL;
2416         _cleanup_close_ int fifo_fd = -1;
2417         Manager *m = userdata;
2418         Inhibitor *i = NULL;
2419         InhibitMode mm;
2420         InhibitWhat w;
2421         pid_t pid;
2422         uid_t uid;
2423         int r;
2424
2425         assert(message);
2426         assert(m);
2427
2428         r = sd_bus_message_read(message, "ssss", &what, &who, &why, &mode);
2429         if (r < 0)
2430                 return r;
2431
2432         w = inhibit_what_from_string(what);
2433         if (w <= 0)
2434                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid what specification %s", what);
2435
2436         mm = inhibit_mode_from_string(mode);
2437         if (mm < 0)
2438                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid mode specification %s", mode);
2439
2440         /* Delay is only supported for shutdown/sleep */
2441         if (mm == INHIBIT_DELAY && (w & ~(INHIBIT_SHUTDOWN|INHIBIT_SLEEP)))
2442                 return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Delay inhibitors only supported for shutdown and sleep");
2443
2444         /* Don't allow taking delay locks while we are already
2445          * executing the operation. We shouldn't create the impression
2446          * that the lock was successful if the machine is about to go
2447          * down/suspend any moment. */
2448         if (m->action_what & w)
2449                 return sd_bus_error_setf(error, BUS_ERROR_OPERATION_IN_PROGRESS, "The operation inhibition has been requested for is already running");
2450
2451         r = bus_verify_polkit_async(
2452                         message,
2453                         CAP_SYS_BOOT,
2454                         w == INHIBIT_SHUTDOWN             ? (mm == INHIBIT_BLOCK ? "org.freedesktop.login1.inhibit-block-shutdown" : "org.freedesktop.login1.inhibit-delay-shutdown") :
2455                         w == INHIBIT_SLEEP                ? (mm == INHIBIT_BLOCK ? "org.freedesktop.login1.inhibit-block-sleep"    : "org.freedesktop.login1.inhibit-delay-sleep") :
2456                         w == INHIBIT_IDLE                 ? "org.freedesktop.login1.inhibit-block-idle" :
2457                         w == INHIBIT_HANDLE_POWER_KEY     ? "org.freedesktop.login1.inhibit-handle-power-key" :
2458                         w == INHIBIT_HANDLE_SUSPEND_KEY   ? "org.freedesktop.login1.inhibit-handle-suspend-key" :
2459                         w == INHIBIT_HANDLE_HIBERNATE_KEY ? "org.freedesktop.login1.inhibit-handle-hibernate-key" :
2460                                                             "org.freedesktop.login1.inhibit-handle-lid-switch",
2461                         NULL,
2462                         false,
2463                         UID_INVALID,
2464                         &m->polkit_registry,
2465                         error);
2466         if (r < 0)
2467                 return r;
2468         if (r == 0)
2469                 return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
2470
2471         r = sd_bus_query_sender_creds(message, SD_BUS_CREDS_EUID|SD_BUS_CREDS_PID, &creds);
2472         if (r < 0)
2473                 return r;
2474
2475         r = sd_bus_creds_get_euid(creds, &uid);
2476         if (r < 0)
2477                 return r;
2478
2479         r = sd_bus_creds_get_pid(creds, &pid);
2480         if (r < 0)
2481                 return r;
2482
2483         if (hashmap_size(m->inhibitors) >= m->inhibitors_max)
2484                 return sd_bus_error_setf(error, SD_BUS_ERROR_LIMITS_EXCEEDED, "Maximum number of inhibitors (%" PRIu64 ") reached, refusing further inhibitors.", m->inhibitors_max);
2485
2486         do {
2487                 id = mfree(id);
2488
2489                 if (asprintf(&id, "%lu", ++m->inhibit_counter) < 0)
2490                         return -ENOMEM;
2491
2492         } while (hashmap_get(m->inhibitors, id));
2493
2494         r = manager_add_inhibitor(m, id, &i);
2495         if (r < 0)
2496                 return r;
2497
2498         i->what = w;
2499         i->mode = mm;
2500         i->pid = pid;
2501         i->uid = uid;
2502         i->why = strdup(why);
2503         i->who = strdup(who);
2504
2505         if (!i->why || !i->who) {
2506                 r = -ENOMEM;
2507                 goto fail;
2508         }
2509
2510         fifo_fd = inhibitor_create_fifo(i);
2511         if (fifo_fd < 0) {
2512                 r = fifo_fd;
2513                 goto fail;
2514         }
2515
2516         inhibitor_start(i);
2517
2518         return sd_bus_reply_method_return(message, "h", fifo_fd);
2519
2520 fail:
2521         if (i)
2522                 inhibitor_free(i);
2523
2524         return r;
2525 }
2526
2527 const sd_bus_vtable manager_vtable[] = {
2528         SD_BUS_VTABLE_START(0),
2529
2530         SD_BUS_WRITABLE_PROPERTY("EnableWallMessages", "b", NULL, NULL, offsetof(Manager, enable_wall_messages), 0),
2531         SD_BUS_WRITABLE_PROPERTY("WallMessage", "s", NULL, NULL, offsetof(Manager, wall_message), 0),
2532
2533         SD_BUS_PROPERTY("NAutoVTs", "u", NULL, offsetof(Manager, n_autovts), SD_BUS_VTABLE_PROPERTY_CONST),
2534         SD_BUS_PROPERTY("KillOnlyUsers", "as", NULL, offsetof(Manager, kill_only_users), SD_BUS_VTABLE_PROPERTY_CONST),
2535         SD_BUS_PROPERTY("KillExcludeUsers", "as", NULL, offsetof(Manager, kill_exclude_users), SD_BUS_VTABLE_PROPERTY_CONST),
2536         SD_BUS_PROPERTY("KillUserProcesses", "b", NULL, offsetof(Manager, kill_user_processes), SD_BUS_VTABLE_PROPERTY_CONST),
2537         SD_BUS_PROPERTY("RebootToFirmwareSetup", "b", property_get_reboot_to_firmware_setup, 0, SD_BUS_VTABLE_PROPERTY_CONST),
2538         SD_BUS_PROPERTY("IdleHint", "b", property_get_idle_hint, 0, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
2539         SD_BUS_PROPERTY("IdleSinceHint", "t", property_get_idle_since_hint, 0, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
2540         SD_BUS_PROPERTY("IdleSinceHintMonotonic", "t", property_get_idle_since_hint, 0, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
2541         SD_BUS_PROPERTY("BlockInhibited", "s", property_get_inhibited, 0, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
2542         SD_BUS_PROPERTY("DelayInhibited", "s", property_get_inhibited, 0, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
2543         SD_BUS_PROPERTY("InhibitDelayMaxUSec", "t", NULL, offsetof(Manager, inhibit_delay_max), SD_BUS_VTABLE_PROPERTY_CONST),
2544         SD_BUS_PROPERTY("HandlePowerKey", "s", property_get_handle_action, offsetof(Manager, handle_power_key), SD_BUS_VTABLE_PROPERTY_CONST),
2545         SD_BUS_PROPERTY("HandleSuspendKey", "s", property_get_handle_action, offsetof(Manager, handle_suspend_key), SD_BUS_VTABLE_PROPERTY_CONST),
2546         SD_BUS_PROPERTY("HandleHibernateKey", "s", property_get_handle_action, offsetof(Manager, handle_hibernate_key), SD_BUS_VTABLE_PROPERTY_CONST),
2547         SD_BUS_PROPERTY("HandleLidSwitch", "s", property_get_handle_action, offsetof(Manager, handle_lid_switch), SD_BUS_VTABLE_PROPERTY_CONST),
2548         SD_BUS_PROPERTY("HandleLidSwitchDocked", "s", property_get_handle_action, offsetof(Manager, handle_lid_switch_docked), SD_BUS_VTABLE_PROPERTY_CONST),
2549         SD_BUS_PROPERTY("HoldoffTimeoutUSec", "t", NULL, offsetof(Manager, holdoff_timeout_usec), SD_BUS_VTABLE_PROPERTY_CONST),
2550         SD_BUS_PROPERTY("IdleAction", "s", property_get_handle_action, offsetof(Manager, idle_action), SD_BUS_VTABLE_PROPERTY_CONST),
2551         SD_BUS_PROPERTY("IdleActionUSec", "t", NULL, offsetof(Manager, idle_action_usec), SD_BUS_VTABLE_PROPERTY_CONST),
2552         SD_BUS_PROPERTY("PreparingForShutdown", "b", property_get_preparing, 0, 0),
2553         SD_BUS_PROPERTY("PreparingForSleep", "b", property_get_preparing, 0, 0),
2554         SD_BUS_PROPERTY("ScheduledShutdown", "(st)", property_get_scheduled_shutdown, 0, 0),
2555         SD_BUS_PROPERTY("Docked", "b", property_get_docked, 0, 0),
2556         SD_BUS_PROPERTY("RemoveIPC", "b", bus_property_get_bool, offsetof(Manager, remove_ipc), SD_BUS_VTABLE_PROPERTY_CONST),
2557         SD_BUS_PROPERTY("RuntimeDirectorySize", "t", bus_property_get_size, offsetof(Manager, runtime_dir_size), SD_BUS_VTABLE_PROPERTY_CONST),
2558         SD_BUS_PROPERTY("InhibitorsMax", "t", NULL, offsetof(Manager, inhibitors_max), SD_BUS_VTABLE_PROPERTY_CONST),
2559         SD_BUS_PROPERTY("NCurrentInhibitors", "t", property_get_current_inhibitors, 0, 0),
2560         SD_BUS_PROPERTY("SessionsMax", "t", NULL, offsetof(Manager, sessions_max), SD_BUS_VTABLE_PROPERTY_CONST),
2561         SD_BUS_PROPERTY("NCurrentSessions", "t", property_get_current_sessions, 0, 0),
2562         SD_BUS_PROPERTY("UserTasksMax", "t", NULL, offsetof(Manager, user_tasks_max), SD_BUS_VTABLE_PROPERTY_CONST),
2563
2564         SD_BUS_METHOD("GetSession", "s", "o", method_get_session, SD_BUS_VTABLE_UNPRIVILEGED),
2565         SD_BUS_METHOD("GetSessionByPID", "u", "o", method_get_session_by_pid, SD_BUS_VTABLE_UNPRIVILEGED),
2566         SD_BUS_METHOD("GetUser", "u", "o", method_get_user, SD_BUS_VTABLE_UNPRIVILEGED),
2567         SD_BUS_METHOD("GetUserByPID", "u", "o", method_get_user_by_pid, SD_BUS_VTABLE_UNPRIVILEGED),
2568         SD_BUS_METHOD("GetSeat", "s", "o", method_get_seat, SD_BUS_VTABLE_UNPRIVILEGED),
2569         SD_BUS_METHOD("ListSessions", NULL, "a(susso)", method_list_sessions, SD_BUS_VTABLE_UNPRIVILEGED),
2570         SD_BUS_METHOD("ListUsers", NULL, "a(uso)", method_list_users, SD_BUS_VTABLE_UNPRIVILEGED),
2571         SD_BUS_METHOD("ListSeats", NULL, "a(so)", method_list_seats, SD_BUS_VTABLE_UNPRIVILEGED),
2572         SD_BUS_METHOD("ListInhibitors", NULL, "a(ssssuu)", method_list_inhibitors, SD_BUS_VTABLE_UNPRIVILEGED),
2573         SD_BUS_METHOD("CreateSession", "uusssssussbssa(sv)", "soshusub", method_create_session, 0),
2574         SD_BUS_METHOD("ReleaseSession", "s", NULL, method_release_session, 0),
2575         SD_BUS_METHOD("ActivateSession", "s", NULL, method_activate_session, SD_BUS_VTABLE_UNPRIVILEGED),
2576         SD_BUS_METHOD("ActivateSessionOnSeat", "ss", NULL, method_activate_session_on_seat, SD_BUS_VTABLE_UNPRIVILEGED),
2577         SD_BUS_METHOD("LockSession", "s", NULL, method_lock_session, SD_BUS_VTABLE_UNPRIVILEGED),
2578         SD_BUS_METHOD("UnlockSession", "s", NULL, method_lock_session, SD_BUS_VTABLE_UNPRIVILEGED),
2579         SD_BUS_METHOD("LockSessions", NULL, NULL, method_lock_sessions, SD_BUS_VTABLE_UNPRIVILEGED),
2580         SD_BUS_METHOD("UnlockSessions", NULL, NULL, method_lock_sessions, SD_BUS_VTABLE_UNPRIVILEGED),
2581         SD_BUS_METHOD("KillSession", "ssi", NULL, method_kill_session, SD_BUS_VTABLE_UNPRIVILEGED),
2582         SD_BUS_METHOD("KillUser", "ui", NULL, method_kill_user, SD_BUS_VTABLE_UNPRIVILEGED),
2583         SD_BUS_METHOD("TerminateSession", "s", NULL, method_terminate_session, SD_BUS_VTABLE_UNPRIVILEGED),
2584         SD_BUS_METHOD("TerminateUser", "u", NULL, method_terminate_user, SD_BUS_VTABLE_UNPRIVILEGED),
2585         SD_BUS_METHOD("TerminateSeat", "s", NULL, method_terminate_seat, SD_BUS_VTABLE_UNPRIVILEGED),
2586         SD_BUS_METHOD("SetUserLinger", "ubb", NULL, method_set_user_linger, SD_BUS_VTABLE_UNPRIVILEGED),
2587         SD_BUS_METHOD("AttachDevice", "ssb", NULL, method_attach_device, SD_BUS_VTABLE_UNPRIVILEGED),
2588         SD_BUS_METHOD("FlushDevices", "b", NULL, method_flush_devices, SD_BUS_VTABLE_UNPRIVILEGED),
2589         SD_BUS_METHOD("PowerOff", "b", NULL, method_poweroff, SD_BUS_VTABLE_UNPRIVILEGED),
2590         SD_BUS_METHOD("Reboot", "b", NULL, method_reboot, SD_BUS_VTABLE_UNPRIVILEGED),
2591         SD_BUS_METHOD("Suspend", "b", NULL, method_suspend, SD_BUS_VTABLE_UNPRIVILEGED),
2592         SD_BUS_METHOD("Hibernate", "b", NULL, method_hibernate, SD_BUS_VTABLE_UNPRIVILEGED),
2593         SD_BUS_METHOD("HybridSleep", "b", NULL, method_hybrid_sleep, SD_BUS_VTABLE_UNPRIVILEGED),
2594         SD_BUS_METHOD("CanPowerOff", NULL, "s", method_can_poweroff, SD_BUS_VTABLE_UNPRIVILEGED),
2595         SD_BUS_METHOD("CanReboot", NULL, "s", method_can_reboot, SD_BUS_VTABLE_UNPRIVILEGED),
2596         SD_BUS_METHOD("CanSuspend", NULL, "s", method_can_suspend, SD_BUS_VTABLE_UNPRIVILEGED),
2597         SD_BUS_METHOD("CanHibernate", NULL, "s", method_can_hibernate, SD_BUS_VTABLE_UNPRIVILEGED),
2598         SD_BUS_METHOD("CanHybridSleep", NULL, "s", method_can_hybrid_sleep, SD_BUS_VTABLE_UNPRIVILEGED),
2599         SD_BUS_METHOD("ScheduleShutdown", "st", NULL, method_schedule_shutdown, SD_BUS_VTABLE_UNPRIVILEGED),
2600         SD_BUS_METHOD("CancelScheduledShutdown", NULL, "b", method_cancel_scheduled_shutdown, SD_BUS_VTABLE_UNPRIVILEGED),
2601         SD_BUS_METHOD("Inhibit", "ssss", "h", method_inhibit, SD_BUS_VTABLE_UNPRIVILEGED),
2602         SD_BUS_METHOD("CanRebootToFirmwareSetup", NULL, "s", method_can_reboot_to_firmware_setup, SD_BUS_VTABLE_UNPRIVILEGED),
2603         SD_BUS_METHOD("SetRebootToFirmwareSetup", "b", NULL, method_set_reboot_to_firmware_setup, SD_BUS_VTABLE_UNPRIVILEGED),
2604         SD_BUS_METHOD("SetWallMessage", "sb", NULL, method_set_wall_message, SD_BUS_VTABLE_UNPRIVILEGED),
2605
2606         SD_BUS_SIGNAL("SessionNew", "so", 0),
2607         SD_BUS_SIGNAL("SessionRemoved", "so", 0),
2608         SD_BUS_SIGNAL("UserNew", "uo", 0),
2609         SD_BUS_SIGNAL("UserRemoved", "uo", 0),
2610         SD_BUS_SIGNAL("SeatNew", "so", 0),
2611         SD_BUS_SIGNAL("SeatRemoved", "so", 0),
2612         SD_BUS_SIGNAL("PrepareForShutdown", "b", 0),
2613         SD_BUS_SIGNAL("PrepareForSleep", "b", 0),
2614
2615         SD_BUS_VTABLE_END
2616 };
2617
2618 static int session_jobs_reply(Session *s, const char *unit, const char *result) {
2619         int r = 0;
2620
2621         assert(s);
2622         assert(unit);
2623
2624         if (!s->started)
2625                 return r;
2626
2627         if (streq(result, "done"))
2628                 r = session_send_create_reply(s, NULL);
2629         else {
2630                 _cleanup_(sd_bus_error_free) sd_bus_error e = SD_BUS_ERROR_NULL;
2631
2632                 sd_bus_error_setf(&e, BUS_ERROR_JOB_FAILED, "Start job for unit %s failed with '%s'", unit, result);
2633                 r = session_send_create_reply(s, &e);
2634         }
2635
2636         return r;
2637 }
2638
2639 int match_job_removed(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2640         const char *path, *result, *unit;
2641         Manager *m = userdata;
2642         Session *session;
2643         uint32_t id;
2644         User *user;
2645         int r;
2646
2647         assert(message);
2648         assert(m);
2649
2650         r = sd_bus_message_read(message, "uoss", &id, &path, &unit, &result);
2651         if (r < 0) {
2652                 bus_log_parse_error(r);
2653                 return 0;
2654         }
2655
2656         if (m->action_job && streq(m->action_job, path)) {
2657                 log_info("Operation '%s' finished.", inhibit_what_to_string(m->action_what));
2658
2659                 /* Tell people that they now may take a lock again */
2660                 send_prepare_for(m, m->action_what, false);
2661
2662                 m->action_job = mfree(m->action_job);
2663                 m->action_unit = NULL;
2664                 m->action_what = 0;
2665                 return 0;
2666         }
2667
2668         session = hashmap_get(m->session_units, unit);
2669         if (session && streq_ptr(path, session->scope_job)) {
2670                 session->scope_job = mfree(session->scope_job);
2671                 session_jobs_reply(session, unit, result);
2672
2673                 session_save(session);
2674                 user_save(session->user);
2675                 session_add_to_gc_queue(session);
2676         }
2677
2678         user = hashmap_get(m->user_units, unit);
2679         if (user &&
2680             (streq_ptr(path, user->service_job) ||
2681              streq_ptr(path, user->slice_job))) {
2682
2683                 if (streq_ptr(path, user->service_job))
2684                         user->service_job = mfree(user->service_job);
2685
2686                 if (streq_ptr(path, user->slice_job))
2687                         user->slice_job = mfree(user->slice_job);
2688
2689                 LIST_FOREACH(sessions_by_user, session, user->sessions)
2690                         session_jobs_reply(session, unit, result);
2691
2692                 user_save(user);
2693                 user_add_to_gc_queue(user);
2694         }
2695
2696         return 0;
2697 }
2698
2699 int match_unit_removed(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2700         const char *path, *unit;
2701         Manager *m = userdata;
2702         Session *session;
2703         User *user;
2704         int r;
2705
2706         assert(message);
2707         assert(m);
2708
2709         r = sd_bus_message_read(message, "so", &unit, &path);
2710         if (r < 0) {
2711                 bus_log_parse_error(r);
2712                 return 0;
2713         }
2714
2715         session = hashmap_get(m->session_units, unit);
2716         if (session)
2717                 session_add_to_gc_queue(session);
2718
2719         user = hashmap_get(m->user_units, unit);
2720         if (user)
2721                 user_add_to_gc_queue(user);
2722
2723         return 0;
2724 }
2725
2726 int match_properties_changed(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2727         _cleanup_free_ char *unit = NULL;
2728         Manager *m = userdata;
2729         const char *path;
2730         Session *session;
2731         User *user;
2732         int r;
2733
2734         assert(message);
2735         assert(m);
2736
2737         path = sd_bus_message_get_path(message);
2738         if (!path)
2739                 return 0;
2740
2741         r = unit_name_from_dbus_path(path, &unit);
2742         if (r == -EINVAL) /* not a unit */
2743                 return 0;
2744         if (r < 0) {
2745                 log_oom();
2746                 return 0;
2747         }
2748
2749         session = hashmap_get(m->session_units, unit);
2750         if (session)
2751                 session_add_to_gc_queue(session);
2752
2753         user = hashmap_get(m->user_units, unit);
2754         if (user)
2755                 user_add_to_gc_queue(user);
2756
2757         return 0;
2758 }
2759
2760 int match_reloading(sd_bus_message *message, void *userdata, sd_bus_error *error) {
2761         Manager *m = userdata;
2762         Session *session;
2763         Iterator i;
2764         int b, r;
2765
2766         assert(message);
2767         assert(m);
2768
2769         r = sd_bus_message_read(message, "b", &b);
2770         if (r < 0) {
2771                 bus_log_parse_error(r);
2772                 return 0;
2773         }
2774
2775         if (b)
2776                 return 0;
2777
2778         /* systemd finished reloading, let's recheck all our sessions */
2779         log_debug("System manager has been reloaded, rechecking sessions...");
2780
2781         HASHMAP_FOREACH(session, m->sessions, i)
2782                 session_add_to_gc_queue(session);
2783
2784         return 0;
2785 }
2786
2787 int manager_send_changed(Manager *manager, const char *property, ...) {
2788         char **l;
2789
2790         assert(manager);
2791
2792         l = strv_from_stdarg_alloca(property);
2793
2794         return sd_bus_emit_properties_changed_strv(
2795                         manager->bus,
2796                         "/org/freedesktop/login1",
2797                         "org.freedesktop.login1.Manager",
2798                         l);
2799 }
2800
2801 static int strdup_job(sd_bus_message *reply, char **job) {
2802         const char *j;
2803         char *copy;
2804         int r;
2805
2806         r = sd_bus_message_read(reply, "o", &j);
2807         if (r < 0)
2808                 return r;
2809
2810         copy = strdup(j);
2811         if (!copy)
2812                 return -ENOMEM;
2813
2814         *job = copy;
2815         return 1;
2816 }
2817
2818 int manager_start_slice(
2819                 Manager *manager,
2820                 const char *slice,
2821                 const char *description,
2822                 const char *after,
2823                 const char *after2,
2824                 uint64_t tasks_max,
2825                 sd_bus_error *error,
2826                 char **job) {
2827
2828         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL, *reply = NULL;
2829         int r;
2830
2831         assert(manager);
2832         assert(slice);
2833         assert(job);
2834
2835         r = sd_bus_message_new_method_call(
2836                         manager->bus,
2837                         &m,
2838                         "org.freedesktop.systemd1",
2839                         "/org/freedesktop/systemd1",
2840                         "org.freedesktop.systemd1.Manager",
2841                         "StartTransientUnit");
2842         if (r < 0)
2843                 return r;
2844
2845         r = sd_bus_message_append(m, "ss", strempty(slice), "fail");
2846         if (r < 0)
2847                 return r;
2848
2849         r = sd_bus_message_open_container(m, 'a', "(sv)");
2850         if (r < 0)
2851                 return r;
2852
2853         if (!isempty(description)) {
2854                 r = sd_bus_message_append(m, "(sv)", "Description", "s", description);
2855                 if (r < 0)
2856                         return r;
2857         }
2858
2859         if (!isempty(after)) {
2860                 r = sd_bus_message_append(m, "(sv)", "After", "as", 1, after);
2861                 if (r < 0)
2862                         return r;
2863         }
2864
2865         if (!isempty(after2)) {
2866                 r = sd_bus_message_append(m, "(sv)", "After", "as", 1, after2);
2867                 if (r < 0)
2868                         return r;
2869         }
2870
2871         r = sd_bus_message_append(m, "(sv)", "TasksMax", "t", tasks_max);
2872         if (r < 0)
2873                 return r;
2874
2875         r = sd_bus_message_close_container(m);
2876         if (r < 0)
2877                 return r;
2878
2879         r = sd_bus_message_append(m, "a(sa(sv))", 0);
2880         if (r < 0)
2881                 return r;
2882
2883         r = sd_bus_call(manager->bus, m, 0, error, &reply);
2884         if (r < 0)
2885                 return r;
2886
2887         return strdup_job(reply, job);
2888 }
2889
2890 int manager_start_scope(
2891                 Manager *manager,
2892                 const char *scope,
2893                 pid_t pid,
2894                 const char *slice,
2895                 const char *description,
2896                 const char *after,
2897                 const char *after2,
2898                 uint64_t tasks_max,
2899                 sd_bus_error *error,
2900                 char **job) {
2901
2902         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL, *reply = NULL;
2903         int r;
2904
2905         assert(manager);
2906         assert(scope);
2907         assert(pid > 1);
2908         assert(job);
2909
2910         r = sd_bus_message_new_method_call(
2911                         manager->bus,
2912                         &m,
2913                         "org.freedesktop.systemd1",
2914                         "/org/freedesktop/systemd1",
2915                         "org.freedesktop.systemd1.Manager",
2916                         "StartTransientUnit");
2917         if (r < 0)
2918                 return r;
2919
2920         r = sd_bus_message_append(m, "ss", strempty(scope), "fail");
2921         if (r < 0)
2922                 return r;
2923
2924         r = sd_bus_message_open_container(m, 'a', "(sv)");
2925         if (r < 0)
2926                 return r;
2927
2928         if (!isempty(slice)) {
2929                 r = sd_bus_message_append(m, "(sv)", "Slice", "s", slice);
2930                 if (r < 0)
2931                         return r;
2932         }
2933
2934         if (!isempty(description)) {
2935                 r = sd_bus_message_append(m, "(sv)", "Description", "s", description);
2936                 if (r < 0)
2937                         return r;
2938         }
2939
2940         if (!isempty(after)) {
2941                 r = sd_bus_message_append(m, "(sv)", "After", "as", 1, after);
2942                 if (r < 0)
2943                         return r;
2944         }
2945
2946         if (!isempty(after2)) {
2947                 r = sd_bus_message_append(m, "(sv)", "After", "as", 1, after2);
2948                 if (r < 0)
2949                         return r;
2950         }
2951
2952         /* cgroup empty notification is not available in containers
2953          * currently. To make this less problematic, let's shorten the
2954          * stop timeout for sessions, so that we don't wait
2955          * forever. */
2956
2957         /* Make sure that the session shells are terminated with
2958          * SIGHUP since bash and friends tend to ignore SIGTERM */
2959         r = sd_bus_message_append(m, "(sv)", "SendSIGHUP", "b", true);
2960         if (r < 0)
2961                 return r;
2962
2963         r = sd_bus_message_append(m, "(sv)", "PIDs", "au", 1, pid);
2964         if (r < 0)
2965                 return r;
2966
2967         r = sd_bus_message_append(m, "(sv)", "TasksMax", "t", tasks_max);
2968         if (r < 0)
2969                 return r;
2970
2971         r = sd_bus_message_close_container(m);
2972         if (r < 0)
2973                 return r;
2974
2975         r = sd_bus_message_append(m, "a(sa(sv))", 0);
2976         if (r < 0)
2977                 return r;
2978
2979         r = sd_bus_call(manager->bus, m, 0, error, &reply);
2980         if (r < 0)
2981                 return r;
2982
2983         return strdup_job(reply, job);
2984 }
2985
2986 int manager_start_unit(Manager *manager, const char *unit, sd_bus_error *error, char **job) {
2987         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
2988         int r;
2989
2990         assert(manager);
2991         assert(unit);
2992         assert(job);
2993
2994         r = sd_bus_call_method(
2995                         manager->bus,
2996                         "org.freedesktop.systemd1",
2997                         "/org/freedesktop/systemd1",
2998                         "org.freedesktop.systemd1.Manager",
2999                         "StartUnit",
3000                         error,
3001                         &reply,
3002                         "ss", unit, "replace");
3003         if (r < 0)
3004                 return r;
3005
3006         return strdup_job(reply, job);
3007 }
3008
3009 int manager_stop_unit(Manager *manager, const char *unit, sd_bus_error *error, char **job) {
3010         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
3011         int r;
3012
3013         assert(manager);
3014         assert(unit);
3015         assert(job);
3016
3017         r = sd_bus_call_method(
3018                         manager->bus,
3019                         "org.freedesktop.systemd1",
3020                         "/org/freedesktop/systemd1",
3021                         "org.freedesktop.systemd1.Manager",
3022                         "StopUnit",
3023                         error,
3024                         &reply,
3025                         "ss", unit, "fail");
3026         if (r < 0) {
3027                 if (sd_bus_error_has_name(error, BUS_ERROR_NO_SUCH_UNIT) ||
3028                     sd_bus_error_has_name(error, BUS_ERROR_LOAD_FAILED)) {
3029
3030                         *job = NULL;
3031                         sd_bus_error_free(error);
3032                         return 0;
3033                 }
3034
3035                 return r;
3036         }
3037
3038         return strdup_job(reply, job);
3039 }
3040
3041 int manager_abandon_scope(Manager *manager, const char *scope, sd_bus_error *error) {
3042         _cleanup_free_ char *path = NULL;
3043         int r;
3044
3045         assert(manager);
3046         assert(scope);
3047
3048         path = unit_dbus_path_from_name(scope);
3049         if (!path)
3050                 return -ENOMEM;
3051
3052         r = sd_bus_call_method(
3053                         manager->bus,
3054                         "org.freedesktop.systemd1",
3055                         path,
3056                         "org.freedesktop.systemd1.Scope",
3057                         "Abandon",
3058                         error,
3059                         NULL,
3060                         NULL);
3061         if (r < 0) {
3062                 if (sd_bus_error_has_name(error, BUS_ERROR_NO_SUCH_UNIT) ||
3063                     sd_bus_error_has_name(error, BUS_ERROR_LOAD_FAILED) ||
3064                     sd_bus_error_has_name(error, BUS_ERROR_SCOPE_NOT_RUNNING)) {
3065                         sd_bus_error_free(error);
3066                         return 0;
3067                 }
3068
3069                 return r;
3070         }
3071
3072         return 1;
3073 }
3074
3075 int manager_kill_unit(Manager *manager, const char *unit, KillWho who, int signo, sd_bus_error *error) {
3076         assert(manager);
3077         assert(unit);
3078
3079         return sd_bus_call_method(
3080                         manager->bus,
3081                         "org.freedesktop.systemd1",
3082                         "/org/freedesktop/systemd1",
3083                         "org.freedesktop.systemd1.Manager",
3084                         "KillUnit",
3085                         error,
3086                         NULL,
3087                         "ssi", unit, who == KILL_LEADER ? "main" : "all", signo);
3088 }
3089
3090 int manager_unit_is_active(Manager *manager, const char *unit) {
3091         _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
3092         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
3093         _cleanup_free_ char *path = NULL;
3094         const char *state;
3095         int r;
3096
3097         assert(manager);
3098         assert(unit);
3099
3100         path = unit_dbus_path_from_name(unit);
3101         if (!path)
3102                 return -ENOMEM;
3103
3104         r = sd_bus_get_property(
3105                         manager->bus,
3106                         "org.freedesktop.systemd1",
3107                         path,
3108                         "org.freedesktop.systemd1.Unit",
3109                         "ActiveState",
3110                         &error,
3111                         &reply,
3112                         "s");
3113         if (r < 0) {
3114                 /* systemd might have droppped off momentarily, let's
3115                  * not make this an error */
3116                 if (sd_bus_error_has_name(&error, SD_BUS_ERROR_NO_REPLY) ||
3117                     sd_bus_error_has_name(&error, SD_BUS_ERROR_DISCONNECTED))
3118                         return true;
3119
3120                 /* If the unit is already unloaded then it's not
3121                  * active */
3122                 if (sd_bus_error_has_name(&error, BUS_ERROR_NO_SUCH_UNIT) ||
3123                     sd_bus_error_has_name(&error, BUS_ERROR_LOAD_FAILED))
3124                         return false;
3125
3126                 return r;
3127         }
3128
3129         r = sd_bus_message_read(reply, "s", &state);
3130         if (r < 0)
3131                 return -EINVAL;
3132
3133         return !streq(state, "inactive") && !streq(state, "failed");
3134 }
3135
3136 int manager_job_is_active(Manager *manager, const char *path) {
3137         _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
3138         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
3139         int r;
3140
3141         assert(manager);
3142         assert(path);
3143
3144         r = sd_bus_get_property(
3145                         manager->bus,
3146                         "org.freedesktop.systemd1",
3147                         path,
3148                         "org.freedesktop.systemd1.Job",
3149                         "State",
3150                         &error,
3151                         &reply,
3152                         "s");
3153         if (r < 0) {
3154                 if (sd_bus_error_has_name(&error, SD_BUS_ERROR_NO_REPLY) ||
3155                     sd_bus_error_has_name(&error, SD_BUS_ERROR_DISCONNECTED))
3156                         return true;
3157
3158                 if (sd_bus_error_has_name(&error, SD_BUS_ERROR_UNKNOWN_OBJECT))
3159                         return false;
3160
3161                 return r;
3162         }
3163
3164         /* We don't actually care about the state really. The fact
3165          * that we could read the job state is enough for us */
3166
3167         return true;
3168 }