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) {
207 u = hashmap_get(m->users, ULONG_TO_PTR((unsigned long) uid));
215 u = user_new(m, uid, gid, name);
225 int manager_add_user_by_name(Manager *m, const char *name, User **_user) {
234 return errno ? -errno : -ENOENT;
236 return manager_add_user(m, p->pw_uid, p->pw_gid, name, _user);
239 int manager_add_user_by_uid(Manager *m, uid_t uid, User **_user) {
247 return errno ? -errno : -ENOENT;
249 return manager_add_user(m, uid, p->pw_gid, p->pw_name, _user);
252 int manager_process_device(Manager *m, struct udev_device *d) {
258 if (streq_ptr(udev_device_get_action(d), "remove")) {
260 device = hashmap_get(m->devices, udev_device_get_syspath(d));
264 seat_add_to_gc_queue(device->seat);
271 sn = udev_device_get_property_value(d, "SEAT");
275 if (!startswith(sn, "seat"))
278 r = manager_add_device(m, udev_device_get_syspath(d), &device);
282 r = manager_add_seat(m, sn, &seat);
290 device_attach(device, seat);
296 int manager_enumerate_devices(Manager *m) {
297 struct udev_list_entry *item = NULL, *first = NULL;
298 struct udev_enumerate *e;
303 /* Loads devices from udev and creates seats for them as
306 e = udev_enumerate_new(m->udev);
312 r = udev_enumerate_add_match_subsystem(e, "graphics");
316 r = udev_enumerate_add_match_tag(e, "seat");
320 r = udev_enumerate_scan_devices(e);
324 first = udev_enumerate_get_list_entry(e);
325 udev_list_entry_foreach(item, first) {
326 struct udev_device *d;
329 d = udev_device_new_from_syspath(m->udev, udev_list_entry_get_name(item));
335 k = manager_process_device(m, d);
336 udev_device_unref(d);
344 udev_enumerate_unref(e);
349 int manager_enumerate_seats(Manager *m) {
356 /* This loads data about seats stored on disk, but does not
357 * actually create any seats. Removes data of seats that no
360 d = opendir("/run/systemd/seats");
365 log_error("Failed to open /run/systemd/seats: %m");
369 while ((de = readdir(d))) {
373 if (!dirent_is_file(de))
376 if (!startswith(de->d_name, "seat"))
379 s = hashmap_get(m->seats, de->d_name);
381 unlinkat(dirfd(d), de->d_name, 0);
395 static int manager_enumerate_users_from_cgroup(Manager *m) {
400 r = cg_enumerate_subgroups(SYSTEMD_CGROUP_CONTROLLER, m->cgroup_path, &d);
405 log_error("Failed to open %s: %s", m->cgroup_path, strerror(-r));
409 while ((r = cg_read_subgroup(d, &name)) > 0) {
413 k = manager_add_user_by_name(m, name, &user);
420 user_add_to_gc_queue(user);
422 if (!user->cgroup_path)
423 if (asprintf(&user->cgroup_path, "%s/%s", m->cgroup_path, name) < 0) {
438 static int manager_enumerate_linger_users(Manager *m) {
443 d = opendir("/var/lib/systemd/linger");
448 log_error("Failed to open /var/lib/systemd/linger/: %m");
452 while ((de = readdir(d))) {
455 if (!dirent_is_file(de))
458 k = manager_add_user_by_name(m, de->d_name, NULL);
460 log_notice("Couldn't add lingering user %s: %s", de->d_name, strerror(-k));
470 int manager_enumerate_users(Manager *m) {
477 /* First, enumerate user cgroups */
478 r = manager_enumerate_users_from_cgroup(m);
480 /* Second, add lingering users on top */
481 k = manager_enumerate_linger_users(m);
485 /* Third, read in user data stored on disk */
486 d = opendir("/run/systemd/users");
491 log_error("Failed to open /run/systemd/users: %m");
495 while ((de = readdir(d))) {
499 if (!dirent_is_file(de))
502 k = safe_atolu(de->d_name, &ul);
504 log_error("Failed to parse file name %s: %s", de->d_name, strerror(-k));
508 u = hashmap_get(m->users, ULONG_TO_PTR(ul));
510 unlinkat(dirfd(d), de->d_name, 0);
524 static int manager_enumerate_sessions_from_cgroup(Manager *m) {
529 HASHMAP_FOREACH(u, m->users, i) {
537 k = cg_enumerate_subgroups(SYSTEMD_CGROUP_CONTROLLER, u->cgroup_path, &d);
542 log_error("Failed to open %s: %s", u->cgroup_path, strerror(-k));
547 while ((k = cg_read_subgroup(d, &name)) > 0) {
550 k = manager_add_session(m, u, name, &session);
556 session_add_to_gc_queue(session);
558 if (!session->cgroup_path)
559 if (asprintf(&session->cgroup_path, "%s/%s", u->cgroup_path, name) < 0) {
577 int manager_enumerate_sessions(Manager *m) {
584 /* First enumerate session cgroups */
585 r = manager_enumerate_sessions_from_cgroup(m);
587 /* Second, read in session data stored on disk */
588 d = opendir("/run/systemd/sessions");
593 log_error("Failed to open /run/systemd/sessions: %m");
597 while ((de = readdir(d))) {
601 if (!dirent_is_file(de))
604 s = hashmap_get(m->sessions, de->d_name);
606 unlinkat(dirfd(d), de->d_name, 0);
620 int manager_dispatch_udev(Manager *m) {
621 struct udev_device *d;
626 d = udev_monitor_receive_device(m->udev_monitor);
630 r = manager_process_device(m, d);
631 udev_device_unref(d);
636 int manager_dispatch_console(Manager *m) {
640 seat_read_active_vt(m->vtconsole);
645 static int vt_is_busy(int vtnr) {
646 struct vt_stat vt_stat;
651 fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
655 if (ioctl(fd, VT_GETSTATE, &vt_stat) < 0)
658 r = !!(vt_stat.v_state & (1 << vtnr));
660 close_nointr_nofail(fd);
665 int manager_spawn_autovt(Manager *m, int vtnr) {
670 r = vt_is_busy(vtnr);
679 static DBusHandlerResult login_message_handler(
680 DBusConnection *connection,
681 DBusMessage *message,
684 return DBUS_HANDLER_RESULT_HANDLED;
687 static DBusHandlerResult login_message_filter(
688 DBusConnection *connection,
689 DBusMessage *message,
692 return DBUS_HANDLER_RESULT_HANDLED;
695 static int manager_connect_bus(Manager *m) {
696 const DBusObjectPathVTable login_vtable = {
697 .message_function = login_message_handler
702 struct epoll_event ev;
706 assert(m->bus_fd < 0);
708 dbus_error_init(&error);
710 m->bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, &error);
712 log_error("Failed to get system D-Bus connection: %s", error.message);
717 if (!dbus_connection_register_object_path(m->bus, "/org/freedesktop/login1", &login_vtable, NULL)) {
718 log_error("Not enough memory");
723 if (!dbus_connection_add_filter(m->bus, login_message_filter, m, NULL)) {
724 log_error("Not enough memory");
729 dbus_bus_add_match(m->bus,
731 "interface='org.freedesktop.systemd1.Agent',"
733 "path='/org/freedesktop/systemd1/agent'",
736 if (dbus_error_is_set(&error)) {
737 log_error("Failed to register match: %s", error.message);
742 if (dbus_bus_request_name(m->bus, "org.freedesktop.login1", DBUS_NAME_FLAG_DO_NOT_QUEUE, &error) < 0) {
743 log_error("Failed to register name on bus: %s", error.message);
748 m->bus_fd = bus_loop_open(m->bus);
756 ev.data.u32 = FD_BUS;
758 if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, m->bus_fd, &ev) < 0)
764 dbus_error_free(&error);
769 static int manager_connect_console(Manager *m) {
770 struct epoll_event ev;
773 assert(m->console_active_fd < 0);
775 m->console_active_fd = open("/sys/class/tty/tty0/active", O_RDONLY|O_NOCTTY|O_CLOEXEC);
776 if (m->console_active_fd < 0) {
777 log_error("Failed to open /sys/class/tty/tty0/active: %m");
783 ev.data.u32 = FD_CONSOLE;
785 if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, m->console_active_fd, &ev) < 0)
791 static int manager_connect_udev(Manager *m) {
792 struct epoll_event ev;
796 assert(!m->udev_monitor);
798 m->udev_monitor = udev_monitor_new_from_netlink(m->udev, "udev");
799 if (!m->udev_monitor)
802 r = udev_monitor_filter_add_match_tag(m->udev_monitor, "seat");
806 r = udev_monitor_filter_add_match_subsystem_devtype(m->udev_monitor, "graphics", NULL);
810 r = udev_monitor_enable_receiving(m->udev_monitor);
814 m->udev_fd = udev_monitor_get_fd(m->udev_monitor);
818 ev.data.u32 = FD_UDEV;
820 if (epoll_ctl(m->epoll_fd, EPOLL_CTL_ADD, m->udev_fd, &ev) < 0)
826 void manager_gc(Manager *m) {
833 while ((seat = m->seat_gc_queue)) {
834 LIST_REMOVE(Seat, gc_queue, m->seat_gc_queue, seat);
835 seat->in_gc_queue = false;
837 if (seat_check_gc(seat) == 0) {
843 while ((session = m->session_gc_queue)) {
844 LIST_REMOVE(Session, gc_queue, m->session_gc_queue, session);
845 session->in_gc_queue = false;
847 if (session_check_gc(session) == 0) {
848 session_stop(session);
849 session_free(session);
853 while ((user = m->user_gc_queue)) {
854 LIST_REMOVE(User, gc_queue, m->user_gc_queue, user);
855 user->in_gc_queue = false;
857 if (user_check_gc(user) == 0) {
864 int manager_startup(Manager *m) {
872 assert(m->epoll_fd <= 0);
874 m->epoll_fd = epoll_create1(EPOLL_CLOEXEC);
878 /* Connect to udev */
879 r = manager_connect_udev(m);
883 /* Connect to console */
884 r = manager_connect_console(m);
888 /* Connect to the bus */
889 r = manager_connect_bus(m);
893 /* Instantiate magic seat 0 */
894 r = manager_add_seat(m, "seat0", &m->vtconsole);
898 /* Deserialize state */
899 manager_enumerate_devices(m);
900 manager_enumerate_seats(m);
901 manager_enumerate_users(m);
902 manager_enumerate_sessions(m);
904 /* Get rid of objects that are no longer used */
907 /* And start everything */
908 HASHMAP_FOREACH(seat, m->seats, i)
911 HASHMAP_FOREACH(user, m->users, i)
914 HASHMAP_FOREACH(session, m->sessions, i)
915 session_start(session);
920 int manager_run(Manager *m) {
924 struct epoll_event event;
929 if (dbus_connection_dispatch(m->bus) != DBUS_DISPATCH_COMPLETE)
932 n = epoll_wait(m->epoll_fd, &event, 1, -1);
934 log_error("epoll() failed: %m");
938 switch (event.data.u32) {
941 manager_dispatch_udev(m);
945 manager_dispatch_console(m);
949 bus_loop_dispatch(m->bus_fd);
957 int main(int argc, char *argv[]) {
961 log_set_target(LOG_TARGET_AUTO);
962 log_parse_environment();
966 log_error("This program takes no arguments.");
975 log_error("Out of memory");
980 r = manager_startup(m);
982 log_error("Failed to fully start up daemon: %s", strerror(-r));
992 return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;