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(CORE, core),
46 DECLARE_LIMIT(CPU, cpu),
47 DECLARE_LIMIT(DATA, data),
48 DECLARE_LIMIT(FSIZE, fsize),
49 DECLARE_LIMIT(LOCKS, locks),
50 DECLARE_LIMIT(MEMLOCK, memlock),
51 DECLARE_LIMIT(NOFILE, nofile),
52 DECLARE_LIMIT(NPROC, nproc),
53 DECLARE_LIMIT(RSS, rss),
54 DECLARE_LIMIT(STACK, stack),
58 readValue(int fd, char const *filename)
61 size_t len = Eread(fd, buf, sizeof(buf)-1);
66 if (strncmp(buf, "inf", 3)==0) return RLIM_INFINITY;
67 res = strtol(buf, &errptr, 0);
71 case 'M' : res *= 1024; /* fallthrough */
72 case 'K' : res *= 1024; ++errptr; break;
73 case 'm' : res *= 1000; /* fallthrough */
74 case 'k' : res *= 1000; ++errptr; break;
79 if (errptr==buf || (*errptr!='\0' && *errptr!='\n')) {
80 WRITE_MSG(2, "Invalid limit in '");
81 WRITE_STR(2, filename);
90 readSingleLimit(struct rlimit *lim, char const *fname_base)
92 size_t fname_len = strlen(fname_base);
93 char fname[fname_len + sizeof(".hard")];
95 bool is_modified = false;
97 strcpy(fname, fname_base);
99 fd = open(fname, O_RDONLY);
101 rlim_t tmp = readValue(fd, fname_base);
109 strcpy(fname+fname_len, ".hard");
110 fd = open(fname, O_RDONLY);
112 lim->rlim_max = readValue(fd, fname_base);
118 strcpy(fname+fname_len, ".soft");
119 fd = open(fname, O_RDONLY);
121 lim->rlim_cur = readValue(fd, fname_base);
128 lim->rlim_max!=RLIM_INFINITY &&
129 (lim->rlim_cur==RLIM_INFINITY ||
130 lim->rlim_cur>lim->rlim_max))
131 lim->rlim_cur = lim->rlim_max;
137 showHelp(int fd, char const *cmd, int res)
139 WRITE_MSG(fd, "Usage: ");
142 "<ulimit-cfgdir> <cmd> <argv>*\n\n"
143 "Please report bugs to " PACKAGE_BUGREPORT "\n");
151 "exec-ulimit " VERSION " -- executes programs with resource limits\n"
152 "This program is part of " PACKAGE_STRING "\n\n"
153 "Copyright (C) 2003 Enrico Scholz\n"
154 VERSION_COPYRIGHT_DISCLAIMER);
158 int main(int argc, char *argv[])
161 int cur_fd = Eopen(".", O_RDONLY, 0);
164 if (strcmp(argv[1], "--help")==0) showHelp(1,argv[0],0);
165 if (strcmp(argv[1], "--version")==0) showVersion();
169 WRITE_MSG(2, "Bad parameter count; use '--help' for more information.\n");
173 if (chdir(argv[1])!=-1) {
174 for (i=0; i<sizeof(LIMITS)/sizeof(LIMITS[0]); ++i) {
177 Egetrlimit(LIMITS[i].code, &limit);
178 if (readSingleLimit(&limit, LIMITS[i].fname))
179 Esetrlimit(LIMITS[i].code, &limit);
185 Eexecv(argv[2], argv+2);