suidgw.c

   1 /*
   2  *  A simple gateway for set-uid scripts
   3  *
   4  *  (c) 2013 Martin Mares <mj@ucw.cz>
   5  */
   6
   7 #undef DEBUG
   8
   9 #define _GNU_SOURCE
  10
  11 #include <errno.h>
  12 #include <fcntl.h>
  13 #include <limits.h>
  14 #include <pwd.h>
  15 #include <stdarg.h>
  16 #include <stdbool.h>
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19 #include <string.h>
  20 #include <syslog.h>
  21 #include <sys/stat.h>
  22 #include <unistd.h>
  23
  24 #define NONRET __attribute__((noreturn))
  25 #define UNUSED __attribute__((unused))
  26
  27 #ifdef DEBUG
  28 #define DBG(...) printf(__VA_ARGS__)
  29 #else
  30 #define DBG(...) do { } while (0)
  31 #endif
  32
  33 static char program_name[PATH_MAX];
  34 static char script_path[PATH_MAX];
  35 static uid_t use_uid;
  36 static gid_t use_gid;
  37
  38 static const char * const script_dirs[] = {
  39   "/usr/local/lib/suidgw",
  40   "/usr/lib/suidgw",
  41 #ifdef DEBUG
  42   "./tests/scripts",
  43 #endif
  44   NULL
  45 };
  46
  47 static char env_orig_uid[32], env_orig_gid[32];
  48
  49 static char *sanitized_env[] = {
  50   "PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin",
  51   env_orig_uid,
  52   env_orig_gid,
  53   NULL
  54 };
  55
  56 static void NONRET die(const char *fmt, ...)
  57 {
  58   va_list args;
  59   va_start(args, fmt);
  60   fprintf(stderr, "suidgw: ");
  61   vfprintf(stderr, fmt, args);
  62   fputc('\n', stderr);
  63   exit(1);
  64 }
  65
  66 static void sanitize_fds(void)
  67 {
  68   // Make sure that nobody is playing dirty games with our file descriptors
  69   int fd;
  70   do
  71     {
  72       fd = open("/dev/null", O_RDWR);
  73       if (fd < 0)
  74         die("Cannot open /dev/null: %m");
  75     }
  76   while (fd <= 2);
  77   close(fd);
  78 }
  79
  80 static void sanitize_env(void)
  81 {
  82   snprintf(env_orig_uid, sizeof(env_orig_uid), "ORIG_UID=%d", (int) getuid());
  83   snprintf(env_orig_gid, sizeof(env_orig_gid), "ORIG_GID=%d", (int) getgid());
  84 }
  85
  86 static bool get_program_name(const char *arg0)
  87 {
  88   // If arg0 is a path, extract the last component
  89   const char *p = strrchr(arg0, '/');
  90   if (p)
  91     p++;
  92   else
  93     p = arg0;
  94
  95   // Reject empty and oversized names
  96   if (!p[0] || strlen(p) >= PATH_MAX)
  97     return 0;
  98
  99   // Reject invalid characters
 100   for (const char *q = p; *q; q++)
 101     {
 102       int c = *q;
 103       if (! (c >= 'a' && c <= 'z' ||
 104              c >= 'A' && c <= 'Z' ||
 105              c >= '0' && c <= '9' ||
 106              c == '-' ||
 107              c == '_'))
 108         return 0;
 109     }
 110
 111   DBG("Program name: <%s>\n", p);
 112   strcpy(program_name, p);
 113   return 1;
 114 }
 115
 116 static void find_script(void)
 117 {
 118   for (int i = 0; script_dirs[i]; i++)
 119     {
 120       int len = snprintf(script_path, PATH_MAX, "%s/%s", script_dirs[i], program_name);
 121       if (len >= PATH_MAX)
 122         continue;
 123       DBG("Trying <%s>\n", script_path);
 124       if (access(script_path, X_OK) >= 0)
 125         {
 126           DBG("Found <%s>\n", script_path);
 127           return;
 128         }
 129       if (errno != ENOENT)
 130         die("Cannot run %s: %m", script_path);
 131     }
 132
 133   die("Unable to find script %s", program_name);
 134 }
 135
 136 static void check_stat(void)
 137 {
 138   struct stat st;
 139   if (stat(script_path, &st) < 0)
 140     die("Cannot stat %s: %m", script_path);
 141
 142   if (!S_ISREG(st.st_mode))
 143     die("Not a regular file: %s", script_path);
 144   if (!(st.st_mode & (S_ISUID | S_ISGID)))
 145     die("Missing suid/sgid: %s", script_path);
 146
 147   use_uid = (st.st_mode & S_ISUID) ? st.st_uid : getuid();
 148   use_gid = (st.st_mode & S_ISGID) ? st.st_gid : getgid();
 149   DBG("Will use uid=%d gid=%d\n", (int) use_uid, (int) use_gid);
 150 }
 151
 152 static void switch_ugid(void)
 153 {
 154   if (setresgid(use_gid, use_gid, use_gid) < 0)
 155     die("Failed to set group id: %m");
 156
 157   if (setresuid(use_uid, use_uid, use_uid) < 0)
 158     die("Failed to set user id: %m");
 159 }
 160
 161 int main(int argc UNUSED, char **argv)
 162 {
 163   sanitize_fds();
 164   sanitize_env();
 165
 166   if (geteuid())
 167     die("Must be run setuid");
 168
 169   struct passwd *pwd = getpwuid(getuid());
 170   if (!pwd)
 171     die("You don't exist");
 172
 173   openlog("suidgw", LOG_NDELAY | LOG_PID, LOG_AUTH);
 174
 175   if (!get_program_name(argv[0]))
 176     die("Unable to parse program name %s", argv[0]);
 177
 178   find_script();
 179   check_stat();
 180   switch_ugid();
 181
 182   syslog(LOG_INFO, "User %s executes %s with uid=%d, gid=%d",
 183     pwd->pw_name, script_path, (int) use_uid, (int) use_gid);
 184
 185   if (execve(script_path, argv, sanitized_env) < 0)
 186     die("Cannot execute %s: %m", script_path);
 187
 188   die("This must never happen");
 189 }