This is a live mirror of the Perl 5 development currently hosted at https://github.com/perl/perl5
more complete pseudo-fork() support for Windows
[perl5.git] / os2 / os2.c
index 6b814de..8a17ae7 100644 (file)
--- a/os2/os2.c
+++ b/os2/os2.c
@@ -3,6 +3,10 @@
 #define INCL_DOSFILEMGR
 #define INCL_DOSMEMMGR
 #define INCL_DOSERRORS
+/* These 3 are needed for compile if os2.h includes os2tk.h, not os2emx.h */
+#define INCL_DOSPROCESS
+#define SPU_DISABLESUPPRESSION          0
+#define SPU_ENABLESUPPRESSION           1
 #include <os2.h>
 
 #include <sys/uflags.h>
@@ -412,6 +416,7 @@ result(int flag, int pid)
 #define EXECF_EXEC 1
 #define EXECF_TRUEEXEC 2
 #define EXECF_SPAWN_NOWAIT 3
+#define EXECF_SPAWN_BYFLAG 4
 
 /* const char* const ptypes[] = { "FS", "DOS", "VIO", "PM", "DETACH" }; */
 
@@ -459,11 +464,12 @@ static ULONG os2_mytype;
 /* global PL_Argv[] contains arguments. */
 
 int
-do_spawn_ve(really, flag, execf, inicmd)
+do_spawn_ve(really, flag, execf, inicmd, addflag)
 SV *really;
 U32 flag;
 U32 execf;
 char *inicmd;
+U32 addflag;
 {
     dTHR;
        int trueflag = flag;
@@ -476,6 +482,7 @@ char *inicmd;
        char **argsp = fargs;
        char nargs = 4;
        int force_shell;
+       int new_stderr = -1, nostderr = 0, fl_stderr;
        STRLEN n_a;
        
        if (flag == P_WAIT)
@@ -558,6 +565,24 @@ char *inicmd;
            }
        }
 
+       if (addflag) {
+           addflag = 0;
+           new_stderr = dup(2);                /* Preserve stderr */
+           if (new_stderr == -1) {
+               if (errno == EBADF)
+                   nostderr = 1;
+               else {
+                   rc = -1;
+                   goto finish;
+               }
+           } else
+               fl_stderr = fcntl(2, F_GETFD);
+           rc = dup2(1,2);
+           if (rc == -1)
+               goto finish;
+           fcntl(new_stderr, F_SETFD, FD_CLOEXEC);
+       }
+
 #if 0
        rc = result(trueflag, spawnvp(flag,tmps,PL_Argv));
 #else
@@ -567,7 +592,7 @@ char *inicmd;
            rc = spawnvp(trueflag | P_OVERLAY,tmps,PL_Argv);
        else if (execf == EXECF_SPAWN_NOWAIT)
            rc = spawnvp(flag,tmps,PL_Argv);
-        else                           /* EXECF_SPAWN */
+        else                           /* EXECF_SPAWN, EXECF_SPAWN_BYFLAG */
            rc = result(trueflag, 
                        spawnvp(flag,tmps,PL_Argv));
 #endif 
@@ -593,7 +618,9 @@ char *inicmd;
                     if (l >= sizeof scrbuf) {
                        Safefree(scr);
                      longbuf:
-                       croak("Size of scriptname too big: %d", l);
+                       warn("Size of scriptname too big: %d", l);
+                      rc = -1;
+                      goto finish;
                     }
                     strcpy(scrbuf, scr);
                     Safefree(scr);
@@ -772,67 +799,34 @@ char *inicmd;
                goto retry;
            }
        }
