minor optimizations
[util-vserver.git] / util-vserver / src / vps.c
index a02b09f..48bfaa8 100644 (file)
 #endif
 
 #include "util.h"
-#include "wrappers.h"
-#include "wrappers-vserver.h"
 #include "pathconfig.h"
 
 #include <lib/vserver.h>
 #include <lib/fmt.h>
 #include <assert.h>
+#include <fcntl.h>
+#include <signal.h>
+
+#define ENSC_WRAPPERS_VSERVER  1
+#define ENSC_WRAPPERS_STDLIB   1
+#define ENSC_WRAPPERS_UNISTD   1
+#define ENSC_WRAPPERS_FCNTL    1
+#include <wrappers.h>
 
 #define CTXNR_WIDTH    5
 #define HUNK_SIZE      0x4000
@@ -71,14 +77,21 @@ showVersion()
 static size_t
 writeContextInfo(xid_t ctx, char const *name)
 {
-  char         buf[sizeof(ctx)*3+1];
-  size_t       l   = utilvserver_fmt_ulong(buf, ctx);
   size_t       l1  = name==0 ? 0 : strlen(name);
   size_t       res = CTXNR_WIDTH + 1;
+  
+  if (ctx==VC_NOCTX) {
+    if (3<CTXNR_WIDTH) write(1, CONTEXT_PLACE, CTXNR_WIDTH-3);
+    write(1, "ERR ", 4);
+  }
+  else {
+    char       buf[sizeof(ctx)*3+1];
+    size_t     l = utilvserver_fmt_ulong(buf, ctx);
 
-  if (l<CTXNR_WIDTH) write(1, CONTEXT_PLACE, CTXNR_WIDTH-l);
-  write(1, buf, l);
-  write(1, " ", 1);
+    if (l<CTXNR_WIDTH) write(1, CONTEXT_PLACE, CTXNR_WIDTH-l);
+    write(1, buf, l);
+    write(1, " ", 1);
+  }
 
   if (l1!=0) {
     assert(name!=0);
@@ -96,7 +109,7 @@ extractCtx(char *pid_str)
   while (*pid_str==' ') ++pid_str;
   pid = atoi(pid_str);
 
-  return vc_X_getctx(pid);
+  return vc_get_task_xid(pid);
 }
 
 static char const *
@@ -197,6 +210,7 @@ processOutput(char *data, size_t len)
     write(1, data, pid_end);
     l = writeContextInfo(ctx, vserver_name);
     if (l<CONTEXT_WIDTH) write(1, CONTEXT_PLACE, CONTEXT_WIDTH-l);
+    else                 write(1, " ", 1);
     write(1, data+pid_end, eol_pos-(data+pid_end));
     write(1, "\n", 1);
     
@@ -211,14 +225,19 @@ int main(int argc, char *argv[])
   pid_t                pid;
   char *       data;
   size_t       len;
+  char const * errptr;
 
   if (argc>1) {
     if (strcmp(argv[1], "--help")   ==0) showHelp(1, argv[0], 0);
     if (strcmp(argv[1], "--version")==0) showVersion();
   }
-    
-  if (vc_X_getctx(0)!=1)
-    Evc_new_s_context(1, vc_get_securecaps(), 0);
+
+  signal(SIGCHLD, SIG_DFL);
+
+  if (!switchToWatchXid(&errptr)) {
+    perror(errptr);
+    exit(wrapper_exit_code);
+  }
 
   Epipe(p);
   pid = Efork();
@@ -241,8 +260,5 @@ int main(int argc, char *argv[])
   Eclose(p[0]);
   
   processOutput(data, len);
-
-  exitLikeProcess(pid);
-  perror("exitLikeProcess()");
-  return wrapper_exit_code;
+  exitLikeProcess(pid, "ps", wrapper_exit_code);
 }