Subversion Repositories SvarDOS

Rev

Rev 572 | Rev 577 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
421 mateuszvis 1
/* This file is part of the SvarCOM project and is published under the terms
2
 * of the MIT license.
3
 *
571 mateuszvis 4
 * Copyright (C) 2021-2022 Mateusz Viste
421 mateuszvis 5
 *
6
 * Permission is hereby granted, free of charge, to any person obtaining a
7
 * copy of this software and associated documentation files (the "Software"),
8
 * to deal in the Software without restriction, including without limitation
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
11
 * Software is furnished to do so, subject to the following conditions:
12
 *
13
 * The above copyright notice and this permission notice shall be included in
14
 * all copies or substantial portions of the Software.
15
 *
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,
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
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
22
 * DEALINGS IN THE SOFTWARE.
23
 */
24
 
349 mateuszvis 25
#include <i86.h>
26
#include <dos.h>
27
#include <stdio.h>
350 mateuszvis 28
#include <stdlib.h>
349 mateuszvis 29
#include <string.h>
30
 
31
#include <process.h>
32
 
352 mateuszvis 33
#include "cmd.h"
366 mateuszvis 34
#include "env.h"
352 mateuszvis 35
#include "helpers.h"
402 mateuszvis 36
#include "redir.h"
351 mateuszvis 37
#include "rmodinit.h"
448 mateuszvis 38
#include "sayonara.h"
349 mateuszvis 39
 
479 mateuszvis 40
#include "rmodcore.h" /* rmod binary inside a BUFFER array */
443 mateuszvis 41
 
572 mateuszvis 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
 
349 mateuszvis 51
struct config {
449 mateuszvis 52
  unsigned char flags; /* command.com flags, as defined in rmodinit.h */
443 mateuszvis 53
  char *execcmd;
410 mateuszvis 54
  unsigned short envsiz;
443 mateuszvis 55
};
349 mateuszvis 56
 
490 mateuszvis 57
/* max length of the cmdline storage (bytes) - includes also max length of
58
 * line loaded from a BAT file (no more than 255 bytes!) */
59
#define CMDLINE_MAXLEN 255
349 mateuszvis 60
 
490 mateuszvis 61
 
62
/* sets guard values at a few places in memory for later detection of
63
 * overflows via memguard_check() */
500 mateuszvis 64
static void memguard_set(char *cmdlinebuf) {
490 mateuszvis 65
  BUFFER[sizeof(BUFFER) - 1] = 0xC7;
500 mateuszvis 66
  cmdlinebuf[CMDLINE_MAXLEN] = 0xC7;
490 mateuszvis 67
}
68
 
69
 
70
/* checks for valguards at specific memory locations, returns 0 on success */
500 mateuszvis 71
static int memguard_check(unsigned short rmodseg, char *cmdlinebuf) {
490 mateuszvis 72
  /* check RMOD signature (would be overwritten in case of stack overflow */
73
  static char msg[] = "!! MEMORY CORRUPTION ## DETECTED !!";
74
  unsigned short far *rmodsig = MK_FP(rmodseg, 0x100 + 6);
548 mateuszvis 75
  unsigned char far *rmod = MK_FP(rmodseg, 0);
76
 
490 mateuszvis 77
  if (*rmodsig != 0x2019) {
78
    msg[22] = '1';
548 mateuszvis 79
    goto FAIL;
490 mateuszvis 80
  }
548 mateuszvis 81
 
500 mateuszvis 82
  /* check last BUFFER byte */
490 mateuszvis 83
  if (BUFFER[sizeof(BUFFER) - 1] != 0xC7) {
84
    msg[22] = '2';
548 mateuszvis 85
    goto FAIL;
490 mateuszvis 86
  }
548 mateuszvis 87
 
500 mateuszvis 88
  /* check last cmdlinebuf byte */
89
  if (cmdlinebuf[CMDLINE_MAXLEN] != 0xC7) {
490 mateuszvis 90
    msg[22] = '3';
548 mateuszvis 91
    goto FAIL;
490 mateuszvis 92
  }
548 mateuszvis 93
 
94
  /* check rmod exec buf */
95
  if (rmod[RMOD_OFFSET_EXECPROG + 127] != 0) {
96
    msg[22] = '4';
97
    goto FAIL;
98
  }
99
 
100
  /* check rmod exec stdin buf */
101
  if (rmod[RMOD_OFFSET_STDINFILE + 127] != 0) {
102
    msg[22] = '5';
103
    goto FAIL;
104
  }
105
 
106
  /* check rmod exec stdout buf */
107
  if (rmod[RMOD_OFFSET_STDOUTFILE + 127] != 0) {
108
    msg[22] = '6';
109
    goto FAIL;
110
  }
111
 
112
  /* else all good */
490 mateuszvis 113
  return(0);
548 mateuszvis 114
 
115
  /* error handling */
116
  FAIL:
117
  outputnl(msg);
118
  return(1);
490 mateuszvis 119
}
120
 
121
 
