WebSVN – SvarDOS – Diff – /svarcom/trunk/command.c


/* This file is part of the SvarCOM project and is published under the terms
 * of the MIT license.
 *
 * Copyright (C) 2021-2022 Mateusz Viste
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */
 
#include <i86.h>
#include <dos.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
 
#include <process.h>
 
#include "cmd.h"
#include "env.h"
#include "helpers.h"
#include "redir.h"
#include "rmodinit.h"
#include "sayonara.h"
 
#include "rmodcore.h" /* rmod binary inside a BUFFER array */
 
/* this version byte is used to tag RMOD so I can easily make sure that
 * the RMOD struct I find in memory is one that I know. Should the version
 * mismatch, then it would likely mean that SvarCOM has been upgraded and
 * RMOD should not be accessed as its structure might no longer be in sync
 * with what I think it is.
 *          *** INCREMENT THIS AT EACH NEW SVARCOM RELEASE! ***          */
#define BYTE_VERSION 3
 
 
struct config {
  unsigned char flags; /* command.com flags, as defined in rmodinit.h */
  char *execcmd;
  unsigned short envsiz;
};
 
/* max length of the cmdline storage (bytes) - includes also max length of
 * line loaded from a BAT file (no more than 255 bytes!) */
#define CMDLINE_MAXLEN 255
 
 
/* sets guard values at a few places in memory for later detection of
 * overflows via memguard_check() */