-       if (rc < 0 && PL_dowarn)
-           warn("Can't %s \"%s\": %s\n", 
+       if (rc < 0 && ckWARN(WARN_EXEC))
+           Perl_warner(aTHX_ WARN_EXEC, "Can't %s \"%s\": %s\n", 
                 ((execf != EXECF_EXEC && execf != EXECF_TRUEEXEC) 
                  ? "spawn" : "exec"),
                 PL_Argv[0], Strerror(errno));
        if (rc < 0 && (execf != EXECF_SPAWN_NOWAIT) 
            && ((trueflag & 0xFF) == P_WAIT)) 
-           rc = 255 << 8; /* Emulate the fork(). */
-
-    return rc;
-}
-
-/* Array spawn.  */
-int
-do_aspawn(really,mark,sp)
-SV *really;
-register SV **mark;
-register SV **sp;
-{
-    dTHR;
-    register char **a;
-    char *tmps = NULL;
-    int rc;
-    int flag = P_WAIT, trueflag, err, secondtry = 0;
-    STRLEN n_a;
-
-    if (sp > mark) {
-       New(1301,PL_Argv, sp - mark + 3, char*);
-       a = PL_Argv;
-
-       if (mark < sp && SvNIOKp(*(mark+1)) && !SvPOKp(*(mark+1))) {
-               ++mark;
-               flag = SvIVx(*mark);
-       }
-
-       while (++mark <= sp) {
-           if (*mark)
-               *a++ = SvPVx(*mark, n_a);
-           else
-               *a++ = "";
-       }
-       *a = Nullch;
-
-       rc = do_spawn_ve(really, flag, EXECF_SPAWN, NULL);
-    } else
-       rc = -1;
-    do_execfree();
+           rc = -1;
+
+  finish:
+    if (new_stderr != -1) {    /* How can we use error codes? */
+       dup2(new_stderr, 2);
+       close(new_stderr);
+       fcntl(2, F_SETFD, fl_stderr);
+    } else if (nostderr)
+       close(2);
     return rc;
 }
 
 /* Try converting 1-arg form to (usually shell-less) multi-arg form. */
 int
-do_spawn2(cmd, execf)
-char *cmd;
-int execf;
+do_spawn3(char *cmd, int execf, int flag)
 {
     register char **a;
     register char *s;
     char flags[10];
     char *shell, *copt, *news = NULL;
-    int rc, err, seenspace = 0;
+    int rc, err, seenspace = 0, mergestderr = 0;
     char fullcmd[MAXNAMLEN + 1];
 
 #ifdef TRYSHELL
@@ -885,6 +879,18 @@ int execf;
                break;
            } else if (*s == '\\' && !seenspace) {
                continue;               /* Allow backslashes in names */
+           } else if (*s == '>' && s >= cmd + 3
+                       && s[-1] == '2' && s[1] == '&' && s[2] == '1'
+                       && isSPACE(s[-2]) ) {
+               char *t = s + 3;
+
+               while (*t && isSPACE(*t))
+                   t++;
+               if (!*t) {
+                   s[-2] = '\0';
+                   mergestderr = 1;
+                   break;              /* Allow 2>&1 as the last thing */
+               }
            }
            /* We do not convert this to do_spawn_ve since shell
               should be smart enough to start itself gloriously. */
@@ -895,15 +901,18 @@ int execf;
                 rc = spawnl(P_OVERLAY,shell,shell,copt,cmd,(char*)0);
            else if (execf == EXECF_SPAWN_NOWAIT)
                 rc = spawnl(P_NOWAIT,shell,shell,copt,cmd,(char*)0);
+           else if (execf == EXECF_SPAWN_BYFLAG)
+                rc = spawnl(flag,shell,shell,copt,cmd,(char*)0);
            else {
                /* In the ak code internal P_NOWAIT is P_WAIT ??? */
                rc = result(P_WAIT,
                            spawnl(P_NOWAIT,shell,shell,copt,cmd,(char*)0));
-               if (rc < 0 && PL_dowarn)
-                   warn("Can't %s \"%s\": %s", 
+               if (rc < 0 && ckWARN(WARN_EXEC))
+                   Perl_warner(aTHX_ WARN_EXEC, "Can't %s \"%s\": %s", 
                         (execf == EXECF_SPAWN ? "spawn" : "exec"),
                         shell, Strerror(errno));
-               if (rc < 0) rc = 255 << 8; /* Emulate the fork(). */
+               if (rc < 0)
+                   rc = -1;
            }
            if (news)
                Safefree(news);
@@ -927,7 +936,7 @@ int execf;
     }
     *a = Nullch;
     if (PL_Argv[0])
