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