static void memguard_set(char *cmdlinebuf) {
  BUFFER[sizeof(BUFFER) - 1] = 0xC7;
  cmdlinebuf[CMDLINE_MAXLEN] = 0xC7;
}
 
 
/* checks for valguards at specific memory locations, returns 0 on success */
static int memguard_check(unsigned short rmodseg, char *cmdlinebuf) {
  /* check RMOD signature (would be overwritten in case of stack overflow */
  static char msg[] = "!! MEMORY CORRUPTION ## DETECTED !!";
  unsigned short far *rmodsig = MK_FP(rmodseg, 0x100 + 6);
  unsigned char far *rmod = MK_FP(rmodseg, 0);
 
  if (*rmodsig != 0x2019) {
    msg[22] = '1';
    goto FAIL;
  }
 
  /* check last BUFFER byte */
  if (BUFFER[sizeof(BUFFER) - 1] != 0xC7) {
    msg[22] = '2';
    goto FAIL;
  }
 
  /* check last cmdlinebuf byte */
  if (cmdlinebuf[CMDLINE_MAXLEN] != 0xC7) {
    msg[22] = '3';
    goto FAIL;
  }
 
  /* check rmod exec buf */
  if (rmod[RMOD_OFFSET_EXECPROG + 127] != 0) {
    msg[22] = '4';
    goto FAIL;
  }
 
  /* check rmod exec stdin buf */
  if (rmod[RMOD_OFFSET_STDINFILE + 127] != 0) {
    msg[22] = '5';
    goto FAIL;
  }
 
  /* check rmod exec stdout buf */
  if (rmod[RMOD_OFFSET_STDOUTFILE + 127] != 0) {
    msg[22] = '6';
    goto FAIL;
  }
 
  /* else all good */
  return(0);
 
  /* error handling */
  FAIL:
  outputnl(msg);
  return(1);
}
 
 
/* parses command line the hard way (directly from PSP) */
static void parse_argv(struct config *cfg) {
  const unsigned char *cmdlinelen = (void *)0x80;
  char *cmdline = (void *)0x81;
 
  memset(cfg, 0, sizeof(*cfg));
 
  /* set a NULL terminator on cmdline */
  cmdline[*cmdlinelen] = 0;
 
  while (*cmdline != 0) {
 
    /* skip over any leading spaces */
    if (*cmdline == ' ') {
      cmdline++;
      continue;
    }
 
    if (*cmdline != '/') {
      output("Invalid parameter: ");
      outputnl(cmdline);
      goto SKIP_TO_NEXT_ARG;
    }
 
    /* got a slash */
    cmdline++;  /* skip the slash */
    switch (*cmdline) {
      case 'c': /* /C = execute command and quit */
      case 'C':
        cfg->flags |= FLAG_EXEC_AND_QUIT;
        /* FALLTHRU */
      case 'k': /* /K = execute command and keep running */
      case 'K':
        cfg->execcmd = cmdline + 1;
        return;
 
      case 'd': /* /D = skip autoexec.bat processing */
      case 'D':
        cfg->flags |= FLAG_SKIP_AUTOEXEC;
        break;
 
      case 'e': /* preset the initial size of the environment block */
      case 'E':
        cmdline++;
        if (*cmdline == ':') cmdline++; /* could be /E:size */
        atous(&(cfg->envsiz), cmdline);
        if (cfg->envsiz < 64) cfg->envsiz = 0;
        break;
 
      case 'p': /* permanent shell (can't exit + run autoexec.bat) */
      case 'P':
        cfg->flags |= FLAG_PERMANENT;
        break;
 
      case '?':
        outputnl("Starts the SvarCOM command interpreter");
        outputnl("");
        outputnl("COMMAND /E:nnn [/[C|K] [/P] [/D] command]");
        outputnl("");
        outputnl("/D      Skip AUTOEXEC.BAT processing (makes sense only with /P)");
        outputnl("/E:nnn  Sets the environment size to nnn bytes");
        outputnl("/P      Makes the new command interpreter permanent and run AUTOEXEC.BAT");
        outputnl("/C      Executes the specified command and returns");
        outputnl("/K      Executes the specified command and continues running");
        exit(1);
        break;
 
      default:
        output("Invalid switch: /");
        outputnl(cmdline);
        break;
    }
 
    /* move to next argument or quit processing if end of cmdline */
    SKIP_TO_NEXT_ARG:
    while ((*cmdline != 0) && (*cmdline != ' ') && (*cmdline != '/')) cmdline++;
  }
}
 
 
/* builds the prompt string and displays it. buff is filled with a zero-terminated copy of the prompt. */
static void build_and_display_prompt(char *buff, unsigned short envseg) {
  char *s = buff;
  /* locate the prompt variable or use the default pattern */
  const char far *fmt = env_lookup_val(envseg, "PROMPT");
  if ((fmt == NULL) || (*fmt == 0)) fmt = "$p$g"; /* fallback to default if empty */
  /* build the prompt string based on pattern */
  for (; *fmt != 0; fmt++) {
    if (*fmt != '$') {
      *s = *fmt;
      s++;
      continue;
    }
    /* escape code ($P, etc) */
    fmt++;
    switch (*fmt) {
      case 'Q':  /* $Q = = (equal sign) */
      case 'q':
        *s = '=';
        s++;
        break;
      case '$':  /* $$ = $ (dollar sign) */
        *s = '$';
        s++;
        break;
      case 'T':  /* $t = current time */
      case 't':
        s += sprintf(s, "00:00"); /* TODO */
        break;
      case 'D':  /* $D = current date */
      case 'd':
        s += sprintf(s, "1985-07-29"); /* TODO */
        break;
      case 'P':  /* $P = current drive and path */
      case 'p':
        _asm {
          mov ah, 0x19    /* DOS 1+ - GET CURRENT DRIVE */
          int 0x21
          mov bx, s
          mov [bx], al  /* AL = drive (00 = A:, 01 = B:, etc */
        }
        *s += 'A';
        s++;
        *s = ':';
        s++;
        *s = '\\';
        s++;
        _asm {
          mov ah, 0x47    /* DOS 2+ - CWD - GET CURRENT DIRECTORY */
          xor dl,dl       /* DL = drive number (00h = default, 01h = A:, etc) */
          mov si, s       /* DS:SI -> 64-byte buffer for ASCIZ pathname */
          int 0x21
          jc DONE         /* leave path empty on error */
          /* move s ptr forward to end (0-termintor) of pathname */
          NEXTBYTE:
          mov si, s
          cmp byte ptr [si], 0
          je DONE
          inc s
          jmp NEXTBYTE
          DONE:
        }
        break;
      case 'V':  /* $V = DOS version number */
      case 'v':
        s += sprintf(s, "VER"); /* TODO */
        break;
      case 'N':  /* $N = current drive */
      case 'n':
        _asm {
          mov ah, 0x19    /* DOS 1+ - GET CURRENT DRIVE */
          int 0x21
          mov bx, s
          mov [bx], al  /* AL = drive (00 = A:, 01 = B:, etc */
        }
        *s += 'A';
        s++;
        break;
      case 'G':  /* $G = > (greater-than sign) */
      case 'g':
        *s = '>';
        s++;
        break;
      case 'L':  /* $L = < (less-than sign) */
      case 'l':
        *s = '<';
        s++;
        break;
      case 'B':  /* $B = | (pipe) */
      case 'b':
        *s = '|';
        s++;
        break;
      case 'H':  /* $H = backspace (erases previous character) */
      case 'h':
        *s = '\b';
        s++;
        break;
      case 'E':  /* $E = Escape code (ASCII 27) */
      case 'e':
        *s = 27;
        s++;
        break;
      case '_':  /* $_ = CR+LF */
        *s = '\r';
        s++;
        *s = '\n';
        s++;
        break;
    }
  }
  *s = 0;
  output(buff);
}
 
 
static void run_as_external(char *buff, const char *cmdline, unsigned short envseg, struct rmod_props far *rmod, struct redir_data *redir) {
  char *cmdfile = buff + 512;
  const char far *pathptr;
  int lookup;
  unsigned short i;
  const char *ext;
  char *cmd = buff + 1024;
  const char *cmdtail;
  char far *rmod_execprog = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPROG);
  char far *rmod_cmdtail = MK_FP(rmod->rmodseg, 0x81);
  _Packed struct {
    unsigned short envseg;
    unsigned long cmdtail;
    unsigned long fcb1;
    unsigned long fcb2;
  } far *ExecParam = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPARAM);
 
  /* find cmd and cmdtail */
  i = 0;
  cmdtail = cmdline;
  while (*cmdtail == ' ') cmdtail++; /* skip any leading spaces */
  while ((*cmdtail != ' ') && (*cmdtail != '/') && (*cmdtail != '+') && (*cmdtail != 0)) {
    cmd[i++] = *cmdtail;
    cmdtail++;
  }
  cmd[i] = 0;
 
  /* is this a command in curdir? */
  lookup = lookup_cmd(cmdfile, cmd, NULL, &ext);
  if (lookup == 0) {
    /* printf("FOUND LOCAL EXEC FILE: '%s'\r\n", cmdfile); */
    goto RUNCMDFILE;
  } else if (lookup == -2) {
    /* puts("NOT FOUND"); */
    return;
  }
 
  /* try matching something in PATH */
  pathptr = env_lookup_val(envseg, "PATH");
 
  /* try each path in %PATH% */
  while (pathptr) {
    for (i = 0;; i++) {
      buff[i] = *pathptr;
      if ((buff[i] == 0) || (buff[i] == ';')) break;
      pathptr++;
    }
    buff[i] = 0;
    lookup = lookup_cmd(cmdfile, cmd, buff, &ext);
    if (lookup == 0) goto RUNCMDFILE;
    if (lookup == -2) return;
    if (*pathptr == ';') {
      pathptr++;
    } else {
      break;
    }
  }
 
  /* last chance: is it an executable link? (trim extension from cmd first) */
  for (i = 0; (cmd[i] != 0) && (cmd[i] != '.') && (i < 9); i++) buff[128 + i] = cmd[i];
  buff[128 + i] = 0;
  if ((i < 9) && (link_computefname(buff, buff + 128, envseg) == 0)) {
    /* try opening the link file (if it exists) and read it into buff */
    i = 0;
    _asm {
      push ax
      push bx
      push cx
      push dx
 
      mov ax, 0x3d00  /* DOS 2+ - OPEN EXISTING FILE, READ-ONLY */
      mov dx, buff    /* file name */
      int 0x21
      jc ERR_FOPEN
      /* file handle in AX, read from file now */
      mov bx, ax      /* file handle */
      mov ah, 0x3f    /* Read from file via handle bx */
      mov cx, 128     /* up to 128 bytes */
      /* mov dx, buff */ /* dest buffer (already set) */
      int 0x21        /* read up to 256 bytes from file and write to buff */
      jc ERR_READ
      mov i, ax
      ERR_READ:
      mov ah, 0x3e    /* close file handle in BX */
      int 0x21
      ERR_FOPEN:
 
      pop dx
      pop cx
      pop bx
      pop ax
    }
 
    /* did I read anything? */
    if (i != 0) {
      buff[i] = 0;
      /* trim buff at first \n or \r, just in case someone fiddled with the
       * link file using a text editor */
      for (i = 0; (buff[i] != 0) && (buff[i] != '\r') && (buff[i] != '\n'); i++);
      buff[i] = 0;
      /* lookup check */
      if (buff[0] != 0) {
        lookup = lookup_cmd(cmdfile, cmd, buff, &ext);
        if (lookup == 0) goto RUNCMDFILE;
      }
    }
  }
 
  /* all failed (ie. executable file not found) */
  return;
 
  RUNCMDFILE:
 
  /* special handling of batch files */
  if ((ext != NULL) && (imatch(ext, "bat"))) {
    /* copy truename of the bat file to rmod buff */
    _fstrcpy(rmod->batfile, cmdfile);
 
    /* explode args of the bat file and store them in rmod buff */
    cmd_explode(buff, cmdline, NULL);
    _fmemcpy(rmod->batargv, buff, sizeof(rmod->batargv));
 
    /* reset the 'next line to execute' counter */
    rmod->batnextline = 0;
    /* remember the echo flag (in case bat file disables echo) */
    rmod->flags &= ~FLAG_ECHO_BEFORE_BAT;
    if (rmod->flags & FLAG_ECHOFLAG) rmod->flags |= FLAG_ECHO_BEFORE_BAT;
    return;
  }
 
  /* copy full filename to execute, along with redirected files (if any) */
  _fstrcpy(rmod_execprog, cmdfile);
 
  /* copy stdin file if a redirection is needed */
  if (redir->stdinfile) {
    char far *farptr = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDINFILE);
    _fstrcpy(farptr, redir->stdinfile);
  }
 
  /* same for stdout file */
  if (redir->stdoutfile) {
    char far *farptr = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDOUTFILE);
    unsigned short far *farptr16 = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDOUTAPP);
    _fstrcpy(farptr, redir->stdoutfile);
    /* openflag */
    *farptr16 = redir->stdout_openflag;
  }
 
  /* copy cmdtail to rmod's PSP and compute its len */
  for (i = 0; cmdtail[i] != 0; i++) rmod_cmdtail[i] = cmdtail[i];
  rmod_cmdtail[i] = '\r';
  rmod_cmdtail[-1] = i;
 
  /* set up rmod to execute the command */
 
  ExecParam->envseg = envseg;
  ExecParam->cmdtail = (unsigned long)MK_FP(rmod->rmodseg, 0x80); /* farptr, must be in PSP format (lenbyte args \r) */
  ExecParam->fcb1 = 0; /* TODO farptr */
  ExecParam->fcb2 = 0; /* TODO farptr */
  exit(0); /* let rmod do the job now */
}
 
 
static void set_comspec_to_self(unsigned short envseg) {
  unsigned short *psp_envseg = (void *)(0x2c); /* pointer to my env segment field in the PSP */
  char far *myenv = MK_FP(*psp_envseg, 0);
  unsigned short varcount;
  char buff[256] = "COMSPEC=";
  char *buffptr = buff + 8;
  /* who am i? look into my own environment, at the end of it should be my EXEPATH string */
  while (*myenv != 0) {
    /* consume a NULL-terminated string */
    while (*myenv != 0) myenv++;
    /* move to next string */
    myenv++;
  }
  /* get next word, if 1 then EXEPATH follows */
  myenv++;
  varcount = *myenv;
  myenv++;
  varcount |= (*myenv << 8);
  myenv++;
  if (varcount != 1) return; /* NO EXEPATH FOUND */
  while (*myenv != 0) {
    *buffptr = *myenv;
    buffptr++;
    myenv++;
  }
  *buffptr = 0;
  /* printf("EXEPATH: '%s'\r\n", buff); */
  env_setvar(envseg, buff);
}
 
 
/* wait for user input */
static void cmdline_getinput(unsigned short inpseg, unsigned short inpoff) {
  _asm {
    push ax
    push bx
    push cx
    push dx
    push ds
 
    /* is DOSKEY support present? (INT 2Fh, AX=4800h, returns non-zero in AL if present) */
    mov ax, 0x4800
    int 0x2f
    mov bl, al /* save doskey status in BL */
 
    /* set up buffered input to inpseg:inpoff */
    mov ax, inpseg
    push ax
    pop ds
    mov dx, inpoff
 
    /* execute either DOS input or DOSKEY */
    test bl, bl /* zf set if no DOSKEY present */
    jnz DOSKEY
 
    mov ah, 0x0a
    int 0x21
    jmp short DONE
 
    DOSKEY:
    mov ax, 0x4810
    int 0x2f
 
    DONE:
    /* terminate command with a CR/LF */
    mov ah, 0x02 /* display character in dl */
    mov dl, 0x0d
    int 0x21
    mov dl, 0x0a
    int 0x21
 
    pop ds
    pop dx
    pop cx
    pop bx
    pop ax
  }
}
 
 
/* fetches a line from batch file and write it to buff (NULL-terminated),
 * increments rmod counter and returns 0 on success. */
