use $(LIBENSCVECTOR) instead of libensc_vector.a
[util-vserver.git] / util-vserver / src / exec-ulimit.c
index 8421943..a487963 100644 (file)
 #  include <config.h>
 #endif
 
-#include "wrappers.h"
 #include "util.h"
 
 #include <sys/time.h>
 #include <sys/resource.h>
 #include <unistd.h>
+#include <fcntl.h>
+#include <errno.h>
+
+#define ENSC_WRAPPERS_PREFIX   "exec-ulimit: "
+#define ENSC_WRAPPERS_UNISTD   1
+#define ENSC_WRAPPERS_FCNTL    1
+#define ENSC_WRAPPERS_RESOURCE 1
+#include <wrappers.h>
 
 #define DECLARE_LIMIT(RES,FNAME) { #FNAME, RLIMIT_##RES }
 
@@ -35,6 +42,7 @@ static struct {
     char const *fname;
     int                code;
 } const LIMITS[] = {
+  DECLARE_LIMIT(CORE,    core),
   DECLARE_LIMIT(CPU,     cpu),
   DECLARE_LIMIT(DATA,    data),
   DECLARE_LIMIT(FSIZE,   fsize),
@@ -58,6 +66,16 @@ readValue(int fd, char const *filename)
   if (strncmp(buf, "inf", 3)==0) return RLIM_INFINITY;
   res = strtol(buf, &errptr, 0);
 
+  if (errptr!=buf) {
+    switch (*errptr) {
+      case 'M' :  res *= 1024; /* fallthrough */
+      case 'K' :  res *= 1024; ++errptr; break;
+      case 'm' :  res *= 1000; /* fallthrough */
+      case 'k' :  res *= 1000; ++errptr; break;
+      default  :  break;
+    }
+  }
+
   if (errptr==buf || (*errptr!='\0' && *errptr!='\n')) {
     WRITE_MSG(2, "Invalid limit in '");
     WRITE_STR(2, filename);
@@ -115,13 +133,40 @@ readSingleLimit(struct rlimit *lim, char const *fname_base)
   return is_modified;
 }
 
+static void
+showHelp(int fd, char const *cmd, int res)
+{
+  WRITE_MSG(fd, "Usage:  ");
+  WRITE_STR(fd, cmd);
+  WRITE_STR(fd,
+           "<ulimit-cfgdir> <cmd> <argv>*\n\n"
+           "Please report bugs to " PACKAGE_BUGREPORT "\n");
+  exit(res);
+}
+
+static void
+showVersion()
+{
+  WRITE_MSG(1,
+           "exec-ulimit " VERSION " -- executes programs with resource limits\n"
+           "This program is part of " PACKAGE_STRING "\n\n"
+           "Copyright (C) 2003 Enrico Scholz\n"
+           VERSION_COPYRIGHT_DISCLAIMER);
+  exit(0);
+}
+
 int main(int argc, char *argv[])
 {
   size_t               i;
   int                  cur_fd = Eopen(".", O_RDONLY, 0);
-  
+
+  if (argc==2) {
+    if (strcmp(argv[1], "--help")==0)    showHelp(1,argv[0],0);
+    if (strcmp(argv[1], "--version")==0) showVersion();
+  }
+
   if (argc<3) {
-    WRITE_MSG(2, "Usage: exec-ulimit <ulimit-cfgdir> <cmd> <argv>*\n");
+    WRITE_MSG(2, "Bad parameter count; use '--help' for more information.\n");
     exit(255);
   }