443 mateuszvis 122
/* parses command line the hard way (directly from PSP) */
123
static void parse_argv(struct config *cfg) {
491 mateuszvis 124
  const unsigned char *cmdlinelen = (void *)0x80;
125
  char *cmdline = (void *)0x81;
443 mateuszvis 126
 
349 mateuszvis 127
  memset(cfg, 0, sizeof(*cfg));
350 mateuszvis 128
 
443 mateuszvis 129
  /* set a NULL terminator on cmdline */
130
  cmdline[*cmdlinelen] = 0;
131
 
491 mateuszvis 132
  while (*cmdline != 0) {
443 mateuszvis 133
 
134
    /* skip over any leading spaces */
491 mateuszvis 135
    if (*cmdline == ' ') {
136
      cmdline++;
137
      continue;
349 mateuszvis 138
    }
443 mateuszvis 139
 
491 mateuszvis 140
    if (*cmdline != '/') {
443 mateuszvis 141
      output("Invalid parameter: ");
491 mateuszvis 142
      outputnl(cmdline);
143
      goto SKIP_TO_NEXT_ARG;
144
    }
443 mateuszvis 145
 
491 mateuszvis 146
    /* got a slash */
147
    cmdline++;  /* skip the slash */
148
    switch (*cmdline) {
149
      case 'c': /* /C = execute command and quit */
150
      case 'C':
151
        cfg->flags |= FLAG_EXEC_AND_QUIT;
152
        /* FALLTHRU */
153
      case 'k': /* /K = execute command and keep running */
154
      case 'K':
155
        cfg->execcmd = cmdline + 1;
156
        return;
443 mateuszvis 157
 
494 mateuszvis 158
      case 'd': /* /D = skip autoexec.bat processing */
159
      case 'D':
160
        cfg->flags |= FLAG_SKIP_AUTOEXEC;
161
        break;
162
 
491 mateuszvis 163
      case 'e': /* preset the initial size of the environment block */
164
      case 'E':
165
        cmdline++;
166
        if (*cmdline == ':') cmdline++; /* could be /E:size */
167
        atous(&(cfg->envsiz), cmdline);
168
        if (cfg->envsiz < 64) cfg->envsiz = 0;
169
        break;
449 mateuszvis 170
 
491 mateuszvis 171
      case 'p': /* permanent shell (can't exit + run autoexec.bat) */
172
      case 'P':
173
        cfg->flags |= FLAG_PERMANENT;
174
        break;
444 mateuszvis 175
 
491 mateuszvis 176
      case '?':
177
        outputnl("Starts the SvarCOM command interpreter");
178
        outputnl("");
494 mateuszvis 179
        outputnl("COMMAND /E:nnn [/[C|K] [/P] [/D] command]");
491 mateuszvis 180
        outputnl("");
494 mateuszvis 181
        outputnl("/D      Skip AUTOEXEC.BAT processing (makes sense only with /P)");
182
        outputnl("/E:nnn  Sets the environment size to nnn bytes");
183
        outputnl("/P      Makes the new command interpreter permanent and run AUTOEXEC.BAT");
184
        outputnl("/C      Executes the specified command and returns");
185
        outputnl("/K      Executes the specified command and continues running");
491 mateuszvis 186
        exit(1);
187
        break;
188
 
189
      default:
494 mateuszvis 190
        output("Invalid switch: /");
491 mateuszvis 191
        outputnl(cmdline);
192
        break;
350 mateuszvis 193
    }
443 mateuszvis 194
 
195
    /* move to next argument or quit processing if end of cmdline */
491 mateuszvis 196
    SKIP_TO_NEXT_ARG:
197
    while ((*cmdline != 0) && (*cmdline != ' ') && (*cmdline != '/')) cmdline++;
349 mateuszvis 198
  }
199
}
200
 
201
 
