Fix environment vars after setuid

43146158 · gfleury · d74db187 · 43146158
Commit 43146158 authored Dec 21, 2014 by gfleury
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

main.c main.c +5 -1

No files found.
--- a/main.c
+++ b/main.c
@@ -566,7 +566,11 @@ int change_user_group () {
    if (setuid (pw->pw_uid) < 0) {
      fprintf (stderr, "change_user_group: failed to assume identity of user %s\n", username);
      return -1;
-    }
+    } else {
+      pw = getpwuid(getuid());
+      setenv("USER", pw->pw_name, 1);
+      setenv("HOME", pw->pw_dir, 1);
+      setenv("SHELL", pw->pw_shell, 1);
  }
  return 0;
 }