sbase/xargs.c

280 lines
4.4 KiB
C
Raw Normal View History

2014-01-03 11:52:47 +00:00
/* See LICENSE file for copyright and license details. */
2015-02-14 20:02:41 +00:00
#include <sys/wait.h>
#include <errno.h>
#include <limits.h>
#include <stdint.h>
2014-01-03 11:52:47 +00:00
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
2014-01-03 11:52:47 +00:00
#include "util.h"
#define NARGS 10000
2014-01-03 11:52:47 +00:00
static int inputc(void);
2014-01-07 14:56:05 +00:00
static void fillargbuf(int);
static int eatspace(void);
2014-01-03 11:52:47 +00:00
static int parsequote(int);
static int parseescape(void);
2014-01-03 11:52:47 +00:00
static char *poparg(void);
static void waitchld(void);
2014-01-08 20:27:27 +00:00
static void spawn(void);
2014-01-03 11:52:47 +00:00
2014-01-08 20:29:34 +00:00
static size_t argbsz;
2014-01-03 11:52:47 +00:00
static size_t argbpos;
static size_t maxargs = 0;
static int nerrors = 0;
static int rflag = 0, nflag = 0, tflag = 0, xflag = 0;
static char *argb;
static char *cmd[NARGS];
static char *eofstr;
2014-01-03 11:52:47 +00:00
static int
inputc(void)
{
int ch;
ch = getc(stdin);
if (ch == EOF && ferror(stdin))
eprintf("getc <stdin>:");
2014-01-03 11:52:47 +00:00
return ch;
2014-01-03 11:52:47 +00:00
}
static void
2014-01-07 14:56:05 +00:00
fillargbuf(int ch)
2014-01-03 11:52:47 +00:00
{
if (argbpos >= argbsz) {
2014-01-08 20:29:34 +00:00
argbsz = argbpos == 0 ? 1 : argbsz * 2;
argb = erealloc(argb, argbsz);
2014-01-03 11:52:47 +00:00
}
argb[argbpos] = ch;
}
static int
2014-01-03 11:52:47 +00:00
eatspace(void)
{
int ch;
while ((ch = inputc()) != EOF) {
switch (ch) {
case ' ': case '\t': case '\n':
break;
default:
ungetc(ch, stdin);
return ch;
2014-01-03 11:52:47 +00:00
}
}
return -1;
2014-01-03 11:52:47 +00:00
}
static int
parsequote(int q)
{
int ch;
while ((ch = inputc()) != EOF) {
2014-01-08 20:22:10 +00:00
if (ch == q)
2014-01-03 11:52:47 +00:00
return 0;
if (ch != '\n') {
2014-01-07 14:56:05 +00:00
fillargbuf(ch);
2014-01-03 11:52:47 +00:00
argbpos++;
}
}
2014-01-03 11:52:47 +00:00
return -1;
}
static int
2014-01-03 11:52:47 +00:00
parseescape(void)
{
int ch;
if ((ch = inputc()) != EOF) {
2014-01-07 14:56:05 +00:00
fillargbuf(ch);
2014-01-03 11:52:47 +00:00
argbpos++;
return ch;
2014-01-03 11:52:47 +00:00
}
return -1;
2014-01-03 11:52:47 +00:00
}
static char *
poparg(void)
{
int ch;
argbpos = 0;
2014-11-19 19:59:37 +00:00
if (eatspace() < 0)
return NULL;
2014-01-03 11:52:47 +00:00
while ((ch = inputc()) != EOF) {
switch (ch) {
case ' ': case '\t': case '\n':
2014-01-07 11:53:55 +00:00
goto out;
2014-01-03 11:52:47 +00:00
case '\'':
2014-11-19 19:59:37 +00:00
if (parsequote('\'') < 0)
eprintf("unterminated single quote\n");
2014-01-03 11:52:47 +00:00
break;
case '\"':
2014-11-19 19:59:37 +00:00
if (parsequote('\"') < 0)
eprintf("unterminated double quote\n");
2014-01-03 11:52:47 +00:00
break;
case '\\':
2014-11-19 19:59:37 +00:00
if (parseescape() < 0)
eprintf("backslash at EOF\n");
2014-01-03 11:52:47 +00:00
break;
default:
2014-01-07 14:56:05 +00:00
fillargbuf(ch);
2014-01-03 11:52:47 +00:00
argbpos++;
break;
}
}
2014-01-07 11:53:55 +00:00
out:
fillargbuf('\0');
return (eofstr && !strcmp(argb, eofstr)) ? NULL : argb;
2014-01-03 11:52:47 +00:00
}
static void
waitchld(void)
2014-01-03 11:52:47 +00:00
{
int status;
2014-01-03 11:52:47 +00:00
wait(&status);
if (WIFEXITED(status)) {
if (WEXITSTATUS(status) == 255)
exit(124);
if (WEXITSTATUS(status) == 127 ||
WEXITSTATUS(status) == 126)
exit(WEXITSTATUS(status));
if (status)
nerrors++;
}
if (WIFSIGNALED(status))
exit(125);
2014-01-03 11:52:47 +00:00
}
static void
2014-01-08 20:27:27 +00:00
spawn(void)
{
2014-01-27 15:16:43 +00:00
int savederrno;
int first = 1;
char **p;
if (tflag) {
for (p = cmd; *p; p++) {
if (!first)
fputc(' ', stderr);
fputs(*p, stderr);
first = 0;
}
fputc('\n', stderr);
}
switch (fork()) {
case -1:
eprintf("fork:");
case 0:
execvp(*cmd, cmd);
2014-01-27 15:16:43 +00:00
savederrno = errno;
weprintf("execvp %s:", *cmd);
_exit(126 + (savederrno == ENOENT));
}
waitchld();
}
2015-03-07 12:36:40 +00:00
static void
usage(void)
{
eprintf("usage: %s [-rtx] [-E eofstr] [-n num] [-s num] [cmd [arg ...]]\n", argv0);
2015-03-07 12:36:40 +00:00
}
int
main(int argc, char *argv[])
{
int ret = 0, leftover = 0, i;
size_t argsz, argmaxsz;
size_t arglen, a;
2015-03-07 12:36:40 +00:00
char *arg = "";
if ((argmaxsz = sysconf(_SC_ARG_MAX)) == (size_t)-1)
argmaxsz = _POSIX_ARG_MAX;
/* Leave some room for environment variables */
argmaxsz -= 4096;
2015-03-07 12:36:40 +00:00
ARGBEGIN {
case 'n':
nflag = 1;
maxargs = estrtonum(EARGF(usage()), 1, MIN(SIZE_MAX, LLONG_MAX));
2015-03-07 12:36:40 +00:00
break;
case 'r':
rflag = 1;
break;
case 's':
argmaxsz = estrtonum(EARGF(usage()), 1, MIN(SIZE_MAX, LLONG_MAX));
break;
case 't':
tflag = 1;
break;
case 'x':
xflag = 1;
break;
2015-03-07 12:36:40 +00:00
case 'E':
eofstr = EARGF(usage());
break;
default:
usage();
} ARGEND
2015-03-07 12:36:40 +00:00
do {
argsz = 0; i = 0; a = 0;
if (argc) {
2015-03-07 12:36:40 +00:00
for (; i < argc; i++) {
cmd[i] = estrdup(argv[i]);
argsz += strlen(cmd[i]) + 1;
}
} else {
cmd[i] = estrdup("/bin/echo");
2015-03-27 21:45:03 +00:00
argsz += strlen("/bin/echo") + 1;
2015-03-07 12:36:40 +00:00
i++;
}
while (leftover || (arg = poparg())) {
2015-03-27 21:45:03 +00:00
arglen = strlen(arg);
if (argsz + arglen >= argmaxsz || i >= NARGS - 1) {
if (arglen >= argmaxsz) {
weprintf("insufficient argument space\n");
if (xflag)
exit(1);
}
2015-03-07 12:36:40 +00:00
leftover = 1;
break;
}
cmd[i] = estrdup(arg);
2015-03-27 21:45:03 +00:00
argsz += arglen + 1;
2015-03-07 12:36:40 +00:00
i++;
a++;
leftover = 0;
if (nflag && a >= maxargs)
2015-03-07 12:36:40 +00:00
break;
}
cmd[i] = NULL;
if (a >= maxargs && nflag)
2015-03-07 12:36:40 +00:00
spawn();
else if (!a || (i == 1 && rflag))
2015-03-07 12:36:40 +00:00
;
else
spawn();
for (; i >= 0; i--)
free(cmd[i]);
} while (arg);
free(argb);
if (nerrors || (fshut(stdin, "<stdin>") | fshut(stdout, "<stdout>")))
ret = 123;
return ret;
2015-03-07 12:36:40 +00:00
}