474 mateuszvis 202
/* builds the prompt string and displays it. buff is filled with a zero-terminated copy of the prompt. */
203
static void build_and_display_prompt(char *buff, unsigned short envseg) {
204
  char *s = buff;
370 mateuszvis 205
  /* locate the prompt variable or use the default pattern */
438 mateuszvis 206
  const char far *fmt = env_lookup_val(envseg, "PROMPT");
370 mateuszvis 207
  if ((fmt == NULL) || (*fmt == 0)) fmt = "$p$g"; /* fallback to default if empty */
208
  /* build the prompt string based on pattern */
354 mateuszvis 209
  for (; *fmt != 0; fmt++) {
210
    if (*fmt != '$') {
211
      *s = *fmt;
212
      s++;
213
      continue;
214
    }
215
    /* escape code ($P, etc) */
216
    fmt++;
217
    switch (*fmt) {
218
      case 'Q':  /* $Q = = (equal sign) */
219
      case 'q':
220
        *s = '=';
221
        s++;
222
        break;
223
      case '$':  /* $$ = $ (dollar sign) */
224
        *s = '$';
225
        s++;
226
        break;
227
      case 'T':  /* $t = current time */
228
      case 't':
229
        s += sprintf(s, "00:00"); /* TODO */
230
        break;
231
      case 'D':  /* $D = current date */
232
      case 'd':
233
        s += sprintf(s, "1985-07-29"); /* TODO */
234
        break;
235
      case 'P':  /* $P = current drive and path */
236
      case 'p':
237
        _asm {
238
          mov ah, 0x19    /* DOS 1+ - GET CURRENT DRIVE */
239
          int 0x21
240
          mov bx, s
241
          mov [bx], al  /* AL = drive (00 = A:, 01 = B:, etc */
242
        }
243
        *s += 'A';
244
        s++;
245
        *s = ':';
246
        s++;
247
        *s = '\\';
248
        s++;
249
        _asm {
250
          mov ah, 0x47    /* DOS 2+ - CWD - GET CURRENT DIRECTORY */
251
          xor dl,dl       /* DL = drive number (00h = default, 01h = A:, etc) */
252
          mov si, s       /* DS:SI -> 64-byte buffer for ASCIZ pathname */
253
          int 0x21
526 mateuszvis 254
          jc DONE         /* leave path empty on error */
255
          /* move s ptr forward to end (0-termintor) of pathname */
256
          NEXTBYTE:
257
          mov si, s
258
          cmp byte ptr [si], 0
259
          je DONE
260
          inc s
261
          jmp NEXTBYTE
262
          DONE:
354 mateuszvis 263
        }
264
        break;
265
      case 'V':  /* $V = DOS version number */
266
      case 'v':
267
        s += sprintf(s, "VER"); /* TODO */
268
        break;
269
      case 'N':  /* $N = current drive */
270
      case 'n':
271
        _asm {
272
          mov ah, 0x19    /* DOS 1+ - GET CURRENT DRIVE */
273
          int 0x21
274
          mov bx, s
275
          mov [bx], al  /* AL = drive (00 = A:, 01 = B:, etc */
276
        }
277
        *s += 'A';
278
        s++;
279
        break;
280
      case 'G':  /* $G = > (greater-than sign) */
281
      case 'g':
282
        *s = '>';
283
        s++;
284
        break;
285
      case 'L':  /* $L = < (less-than sign) */
286
      case 'l':
287
        *s = '<';
288
        s++;
289
        break;
290
      case 'B':  /* $B = | (pipe) */
291
      case 'b':
292
        *s = '|';
293
        s++;
294
        break;
295
      case 'H':  /* $H = backspace (erases previous character) */
296
      case 'h':
297
        *s = '\b';
298
        s++;
299
        break;
300
      case 'E':  /* $E = Escape code (ASCII 27) */
301
      case 'e':
302
        *s = 27;
303
        s++;
304
        break;
305
      case '_':  /* $_ = CR+LF */
306
        *s = '\r';
307
        s++;
308
        *s = '\n';
309
        s++;
310
        break;
311
    }
312
  }
474 mateuszvis 313
  *s = 0;
314
  output(buff);
354 mateuszvis 315
}
349 mateuszvis 316
 
317
 
576 mateuszvis 318
static void run_as_external(char *buff, const char *cmdline, unsigned short envseg, struct rmod_props far *rmod, struct redir_data *redir, unsigned char delete_stdin_file) {
472 mateuszvis 319
  char *cmdfile = buff + 512;
458 mateuszvis 320
  const char far *pathptr;
321
  int lookup;
322
  unsigned short i;
323
  const char *ext;
508 mateuszvis 324
  char *cmd = buff + 1024;
479 mateuszvis 325
  const char *cmdtail;
461 mateuszvis 326
  char far *rmod_execprog = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPROG);
327
  char far *rmod_cmdtail = MK_FP(rmod->rmodseg, 0x81);
328
  _Packed struct {
329
    unsigned short envseg;
330
    unsigned long cmdtail;
331
    unsigned long fcb1;
332
    unsigned long fcb2;
333
  } far *ExecParam = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPARAM);
364 mateuszvis 334
 
472 mateuszvis 335
  /* find cmd and cmdtail */
336
  i = 0;
337
  cmdtail = cmdline;
338
  while (*cmdtail == ' ') cmdtail++; /* skip any leading spaces */
339
  while ((*cmdtail != ' ') && (*cmdtail != '/') && (*cmdtail != '+') && (*cmdtail != 0)) {
340
    cmd[i++] = *cmdtail;
341
    cmdtail++;
342
  }
343
  cmd[i] = 0;
364 mateuszvis 344
 
458 mateuszvis 345
  /* is this a command in curdir? */
472 mateuszvis 346
  lookup = lookup_cmd(cmdfile, cmd, NULL, &ext);
458 mateuszvis 347
  if (lookup == 0) {
348
    /* printf("FOUND LOCAL EXEC FILE: '%s'\r\n", cmdfile); */
349
    goto RUNCMDFILE;
350
  } else if (lookup == -2) {
351
    /* puts("NOT FOUND"); */
352
    return;
353
  }
354
 
355
  /* try matching something in PATH */
356
  pathptr = env_lookup_val(envseg, "PATH");
357
 
358
  /* try each path in %PATH% */
571 mateuszvis 359
  while (pathptr) {
458 mateuszvis 360
    for (i = 0;; i++) {
361
      buff[i] = *pathptr;
362
      if ((buff[i] == 0) || (buff[i] == ';')) break;
363
      pathptr++;
364
    }
365
    buff[i] = 0;
472 mateuszvis 366
    lookup = lookup_cmd(cmdfile, cmd, buff, &ext);
571 mateuszvis 367
    if (lookup == 0) goto RUNCMDFILE;
458 mateuszvis 368
    if (lookup == -2) return;
369
    if (*pathptr == ';') {
370
      pathptr++;
371
    } else {
571 mateuszvis 372
      break;
458 mateuszvis 373
    }
374
  }
