CHeck the output of setgid and setuid.

2023-01-20 09:26:31 -05:00
parent 20323ba64e
commit 2a06eb791a
1 changed files with 9 additions and 3 deletions
@@ -93,7 +93,7 @@ static inline bool_t __get_input(char *input, size_t max_len, bool_t hide)
        } else if (c == '\033') {
            c = getchar();
            if (c == '[') {
-                c = getchar(); // Get the char, and ignore it.
+                getchar(); // Get the char, and ignore it.
            } else if (c == '^') {
                c = getchar(); // Get the char.
                if (c == 'C') {
@@ -199,10 +199,16 @@ int main(int argc, char **argv)
    }

    // Set the group id.
-    setgid(pwd->pw_gid);
+    if (setgid(pwd->pw_gid) < 0) {
+        printf("login: Failed to change group id: %s\n", strerror(errno));
+        return 1;
+    }

    // Set the user id.
-    setuid(pwd->pw_uid);
+    if (setuid(pwd->pw_uid) < 0) {
+        printf("login: Failed to change user id: %s\n", strerror(errno));
+        return 1;
+    }

    printf("\n");