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_EXEC_ARGS 5
  49
  50 int main(int argc, char **argv, char **envp)
  51 {
  52     char *slice_name;
  53     char **argv2;
  54     int argc2;
  55     char slice_id_str[256];
  56     unsigned int slice_xid;
  57
  58     slice_xid = getuid();
  59     slice_name = get_current_username(slice_xid);
  60     if (!slice_name) {
  61         fprintf(stderr,"Could not look up slice name\n");
  62         goto out_exception;
  63     }
  64
  65     argv2 = extend_argv(argc, argv, NUM_VSERVER_EXEC_ARGS);
  66     if (!argv2) goto out_exception;
  67
  68
  69     // Populate arguments
  70     snprintf(slice_id_str, 255, "%u", slice_xid);
  71     argv2[0] = strdup(VSERVER_PATH);
  72     argv2[1] = strdup(slice_name);
  73     argv2[2] = strdup("exec");
  74     argv2[3] = strdup("/bin/su");
  75     argv2[4] = strdup("-");
  76     argv2[5] = strdup(slice_name);
  77
  78     if (setuid(geteuid())) goto out_exception;
  79
  80     /*
  81     char **env_n;
  82     FILE *log_file;
  83     log_file = fopen("/tmp/arglogs","w");
  84     for (env_n = envp;*env_n;env_n++) {
  85         fprintf(log_file,"Env: %s\n",*env_n);
  86     }
  87     fclose(log_file);
  88     */
  89
  90     execve(VSERVER_PATH, argv2, envp);
  91
  92 out_exception:
  93     printf("%s\n", strerror(errno));
  94     return errno;
  95 }