merged from UV_NAMESPACE_AFTER_CHROOT branch
[util-vserver.git] / util-vserver / src / vcontext.c
index 4a9d12e..0d979cf 100644 (file)
@@ -24,8 +24,6 @@
 #include "lib/internal.h"
 #include "lib_internal/jail.h"
 
-#include "linuxcaps.h"
-
 #include <vserver.h>
 #include <getopt.h>
 #include <fcntl.h>
@@ -33,6 +31,7 @@
 #include <sys/socket.h>
 #include <sys/un.h>
 #include <assert.h>
+#include <signal.h>
 
 #define ENSC_WRAPPERS_PREFIX   "vcontext: "
 #define ENSC_WRAPPERS_UNISTD   1
@@ -47,7 +46,7 @@
 #define CMD_XID                        0x4000
 #define CMD_CREATE             0x4001
 #define CMD_MIGRATE            0x4003
-#define CMD_FAKEINIT           0x4002
+#define CMD_INITPID            0x4002
 #define CMD_DISCONNECT         0x4004
 #define CMD_UID                        0x4005
 #define CMD_CHROOT             0x4006
 #define CMD_SYNCMSG            0x4009
 #define CMD_MIGRATESELF                0x400a
 #define CMD_ENDSETUP           0x400b
+#define CMD_SILENTEXIST                0x400c
+#define CMD_NAMESPACE          0x400d
 
-#if 0
-#define CMD_CCAP               0x2002
-#define CMD_BCAP               0x2005
-#endif
 
 struct option const
 CMDLINE_OPTIONS[] = {
@@ -71,16 +68,19 @@ CMDLINE_OPTIONS[] = {
   { "create",     no_argument,       0, CMD_CREATE },
   { "migrate",    no_argument,       0, CMD_MIGRATE },
   { "migrate-self", no_argument,               0, CMD_MIGRATESELF },
-  { "fakeinit",     no_argument,               0, CMD_FAKEINIT },
+  { "initpid",      no_argument,               0, CMD_INITPID },
   { "endsetup",     no_argument,        0, CMD_ENDSETUP },
   { "disconnect",   no_argument,       0, CMD_DISCONNECT },
   { "silent",       no_argument,               0, CMD_SILENT },
+  { "silentexist",  no_argument,               0, CMD_SILENTEXIST },
   { "uid",          required_argument,  0, CMD_UID },
   { "chroot",       no_argument,               0, CMD_CHROOT },
+  { "namespace",    no_argument,               0, CMD_NAMESPACE },
   { "syncsock",     required_argument,         0, CMD_SYNCSOCK },
   { "syncmsg",      required_argument,         0, CMD_SYNCMSG },
-//  { "ccap",         required_argument,  0, CMD_CCAP },
-//  { "bcap",         required_argument,  0, CMD_BCAP },
+#if 1  
+  { "fakeinit",     no_argument,               0, CMD_INITPID },       // compatibility
+#endif  
   { 0,0,0,0 },
 };
 
@@ -90,14 +90,15 @@ struct Arguments {
     bool               do_migrateself;
     bool               do_disconnect;
     bool               do_endsetup;
-    bool               is_fakeinit;
+    bool               is_initpid;
+    bool               is_silentexist;
+    bool               set_namespace;
     int                        verbosity;
     bool               do_chroot;
     uid_t              uid;
     xid_t              xid;
     char const *       sync_sock;
     char const *       sync_msg;
-    //struct vc_ctx_caps  caps;
 };
 
 int            wrapper_exit_code = 255;
@@ -115,12 +116,15 @@ showHelp(int fd, char const *cmd, int res)
            "\n"
            "<opts> can be:\n"
            "    --chroot        ...  chroot into current directory\n"
+           "    --namespace     ...  execute namespace management operations\n"
            "    --uid <uid>     ...  change uid\n"
-           "    --fakeinit      ...  set current process as general process reaper\n"
+           "    --initpid       ...  set current process as general process reaper\n"
            "                         for ctx (possible for --migrate only)\n"
            "    --endsetup      ...  clear the setup flag; usefully for migrate only\n"
            "    --disconnect    ...  start program in background\n"
            "    --silent        ...  be silent\n"
+           "    --silentexist   ...  be silent when context exists already; usefully\n"
+           "                         for '--create' only\n"
            "    --syncsock <file-name>\n"
            "                    ...  before executing the program, send a message\n"
            "                         to the socket and wait until it closes.\n"
@@ -147,51 +151,7 @@ showVersion()
   exit(0);
 }
 