375
 
571 mateuszvis 376
  /* last chance: is it an executable link? (trim extension from cmd first) */
377
  for (i = 0; (cmd[i] != 0) && (cmd[i] != '.') && (i < 9); i++) buff[128 + i] = cmd[i];
378
  buff[128 + i] = 0;
379
  if ((i < 9) && (link_computefname(buff, buff + 128, envseg) == 0)) {
380
    /* try opening the link file (if it exists) and read it into buff */
381
    i = 0;
382
    _asm {
383
      push ax
384
      push bx
385
      push cx
386
      push dx
387
 
388
      mov ax, 0x3d00  /* DOS 2+ - OPEN EXISTING FILE, READ-ONLY */
389
      mov dx, buff    /* file name */
390
      int 0x21
391
      jc ERR_FOPEN
392
      /* file handle in AX, read from file now */
393
      mov bx, ax      /* file handle */
394
      mov ah, 0x3f    /* Read from file via handle bx */
395
      mov cx, 128     /* up to 128 bytes */
396
      /* mov dx, buff */ /* dest buffer (already set) */
397
      int 0x21        /* read up to 256 bytes from file and write to buff */
398
      jc ERR_READ
399
      mov i, ax
400
      ERR_READ:
401
      mov ah, 0x3e    /* close file handle in BX */
402
      int 0x21
403
      ERR_FOPEN:
404
 
405
      pop dx
406
      pop cx
407
      pop bx
408
      pop ax
409
    }
410
 
411
    /* did I read anything? */
412
    if (i != 0) {
413
      buff[i] = 0;
414
      /* trim buff at first \n or \r, just in case someone fiddled with the
415
       * link file using a text editor */
416
      for (i = 0; (buff[i] != 0) && (buff[i] != '\r') && (buff[i] != '\n'); i++);
417
      buff[i] = 0;
418
      /* lookup check */
419
      if (buff[0] != 0) {
420
        lookup = lookup_cmd(cmdfile, cmd, buff, &ext);
421
        if (lookup == 0) goto RUNCMDFILE;
422
      }
423
    }
424
  }
425
 
426
  /* all failed (ie. executable file not found) */
427
  return;
428
 
458 mateuszvis 429
  RUNCMDFILE:
430
 
469 mateuszvis 431
  /* special handling of batch files */
432
  if ((ext != NULL) && (imatch(ext, "bat"))) {
433
    /* copy truename of the bat file to rmod buff */
548 mateuszvis 434
    _fstrcpy(rmod->batfile, cmdfile);
508 mateuszvis 435
 
436
    /* explode args of the bat file and store them in rmod buff */
437
    cmd_explode(buff, cmdline, NULL);
438
    _fmemcpy(rmod->batargv, buff, sizeof(rmod->batargv));
439
 
469 mateuszvis 440
    /* reset the 'next line to execute' counter */
441
    rmod->batnextline = 0;
474 mateuszvis 442
    /* remember the echo flag (in case bat file disables echo) */
443
    rmod->flags &= ~FLAG_ECHO_BEFORE_BAT;
475 mateuszvis 444
    if (rmod->flags & FLAG_ECHOFLAG) rmod->flags |= FLAG_ECHO_BEFORE_BAT;
469 mateuszvis 445
    return;
446
  }
447
 
517 mateuszvis 448
  /* copy full filename to execute, along with redirected files (if any) */
548 mateuszvis 449
  _fstrcpy(rmod_execprog, cmdfile);
450
 
451
  /* copy stdin file if a redirection is needed */
517 mateuszvis 452
  if (redir->stdinfile) {
548 mateuszvis 453
    char far *farptr = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDINFILE);
576 mateuszvis 454
    char far *delstdin = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDIN_DEL);
548 mateuszvis 455
    _fstrcpy(farptr, redir->stdinfile);
576 mateuszvis 456
    if (delete_stdin_file) {
457
      *delstdin = redir->stdinfile[0];
458
    } else {
459
      *delstdin = 0;
460
    }
517 mateuszvis 461
  }
548 mateuszvis 462
 
463
  /* same for stdout file */
517 mateuszvis 464
  if (redir->stdoutfile) {
548 mateuszvis 465
    char far *farptr = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDOUTFILE);
466
    unsigned short far *farptr16 = MK_FP(rmod->rmodseg, RMOD_OFFSET_STDOUTAPP);
467
    _fstrcpy(farptr, redir->stdoutfile);
517 mateuszvis 468
    /* openflag */
548 mateuszvis 469
    *farptr16 = redir->stdout_openflag;
517 mateuszvis 470
  }
461 mateuszvis 471
 
472
  /* copy cmdtail to rmod's PSP and compute its len */
473
  for (i = 0; cmdtail[i] != 0; i++) rmod_cmdtail[i] = cmdtail[i];
474
  rmod_cmdtail[i] = '\r';
475
  rmod_cmdtail[-1] = i;
476
 
477
  /* set up rmod to execute the command */
478
 
464 mateuszvis 479
  ExecParam->envseg = envseg;
