1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2011 Lennart Poettering
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.
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.
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/>.
28 #include <sys/epoll.h>
29 #include <sys/ioctl.h>
33 #include "dbus-common.h"
34 #include "dbus-loop.h"
42 Manager *manager_new(void) {
49 m->console_active_fd = -1;
55 m->devices = hashmap_new(string_hash_func, string_compare_func);
56 m->seats = hashmap_new(string_hash_func, string_compare_func);
57 m->sessions = hashmap_new(string_hash_func, string_compare_func);
58 m->users = hashmap_new(trivial_hash_func, trivial_compare_func);
60 if (!m->devices || !m->seats || !m->sessions || !m->users) {
71 if (cg_get_user_path(&m->cgroup_path) < 0) {
79 void manager_free(Manager *m) {
87 while ((session = hashmap_first(m->sessions)))
88 session_free(session);
90 while ((u = hashmap_first(m->users)))
93 while ((d = hashmap_first(m->devices)))
96 while ((s = hashmap_first(m->seats)))
99 hashmap_free(m->sessions);
100 hashmap_free(m->users);
101 hashmap_free(m->devices);
102 hashmap_free(m->seats);
104 if (m->console_active_fd >= 0)
105 close_nointr_nofail(m->console_active_fd);
108 udev_monitor_unref(m->udev_monitor);
114 dbus_connection_flush(m->bus);
115 dbus_connection_close(m->bus);
116 dbus_connection_unref(m->bus);
120 close_nointr_nofail(m->bus_fd);
122 if (m->epoll_fd >= 0)
123 close_nointr_nofail(m->epoll_fd);
125 free(m->cgroup_path);
129 int manager_add_device(Manager *m, const char *sysfs, Device **_device) {
135 d = hashmap_get(m->devices, sysfs);
143 d = device_new(m, sysfs);
153 int manager_add_seat(Manager *m, const char *id, Seat **_seat) {
159 s = hashmap_get(m->seats, id);
177 int manager_add_session(Manager *m, User *u, const char *id, Session **_session) {
183 s = hashmap_get(m->sessions, id);
191 s = session_new(m, u, id);
201 int manager_add_user(Manager *m, uid_t uid, gid_t gid, const char *name, User **_user) {
208 u = hashmap_get(m->users, ULONG_TO_PTR((unsigned long) uid));
216 u = user_new(m, uid, gid, name);
232 int manager_add_user_by_name(Manager *m, const char *name, User **_user) {
241 return errno ? -errno : -ENOENT;
243 return manager_add_user(m, p->pw_uid, p->pw_gid, name, _user);
246 int manager_add_user_by_uid(Manager *m, uid_t uid, User **_user) {
254 return errno ? -errno : -ENOENT;
256 return manager_add_user(m, uid, p->pw_gid, p->pw_name, _user);
259 int manager_process_device(Manager *m, struct udev_device *d) {
267 sn = udev_device_get_property_value(d, "SEAT");
271 r = manager_add_device(m, udev_device_get_syspath(d), &device);
275 r = manager_add_seat(m, sn, &seat);
279 device_attach(device, seat);
283 int manager_enumerate_devices(Manager *m) {
284 struct udev_list_entry *item = NULL, *first = NULL;
285 struct udev_enumerate *e;
290 /* Loads devices from udev and creates seats for them as
293 e = udev_enumerate_new(m->udev);
297 if (udev_enumerate_add_match_subsystem(e, "graphics") < 0)
300 if (udev_enumerate_add_match_tag(e, "seat") < 0)
303 if (udev_enumerate_scan_devices(e) < 0)
306 first = udev_enumerate_get_list_entry(e);
307 udev_list_entry_foreach(item, first) {
308 struct udev_device *d;
311 d = udev_device_new_from_syspath(m->udev, udev_list_entry_get_name(item));
315 k = manager_process_device(m, d);
316 udev_device_unref(d);
324 udev_enumerate_unref(e);
329 int manager_enumerate_seats(Manager *m) {
336 /* This loads data about seats stored on disk, but does not
337 * actually create any seats. Removes data of seats that no
340 d = opendir("/run/systemd/seats");
345 log_error("Failed to open /run/systemd/seats: %m");
349 while ((de = readdir(d))) {
353 if (!dirent_is_file(de))
356 s = hashmap_get(m->seats, de->d_name);
358 unlinkat(dirfd(d), de->d_name, 0);
372 static int manager_enumerate_users_from_cgroup(Manager *m) {
377 r = cg_enumerate_subgroups(SYSTEMD_CGROUP_CONTROLLER, m->cgroup_path, &d);
382 log_error("Failed to open %s: %s", m->cgroup_path, strerror(-r));
386 while ((r = cg_read_subgroup(d, &name)) > 0) {
389 r = manager_add_user_by_name(m, name, &user);
395 if (user->cgroup_path)
398 if (asprintf(&user->cgroup_path, "%s/%s", m->cgroup_path, name) < 0) {
410 int manager_enumerate_users(Manager *m) {
417 r = manager_enumerate_users_from_cgroup(m);
419 d = opendir("/run/systemd/users");
424 log_error("Failed to open /run/systemd/users: %m");
428 while ((de = readdir(d))) {
433 if (!dirent_is_file(de))
436 k = safe_atolu(de->d_name, &ul);
438 log_error("Failed to parse file name %s: %s", de->d_name, strerror(-k));
442 u = hashmap_get(m->users, ULONG_TO_PTR(ul));
444 unlinkat(dirfd(d), de->d_name, 0);
458 static int manager_enumerate_sessions_from_cgroup(Manager *m) {
463 HASHMAP_FOREACH(u, m->users, i) {
468 k = cg_enumerate_subgroups(SYSTEMD_CGROUP_CONTROLLER, u->cgroup_path, &d);
473 log_error("Failed to open %s: %s", u->cgroup_path, strerror(-k));
478 while ((k = cg_read_subgroup(d, &name)) > 0) {
481 k = manager_add_session(m, u, name, &session);
487 if (session->cgroup_path)
490 if (asprintf(&session->cgroup_path, "%s/%s", u->cgroup_path, name) < 0) {
505 int manager_enumerate_sessions(Manager *m) {
512 r = manager_enumerate_sessions_from_cgroup(m);
514 d = opendir("/run/systemd/sessions");
519 log_error("Failed to open /run/systemd/sessions: %m");
523 while ((de = readdir(d))) {
527 if (!dirent_is_file(de))
530 s = hashmap_get(m->sessions, de->d_name);
532 unlinkat(dirfd(d), de->d_name, 0);
546 int manager_start_linger_users(Manager *m) {
551 d = opendir("/var/lib/systemd/linger");
556 log_error("Failed to open /var/lib/systemd/linger/: %m");
560 while ((de = readdir(d))) {
563 if (!dirent_is_file(de))
566 k = manager_add_user_by_name(m, de->d_name, NULL);
568 log_notice("Couldn't add lingering user %s: %s", de->d_name, strerror(-k));
578 int manager_dispatch_udev(Manager *m) {
579 struct udev_device *d;
584 d = udev_monitor_receive_device(m->udev_monitor);
588 r = manager_process_device(m, d);
589 udev_device_unref(d);
594 int manager_dispatch_console(Manager *m) {
601 lseek(m->console_active_fd, SEEK_SET, 0);
603 k = read(m->console_active_fd, t, sizeof(t)-1);
605 log_error("Failed to read current console: %s", k < 0 ? strerror(-errno) : "EOF");
606 return k < 0 ? -errno : -EIO;
610 if (!startswith(t, "tty")) {
611 log_error("Hm, /sys/class/tty/tty0/active is badly formatted.");
615 r = safe_atoi(t+3, &vtnr);
617 log_error("Failed to parse VT number %s", t+3);
622 log_error("VT number invalid: %s", t+3);
627 seat_active_vt_changed(m->vtconsole, vtnr);
632 static int vt_is_busy(int vtnr) {
633 struct vt_stat vt_stat;
638 fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
642 if (ioctl(fd, VT_GETSTATE, &vt_stat) < 0)
645 r = !!(vt_stat.v_state & (1 << vtnr));
647 close_nointr_nofail(fd);
652 int manager_spawn_autovt(Manager *m, int vtnr) {
657 r = vt_is_busy(vtnr);
666 static DBusHandlerResult login_message_handler(
667 DBusConnection *connection,
668 DBusMessage *message,
671 return DBUS_HANDLER_RESULT_HANDLED;
674 static DBusHandlerResult login_message_filter(
675 DBusConnection *connection,
676 DBusMessage *message,
679 return DBUS_HANDLER_RESULT_HANDLED;
682 static int manager_connect_bus(Manager *m) {
683 const DBusObjectPathVTable login_vtable = {
684 .message_function = login_message_handler
689 struct epoll_event ev;
693 assert(m->bus_fd < 0);
695 dbus_error_init(&error);
697 m->bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, &error);
699 log_error("Failed to get system D-Bus connection: %s", error.message);
704 if (!dbus_connection_register_object_path(m->bus, "/org/freedesktop/login1", &login_vtable, NULL)) {
705 log_error("Not enough memory");
710 if (!dbus_connection_add_filter(m->bus, login_message_filter, m, NULL)) {
711 log_error("Not enough memory");
716 dbus_bus_add_match(m->bus,
718 "interface='org.freedesktop.systemd1.Agent',"
720 "path='/org/freedesktop/systemd1/agent'",
723 if (dbus_error_is_set(&error)) {
724 log_error("Failed to register match: %s", error.message);
729 if (dbus_bus_request_name(m->bus, "org.freedesktop.login1", DBUS_NAME_FLAG_DO_NOT_QUEUE, &error) < 0) {
730 log_error("Failed to register name on bus: %s", error.message);
735 m->bus_fd = bus_loop_open(m->bus);
743 ev.data.u32 = FD_BUS;
745 if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, m->bus_fd, &ev) < 0)
751 dbus_error_free(&error);
756 static int manager_connect_console(Manager *m) {
757 struct epoll_event ev;
760 assert(m->console_active_fd < 0);
762 m->console_active_fd = open("/sys/class/tty/tty0/active", O_RDONLY|O_NOCTTY|O_CLOEXEC);
763 if (m->console_active_fd < 0) {
764 log_error("Failed to open /sys/class/tty/tty0/active: %m");
770 ev.data.u32 = FD_CONSOLE;
772 if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, m->console_active_fd, &ev) < 0)
778 static int manager_connect_udev(Manager *m) {
779 struct epoll_event ev;
782 assert(!m->udev_monitor);
784 m->udev_monitor = udev_monitor_new_from_netlink(m->udev, "udev");
785 if (!m->udev_monitor)
788 udev_monitor_filter_add_match_tag(m->udev_monitor, "seat");
790 if (udev_monitor_filter_add_match_subsystem_devtype(m->udev_monitor, "graphics", NULL) < 0)
793 if (udev_monitor_enable_receiving(m->udev_monitor) < 0)
796 m->udev_fd = udev_monitor_get_fd(m->udev_monitor);
800 ev.data.u32 = FD_UDEV;
802 if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, m->udev_fd, &ev) < 0)
808 int manager_startup(Manager *m) {
812 assert(m->epoll_fd <= 0);
814 m->epoll_fd = epoll_create1(EPOLL_CLOEXEC);
818 /* Connect to udev */
819 r = manager_connect_udev(m);
823 /* Connect to console */
824 r = manager_connect_console(m);
828 /* Connect to the bus */
829 r = manager_connect_bus(m);
833 /* Deserialize state */
834 manager_enumerate_devices(m);
835 manager_enumerate_seats(m);
836 manager_enumerate_users(m);
837 manager_enumerate_sessions(m);
839 /* Spawn lingering users */
840 manager_start_linger_users(m);
845 int manager_run(Manager *m) {
849 struct epoll_event event;
852 if (dbus_connection_dispatch(m->bus) != DBUS_DISPATCH_COMPLETE)
855 n = epoll_wait(m->epoll_fd, &event, 1, -1);
857 log_error("epoll() failed: %m");
861 switch (event.data.u32) {
864 manager_dispatch_udev(m);
868 manager_dispatch_console(m);
872 bus_loop_dispatch(m->bus_fd);
880 int main(int argc, char *argv[]) {
884 log_set_target(LOG_TARGET_AUTO);
885 log_parse_environment();
889 log_error("This program takes no arguments.");
898 log_error("Out of memory");
903 r = manager_startup(m);
905 log_error("Failed to fully start up daemon: %s", strerror(-r));
915 return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;