-       rc = do_spawn_ve(NULL, 0, execf, cmd);
+       rc = do_spawn_ve(NULL, flag, execf, cmd, mergestderr);
     else
        rc = -1;
     if (news)
@@ -936,25 +945,67 @@ int execf;
     return rc;
 }
 
+/* Array spawn.  */
+int
+do_aspawn(really,mark,sp)
+SV *really;
+register SV **mark;
+register SV **sp;
+{
+    dTHR;
+    register char **a;
+    int rc;
+    int flag = P_WAIT, flag_set = 0;
+    STRLEN n_a;
+
+    if (sp > mark) {
+       New(1301,PL_Argv, sp - mark + 3, char*);
+       a = PL_Argv;
+
+       if (mark < sp && SvNIOKp(*(mark+1)) && !SvPOKp(*(mark+1))) {
+               ++mark;
+               flag = SvIVx(*mark);
+               flag_set = 1;
+
+       }
+
+       while (++mark <= sp) {
+           if (*mark)
+               *a++ = SvPVx(*mark, n_a);
+           else
+               *a++ = "";
+       }
+       *a = Nullch;
+
+       if (flag_set && (a == PL_Argv + 1)) { /* One arg? */
+           rc = do_spawn3(a[-1], EXECF_SPAWN_BYFLAG, flag);
+       } else
+           rc = do_spawn_ve(really, flag, EXECF_SPAWN, NULL, 0);
+    } else
+       rc = -1;
+    do_execfree();
+    return rc;
+}
+
 int
 do_spawn(cmd)
 char *cmd;
 {
-    return do_spawn2(cmd, EXECF_SPAWN);
+    return do_spawn3(cmd, EXECF_SPAWN, 0);
 }
 
 int
 do_spawn_nowait(cmd)
 char *cmd;
 {
-    return do_spawn2(cmd, EXECF_SPAWN_NOWAIT);
+    return do_spawn3(cmd, EXECF_SPAWN_NOWAIT,0);
 }
 
 bool
 do_exec(cmd)
 char *cmd;
 {
-    do_spawn2(cmd, EXECF_EXEC);
+    do_spawn3(cmd, EXECF_EXEC, 0);
     return FALSE;
 }
 
@@ -962,7 +1013,7 @@ bool
 os2exec(cmd)
 char *cmd;
 {
-    return do_spawn2(cmd, EXECF_TRUEEXEC);
+    return do_spawn3(cmd, EXECF_TRUEEXEC, 0);
 }
 
 PerlIO *
@@ -977,6 +1028,7 @@ char       *mode;
     register I32 pid, rc;
     PerlIO *res;
     SV *sv;
+    int fh_fl;
     
     /* `this' is what we use in the parent, `that' in the child. */
     this = (*mode == 'w');
@@ -988,26 +1040,51 @@ char     *mode;
     if (pipe(p) < 0)
        return Nullfp;
     /* Now we need to spawn the child. */
+    if (p[this] == (*mode == 'r')) {   /* if fh 0/1 was initially closed. */
+       int new = dup(p[this]);
+
+       if (new == -1)
+           goto closepipes;
+       close(p[this]);
+       p[this] = new;
+    }
     newfd = dup(*mode == 'r');         /* Preserve std* */
