src/vsh.c

   1 /* Version 2 of vsh. Basically a wrapper around 'vserver <slice name> enter.' */
   2
   3 #include <unistd.h>
   4 #include <pwd.h>
   5 #include <string.h>
   6 #include <sys/types.h>
   7 #include <sys/stat.h>
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include <errno.h>
  11
  12 #define VSH_PATH    "/usr/sbin/vsh"
  13 #ifndef PATH_MAX
  14 #define PATH_MAX    4096
  15 #endif
  16
  17 #define VSERVER_PATH  "/usr/sbin/vserver"
  18
  19 char* get_current_username (unsigned int uid)
  20 {
  21     struct passwd *passwd_entry;
  22     if ((passwd_entry = getpwuid(uid)) == NULL) {
  23         fprintf(stderr, "Could not look up user record for %d\n", uid);
  24         return NULL;
  25     }
  26
  27     return (strdup(passwd_entry->pw_name));
  28 }
  29
  30 char **extend_argv(int argc, char **argv, int num_extra_args) {
  31     int argc2, i;
  32     char **argv2;
  33
  34     argc2 = argc + num_extra_args;
  35     argv2 = (char **) malloc((argc2 + 1) * sizeof(char *));
  36
  37     if (!argv2)
  38         return (char **) NULL;
  39
  40     for (i=0; i<argc; i++) {
  41         argv2[i+num_extra_args]=strdup(argv[i]);
  42     }
  43     argv2[argc2]=NULL;
  44
  45     return argv2;
  46 }
  47
  48 #define NUM_VSERVER_SUEXEC_ARGS 5
  49
  50 int main(int argc, char **argv)
  51 {
  52     char *slice_name;
  53     char **argv2;
  54     int argc2;
  55     char slice_id_str[256];
  56     unsigned int slice_xid;
  57     char *envp[] = { NULL, NULL };
  58     char home_env_str[256];
  59
  60     slice_xid = getuid();
  61     slice_name = get_current_username(slice_xid);
  62     if (!slice_name) {
  63         fprintf(stderr,"Could not look up slice name\n");
  64         goto out_exception;
  65     }
  66
  67     argv2 = extend_argv(argc, argv, NUM_VSERVER_SUEXEC_ARGS);
  68     if (!argv2) goto out_exception;
  69
  70
  71     // Populate arguments
  72     snprintf(slice_id_str, 255, "%u", slice_xid);
  73     argv2[0] = strdup(VSERVER_PATH);
  74     argv2[1] = strdup(slice_name);
  75     argv2[2] = strdup("suexec");
  76     argv2[3] = strdup(slice_id_str);
  77     argv2[4] = strdup("/bin/bash");
  78     argv2[5] = strdup("-login");
  79
  80     snprintf(home_env_str, 255, "HOME=/home/%s", slice_name);
  81     envp[0] = home_env_str;
  82
  83     if (setuid(geteuid())) goto out_exception;
  84
  85     execve(VSERVER_PATH, argv2, envp);
  86
  87 out_exception:
  88     printf("%s\n", strerror(errno));
  89     return errno;
  90 }