461 mateuszvis 480
  ExecParam->cmdtail = (unsigned long)MK_FP(rmod->rmodseg, 0x80); /* farptr, must be in PSP format (lenbyte args \r) */
481
  ExecParam->fcb1 = 0; /* TODO farptr */
482
  ExecParam->fcb2 = 0; /* TODO farptr */
483
  exit(0); /* let rmod do the job now */
364 mateuszvis 484
}
485
 
486
 
367 mateuszvis 487
static void set_comspec_to_self(unsigned short envseg) {
488
  unsigned short *psp_envseg = (void *)(0x2c); /* pointer to my env segment field in the PSP */
489
  char far *myenv = MK_FP(*psp_envseg, 0);
490
  unsigned short varcount;
491
  char buff[256] = "COMSPEC=";
492
  char *buffptr = buff + 8;
493
  /* who am i? look into my own environment, at the end of it should be my EXEPATH string */
494
  while (*myenv != 0) {
495
    /* consume a NULL-terminated string */
496
    while (*myenv != 0) myenv++;
497
    /* move to next string */
498
    myenv++;
499
  }
500
  /* get next word, if 1 then EXEPATH follows */
501
  myenv++;
502
  varcount = *myenv;
503
  myenv++;
504
  varcount |= (*myenv << 8);
505
  myenv++;
506
  if (varcount != 1) return; /* NO EXEPATH FOUND */
507
  while (*myenv != 0) {
508
    *buffptr = *myenv;
509
    buffptr++;
510
    myenv++;
511
  }
512
  *buffptr = 0;
513
  /* printf("EXEPATH: '%s'\r\n", buff); */
514
  env_setvar(envseg, buff);
515
}
516
 
517
 
450 mateuszvis 518
/* wait for user input */
519
static void cmdline_getinput(unsigned short inpseg, unsigned short inpoff) {
520
  _asm {
521
    push ax
522
    push bx
523
    push cx
524
    push dx
525
    push ds
526
 
527
    /* is DOSKEY support present? (INT 2Fh, AX=4800h, returns non-zero in AL if present) */
528
    mov ax, 0x4800
529
    int 0x2f
530
    mov bl, al /* save doskey status in BL */
531
 
532
    /* set up buffered input to inpseg:inpoff */
533
    mov ax, inpseg
534
    push ax
535
    pop ds
536
    mov dx, inpoff
537
 
538
    /* execute either DOS input or DOSKEY */
539
    test bl, bl /* zf set if no DOSKEY present */
540
    jnz DOSKEY
541
 
542
    mov ah, 0x0a
543
    int 0x21
544
    jmp short DONE
545
 
546
    DOSKEY:
547
    mov ax, 0x4810
548
    int 0x2f
549
 
550
    DONE:
551
    /* terminate command with a CR/LF */
552
    mov ah, 0x02 /* display character in dl */
553
    mov dl, 0x0d
554
    int 0x21
555
    mov dl, 0x0a
556
    int 0x21
557
 
558
    pop ds
559
    pop dx
560
    pop cx
561
    pop bx
562
    pop ax
563
  }
564
}
565
 
566
 
479 mateuszvis 567
/* fetches a line from batch file and write it to buff (NULL-terminated),
568
 * increments rmod counter and returns 0 on success. */
484 mateuszvis 569
static int getbatcmd(char *buff, unsigned char buffmaxlen, struct rmod_props far *rmod) {
469 mateuszvis 570
  unsigned short i;
474 mateuszvis 571
  unsigned short batname_seg = FP_SEG(rmod->batfile);
572
  unsigned short batname_off = FP_OFF(rmod->batfile);
573
  unsigned short filepos_cx = rmod->batnextline >> 16;
574
  unsigned short filepos_dx = rmod->batnextline & 0xffff;
575
  unsigned char blen = 0;
505 mateuszvis 576
  unsigned short errv = 0;
474 mateuszvis 577
 
578
  /* open file, jump to offset filpos, and read data into buff.
579
   * result in blen (unchanged if EOF or failure). */
580
  _asm {
581
    push ax
582
    push bx
583
    push cx
584
    push dx
585
 
586
    /* open file (read-only) */
505 mateuszvis 587
    mov bx, 0xffff        /* preset BX to 0xffff to detect error conditions */
474 mateuszvis 588
    mov dx, batname_off
589
    mov ax, batname_seg
590
    push ds     /* save DS */
591
    mov ds, ax
592
    mov ax, 0x3d00
593
    int 0x21    /* handle in ax on success */
594
    pop ds      /* restore DS */
505 mateuszvis 595
    jc ERR
474 mateuszvis 596
    mov bx, ax  /* save handle to bx */
597
 
598
    /* jump to file offset CX:DX */
599
    mov ax, 0x4200
600
    mov cx, filepos_cx
601
    mov dx, filepos_dx
602
    int 0x21  /* CF clear on success, DX:AX set to cur pos */
505 mateuszvis 603
    jc ERR
474 mateuszvis 604
 
605
    /* read the line into buff */
606
    mov ah, 0x3f
484 mateuszvis 607
    xor ch, ch
608
    mov cl, buffmaxlen
474 mateuszvis 609
    mov dx, buff
610
    int 0x21 /* CF clear on success, AX=number of bytes read */
505 mateuszvis 611
    jc ERR
474 mateuszvis 612
    mov blen, al
505 mateuszvis 613
    jmp CLOSEANDQUIT
474 mateuszvis 614
 
505 mateuszvis 615
    ERR:
616
    mov errv, ax
617
 
474 mateuszvis 618
    CLOSEANDQUIT:
505 mateuszvis 619
    /* close file (if bx contains a handle) */
620
    cmp bx, 0xffff
621
    je DONE
474 mateuszvis 622
    mov ah, 0x3e
623
    int 0x21
624
 
625
    DONE:
626
    pop dx
627
    pop cx
628
    pop bx
629
    pop ax
469 mateuszvis 630
  }
470 mateuszvis 631
 
474 mateuszvis 632
  /* printf("blen=%u filepos_cx=%u filepos_dx=%u\r\n", blen, filepos_cx, filepos_dx); */
470 mateuszvis 633
 
538 mateuszvis 634
  if (errv != 0) nls_outputnl_doserr(errv);
505 mateuszvis 635
 
474 mateuszvis 636
  /* on EOF - abort processing the bat file */
637
  if (blen == 0) goto OOPS;
638
 
639
  /* find nearest \n to inc batch offset and replace \r by NULL terminator
640
   * I support all CR/LF, CR- and LF-terminated batch files */
641
  for (i = 0; i < blen; i++) {
642
    if ((buff[i] == '\r') || (buff[i] == '\n')) {
643
      if ((buff[i] == '\r') && ((i+1) < blen) && (buff[i+1] == '\n')) rmod->batnextline += 1;
644
      break;
645
    }
646
  }
647
  buff[i] = 0;
648
  rmod->batnextline += i + 1;
649
 
650
  return(0);
651
 
652
  OOPS:
653
  rmod->batfile[0] = 0;
654
  rmod->batnextline = 0;
655
  return(-1);
469 mateuszvis 656
}
657
 