-static inline ALWAYSINLINE int
-initSync(int p[2], bool do_disconnect)
-{
-  if (!do_disconnect) return 0;
-
-  Epipe(p);
-  fcntl(p[1], F_SETFD, FD_CLOEXEC);
-  return Efork();
-}
-
-static inline ALWAYSINLINE void
-doSyncStage1(int p[2], bool do_disconnect)
-{
-  int  fd;
-
-  if (!do_disconnect) return;
-  
-  fd = Eopen("/dev/null", O_RDONLY, 0);
-  Esetsid();
-  Edup2(fd, 0);
-  Eclose(p[0]);
-  if (fd!=0) Eclose(fd);
-  Ewrite(p[1], ".", 1);
-}
-
-static inline ALWAYSINLINE void
-doSyncStage2(int p[2], bool do_disconnect)
-{
-  if (!do_disconnect) return;
-
-  Ewrite(p[1], "X", 1);
-}
-
-static void
-waitOnSync(pid_t pid, int p[2])
-{
-  int          c;
-  size_t       l;
-
-  Eclose(p[1]);
-  l = Eread(p[0], &c, 1);
-  if (l!=1) exitLikeProcess(pid);
-  l = Eread(p[0], &c, 1);
-  if (l!=0) exitLikeProcess(pid);
-}
+#include "context-sync.hc"
 
 static inline ALWAYSINLINE void
 tellContext(xid_t ctx, bool do_it)
@@ -203,9 +163,9 @@ tellContext(xid_t ctx, bool do_it)
 
   l = utilvserver_fmt_long(buf,ctx);
 
-  WRITE_MSG(2, "New security context is ");
-  write(2, buf, l);
-  WRITE_MSG(2, "\n");
+  WRITE_MSG(1, "New security context is ");
+  write(1, buf, l);
+  WRITE_MSG(1, "\n");
 }
 
 static int
@@ -229,14 +189,16 @@ setFlags(struct Arguments const *args, xid_t xid)
 {
   struct vc_ctx_flags  flags = { 0,0 };
 
-  if (args->is_fakeinit)
-    flags.mask |= VC_VXF_STATE_INIT;
+  if (args->is_initpid)
+    flags.mask |=  VC_VXF_STATE_INIT;
 
   if (args->do_endsetup)
-    flags.mask |= VC_VXF_STATE_SETUP;
+    flags.mask |=  VC_VXF_STATE_SETUP;
 
-  if (flags.mask!=0) 
-    Evc_set_flags(xid, &flags);
+  if (flags.mask!=0) {
+    DPRINTF("set_flags: mask=%08llx, flag=%08llx\n", flags.mask, flags.flagword);
+    Evc_set_cflags(xid, &flags);
+  }
 }
 
 static void
@@ -250,101 +212,36 @@ doExternalSync(int fd, char const *msg)
   Eshutdown(fd, SHUT_WR);
 
   if (TEMP_FAILURE_RETRY(recv(fd, &c, 1, MSG_NOSIGNAL))!=0) {
-    WRITE_MSG(2, "vcontext: unexpected external synchronization event\n");
+    WRITE_MSG(2, ENSC_WRAPPERS_PREFIX "unexpected external synchronization event\n");
     exit(255);
   }
 
   Eclose(fd);
 }
 