static int getbatcmd(char *buff, unsigned char buffmaxlen, struct rmod_props far *rmod) {
  unsigned short i;
  unsigned short batname_seg = FP_SEG(rmod->batfile);
  unsigned short batname_off = FP_OFF(rmod->batfile);
  unsigned short filepos_cx = rmod->batnextline >> 16;
  unsigned short filepos_dx = rmod->batnextline & 0xffff;
  unsigned char blen = 0;
  unsigned short errv = 0;
 
  /* open file, jump to offset filpos, and read data into buff.
   * result in blen (unchanged if EOF or failure). */
  _asm {
    push ax
    push bx
    push cx
    push dx
 
    /* open file (read-only) */
    mov bx, 0xffff        /* preset BX to 0xffff to detect error conditions */
    mov dx, batname_off
    mov ax, batname_seg
    push ds     /* save DS */
    mov ds, ax
    mov ax, 0x3d00
    int 0x21    /* handle in ax on success */
    pop ds      /* restore DS */
    jc ERR
    mov bx, ax  /* save handle to bx */
 
    /* jump to file offset CX:DX */
    mov ax, 0x4200
    mov cx, filepos_cx
    mov dx, filepos_dx
    int 0x21  /* CF clear on success, DX:AX set to cur pos */
    jc ERR
 
    /* read the line into buff */
    mov ah, 0x3f
    xor ch, ch
    mov cl, buffmaxlen
    mov dx, buff
    int 0x21 /* CF clear on success, AX=number of bytes read */
    jc ERR
    mov blen, al
    jmp CLOSEANDQUIT
 
    ERR:
    mov errv, ax
 
    CLOSEANDQUIT:
    /* close file (if bx contains a handle) */
    cmp bx, 0xffff
    je DONE
    mov ah, 0x3e
    int 0x21
 
    DONE:
    pop dx
    pop cx
    pop bx
    pop ax
  }
 
  /* printf("blen=%u filepos_cx=%u filepos_dx=%u\r\n", blen, filepos_cx, filepos_dx); */
 
  if (errv != 0) nls_outputnl_doserr(errv);
 
  /* on EOF - abort processing the bat file */
  if (blen == 0) goto OOPS;
 
  /* find nearest \n to inc batch offset and replace \r by NULL terminator
   * I support all CR/LF, CR- and LF-terminated batch files */
  for (i = 0; i < blen; i++) {
    if ((buff[i] == '\r') || (buff[i] == '\n')) {
      if ((buff[i] == '\r') && ((i+1) < blen) && (buff[i+1] == '\n')) rmod->batnextline += 1;
      break;
    }
  }
  buff[i] = 0;
  rmod->batnextline += i + 1;
 
  return(0);
 
  OOPS:
  rmod->batfile[0] = 0;
  rmod->batnextline = 0;
  return(-1);
}
 
 
/* replaces %-variables in a BAT line with resolved values:
 * %PATH%       -> replaced by the contend of the PATH env variable
 * %UNDEFINED%  -> undefined variables are replaced by nothing ("")
 * %NOTCLOSED   -> NOTCLOSED
 * %1           -> first argument of the batch file (or nothing if no arg) */
static void batpercrepl(char *res, unsigned short ressz, const char *line, const struct rmod_props far *rmod, unsigned short envseg) {
  unsigned short lastperc = 0xffff;
  unsigned short reslen = 0;
 
  if (ressz == 0) return;
  ressz--; /* reserve one byte for the NULL terminator */
 
  for (; (reslen < ressz) && (*line != 0); line++) {
    /* if not a percent, I don't care */
    if (*line != '%') {
      res[reslen++] = *line;
      continue;
    }
 
    /* *** perc char handling *** */
 
    /* closing perc? */
    if (lastperc != 0xffff) {
      /* %% is '%' */
      if (lastperc == reslen) {
        res[reslen++] = '%';
      } else {   /* otherwise variable name */
        const char far *ptr;
        res[reslen] = 0;
        reslen = lastperc;
        ptr = env_lookup_val(envseg, res + reslen);
        if (ptr != NULL) {
          while ((*ptr != 0) && (reslen < ressz)) {
            res[reslen++] = *ptr;
            ptr++;
          }
        }
      }
      lastperc = 0xffff;
      continue;
    }
 
    /* digit? (bat arg) */
    if ((line[1] >= '0') && (line[1] <= '9')) {
      unsigned short argid = line[1] - '0';
      unsigned short i;
      const char far *argv = rmod->batargv;
 
      /* locate the proper arg */
      for (i = 0; i != argid; i++) {
        /* if string is 0, then end of list reached */
        if (*argv == 0) break;
        /* jump to next arg */
        while (*argv != 0) argv++;
        argv++;
      }
 
      /* copy the arg to result */
      for (i = 0; (argv[i] != 0) && (reslen < ressz); i++) {
        res[reslen++] = argv[i];
      }
      line++;  /* skip the digit */
      continue;
    }
 
    /* opening perc */
    lastperc = reslen;
 
  }
 
  res[reslen] = 0;
}
 
 
int main(void) {
  static struct config cfg;
  static unsigned short far *rmod_envseg;
  static unsigned short far *lastexitcode;
  static struct rmod_props far *rmod;
  static char cmdlinebuf[CMDLINE_MAXLEN + 2]; /* 1 extra byte for 0-terminator and another for memguard */
  static char *cmdline;
  static struct redir_data redirprops;
  static enum cmd_result cmdres;
  static unsigned short i; /* general-purpose variable for short-lived things */
 
  rmod = rmod_find(BUFFER_len);
  if (rmod == NULL) {
    /* look at command line parameters (in case env size if set there) */
    parse_argv(&cfg);
    rmod = rmod_install(cfg.envsiz, BUFFER, BUFFER_len);
    if (rmod == NULL) {
      outputnl("ERROR: rmod_install() failed");
      return(1);
    }
    /* copy flags to rmod's storage (and enable ECHO) */
    rmod->flags = cfg.flags | FLAG_ECHOFLAG;
    /* printf("rmod installed at %Fp\r\n", rmod); */
    rmod->version = BYTE_VERSION;
  } else {
    /* printf("rmod found at %Fp\r\n", rmod); */
    /* if I was spawned by rmod and FLAG_EXEC_AND_QUIT is set, then I should
     * die asap, because the command has been executed already, so I no longer
     * have a purpose in life */
    if (rmod->flags & FLAG_EXEC_AND_QUIT) sayonara(rmod);
    /* */
    if (rmod->version != BYTE_VERSION) {
      outputnl("SVARCOM VERSION CHANGED. SYSTEM HALTED. PLEASE REBOOT YOUR COMPUTER.");
      _asm {
        HALT:
        hlt
        jmp HALT
      }
    }
  }
 
  /* install a few guardvals in memory to detect some cases of overflows */
  memguard_set(cmdlinebuf);
 
  rmod_envseg = MK_FP(rmod->rmodseg, RMOD_OFFSET_ENVSEG);
  lastexitcode = MK_FP(rmod->rmodseg, RMOD_OFFSET_LEXITCODE);
 
  /* make COMPSEC point to myself */
  set_comspec_to_self(*rmod_envseg);
 
/*  {
    unsigned short envsiz;
    unsigned short far *sizptr = MK_FP(*rmod_envseg - 1, 3);
    envsiz = *sizptr;
    envsiz *= 16;
    printf("rmod_inpbuff at %04X:%04X, env_seg at %04X:0000 (env_size = %u bytes)\r\n", rmod->rmodseg, RMOD_OFFSET_INPBUFF, *rmod_envseg, envsiz);
  }*/
 
  /* on /P check for the presence of AUTOEXEC.BAT and execute it if found,
   * but skip this check if /D was also passed */
  if ((cfg.flags & (FLAG_PERMANENT | FLAG_SKIP_AUTOEXEC)) == FLAG_PERMANENT) {
    if (file_getattr("AUTOEXEC.BAT") >= 0) cfg.execcmd = "AUTOEXEC.BAT";
  }
 
  do {
    /* terminate previous command with a CR/LF if ECHO ON (but not during BAT processing) */
    if ((rmod->flags & FLAG_ECHOFLAG) && (rmod->batfile[0] == 0)) outputnl("");
 
    SKIP_NEWLINE:
 
    /* cancel any redirections that may have been set up before */
    redir_revert();
 
    /* memory check */
    memguard_check(rmod->rmodseg, cmdlinebuf);
 
    /* preset cmdline to point at the dedicated buffer */
    cmdline = cmdlinebuf;
 
    /* (re)load translation strings if needed */
    nls_langreload(BUFFER, *rmod_envseg);
 
    /* load awaiting command, if any (used to run piped commands) */
    if (rmod->awaitingcmd[0] != 0) {
      _fstrcpy(cmdline, rmod->awaitingcmd);
      rmod->awaitingcmd[0] = 0;
      goto EXEC_CMDLINE;
    }
 
    /* skip user input if I have a command to exec (/C or /K) */
    if (cfg.execcmd != NULL) {
      cmdline = cfg.execcmd;
      cfg.execcmd = NULL;
      goto EXEC_CMDLINE;
    }
 
    /* if batch file is being executed -> fetch next line */
    if (rmod->batfile[0] != 0) {
      if (getbatcmd(BUFFER, CMDLINE_MAXLEN, rmod) != 0) { /* end of batch */
        /* restore echo flag as it was before running the bat file */
        rmod->flags &= ~FLAG_ECHOFLAG;
        if (rmod->flags & FLAG_ECHO_BEFORE_BAT) rmod->flags |= FLAG_ECHOFLAG;
        continue;
      }
      /* %-decoding of variables (%PATH%, %1, %%...), result in cmdline */
      batpercrepl(cmdline, CMDLINE_MAXLEN, BUFFER, rmod, *rmod_envseg);
      /* skip any leading spaces */
      while (*cmdline == ' ') cmdline++;
      /* output prompt and command on screen if echo on and command is not
       * inhibiting it with the @ prefix */
      if ((rmod->flags & FLAG_ECHOFLAG) && (cmdline[0] != '@')) {
        build_and_display_prompt(BUFFER, *rmod_envseg);
        outputnl(cmdline);
      }
      /* skip the @ prefix if present, it is no longer useful */
      if (cmdline[0] == '@') cmdline++;
    } else {
      /* interactive mode: display prompt (if echo enabled) and wait for user
       * command line */
      if (rmod->flags & FLAG_ECHOFLAG) build_and_display_prompt(BUFFER, *rmod_envseg);
      /* collect user input */
      cmdline_getinput(FP_SEG(rmod->inputbuf), FP_OFF(rmod->inputbuf));
      /* copy it to local cmdline */
      if (rmod->inputbuf[1] != 0) _fmemcpy(cmdline, rmod->inputbuf + 2, rmod->inputbuf[1]);
      cmdline[(unsigned)(rmod->inputbuf[1])] = 0; /* zero-terminate local buff (oriignal is '\r'-terminated) */
    }
 
    /* if nothing entered, loop again (but without appending an extra CR/LF) */
    if (cmdline[0] == 0) goto SKIP_NEWLINE;
 
    /* I jump here when I need to exec an initial command (/C or /K) */
    EXEC_CMDLINE:
 
    /* move pointer forward to skip over any leading spaces */
    while (*cmdline == ' ') cmdline++;
 
    /* update rmod's ptr to COMPSPEC so it is always up to date */
    rmod_updatecomspecptr(rmod->rmodseg, *rmod_envseg);
 
    /* handle redirections (if any) */
    i = redir_parsecmd(&redirprops, cmdline, rmod->awaitingcmd);
    if (i != 0) {
      nls_outputnl_doserr(i);
      rmod->awaitingcmd[0] = 0;
      continue;
    }
 
    /* try matching (and executing) an internal command */
    cmdres = cmd_process(rmod, *rmod_envseg, cmdline, BUFFER, sizeof(BUFFER), &redirprops);
    if ((cmdres == CMD_OK) || (cmdres == CMD_FAIL)) {
      /* internal command executed */
      continue;
    } else if (cmdres == CMD_CHANGED) { /* cmdline changed, needs to be reprocessed */
      goto EXEC_CMDLINE;
    } else if (cmdres == CMD_NOTFOUND) {
      /* this was not an internal command, try matching an external command */
      run_as_external(BUFFER, cmdline, *rmod_envseg, rmod, &redirprops);
      /* perhaps this is a newly launched BAT file */
      if ((rmod->batfile[0] != 0) && (rmod->batnextline == 0)) goto SKIP_NEWLINE;
      /* run_as_external() does not return on success, if I am still alive then
       * external command failed to execute */
      outputnl("Bad command or file name");
      continue;
    }
 
    /* I should never ever land here */
    outputnl("INTERNAL ERR: INVALID CMDRES");
 
  } while ((rmod->flags & FLAG_EXEC_AND_QUIT) == 0);
 
  sayonara(rmod);
  return(0);
}
 

