--- /dev/null
+// $Id$ --*- c -*--
+
+// Copyright (C) 2003 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
+//
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation; version 2 of the License.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+
+
+#ifdef HAVE_CONFIG_H
+# include <config.h>
+#endif
+
+#include "wrappers.h"
+#include "util.h"
+
+#include <sys/time.h>
+#include <sys/resource.h>
+#include <unistd.h>
+
+#define DECLARE_LIMIT(RES,FNAME) { #FNAME, RLIMIT_##RES }
+
+int wrapper_exit_code = 255;
+
+static struct {
+ char const *fname;
+ int code;
+} const LIMITS[] = {
+ DECLARE_LIMIT(CPU, cpu),
+ DECLARE_LIMIT(DATA, data),
+ DECLARE_LIMIT(FSIZE, fsize),
+ DECLARE_LIMIT(LOCKS, locks),
+ DECLARE_LIMIT(MEMLOCK, memlock),
+ DECLARE_LIMIT(NOFILE, nofile),
+ DECLARE_LIMIT(NPROC, nproc),
+ DECLARE_LIMIT(RSS, rss),
+ DECLARE_LIMIT(STACK, stack),
+};
+
+static rlim_t
+readValue(int fd, char const *filename)
+{
+ char buf[128];
+ size_t len = Eread(fd, buf, sizeof(buf)-1);
+ long int res;
+ char * errptr;
+
+ buf[len] = '\0';
+ if (strncmp(buf, "inf", 3)==0) return RLIM_INFINITY;
+ res = strtol(buf, &errptr, 0);
+
+ if (errptr==buf || (*errptr!='\0' && *errptr!='\n')) {
+ WRITE_MSG(2, "Invalid limit in '");
+ WRITE_STR(2, filename);
+ WRITE_STR(2, "'\n");
+ exit(255);
+ }
+
+ return res;
+}
+
+static bool
+readSingleLimit(struct rlimit *lim, char const *fname_base)
+{
+ size_t fname_len = strlen(fname_base);
+ char fname[fname_len + sizeof(".hard")];
+ int fd;
+ bool is_modified = false;
+
+ strcpy(fname, fname_base);
+
+ fd = open(fname, O_RDONLY);
+ if (fd!=-1) {
+ rlim_t tmp = readValue(fd, fname_base);
+ lim->rlim_cur = tmp;
+ lim->rlim_max = tmp;
+ Eclose(fd);
+
+ is_modified = true;
+ }
+
+ strcpy(fname+fname_len, ".hard");
+ fd = open(fname, O_RDONLY);
+ if (fd!=-1) {
+ lim->rlim_max = readValue(fd, fname_base);
+ Eclose(fd);
+
+ is_modified = true;
+ }
+
+ strcpy(fname+fname_len, ".soft");
+ fd = open(fname, O_RDONLY);
+ if (fd!=-1) {
+ lim->rlim_cur = readValue(fd, fname_base);
+ Eclose(fd);
+
+ is_modified = true;
+ }
+
+ if (is_modified &&
+ lim->rlim_max!=RLIM_INFINITY &&
+ (lim->rlim_cur==RLIM_INFINITY ||
+ lim->rlim_cur>lim->rlim_max))
+ lim->rlim_cur = lim->rlim_max;
+
+ return is_modified;
+}
+
+int main(int argc, char *argv[])
+{
+ size_t i;
+ int cur_fd = Eopen(".", O_RDONLY, 0);
+
+ if (argc<3) {
+ WRITE_MSG(2, "Usage: exec-ulimit <ulimit-cfgdir> <cmd> <argv>*\n");
+ exit(255);
+ }
+
+ Echdir(argv[1]);
+
+ for (i=0; i<sizeof(LIMITS)/sizeof(LIMITS[0]); ++i) {
+ struct rlimit limit;
+
+ Egetrlimit(LIMITS[i].code, &limit);
+ if (readSingleLimit(&limit, LIMITS[i].fname))
+ Esetrlimit(LIMITS[i].code, &limit);
+ }
+
+ Efchdir(cur_fd);
+ Eclose(cur_fd);
+
+ Eexecv(argv[2], argv+2);
+}