-static void UNUSED
-dropPrivs(xid_t xid)
-{
-  extern int capget (struct __user_cap_header_struct *,
-                    struct __user_cap_data_struct *);
-  extern int capset (struct __user_cap_header_struct *,
-                    struct __user_cap_data_struct *);
-  struct vc_ctx_caps                   caps;
-  struct __user_cap_header_struct      header;
-  struct __user_cap_data_struct                user;
-
-  header.version = _LINUX_CAPABILITY_VERSION;
-  header.pid     = 0;
-  
-  Evc_get_ccaps(xid, &caps);
-  if (capget(&header, &user)==-1)
-    perror(ENSC_WRAPPERS_PREFIX "capget()");
-  else if ( (user.permitted   &= caps.ccaps),
-           (user.effective   &= user.permitted),
-           (user.inheritable &= user.permitted),
-           capset(&header, &user)==-1 )
-    perror(ENSC_WRAPPERS_PREFIX "capset()");
-  else
-    return;
-
-  exit(wrapper_exit_code);
-}
-
-#if 0
-static void
-parseBCaps(char const *str, struct vc_ctx_caps *caps)
-{
-  struct vc_err_listparser     err;
-  int                          rc;
-
-  rc = vc_list2bcap(str,0, &err, caps);
-  
-  if (rc==-1) {
-    WRITE_MSG(2, "Unknown bcap '");
-    write(2, err.ptr, err.len);
-    WRITE_MSG(2, "'\n");
-    exit(wrapper_exit_code);
-  }
-}
-
-static void
-parseCCaps(char const *str, struct vc_ctx_caps *caps)
-{
-  struct vc_err_listparser     err;
-  int                          rc;
-
-  rc = vc_list2ccap(str,0, &err, caps);
-  
-  if (rc==-1) {
-    WRITE_MSG(2, "Unknown ccap '");
-    write(2, err.ptr, err.len);
-    WRITE_MSG(2, "'\n");
-    exit(wrapper_exit_code);
-  }
-}
-
-static void
-parseSecure(struct vc_ctx_flags UNUSED * flags,
-           struct vc_ctx_caps  UNUSED * caps)
-{
-  caps->ccaps = ~0;
-  caps->cmask = ~0;
-  caps->bcaps = ~vc_get_securecaps();
-}
-#endif
-
 static inline ALWAYSINLINE int
 doit(struct Arguments const *args, char *argv[])
 {
-  int                  p[2];
+  int                  p[2][2];
   pid_t                        pid = initSync(p, args->do_disconnect);
   
   if (pid==0) {
     xid_t                      xid;
     int                                ext_sync_fd = connectExternalSync(args->sync_sock);
+
+    doSyncStage0(p, args->do_disconnect);
     
     if (args->do_create) {
-      xid = vc_create_context(args->xid);
+      xid = vc_ctx_create(args->xid);
       if (xid==VC_NOCTX) {
-       if (errno==EEXIST) return 254;
-       else {
-         perror(ENSC_WRAPPERS_PREFIX "vc_create_context()");
-         return wrapper_exit_code;
+       switch (errno) {
+         case EEXIST   :
+           if (!args->is_silentexist)
+             perror(ENSC_WRAPPERS_PREFIX "vc_create_context()");
+           return 254;
+         default       :
+           perror(ENSC_WRAPPERS_PREFIX "vc_create_context()");
+           return wrapper_exit_code;
        }
       }
       tellContext(xid, args->verbosity>=1);
@@ -352,45 +249,40 @@ doit(struct Arguments const *args, char *argv[])
     else
       xid = args->xid;
 
-    if (args->do_chroot)
+    if (args->do_chroot) {
       Echroot(".");
+      if (args->set_namespace) {
+       if (args->do_migrateself)  Evc_set_namespace();
+       else if (args->do_migrate) Evc_enter_namespace(xid);
+      }
+    }
 
     setFlags(args, xid);
 
-#if 0
-    if (args->do_migrate ||  args->do_migrateself)
-      dropPrivs(xid);
-#endif
-
     if (args->do_migrate && !args->do_migrateself)
-      Evc_migrate_context(xid);
+      Evc_ctx_migrate(xid);
 
     if (args->uid!=(uid_t)(-1) && getuid()!=args->uid) {
       Esetuid(args->uid);
       if (getuid()!=args->uid) {
-       WRITE_MSG(2, "vcontext: Something went wrong while changing the UID\n");
+       WRITE_MSG(2, ENSC_WRAPPERS_PREFIX "Something went wrong while changing the UID\n");
        exit(255);
       }
     }
     
     doExternalSync(ext_sync_fd, args->sync_msg);
     doSyncStage1(p, args->do_disconnect);
+    DPRINTF("doit: pid=%u, ppid=%u\n", getpid(), getppid());
     execvp (argv[optind],argv+optind);
     doSyncStage2(p, args->do_disconnect);
 
-    PERROR_Q("chcontext: execvp", argv[optind]);
+    PERROR_Q(ENSC_WRAPPERS_PREFIX "execvp", argv[optind]);
     exit(255);
   }
 
   assert(args->do_disconnect);
-
-#warning add an additional sync here  
-  if (!jailIntoTempDir()) {
-    perror(ENSC_WRAPPERS_PREFIX "jailIntoTempDir()");
-    exit(255);
-  }
     
-  waitOnSync(pid, p);
+  waitOnSync(pid, p, args->xid!=VC_DYNAMIC_XID && args->do_migrate);
   return EXIT_SUCCESS;
 }
 
@@ -402,7 +294,9 @@ int main (int argc, char *argv[])
     .do_migrateself = false,
     .do_disconnect  = false,
     .do_endsetup    = false,
-    .is_fakeinit    = false,
+    .is_initpid     = false,
+    .is_silentexist = false,
+    .set_namespace  = false,
     .verbosity      = 1,
     .uid            = -1,
     .xid            = VC_DYNAMIC_XID,
@@ -416,22 +310,19 @@ int main (int argc, char *argv[])
     switch (c) {
       case CMD_HELP            :  showHelp(1, argv[0], 0);
       case CMD_VERSION         :  showVersion();
-      case CMD_CREATE          :  args.do_create     = true; break;
-      case CMD_MIGRATE         :  args.do_migrate    = true; break;
-      case CMD_DISCONNECT      :  args.do_disconnect = true; break;
-      case CMD_ENDSETUP                :  args.do_endsetup   = true; break;
-      case CMD_FAKEINIT                :  args.is_fakeinit   = true; break;
-      case CMD_CHROOT          :  args.do_chroot     = true; break;
-      case CMD_SILENT          :  --args.verbosity;          break;
-      case CMD_XID             :  args.xid           = atol(optarg); break;
-      case CMD_UID             :  args.uid           = atol(optarg); break;
-      case CMD_SYNCSOCK                :  args.sync_sock     = optarg;    break;
-      case CMD_SYNCMSG         :  args.sync_msg      = optarg;    break;
-#if 0  
-      case CMD_CCAP            :  parseCCaps(optarg, &args.caps); break;
-      case CMD_BCAP            :  parseBCaps(optarg, &args.caps); break;
-      case CMD_SECURE          :  parseSecure(&args.caps);        break;
-#endif
+      case CMD_CREATE          :  args.do_create      = true;   break;
+      case CMD_MIGRATE         :  args.do_migrate     = true;   break;
+      case CMD_DISCONNECT      :  args.do_disconnect  = true;   break;
+      case CMD_ENDSETUP                :  args.do_endsetup    = true;   break;
+      case CMD_INITPID         :  args.is_initpid     = true;   break;
+      case CMD_CHROOT          :  args.do_chroot      = true;   break;
+      case CMD_NAMESPACE       :  args.set_namespace  = true;   break;
+      case CMD_SILENTEXIST     :  args.is_silentexist = true;   break;
+      case CMD_SYNCSOCK                :  args.sync_sock      = optarg; break;
+      case CMD_SYNCMSG         :  args.sync_msg       = optarg; break;
+      case CMD_UID             :  args.uid = atol(optarg);      break;
+      case CMD_XID             :  args.xid = Evc_xidopt2xid(optarg,true); break;
+      case CMD_SILENT          :  --args.verbosity; break;
       case CMD_MIGRATESELF     :
        args.do_migrate     = true;
        args.do_migrateself = true;
@@ -446,6 +337,8 @@ int main (int argc, char *argv[])
     }
   }
 
+  signal(SIGCHLD, SIG_DFL);
+  
   if (args.do_migrateself)
     args.xid = Evc_get_task_xid(0);
   
@@ -453,10 +346,10 @@ int main (int argc, char *argv[])
     WRITE_MSG(2, "Neither '--create' nor '--migrate specified; try '--help' for more information\n");
   else if (args.do_create  &&  args.do_migrate)
     WRITE_MSG(2, "Can not specify '--create' and '--migrate' at the same time; try '--help' for more information\n");
-  else if (!args.do_migrate && args.is_fakeinit)
-    WRITE_MSG(2, "'--fakeinit' is possible in combination with '--migrate' only\n");
+  else if (!args.do_migrate && args.is_initpid)
+    WRITE_MSG(2, "'--initpid' is possible in combination with '--migrate' only\n");
   else if (!args.do_create && args.xid==VC_DYNAMIC_XID)
-    WRITE_MSG(2, "vcontext: Can not migrate to an unknown context\n");
+    WRITE_MSG(2, ENSC_WRAPPERS_PREFIX "Can not migrate to an unknown context\n");
   else if (optind>=argc)
     WRITE_MSG(2, "No command given; use '--help' for more information.\n");
   else