658
 
507 mateuszvis 659
/* replaces %-variables in a BAT line with resolved values:
660
 * %PATH%       -> replaced by the contend of the PATH env variable
661
 * %UNDEFINED%  -> undefined variables are replaced by nothing ("")
662
 * %NOTCLOSED   -> NOTCLOSED
663
 * %1           -> first argument of the batch file (or nothing if no arg) */
664
static void batpercrepl(char *res, unsigned short ressz, const char *line, const struct rmod_props far *rmod, unsigned short envseg) {
665
  unsigned short lastperc = 0xffff;
666
  unsigned short reslen = 0;
667
 
668
  if (ressz == 0) return;
669
  ressz--; /* reserve one byte for the NULL terminator */
670
 
671
  for (; (reslen < ressz) && (*line != 0); line++) {
672
    /* if not a percent, I don't care */
673
    if (*line != '%') {
674
      res[reslen++] = *line;
675
      continue;
676
    }
677
 
678
    /* *** perc char handling *** */
679
 
680
    /* closing perc? */
681
    if (lastperc != 0xffff) {
682
      /* %% is '%' */
683
      if (lastperc == reslen) {
684
        res[reslen++] = '%';
685
      } else {   /* otherwise variable name */
686
        const char far *ptr;
687
        res[reslen] = 0;
688
        reslen = lastperc;
689
        ptr = env_lookup_val(envseg, res + reslen);
690
        if (ptr != NULL) {
691
          while ((*ptr != 0) && (reslen < ressz)) {
692
            res[reslen++] = *ptr;
693
            ptr++;
694
          }
695
        }
696
      }
697
      lastperc = 0xffff;
698
      continue;
699
    }
700
 
701
    /* digit? (bat arg) */
702
    if ((line[1] >= '0') && (line[1] <= '9')) {
508 mateuszvis 703
      unsigned short argid = line[1] - '0';
704
      unsigned short i;
705
      const char far *argv = rmod->batargv;
706
 
707
      /* locate the proper arg */
708
      for (i = 0; i != argid; i++) {
709
        /* if string is 0, then end of list reached */
710
        if (*argv == 0) break;
711
        /* jump to next arg */
712
        while (*argv != 0) argv++;
713
        argv++;
714
      }
715
 
716
      /* copy the arg to result */
717
      for (i = 0; (argv[i] != 0) && (reslen < ressz); i++) {
718
        res[reslen++] = argv[i];
719
      }
507 mateuszvis 720
      line++;  /* skip the digit */
721
      continue;
722
    }
723
 
724
    /* opening perc */
725
    lastperc = reslen;
726
 
727
  }
728
 
729
  res[reslen] = 0;
730
}
731
 
732
 
