home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / dmake40.zip / msdos / spawn.c < prev    next >
C/C++ Source or Header  |  1994-10-23  |  13KB  |  424 lines
  1. /* RCS      -- $Header: /u5/dvadura/src/public/dmake/src/msdos/RCS/spawn.c,v 1.1 1994/10/06 17:41:42 dvadura Exp $
  2. -- SYNOPSIS -- spawnvpe code to emulate spawnvpe call common to DOS compilers.
  3. -- 
  4. -- DESCRIPTION
  5. --    This implementation is further integrated into dmake in that it
  6. --    determines the program to execute and if it's extension is either
  7. --    .bat or .ksh it executes it using the appropriate shell based on the
  8. --    setting of .MKSARGS.  If .MKSARGS is set then in addition
  9. --    to the command tail getting built the arguments are also passed in the
  10. --    environment pursuant to the published MKS argument passing conventions.
  11. --    If the variable Swap_on_exec is set and the DOS OS supports it
  12. --    then the dmake executable image is swapped to secondary storage prior
  13. --    to running the child process.  This is requested by setting the
  14. --    appropriate flag in the call to exec.
  15. --
  16. --    This and the exec.asm routine are derived from work that was supplied
  17. --    to me by Kent Williams (williams@umaxc.weeg.uiowa.edu) and by
  18. --      Len Reed, (..!gatech!holos0!lbr or holos0!lbr@gatech.edu., Holos
  19. --    Software, Inc., Tucker, Ga.).  I sincerely acknowledge their help since
  20. --    their Turbo C, and MSC 6.0 code lead directly to this combined
  21. --    swapping exec that hopefully works with either compiler in all memory
  22. --    models.
  23. --
  24. -- AUTHOR
  25. --      Dennis Vadura, dvadura@watdragon.uwaterloo.ca
  26. --      CS DEPT, University of Waterloo, Waterloo, Ont., Canada
  27. --
  28. -- COPYRIGHT
  29. --      Copyright (c) 1992,1994 by Dennis Vadura.  All rights reserved.
  30. -- 
  31. --      This program is free software; you can redistribute it and/or
  32. --      modify it under the terms of the GNU General Public License
  33. --      (version 1), as published by the Free Software Foundation, and
  34. --      found in the file 'LICENSE' included with this distribution.
  35. -- 
  36. --      This program is distributed in the hope that it will be useful,
  37. --      but WITHOUT ANY WARRANTY; without even the implied warrant of
  38. --      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  39. --      GNU General Public License for more details.
  40. -- 
  41. --      You should have received a copy of the GNU General Public License
  42. --      along with this program;  if not, write to the Free Software
  43. --      Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  44. --
  45. -- LOG
  46. --     $Log: spawn.c,v $
  47.  * Revision 1.1  1994/10/06  17:41:42  dvadura
  48.  * dmake Release Version 4.0, Initial revision
  49.  *
  50. */
  51.  
  52. #include <stdio.h>
  53. #include <stdlib.h>
  54.  
  55. #if defined(_MSC_VER) && _MSC_VER >= 600
  56.     /* Ignore the MSC 6.0 library's "const"-riddled prototype
  57.        for spawnvpe.
  58.     */
  59. # define spawnvpe _ignore_msc_spawnvpe
  60. # include <process.h>
  61. # undef spawnvpe
  62.   int spawnvpe(int, char *, char **, char **);
  63. #else
  64. # include <process.h>
  65. #endif
  66.  
  67. #include <dos.h>
  68. #include <errno.h>
  69. #include <string.h>
  70. #include <alloc.h>
  71. #include <fcntl.h>
  72. #include "extern.h"
  73. #include "dosdta.h"
  74. #include "exec.h"
  75. #include "sysintf.h"
  76.  
  77. extern int Interrupted;
  78.  
  79. /* variables and functions local to this file */
  80. static char     *_findexec ANSI((char *, int *));
  81. static char    **_getpath ANSI(());
  82. static char far *_dos_alloc ANSI((uint16));
  83.  
  84. static uint16 _swap_mask;
  85. static int    _mks_args;
  86. static char   dot_com[] = ".COM",
  87.           dot_exe[] = ".EXE",
  88.               dot_bat[] = ".BAT",
  89.           dot_ksh[] = ".KSH";
  90.  
  91. /* Kinds of executables */
  92. #define SCR 1
  93. #define COM 2
  94. #define EXE 4
  95. #define ALL (SCR|COM|EXE)
  96.  
  97. /* How to make a long pointer */
  98. #define CF(x) (char far *)x
  99.  
  100. /* Make sure we know how to get a segment out of a long pointer */
  101. #ifndef FP_SEG
  102. #define FP_SEG(fp)    ((unsigned)((unsigned long)(fp) >> 16))
  103. #endif
  104.  
  105.  
  106. PUBLIC int
  107. spawnvpe(mode, program, av, ep)/*
  108. =================================
  109.    Spawn a process using an environment and a vector of arguments.
  110.    The code computes a new environment, puts the MKS arguments into
  111.    it if need be, and calls the appropriate routines to search the
  112.    path and to invoke the process. */
  113. int  mode;
  114. char *program;
  115. char **av;
  116. char **ep;
  117. {
  118.    char pwd[PATH_MAX+1];
  119.    char **envp = ep;        /* Cause we are going to mess with it. */
  120.    char **argv = av;        /* Same with this one.               */
  121.    char cmdtail[129];
  122.    char far *environment;
  123.    char *tail;
  124.    char *swptmp;
  125.    unsigned int envsize;
  126.    unsigned int cmdsize;
  127.    int  cmdtailen;
  128.    int  i;
  129.    int  doswap;
  130.  
  131.    /* First check to see if we can find the program to execute this way we
  132.     * don't alloc the environment and other such stuff prior to figuring out
  133.     * we don't know how to run the program. */
  134. find_program:
  135.    if((program = _findexec(program, &i)) == NIL(char)) {
  136.       errno = ENOENT;
  137.       return( -1 );
  138.    }
  139.  
  140.    /* i is set to TRUE in _findexec if the exec is a shell
  141.     * script (either .BAT or .KSH file), returns FALSE for all others. */
  142.    if( i && !Packed_shell ) {
  143.       /* Restore the spaces into the command line that were erased by
  144.        * the previous call to Pack_argv.  This enables us to repack the
  145.        * command as a shell command using Pack_argv again. */
  146.       for( i=0; argv[i] != NIL(char); i++ ) {
  147.          int x = strlen(argv[i]);
  148.          if( argv[i+1] != NIL(char) ) argv[i][x] = ' ';
  149.       }
  150.  
  151.       argv = Pack_argv( FALSE, TRUE, *argv );
  152.  
  153.       /* Go and find the program again, I hate goto's but it seems silly to
  154.        * use tail recursion here just for aesthetic purity. */
  155.       program = *argv;
  156.       goto find_program;
  157.    }
  158.  
  159.    /* Compute size of *argv vector for passing as MKS style arguments */
  160.    cmdsize = strlen(*argv)+2;
  161.  
  162.    /* So we have decided on a program to run, therefore pack the command tail
  163.     * and build the environment to pass to the exec code.  This loop packs the
  164.     * DOS command tail, and computes the size of all arguments for the MKS
  165.     * argument passing convention.  Note that we reserve one less byte in the
  166.     * command tail if we are not using MKS style argument passing.
  167.     *
  168.     * Make sure the command tail contains at leat a space.  Some commands fail
  169.     * to work if the command tail is only a \r, STUPID DOS! */
  170.    cmdtailen = ((_mks_args = ((Glob_attr & A_MKSARGS) != 0)) != 0)?3:2;
  171.    tail      = cmdtail+1;
  172.  
  173.    if( argv[1] != NIL(char) )
  174.       for( i=1; argv[i] != NIL(char); i++ ) {
  175.      int arglen = strlen(argv[i]);
  176.  
  177.      cmdsize += arglen+2;        /* Compute all args size for MKS */
  178.  
  179.      if( (cmdtailen += arglen+1) <= 128 ) {
  180.         register char *p = argv[i];
  181.         tail[-1] = ' ';        /* put in the space */
  182.         while( *tail++ = *p++ );    /* put in the arg   */
  183.      }
  184.      else if( !_mks_args ) {
  185.         errno = E2BIG;        /* unless its MKS exit if arglist */
  186.         return(-1);            /* is too long.              */
  187.      }
  188.       }
  189.    else
  190.       *tail++ = ' ';
  191.  
  192.    /* Finish the command tail set up, placing the length in the first byte,
  193.     * and the \r \n \0 at the end for DOS, MKS and us respectively. */
  194.    *cmdtail = tail-cmdtail-2;
  195.    tail[-1] = '\r';
  196.    if( _mks_args ) *tail++ = '\n';
  197.    *tail = '\0';
  198.  
  199.    /* Compute size of environment, skipping any MKS arguments passed in our
  200.     * environment */
  201.    for(; *envp && **envp == '~'; envp++ );
  202.    for(i=0, envsize=_mks_args?cmdsize:1; envp[i] != NIL(char); i++ )
  203.       envsize += strlen(envp[i]) + 1;
  204.  
  205.    /* Check the DOS version number here.  If it is < 3.0 then we don't
  206.     * even want to think about executing the swapping code.   Permanently
  207.     * set swap to 0. */
  208.    doswap = (_osmajor < 3) ? 0 : Swap_on_exec;
  209.  
  210.    /* Set up temporary file for swapping */
  211.    swptmp = doswap?tempnam(NIL(char),"mk"):""; 
  212.  
  213.    /* Allocate an appropriate sized environment block and align it on a
  214.     * paragraph boundary.  It will later get copied to an appropriately low
  215.     * place in the executable image so that when we swap out the environment
  216.     * is still present.  Use
  217.     *    _dos_alloc
  218.     * to allocate the environment segment.  The segment is freed by the call
  219.     * to exec. */
  220.    environment = _dos_alloc( envsize = ((envsize+16)>>4) );
  221.  
  222.    /* First copy the arguments preceeded by ~ character if we are using
  223.     * MKS style argument passing */
  224.    if( _mks_args )
  225.       for(; *argv; argv++) {
  226.          register char *p = *argv;
  227.  
  228.      *environment++ = '~';
  229.      while( *environment++ = *p++ );    /* Far dest, poss near ptr */
  230.       }
  231.  
  232.    /* Now stick in the current evironment vectors. */
  233.    for(; *envp; envp++) {
  234.       register char *p = *envp;
  235.       while( *environment++ = *p++ );        /* Far dest, poss near ptr */
  236.    }
  237.    *environment = '\0';
  238.  
  239.    /* Clear the interrupted flag, and exec  */
  240.    Interrupted = 0;
  241.  
  242.    /* Preserve the current working directory accross a spawn call 
  243.     * DOS is brain dead about this.  This way we have some hope of cleaning
  244.     * up the swapping tempfiles after we return. */
  245.    strcpy(pwd,Get_current_dir());
  246.    i = exec(doswap,CF(program),CF(cmdtail),FP_SEG(environment),CF(swptmp));
  247.    Set_dir(pwd);
  248.  
  249.    /* Now free the temporary file name */
  250.    if( doswap ) FREE(swptmp);
  251.  
  252.    /* If swap was interrupted then quit properly from dmake. */
  253.    if( Interrupted ) Quit();
  254.  
  255.    return(i);
  256. }
  257.  
  258.  
  259. PUBLIC void
  260. Hook_std_writes( file )
  261. char *file;
  262. {
  263.    if( file!= NIL(char) ) {
  264.       int mode = O_BINARY | O_WRONLY | O_CREAT | O_TRUNC;
  265.       int handle;
  266.  
  267.       if (*file == '+') {
  268.           ++file;             /* -F +file means append to file */
  269.           mode = O_BINARY | O_WRONLY | O_CREAT | O_APPEND;
  270.       }
  271.       handle = open(file, mode, S_IREAD | S_IWRITE);
  272.       if (handle < 0) {
  273.           Fatal( "Could not open -F file");
  274.       }
  275.       (void) lseek(handle, 0L, SEEK_END);
  276.       do_hook_std_writes(handle);
  277.    }
  278.    else
  279.       do_unhook_std_writes();
  280. }
  281.  
  282.  
  283. /*
  284. ** _findexec finds executables on the path.
  285. ** Note that it is pretty simple to add support for other executable types
  286. ** shell scripts, etc.
  287. **
  288. ** This follows the command.com behavior very closely.
  289. */
  290. static char *
  291. _findexec( s, is_shell )/*
  292. ==========================
  293.    Cloned closely from code provided by Kent Williams.  Stripped his down to
  294.    a reduced search since dmake doesn't need to recompute the PATH vector
  295.    each time it does the search since it cannot alter the path vector once
  296.    it begins to make recipes.  Also modified it to use findfirst and findnext
  297.    as provided for dirlib package that I got off the net. */
  298. char *s;
  299. int  *is_shell;
  300. {
  301.    unsigned found_flags;
  302.    char     **pathv = NIL(char *);
  303.    char     *ext    = NIL(char);
  304.    char     *buf    = NIL(char);
  305.    char     *p[2];
  306.    char     *dot_scr;
  307.    char        *dot;
  308.  
  309.    p[0] = ""; p[1] = NIL(char);
  310.    if( strchr("./\\", *s) || s[1] == ':' )
  311.       pathv = p;
  312.    else if( (pathv = _getpath()) == NIL(char *) )
  313.       return( NIL(char) );
  314.  
  315.    /* Compute the extension we need if any. */
  316.    if( (dot = strrchr(s,'.')) != NIL(char) &&
  317.         dot > strrchr(s,'/') && dot > strrchr(s,'\\') )
  318.       ext = dot+1;
  319.  
  320.    dot_scr   = _mks_args ? dot_ksh : dot_bat;
  321.    *is_shell = FALSE;
  322.  
  323.    for( found_flags = 0; *pathv && !found_flags; pathv++ ) {
  324.       DTA dta;
  325.  
  326.       if( !ext ) {
  327.      char *name;
  328.      buf = Build_path( *pathv, name=DmStrJoin(s, ".???", -1, FALSE) );
  329.      FREE(name);
  330.       }
  331.       else
  332.      buf = Build_path( *pathv, s );
  333.  
  334.       if( findfirst((char *)strupr(buf), &dta) != NIL(DTA) ) {
  335.      if( !ext ) {
  336.         char *dot;
  337.  
  338.         /* search order is .com .exe (.ksh || .bat)
  339.          * there has to be a '.' */
  340.         do {
  341.            dot = strrchr(dta.name,'.');
  342.            if(0 == strcmp(dot,dot_com))
  343.           found_flags |= COM;
  344.            else if(0 == strcmp(dot,dot_exe))
  345.           found_flags |= EXE;
  346.            else if( 0 == strcmp(dot,dot_scr) )
  347.           found_flags |= SCR;
  348.         } while( found_flags != ALL && findnext(&dta) != NIL(DTA) );
  349.  
  350.         if(found_flags & COM)      ext = dot_com;
  351.         else if(found_flags & EXE) ext = dot_exe;
  352.         else if(found_flags & SCR) {
  353.            ext = dot_scr;
  354.            *is_shell = TRUE;
  355.         }
  356.  
  357.         if( found_flags ) {
  358.            char *name;
  359.            buf = Build_path( *pathv, name=DmStrJoin(s,ext,-1,FALSE) );
  360.            FREE(name);
  361.            strupr(buf);
  362.         }
  363.      }
  364.      else
  365.         found_flags++;
  366.       }
  367.    }
  368.  
  369.    return( found_flags ? buf : NIL(char) );
  370. }
  371.  
  372.  
  373. /*
  374. ** getpath turns the DOS path into a char *vector, It is gotten and
  375. ** transformed only once since dmake can't modify the value of PATH while
  376. ** it is making targets.
  377. */
  378. static char **
  379. _getpath()
  380. {
  381.    static   char **dir = NIL(char *);
  382.    register char *p;
  383.  
  384.    if( !dir ) {
  385.       register char *t;
  386.       int           i;
  387.       char          *semi = NIL(char);
  388.  
  389.       if( (p = getenv("PATH")) == NIL(char) ) p = "";
  390.       for( i=1, t=p; *t; t++ ) if( *t == ';' ) i++;
  391.  
  392.       TALLOC(dir, i+1, char *);
  393.       p   = DmStrDup(p);
  394.  
  395.       for( i=0; p; p = semi ? (semi+1):NIL(char),i++ ){
  396.      if( (semi = strchr(p,';')) != NIL(char) ) *semi = '\0';
  397.      dir[i] = p;
  398.       }
  399.       dir[i]=NIL(char);
  400.    }
  401.  
  402.    return( dir );
  403. }
  404.  
  405.  
  406. static char far *
  407. _dos_alloc( size )/*
  408. ====================
  409.    This routine allocates size paragraphs from DOS.  It changes the memory
  410.    allocation strategy to allocate from the tail and then changes it back.
  411.    to using first fit. */
  412. uint16 size;
  413. {
  414.    union REGS r;
  415.  
  416.    r.h.ah = 0x48;
  417.    r.x.bx = size;
  418.  
  419.    intdos( &r, &r );
  420.    if( r.x.cflag ) No_ram();
  421.    
  422.    return( (char far *) MK_FP(r.x.ax, 0) );
  423. }
  424.