-    if (p[that] != (*mode == 'r')) {
+    if (newfd == -1) {         
+       /* This cannot happen due to fh being bad after pipe(), since
+          pipe() should have created fh 0 and 1 even if they were
+          initially closed.  But we closed p[this] before.  */
+       if (errno != EBADF) {
+         closepipes:
+           close(p[0]);
+           close(p[1]);
+           return Nullfp;
+       }
+    } else
+       fh_fl = fcntl(*mode == 'r', F_GETFD);
+    if (p[that] != (*mode == 'r')) {   /* if fh 0/1 was initially closed. */
        dup2(p[that], *mode == 'r');
        close(p[that]);
     }
     /* Where is `this' and newfd now? */
     fcntl(p[this], F_SETFD, FD_CLOEXEC);
-    fcntl(newfd, F_SETFD, FD_CLOEXEC);
+    if (newfd != -1)
+       fcntl(newfd, F_SETFD, FD_CLOEXEC);
     pid = do_spawn_nowait(cmd);
-    if (newfd != (*mode == 'r')) {
+    if (newfd == -1)
+       close(*mode == 'r');            /* It was closed initially */
+    else if (newfd != (*mode == 'r')) {        /* Probably this check is not needed */
        dup2(newfd, *mode == 'r');      /* Return std* back. */
        close(newfd);
-    }
+       fcntl(*mode == 'r', F_SETFD, fh_fl);
+    } else
+       fcntl(*mode == 'r', F_SETFD, fh_fl);
     if (p[that] == (*mode == 'r'))
        close(p[that]);
     if (pid == -1) {
        close(p[this]);
-       return NULL;
+       return Nullfp;
     }
-    if (p[that] < p[this]) {
+    if (p[that] < p[this]) {           /* Make fh as small as possible */
        dup2(p[this], p[that]);
        close(p[this]);
        p[this] = p[that];
@@ -1173,10 +1250,12 @@ settmppath()
     if (!p) return;
     len = strlen(p);
     tpath = (char *)malloc(len + strlen(TMPPATH1) + 2);
-    strcpy(tpath, p);
-    tpath[len] = '/';
-    strcpy(tpath + len + 1, TMPPATH1);
-    tmppath = tpath;
+    if (tpath) {
+       strcpy(tpath, p);
+       tpath[len] = '/';
+       strcpy(tpath + len + 1, TMPPATH1);
+       tmppath = tpath;
+    }
 }
 
 #include "XSUB.h"
@@ -1248,7 +1327,7 @@ mod2fname(sv)
 #ifdef USE_THREADS
     sum++;                             /* Avoid conflict of DLLs in memory. */
 #endif 
-    sum += PATCHLEVEL * 200 + SUBVERSION * 2;  /*  */
+    sum += PERL_VERSION * 200 + PERL_SUBVERSION * 2;  /*  */
     fname[pos] = 'A' + (sum % 26);
     fname[pos + 1] = 'A' + (sum / 26 % 26);
     fname[pos + 2] = '\0';
@@ -1282,18 +1361,37 @@ os2error(int rc)
                return NULL;
        if (DosGetMessage(NULL, 0, buf, sizeof buf - 1, rc, "OSO001.MSG", &len))
                sprintf(buf, "OS/2 system error code %d=0x%x", rc, rc);
-       else
+       else {
                buf[len] = '\0';
-       if (len > 0 && buf[len - 1] == '\n')
-           buf[len - 1] = '\0';
-       if (len > 1 && buf[len - 2] == '\r')
-           buf[len - 2] = '\0';
-       if (len > 2 && buf[len - 3] == '.')
-           buf[len - 3] = '\0';
+               if (len && buf[len - 1] == '\n')
+                       buf[--len] = 0;
+               if (len && buf[len - 1] == '\r')
+                       buf[--len] = 0;
+               if (len && buf[len - 1] == '.')
+                       buf[--len] = 0;
+       }
        return buf;
 }
 
 char *
+os2_execname(void)
+{
+  char buf[300], *p;
+
+  if (_execname(buf, sizeof buf) != 0)
+       return PL_origargv[0];
+  p = buf;
+  while (*p) {
+    if (*p == '\\')
+       *p = '/';
+    p++;
+  }
+  p = savepv(buf);
+  SAVEFREEPV(p);
+  return p;
+}
+
+char *
 perllib_mangle(char *s, unsigned int l)
 {
     static char *newp, *oldp;
@@ -1993,7 +2091,7 @@ Perl_OS2_init(char **env)
     settmppath();
     OS2_Perl_data.xs_init = &Xs_OS2_init;
     _uflags (_UF_SBRK_MODEL, _UF_SBRK_ARBITRARY);
-    if (environ == NULL) {
+    if (environ == NULL && env) {
        environ = env;
     }
     if ( (shell = getenv("PERL_SH_DRIVE")) ) {