src/login/test-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 <sys/poll.h>
  23 #include <string.h>
  24
  25 #include <systemd/sd-login.h>
  26
  27 #include "util.h"
  28 #include "strv.h"
  29
  30 int main(int argc, char* argv[]) {
  31         int r, k;
  32         uid_t u, u2;
  33         char *seat;
  34         char *session;
  35         char *state;
  36         char *session2;
  37         char *t;
  38         char **seats, **sessions;
  39         uid_t *uids;
  40         unsigned n;
  41         struct pollfd pollfd;
  42         sd_login_monitor *m;
  43
  44         assert_se(sd_pid_get_session(0, &session) == 0);
  45         printf("session = %s\n", session);
  46
  47         assert_se(sd_pid_get_owner_uid(0, &u2) == 0);
  48         printf("user = %lu\n", (unsigned long) u2);
  49
  50         r = sd_uid_get_sessions(u2, false, &sessions);
  51         assert_se(r >= 0);
  52         assert_se(r == (int) strv_length(sessions));
  53         assert_se(t = strv_join(sessions, ", "));
  54         strv_free(sessions);
  55         printf("sessions = %s\n", t);
  56         free(t);
  57
  58         assert_se(r == sd_uid_get_sessions(u2, false, NULL));
  59
  60         r = sd_uid_get_seats(u2, false, &seats);
  61         assert_se(r >= 0);
  62         assert_se(r == (int) strv_length(seats));
  63         assert_se(t = strv_join(seats, ", "));
  64         strv_free(seats);
  65         printf("seats = %s\n", t);
  66         free(t);
  67
  68         assert_se(r == sd_uid_get_seats(u2, false, NULL));
  69
  70         r = sd_session_is_active(session);
  71         assert_se(r >= 0);
  72         printf("active = %s\n", yes_no(r));
  73
  74         assert_se(sd_session_get_uid(session, &u) >= 0);
  75         printf("uid = %lu\n", (unsigned long) u);
  76         assert_se(u == u2);
  77
  78         assert_se(sd_session_get_seat(session, &seat) >= 0);
  79         printf("seat = %s\n", seat);
  80
  81         r = sd_seat_can_multi_session(seat);
  82         assert_se(r >= 0);
  83         printf("can do multi session = %s\n", yes_no(r));
  84
  85         assert_se(sd_uid_get_state(u, &state) >= 0);
  86         printf("state = %s\n", state);
  87
  88         assert_se(sd_uid_is_on_seat(u, 0, seat) > 0);
  89
  90         k = sd_uid_is_on_seat(u, 1, seat);
  91         assert_se(k >= 0);
  92         assert_se(!!r == !!r);
  93
  94         assert_se(sd_seat_get_active(seat, &session2, &u2) >= 0);
  95         printf("session2 = %s\n", session2);
  96         printf("uid2 = %lu\n", (unsigned long) u2);
  97
  98         r = sd_seat_get_sessions(seat, &sessions, &uids, &n);
  99         assert_se(r >= 0);
 100         printf("n_sessions = %i\n", r);
 101         assert_se(r == (int) strv_length(sessions));
 102         assert_se(t = strv_join(sessions, ", "));
 103         strv_free(sessions);
 104         printf("sessions = %s\n", t);
 105         free(t);
 106         printf("uids =");
 107         for (k = 0; k < (int) n; k++)
 108                 printf(" %lu", (unsigned long) uids[k]);
 109         printf("\n");
 110         free(uids);
 111
 112         assert_se(sd_seat_get_sessions(seat, NULL, NULL, NULL) == r);
 113
 114         free(session);
 115         free(state);
 116         free(session2);
 117         free(seat);
 118
 119         r = sd_get_seats(&seats);
 120         assert_se(r >= 0);
 121         assert_se(r == (int) strv_length(seats));
 122         assert_se(t = strv_join(seats, ", "));
 123         strv_free(seats);
 124         printf("n_seats = %i\n", r);
 125         printf("seats = %s\n", t);
 126         free(t);
 127
 128         assert_se(sd_get_seats(NULL) == r);
 129
 130         r = sd_get_sessions(&sessions);
 131         assert_se(r >= 0);
 132         assert_se(r == (int) strv_length(sessions));
 133         assert_se(t = strv_join(sessions, ", "));
 134         strv_free(sessions);
 135         printf("n_sessions = %i\n", r);
 136         printf("sessions = %s\n", t);
 137         free(t);
 138
 139         assert_se(sd_get_sessions(NULL) == r);
 140
 141         r = sd_get_uids(&uids);
 142         assert_se(r >= 0);
 143
 144         printf("uids =");
 145         for (k = 0; k < r; k++)
 146                 printf(" %lu", (unsigned long) uids[k]);
 147         printf("\n");
 148         free(uids);
 149
 150         printf("n_uids = %i\n", r);
 151         assert_se(sd_get_uids(NULL) == r);
 152
 153         r = sd_login_monitor_new("session", &m);
 154         assert_se(r >= 0);
 155
 156         zero(pollfd);
 157         pollfd.fd = sd_login_monitor_get_fd(m);
 158         pollfd.events = POLLIN;
 159
 160         for (n = 0; n < 5; n++) {
 161                 r = poll(&pollfd, 1, -1);
 162                 assert_se(r >= 0);
 163
 164                 sd_login_monitor_flush(m);
 165                 printf("Wake!\n");
 166         }
 167
 168         sd_login_monitor_unref(m);
 169
 170         return 0;
 171 }