src/login/logind-core.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 <fcntl.h>
  21 #include <pwd.h>
  22 #include <sys/ioctl.h>
  23 #include <sys/types.h>
  24 #include <linux/vt.h>
  25
  26 #include "alloc-util.h"
  27 #include "bus-error.h"
  28 #include "bus-util.h"
  29 #include "cgroup-util.h"
  30 #include "fd-util.h"
  31 #include "logind.h"
  32 #include "strv.h"
  33 #include "terminal-util.h"
  34 #include "udev-util.h"
  35 #include "user-util.h"
  36
  37 int manager_add_device(Manager *m, const char *sysfs, bool master, Device **_device) {
  38         Device *d;
  39
  40         assert(m);
  41         assert(sysfs);
  42
  43         d = hashmap_get(m->devices, sysfs);
  44         if (d)
  45                 /* we support adding master-flags, but not removing them */
  46                 d->master = d->master || master;
  47         else {
  48                 d = device_new(m, sysfs, master);
  49                 if (!d)
  50                         return -ENOMEM;
  51         }
  52
  53         if (_device)
  54                 *_device = d;
  55
  56         return 0;
  57 }
  58
  59 int manager_add_seat(Manager *m, const char *id, Seat **_seat) {
  60         Seat *s;
  61
  62         assert(m);
  63         assert(id);
  64
  65         s = hashmap_get(m->seats, id);
  66         if (!s) {
  67                 s = seat_new(m, id);
  68                 if (!s)
  69                         return -ENOMEM;
  70         }
  71
  72         if (_seat)
  73                 *_seat = s;
  74
  75         return 0;
  76 }
  77
  78 int manager_add_session(Manager *m, const char *id, Session **_session) {
  79         Session *s;
  80
  81         assert(m);
  82         assert(id);
  83
  84         s = hashmap_get(m->sessions, id);
  85         if (!s) {
  86                 s = session_new(m, id);
  87                 if (!s)
  88                         return -ENOMEM;
  89         }
  90
  91         if (_session)
  92                 *_session = s;
  93
  94         return 0;
  95 }
  96
  97 int manager_add_user(Manager *m, uid_t uid, gid_t gid, const char *name, User **_user) {
  98         User *u;
  99         int r;
 100
 101         assert(m);
 102         assert(name);
 103
 104         u = hashmap_get(m->users, UID_TO_PTR(uid));
 105         if (!u) {
 106                 r = user_new(&u, m, uid, gid, name);
 107                 if (r < 0)
 108                         return r;
 109         }
 110
 111         if (_user)
 112                 *_user = u;
 113
 114         return 0;
 115 }
 116
 117 int manager_add_user_by_name(Manager *m, const char *name, User **_user) {
 118         uid_t uid;
 119         gid_t gid;
 120         int r;
 121
 122         assert(m);
 123         assert(name);
 124
 125         r = get_user_creds(&name, &uid, &gid, NULL, NULL);
 126         if (r < 0)
 127                 return r;
 128
 129         return manager_add_user(m, uid, gid, name, _user);
 130 }
 131
 132 int manager_add_user_by_uid(Manager *m, uid_t uid, User **_user) {
 133         struct passwd *p;
 134
 135         assert(m);
 136
 137         errno = 0;
 138         p = getpwuid(uid);
 139         if (!p)
 140                 return errno > 0 ? -errno : -ENOENT;
 141
 142         return manager_add_user(m, uid, p->pw_gid, p->pw_name, _user);
 143 }
 144
 145 int manager_add_inhibitor(Manager *m, const char* id, Inhibitor **_inhibitor) {
 146         Inhibitor *i;
 147
 148         assert(m);
 149         assert(id);
 150
 151         i = hashmap_get(m->inhibitors, id);
 152         if (i) {
 153                 if (_inhibitor)
 154                         *_inhibitor = i;
 155
 156                 return 0;
 157         }
 158
 159         i = inhibitor_new(m, id);
 160         if (!i)
 161                 return -ENOMEM;
 162
 163         if (_inhibitor)
 164                 *_inhibitor = i;
 165
 166         return 0;
 167 }
 168
 169 int manager_add_button(Manager *m, const char *name, Button **_button) {
 170         Button *b;
 171
 172         assert(m);
 173         assert(name);
 174
 175         b = hashmap_get(m->buttons, name);
 176         if (!b) {
 177                 b = button_new(m, name);
 178                 if (!b)
 179                         return -ENOMEM;
 180         }
 181
 182         if (_button)
 183                 *_button = b;
 184
 185         return 0;
 186 }
 187
 188 int manager_process_seat_device(Manager *m, struct udev_device *d) {
 189         Device *device;
 190         int r;
 191
 192         assert(m);
 193
 194         if (streq_ptr(udev_device_get_action(d), "remove")) {
 195
 196                 device = hashmap_get(m->devices, udev_device_get_syspath(d));
 197                 if (!device)
 198                         return 0;
 199
 200                 seat_add_to_gc_queue(device->seat);
 201                 device_free(device);
 202
 203         } else {
 204                 const char *sn;
 205                 Seat *seat = NULL;
 206                 bool master;
 207
 208                 sn = udev_device_get_property_value(d, "ID_SEAT");
 209                 if (isempty(sn))
 210                         sn = "seat0";
 211
 212                 if (!seat_name_is_valid(sn)) {
 213                         log_warning("Device with invalid seat name %s found, ignoring.", sn);
 214                         return 0;
 215                 }
 216
 217                 seat = hashmap_get(m->seats, sn);
 218                 master = udev_device_has_tag(d, "master-of-seat");
 219
 220                 /* Ignore non-master devices for unknown seats */
 221                 if (!master && !seat)
 222                         return 0;
 223
 224                 r = manager_add_device(m, udev_device_get_syspath(d), master, &device);
 225                 if (r < 0)
 226                         return r;
 227
 228                 if (!seat) {
 229                         r = manager_add_seat(m, sn, &seat);
 230                         if (r < 0) {
 231                                 if (!device->seat)
 232                                         device_free(device);
 233
 234                                 return r;
 235                         }
 236                 }
 237
 238                 device_attach(device, seat);
 239                 seat_start(seat);
 240         }
 241
 242         return 0;
 243 }
 244
 245 int manager_process_button_device(Manager *m, struct udev_device *d) {
 246         Button *b;
 247
 248         int r;
 249
 250         assert(m);
 251
 252         if (streq_ptr(udev_device_get_action(d), "remove")) {
 253
 254                 b = hashmap_get(m->buttons, udev_device_get_sysname(d));
 255                 if (!b)
 256                         return 0;
 257
 258                 button_free(b);
 259
 260         } else {
 261                 const char *sn;
 262
 263                 r = manager_add_button(m, udev_device_get_sysname(d), &b);
 264                 if (r < 0)
 265                         return r;
 266
 267                 sn = udev_device_get_property_value(d, "ID_SEAT");
 268                 if (isempty(sn))
 269                         sn = "seat0";
 270
 271                 button_set_seat(b, sn);
 272                 button_open(b);
 273         }
 274
 275         return 0;
 276 }
 277
 278 int manager_get_session_by_pid(Manager *m, pid_t pid, Session **session) {
 279         _cleanup_free_ char *unit = NULL;
 280         Session *s;
 281         int r;
 282
 283         assert(m);
 284
 285         if (pid < 1)
 286                 return -EINVAL;
 287
 288         r = cg_pid_get_unit(pid, &unit);
 289         if (r < 0)
 290                 return 0;
 291
 292         s = hashmap_get(m->session_units, unit);
 293         if (!s)
 294                 return 0;
 295
 296         if (session)
 297                 *session = s;
 298         return 1;
 299 }
 300
 301 int manager_get_user_by_pid(Manager *m, pid_t pid, User **user) {
 302         _cleanup_free_ char *unit = NULL;
 303         User *u;
 304         int r;
 305
 306         assert(m);
 307         assert(user);
 308
 309         if (pid < 1)
 310                 return -EINVAL;
 311
 312         r = cg_pid_get_slice(pid, &unit);
 313         if (r < 0)
 314                 return 0;
 315
 316         u = hashmap_get(m->user_units, unit);
 317         if (!u)
 318                 return 0;
 319
 320         *user = u;
 321         return 1;
 322 }
 323
 324 int manager_get_idle_hint(Manager *m, dual_timestamp *t) {
 325         Session *s;
 326         bool idle_hint;
 327         dual_timestamp ts = DUAL_TIMESTAMP_NULL;
 328         Iterator i;
 329
 330         assert(m);
 331
 332         idle_hint = !manager_is_inhibited(m, INHIBIT_IDLE, INHIBIT_BLOCK, t, false, false, 0, NULL);
 333
 334         HASHMAP_FOREACH(s, m->sessions, i) {
 335                 dual_timestamp k;
 336                 int ih;
 337
 338                 ih = session_get_idle_hint(s, &k);
 339                 if (ih < 0)
 340                         return ih;
 341
 342                 if (!ih) {
 343                         if (!idle_hint) {
 344                                 if (k.monotonic < ts.monotonic)
 345                                         ts = k;
 346                         } else {
 347                                 idle_hint = false;
 348                                 ts = k;
 349                         }
 350                 } else if (idle_hint) {
 351
 352                         if (k.monotonic > ts.monotonic)
 353                                 ts = k;
 354                 }
 355         }
 356
 357         if (t)
 358                 *t = ts;
 359
 360         return idle_hint;
 361 }
 362
 363 bool manager_shall_kill(Manager *m, const char *user) {
 364         assert(m);
 365         assert(user);
 366
 367         if (!m->kill_exclude_users && streq(user, "root"))
 368                 return false;
 369
 370         if (strv_contains(m->kill_exclude_users, user))
 371                 return false;
 372
 373         if (!strv_isempty(m->kill_only_users))
 374                 return strv_contains(m->kill_only_users, user);
 375
 376         return m->kill_user_processes;
 377 }
 378
 379 static int vt_is_busy(unsigned int vtnr) {
 380         struct vt_stat vt_stat;
 381         int r = 0;
 382         _cleanup_close_ int fd;
 383
 384         assert(vtnr >= 1);
 385
 386         /* We explicitly open /dev/tty1 here instead of /dev/tty0. If
 387          * we'd open the latter we'd open the foreground tty which
 388          * hence would be unconditionally busy. By opening /dev/tty1
 389          * we avoid this. Since tty1 is special and needs to be an
 390          * explicitly loaded getty or DM this is safe. */
 391
 392         fd = open_terminal("/dev/tty1", O_RDWR|O_NOCTTY|O_CLOEXEC);
 393         if (fd < 0)
 394                 return -errno;
 395
 396         if (ioctl(fd, VT_GETSTATE, &vt_stat) < 0)
 397                 r = -errno;
 398         else
 399                 r = !!(vt_stat.v_state & (1 << vtnr));
 400
 401         return r;
 402 }
 403
 404 int manager_spawn_autovt(Manager *m, unsigned int vtnr) {
 405         _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
 406         char name[sizeof("autovt@tty.service") + DECIMAL_STR_MAX(unsigned int)];
 407         int r;
 408
 409         assert(m);
 410         assert(vtnr >= 1);
 411
 412         if (vtnr > m->n_autovts &&
 413             vtnr != m->reserve_vt)
 414                 return 0;
 415
 416         if (vtnr != m->reserve_vt) {
 417                 /* If this is the reserved TTY, we'll start the getty
 418                  * on it in any case, but otherwise only if it is not
 419                  * busy. */
 420
 421                 r = vt_is_busy(vtnr);
 422                 if (r < 0)
 423                         return r;
 424                 else if (r > 0)
 425                         return -EBUSY;
 426         }
 427
 428         snprintf(name, sizeof(name), "autovt@tty%u.service", vtnr);
 429         r = sd_bus_call_method(
 430                         m->bus,
 431                         "org.freedesktop.systemd1",
 432                         "/org/freedesktop/systemd1",
 433                         "org.freedesktop.systemd1.Manager",
 434                         "StartUnit",
 435                         &error,
 436                         NULL,
 437                         "ss", name, "fail");
 438         if (r < 0)
 439                 log_error("Failed to start %s: %s", name, bus_error_message(&error, r));
 440
 441         return r;
 442 }
 443
 444 static bool manager_is_docked(Manager *m) {
 445         Iterator i;
 446         Button *b;
 447
 448         HASHMAP_FOREACH(b, m->buttons, i)
 449                 if (b->docked)
 450                         return true;
 451
 452         return false;
 453 }
 454
 455 static int manager_count_external_displays(Manager *m) {
 456         _cleanup_udev_enumerate_unref_ struct udev_enumerate *e = NULL;
 457         struct udev_list_entry *item = NULL, *first = NULL;
 458         int r;
 459         int n = 0;
 460
 461         e = udev_enumerate_new(m->udev);
 462         if (!e)
 463                 return -ENOMEM;
 464
 465         r = udev_enumerate_add_match_subsystem(e, "drm");
 466         if (r < 0)
 467                 return r;
 468
 469         r = udev_enumerate_scan_devices(e);
 470         if (r < 0)
 471                 return r;
 472
 473         first = udev_enumerate_get_list_entry(e);
 474         udev_list_entry_foreach(item, first) {
 475                 _cleanup_udev_device_unref_ struct udev_device *d = NULL;
 476                 struct udev_device *p;
 477                 const char *status, *enabled, *dash, *nn, *i;
 478                 bool external = false;
 479
 480                 d = udev_device_new_from_syspath(m->udev, udev_list_entry_get_name(item));
 481                 if (!d)
 482                         return -ENOMEM;
 483
 484                 p = udev_device_get_parent(d);
 485                 if (!p)
 486                         continue;
 487
 488                 /* If the parent shares the same subsystem as the
 489                  * device we are looking at then it is a connector,
 490                  * which is what we are interested in. */
 491                 if (!streq_ptr(udev_device_get_subsystem(p), "drm"))
 492                         continue;
 493
 494                 nn = udev_device_get_sysname(d);
 495                 if (!nn)
 496                         continue;
 497
 498                 /* Ignore internal displays: the type is encoded in
 499                  * the sysfs name, as the second dash separated item
 500                  * (the first is the card name, the last the connector
 501                  * number). We implement a whitelist of external
 502                  * displays here, rather than a whitelist, to ensure
 503                  * we don't block suspends too eagerly. */
 504                 dash = strchr(nn, '-');
 505                 if (!dash)
 506                         continue;
 507
 508                 dash++;
 509                 FOREACH_STRING(i, "VGA-", "DVI-I-", "DVI-D-", "DVI-A-"
 510                                "Composite-", "SVIDEO-", "Component-",
 511                                "DIN-", "DP-", "HDMI-A-", "HDMI-B-", "TV-") {
 512
 513                         if (startswith(dash, i)) {
 514                                 external = true;
 515                                 break;
 516                         }
 517                 }
 518                 if (!external)
 519                         continue;
 520
 521                 /* Ignore ports that are not enabled */
 522                 enabled = udev_device_get_sysattr_value(d, "enabled");
 523                 if (!enabled)
 524                         continue;
 525                 if (!streq_ptr(enabled, "enabled"))
 526                         continue;
 527
 528                 /* We count any connector which is not explicitly
 529                  * "disconnected" as connected. */
 530                 status = udev_device_get_sysattr_value(d, "status");
 531                 if (!streq_ptr(status, "disconnected"))
 532                         n++;
 533         }
 534
 535         return n;
 536 }
 537
 538 bool manager_is_docked_or_external_displays(Manager *m) {
 539         int n;
 540
 541         /* If we are docked don't react to lid closing */
 542         if (manager_is_docked(m)) {
 543                 log_debug("System is docked.");
 544                 return true;
 545         }
 546
 547         /* If we have more than one display connected,
 548          * assume that we are docked. */
 549         n = manager_count_external_displays(m);
 550         if (n < 0)
 551                 log_warning_errno(n, "Display counting failed: %m");
 552         else if (n >= 1) {
 553                 log_debug("External (%i) displays connected.", n);
 554                 return true;
 555         }
 556
 557         return false;
 558 }