From 683d0a5fb8d3241f17a0657ca38c292d2d169b47 Mon Sep 17 00:00:00 2001 From: Enrico Scholz Date: Mon, 20 Oct 2003 22:30:09 +0000 Subject: [PATCH] initial checkin git-svn-id: http://svn.linux-vserver.org/svn/util-vserver/trunk@245 94cd875c-1c1d-0410-91d2-eb244daf1a30 --- util-vserver/src/exec-ulimit.c | 142 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 142 insertions(+) create mode 100644 util-vserver/src/exec-ulimit.c diff --git a/util-vserver/src/exec-ulimit.c b/util-vserver/src/exec-ulimit.c new file mode 100644 index 0000000..f344f0f --- /dev/null +++ b/util-vserver/src/exec-ulimit.c @@ -0,0 +1,142 @@ +// $Id$ --*- c -*-- + +// Copyright (C) 2003 Enrico Scholz +// +// 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 +#endif + +#include "wrappers.h" +#include "util.h" + +#include +#include +#include + +#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 *\n"); + exit(255); + } + + Echdir(argv[1]); + + for (i=0; i