--- ray/src/common/unix_process.c 2019/05/14 18:37:36 3.14 +++ ray/src/common/unix_process.c 2025/10/17 18:14:53 3.20 @@ -1,5 +1,5 @@ #ifndef lint -static const char RCSid[] = "$Id: unix_process.c,v 3.14 2019/05/14 18:37:36 greg Exp $"; +static const char RCSid[] = "$Id: unix_process.c,v 3.20 2025/10/17 18:14:53 greg Exp $"; #endif /* * Routines to communicate with separate process via dual pipes @@ -10,108 +10,145 @@ static const char RCSid[] = "$Id: unix_process.c,v 3.1 #include "copyright.h" -#include #include -#include #include #include "rtprocess.h" #include "rtio.h" +SUBPROC sp_inactive = SP_INACTIVE; + + int open_process( /* open communication to separate process */ -SUBPROC *pd, -char *av[] + SUBPROC *pd, + char *av[] ) { char *compath; int p0[2], p1[2]; - pd->pid = -1; - pd->running = 0; /* not going yet */ + if (pd->pid > 0) + return(-1); /* need to close, first */ + if ((pd->flags&(PF_FILT_INP|PF_FILT_OUT)) == (PF_FILT_INP|PF_FILT_OUT)) + return(-1); /* circular process not supported */ + + pd->flags &= ~PF_RUNNING; /* not running for sure */ + pd->pid = -1; + if (av == NULL) /* cloning operation? */ compath = NULL; else if ((compath = getpath(av[0], getenv("PATH"), X_OK)) == NULL) return(0); - if (pipe(p0) < 0 || pipe(p1) < 0) + + if (pd->flags & PF_FILT_INP) { /* filterning input stream? */ + if ((pd->r < 0) | (pd->r == 1)) + return(-1); + p0[0] = pd->r; + p0[1] = -1; + } else if (pipe(p0) < 0) return(-1); -#ifdef BSD - if (compath != NULL) - pd->pid = vfork(); /* more efficient with exec() */ - else -#endif + + if (pd->flags & PF_FILT_OUT) { /* filtering output stream? */ + if (pd->w < 1) + return(-1); + p1[0] = -1; + p1[1] = pd->w; + } else if (pipe(p1) < 0) + return(-1); pd->pid = fork(); if (pd->pid == 0) { /* if child... */ close(p0[1]); close(p1[0]); if (p0[0] != 0) { /* connect p0 to stdin */ - dup2(p0[0], 0); + if (dup2(p0[0], 0) < 0) + return(-1); close(p0[0]); } if (p1[1] != 1) { /* connect p1 to stdout */ - dup2(p1[1], 1); + if (dup2(p1[1], 1) < 0) + return(-1); close(p1[1]); } if (compath == NULL) /* just cloning? */ return(0); - execv(compath, av); /* else exec command */ + /* else clear streams' FD_CLOEXEC */ + if (p0[0] == 0) + fcntl(0, F_SETFD, 0); + if (p1[1] == 1) + fcntl(1, F_SETFD, 0); + execv(compath, av); /* exec command */ perror(compath); _exit(127); } if (pd->pid == -1) return(-1); - close(p0[0]); - close(p1[1]); - pd->r = p1[0]; - pd->w = p0[1]; + /* connect parent's streams */ + if (!(pd->flags & PF_FILT_INP)) { + close(p0[0]); + pd->r = p1[0]; + } else if (p1[0] != pd->r) { + if (dup2(p1[0], pd->r) < 0) + return(-1); + close(p1[0]); + } + if (!(pd->flags & PF_FILT_OUT)) { + close(p1[1]); + pd->w = p0[1]; + } else if (p0[1] != pd->w) { + if (dup2(p0[1], pd->w) < 0) + return(-1); + close(p0[1]); + } /* * Close write stream on exec to avoid multiprocessing deadlock. * No use in read stream without it, so set flag there as well. * GW: This bug took me two days to figure out!! */ - fcntl(pd->r, F_SETFD, FD_CLOEXEC); - fcntl(pd->w, F_SETFD, FD_CLOEXEC); - pd->running = 1; + if (pd->r > 0) + fcntl(pd->r, F_SETFD, FD_CLOEXEC); + if (pd->w > 1) + fcntl(pd->w, F_SETFD, FD_CLOEXEC); + pd->flags |= PF_RUNNING; return(PIPE_BUF); } int close_processes( /* close pipes and wait for processes to finish */ -SUBPROC pd[], -int nproc + SUBPROC pd[], + int nproc ) { int togo = nproc; - int status, rtn_status = 0; + int status = 0, rtn_status = 0; RT_PID pid; int i; for (i = 0; i < nproc; i++) /* close pipes, first */ - if (pd[i].running) { - close(pd[i].w); - close(pd[i].r); - pd[i].running = 0; + if (pd[i].flags & PF_RUNNING) { + if (pd[i].w >= 0) close(pd[i].w); + if (pd[i].r >= 0) close(pd[i].r); + pd[i].flags &= ~PF_RUNNING; } else togo -= (pd[i].pid < 0); if (nproc == 1) { /* await specific process? */ if (waitpid(pd->pid, &status, 0) != pd->pid) return(-1); - pd->pid = -1; + *pd = sp_inactive; return(status>>8 & 0xff); } /* else unordered wait */ while (togo > 0 && (pid = wait(&status)) >= 0) { for (i = nproc; i-- > 0; ) - if (pd[i].pid == pid) { - pd[i].pid = -1; - --togo; - break; - } + if (pd[i].pid == pid) + break; /* found it! */ if (i < 0) - continue; /* child we don't know? */ + continue; /* or child not in our list */ + pd[i] = sp_inactive; + --togo; status = status>>8 & 0xff; if (status) /* record non-zero status */ rtn_status = status;