vsh.c

   1 /* Version 3 of vsh. Basically a wrapper around 'lxcsu-internal -n -m <slicename>' - Sapan */
   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 LXCSU_PATH  "/usr/sbin/lxcsu-internal"
  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_LXCSU_EXEC_ARGS 2
  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_LXCSU_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(LXCSU_PATH);
  72     argv2[1] = strdup(slice_name);
  73     argv2[2] = strdup("--");
  74
  75     if (setuid(geteuid())) goto out_exception;
  76
  77     execve(LXCSU_PATH, argv2, envp);
  78
  79 out_exception:
  80     printf("%s\n", strerror(errno));
  81     return errno;
  82 }