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>
30 #define ENSC_WRAPPERS_UNISTD 1
31 #define ENSC_WRAPPERS_FCNTL 1
32 #define ENSC_WRAPPERS_RESOURCE 1
35 #define DECLARE_LIMIT(RES,FNAME) { #FNAME, RLIMIT_##RES }
37 int wrapper_exit_code = 255;
43 DECLARE_LIMIT(CPU, cpu),
44 DECLARE_LIMIT(DATA, data),
45 DECLARE_LIMIT(FSIZE, fsize),
46 DECLARE_LIMIT(LOCKS, locks),
47 DECLARE_LIMIT(MEMLOCK, memlock),
48 DECLARE_LIMIT(NOFILE, nofile),
49 DECLARE_LIMIT(NPROC, nproc),
50 DECLARE_LIMIT(RSS, rss),
51 DECLARE_LIMIT(STACK, stack),
55 readValue(int fd, char const *filename)
58 size_t len = Eread(fd, buf, sizeof(buf)-1);
63 if (strncmp(buf, "inf", 3)==0) return RLIM_INFINITY;
64 res = strtol(buf, &errptr, 0);
68 case 'M' : res *= 1024; /* fallthrough */
69 case 'K' : res *= 1024; ++errptr; break;
70 case 'm' : res *= 1000; /* fallthrough */
71 case 'k' : res *= 1000; ++errptr; break;
76 if (errptr==buf || (*errptr!='\0' && *errptr!='\n')) {
77 WRITE_MSG(2, "Invalid limit in '");
78 WRITE_STR(2, filename);
87 readSingleLimit(struct rlimit *lim, char const *fname_base)
89 size_t fname_len = strlen(fname_base);
90 char fname[fname_len + sizeof(".hard")];
92 bool is_modified = false;
94 strcpy(fname, fname_base);
96 fd = open(fname, O_RDONLY);
98 rlim_t tmp = readValue(fd, fname_base);
106 strcpy(fname+fname_len, ".hard");
107 fd = open(fname, O_RDONLY);
109 lim->rlim_max = readValue(fd, fname_base);
115 strcpy(fname+fname_len, ".soft");
116 fd = open(fname, O_RDONLY);
118 lim->rlim_cur = readValue(fd, fname_base);
125 lim->rlim_max!=RLIM_INFINITY &&
126 (lim->rlim_cur==RLIM_INFINITY ||
127 lim->rlim_cur>lim->rlim_max))
128 lim->rlim_cur = lim->rlim_max;
134 showHelp(int fd, char const *cmd, int res)
136 WRITE_MSG(fd, "Usage: ");
139 "<ulimit-cfgdir> <cmd> <argv>*\n\n"
140 "Please report bugs to " PACKAGE_BUGREPORT "\n");
148 "exec-ulimit " VERSION " -- executes programs with resource limits\n"
149 "This program is part of " PACKAGE_STRING "\n\n"
150 "Copyright (C) 2003 Enrico Scholz\n"
151 VERSION_COPYRIGHT_DISCLAIMER);
155 int main(int argc, char *argv[])
158 int cur_fd = Eopen(".", O_RDONLY, 0);
161 if (strcmp(argv[1], "--help")==0) showHelp(1,argv[0],0);
162 if (strcmp(argv[1], "--version")==0) showVersion();
166 WRITE_MSG(2, "Bad parameter count; use '--help' for more information.\n");
170 if (chdir(argv[1])!=-1) {
171 for (i=0; i<sizeof(LIMITS)/sizeof(LIMITS[0]); ++i) {
174 Egetrlimit(LIMITS[i].code, &limit);
175 if (readSingleLimit(&limit, LIMITS[i].fname))
176 Esetrlimit(LIMITS[i].code, &limit);
182 Eexecv(argv[2], argv+2);