Check setgid too after dropping root

swaywm · Jan 16, 2024 · b63aaff · b63aaff
1 parent bb32fd1
commit b63aaff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/shadow.c b/shadow.c
@@ -33,9 +33,9 @@ void initialize_pw_backend(int argc, char **argv) {
 		swaylock_log_errno(LOG_ERROR, "Unable to drop root");
 		exit(EXIT_FAILURE);
 	}
-	if (setuid(0) != -1) {
+	if (setuid(0) != -1 || setgid(0) != -1) {
 		swaylock_log_errno(LOG_ERROR, "Unable to drop root (we shouldn't be "
-			"able to restore it after setuid)");
+			"able to restore it after setuid/setgid)");
 		exit(EXIT_FAILURE);
 	}
 }