3 // Copyright (C) 2003 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; version 2 of the License.
9 // This program is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 // GNU General Public License for more details.
14 // You should have received a copy of the GNU General Public License
15 // along with this program; if not, write to the Free Software
16 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
26 #include <sys/resource.h>
31 #define ENSC_WRAPPERS_PREFIX "exec-ulimit: "
32 #define ENSC_WRAPPERS_UNISTD 1
33 #define ENSC_WRAPPERS_FCNTL 1
34 #define ENSC_WRAPPERS_RESOURCE 1
37 #define DECLARE_LIMIT(RES,FNAME) { #FNAME, RLIMIT_##RES }
39 int wrapper_exit_code = 255;
45 DECLARE_LIMIT(CPU, cpu),
46 DECLARE_LIMIT(DATA, data),
47 DECLARE_LIMIT(FSIZE, fsize),
48 DECLARE_LIMIT(LOCKS, locks),
49 DECLARE_LIMIT(MEMLOCK, memlock),
50 DECLARE_LIMIT(NOFILE, nofile),
51 DECLARE_LIMIT(NPROC, nproc),
52 DECLARE_LIMIT(RSS, rss),
53 DECLARE_LIMIT(STACK, stack),
57 readValue(int fd, char const *filename)
60 size_t len = Eread(fd, buf, sizeof(buf)-1);
65 if (strncmp(buf, "inf", 3)==0) return RLIM_INFINITY;
66 res = strtol(buf, &errptr, 0);
70 case 'M' : res *= 1024; /* fallthrough */
71 case 'K' : res *= 1024; ++errptr; break;
72 case 'm' : res *= 1000; /* fallthrough */
73 case 'k' : res *= 1000; ++errptr; break;
78 if (errptr==buf || (*errptr!='\0' && *errptr!='\n')) {
79 WRITE_MSG(2, "Invalid limit in '");
80 WRITE_STR(2, filename);
89 readSingleLimit(struct rlimit *lim, char const *fname_base)
91 size_t fname_len = strlen(fname_base);
92 char fname[fname_len + sizeof(".hard")];
94 bool is_modified = false;
96 strcpy(fname, fname_base);
98 fd = open(fname, O_RDONLY);
100 rlim_t tmp = readValue(fd, fname_base);
108 strcpy(fname+fname_len, ".hard");
109 fd = open(fname, O_RDONLY);
111 lim->rlim_max = readValue(fd, fname_base);
117 strcpy(fname+fname_len, ".soft");
118 fd = open(fname, O_RDONLY);
120 lim->rlim_cur = readValue(fd, fname_base);
127 lim->rlim_max!=RLIM_INFINITY &&
128 (lim->rlim_cur==RLIM_INFINITY ||
129 lim->rlim_cur>lim->rlim_max))
130 lim->rlim_cur = lim->rlim_max;
136 showHelp(int fd, char const *cmd, int res)
138 WRITE_MSG(fd, "Usage: ");
141 "<ulimit-cfgdir> <cmd> <argv>*\n\n"
142 "Please report bugs to " PACKAGE_BUGREPORT "\n");
150 "exec-ulimit " VERSION " -- executes programs with resource limits\n"
151 "This program is part of " PACKAGE_STRING "\n\n"
152 "Copyright (C) 2003 Enrico Scholz\n"
153 VERSION_COPYRIGHT_DISCLAIMER);
157 int main(int argc, char *argv[])
160 int cur_fd = Eopen(".", O_RDONLY, 0);
163 if (strcmp(argv[1], "--help")==0) showHelp(1,argv[0],0);
164 if (strcmp(argv[1], "--version")==0) showVersion();
168 WRITE_MSG(2, "Bad parameter count; use '--help' for more information.\n");
172 if (chdir(argv[1])!=-1) {
173 for (i=0; i<sizeof(LIMITS)/sizeof(LIMITS[0]); ++i) {
176 Egetrlimit(LIMITS[i].code, &limit);
177 if (readSingleLimit(&limit, LIMITS[i].fname))
178 Esetrlimit(LIMITS[i].code, &limit);
184 Eexecv(argv[2], argv+2);