xref: /freebsd/bin/sh/miscbltin.c (revision 58570ac4a5cbda1359a500deff49887bb1ac0551)
14b88c807SRodney W. Grimes /*-
24b88c807SRodney W. Grimes  * Copyright (c) 1991, 1993
34b88c807SRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
44b88c807SRodney W. Grimes  *
54b88c807SRodney W. Grimes  * This code is derived from software contributed to Berkeley by
64b88c807SRodney W. Grimes  * Kenneth Almquist.
74b88c807SRodney W. Grimes  *
84b88c807SRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
94b88c807SRodney W. Grimes  * modification, are permitted provided that the following conditions
104b88c807SRodney W. Grimes  * are met:
114b88c807SRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
124b88c807SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
134b88c807SRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
144b88c807SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
154b88c807SRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
164b88c807SRodney W. Grimes  * 4. Neither the name of the University nor the names of its contributors
174b88c807SRodney W. Grimes  *    may be used to endorse or promote products derived from this software
184b88c807SRodney W. Grimes  *    without specific prior written permission.
194b88c807SRodney W. Grimes  *
204b88c807SRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
214b88c807SRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
224b88c807SRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
234b88c807SRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
244b88c807SRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
254b88c807SRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
264b88c807SRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
274b88c807SRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
284b88c807SRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
294b88c807SRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
304b88c807SRodney W. Grimes  * SUCH DAMAGE.
314b88c807SRodney W. Grimes  */
324b88c807SRodney W. Grimes 
334b88c807SRodney W. Grimes #ifndef lint
343d7b5b93SPhilippe Charnier #if 0
353d7b5b93SPhilippe Charnier static char sccsid[] = "@(#)miscbltin.c	8.4 (Berkeley) 5/4/95";
363d7b5b93SPhilippe Charnier #endif
374b88c807SRodney W. Grimes #endif /* not lint */
382749b141SDavid E. O'Brien #include <sys/cdefs.h>
392749b141SDavid E. O'Brien __FBSDID("$FreeBSD$");
404b88c807SRodney W. Grimes 
414b88c807SRodney W. Grimes /*
4246be34b9SKris Kennaway  * Miscellaneous builtins.
434b88c807SRodney W. Grimes  */
444b88c807SRodney W. Grimes 
45aa9caaf6SPeter Wemm #include <sys/types.h>
46aa9caaf6SPeter Wemm #include <sys/stat.h>
47aa9caaf6SPeter Wemm #include <sys/time.h>
48aa9caaf6SPeter Wemm #include <sys/resource.h>
49aa9caaf6SPeter Wemm #include <unistd.h>
5016992ff4SPeter Wemm #include <errno.h>
510c1661b7SMaxime Henrion #include <stdint.h>
524417f629SPeter Wemm #include <stdio.h>
531f40b47bSMartin Cracauer #include <stdlib.h>
54aa9caaf6SPeter Wemm 
554b88c807SRodney W. Grimes #include "shell.h"
564b88c807SRodney W. Grimes #include "options.h"
574b88c807SRodney W. Grimes #include "var.h"
584b88c807SRodney W. Grimes #include "output.h"
594b88c807SRodney W. Grimes #include "memalloc.h"
604b88c807SRodney W. Grimes #include "error.h"
614b88c807SRodney W. Grimes #include "mystring.h"
62*58570ac4SJilles Tjoelker #include "syntax.h"
634b88c807SRodney W. Grimes 
644b88c807SRodney W. Grimes #undef eflag
654b88c807SRodney W. Grimes 
6679ea0bd9SStefan Farfeleder int readcmd(int, char **);
6779ea0bd9SStefan Farfeleder int umaskcmd(int, char **);
6879ea0bd9SStefan Farfeleder int ulimitcmd(int, char **);
6979ea0bd9SStefan Farfeleder 
704b88c807SRodney W. Grimes /*
718f0561ccSThomas Gellekum  * The read builtin.  The -r option causes backslashes to be treated like
728f0561ccSThomas Gellekum  * ordinary characters.
734b88c807SRodney W. Grimes  *
744b88c807SRodney W. Grimes  * This uses unbuffered input, which may be avoidable in some cases.
75b6748ec2SStefan Farfeleder  *
76b6748ec2SStefan Farfeleder  * Note that if IFS=' :' then read x y should work so that:
77b6748ec2SStefan Farfeleder  * 'a b'	x='a', y='b'
78b6748ec2SStefan Farfeleder  * ' a b '	x='a', y='b'
79b6748ec2SStefan Farfeleder  * ':b'		x='',  y='b'
80b6748ec2SStefan Farfeleder  * ':'		x='',  y=''
81b6748ec2SStefan Farfeleder  * '::'		x='',  y=''
82b6748ec2SStefan Farfeleder  * ': :'	x='',  y=''
83b6748ec2SStefan Farfeleder  * ':::'	x='',  y='::'
84b6748ec2SStefan Farfeleder  * ':b c:'	x='',  y='b c:'
854b88c807SRodney W. Grimes  */
864b88c807SRodney W. Grimes 
87aa9caaf6SPeter Wemm int
885134c3f7SWarner Losh readcmd(int argc __unused, char **argv __unused)
89aa9caaf6SPeter Wemm {
904b88c807SRodney W. Grimes 	char **ap;
914b88c807SRodney W. Grimes 	int backslash;
924b88c807SRodney W. Grimes 	char c;
938f0561ccSThomas Gellekum 	int rflag;
944b88c807SRodney W. Grimes 	char *prompt;
95384aedabSJilles Tjoelker 	const char *ifs;
964b88c807SRodney W. Grimes 	char *p;
974b88c807SRodney W. Grimes 	int startword;
984b88c807SRodney W. Grimes 	int status;
994b88c807SRodney W. Grimes 	int i;
100b6748ec2SStefan Farfeleder 	int is_ifs;
101b6748ec2SStefan Farfeleder 	int saveall = 0;
102afa53c8dSMike Smith 	struct timeval tv;
103afa53c8dSMike Smith 	char *tvptr;
104afa53c8dSMike Smith 	fd_set ifds;
1054b88c807SRodney W. Grimes 
1068f0561ccSThomas Gellekum 	rflag = 0;
1074b88c807SRodney W. Grimes 	prompt = NULL;
108afa53c8dSMike Smith 	tv.tv_sec = -1;
109afa53c8dSMike Smith 	tv.tv_usec = 0;
1108f0561ccSThomas Gellekum 	while ((i = nextopt("erp:t:")) != '\0') {
111afa53c8dSMike Smith 		switch(i) {
112afa53c8dSMike Smith 		case 'p':
113f01e3d0cSMartin Cracauer 			prompt = shoptarg;
114afa53c8dSMike Smith 			break;
115afa53c8dSMike Smith 		case 'e':
1168f0561ccSThomas Gellekum 			break;
1178f0561ccSThomas Gellekum 		case 'r':
1188f0561ccSThomas Gellekum 			rflag = 1;
119afa53c8dSMike Smith 			break;
120afa53c8dSMike Smith 		case 't':
121f01e3d0cSMartin Cracauer 			tv.tv_sec = strtol(shoptarg, &tvptr, 0);
122f01e3d0cSMartin Cracauer 			if (tvptr == shoptarg)
123afa53c8dSMike Smith 				error("timeout value");
124afa53c8dSMike Smith 			switch(*tvptr) {
125afa53c8dSMike Smith 			case 0:
126afa53c8dSMike Smith 			case 's':
127afa53c8dSMike Smith 				break;
128afa53c8dSMike Smith 			case 'h':
129afa53c8dSMike Smith 				tv.tv_sec *= 60;
130afa53c8dSMike Smith 				/* FALLTHROUGH */
131afa53c8dSMike Smith 			case 'm':
132afa53c8dSMike Smith 				tv.tv_sec *= 60;
133afa53c8dSMike Smith 				break;
134afa53c8dSMike Smith 			default:
135afa53c8dSMike Smith 				error("timeout unit");
136afa53c8dSMike Smith 			}
137afa53c8dSMike Smith 			break;
138afa53c8dSMike Smith 		}
1394b88c807SRodney W. Grimes 	}
1404b88c807SRodney W. Grimes 	if (prompt && isatty(0)) {
1414b88c807SRodney W. Grimes 		out2str(prompt);
1424b88c807SRodney W. Grimes 		flushall();
1434b88c807SRodney W. Grimes 	}
1444b88c807SRodney W. Grimes 	if (*(ap = argptr) == NULL)
1454b88c807SRodney W. Grimes 		error("arg count");
1464b88c807SRodney W. Grimes 	if ((ifs = bltinlookup("IFS", 1)) == NULL)
14786d8da5dSStefan Farfeleder 		ifs = " \t\n";
148afa53c8dSMike Smith 
149afa53c8dSMike Smith 	if (tv.tv_sec >= 0) {
150afa53c8dSMike Smith 		/*
151afa53c8dSMike Smith 		 * Wait for something to become available.
152afa53c8dSMike Smith 		 */
153afa53c8dSMike Smith 		FD_ZERO(&ifds);
154afa53c8dSMike Smith 		FD_SET(0, &ifds);
155afa53c8dSMike Smith 		status = select(1, &ifds, NULL, NULL, &tv);
156afa53c8dSMike Smith 		/*
157afa53c8dSMike Smith 		 * If there's nothing ready, return an error.
158afa53c8dSMike Smith 		 */
159afa53c8dSMike Smith 		if (status <= 0)
160afa53c8dSMike Smith 			return(1);
161afa53c8dSMike Smith 	}
162afa53c8dSMike Smith 
1634b88c807SRodney W. Grimes 	status = 0;
164b6748ec2SStefan Farfeleder 	startword = 2;
1654b88c807SRodney W. Grimes 	backslash = 0;
1664b88c807SRodney W. Grimes 	STARTSTACKSTR(p);
1674b88c807SRodney W. Grimes 	for (;;) {
168e1b4d8d0SSheldon Hearn 		if (read(STDIN_FILENO, &c, 1) != 1) {
1694b88c807SRodney W. Grimes 			status = 1;
1704b88c807SRodney W. Grimes 			break;
1714b88c807SRodney W. Grimes 		}
1724b88c807SRodney W. Grimes 		if (c == '\0')
1734b88c807SRodney W. Grimes 			continue;
1749d37e157SJilles Tjoelker 		CHECKSTRSPACE(1, p);
1754b88c807SRodney W. Grimes 		if (backslash) {
1764b88c807SRodney W. Grimes 			backslash = 0;
177917fdfb1SJilles Tjoelker 			startword = 0;
1784b88c807SRodney W. Grimes 			if (c != '\n')
1799d37e157SJilles Tjoelker 				USTPUTC(c, p);
1804b88c807SRodney W. Grimes 			continue;
1814b88c807SRodney W. Grimes 		}
1828f0561ccSThomas Gellekum 		if (!rflag && c == '\\') {
1834b88c807SRodney W. Grimes 			backslash++;
1844b88c807SRodney W. Grimes 			continue;
1854b88c807SRodney W. Grimes 		}
1864b88c807SRodney W. Grimes 		if (c == '\n')
1874b88c807SRodney W. Grimes 			break;
188b6748ec2SStefan Farfeleder 		if (strchr(ifs, c))
189b6748ec2SStefan Farfeleder 			is_ifs = strchr(" \t\n", c) ? 1 : 2;
190b6748ec2SStefan Farfeleder 		else
191b6748ec2SStefan Farfeleder 			is_ifs = 0;
192b6748ec2SStefan Farfeleder 
193b6748ec2SStefan Farfeleder 		if (startword != 0) {
194b6748ec2SStefan Farfeleder 			if (is_ifs == 1) {
195b6748ec2SStefan Farfeleder 				/* Ignore leading IFS whitespace */
196b6748ec2SStefan Farfeleder 				if (saveall)
1979d37e157SJilles Tjoelker 					USTPUTC(c, p);
1984b88c807SRodney W. Grimes 				continue;
1994b88c807SRodney W. Grimes 			}
200b6748ec2SStefan Farfeleder 			if (is_ifs == 2 && startword == 1) {
201b6748ec2SStefan Farfeleder 				/* Only one non-whitespace IFS per word */
202b6748ec2SStefan Farfeleder 				startword = 2;
203b6748ec2SStefan Farfeleder 				if (saveall)
2049d37e157SJilles Tjoelker 					USTPUTC(c, p);
205b6748ec2SStefan Farfeleder 				continue;
206b6748ec2SStefan Farfeleder 			}
207b6748ec2SStefan Farfeleder 		}
208b6748ec2SStefan Farfeleder 
209b6748ec2SStefan Farfeleder 		if (is_ifs == 0) {
210b6748ec2SStefan Farfeleder 			/* append this character to the current variable */
2114b88c807SRodney W. Grimes 			startword = 0;
212b6748ec2SStefan Farfeleder 			if (saveall)
213b6748ec2SStefan Farfeleder 				/* Not just a spare terminator */
214b6748ec2SStefan Farfeleder 				saveall++;
2159d37e157SJilles Tjoelker 			USTPUTC(c, p);
216b6748ec2SStefan Farfeleder 			continue;
217b6748ec2SStefan Farfeleder 		}
218b6748ec2SStefan Farfeleder 
219b6748ec2SStefan Farfeleder 		/* end of variable... */
220b6748ec2SStefan Farfeleder 		startword = is_ifs;
221b6748ec2SStefan Farfeleder 
222b6748ec2SStefan Farfeleder 		if (ap[1] == NULL) {
223b6748ec2SStefan Farfeleder 			/* Last variable needs all IFS chars */
224b6748ec2SStefan Farfeleder 			saveall++;
2259d37e157SJilles Tjoelker 			USTPUTC(c, p);
226b6748ec2SStefan Farfeleder 			continue;
227b6748ec2SStefan Farfeleder 		}
228b6748ec2SStefan Farfeleder 
2294b88c807SRodney W. Grimes 		STACKSTRNUL(p);
2304b88c807SRodney W. Grimes 		setvar(*ap, stackblock(), 0);
2314b88c807SRodney W. Grimes 		ap++;
2324b88c807SRodney W. Grimes 		STARTSTACKSTR(p);
2334b88c807SRodney W. Grimes 	}
2344b88c807SRodney W. Grimes 	STACKSTRNUL(p);
235b6748ec2SStefan Farfeleder 
236b6748ec2SStefan Farfeleder 	/* Remove trailing IFS chars */
237b6748ec2SStefan Farfeleder 	for (; stackblock() <= --p; *p = 0) {
238b6748ec2SStefan Farfeleder 		if (!strchr(ifs, *p))
239b6748ec2SStefan Farfeleder 			break;
240b6748ec2SStefan Farfeleder 		if (strchr(" \t\n", *p))
241b6748ec2SStefan Farfeleder 			/* Always remove whitespace */
242b6748ec2SStefan Farfeleder 			continue;
243b6748ec2SStefan Farfeleder 		if (saveall > 1)
244b6748ec2SStefan Farfeleder 			/* Don't remove non-whitespace unless it was naked */
245b6748ec2SStefan Farfeleder 			break;
246b6748ec2SStefan Farfeleder 	}
2474b88c807SRodney W. Grimes 	setvar(*ap, stackblock(), 0);
248b6748ec2SStefan Farfeleder 
249b6748ec2SStefan Farfeleder 	/* Set any remaining args to "" */
2504b88c807SRodney W. Grimes 	while (*++ap != NULL)
2514b88c807SRodney W. Grimes 		setvar(*ap, nullstr, 0);
2524b88c807SRodney W. Grimes 	return status;
2534b88c807SRodney W. Grimes }
2544b88c807SRodney W. Grimes 
2554b88c807SRodney W. Grimes 
2564b88c807SRodney W. Grimes 
257aa9caaf6SPeter Wemm int
258384aedabSJilles Tjoelker umaskcmd(int argc __unused, char **argv __unused)
259aa9caaf6SPeter Wemm {
260aa9caaf6SPeter Wemm 	char *ap;
2614b88c807SRodney W. Grimes 	int mask;
2624b88c807SRodney W. Grimes 	int i;
263aa9caaf6SPeter Wemm 	int symbolic_mode = 0;
2644b88c807SRodney W. Grimes 
265aa9caaf6SPeter Wemm 	while ((i = nextopt("S")) != '\0') {
266aa9caaf6SPeter Wemm 		symbolic_mode = 1;
267aa9caaf6SPeter Wemm 	}
268aa9caaf6SPeter Wemm 
2694b88c807SRodney W. Grimes 	INTOFF;
2704b88c807SRodney W. Grimes 	mask = umask(0);
2714b88c807SRodney W. Grimes 	umask(mask);
2724b88c807SRodney W. Grimes 	INTON;
273aa9caaf6SPeter Wemm 
274aa9caaf6SPeter Wemm 	if ((ap = *argptr) == NULL) {
275aa9caaf6SPeter Wemm 		if (symbolic_mode) {
276aa9caaf6SPeter Wemm 			char u[4], g[4], o[4];
277aa9caaf6SPeter Wemm 
278aa9caaf6SPeter Wemm 			i = 0;
279aa9caaf6SPeter Wemm 			if ((mask & S_IRUSR) == 0)
280aa9caaf6SPeter Wemm 				u[i++] = 'r';
281aa9caaf6SPeter Wemm 			if ((mask & S_IWUSR) == 0)
282aa9caaf6SPeter Wemm 				u[i++] = 'w';
283aa9caaf6SPeter Wemm 			if ((mask & S_IXUSR) == 0)
284aa9caaf6SPeter Wemm 				u[i++] = 'x';
285aa9caaf6SPeter Wemm 			u[i] = '\0';
286aa9caaf6SPeter Wemm 
287aa9caaf6SPeter Wemm 			i = 0;
288aa9caaf6SPeter Wemm 			if ((mask & S_IRGRP) == 0)
289aa9caaf6SPeter Wemm 				g[i++] = 'r';
290aa9caaf6SPeter Wemm 			if ((mask & S_IWGRP) == 0)
291aa9caaf6SPeter Wemm 				g[i++] = 'w';
292aa9caaf6SPeter Wemm 			if ((mask & S_IXGRP) == 0)
293aa9caaf6SPeter Wemm 				g[i++] = 'x';
294aa9caaf6SPeter Wemm 			g[i] = '\0';
295aa9caaf6SPeter Wemm 
296aa9caaf6SPeter Wemm 			i = 0;
297aa9caaf6SPeter Wemm 			if ((mask & S_IROTH) == 0)
298aa9caaf6SPeter Wemm 				o[i++] = 'r';
299aa9caaf6SPeter Wemm 			if ((mask & S_IWOTH) == 0)
300aa9caaf6SPeter Wemm 				o[i++] = 'w';
301aa9caaf6SPeter Wemm 			if ((mask & S_IXOTH) == 0)
302aa9caaf6SPeter Wemm 				o[i++] = 'x';
303aa9caaf6SPeter Wemm 			o[i] = '\0';
304aa9caaf6SPeter Wemm 
305aa9caaf6SPeter Wemm 			out1fmt("u=%s,g=%s,o=%s\n", u, g, o);
3064b88c807SRodney W. Grimes 		} else {
307aa9caaf6SPeter Wemm 			out1fmt("%.4o\n", mask);
308aa9caaf6SPeter Wemm 		}
309aa9caaf6SPeter Wemm 	} else {
310*58570ac4SJilles Tjoelker 		if (is_digit(*ap)) {
3114b88c807SRodney W. Grimes 			mask = 0;
3124b88c807SRodney W. Grimes 			do {
313aa9caaf6SPeter Wemm 				if (*ap >= '8' || *ap < '0')
314e4c880afSStefan Farfeleder 					error("Illegal number: %s", *argptr);
315aa9caaf6SPeter Wemm 				mask = (mask << 3) + (*ap - '0');
316aa9caaf6SPeter Wemm 			} while (*++ap != '\0');
3174b88c807SRodney W. Grimes 			umask(mask);
318aa9caaf6SPeter Wemm 		} else {
319aa9caaf6SPeter Wemm 			void *set;
320670528cdSStefan Farfeleder 			INTOFF;
321aa9caaf6SPeter Wemm 			if ((set = setmode (ap)) == 0)
322aa9caaf6SPeter Wemm 				error("Illegal number: %s", ap);
323aa9caaf6SPeter Wemm 
324aa9caaf6SPeter Wemm 			mask = getmode (set, ~mask & 0777);
325aa9caaf6SPeter Wemm 			umask(~mask & 0777);
326a207266eSWarner Losh 			free(set);
327670528cdSStefan Farfeleder 			INTON;
328aa9caaf6SPeter Wemm 		}
3294b88c807SRodney W. Grimes 	}
3304b88c807SRodney W. Grimes 	return 0;
3314b88c807SRodney W. Grimes }
3327a2afe64SJoerg Wunsch 
333aa9caaf6SPeter Wemm /*
334aa9caaf6SPeter Wemm  * ulimit builtin
335aa9caaf6SPeter Wemm  *
336aa9caaf6SPeter Wemm  * This code, originally by Doug Gwyn, Doug Kingston, Eric Gisin, and
337aa9caaf6SPeter Wemm  * Michael Rendell was ripped from pdksh 5.0.8 and hacked for use with
338aa9caaf6SPeter Wemm  * ash by J.T. Conklin.
339aa9caaf6SPeter Wemm  *
340aa9caaf6SPeter Wemm  * Public domain.
341aa9caaf6SPeter Wemm  */
3427a2afe64SJoerg Wunsch 
343aa9caaf6SPeter Wemm struct limits {
344aa9caaf6SPeter Wemm 	const char *name;
34516992ff4SPeter Wemm 	const char *units;
346aa9caaf6SPeter Wemm 	int	cmd;
347aa9caaf6SPeter Wemm 	int	factor;	/* multiply by to get rlim_{cur,max} values */
348aa9caaf6SPeter Wemm 	char	option;
3497a2afe64SJoerg Wunsch };
3507a2afe64SJoerg Wunsch 
351aa9caaf6SPeter Wemm static const struct limits limits[] = {
352aa9caaf6SPeter Wemm #ifdef RLIMIT_CPU
35316992ff4SPeter Wemm 	{ "cpu time",		"seconds",	RLIMIT_CPU,	   1, 't' },
354aa9caaf6SPeter Wemm #endif
355aa9caaf6SPeter Wemm #ifdef RLIMIT_FSIZE
35616992ff4SPeter Wemm 	{ "file size",		"512-blocks",	RLIMIT_FSIZE,	 512, 'f' },
357aa9caaf6SPeter Wemm #endif
358aa9caaf6SPeter Wemm #ifdef RLIMIT_DATA
35916992ff4SPeter Wemm 	{ "data seg size",	"kbytes",	RLIMIT_DATA,	1024, 'd' },
360aa9caaf6SPeter Wemm #endif
361aa9caaf6SPeter Wemm #ifdef RLIMIT_STACK
36216992ff4SPeter Wemm 	{ "stack size",		"kbytes",	RLIMIT_STACK,	1024, 's' },
363aa9caaf6SPeter Wemm #endif
364aa9caaf6SPeter Wemm #ifdef  RLIMIT_CORE
36516992ff4SPeter Wemm 	{ "core file size",	"512-blocks",	RLIMIT_CORE,	 512, 'c' },
366aa9caaf6SPeter Wemm #endif
367aa9caaf6SPeter Wemm #ifdef RLIMIT_RSS
36816992ff4SPeter Wemm 	{ "max memory size",	"kbytes",	RLIMIT_RSS,	1024, 'm' },
369aa9caaf6SPeter Wemm #endif
370aa9caaf6SPeter Wemm #ifdef RLIMIT_MEMLOCK
37116992ff4SPeter Wemm 	{ "locked memory",	"kbytes",	RLIMIT_MEMLOCK, 1024, 'l' },
372aa9caaf6SPeter Wemm #endif
373aa9caaf6SPeter Wemm #ifdef RLIMIT_NPROC
37416992ff4SPeter Wemm 	{ "max user processes",	(char *)0,	RLIMIT_NPROC,      1, 'u' },
375aa9caaf6SPeter Wemm #endif
376aa9caaf6SPeter Wemm #ifdef RLIMIT_NOFILE
37716992ff4SPeter Wemm 	{ "open files",		(char *)0,	RLIMIT_NOFILE,     1, 'n' },
378aa9caaf6SPeter Wemm #endif
379aa9caaf6SPeter Wemm #ifdef RLIMIT_VMEM
38016992ff4SPeter Wemm 	{ "virtual mem size",	"kbytes",	RLIMIT_VMEM,	1024, 'v' },
381aa9caaf6SPeter Wemm #endif
382aa9caaf6SPeter Wemm #ifdef RLIMIT_SWAP
38316992ff4SPeter Wemm 	{ "swap limit",		"kbytes",	RLIMIT_SWAP,	1024, 'w' },
384aa9caaf6SPeter Wemm #endif
3857cde774dSBrian Feldman #ifdef RLIMIT_SBSIZE
3867cde774dSBrian Feldman 	{ "sbsize",		"bytes",	RLIMIT_SBSIZE,	   1, 'b' },
3877cde774dSBrian Feldman #endif
388bc093719SEd Schouten #ifdef RLIMIT_NPTS
389bc093719SEd Schouten 	{ "pseudo-terminals",	(char *)0,	RLIMIT_NPTS,	   1, 'p' },
390bc093719SEd Schouten #endif
39116992ff4SPeter Wemm 	{ (char *) 0,		(char *)0,	0,		   0, '\0' }
3927a2afe64SJoerg Wunsch };
3937a2afe64SJoerg Wunsch 
394aa9caaf6SPeter Wemm int
3955134c3f7SWarner Losh ulimitcmd(int argc __unused, char **argv __unused)
396aa9caaf6SPeter Wemm {
397afb033d5SSteve Price 	int	c;
3980c1661b7SMaxime Henrion 	rlim_t val = 0;
399aa9caaf6SPeter Wemm 	enum { SOFT = 0x1, HARD = 0x2 }
400aa9caaf6SPeter Wemm 			how = SOFT | HARD;
401aa9caaf6SPeter Wemm 	const struct limits	*l;
402aa9caaf6SPeter Wemm 	int		set, all = 0;
403aa9caaf6SPeter Wemm 	int		optc, what;
404aa9caaf6SPeter Wemm 	struct rlimit	limit;
4057a2afe64SJoerg Wunsch 
406aa9caaf6SPeter Wemm 	what = 'f';
407c9253e93SKonstantin Belousov 	while ((optc = nextopt("HSatfdsmcnuvlbpw")) != '\0')
408aa9caaf6SPeter Wemm 		switch (optc) {
4097a2afe64SJoerg Wunsch 		case 'H':
410aa9caaf6SPeter Wemm 			how = HARD;
4117a2afe64SJoerg Wunsch 			break;
4127a2afe64SJoerg Wunsch 		case 'S':
413aa9caaf6SPeter Wemm 			how = SOFT;
4147a2afe64SJoerg Wunsch 			break;
4157a2afe64SJoerg Wunsch 		case 'a':
416aa9caaf6SPeter Wemm 			all = 1;
4177a2afe64SJoerg Wunsch 			break;
418aa9caaf6SPeter Wemm 		default:
419aa9caaf6SPeter Wemm 			what = optc;
4207a2afe64SJoerg Wunsch 		}
4217a2afe64SJoerg Wunsch 
422aa9caaf6SPeter Wemm 	for (l = limits; l->name && l->option != what; l++)
423aa9caaf6SPeter Wemm 		;
424aa9caaf6SPeter Wemm 	if (!l->name)
425abe2dc61STim J. Robbins 		error("internal error (%c)", what);
4267a2afe64SJoerg Wunsch 
427aa9caaf6SPeter Wemm 	set = *argptr ? 1 : 0;
428aa9caaf6SPeter Wemm 	if (set) {
429aa9caaf6SPeter Wemm 		char *p = *argptr;
430aa9caaf6SPeter Wemm 
431aa9caaf6SPeter Wemm 		if (all || argptr[1])
432abe2dc61STim J. Robbins 			error("too many arguments");
433aa9caaf6SPeter Wemm 		if (strcmp(p, "unlimited") == 0)
4347a2afe64SJoerg Wunsch 			val = RLIM_INFINITY;
4357a2afe64SJoerg Wunsch 		else {
4360c1661b7SMaxime Henrion 			val = 0;
437aa9caaf6SPeter Wemm 
438aa9caaf6SPeter Wemm 			while ((c = *p++) >= '0' && c <= '9')
439aa9caaf6SPeter Wemm 			{
440aa9caaf6SPeter Wemm 				val = (val * 10) + (long)(c - '0');
4410c1661b7SMaxime Henrion 				if (val < 0)
442aa9caaf6SPeter Wemm 					break;
4437a2afe64SJoerg Wunsch 			}
444aa9caaf6SPeter Wemm 			if (c)
445abe2dc61STim J. Robbins 				error("bad number");
446aa9caaf6SPeter Wemm 			val *= l->factor;
447aa9caaf6SPeter Wemm 		}
448aa9caaf6SPeter Wemm 	}
449aa9caaf6SPeter Wemm 	if (all) {
450aa9caaf6SPeter Wemm 		for (l = limits; l->name; l++) {
45116992ff4SPeter Wemm 			char optbuf[40];
45216992ff4SPeter Wemm 			if (getrlimit(l->cmd, &limit) < 0)
453abe2dc61STim J. Robbins 				error("can't get limit: %s", strerror(errno));
454aa9caaf6SPeter Wemm 			if (how & SOFT)
455aa9caaf6SPeter Wemm 				val = limit.rlim_cur;
456aa9caaf6SPeter Wemm 			else if (how & HARD)
457aa9caaf6SPeter Wemm 				val = limit.rlim_max;
458aa9caaf6SPeter Wemm 
45916992ff4SPeter Wemm 			if (l->units)
46016992ff4SPeter Wemm 				snprintf(optbuf, sizeof(optbuf),
4614e4e0959SPeter Wemm 					"(%s, -%c) ", l->units, l->option);
46216992ff4SPeter Wemm 			else
46316992ff4SPeter Wemm 				snprintf(optbuf, sizeof(optbuf),
4644e4e0959SPeter Wemm 					"(-%c) ", l->option);
4654e4e0959SPeter Wemm 			out1fmt("%-18s %18s ", l->name, optbuf);
466aa9caaf6SPeter Wemm 			if (val == RLIM_INFINITY)
467258ef734SJilles Tjoelker 				out1str("unlimited\n");
468aa9caaf6SPeter Wemm 			else
469aa9caaf6SPeter Wemm 			{
470aa9caaf6SPeter Wemm 				val /= l->factor;
4710c1661b7SMaxime Henrion 				out1fmt("%jd\n", (intmax_t)val);
4727a2afe64SJoerg Wunsch 			}
4737a2afe64SJoerg Wunsch 		}
4747a2afe64SJoerg Wunsch 		return 0;
4757a2afe64SJoerg Wunsch 	}
476aa9caaf6SPeter Wemm 
47716992ff4SPeter Wemm 	if (getrlimit(l->cmd, &limit) < 0)
478abe2dc61STim J. Robbins 		error("can't get limit: %s", strerror(errno));
479aa9caaf6SPeter Wemm 	if (set) {
480aa9caaf6SPeter Wemm 		if (how & SOFT)
481aa9caaf6SPeter Wemm 			limit.rlim_cur = val;
482aa9caaf6SPeter Wemm 		if (how & HARD)
483aa9caaf6SPeter Wemm 			limit.rlim_max = val;
484aa9caaf6SPeter Wemm 		if (setrlimit(l->cmd, &limit) < 0)
485abe2dc61STim J. Robbins 			error("bad limit: %s", strerror(errno));
486aa9caaf6SPeter Wemm 	} else {
487aa9caaf6SPeter Wemm 		if (how & SOFT)
488aa9caaf6SPeter Wemm 			val = limit.rlim_cur;
489aa9caaf6SPeter Wemm 		else if (how & HARD)
490aa9caaf6SPeter Wemm 			val = limit.rlim_max;
491aa9caaf6SPeter Wemm 
492aa9caaf6SPeter Wemm 		if (val == RLIM_INFINITY)
493258ef734SJilles Tjoelker 			out1str("unlimited\n");
494aa9caaf6SPeter Wemm 		else
495aa9caaf6SPeter Wemm 		{
496aa9caaf6SPeter Wemm 			val /= l->factor;
4970c1661b7SMaxime Henrion 			out1fmt("%jd\n", (intmax_t)val);
498aa9caaf6SPeter Wemm 		}
499aa9caaf6SPeter Wemm 	}
500aa9caaf6SPeter Wemm 	return 0;
501aa9caaf6SPeter Wemm }
502