xref: /freebsd/usr.bin/apply/apply.c (revision fbbd9655e5107c68e4e0146ff22b73d7350475bc)
19b50d902SRodney W. Grimes /*-
29b50d902SRodney W. Grimes  * Copyright (c) 1994
39b50d902SRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
49b50d902SRodney W. Grimes  *
59b50d902SRodney W. Grimes  * This code is derived from software contributed to Berkeley by
69b50d902SRodney W. Grimes  * Jan-Simon Pendry.
79b50d902SRodney W. Grimes  *
89b50d902SRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
99b50d902SRodney W. Grimes  * modification, are permitted provided that the following conditions
109b50d902SRodney W. Grimes  * are met:
119b50d902SRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
129b50d902SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
139b50d902SRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
149b50d902SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
159b50d902SRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
16*fbbd9655SWarner Losh  * 3. Neither the name of the University nor the names of its contributors
179b50d902SRodney W. Grimes  *    may be used to endorse or promote products derived from this software
189b50d902SRodney W. Grimes  *    without specific prior written permission.
199b50d902SRodney W. Grimes  *
209b50d902SRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
219b50d902SRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
229b50d902SRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
239b50d902SRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
249b50d902SRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
259b50d902SRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
269b50d902SRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
279b50d902SRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
289b50d902SRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
299b50d902SRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
309b50d902SRodney W. Grimes  * SUCH DAMAGE.
319b50d902SRodney W. Grimes  */
329b50d902SRodney W. Grimes 
339f5b04e9SDavid Malone #if 0
349b50d902SRodney W. Grimes #ifndef lint
359f5b04e9SDavid Malone static char sccsid[] = "@(#)apply.c	8.4 (Berkeley) 4/4/94";
36b26a5ea6SKris Kennaway #endif
379f5b04e9SDavid Malone #endif
389f5b04e9SDavid Malone 
399f5b04e9SDavid Malone #include <sys/cdefs.h>
409f5b04e9SDavid Malone __FBSDID("$FreeBSD$");
419b50d902SRodney W. Grimes 
42aab65414SBrian Somers #include <sys/types.h>
4312794df2SJaakko Heinonen #include <sys/sbuf.h>
449b50d902SRodney W. Grimes #include <sys/wait.h>
459b50d902SRodney W. Grimes 
469b50d902SRodney W. Grimes #include <ctype.h>
479b50d902SRodney W. Grimes #include <err.h>
4812794df2SJaakko Heinonen #include <errno.h>
499b50d902SRodney W. Grimes #include <paths.h>
509b50d902SRodney W. Grimes #include <signal.h>
519b50d902SRodney W. Grimes #include <stdio.h>
529b50d902SRodney W. Grimes #include <stdlib.h>
539b50d902SRodney W. Grimes #include <string.h>
549b50d902SRodney W. Grimes #include <unistd.h>
559b50d902SRodney W. Grimes 
56f6973480SWill Andrews #define EXEC	"exec "
579b50d902SRodney W. Grimes 
58befcdd43SXin LI static int	exec_shell(const char *, const char *, const char *);
59cdc8ffbbSWill Andrews static void	usage(void);
60cdc8ffbbSWill Andrews 
619b50d902SRodney W. Grimes int
6212794df2SJaakko Heinonen main(int argc, char *argv[])
6312794df2SJaakko Heinonen {
6412794df2SJaakko Heinonen 	struct sbuf *cmdbuf;
6512794df2SJaakko Heinonen 	long arg_max;
66cdc8ffbbSWill Andrews 	int ch, debug, i, magic, n, nargs, offset, rval;
6712794df2SJaakko Heinonen 	size_t cmdsize;
6812794df2SJaakko Heinonen 	char *cmd, *name, *p, *shell, *slashp, *tmpshell;
699b50d902SRodney W. Grimes 
709b50d902SRodney W. Grimes 	debug = 0;
719b50d902SRodney W. Grimes 	magic = '%';		/* Default magic char is `%'. */
729b50d902SRodney W. Grimes 	nargs = -1;
739861e6dfSWill Andrews 	while ((ch = getopt(argc, argv, "a:d0123456789")) != -1)
749b50d902SRodney W. Grimes 		switch (ch) {
759b50d902SRodney W. Grimes 		case 'a':
769b50d902SRodney W. Grimes 			if (optarg[1] != '\0')
779b50d902SRodney W. Grimes 				errx(1,
786ec34d21SPhilippe Charnier 				    "illegal magic character specification");
799b50d902SRodney W. Grimes 			magic = optarg[0];
809b50d902SRodney W. Grimes 			break;
819b50d902SRodney W. Grimes 		case 'd':
829b50d902SRodney W. Grimes 			debug = 1;
839b50d902SRodney W. Grimes 			break;
849b50d902SRodney W. Grimes 		case '0': case '1': case '2': case '3': case '4':
859b50d902SRodney W. Grimes 		case '5': case '6': case '7': case '8': case '9':
869b50d902SRodney W. Grimes 			if (nargs != -1)
879b50d902SRodney W. Grimes 				errx(1,
886ec34d21SPhilippe Charnier 				    "only one -# argument may be specified");
899b50d902SRodney W. Grimes 			nargs = optopt - '0';
909b50d902SRodney W. Grimes 			break;
919b50d902SRodney W. Grimes 		default:
929b50d902SRodney W. Grimes 			usage();
939b50d902SRodney W. Grimes 		}
949b50d902SRodney W. Grimes 	argc -= optind;
959b50d902SRodney W. Grimes 	argv += optind;
969b50d902SRodney W. Grimes 
979b50d902SRodney W. Grimes 	if (argc < 2)
989b50d902SRodney W. Grimes 		usage();
999b50d902SRodney W. Grimes 
1009b50d902SRodney W. Grimes 	/*
1019b50d902SRodney W. Grimes 	 * The command to run is argv[0], and the args are argv[1..].
1029b50d902SRodney W. Grimes 	 * Look for %digit references in the command, remembering the
1039b50d902SRodney W. Grimes 	 * largest one.
1049b50d902SRodney W. Grimes 	 */
1059b50d902SRodney W. Grimes 	for (n = 0, p = argv[0]; *p != '\0'; ++p)
1069b50d902SRodney W. Grimes 		if (p[0] == magic && isdigit(p[1]) && p[1] != '0') {
1079b50d902SRodney W. Grimes 			++p;
1089b50d902SRodney W. Grimes 			if (p[0] - '0' > n)
1099b50d902SRodney W. Grimes 				n = p[0] - '0';
1109b50d902SRodney W. Grimes 		}
1119b50d902SRodney W. Grimes 
1129b50d902SRodney W. Grimes 	/*
113f6973480SWill Andrews 	 * Figure out the shell and name arguments to pass to execl()
114f6973480SWill Andrews 	 * in exec_shell().  Always malloc() shell and just set name
115f6973480SWill Andrews 	 * to point at the last part of shell if there are any backslashes,
116f6973480SWill Andrews 	 * otherwise just set it to point at the space malloc()'d.  If
117f6973480SWill Andrews 	 * SHELL environment variable exists, replace contents of
118f6973480SWill Andrews 	 * shell with it.
119f6973480SWill Andrews 	 */
120f6973480SWill Andrews 	shell = name = NULL;
121f6973480SWill Andrews 	tmpshell = getenv("SHELL");
122f6973480SWill Andrews 	shell = (tmpshell != NULL) ? strdup(tmpshell) : strdup(_PATH_BSHELL);
123f6973480SWill Andrews 	if (shell == NULL)
124f6973480SWill Andrews 		err(1, "strdup() failed");
125f6973480SWill Andrews 	slashp = strrchr(shell, '/');
126f6973480SWill Andrews 	name = (slashp != NULL) ? slashp + 1 : shell;
127f6973480SWill Andrews 
128f6973480SWill Andrews 	/*
1299b50d902SRodney W. Grimes 	 * If there were any %digit references, then use those, otherwise
1309b50d902SRodney W. Grimes 	 * build a new command string with sufficient %digit references at
1319b50d902SRodney W. Grimes 	 * the end to consume (nargs) arguments each time round the loop.
132f6973480SWill Andrews 	 * Allocate enough space to hold the maximum command.  Save the
133f6973480SWill Andrews 	 * size to pass to snprintf().
1349b50d902SRodney W. Grimes 	 */
135cdc8ffbbSWill Andrews 	cmdsize = sizeof(EXEC) - 1 + strlen(argv[0])
136cdc8ffbbSWill Andrews 	    + 9 * (sizeof(" %1") - 1) + 1;
137f6973480SWill Andrews 	if ((cmd = malloc(cmdsize)) == NULL)
1389b50d902SRodney W. Grimes 		err(1, NULL);
1399b50d902SRodney W. Grimes 
1409b50d902SRodney W. Grimes 	if (n == 0) {
1419b50d902SRodney W. Grimes 		/* If nargs not set, default to a single argument. */
1429b50d902SRodney W. Grimes 		if (nargs == -1)
1439b50d902SRodney W. Grimes 			nargs = 1;
1449b50d902SRodney W. Grimes 
1459b50d902SRodney W. Grimes 		p = cmd;
146f6973480SWill Andrews 		offset = snprintf(cmd, cmdsize, EXEC "%s", argv[0]);
14708741a6fSBrian Somers 		if ((size_t)offset >= cmdsize)
14812794df2SJaakko Heinonen 			errx(1, "snprintf() failed");
149f6973480SWill Andrews 		p += offset;
15034bc43d5SWill Andrews 		cmdsize -= offset;
151f6973480SWill Andrews 		for (i = 1; i <= nargs; i++) {
152f6973480SWill Andrews 			offset = snprintf(p, cmdsize, " %c%d", magic, i);
15308741a6fSBrian Somers 			if ((size_t)offset >= cmdsize)
15412794df2SJaakko Heinonen 				errx(1, "snprintf() failed");
155f6973480SWill Andrews 			p += offset;
15634bc43d5SWill Andrews 			cmdsize -= offset;
157f6973480SWill Andrews 		}
1589b50d902SRodney W. Grimes 
1599b50d902SRodney W. Grimes 		/*
1609b50d902SRodney W. Grimes 		 * If nargs set to the special value 0, eat a single
1619b50d902SRodney W. Grimes 		 * argument for each command execution.
1629b50d902SRodney W. Grimes 		 */
1639b50d902SRodney W. Grimes 		if (nargs == 0)
1649b50d902SRodney W. Grimes 			nargs = 1;
1659b50d902SRodney W. Grimes 	} else {
166f6973480SWill Andrews 		offset = snprintf(cmd, cmdsize, EXEC "%s", argv[0]);
16708741a6fSBrian Somers 		if ((size_t)offset >= cmdsize)
16812794df2SJaakko Heinonen 			errx(1, "snprintf() failed");
1699b50d902SRodney W. Grimes 		nargs = n;
1709b50d902SRodney W. Grimes 	}
1719b50d902SRodney W. Grimes 
17212794df2SJaakko Heinonen 	cmdbuf = sbuf_new(NULL, NULL, 1024, SBUF_AUTOEXTEND);
17312794df2SJaakko Heinonen 	if (cmdbuf == NULL)
1749b50d902SRodney W. Grimes 		err(1, NULL);
1759b50d902SRodney W. Grimes 
17612794df2SJaakko Heinonen 	arg_max = sysconf(_SC_ARG_MAX);
17712794df2SJaakko Heinonen 
1789b50d902SRodney W. Grimes 	/*
1799b50d902SRodney W. Grimes 	 * (argc) and (argv) are still offset by one to make it simpler to
1809b50d902SRodney W. Grimes 	 * expand %digit references.  At the end of the loop check for (argc)
1819b50d902SRodney W. Grimes 	 * equals 1 means that all the (argv) has been consumed.
1829b50d902SRodney W. Grimes 	 */
1839b50d902SRodney W. Grimes 	for (rval = 0; argc > nargs; argc -= nargs, argv += nargs) {
18412794df2SJaakko Heinonen 		sbuf_clear(cmdbuf);
1859b50d902SRodney W. Grimes 		/* Expand command argv references. */
18612794df2SJaakko Heinonen 		for (p = cmd; *p != '\0'; ++p) {
187f6973480SWill Andrews 			if (p[0] == magic && isdigit(p[1]) && p[1] != '0') {
18812794df2SJaakko Heinonen 				if (sbuf_cat(cmdbuf, argv[(++p)[0] - '0'])
18912794df2SJaakko Heinonen 				    == -1)
19012794df2SJaakko Heinonen 					errc(1, ENOMEM, "sbuf");
19112794df2SJaakko Heinonen 			} else {
19212794df2SJaakko Heinonen 				if (sbuf_putc(cmdbuf, *p) == -1)
19312794df2SJaakko Heinonen 					errc(1, ENOMEM, "sbuf");
19412794df2SJaakko Heinonen 			}
19512794df2SJaakko Heinonen 			if (sbuf_len(cmdbuf) > arg_max)
19612794df2SJaakko Heinonen 				errc(1, E2BIG, NULL);
19712794df2SJaakko Heinonen 		}
1989b50d902SRodney W. Grimes 
1999b50d902SRodney W. Grimes 		/* Terminate the command string. */
20012794df2SJaakko Heinonen 		sbuf_finish(cmdbuf);
2019b50d902SRodney W. Grimes 
2029b50d902SRodney W. Grimes 		/* Run the command. */
2039b50d902SRodney W. Grimes 		if (debug)
20412794df2SJaakko Heinonen 			(void)printf("%s\n", sbuf_data(cmdbuf));
2059b50d902SRodney W. Grimes 		else
20612794df2SJaakko Heinonen 			if (exec_shell(sbuf_data(cmdbuf), shell, name))
2079b50d902SRodney W. Grimes 				rval = 1;
2089b50d902SRodney W. Grimes 	}
2099b50d902SRodney W. Grimes 
2109b50d902SRodney W. Grimes 	if (argc != 1)
2119b50d902SRodney W. Grimes 		errx(1, "expecting additional argument%s after \"%s\"",
2129b50d902SRodney W. Grimes 		    (nargs - argc) ? "s" : "", argv[argc - 1]);
213f6973480SWill Andrews 	free(cmd);
21412794df2SJaakko Heinonen 	sbuf_delete(cmdbuf);
215f6973480SWill Andrews 	free(shell);
2169b50d902SRodney W. Grimes 	exit(rval);
2179b50d902SRodney W. Grimes }
2189b50d902SRodney W. Grimes 
2199b50d902SRodney W. Grimes /*
220f6973480SWill Andrews  * exec_shell --
221f6973480SWill Andrews  * 	Execute a shell command using passed use_shell and use_name
222f6973480SWill Andrews  * 	arguments.
2239b50d902SRodney W. Grimes  */
224f6973480SWill Andrews static int
225befcdd43SXin LI exec_shell(const char *command, const char *use_shell, const char *use_name)
2269b50d902SRodney W. Grimes {
2279b50d902SRodney W. Grimes 	pid_t pid;
228ec10002dSEivind Eklund 	int omask, pstat;
2299b50d902SRodney W. Grimes 	sig_t intsave, quitsave;
2309b50d902SRodney W. Grimes 
2319861e6dfSWill Andrews 	if (!command)		/* just checking... */
2329b50d902SRodney W. Grimes 		return(1);
2339b50d902SRodney W. Grimes 
2349b50d902SRodney W. Grimes 	omask = sigblock(sigmask(SIGCHLD));
2356ad13586SKris Kennaway 	switch(pid = vfork()) {
2369b50d902SRodney W. Grimes 	case -1:			/* error */
2376ad13586SKris Kennaway 		err(1, "vfork");
2389b50d902SRodney W. Grimes 	case 0:				/* child */
2399b50d902SRodney W. Grimes 		(void)sigsetmask(omask);
2407bc6d015SBrian Somers 		execl(use_shell, use_name, "-c", command, (char *)NULL);
241f6973480SWill Andrews 		warn("%s", use_shell);
2426ad13586SKris Kennaway 		_exit(1);
2439b50d902SRodney W. Grimes 	}
2449b50d902SRodney W. Grimes 	intsave = signal(SIGINT, SIG_IGN);
2459b50d902SRodney W. Grimes 	quitsave = signal(SIGQUIT, SIG_IGN);
246ec10002dSEivind Eklund 	pid = waitpid(pid, &pstat, 0);
2479b50d902SRodney W. Grimes 	(void)sigsetmask(omask);
2489b50d902SRodney W. Grimes 	(void)signal(SIGINT, intsave);
2499b50d902SRodney W. Grimes 	(void)signal(SIGQUIT, quitsave);
250ec10002dSEivind Eklund 	return(pid == -1 ? -1 : pstat);
2519b50d902SRodney W. Grimes }
2529b50d902SRodney W. Grimes 
253befcdd43SXin LI static void
254da8de1e2SAlfred Perlstein usage(void)
2559b50d902SRodney W. Grimes {
2569b50d902SRodney W. Grimes 
2579b50d902SRodney W. Grimes 	(void)fprintf(stderr,
2589861e6dfSWill Andrews 	"usage: apply [-a magic] [-d] [-0123456789] command arguments ...\n");
2599b50d902SRodney W. Grimes 	exit(1);
2609b50d902SRodney W. Grimes }
261