The #4880 was too eager, the $n test is useful too.
[p5sagit/p5-mst-13.2.git] / os2 / os2.c
index ce1f209..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>
@@ -162,7 +166,8 @@ int
 os2_cond_wait(perl_cond *c, perl_mutex *m)
 {                                              
     int rc;
-    if ((rc = DosResetEventSem(*c,&PL_na)) && (rc != ERROR_ALREADY_RESET))
+    STRLEN n_a;
+    if ((rc = DosResetEventSem(*c,&n_a)) && (rc != ERROR_ALREADY_RESET))
        croak("panic: COND_WAIT-reset: rc=%i", rc);             
     if (m) MUTEX_UNLOCK(m);                                    
     if (CheckOSError(DosWaitEventSem(*c,SEM_INDEFINITE_WAIT))
@@ -411,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" }; */
 
@@ -458,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;
@@ -475,6 +482,8 @@ 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)
                flag = P_NOWAIT;
@@ -489,7 +498,7 @@ char *inicmd;
            ) /* will spawnvp use PATH? */
            TAINT_ENV();        /* testing IFS here is overkill, probably */
        /* We should check PERL_SH* and PERLLIB_* as well? */
-       if (!really || !*(tmps = SvPV(really, PL_na)))
+       if (!really || !*(tmps = SvPV(really, n_a)))
            tmps = PL_Argv[0];
 
       reread:
@@ -556,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
@@ -565,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 
@@ -591,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);
@@ -770,66 +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;
-
-    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, PL_na);
-           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
@@ -882,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. */
@@ -892,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);
@@ -924,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)
@@ -933,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;
 }
 
@@ -959,7 +1013,7 @@ bool
 os2exec(cmd)
 char *cmd;
 {
-    return do_spawn2(cmd, EXECF_TRUEEXEC);
+    return do_spawn3(cmd, EXECF_TRUEEXEC, 0);
 }
 
 PerlIO *
@@ -974,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');
@@ -985,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];
@@ -1170,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"
@@ -1184,8 +1266,9 @@ XS(XS_File__Copy_syscopy)
     if (items < 2 || items > 3)
        croak("Usage: File::Copy::syscopy(src,dst,flag=0)");
     {
-       char *  src = (char *)SvPV(ST(0),PL_na);
-       char *  dst = (char *)SvPV(ST(1),PL_na);
+       STRLEN n_a;
+       char *  src = (char *)SvPV(ST(0),n_a);
+       char *  dst = (char *)SvPV(ST(1),n_a);
        U32     flag;
        int     RETVAL, rc;
 
@@ -1214,6 +1297,7 @@ mod2fname(sv)
     AV  *av;
     SV  *svp;
     char *s;
+    STRLEN n_a;
 
     if (!SvROK(sv)) croak("Not a reference given to mod2fname");
     sv = SvRV(sv);
@@ -1224,7 +1308,7 @@ mod2fname(sv)
     if (avlen < 0) 
       croak("Empty array reference given to mod2fname");
 
-    s = SvPV(*av_fetch((AV*)sv, avlen, FALSE), PL_na);
+    s = SvPV(*av_fetch((AV*)sv, avlen, FALSE), n_a);
     strncpy(fname, s, 8);
     len = strlen(s);
     if (len < 6) pos = len;
@@ -1234,7 +1318,7 @@ mod2fname(sv)
     }
     avlen --;
     while (avlen >= 0) {
-       s = SvPV(*av_fetch((AV*)sv, avlen, FALSE), PL_na);
+       s = SvPV(*av_fetch((AV*)sv, avlen, FALSE), n_a);
        while (*s) {
            sum = 33 * sum + *(s++);    /* 7 is primitive mod 13. */
        }
@@ -1243,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';
@@ -1277,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;
@@ -1473,9 +1576,10 @@ XS(XS_OS2_Errors2Drive)
     if (items != 1)
        croak("Usage: OS2::Errors2Drive(drive)");
     {
+       STRLEN n_a;
        SV  *sv = ST(0);
        int     suppress = SvOK(sv);
-       char    *s = suppress ? SvPV(sv, PL_na) : NULL;
+       char    *s = suppress ? SvPV(sv, n_a) : NULL;
        char    drive = (s ? *s : 0);
        unsigned long rc;
 
@@ -1660,7 +1764,8 @@ XS(XS_Cwd_sys_chdir)
     if (items != 1)
        croak("Usage: Cwd::sys_chdir(path)");
     {
-       char *  path = (char *)SvPV(ST(0),PL_na);
+       STRLEN n_a;
+       char *  path = (char *)SvPV(ST(0),n_a);
        bool    RETVAL;
 
        RETVAL = sys_chdir(path);
@@ -1676,7 +1781,8 @@ XS(XS_Cwd_change_drive)
     if (items != 1)
        croak("Usage: Cwd::change_drive(d)");
     {
-       char    d = (char)*SvPV(ST(0),PL_na);
+       STRLEN n_a;
+       char    d = (char)*SvPV(ST(0),n_a);
        bool    RETVAL;
 
        RETVAL = change_drive(d);
@@ -1692,7 +1798,8 @@ XS(XS_Cwd_sys_is_absolute)
     if (items != 1)
        croak("Usage: Cwd::sys_is_absolute(path)");
     {
-       char *  path = (char *)SvPV(ST(0),PL_na);
+       STRLEN n_a;
+       char *  path = (char *)SvPV(ST(0),n_a);
        bool    RETVAL;
 
        RETVAL = sys_is_absolute(path);
@@ -1708,7 +1815,8 @@ XS(XS_Cwd_sys_is_rooted)
     if (items != 1)
        croak("Usage: Cwd::sys_is_rooted(path)");
     {
-       char *  path = (char *)SvPV(ST(0),PL_na);
+       STRLEN n_a;
+       char *  path = (char *)SvPV(ST(0),n_a);
        bool    RETVAL;
 
        RETVAL = sys_is_rooted(path);
@@ -1724,7 +1832,8 @@ XS(XS_Cwd_sys_is_relative)
     if (items != 1)
        croak("Usage: Cwd::sys_is_relative(path)");
     {
-       char *  path = (char *)SvPV(ST(0),PL_na);
+       STRLEN n_a;
+       char *  path = (char *)SvPV(ST(0),n_a);
        bool    RETVAL;
 
        RETVAL = sys_is_relative(path);
@@ -1755,7 +1864,8 @@ XS(XS_Cwd_sys_abspath)
     if (items < 1 || items > 2)
        croak("Usage: Cwd::sys_abspath(path, dir = NULL)");
     {
-       char *  path = (char *)SvPV(ST(0),PL_na);
+       STRLEN n_a;
+       char *  path = (char *)SvPV(ST(0),n_a);
        char *  dir;
        char p[MAXPATHLEN];
        char *  RETVAL;
@@ -1763,7 +1873,7 @@ XS(XS_Cwd_sys_abspath)
        if (items < 2)
            dir = NULL;
        else {
-           dir = (char *)SvPV(ST(1),PL_na);
+           dir = (char *)SvPV(ST(1),n_a);
        }
        if (path[0] == '.' && (path[1] == '/' || path[1] == '\\')) {
            path += 2;
@@ -1903,7 +2013,8 @@ XS(XS_Cwd_extLibpath_set)
     if (items < 1 || items > 2)
        croak("Usage: Cwd::extLibpath_set(s, type = 0)");
     {
-       char *  s = (char *)SvPV(ST(0),PL_na);
+       STRLEN n_a;
+       char *  s = (char *)SvPV(ST(0),n_a);
        bool    type;
        U32     rc;
        bool    RETVAL;
@@ -1980,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")) ) {