#include <stdio.h>
#include <signal.h>
#include <errno.h>
#include <sys/types.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifdef HAVE_STRERROR
#include <string.h>
#else
extern char *strerror();
#endif
#ifdef _POSIX_VERSION
#include <sys/wait.h>
#define PID_T pid_t
#else
#define WIFEXITED(s) (((s) & 0377) == 0)
#define WIFSTOPPED(s) (((s) & 0377) == 0177)
#define WIFSIGNALED(s) (((s) & 0377) != 0 && (((s) & 0377) != 0177))
#define WEXITSTATUS(s) (((s) >> 8) & 0377)
#define WTERMSIG(s) ((s) & 0177)
#define WSTOPSIG(s) (((s) >> 8) & 0377)
#ifndef WCOREFLAG
#define WCOREFLAG 0200
#endif
#define PID_T int
#endif
#ifndef WCOREFLAG
#ifdef WCOREFLG
#define WCOREFLAG WCOREFLG
#endif
#endif
#ifndef WCOREDUMP
#ifdef WCOREFLAG
#define WCOREDUMP(s) ((s) & WCOREFLAG)
#else
#define WCOREDUMP(s) (0)
#endif
#endif
#include "pipeline.h"
#ifdef __STDC__
#define P(parms) parms
#else
#define P(parms) ()
#define const
#endif
#define error c_error
extern void error P((const char *, const char *, const char *, const char *));
extern void c_fatal P((const char *, const char *, const char *, const char *));
static void sys_fatal P((const char *));
static const char *xstrsignal P((int));
static char *i_to_a P((int));
#if defined(__MSDOS__) \
|| (defined(_WIN32) && !defined(_UWIN) && !defined(__CYGWIN32__))
#include <process.h>
#include <fcntl.h>
#include <ctype.h>
#include <string.h>
#include "nonposix.h"
static int child_interrupted;
static RETSIGTYPE signal_catcher (int signo)
{
child_interrupted++;
}
static const char *sh = "sh";
static const char *command = "command";
const char *
system_shell_name (void)
{
static const char *shell_name;
if (shell_name == NULL)
{
int sh_found = spawnlp (P_WAIT, sh, sh, "-c", ":", NULL) == 0;
if (sh_found)
shell_name = sh;
else
shell_name = command;
}
return shell_name;
}
const char *
system_shell_dash_c (void)
{
if (strcmp (system_shell_name(), sh) == 0)
return "-c";
else
return "/c";
}
int
is_system_shell (const char *shell)
{
size_t shlen;
size_t ibase = 0, idot, i;
if (!shell)
return 0;
idot = shlen = strlen(shell);
for (i = 0; i < shlen; i++)
{
if (shell[i] == '.')
idot = i;
else if (shell[i] == '/' || shell[i] == '\\' || shell[i] == ':')
{
ibase = i + 1;
idot = shlen;
}
}
return
(strncasecmp (shell + ibase, system_shell_name (), idot - ibase) == 0
&& (idot == shlen
|| strcasecmp (shell + idot, ".exe") == 0
|| strcasecmp (shell + idot, ".com") == 0));
}
int run_pipeline (ncommands, commands, no_pipe)
int ncommands;
char ***commands;
int no_pipe;
{
int save_stdin = dup(0);
int save_stdout = dup(1);
char *tmpfiles[2];
char tem1[L_tmpnam], tem2[L_tmpnam];
int infile = 0;
int outfile = 1;
int i, f, ret = 0;
tmpfiles[0] = tmpnam(tem1);
tmpfiles[1] = tmpnam(tem2);
for (i = 0; i < ncommands; i++)
{
int exit_status;
RETSIGTYPE (*prev_handler)(int);
if (i)
{
f = open(tmpfiles[infile], O_RDONLY|O_BINARY, 0666);
if (f < 0)
sys_fatal("open stdin");
if (dup2(f, 0) < 0)
sys_fatal("dup2 stdin");
if (close(f) < 0)
sys_fatal("close stdin");
}
if ((i < ncommands - 1) && !no_pipe)
{
f = open(tmpfiles[outfile], O_WRONLY|O_CREAT|O_TRUNC|O_BINARY, 0666);
if (f < 0)
sys_fatal("open stdout");
if (dup2(f, 1) < 0)
sys_fatal("dup2 stdout");
if (close(f) < 0)
sys_fatal("close stdout");
}
else if (dup2(save_stdout, 1) < 0)
sys_fatal("restore stdout");
child_interrupted = 0;
prev_handler = signal(SIGINT, signal_catcher);
exit_status = spawnvp(P_WAIT, commands[i][0], commands[i]);
signal(SIGINT, prev_handler);
if (child_interrupted)
{
error("%1: Interrupted", commands[i][0], (char *)0, (char *)0);
ret |= 2;
}
else if (exit_status < 0)
{
error("couldn't exec %1: %2", commands[i][0],
strerror(errno), (char *)0);
fflush(stderr);
ret |= 4;
}
if (exit_status != 0)
ret |= 1;
if (ret != 0)
break;
infile = 1 - infile;
outfile = 1 - outfile;
}
if (dup2(save_stdin, 0) < 0)
sys_fatal("restore stdin");
unlink(tmpfiles[0]);
unlink(tmpfiles[1]);
return ret;
}
#else
int run_pipeline(ncommands, commands, no_pipe)
int ncommands;
char ***commands;
int no_pipe;
{
int i;
int last_input = 0;
PID_T pids[MAX_COMMANDS];
int ret = 0;
int proc_count = ncommands;
for (i = 0; i < ncommands; i++) {
int pdes[2];
PID_T pid;
if ((i != ncommands - 1) && !no_pipe) {
if (pipe(pdes) < 0)
sys_fatal("pipe");
}
pid = fork();
if (pid < 0)
sys_fatal("fork");
if (pid == 0) {
if (last_input != 0) {
if (close(0) < 0)
sys_fatal("close");
if (dup(last_input) < 0)
sys_fatal("dup");
if (close(last_input) < 0)
sys_fatal("close");
}
if ((i != ncommands - 1) && !no_pipe) {
if (close(1) < 0)
sys_fatal("close");
if (dup(pdes[1]) < 0)
sys_fatal("dup");
if (close(pdes[1]) < 0)
sys_fatal("close");
if (close(pdes[0]))
sys_fatal("close");
}
execvp(commands[i][0], commands[i]);
error("couldn't exec %1: %2", commands[i][0],
strerror(errno), (char *)0);
fflush(stderr);
_exit(EXEC_FAILED_EXIT_STATUS);
}
if (last_input != 0) {
if (close(last_input) < 0)
sys_fatal("close");
}
if ((i != ncommands - 1) && !no_pipe) {
if (close(pdes[1]) < 0)
sys_fatal("close");
last_input = pdes[0];
}
pids[i] = pid;
}
while (proc_count > 0) {
int status;
PID_T pid = wait(&status);
if (pid < 0)
sys_fatal("wait");
for (i = 0; i < ncommands; i++)
if (pids[i] == pid) {
pids[i] = -1;
--proc_count;
if (WIFSIGNALED(status)) {
int sig = WTERMSIG(status);
#ifdef SIGPIPE
if (sig == SIGPIPE) {
if (i == ncommands - 1) {
int j;
for (j = 0; j < ncommands; j++)
if (pids[j] > 0)
(void)kill(pids[j], SIGPIPE);
}
}
else
#endif
{
error("%1: %2%3",
commands[i][0],
xstrsignal(sig),
WCOREDUMP(status) ? " (core dumped)" : "");
ret |= 2;
}
}
else if (WIFEXITED(status)) {
int exit_status = WEXITSTATUS(status);
if (exit_status == EXEC_FAILED_EXIT_STATUS)
ret |= 4;
else if (exit_status != 0)
ret |= 1;
}
else
error("unexpected status %1",
i_to_a(status), (char *)0, (char *)0);
break;
}
}
return ret;
}
#endif
static void sys_fatal(s)
const char *s;
{
c_fatal("%1: %2", s, strerror(errno), (char *)0);
}
static char *i_to_a(n)
int n;
{
static char buf[12];
sprintf(buf, "%d", n);
return buf;
}
static const char *xstrsignal(n)
int n;
{
static char buf[sizeof("Signal ") + 1 + sizeof(int)*3];
#ifdef NSIG
#ifdef SYS_SIGLIST_DECLARED
if (n >= 0 && n < NSIG && sys_siglist[n] != 0)
return sys_siglist[n];
#endif
#endif
sprintf(buf, "Signal %d", n);
return buf;
}