login: track login class (i.e. one of "user", "greeter", "lock-screen") for each...

[elogind.git] / src / login / logind-session.c

diff --git a/src/login/logind-session.c b/src/login/logind-session.c
index bb802e5fc293176cbe9da97bb9fafc6f3205f53b..af9c12dcd57d22c35eff72c5c500506115f4ed71 100644 (file)
--- a/src/login/logind-session.c
+++ b/src/login/logind-session.c
@@ -145,6 +145,11 @@ int session_save(Session *s) {
                         "TYPE=%s\n",
                         session_type_to_string(s->type));
 
                         "TYPE=%s\n",
                         session_type_to_string(s->type));
 

+        if (s->class >= 0)
+                fprintf(f,
+                        "CLASS=%s\n",
+                        session_class_to_string(s->class));
+

         if (s->cgroup_path)
                 fprintf(f,
                         "CGROUP=%s\n",
         if (s->cgroup_path)
                 fprintf(f,
                         "CGROUP=%s\n",


@@ -225,7 +230,8 @@ int session_load(Session *s) {
                 *vtnr = NULL,
                 *leader = NULL,
                 *audit_id = NULL,
                 *vtnr = NULL,
                 *leader = NULL,
                 *audit_id = NULL,

-                *type = NULL;
+                *type = NULL,
+                *class = NULL;

 
         int k, r;
 
 
         int k, r;
 


@@ -245,6 +251,7 @@ int session_load(Session *s) {
                            "VTNR",           &vtnr,
                            "LEADER",         &leader,
                            "TYPE",           &type,
                            "VTNR",           &vtnr,
                            "LEADER",         &leader,
                            "TYPE",           &type,

+                           "CLASS",          &class,

                            NULL);
 
         if (r < 0)
                            NULL);
 
         if (r < 0)


@@ -297,6 +304,14 @@ int session_load(Session *s) {
                         s->type = t;
         }
 
                         s->type = t;
         }
 

+        if (class) {
+                SessionClass c;
+
+                c = session_class_from_string(class);
+                if (c >= 0)
+                        s->class = c;
+        }
+

         if (s->fifo_path) {
                 int fd;
 
         if (s->fifo_path) {
                 int fd;
 


@@ -434,7 +449,7 @@ static int session_create_one_group(Session *s, const char *controller, const ch
         if (r < 0)
                 return r;
 
         if (r < 0)
                 return r;
 

-        r = cg_set_task_access(controller, path, 0644, s->user->uid, s->user->gid);
+        r = cg_set_task_access(controller, path, 0644, s->user->uid, s->user->gid, -1);

         if (r >= 0)
                 r = cg_set_group_access(controller, path, 0755, s->user->uid, s->user->gid);
 
         if (r >= 0)
                 r = cg_set_group_access(controller, path, 0755, s->user->uid, s->user->gid);
 


@@ -601,6 +616,23 @@ static int session_terminate_cgroup(Session *s) {
                         log_error("Failed to kill session cgroup: %s", strerror(-r));
 
         } else {
                         log_error("Failed to kill session cgroup: %s", strerror(-r));
 
         } else {

+                if (s->leader > 0) {
+                        Session *t;
+
+                        /* We still send a HUP to the leader process,
+                         * even if we are not supposed to kill the
+                         * whole cgroup. But let's first check the
+                         * leader still exists and belongs to our
+                         * session... */
+
+                        r = manager_get_session_by_pid(s->manager, s->leader, &t);
+                        if (r > 0 && t == s) {
+                                kill(s->leader, SIGTERM); /* for normal processes */
+                                kill(s->leader, SIGHUP);  /* for shells */
+                                kill(s->leader, SIGCONT); /* in case they are stopped */
+                        }
+                }
+

                 r = cg_is_empty_recursive(SYSTEMD_CGROUP_CONTROLLER, s->cgroup_path, true);
                 if (r < 0)
                         log_error("Failed to check session cgroup: %s", strerror(-r));
                 r = cg_is_empty_recursive(SYSTEMD_CGROUP_CONTROLLER, s->cgroup_path, true);
                 if (r < 0)
                         log_error("Failed to check session cgroup: %s", strerror(-r));


@@ -608,8 +640,7 @@ static int session_terminate_cgroup(Session *s) {
                         r = cg_delete(SYSTEMD_CGROUP_CONTROLLER, s->cgroup_path);
                         if (r < 0)
                                 log_error("Failed to delete session cgroup: %s", strerror(-r));
                         r = cg_delete(SYSTEMD_CGROUP_CONTROLLER, s->cgroup_path);
                         if (r < 0)
                                 log_error("Failed to delete session cgroup: %s", strerror(-r));

-                } else
-                        r = -EBUSY;
+                }

         }
 
         STRV_FOREACH(k, s->user->manager->controllers)
         }
 
         STRV_FOREACH(k, s->user->manager->controllers)


@@ -620,7 +651,7 @@ static int session_terminate_cgroup(Session *s) {
         free(s->cgroup_path);
         s->cgroup_path = NULL;
 
         free(s->cgroup_path);
         s->cgroup_path = NULL;
 

-        return r;
+        return 0;

 }
 
 static int session_unlink_x11_socket(Session *s) {
 }
 
 static int session_unlink_x11_socket(Session *s) {


@@ -931,6 +962,14 @@ static const char* const session_type_table[_SESSION_TYPE_MAX] = {
 
 DEFINE_STRING_TABLE_LOOKUP(session_type, SessionType);
 
 
 DEFINE_STRING_TABLE_LOOKUP(session_type, SessionType);
 

+static const char* const session_class_table[_SESSION_CLASS_MAX] = {
+        [SESSION_USER] = "user",
+        [SESSION_GREETER] = "greeter",
+        [SESSION_LOCK_SCREEN] = "lock-screen"
+};
+
+DEFINE_STRING_TABLE_LOOKUP(session_class, SessionClass);
+

 static const char* const kill_who_table[_KILL_WHO_MAX] = {
         [KILL_LEADER] = "leader",
         [KILL_ALL] = "all"
 static const char* const kill_who_table[_KILL_WHO_MAX] = {
         [KILL_LEADER] = "leader",
         [KILL_ALL] = "all"