443 mateuszvis 733
int main(void) {
372 mateuszvis 734
  static struct config cfg;
735
  static unsigned short far *rmod_envseg;
736
  static unsigned short far *lastexitcode;
449 mateuszvis 737
  static struct rmod_props far *rmod;
500 mateuszvis 738
  static char cmdlinebuf[CMDLINE_MAXLEN + 2]; /* 1 extra byte for 0-terminator and another for memguard */
479 mateuszvis 739
  static char *cmdline;
517 mateuszvis 740
  static struct redir_data redirprops;
533 mateuszvis 741
  static enum cmd_result cmdres;
543 mateuszvis 742
  static unsigned short i; /* general-purpose variable for short-lived things */
576 mateuszvis 743
  static unsigned char delete_stdin_file;
349 mateuszvis 744
 
479 mateuszvis 745
  rmod = rmod_find(BUFFER_len);
449 mateuszvis 746
  if (rmod == NULL) {
485 mateuszvis 747
    /* look at command line parameters (in case env size if set there) */
748
    parse_argv(&cfg);
479 mateuszvis 749
    rmod = rmod_install(cfg.envsiz, BUFFER, BUFFER_len);
449 mateuszvis 750
    if (rmod == NULL) {
369 mateuszvis 751
      outputnl("ERROR: rmod_install() failed");
349 mateuszvis 752
      return(1);
753
    }
475 mateuszvis 754
    /* copy flags to rmod's storage (and enable ECHO) */
755
    rmod->flags = cfg.flags | FLAG_ECHOFLAG;
465 mateuszvis 756
    /* printf("rmod installed at %Fp\r\n", rmod); */
572 mateuszvis 757
    rmod->version = BYTE_VERSION;
349 mateuszvis 758
  } else {
465 mateuszvis 759
    /* printf("rmod found at %Fp\r\n", rmod); */
760
    /* if I was spawned by rmod and FLAG_EXEC_AND_QUIT is set, then I should
761
     * die asap, because the command has been executed already, so I no longer
762
     * have a purpose in life */
469 mateuszvis 763
    if (rmod->flags & FLAG_EXEC_AND_QUIT) sayonara(rmod);
572 mateuszvis 764
    /* */
765
    if (rmod->version != BYTE_VERSION) {
766
      outputnl("SVARCOM VERSION CHANGED. SYSTEM HALTED. PLEASE REBOOT YOUR COMPUTER.");
767
      _asm {
768
        HALT:
769
        hlt
770
        jmp HALT
771
      }
772
    }
349 mateuszvis 773
  }
774
 
490 mateuszvis 775
  /* install a few guardvals in memory to detect some cases of overflows */
500 mateuszvis 776
  memguard_set(cmdlinebuf);
490 mateuszvis 777
 
465 mateuszvis 778
  rmod_envseg = MK_FP(rmod->rmodseg, RMOD_OFFSET_ENVSEG);
779
  lastexitcode = MK_FP(rmod->rmodseg, RMOD_OFFSET_LEXITCODE);
350 mateuszvis 780
 
367 mateuszvis 781
  /* make COMPSEC point to myself */
782
  set_comspec_to_self(*rmod_envseg);
783
 
369 mateuszvis 784
/*  {
350 mateuszvis 785
    unsigned short envsiz;
786
    unsigned short far *sizptr = MK_FP(*rmod_envseg - 1, 3);
787
    envsiz = *sizptr;
788
    envsiz *= 16;
465 mateuszvis 789
    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);
369 mateuszvis 790
  }*/
349 mateuszvis 791
 
494 mateuszvis 792
  /* on /P check for the presence of AUTOEXEC.BAT and execute it if found,
793
   * but skip this check if /D was also passed */
794
  if ((cfg.flags & (FLAG_PERMANENT | FLAG_SKIP_AUTOEXEC)) == FLAG_PERMANENT) {
483 mateuszvis 795
    if (file_getattr("AUTOEXEC.BAT") >= 0) cfg.execcmd = "AUTOEXEC.BAT";
796
  }
797
 
443 mateuszvis 798
  do {
480 mateuszvis 799
    /* terminate previous command with a CR/LF if ECHO ON (but not during BAT processing) */
483 mateuszvis 800
    if ((rmod->flags & FLAG_ECHOFLAG) && (rmod->batfile[0] == 0)) outputnl("");
474 mateuszvis 801
 
802
    SKIP_NEWLINE:
803
 
490 mateuszvis 804
    /* memory check */
500 mateuszvis 805
    memguard_check(rmod->rmodseg, cmdlinebuf);
474 mateuszvis 806
 
500 mateuszvis 807
    /* preset cmdline to point at the dedicated buffer */
808
    cmdline = cmdlinebuf;
490 mateuszvis 809
 
437 mateuszvis 810
    /* (re)load translation strings if needed */
811
    nls_langreload(BUFFER, *rmod_envseg);
812
 
543 mateuszvis 813
    /* load awaiting command, if any (used to run piped commands) */
814
    if (rmod->awaitingcmd[0] != 0) {
815
      _fstrcpy(cmdline, rmod->awaitingcmd);
816
      rmod->awaitingcmd[0] = 0;
576 mateuszvis 817
      delete_stdin_file = 1;
543 mateuszvis 818
      goto EXEC_CMDLINE;
576 mateuszvis 819
    } else {
820
      delete_stdin_file = 0;
543 mateuszvis 821
    }
822
 
443 mateuszvis 823
    /* skip user input if I have a command to exec (/C or /K) */
824
    if (cfg.execcmd != NULL) {
825
      cmdline = cfg.execcmd;
826
      cfg.execcmd = NULL;
827
      goto EXEC_CMDLINE;
828
    }
829
 
469 mateuszvis 830
    /* if batch file is being executed -> fetch next line */
831
    if (rmod->batfile[0] != 0) {
507 mateuszvis 832
      if (getbatcmd(BUFFER, CMDLINE_MAXLEN, rmod) != 0) { /* end of batch */
474 mateuszvis 833
        /* restore echo flag as it was before running the bat file */
475 mateuszvis 834
        rmod->flags &= ~FLAG_ECHOFLAG;
835
        if (rmod->flags & FLAG_ECHO_BEFORE_BAT) rmod->flags |= FLAG_ECHOFLAG;
474 mateuszvis 836
        continue;
837
      }
507 mateuszvis 838
      /* %-decoding of variables (%PATH%, %1, %%...), result in cmdline */
839
      batpercrepl(cmdline, CMDLINE_MAXLEN, BUFFER, rmod, *rmod_envseg);
480 mateuszvis 840
      /* skip any leading spaces */
841
      while (*cmdline == ' ') cmdline++;
474 mateuszvis 842
      /* output prompt and command on screen if echo on and command is not
843
       * inhibiting it with the @ prefix */
479 mateuszvis 844
      if ((rmod->flags & FLAG_ECHOFLAG) && (cmdline[0] != '@')) {
474 mateuszvis 845
        build_and_display_prompt(BUFFER, *rmod_envseg);
479 mateuszvis 846
        outputnl(cmdline);
474 mateuszvis 847
      }
479 mateuszvis 848
      /* skip the @ prefix if present, it is no longer useful */
849
      if (cmdline[0] == '@') cmdline++;
469 mateuszvis 850
    } else {
474 mateuszvis 851
      /* interactive mode: display prompt (if echo enabled) and wait for user
852
       * command line */
475 mateuszvis 853
      if (rmod->flags & FLAG_ECHOFLAG) build_and_display_prompt(BUFFER, *rmod_envseg);
474 mateuszvis 854
      /* collect user input */
469 mateuszvis 855
      cmdline_getinput(FP_SEG(rmod->inputbuf), FP_OFF(rmod->inputbuf));
479 mateuszvis 856
      /* copy it to local cmdline */
857
      if (rmod->inputbuf[1] != 0) _fmemcpy(cmdline, rmod->inputbuf + 2, rmod->inputbuf[1]);
858
      cmdline[(unsigned)(rmod->inputbuf[1])] = 0; /* zero-terminate local buff (oriignal is '\r'-terminated) */
469 mateuszvis 859
    }
349 mateuszvis 860
 
405 mateuszvis 861
    /* if nothing entered, loop again (but without appending an extra CR/LF) */
479 mateuszvis 862
    if (cmdline[0] == 0) goto SKIP_NEWLINE;
349 mateuszvis 863
 
443 mateuszvis 864
    /* I jump here when I need to exec an initial command (/C or /K) */
865
    EXEC_CMDLINE:
866
 
364 mateuszvis 867
    /* move pointer forward to skip over any leading spaces */
868
    while (*cmdline == ' ') cmdline++;
349 mateuszvis 869
 
367 mateuszvis 870
    /* update rmod's ptr to COMPSPEC so it is always up to date */
465 mateuszvis 871
    rmod_updatecomspecptr(rmod->rmodseg, *rmod_envseg);
367 mateuszvis 872
 
402 mateuszvis 873
    /* handle redirections (if any) */
543 mateuszvis 874
    i = redir_parsecmd(&redirprops, cmdline, rmod->awaitingcmd);
875
    if (i != 0) {
876
      nls_outputnl_doserr(i);
877
      rmod->awaitingcmd[0] = 0;
878
      continue;
879
    }
402 mateuszvis 880
 
364 mateuszvis 881
    /* try matching (and executing) an internal command */
576 mateuszvis 882
    cmdres = cmd_process(rmod, *rmod_envseg, cmdline, BUFFER, sizeof(BUFFER), &redirprops, delete_stdin_file);
533 mateuszvis 883
    if ((cmdres == CMD_OK) || (cmdres == CMD_FAIL)) {
443 mateuszvis 884
      /* internal command executed */
885
      continue;
533 mateuszvis 886
    } else if (cmdres == CMD_CHANGED) { /* cmdline changed, needs to be reprocessed */
887
      goto EXEC_CMDLINE;
888
    } else if (cmdres == CMD_NOTFOUND) {
889
      /* this was not an internal command, try matching an external command */
576 mateuszvis 890
      run_as_external(BUFFER, cmdline, *rmod_envseg, rmod, &redirprops, delete_stdin_file);
533 mateuszvis 891
      /* perhaps this is a newly launched BAT file */
892
      if ((rmod->batfile[0] != 0) && (rmod->batnextline == 0)) goto SKIP_NEWLINE;
893
      /* run_as_external() does not return on success, if I am still alive then
894
       * external command failed to execute */
895
      outputnl("Bad command or file name");
896
      continue;
353 mateuszvis 897
    }
352 mateuszvis 898
 
533 mateuszvis 899
    /* I should never ever land here */
900
    outputnl("INTERNAL ERR: INVALID CMDRES");
349 mateuszvis 901
 
449 mateuszvis 902
  } while ((rmod->flags & FLAG_EXEC_AND_QUIT) == 0);
349 mateuszvis 903
 
449 mateuszvis 904
  sayonara(rmod);
349 mateuszvis 905
  return(0);
906
}