1 files changed, 116 insertions, 0 deletions
diff --git a/softlimit.c b/softlimit.c
new file mode 100644
index 0000000..fb3233f
--- /dev/null
+++ b/softlimit.c
@@ -0,0 +1,116 @@
+#include <sys/types.h>
+#include <sys/resource.h>
+#include "ninitfeatures.h"
+
+static char *arg;
+
+static void doit(int resource)
+{
+  unsigned long u;
+  struct rlimit r;
+
+  if (getrlimit(resource,&r) == -1) {
+    carp("g","etrlimit failed");
+    return;
+  }
+  
+  if (arg[0]=='=' && arg[1]==0)
+    r.rlim_cur = r.rlim_max;
+  else {
+    if (arg[scan_ulong(arg,&u)]) {
+      carp("invalid line: ", arg-1);
+      return;
+    }
+    r.rlim_cur = u;
+    if (r.rlim_cur > r.rlim_max)
+      r.rlim_cur = r.rlim_max;
+  }
+
+  if (setrlimit(resource,&r) == -1)
+    carp("s","etrlimit failed");
+}
+
+void softlimit(char **argv) /*EXTRACT_INCL*/ {
+  while (*argv) {
+    arg = argv[0] + 1;
+    switch(argv[0][0]) {
+      case 0: break;
+      case 'a':
+#ifdef RLIMIT_AS
+        doit(RLIMIT_AS);
+#endif
+#ifdef RLIMIT_VMEM
+        doit(RLIMIT_VMEM);
+#endif
+        break;
+      case 'c':
+#ifdef RLIMIT_CORE
+        doit(RLIMIT_CORE);
+#endif
+        break;
+      case 'd':
+#ifdef RLIMIT_DATA
+        doit(RLIMIT_DATA);
+#endif
+        break;
+      case 'f':
+#ifdef RLIMIT_FSIZE
+        doit(RLIMIT_FSIZE);
+#endif
+        break;
+      case 'l':
+#ifdef RLIMIT_MEMLOCK
+        doit(RLIMIT_MEMLOCK);
+#endif
+        break;
+      case 'm':
+#ifdef RLIMIT_DATA
+        doit(RLIMIT_DATA);
+#endif
+#ifdef RLIMIT_STACK
+        doit(RLIMIT_STACK);
+#endif
+#ifdef RLIMIT_MEMLOCK
+        doit(RLIMIT_MEMLOCK);
+#endif
+#ifdef RLIMIT_VMEM
+        doit(RLIMIT_VMEM);
+#endif
+#ifdef RLIMIT_AS
+        doit(RLIMIT_AS);
+#endif
+	break;
+      case 'o':
+#ifdef RLIMIT_NOFILE
+        doit(RLIMIT_NOFILE);
+#endif
+#ifdef RLIMIT_OFILE
+        doit(RLIMIT_OFILE);
+#endif
+        break;
+      case 'p':
+#ifdef RLIMIT_NPROC
+        doit(RLIMIT_NPROC);
+#endif
+        break;
+      case 'r':
+#ifdef RLIMIT_RSS
+        doit(RLIMIT_RSS);
+#endif
+        break;
+      case 's':
+#ifdef RLIMIT_STACK
+        doit(RLIMIT_STACK);
+#endif
+        break;
+      case 't':
+#ifdef RLIMIT_CPU
+        doit(RLIMIT_CPU);
+#endif
+        break;
+      default:
+	carp("invalid line: ", argv[0]);
+    }
+    argv++;
+  }
+}