src/login/sd-login.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2011 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU General Public License as published by
  10   the Free Software Foundation; either version 2 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   General Public License for more details.
  17
  18   You should have received a copy of the GNU General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <unistd.h>
  23 #include <string.h>
  24 #include <errno.h>
  25 #include <sys/inotify.h>
  26
  27 #include "util.h"
  28 #include "cgroup-util.h"
  29 #include "macro.h"
  30 #include "sd-login.h"
  31 #include "strv.h"
  32
  33 static int pid_get_cgroup(pid_t pid, char **root, char **cgroup) {
  34         char *cg_process, *cg_init, *p;
  35         int r;
  36
  37         if (pid == 0)
  38                 pid = getpid();
  39
  40         if (pid <= 0)
  41                 return -EINVAL;
  42
  43         r = cg_get_by_pid(SYSTEMD_CGROUP_CONTROLLER, pid, &cg_process);
  44         if (r < 0)
  45                 return r;
  46
  47         r = cg_get_by_pid(SYSTEMD_CGROUP_CONTROLLER, 1, &cg_init);
  48         if (r < 0) {
  49                 free(cg_process);
  50                 return r;
  51         }
  52
  53         if (endswith(cg_init, "/system"))
  54                 cg_init[strlen(cg_init)-7] = 0;
  55         else if (streq(cg_init, "/"))
  56                 cg_init[0] = 0;
  57
  58         if (startswith(cg_process, cg_init))
  59                 p = cg_process + strlen(cg_init);
  60         else
  61                 p = cg_process;
  62
  63         free(cg_init);
  64
  65         if (cgroup) {
  66                 char* c;
  67
  68                 c = strdup(p);
  69                 if (!c) {
  70                         free(cg_process);
  71                         return -ENOMEM;
  72                 }
  73
  74                 *cgroup = c;
  75         }
  76
  77         if (root) {
  78                 cg_process[p-cg_process] = 0;
  79                 *root = cg_process;
  80         } else
  81                 free(cg_process);
  82
  83         return 0;
  84 }
  85
  86 _public_ int sd_pid_get_session(pid_t pid, char **session) {
  87         int r;
  88         char *cgroup, *p;
  89
  90         if (!session)
  91                 return -EINVAL;
  92
  93         r = pid_get_cgroup(pid, NULL, &cgroup);
  94         if (r < 0)
  95                 return r;
  96
  97         if (!startswith(cgroup, "/user/")) {
  98                 free(cgroup);
  99                 return -ENOENT;
 100         }
 101
 102         p = strchr(cgroup + 6, '/');
 103         if (!p) {
 104                 free(cgroup);
 105                 return -ENOENT;
 106         }
 107
 108         p++;
 109         if (startswith(p, "shared/") || streq(p, "shared")) {
 110                 free(cgroup);
 111                 return -ENOENT;
 112         }
 113
 114         p = strndup(p, strcspn(p, "/"));
 115         free(cgroup);
 116
 117         if (!p)
 118                 return -ENOMEM;
 119
 120         *session = p;
 121         return 0;
 122 }
 123
 124 _public_ int sd_pid_get_unit(pid_t pid, char **unit) {
 125         int r;
 126         char *cgroup, *p, *at, *b;
 127         size_t k;
 128
 129         if (!unit)
 130                 return -EINVAL;
 131
 132         r = pid_get_cgroup(pid, NULL, &cgroup);
 133         if (r < 0)
 134                 return r;
 135
 136         if (!startswith(cgroup, "/system/")) {
 137                 free(cgroup);
 138                 return -ENOENT;
 139         }
 140
 141         p = cgroup + 8;
 142         k = strcspn(p, "/");
 143
 144         at = memchr(p, '@', k);
 145         if (at && at[1] == '.') {
 146                 size_t j;
 147
 148                 /* This is a templated service */
 149                 if (p[k] != '/') {
 150                         free(cgroup);
 151                         return -EIO;
 152                 }
 153
 154                 j = strcspn(p+k+1, "/");
 155
 156                 b = malloc(k + j + 1);
 157
 158                 if (b) {
 159                         memcpy(b, p, at - p + 1);
 160                         memcpy(b + (at - p) + 1, p + k + 1, j);
 161                         memcpy(b + (at - p) + 1 + j, at + 1, k - (at - p) - 1);
 162                         b[k+j] = 0;
 163                 }
 164         } else
 165                   b = strndup(p, k);
 166
 167         free(cgroup);
 168
 169         if (!b)
 170                 return -ENOMEM;
 171
 172         *unit = b;
 173         return 0;
 174 }
 175
 176 _public_ int sd_pid_get_owner_uid(pid_t pid, uid_t *uid) {
 177         int r;
 178         char *root, *cgroup, *p, *cc;
 179         struct stat st;
 180
 181         if (!uid)
 182                 return -EINVAL;
 183
 184         r = pid_get_cgroup(pid, &root, &cgroup);
 185         if (r < 0)
 186                 return r;
 187
 188         if (!startswith(cgroup, "/user/")) {
 189                 free(cgroup);
 190                 free(root);
 191                 return -ENOENT;
 192         }
 193
 194         p = strchr(cgroup + 6, '/');
 195         if (!p) {
 196                 free(cgroup);
 197                 return -ENOENT;
 198         }
 199
 200         p++;
 201         p += strcspn(p, "/");
 202         *p = 0;
 203
 204         r = cg_get_path(SYSTEMD_CGROUP_CONTROLLER, root, cgroup, &cc);
 205         free(root);
 206         free(cgroup);
 207
 208         if (r < 0)
 209                 return -ENOMEM;
 210
 211         r = lstat(cc, &st);
 212         free(cc);
 213
 214         if (r < 0)
 215                 return -errno;
 216
 217         if (!S_ISDIR(st.st_mode))
 218                 return -ENOTDIR;
 219
 220         *uid = st.st_uid;
 221         return 0;
 222 }
 223
 224 _public_ int sd_uid_get_state(uid_t uid, char**state) {
 225         char *p, *s = NULL;
 226         int r;
 227
 228         if (!state)
 229                 return -EINVAL;
 230
 231         if (asprintf(&p, "/run/systemd/users/%lu", (unsigned long) uid) < 0)
 232                 return -ENOMEM;
 233
 234         r = parse_env_file(p, NEWLINE, "STATE", &s, NULL);
 235         free(p);
 236
 237         if (r == -ENOENT) {
 238                 free(s);
 239                 s = strdup("offline");
 240                 if (!s)
 241                         return -ENOMEM;
 242
 243                 *state = s;
 244                 return 0;
 245         } else if (r < 0) {
 246                 free(s);
 247                 return r;
 248         } else if (!s)
 249                 return -EIO;
 250
 251         *state = s;
 252         return 0;
 253 }
 254
 255 _public_ int sd_uid_is_on_seat(uid_t uid, int require_active, const char *seat) {
 256         char *p, *w, *t, *state, *s = NULL;
 257         size_t l;
 258         int r;
 259         const char *variable;
 260
 261         if (!seat)
 262                 return -EINVAL;
 263
 264         variable = require_active ? "ACTIVE_UID" : "UIDS";
 265
 266         p = strappend("/run/systemd/seats/", seat);
 267         if (!p)
 268                 return -ENOMEM;
 269
 270         r = parse_env_file(p, NEWLINE, variable, &s, NULL);
 271         free(p);
 272
 273         if (r < 0) {
 274                 free(s);
 275                 return r;
 276         }
 277
 278         if (!s)
 279                 return -EIO;
 280
 281         if (asprintf(&t, "%lu", (unsigned long) uid) < 0) {
 282                 free(s);
 283                 return -ENOMEM;
 284         }
 285
 286         FOREACH_WORD(w, l, s, state) {
 287                 if (strncmp(t, w, l) == 0) {
 288                         free(s);
 289                         free(t);
 290
 291                         return 1;
 292                 }
 293         }
 294
 295         free(s);
 296         free(t);
 297
 298         return 0;
 299 }
 300
 301 static int uid_get_array(uid_t uid, const char *variable, char ***array) {
 302         char *p, *s = NULL;
 303         char **a;
 304         int r;
 305
 306         if (asprintf(&p, "/run/systemd/users/%lu", (unsigned long) uid) < 0)
 307                 return -ENOMEM;
 308
 309         r = parse_env_file(p, NEWLINE,
 310                            variable, &s,
 311                            NULL);
 312         free(p);
 313
 314         if (r < 0) {
 315                 free(s);
 316
 317                 if (r == -ENOENT) {
 318                         if (array)
 319                                 *array = NULL;
 320                         return 0;
 321                 }
 322
 323                 return r;
 324         }
 325
 326         if (!s) {
 327                 if (array)
 328                         *array = NULL;
 329                 return 0;
 330         }
 331
 332         a = strv_split(s, " ");
 333         free(s);
 334
 335         if (!a)
 336                 return -ENOMEM;
 337
 338         strv_uniq(a);
 339         r = strv_length(a);
 340
 341         if (array)
 342                 *array = a;
 343         else
 344                 strv_free(a);
 345
 346         return r;
 347 }
 348
 349 _public_ int sd_uid_get_sessions(uid_t uid, int require_active, char ***sessions) {
 350         return uid_get_array(uid, require_active ? "ACTIVE_SESSIONS" : "SESSIONS", sessions);
 351 }
 352
 353 _public_ int sd_uid_get_seats(uid_t uid, int require_active, char ***seats) {
 354         return uid_get_array(uid, require_active ? "ACTIVE_SEATS" : "SEATS", seats);
 355 }
 356
 357 static int file_of_session(const char *session, char **_p) {
 358         char *p;
 359         int r;
 360
 361         assert(_p);
 362
 363         if (session)
 364                 p = strappend("/run/systemd/sessions/", session);
 365         else {
 366                 char *buf;
 367
 368                 r = sd_pid_get_session(0, &buf);
 369                 if (r < 0)
 370                         return r;
 371
 372                 p = strappend("/run/systemd/sessions/", buf);
 373                 free(buf);
 374         }
 375
 376         if (!p)
 377                 return -ENOMEM;
 378
 379         *_p = p;
 380         return 0;
 381 }
 382
 383 _public_ int sd_session_is_active(const char *session) {
 384         int r;
 385         char *p, *s = NULL;
 386
 387         r = file_of_session(session, &p);
 388         if (r < 0)
 389                 return r;
 390
 391         r = parse_env_file(p, NEWLINE, "ACTIVE", &s, NULL);
 392         free(p);
 393
 394         if (r < 0) {
 395                 free(s);
 396                 return r;
 397         }
 398
 399         if (!s)
 400                 return -EIO;
 401
 402         r = parse_boolean(s);
 403         free(s);
 404
 405         return r;
 406 }
 407
 408 _public_ int sd_session_get_uid(const char *session, uid_t *uid) {
 409         int r;
 410         char *p, *s = NULL;
 411
 412         if (!uid)
 413                 return -EINVAL;
 414
 415         r = file_of_session(session, &p);
 416         if (r < 0)
 417                 return r;
 418
 419         r = parse_env_file(p, NEWLINE, "UID", &s, NULL);
 420         free(p);
 421
 422         if (r < 0) {
 423                 free(s);
 424                 return r;
 425         }
 426
 427         if (!s)
 428                 return -EIO;
 429
 430         r = parse_uid(s, uid);
 431         free(s);
 432
 433         return r;
 434 }
 435
 436 static int session_get_string(const char *session, const char *field, char **value) {
 437         char *p, *s = NULL;
 438         int r;
 439
 440         if (!value)
 441                 return -EINVAL;
 442
 443         r = file_of_session(session, &p);
 444         if (r < 0)
 445                 return r;
 446
 447         r = parse_env_file(p, NEWLINE, field, &s, NULL);
 448         free(p);
 449
 450         if (r < 0) {
 451                 free(s);
 452                 return r;
 453         }
 454
 455         if (isempty(s))
 456                 return -ENOENT;
 457
 458         *value = s;
 459         return 0;
 460 }
 461
 462 _public_ int sd_session_get_seat(const char *session, char **seat) {
 463         return session_get_string(session, "SEAT", seat);
 464 }
 465
 466 _public_ int sd_session_get_service(const char *session, char **service) {
 467         return session_get_string(session, "SERVICE", service);
 468 }
 469
 470 _public_ int sd_session_get_type(const char *session, char **type) {
 471         return session_get_string(session, "TYPE", type);
 472 }
 473
 474 _public_ int sd_session_get_class(const char *session, char **class) {
 475         return session_get_string(session, "CLASS", class);
 476 }
 477
 478 _public_ int sd_session_get_display(const char *session, char **display) {
 479         return session_get_string(session, "DISPLAY", display);
 480 }
 481
 482 static int file_of_seat(const char *seat, char **_p) {
 483         char *p;
 484         int r;
 485
 486         assert(_p);
 487
 488         if (seat)
 489                 p = strappend("/run/systemd/seats/", seat);
 490         else {
 491                 char *buf;
 492
 493                 r = sd_session_get_seat(NULL, &buf);
 494                 if (r < 0)
 495                         return r;
 496
 497                 p = strappend("/run/systemd/seats/", buf);
 498                 free(buf);
 499         }
 500
 501         if (!p)
 502                 return -ENOMEM;
 503
 504         *_p = p;
 505         return 0;
 506 }
 507
 508 _public_ int sd_seat_get_active(const char *seat, char **session, uid_t *uid) {
 509         char *p, *s = NULL, *t = NULL;
 510         int r;
 511
 512         if (!session && !uid)
 513                 return -EINVAL;
 514
 515         r = file_of_seat(seat, &p);
 516         if (r < 0)
 517                 return r;
 518
 519         r = parse_env_file(p, NEWLINE,
 520                            "ACTIVE", &s,
 521                            "ACTIVE_UID", &t,
 522                            NULL);
 523         free(p);
 524
 525         if (r < 0) {
 526                 free(s);
 527                 free(t);
 528                 return r;
 529         }
 530
 531         if (session && !s)  {
 532                 free(t);
 533                 return -ENOENT;
 534         }
 535
 536         if (uid && !t) {
 537                 free(s);
 538                 return -ENOENT;
 539         }
 540
 541         if (uid && t) {
 542                 r = parse_uid(t, uid);
 543                 if (r < 0) {
 544                         free(t);
 545                         free(s);
 546                         return r;
 547                 }
 548         }
 549
 550         free(t);
 551
 552         if (session && s)
 553                 *session = s;
 554         else
 555                 free(s);
 556
 557         return 0;
 558 }
 559
 560 _public_ int sd_seat_get_sessions(const char *seat, char ***sessions, uid_t **uids, unsigned *n_uids) {
 561         char *p, *s = NULL, *t = NULL, **a = NULL;
 562         uid_t *b = NULL;
 563         unsigned n = 0;
 564         int r;
 565
 566         r = file_of_seat(seat, &p);
 567         if (r < 0)
 568                 return r;
 569
 570         r = parse_env_file(p, NEWLINE,
 571                            "SESSIONS", &s,
 572                            "ACTIVE_SESSIONS", &t,
 573                            NULL);
 574         free(p);
 575
 576         if (r < 0) {
 577                 free(s);
 578                 free(t);
 579                 return r;
 580         }
 581
 582         if (s) {
 583                 a = strv_split(s, " ");
 584                 if (!a) {
 585                         free(s);
 586                         free(t);
 587                         return -ENOMEM;
 588                 }
 589         }
 590
 591         free(s);
 592
 593         if (uids && t) {
 594                 char *w, *state;
 595                 size_t l;
 596
 597                 FOREACH_WORD(w, l, t, state)
 598                         n++;
 599
 600                 if (n == 0)
 601                         b = NULL;
 602                 else {
 603                         unsigned i = 0;
 604
 605                         b = new(uid_t, n);
 606                         if (!b) {
 607                                 strv_free(a);
 608                                 return -ENOMEM;
 609                         }
 610
 611                         FOREACH_WORD(w, l, t, state) {
 612                                 char *k;
 613
 614                                 k = strndup(w, l);
 615                                 if (!k) {
 616                                         free(t);
 617                                         free(b);
 618                                         strv_free(a);
 619                                         return -ENOMEM;
 620                                 }
 621
 622                                 r = parse_uid(k, b + i);
 623                                 free(k);
 624                                 if (r < 0)
 625                                         continue;
 626
 627                                 i++;
 628                         }
 629                 }
 630         }
 631
 632         free(t);
 633
 634         r = strv_length(a);
 635
 636         if (sessions)
 637                 *sessions = a;
 638         else
 639                 strv_free(a);
 640
 641         if (uids)
 642                 *uids = b;
 643
 644         if (n_uids)
 645                 *n_uids = n;
 646
 647         return r;
 648 }
 649
 650 _public_ int sd_seat_can_multi_session(const char *seat) {
 651         char *p, *s = NULL;
 652         int r;
 653
 654         r = file_of_seat(seat, &p);
 655         if (r < 0)
 656                 return r;
 657
 658         r = parse_env_file(p, NEWLINE,
 659                            "CAN_MULTI_SESSION", &s,
 660                            NULL);
 661         free(p);
 662
 663         if (r < 0) {
 664                 free(s);
 665                 return r;
 666         }
 667
 668         if (s) {
 669                 r = parse_boolean(s);
 670                 free(s);
 671         } else
 672                 r = 0;
 673
 674         return r;
 675 }
 676
 677 _public_ int sd_get_seats(char ***seats) {
 678         return get_files_in_directory("/run/systemd/seats/", seats);
 679 }
 680
 681 _public_ int sd_get_sessions(char ***sessions) {
 682         return get_files_in_directory("/run/systemd/sessions/", sessions);
 683 }
 684
 685 _public_ int sd_get_uids(uid_t **users) {
 686         DIR *d;
 687         int r = 0;
 688         unsigned n = 0;
 689         uid_t *l = NULL;
 690
 691         d = opendir("/run/systemd/users/");
 692         if (!d)
 693                 return -errno;
 694
 695         for (;;) {
 696                 struct dirent buffer, *de;
 697                 int k;
 698                 uid_t uid;
 699
 700                 k = readdir_r(d, &buffer, &de);
 701                 if (k != 0) {
 702                         r = -k;
 703                         goto finish;
 704                 }
 705
 706                 if (!de)
 707                         break;
 708
 709                 dirent_ensure_type(d, de);
 710
 711                 if (!dirent_is_file(de))
 712                         continue;
 713
 714                 k = parse_uid(de->d_name, &uid);
 715                 if (k < 0)
 716                         continue;
 717
 718                 if (users) {
 719                         if ((unsigned) r >= n) {
 720                                 uid_t *t;
 721
 722                                 n = MAX(16, 2*r);
 723                                 t = realloc(l, sizeof(uid_t) * n);
 724                                 if (!t) {
 725                                         r = -ENOMEM;
 726                                         goto finish;
 727                                 }
 728
 729                                 l = t;
 730                         }
 731
 732                         assert((unsigned) r < n);
 733                         l[r++] = uid;
 734                 } else
 735                         r++;
 736         }
 737
 738 finish:
 739         if (d)
 740                 closedir(d);
 741
 742         if (r >= 0) {
 743                 if (users)
 744                         *users = l;
 745         } else
 746                 free(l);
 747
 748         return r;
 749 }
 750
 751 static inline int MONITOR_TO_FD(sd_login_monitor *m) {
 752         return (int) (unsigned long) m - 1;
 753 }
 754
 755 static inline sd_login_monitor* FD_TO_MONITOR(int fd) {
 756         return (sd_login_monitor*) (unsigned long) (fd + 1);
 757 }
 758
 759 _public_ int sd_login_monitor_new(const char *category, sd_login_monitor **m) {
 760         int fd, k;
 761         bool good = false;
 762
 763         if (!m)
 764                 return -EINVAL;
 765
 766         fd = inotify_init1(IN_NONBLOCK|IN_CLOEXEC);
 767         if (fd < 0)
 768                 return errno;
 769
 770         if (!category || streq(category, "seat")) {
 771                 k = inotify_add_watch(fd, "/run/systemd/seats/", IN_MOVED_TO|IN_DELETE);
 772                 if (k < 0) {
 773                         close_nointr_nofail(fd);
 774                         return -errno;
 775                 }
 776
 777                 good = true;
 778         }
 779
 780         if (!category || streq(category, "session")) {
 781                 k = inotify_add_watch(fd, "/run/systemd/sessions/", IN_MOVED_TO|IN_DELETE);
 782                 if (k < 0) {
 783                         close_nointr_nofail(fd);
 784                         return -errno;
 785                 }
 786
 787                 good = true;
 788         }
 789
 790         if (!category || streq(category, "uid")) {
 791                 k = inotify_add_watch(fd, "/run/systemd/users/", IN_MOVED_TO|IN_DELETE);
 792                 if (k < 0) {
 793                         close_nointr_nofail(fd);
 794                         return -errno;
 795                 }
 796
 797                 good = true;
 798         }
 799
 800         if (!good) {
 801                 close_nointr(fd);
 802                 return -EINVAL;
 803         }
 804
 805         *m = FD_TO_MONITOR(fd);
 806         return 0;
 807 }
 808
 809 _public_ sd_login_monitor* sd_login_monitor_unref(sd_login_monitor *m) {
 810         int fd;
 811
 812         if (!m)
 813                 return NULL;
 814
 815         fd = MONITOR_TO_FD(m);
 816         close_nointr(fd);
 817
 818         return NULL;
 819 }
 820
 821 _public_ int sd_login_monitor_flush(sd_login_monitor *m) {
 822
 823         if (!m)
 824                 return -EINVAL;
 825
 826         return flush_fd(MONITOR_TO_FD(m));
 827 }
 828
 829 _public_ int sd_login_monitor_get_fd(sd_login_monitor *m) {
 830
 831         if (!m)
 832                 return -EINVAL;
 833
 834         return MONITOR_TO_FD(m);
 835 }