Rev 571	Rev 572
1	`/* This file is part of the SvarCOM project and is published under the terms`	1	`/* This file is part of the SvarCOM project and is published under the terms`
2	`* of the MIT license.`	2	`* of the MIT license.`
3	`*`	3	`*`
4	`* Copyright (C) 2021-2022 Mateusz Viste`	4	`* Copyright (C) 2021-2022 Mateusz Viste`
5	`*`	5	`*`
6	`* Permission is hereby granted, free of charge, to any person obtaining a`	6	`* Permission is hereby granted, free of charge, to any person obtaining a`
7	`* copy of this software and associated documentation files (the "Software"),`	7	`* copy of this software and associated documentation files (the "Software"),`
8	`* to deal in the Software without restriction, including without limitation`	8	`* to deal in the Software without restriction, including without limitation`
9	`* the rights to use, copy, modify, merge, publish, distribute, sublicense,`	9	`* the rights to use, copy, modify, merge, publish, distribute, sublicense,`
10	`* and/or sell copies of the Software, and to permit persons to whom the`	10	`* and/or sell copies of the Software, and to permit persons to whom the`
11	`* Software is furnished to do so, subject to the following conditions:`	11	`* Software is furnished to do so, subject to the following conditions:`
12	`*`	12	`*`
13	`* The above copyright notice and this permission notice shall be included in`	13	`* The above copyright notice and this permission notice shall be included in`
14	`* all copies or substantial portions of the Software.`	14	`* all copies or substantial portions of the Software.`
15	`*`	15	`*`
16	`* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR`	16	`* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR`
17	`* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,`	17	`* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,`
18	`* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE`	18	`* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE`
19	`* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER`	19	`* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER`
20	`* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING`	20	`* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING`
21	`* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER`	21	`* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER`
22	`* DEALINGS IN THE SOFTWARE.`	22	`* DEALINGS IN THE SOFTWARE.`
23	`*/`	23	`*/`
24		24
25	`#include <i86.h>`	25	`#include <i86.h>`
26	`#include <dos.h>`	26	`#include <dos.h>`
27	`#include <stdio.h>`	27	`#include <stdio.h>`
28	`#include <stdlib.h>`	28	`#include <stdlib.h>`
29	`#include <string.h>`	29	`#include <string.h>`
30		30
31	`#include <process.h>`	31	`#include <process.h>`
32		32
33	`#include "cmd.h"`	33	`#include "cmd.h"`
34	`#include "env.h"`	34	`#include "env.h"`
35	`#include "helpers.h"`	35	`#include "helpers.h"`
36	`#include "redir.h"`	36	`#include "redir.h"`
37	`#include "rmodinit.h"`	37	`#include "rmodinit.h"`
38	`#include "sayonara.h"`	38	`#include "sayonara.h"`
39		39
40	`#include "rmodcore.h" /* rmod binary inside a BUFFER array */`	40	`#include "rmodcore.h" /* rmod binary inside a BUFFER array */`
41		41
-		42	`/* this version byte is used to tag RMOD so I can easily make sure that`
-		43	`* the RMOD struct I find in memory is one that I know. Should the version`
-		44	`* mismatch, then it would likely mean that SvarCOM has been upgraded and`
-		45	`* RMOD should not be accessed as its structure might no longer be in sync`
-		46	`* with what I think it is.`
-		47	`* * INCREMENT THIS AT EACH NEW SVARCOM RELEASE! * */`
-		48	`#define BYTE_VERSION 3`
-		49
-		50
42	`struct config {`	51	`struct config {`
43	`unsigned char flags; /* command.com flags, as defined in rmodinit.h */`	52	`unsigned char flags; /* command.com flags, as defined in rmodinit.h */`
44	`char *execcmd;`	53	`char *execcmd;`
45	`unsigned short envsiz;`	54	`unsigned short envsiz;`
46	`};`	55	`};`
47		56
48	`/* max length of the cmdline storage (bytes) - includes also max length of`	57	`/* max length of the cmdline storage (bytes) - includes also max length of`
49	`* line loaded from a BAT file (no more than 255 bytes!) */`	58	`* line loaded from a BAT file (no more than 255 bytes!) */`
50	`#define CMDLINE_MAXLEN 255`	59	`#define CMDLINE_MAXLEN 255`
51		60
52		61
53	`/* sets guard values at a few places in memory for later detection of`	62	`/* sets guard values at a few places in memory for later detection of`
54	`* overflows via memguard_check() */`	63	`* overflows via memguard_check() */`
55	`static void memguard_set(char *cmdlinebuf) {`	64	`static void memguard_set(char *cmdlinebuf) {`
56	`BUFFER[sizeof(BUFFER) - 1] = 0xC7;`	65	`BUFFER[sizeof(BUFFER) - 1] = 0xC7;`
57	`cmdlinebuf[CMDLINE_MAXLEN] = 0xC7;`	66	`cmdlinebuf[CMDLINE_MAXLEN] = 0xC7;`
58	`}`	67	`}`
59		68
60		69
61	`/* checks for valguards at specific memory locations, returns 0 on success */`	70	`/* checks for valguards at specific memory locations, returns 0 on success */`
62	`static int memguard_check(unsigned short rmodseg, char *cmdlinebuf) {`	71	`static int memguard_check(unsigned short rmodseg, char *cmdlinebuf) {`
63	`/* check RMOD signature (would be overwritten in case of stack overflow */`	72	`/* check RMOD signature (would be overwritten in case of stack overflow */`
64	`static char msg[] = "!! MEMORY CORRUPTION ## DETECTED !!";`	73	`static char msg[] = "!! MEMORY CORRUPTION ## DETECTED !!";`
65	`unsigned short far *rmodsig = MK_FP(rmodseg, 0x100 + 6);`	74	`unsigned short far *rmodsig = MK_FP(rmodseg, 0x100 + 6);`
66	`unsigned char far *rmod = MK_FP(rmodseg, 0);`	75	`unsigned char far *rmod = MK_FP(rmodseg, 0);`
67		76
68	`if (*rmodsig != 0x2019) {`	77	`if (*rmodsig != 0x2019) {`
69	`msg[22] = '1';`	78	`msg[22] = '1';`
70	`goto FAIL;`	79	`goto FAIL;`
71	`}`	80	`}`
72		81
73	`/* check last BUFFER byte */`	82	`/* check last BUFFER byte */`
74	`if (BUFFER[sizeof(BUFFER) - 1] != 0xC7) {`	83	`if (BUFFER[sizeof(BUFFER) - 1] != 0xC7) {`
75	`msg[22] = '2';`	84	`msg[22] = '2';`
76	`goto FAIL;`	85	`goto FAIL;`
77	`}`	86	`}`
78		87
79	`/* check last cmdlinebuf byte */`	88	`/* check last cmdlinebuf byte */`
80	`if (cmdlinebuf[CMDLINE_MAXLEN] != 0xC7) {`	89	`if (cmdlinebuf[CMDLINE_MAXLEN] != 0xC7) {`
81	`msg[22] = '3';`	90	`msg[22] = '3';`
82	`goto FAIL;`	91	`goto FAIL;`
83	`}`	92	`}`
84		93
85	`/* check rmod exec buf */`	94	`/* check rmod exec buf */`
86	`if (rmod[RMOD_OFFSET_EXECPROG + 127] != 0) {`	95	`if (rmod[RMOD_OFFSET_EXECPROG + 127] != 0) {`
87	`msg[22] = '4';`	96	`msg[22] = '4';`
88	`goto FAIL;`	97	`goto FAIL;`
89	`}`	98	`}`
90		99
91	`/* check rmod exec stdin buf */`	100	`/* check rmod exec stdin buf */`
92	`if (rmod[RMOD_OFFSET_STDINFILE + 127] != 0) {`	101	`if (rmod[RMOD_OFFSET_STDINFILE + 127] != 0) {`
93	`msg[22] = '5';`	102	`msg[22] = '5';`
94	`goto FAIL;`	103	`goto FAIL;`
95	`}`	104	`}`
96		105
97	`/* check rmod exec stdout buf */`	106	`/* check rmod exec stdout buf */`
98	`if (rmod[RMOD_OFFSET_STDOUTFILE + 127] != 0) {`	107	`if (rmod[RMOD_OFFSET_STDOUTFILE + 127] != 0) {`
99	`msg[22] = '6';`	108	`msg[22] = '6';`
100	`goto FAIL;`	109	`goto FAIL;`
101	`}`	110	`}`
102		111
103	`/* else all good */`	112	`/* else all good */`
104	`return(0);`	113	`return(0);`
105		114
106	`/* error handling */`	115	`/* error handling */`
107	`FAIL:`	116	`FAIL:`
108	`outputnl(msg);`	117	`outputnl(msg);`
109	`return(1);`	118	`return(1);`
110	`}`	119	`}`
111		120
112		121
113	`/* parses command line the hard way (directly from PSP) */`	122	`/* parses command line the hard way (directly from PSP) */`
114	`static void parse_argv(struct config *cfg) {`	123	`static void parse_argv(struct config *cfg) {`
115	`const unsigned char cmdlinelen = (void )0x80;`	124	`const unsigned char cmdlinelen = (void )0x80;`
116	`char cmdline = (void )0x81;`	125	`char cmdline = (void )0x81;`
117		126
118	`memset(cfg, 0, sizeof(*cfg));`	127	`memset(cfg, 0, sizeof(*cfg));`
119		128
120	`/* set a NULL terminator on cmdline */`	129	`/* set a NULL terminator on cmdline */`
121	`cmdline[*cmdlinelen] = 0;`	130	`cmdline[*cmdlinelen] = 0;`
122		131
123	`while (*cmdline != 0) {`	132	`while (*cmdline != 0) {`
124		133
125	`/* skip over any leading spaces */`	134	`/* skip over any leading spaces */`
126	`if (*cmdline == ' ') {`	135	`if (*cmdline == ' ') {`
127	`cmdline++;`	136	`cmdline++;`
128	`continue;`	137	`continue;`
129	`}`	138	`}`
130		139
131	`if (*cmdline != '/') {`	140	`if (*cmdline != '/') {`
132	`output("Invalid parameter: ");`	141	`output("Invalid parameter: ");`
133	`outputnl(cmdline);`	142	`outputnl(cmdline);`
134	`goto SKIP_TO_NEXT_ARG;`	143	`goto SKIP_TO_NEXT_ARG;`
135	`}`	144	`}`
136		145
137	`/* got a slash */`	146	`/* got a slash */`
138	`cmdline++; /* skip the slash */`	147	`cmdline++; /* skip the slash */`
139	`switch (*cmdline) {`	148	`switch (*cmdline) {`
140	`case 'c': /* /C = execute command and quit */`	149	`case 'c': /* /C = execute command and quit */`
141	`case 'C':`	150	`case 'C':`
142	`cfg->flags \|= FLAG_EXEC_AND_QUIT;`	151	`cfg->flags \|= FLAG_EXEC_AND_QUIT;`
143	`/* FALLTHRU */`	152	`/* FALLTHRU */`
144	`case 'k': /* /K = execute command and keep running */`	153	`case 'k': /* /K = execute command and keep running */`
145	`case 'K':`	154	`case 'K':`
146	`cfg->execcmd = cmdline + 1;`	155	`cfg->execcmd = cmdline + 1;`
147	`return;`	156	`return;`
148		157
149	`case 'd': /* /D = skip autoexec.bat processing */`	158	`case 'd': /* /D = skip autoexec.bat processing */`
150	`case 'D':`	159	`case 'D':`
151	`cfg->flags \|= FLAG_SKIP_AUTOEXEC;`	160	`cfg->flags \|= FLAG_SKIP_AUTOEXEC;`
152	`break;`	161	`break;`
153		162
154	`case 'e': /* preset the initial size of the environment block */`	163	`case 'e': /* preset the initial size of the environment block */`
155	`case 'E':`	164	`case 'E':`
156	`cmdline++;`	165	`cmdline++;`
157	`if (cmdline == ':') cmdline++; / could be /E:size */`	166	`if (cmdline == ':') cmdline++; / could be /E:size */`
158	`atous(&(cfg->envsiz), cmdline);`	167	`atous(&(cfg->envsiz), cmdline);`
159	`if (cfg->envsiz < 64) cfg->envsiz = 0;`	168	`if (cfg->envsiz < 64) cfg->envsiz = 0;`
160	`break;`	169	`break;`
161		170
162	`case 'p': /* permanent shell (can't exit + run autoexec.bat) */`	171	`case 'p': /* permanent shell (can't exit + run autoexec.bat) */`
163	`case 'P':`	172	`case 'P':`
164	`cfg->flags \|= FLAG_PERMANENT;`	173	`cfg->flags \|= FLAG_PERMANENT;`
165	`break;`	174	`break;`
166		175
167	`case '?':`	176	`case '?':`
168	`outputnl("Starts the SvarCOM command interpreter");`	177	`outputnl("Starts the SvarCOM command interpreter");`
169	`outputnl("");`	178	`outputnl("");`
170	`outputnl("COMMAND /E:nnn [/[C\|K] [/P] [/D] command]");`	179	`outputnl("COMMAND /E:nnn [/[C\|K] [/P] [/D] command]");`
171	`outputnl("");`	180	`outputnl("");`
172	`outputnl("/D Skip AUTOEXEC.BAT processing (makes sense only with /P)");`	181	`outputnl("/D Skip AUTOEXEC.BAT processing (makes sense only with /P)");`
173	`outputnl("/E:nnn Sets the environment size to nnn bytes");`	182	`outputnl("/E:nnn Sets the environment size to nnn bytes");`
174	`outputnl("/P Makes the new command interpreter permanent and run AUTOEXEC.BAT");`	183	`outputnl("/P Makes the new command interpreter permanent and run AUTOEXEC.BAT");`
175	`outputnl("/C Executes the specified command and returns");`	184	`outputnl("/C Executes the specified command and returns");`
176	`outputnl("/K Executes the specified command and continues running");`	185	`outputnl("/K Executes the specified command and continues running");`
177	`exit(1);`	186	`exit(1);`
178	`break;`	187	`break;`
179		188
180	`default:`	189	`default:`
181	`output("Invalid switch: /");`	190	`output("Invalid switch: /");`
182	`outputnl(cmdline);`	191	`outputnl(cmdline);`
183	`break;`	192	`break;`
184	`}`	193	`}`
185		194
186	`/* move to next argument or quit processing if end of cmdline */`	195	`/* move to next argument or quit processing if end of cmdline */`
187	`SKIP_TO_NEXT_ARG:`	196	`SKIP_TO_NEXT_ARG:`
188	`while ((cmdline != 0) && (cmdline != ' ') && (*cmdline != '/')) cmdline++;`	197	`while ((cmdline != 0) && (cmdline != ' ') && (*cmdline != '/')) cmdline++;`
189	`}`	198	`}`
190	`}`	199	`}`
191		200
192		201
193	`/* builds the prompt string and displays it. buff is filled with a zero-terminated copy of the prompt. */`	202	`/* builds the prompt string and displays it. buff is filled with a zero-terminated copy of the prompt. */`
194	`static void build_and_display_prompt(char *buff, unsigned short envseg) {`	203	`static void build_and_display_prompt(char *buff, unsigned short envseg) {`
195	`char *s = buff;`	204	`char *s = buff;`
196	`/* locate the prompt variable or use the default pattern */`	205	`/* locate the prompt variable or use the default pattern */`
197	`const char far *fmt = env_lookup_val(envseg, "PROMPT");`	206	`const char far *fmt = env_lookup_val(envseg, "PROMPT");`
198	`if ((fmt == NULL) \|\| (fmt == 0)) fmt = "$p$g"; / fallback to default if empty */`	207	`if ((fmt == NULL) \|\| (fmt == 0)) fmt = "$p$g"; / fallback to default if empty */`
199	`/* build the prompt string based on pattern */`	208	`/* build the prompt string based on pattern */`
200	`for (; *fmt != 0; fmt++) {`	209	`for (; *fmt != 0; fmt++) {`
201	`if (*fmt != '$') {`	210	`if (*fmt != '$') {`
202	`s = fmt;`	211	`s = fmt;`
203	`s++;`	212	`s++;`
204	`continue;`	213	`continue;`
205	`}`	214	`}`
206	`/* escape code ($P, etc) */`	215	`/* escape code ($P, etc) */`
207	`fmt++;`	216	`fmt++;`
208	`switch (*fmt) {`	217	`switch (*fmt) {`
209	`case 'Q': /* $Q = = (equal sign) */`	218	`case 'Q': /* $Q = = (equal sign) */`
210	`case 'q':`	219	`case 'q':`
211	`*s = '=';`	220	`*s = '=';`
212	`s++;`	221	`s++;`
213	`break;`	222	`break;`
214	`case '$': /* $$ = $ (dollar sign) */`	223	`case '$': /* $$ = $ (dollar sign) */`
215	`*s = '$';`	224	`*s = '$';`
216	`s++;`	225	`s++;`
217	`break;`	226	`break;`
218	`case 'T': /* $t = current time */`	227	`case 'T': /* $t = current time */`
219	`case 't':`	228	`case 't':`
220	`s += sprintf(s, "00:00"); /* TODO */`	229	`s += sprintf(s, "00:00"); /* TODO */`
221	`break;`	230	`break;`
222	`case 'D': /* $D = current date */`	231	`case 'D': /* $D = current date */`
223	`case 'd':`	232	`case 'd':`
224	`s += sprintf(s, "1985-07-29"); /* TODO */`	233	`s += sprintf(s, "1985-07-29"); /* TODO */`
225	`break;`	234	`break;`
226	`case 'P': /* $P = current drive and path */`	235	`case 'P': /* $P = current drive and path */`
227	`case 'p':`	236	`case 'p':`
228	`_asm {`	237	`_asm {`
229	`mov ah, 0x19 /* DOS 1+ - GET CURRENT DRIVE */`	238	`mov ah, 0x19 /* DOS 1+ - GET CURRENT DRIVE */`
230	`int 0x21`	239	`int 0x21`
231	`mov bx, s`	240	`mov bx, s`
232	`mov [bx], al /* AL = drive (00 = A:, 01 = B:, etc */`	241	`mov [bx], al /* AL = drive (00 = A:, 01 = B:, etc */`
233	`}`	242	`}`
234	`*s += 'A';`	243	`*s += 'A';`
235	`s++;`	244	`s++;`
236	`*s = ':';`	245	`*s = ':';`
237	`s++;`	246	`s++;`
238	`*s = '\\';`	247	`*s = '\\';`
239	`s++;`	248	`s++;`
240	`_asm {`	249	`_asm {`
241	`mov ah, 0x47 /* DOS 2+ - CWD - GET CURRENT DIRECTORY */`	250	`mov ah, 0x47 /* DOS 2+ - CWD - GET CURRENT DIRECTORY */`
242	`xor dl,dl /* DL = drive number (00h = default, 01h = A:, etc) */`	251	`xor dl,dl /* DL = drive number (00h = default, 01h = A:, etc) */`
243	`mov si, s /* DS:SI -> 64-byte buffer for ASCIZ pathname */`	252	`mov si, s /* DS:SI -> 64-byte buffer for ASCIZ pathname */`
244	`int 0x21`	253	`int 0x21`
245	`jc DONE /* leave path empty on error */`	254	`jc DONE /* leave path empty on error */`
246	`/* move s ptr forward to end (0-termintor) of pathname */`	255	`/* move s ptr forward to end (0-termintor) of pathname */`
247	`NEXTBYTE:`	256	`NEXTBYTE:`
248	`mov si, s`	257	`mov si, s`
249	`cmp byte ptr [si], 0`	258	`cmp byte ptr [si], 0`
250	`je DONE`	259	`je DONE`
251	`inc s`	260	`inc s`
252	`jmp NEXTBYTE`	261	`jmp NEXTBYTE`
253	`DONE:`	262	`DONE:`
254	`}`	263	`}`
255	`break;`	264	`break;`
256	`case 'V': /* $V = DOS version number */`	265	`case 'V': /* $V = DOS version number */`
257	`case 'v':`	266	`case 'v':`
258	`s += sprintf(s, "VER"); /* TODO */`	267	`s += sprintf(s, "VER"); /* TODO */`
259	`break;`	268	`break;`
260	`case 'N': /* $N = current drive */`	269	`case 'N': /* $N = current drive */`
261	`case 'n':`	270	`case 'n':`
262	`_asm {`	271	`_asm {`
263	`mov ah, 0x19 /* DOS 1+ - GET CURRENT DRIVE */`	272	`mov ah, 0x19 /* DOS 1+ - GET CURRENT DRIVE */`
264	`int 0x21`	273	`int 0x21`
265	`mov bx, s`	274	`mov bx, s`
266	`mov [bx], al /* AL = drive (00 = A:, 01 = B:, etc */`	275	`mov [bx], al /* AL = drive (00 = A:, 01 = B:, etc */`
267	`}`	276	`}`
268	`*s += 'A';`	277	`*s += 'A';`
269	`s++;`	278	`s++;`
270	`break;`	279	`break;`
271	`case 'G': /* $G = > (greater-than sign) */`	280	`case 'G': /* $G = > (greater-than sign) */`
272	`case 'g':`	281	`case 'g':`
273	`*s = '>';`	282	`*s = '>';`
274	`s++;`	283	`s++;`
275	`break;`	284	`break;`
276	`case 'L': /* $L = < (less-than sign) */`	285	`case 'L': /* $L = < (less-than sign) */`
277	`case 'l':`	286	`case 'l':`
278	`*s = '<';`	287	`*s = '<';`
279	`s++;`	288	`s++;`
280	`break;`	289	`break;`
281	`case 'B': /* $B = \| (pipe) */`	290	`case 'B': /* $B = \| (pipe) */`
282	`case 'b':`	291	`case 'b':`
283	`*s = '\|';`	292	`*s = '\|';`
284	`s++;`	293	`s++;`
285	`break;`	294	`break;`
286	`case 'H': /* $H = backspace (erases previous character) */`	295	`case 'H': /* $H = backspace (erases previous character) */`
287	`case 'h':`	296	`case 'h':`
288	`*s = '\b';`	297	`*s = '\b';`
289	`s++;`	298	`s++;`
290	`break;`	299	`break;`
291	`case 'E': /* $E = Escape code (ASCII 27) */`	300	`case 'E': /* $E = Escape code (ASCII 27) */`
292	`case 'e':`	301	`case 'e':`
293	`*s = 27;`	302	`*s = 27;`
294	`s++;`	303	`s++;`
295	`break;`	304	`break;`
296	`case '_': /* $_ = CR+LF */`	305	`case '_': /* $_ = CR+LF */`
297	`*s = '\r';`	306	`*s = '\r';`
298	`s++;`	307	`s++;`
299	`*s = '\n';`	308	`*s = '\n';`
300	`s++;`	309	`s++;`
301	`break;`	310	`break;`
302	`}`	311	`}`
303	`}`	312	`}`
304	`*s = 0;`	313	`*s = 0;`
305	`output(buff);`	314	`output(buff);`
306	`}`	315	`}`
307		316
308		317
309	`static void run_as_external(char buff, const char cmdline, unsigned short envseg, struct rmod_props far rmod, struct redir_data redir) {`	318	`static void run_as_external(char buff, const char cmdline, unsigned short envseg, struct rmod_props far rmod, struct redir_data redir) {`
310	`char *cmdfile = buff + 512;`	319	`char *cmdfile = buff + 512;`
311	`const char far *pathptr;`	320	`const char far *pathptr;`
312	`int lookup;`	321	`int lookup;`
313	`unsigned short i;`	322	`unsigned short i;`
314	`const char *ext;`	323	`const char *ext;`
315	`char *cmd = buff + 1024;`	324	`char *cmd = buff + 1024;`
316	`const char *cmdtail;`	325	`const char *cmdtail;`
317	`char far *rmod_execprog = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPROG);`	326	`char far *rmod_execprog = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPROG);`
318	`char far *rmod_cmdtail = MK_FP(rmod->rmodseg, 0x81);`	327	`char far *rmod_cmdtail = MK_FP(rmod->rmodseg, 0x81);`
319	`_Packed struct {`	328	`_Packed struct {`
320	`unsigned short envseg;`	329	`unsigned short envseg;`
321	`unsigned long cmdtail;`	330	`unsigned long cmdtail;`
322	`unsigned long fcb1;`	331	`unsigned long fcb1;`
323	`unsigned long fcb2;`	332	`unsigned long fcb2;`
324	`} far *ExecParam = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPARAM);`	333	`} far *ExecParam = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPARAM);`
325		334
326	`/* find cmd and cmdtail */`	335	`/* find cmd and cmdtail */`
327	`i = 0;`	336	`i = 0;`
328	`cmdtail = cmdline;`	337	`cmdtail = cmdline;`
329	`while (cmdtail == ' ') cmdtail++; / skip any leading spaces */`	338	`while (cmdtail == ' ') cmdtail++; / skip any leading spaces */`
330	`while ((cmdtail != ' ') && (cmdtail != '/') && (cmdtail != '+') && (cmdtail != 0)) {`	339	`while ((cmdtail != ' ') && (cmdtail != '/') && (cmdtail != '+') && (cmdtail != 0)) {`
331	`cmd[i++] = *cmdtail;`	340	`cmd[i++] = *cmdtail;`
332	`cmdtail++;`	341	`cmdtail++;`
333	`}`	342	`}`
334	`cmd[i] = 0;`	343	`cmd[i] = 0;`
335		344
336	`/* is this a command in curdir? */`	345	`/* is this a command in curdir? */`
337	`lookup = lookup_cmd(cmdfile, cmd, NULL, &ext);`	346	`lookup = lookup_cmd(cmdfile, cmd, NULL, &ext);`
338	`if (lookup == 0) {`	347	`if (lookup == 0) {`
339	`/* printf("FOUND LOCAL EXEC FILE: '%s'\r\n", cmdfile); */`	348	`/* printf("FOUND LOCAL EXEC FILE: '%s'\r\n", cmdfile); */`
340	`goto RUNCMDFILE;`	349	`goto RUNCMDFILE;`
341	`} else if (lookup == -2) {`	350	`} else if (lookup == -2) {`
342	`/* puts("NOT FOUND"); */`	351	`/* puts("NOT FOUND"); */`
343	`return;`	352	`return;`
344	`}`	353	`}`
345		354
346	`/* try matching something in PATH */`	355	`/* try matching something in PATH */`
347	`pathptr = env_lookup_val(envseg, "PATH");`	356	`pathptr = env_lookup_val(envseg, "PATH");`
348		357
349	`/* try each path in %PATH% */`	358	`/* try each path in %PATH% */`
350	`while (pathptr) {`	359	`while (pathptr) {`
351	`for (i = 0;; i++) {`	360	`for (i = 0;; i++) {`
352	`buff[i] = *pathptr;`	361	`buff[i] = *pathptr;`
353	`if ((buff[i] == 0) \|\| (buff[i] == ';')) break;`	362	`if ((buff[i] == 0) \|\| (buff[i] == ';')) break;`
354	`pathptr++;`	363	`pathptr++;`
355	`}`	364	`}`
356	`buff[i] = 0;`	365	`buff[i] = 0;`
357	`lookup = lookup_cmd(cmdfile, cmd, buff, &ext);`	366	`lookup = lookup_cmd(cmdfile, cmd, buff, &ext);`
358	`if (lookup == 0) goto RUNCMDFILE;`	367	`if (lookup == 0) goto RUNCMDFILE;`
359	`if (lookup == -2) return;`	368	`if (lookup == -2) return;`
360	`if (*pathptr == ';') {`	369	`if (*pathptr == ';') {`
361	`pathptr++;`	370	`pathptr++;`
362	`} else {`	371	`} else {`
363	`break;`	372	`break;`
364	`}`	373	`}`
365	`}`	374	`}`
366		375
367	`/* last chance: is it an executable link? (trim extension from cmd first) */`	376	`/* last chance: is it an executable link? (trim extension from cmd first) */`
368	`for (i = 0; (cmd[i] != 0) && (cmd[i] != '.') && (i < 9); i++) buff[128 + i] = cmd[i];`	377	`for (i = 0; (cmd[i] != 0) && (cmd[i] != '.') && (i < 9); i++) buff[128 + i] = cmd[i];`
369	`buff[128 + i] = 0;`	378	`buff[128 + i] = 0;`
370	`if ((i < 9) && (link_computefname(buff, buff + 128, envseg) == 0)) {`	379	`if ((i < 9) && (link_computefname(buff, buff + 128, envseg) == 0)) {`
371	`/* try opening the link file (if it exists) and read it into buff */`	380	`/* try opening the link file (if it exists) and read it into buff */`
372	`i = 0;`	381	`i = 0;`
373	`_asm {`	382	`_asm {`
374	`push ax`	383	`push ax`
375	`push bx`	384	`push bx`
376	`push cx`	385	`push cx`
377	`push dx`	386	`push dx`
378		387
379	`mov ax, 0x3d00 /* DOS 2+ - OPEN EXISTING FILE, READ-ONLY */`	388	`mov ax, 0x3d00 /* DOS 2+ - OPEN EXISTING FILE, READ-ONLY */`
380	`mov dx, buff /* file name */`	389	`mov dx, buff /* file name */`
381	`int 0x21`	390	`int 0x21`
382	`jc ERR_FOPEN`	391	`jc ERR_FOPEN`
383	`/* file handle in AX, read from file now */`	392	`/* file handle in AX, read from file now */`
384	`mov bx, ax /* file handle */`	393	`mov bx, ax /* file handle */`
385	`mov ah, 0x3f /* Read from file via handle bx */`	394	`mov ah, 0x3f /* Read from file via handle bx */`
386	`mov cx, 128 /* up to 128 bytes */`	395	`mov cx, 128 /* up to 128 bytes */`
387	`/* mov dx, buff / / dest buffer (already set) */`	396	`/* mov dx, buff / / dest buffer (already set) */`
388	`int 0x21 /* read up to 256 bytes from file and write to buff */`	397	`int 0x21 /* read up to 256 bytes from file and write to buff */`
389	`jc ERR_READ`	398	`jc ERR_READ`
390	`mov i, ax`	399	`mov i, ax`
391	`ERR_READ:`	400	`ERR_READ:`
392	`mov ah, 0x3e /* close file handle in BX */`	401	`mov ah, 0x3e /* close file handle in BX */`
393	`int 0x21`	402	`int 0x21`
394	`ERR_FOPEN:`	403	`ERR_FOPEN:`
395		404
396	`pop dx`	405	`pop dx`
397	`pop cx`	406	`pop cx`
398	`pop bx`	407	`pop bx`
399	`pop ax`	408	`pop ax`
400	`}`	409	`}`
401		410
402	`/* did I read anything? */`	411	`/* did I read anything? */`
403	`if (i != 0) {`	412	`if (i != 0) {`
404	`buff[i] = 0;`	413	`buff[i] = 0;`
405	`/* trim buff at first \n or \r, just in case someone fiddled with the`	414	`/* trim buff at first \n or \r, just in case someone fiddled with the`
406	`* link file using a text editor */`	415	`* link file using a text editor */`
407	`for (i = 0; (buff[i] != 0) && (buff[i] != '\r') && (buff[i] != '\n'); i++);`	416	`for (i = 0; (buff[i] != 0) && (buff[i] != '\r') && (buff[i] != '\n'); i++);`
408	`buff[i] = 0;`	417	`buff[i] = 0;`
409	`/* lookup check */`	418	`/* lookup check */`
410	`if (buff[0] != 0) {`	419	`if (buff[0] != 0) {`
411	`lookup = lookup_cmd(cmdfile, cmd, buff, &ext);`	420	`lookup = lookup_cmd(cmdfile, cmd, buff, &ext);`
412	`if (lookup == 0) goto RUNCMDFILE;`	421	`if (lookup == 0) goto RUNCMDFILE;`
413	`}`	422	`}`
414	`}`	423	`}`
415	`}`	424	`}`
416		425
417	`/* all failed (ie. executable file not found) */`	426	`/* all failed (ie. executable file not found) */`
418	`return;`	427	`return;`
419		428
420	`RUNCMDFILE:`	429	`RUNCMDFILE:`
421		430
422	`/* special handling of batch files */`	431	`/* special handling of batch files */`
423	`if ((ext != NULL) && (imatch(ext, "bat"))) {`	432	`if ((ext != NULL) && (imatch(ext, "bat"))) {`
424	`/* copy truename of the bat file to rmod buff */`	433	`/* copy truename of the bat file to rmod buff */`
425	`_fstrcpy(rmod->batfile, cmdfile);`	434	`_fstrcpy(rmod->batfile, cmdfile);`
426		435
427	`/* explode args of the bat file and store them in rmod buff */`	436	`/* explode args of the bat file and store them in rmod buff */`
428	`cmd_explode(buff, cmdline, NULL);`	437	`cmd_explode(buff, cmdline, NULL);`
429	`_fmemcpy(rmod->batargv, buff, sizeof(rmod->batargv));`	438	`_fmemcpy(rmod->batargv, buff, sizeof(rmod->batargv));`
430		439
431	`/* reset the 'next line to execute' counter */`	440	`/* reset the 'next line to execute' counter */`
432	`rmod->batnextline = 0;`	441	`rmod->batnextline = 0;`
433	`/* remember the echo flag (in case bat file disables echo) */`	442	`/* remember the echo flag (in case bat file disables echo) */`
434	`rmod->flags &= ~FLAG_ECHO_BEFORE_BAT;`	443	`rmod->flags &= ~FLAG_ECHO_BEFORE_BAT;`
435	`if (rmod->flags & FLAG_ECHOFLAG) rmod->flags \|= FLAG_ECHO_BEFORE_BAT;`	444	`if (rmod->flags & FLAG_ECHOFLAG) rmod->flags \|= FLAG_ECHO_BEFORE_BAT;`
436	`return;`	445	`return;`
437	`}`	446	`}`
438		447
439	`/* copy full filename to execute, along with redirected files (if any) */`	448	`/* copy full filename to execute, along with redirected files (if any) */`
440	`_fstrcpy(rmod_execprog, cmdfile);`	449	`_fstrcpy(rmod_execprog, cmdfile);`
441		450
442	`/* copy stdin file if a redirection is needed */`	451	`/* copy stdin file if a redirection is needed */`
443	`if (redir->stdinfile) {`	452	`if (redir->stdinfile) {`
444	`char far *farptr = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDINFILE);`	453	`char far *farptr = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDINFILE);`
445	`_fstrcpy(farptr, redir->stdinfile);`	454	`_fstrcpy(farptr, redir->stdinfile);`
446	`}`	455	`}`
447		456
448	`/* same for stdout file */`	457	`/* same for stdout file */`
449	`if (redir->stdoutfile) {`	458	`if (redir->stdoutfile) {`
450	`char far *farptr = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDOUTFILE);`	459	`char far *farptr = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDOUTFILE);`
451	`unsigned short far *farptr16 = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDOUTAPP);`	460	`unsigned short far *farptr16 = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDOUTAPP);`
452	`_fstrcpy(farptr, redir->stdoutfile);`	461	`_fstrcpy(farptr, redir->stdoutfile);`
453	`/* openflag */`	462	`/* openflag */`
454	`*farptr16 = redir->stdout_openflag;`	463	`*farptr16 = redir->stdout_openflag;`
455	`}`	464	`}`
456		465
457	`/* copy cmdtail to rmod's PSP and compute its len */`	466	`/* copy cmdtail to rmod's PSP and compute its len */`
458	`for (i = 0; cmdtail[i] != 0; i++) rmod_cmdtail[i] = cmdtail[i];`	467	`for (i = 0; cmdtail[i] != 0; i++) rmod_cmdtail[i] = cmdtail[i];`
459	`rmod_cmdtail[i] = '\r';`	468	`rmod_cmdtail[i] = '\r';`
460	`rmod_cmdtail[-1] = i;`	469	`rmod_cmdtail[-1] = i;`
461		470
462	`/* set up rmod to execute the command */`	471	`/* set up rmod to execute the command */`
463		472
464	`ExecParam->envseg = envseg;`	473	`ExecParam->envseg = envseg;`
465	`ExecParam->cmdtail = (unsigned long)MK_FP(rmod->rmodseg, 0x80); /* farptr, must be in PSP format (lenbyte args \r) */`	474	`ExecParam->cmdtail = (unsigned long)MK_FP(rmod->rmodseg, 0x80); /* farptr, must be in PSP format (lenbyte args \r) */`
466	`ExecParam->fcb1 = 0; /* TODO farptr */`	475	`ExecParam->fcb1 = 0; /* TODO farptr */`
467	`ExecParam->fcb2 = 0; /* TODO farptr */`	476	`ExecParam->fcb2 = 0; /* TODO farptr */`
468	`exit(0); /* let rmod do the job now */`	477	`exit(0); /* let rmod do the job now */`
469	`}`	478	`}`
470		479
471		480
472	`static void set_comspec_to_self(unsigned short envseg) {`	481	`static void set_comspec_to_self(unsigned short envseg) {`
473	`unsigned short psp_envseg = (void )(0x2c); /* pointer to my env segment field in the PSP */`	482	`unsigned short psp_envseg = (void )(0x2c); /* pointer to my env segment field in the PSP */`
474	`char far myenv = MK_FP(psp_envseg, 0);`	483	`char far myenv = MK_FP(psp_envseg, 0);`
475	`unsigned short varcount;`	484	`unsigned short varcount;`
476	`char buff[256] = "COMSPEC=";`	485	`char buff[256] = "COMSPEC=";`
477	`char *buffptr = buff + 8;`	486	`char *buffptr = buff + 8;`
478	`/* who am i? look into my own environment, at the end of it should be my EXEPATH string */`	487	`/* who am i? look into my own environment, at the end of it should be my EXEPATH string */`
479	`while (*myenv != 0) {`	488	`while (*myenv != 0) {`
480	`/* consume a NULL-terminated string */`	489	`/* consume a NULL-terminated string */`
481	`while (*myenv != 0) myenv++;`	490	`while (*myenv != 0) myenv++;`
482	`/* move to next string */`	491	`/* move to next string */`
483	`myenv++;`	492	`myenv++;`
484	`}`	493	`}`
485	`/* get next word, if 1 then EXEPATH follows */`	494	`/* get next word, if 1 then EXEPATH follows */`
486	`myenv++;`	495	`myenv++;`
487	`varcount = *myenv;`	496	`varcount = *myenv;`
488	`myenv++;`	497	`myenv++;`
489	`varcount \|= (*myenv << 8);`	498	`varcount \|= (*myenv << 8);`
490	`myenv++;`	499	`myenv++;`
491	`if (varcount != 1) return; /* NO EXEPATH FOUND */`	500	`if (varcount != 1) return; /* NO EXEPATH FOUND */`

Subversion Repositories SvarDOS

(root)/svarcom/trunk/command.c – Rev 571 → 572