Subversion Repositories SvarDOS

Rev

Rev 483 | Rev 485 | 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
 *
4
 * Copyright (C) 2021 Mateusz Viste
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
/*
26
 * SvarCOM is a command-line interpreter.
27
 *
28
 * a little memory area is allocated as high as possible. it contains:
29
 *  - a signature (like XMS drivers do)
30
 *  - a routine for exec'ing programs
31
 *  - a "last command" buffer for input history
32
 *
33
 * when svarcom starts, it tries locating the routine in memory.
34
 *
35
 * if found:
36
 *   waits for user input and processes it. if execing something is required, set the "next exec" field in routine's memory and quit.
37
 *
38
 * if not found:
39
 *   installs it by creating a new PSP, set int 22 vector to the routine, set my "parent PSP" to the routine
40
 *   and quit.
41
 *
42
 * PSP structure
43
 * http://www.piclist.com/techref/dos/psps.htm
44
 *
45
 *
46
 *
47
 * === MCB ===
48
 *
49
 * each time that DOS allocates memory, it prefixes the allocated memory with
50
 * a 16-bytes structure called a "Memory Control Block" (MCB). This control
51
 * block has the following structure:
52
 *
53
 * Offset  Size     Description
54
 *   00h   byte     'M' =  non-last member of the MCB chain
55
 *                  'Z' = indicates last entry in MCB chain
56
 *                  other values cause "Memory Allocation Failure" on exit
57
 *   01h   word     PSP segment address of the owner (Process Id)
58
 *                  possible values:
59
 *                    0 = free
60
 *                    8 = Allocated by DOS before first user pgm loaded
61
 *                    other = Process Id/PSP segment address of owner
62
 *   03h  word      number of paragraphs related to this MCB (excluding MCB)
63
 *   05h  11 bytes  reserved
64
 *   10h  ...       start of actual allocated memory block
65
 */
66
 
67
#include <i86.h>
68
#include <dos.h>
69
#include <stdio.h>
350 mateuszvis 70
#include <stdlib.h>
349 mateuszvis 71
#include <string.h>
72
 
73
#include <process.h>
74
 
352 mateuszvis 75
#include "cmd.h"
366 mateuszvis 76
#include "env.h"
352 mateuszvis 77
#include "helpers.h"
402 mateuszvis 78
#include "redir.h"
351 mateuszvis 79
#include "rmodinit.h"
448 mateuszvis 80
#include "sayonara.h"
349 mateuszvis 81
 
479 mateuszvis 82
#include "rmodcore.h" /* rmod binary inside a BUFFER array */
443 mateuszvis 83
 
349 mateuszvis 84
struct config {
449 mateuszvis 85
  unsigned char flags; /* command.com flags, as defined in rmodinit.h */
443 mateuszvis 86
  char *execcmd;
410 mateuszvis 87
  unsigned short envsiz;
443 mateuszvis 88
};
349 mateuszvis 89
 
90
 
443 mateuszvis 91
/* parses command line the hard way (directly from PSP) */
92
static void parse_argv(struct config *cfg) {
93
  unsigned short i;
94
  const unsigned char *cmdlinelen = (unsigned char *)0x80;
95
  char *cmdline = (char *)0x81;
96
 
349 mateuszvis 97
  memset(cfg, 0, sizeof(*cfg));
350 mateuszvis 98
 
443 mateuszvis 99
  /* set a NULL terminator on cmdline */
100
  cmdline[*cmdlinelen] = 0;
101
 
102
  for (i = 0;;) {
103
 
104
    /* skip over any leading spaces */
105
    for (;; i++) {
106
      if (cmdline[i] == 0) return;
107
      if (cmdline[i] != ' ') break;
349 mateuszvis 108
    }
443 mateuszvis 109
 
110
    if (cmdline[i] != '/') {
111
      output("Invalid parameter: ");
112
      outputnl(cmdline + i);
113
      /* exit(1); */
114
    } else {
115
      i++;        /* skip the slash */
116
      switch (cmdline[i]) {
117
        case 'c': /* /C = execute command and quit */
118
        case 'C':
119
          cfg->flags |= FLAG_EXEC_AND_QUIT;
120
          /* FALLTHRU */
121
        case 'k': /* /K = execute command and keep running */
122
        case 'K':
123
          cfg->execcmd = cmdline + i + 1;
124
          return;
125
 
126
        case 'e': /* preset the initial size of the environment block */
127
        case 'E':
128
          i++;
129
          if (cmdline[i] == ':') i++; /* could be /E:size */
130
          atous(&(cfg->envsiz), cmdline + i);
131
          if (cfg->envsiz < 64) cfg->envsiz = 0;
132
          break;
133
 
483 mateuszvis 134
        case 'p': /* permanent shell (can't exit + run autoexec.bat) */
449 mateuszvis 135
        case 'P':
136
          cfg->flags |= FLAG_PERMANENT;
137
          break;
138
 
444 mateuszvis 139
        case '?':
140
          outputnl("Starts the SvarCOM command interpreter");
141
          outputnl("");
142
          outputnl("COMMAND /E:nnn [/[C|K] command]");
143
          outputnl("");
144
          outputnl("/E:nnn     Sets the environment size to nnn bytes");
449 mateuszvis 145
          outputnl("/P         Makes the new command interpreter permanent (can't exit)");
444 mateuszvis 146
          outputnl("/C         Executes the specified command and returns");
147
          outputnl("/K         Executes the specified command and continues running");
148
          exit(1);
149
          break;
150
 
443 mateuszvis 151
        default:
152
          output("Invalid switch:");
153
          output(" ");
154
          outputnl(cmdline + i);
155
          exit(1);
156
          break;
410 mateuszvis 157
      }
350 mateuszvis 158
    }
443 mateuszvis 159
 
160
    /* move to next argument or quit processing if end of cmdline */
161
    for (i++; (cmdline[i] != 0) && (cmdline[i] != ' ') && (cmdline[i] != '/'); i++);
162
 
349 mateuszvis 163
  }
164
}
165
 
166
 
474 mateuszvis 167
/* builds the prompt string and displays it. buff is filled with a zero-terminated copy of the prompt. */
168
static void build_and_display_prompt(char *buff, unsigned short envseg) {
169
  char *s = buff;
370 mateuszvis 170
  /* locate the prompt variable or use the default pattern */
438 mateuszvis 171
  const char far *fmt = env_lookup_val(envseg, "PROMPT");
370 mateuszvis 172
  if ((fmt == NULL) || (*fmt == 0)) fmt = "$p$g"; /* fallback to default if empty */
173
  /* build the prompt string based on pattern */
354 mateuszvis 174
  for (; *fmt != 0; fmt++) {
175
    if (*fmt != '$') {
176
      *s = *fmt;
177
      s++;
178
      continue;
179
    }
180
    /* escape code ($P, etc) */
181
    fmt++;
182
    switch (*fmt) {
183
      case 'Q':  /* $Q = = (equal sign) */
184
      case 'q':
185
        *s = '=';
186
        s++;
187
        break;
188
      case '$':  /* $$ = $ (dollar sign) */
189
        *s = '$';
190
        s++;
191
        break;
192
      case 'T':  /* $t = current time */
193
      case 't':
194
        s += sprintf(s, "00:00"); /* TODO */
195
        break;
196
      case 'D':  /* $D = current date */
197
      case 'd':
198
        s += sprintf(s, "1985-07-29"); /* TODO */
199
        break;
200
      case 'P':  /* $P = current drive and path */
201
      case 'p':
202
        _asm {
203
          mov ah, 0x19    /* DOS 1+ - GET CURRENT DRIVE */
204
          int 0x21
205
          mov bx, s
206
          mov [bx], al  /* AL = drive (00 = A:, 01 = B:, etc */
207
        }
208
        *s += 'A';
209
        s++;
210
        *s = ':';
211
        s++;
212
        *s = '\\';
213
        s++;
214
        _asm {
215
          mov ah, 0x47    /* DOS 2+ - CWD - GET CURRENT DIRECTORY */
216
          xor dl,dl       /* DL = drive number (00h = default, 01h = A:, etc) */
217
          mov si, s       /* DS:SI -> 64-byte buffer for ASCIZ pathname */
218
          int 0x21
219
        }
220
        while (*s != 0) s++; /* move ptr forward to end of pathname */
221
        break;
222
      case 'V':  /* $V = DOS version number */
223
      case 'v':
224
        s += sprintf(s, "VER"); /* TODO */
225
        break;
226
      case 'N':  /* $N = current drive */
227
      case 'n':
228
        _asm {
229
          mov ah, 0x19    /* DOS 1+ - GET CURRENT DRIVE */
230
          int 0x21
231
          mov bx, s
232
          mov [bx], al  /* AL = drive (00 = A:, 01 = B:, etc */
233
        }
234
        *s += 'A';
235
        s++;
236
        break;
237
      case 'G':  /* $G = > (greater-than sign) */
238
      case 'g':
239
        *s = '>';
240
        s++;
241
        break;
242
      case 'L':  /* $L = < (less-than sign) */
243
      case 'l':
244
        *s = '<';
245
        s++;
246
        break;
247
      case 'B':  /* $B = | (pipe) */
248
      case 'b':
249
        *s = '|';
250
        s++;
251
        break;
252
      case 'H':  /* $H = backspace (erases previous character) */
253
      case 'h':
254
        *s = '\b';
255
        s++;
256
        break;
257
      case 'E':  /* $E = Escape code (ASCII 27) */
258
      case 'e':
259
        *s = 27;
260
        s++;
261
        break;
262
      case '_':  /* $_ = CR+LF */
263
        *s = '\r';
264
        s++;
265
        *s = '\n';
266
        s++;
267
        break;
268
    }
269
  }
474 mateuszvis 270
  *s = 0;
271
  output(buff);
354 mateuszvis 272
}
349 mateuszvis 273
 
274
 
458 mateuszvis 275
/* tries locating executable fname in path and fille res with result. returns 0 on success,
276
 * -1 on failed match and -2 on failed match + "don't even try with other paths"
277
 * format is filled the offset where extension starts in fname (-1 if not found) */
278
int lookup_cmd(char *res, const char *fname, const char *path, const char **extptr) {
279
  unsigned short lastbslash = 0xffff;
280
  unsigned short i, len;
281
  unsigned char explicitpath = 0;
282
 
283
  /* does the original fname had an explicit path prefix or explicit ext? */
284
  *extptr = NULL;
285
  for (i = 0; fname[i] != 0; i++) {
286
    switch (fname[i]) {
287
      case ':':
288
      case '\\':
289
        explicitpath = 1;
290
        *extptr = NULL; /* extension is the last dot AFTER all path delimiters */
291
        break;
292
      case '.':
293
        *extptr = fname + i + 1;
294
        break;
295
    }
296
  }
297
 
298
  /* normalize filename */
299
  if (file_truename(fname, res) != 0) return(-2);
300
 
301
  /* printf("truename: %s\r\n", res); */
302
 
303
  /* figure out where the command starts and if it has an explicit extension */
304
  for (len = 0; res[len] != 0; len++) {
305
    switch (res[len]) {
306
      case '?':   /* abort on any wildcard character */
307
      case '*':
308
        return(-2);
309
      case '\\':
310
        lastbslash = len;
311
        break;
312
    }
313
  }
314
 
315
  /* printf("lastbslash=%u\r\n", lastbslash); */
316
 
317
  /* if no path prefix in fname (':' or backslash), then assemble path+filename */
318
  if (!explicitpath) {
319
    if (path != NULL) {
320
      i = strlen(path);
321
    } else {
322
      i = 0;
323
    }
324
    if ((i != 0) && (path[i - 1] != '\\')) i++; /* add a byte for inserting a bkslash after path */
325
    memmove(res + i, res + lastbslash + 1, len - lastbslash);
326
    if (i != 0) {
327
      memmove(res, path, i);
328
      res[i - 1] = '\\';
329
    }
330
  }
331
 
332
  /* if no extension was initially provided, try matching COM, EXE, BAT */
333
  if (*extptr == NULL) {
334
    const char *ext[] = {".COM", ".EXE", ".BAT", NULL};
335
    len = strlen(res);
336
    for (i = 0; ext[i] != NULL; i++) {
337
      strcpy(res + len, ext[i]);
338
      /* printf("? '%s'\r\n", res); */
339
      *extptr = ext[i] + 1;
340
      if (file_getattr(res) >= 0) return(0);
341
    }
342
  } else { /* try finding it as-is */
343
    /* printf("? '%s'\r\n", res); */
344
    if (file_getattr(res) >= 0) return(0);
345
  }
346
 
347
  /* not found */
348
  if (explicitpath) return(-2); /* don't bother trying other paths, the caller had its own path preset anyway */
349
  return(-1);
350
}
351
 
352
 
479 mateuszvis 353
static void run_as_external(char *buff, const char *cmdline, unsigned short envseg, struct rmod_props far *rmod) {
472 mateuszvis 354
  char *cmdfile = buff + 512;
458 mateuszvis 355
  const char far *pathptr;
356
  int lookup;
357
  unsigned short i;
358
  const char *ext;
472 mateuszvis 359
  char *cmd = buff + 256;
479 mateuszvis 360
  const char *cmdtail;
461 mateuszvis 361
  char far *rmod_execprog = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPROG);
362
  char far *rmod_cmdtail = MK_FP(rmod->rmodseg, 0x81);
363
  _Packed struct {
364
    unsigned short envseg;
365
    unsigned long cmdtail;
366
    unsigned long fcb1;
367
    unsigned long fcb2;
368
  } far *ExecParam = MK_FP(rmod->rmodseg, RMOD_OFFSET_EXECPARAM);
364 mateuszvis 369
 
472 mateuszvis 370
  /* find cmd and cmdtail */
371
  i = 0;
372
  cmdtail = cmdline;
373
  while (*cmdtail == ' ') cmdtail++; /* skip any leading spaces */
374
  while ((*cmdtail != ' ') && (*cmdtail != '/') && (*cmdtail != '+') && (*cmdtail != 0)) {
375
    cmd[i++] = *cmdtail;
376
    cmdtail++;
377
  }
378
  cmd[i] = 0;
364 mateuszvis 379
 
458 mateuszvis 380
  /* is this a command in curdir? */
472 mateuszvis 381
  lookup = lookup_cmd(cmdfile, cmd, NULL, &ext);
458 mateuszvis 382
  if (lookup == 0) {
383
    /* printf("FOUND LOCAL EXEC FILE: '%s'\r\n", cmdfile); */
384
    goto RUNCMDFILE;
385
  } else if (lookup == -2) {
386
    /* puts("NOT FOUND"); */
387
    return;
388
  }
389
 
390
  /* try matching something in PATH */
391
  pathptr = env_lookup_val(envseg, "PATH");
392
  if (pathptr == NULL) return;
393
 
394
  /* try each path in %PATH% */
395
  for (;;) {
396
    for (i = 0;; i++) {
397
      buff[i] = *pathptr;
398
      if ((buff[i] == 0) || (buff[i] == ';')) break;
399
      pathptr++;
400
    }
401
    buff[i] = 0;
472 mateuszvis 402
    lookup = lookup_cmd(cmdfile, cmd, buff, &ext);
458 mateuszvis 403
    if (lookup == 0) break;
404
    if (lookup == -2) return;
405
    if (*pathptr == ';') {
406
      pathptr++;
407
    } else {
408
      return;
409
    }
410
  }
411
 
412
  RUNCMDFILE:
413
 
469 mateuszvis 414
  /* special handling of batch files */
415
  if ((ext != NULL) && (imatch(ext, "bat"))) {
416
    /* copy truename of the bat file to rmod buff */
417
    for (i = 0; cmdfile[i] != 0; i++) rmod->batfile[i] = cmdfile[i];
418
    rmod->batfile[i] = 0;
419
    /* copy args of the bat file to rmod buff */
420
    for (i = 0; cmdtail[i] != 0; i++) rmod->batargs[i] = cmdtail[i];
421
    /* reset the 'next line to execute' counter */
422
    rmod->batnextline = 0;
474 mateuszvis 423
    /* remember the echo flag (in case bat file disables echo) */
424
    rmod->flags &= ~FLAG_ECHO_BEFORE_BAT;
475 mateuszvis 425
    if (rmod->flags & FLAG_ECHOFLAG) rmod->flags |= FLAG_ECHO_BEFORE_BAT;
469 mateuszvis 426
    return;
427
  }
428
 
461 mateuszvis 429
  /* copy full filename to execute */
430
  for (i = 0; cmdfile[i] != 0; i++) rmod_execprog[i] = cmdfile[i];
431
  rmod_execprog[i] = 0;
432
 
433
  /* copy cmdtail to rmod's PSP and compute its len */
434
  for (i = 0; cmdtail[i] != 0; i++) rmod_cmdtail[i] = cmdtail[i];
435
  rmod_cmdtail[i] = '\r';
436
  rmod_cmdtail[-1] = i;
437
 
438
  /* set up rmod to execute the command */
439
 
464 mateuszvis 440
  ExecParam->envseg = envseg;
461 mateuszvis 441
  ExecParam->cmdtail = (unsigned long)MK_FP(rmod->rmodseg, 0x80); /* farptr, must be in PSP format (lenbyte args \r) */
442
  ExecParam->fcb1 = 0; /* TODO farptr */
443
  ExecParam->fcb2 = 0; /* TODO farptr */
444
  exit(0); /* let rmod do the job now */
364 mateuszvis 445
}
446
 
447
 
367 mateuszvis 448
static void set_comspec_to_self(unsigned short envseg) {
449
  unsigned short *psp_envseg = (void *)(0x2c); /* pointer to my env segment field in the PSP */
450
  char far *myenv = MK_FP(*psp_envseg, 0);
451
  unsigned short varcount;
452
  char buff[256] = "COMSPEC=";
453
  char *buffptr = buff + 8;
454
  /* who am i? look into my own environment, at the end of it should be my EXEPATH string */
455
  while (*myenv != 0) {
456
    /* consume a NULL-terminated string */
457
    while (*myenv != 0) myenv++;
458
    /* move to next string */
459
    myenv++;
460
  }
461
  /* get next word, if 1 then EXEPATH follows */
462
  myenv++;
463
  varcount = *myenv;
464
  myenv++;
465
  varcount |= (*myenv << 8);
466
  myenv++;
467
  if (varcount != 1) return; /* NO EXEPATH FOUND */
468
  while (*myenv != 0) {
469
    *buffptr = *myenv;
470
    buffptr++;
471
    myenv++;
472
  }
473
  *buffptr = 0;
474
  /* printf("EXEPATH: '%s'\r\n", buff); */
475
  env_setvar(envseg, buff);
476
}
477
 
478
 
450 mateuszvis 479
/* wait for user input */
480
static void cmdline_getinput(unsigned short inpseg, unsigned short inpoff) {
481
  _asm {
482
    push ax
483
    push bx
484
    push cx
485
    push dx
486
    push ds
487
 
488
    /* is DOSKEY support present? (INT 2Fh, AX=4800h, returns non-zero in AL if present) */
489
    mov ax, 0x4800
490
    int 0x2f
491
    mov bl, al /* save doskey status in BL */
492
 
493
    /* set up buffered input to inpseg:inpoff */
494
    mov ax, inpseg
495
    push ax
496
    pop ds
497
    mov dx, inpoff
498
 
499
    /* execute either DOS input or DOSKEY */
500
    test bl, bl /* zf set if no DOSKEY present */
501
    jnz DOSKEY
502
 
503
    mov ah, 0x0a
504
    int 0x21
505
    jmp short DONE
506
 
507
    DOSKEY:
508
    mov ax, 0x4810
509
    int 0x2f
510
 
511
    DONE:
512
    /* terminate command with a CR/LF */
513
    mov ah, 0x02 /* display character in dl */
514
    mov dl, 0x0d
515
    int 0x21
516
    mov dl, 0x0a
517
    int 0x21
518
 
519
    pop ds
520
    pop dx
521
    pop cx
522
    pop bx
523
    pop ax
524
  }
525
}
526
 
527
 
479 mateuszvis 528
/* fetches a line from batch file and write it to buff (NULL-terminated),
529
 * increments rmod counter and returns 0 on success. */
484 mateuszvis 530
static int getbatcmd(char *buff, unsigned char buffmaxlen, struct rmod_props far *rmod) {
469 mateuszvis 531
  unsigned short i;
474 mateuszvis 532
  unsigned short batname_seg = FP_SEG(rmod->batfile);
533
  unsigned short batname_off = FP_OFF(rmod->batfile);
534
  unsigned short filepos_cx = rmod->batnextline >> 16;
535
  unsigned short filepos_dx = rmod->batnextline & 0xffff;
536
  unsigned char blen = 0;
537
 
538
  /* open file, jump to offset filpos, and read data into buff.
539
   * result in blen (unchanged if EOF or failure). */
540
  _asm {
541
    push ax
542
    push bx
543
    push cx
544
    push dx
545
 
546
    /* open file (read-only) */
547
    mov dx, batname_off
548
    mov ax, batname_seg
549
    push ds     /* save DS */
550
    mov ds, ax
551
    mov ax, 0x3d00
552
    int 0x21    /* handle in ax on success */
553
    pop ds      /* restore DS */
554
    jc DONE
555
    mov bx, ax  /* save handle to bx */
556
 
557
    /* jump to file offset CX:DX */
558
    mov ax, 0x4200
559
    mov cx, filepos_cx
560
    mov dx, filepos_dx
561
    int 0x21  /* CF clear on success, DX:AX set to cur pos */
562
    jc CLOSEANDQUIT
563
 
564
    /* read the line into buff */
565
    mov ah, 0x3f
484 mateuszvis 566
    xor ch, ch
567
    mov cl, buffmaxlen
474 mateuszvis 568
    mov dx, buff
569
    int 0x21 /* CF clear on success, AX=number of bytes read */
570
    jc CLOSEANDQUIT
571
    mov blen, al
572
 
573
    CLOSEANDQUIT:
574
    /* close file (handle in bx) */
575
    mov ah, 0x3e
576
    int 0x21
577
 
578
    DONE:
579
    pop dx
580
    pop cx
581
    pop bx
582
    pop ax
469 mateuszvis 583
  }
470 mateuszvis 584
 
474 mateuszvis 585
  /* printf("blen=%u filepos_cx=%u filepos_dx=%u\r\n", blen, filepos_cx, filepos_dx); */
470 mateuszvis 586
 
474 mateuszvis 587
  /* on EOF - abort processing the bat file */
588
  if (blen == 0) goto OOPS;
589
 
590
  /* find nearest \n to inc batch offset and replace \r by NULL terminator
591
   * I support all CR/LF, CR- and LF-terminated batch files */
592
  for (i = 0; i < blen; i++) {
593
    if ((buff[i] == '\r') || (buff[i] == '\n')) {
594
      if ((buff[i] == '\r') && ((i+1) < blen) && (buff[i+1] == '\n')) rmod->batnextline += 1;
595
      break;
596
    }
597
  }
598
  buff[i] = 0;
599
  rmod->batnextline += i + 1;
600
 
601
  return(0);
602
 
603
  OOPS:
604
  rmod->batfile[0] = 0;
605
  rmod->batnextline = 0;
606
  return(-1);
469 mateuszvis 607
}
608
 
609
 
484 mateuszvis 610
/* max length of the cmdline storage (bytes) - includes also max length of
611
 * line loaded from a BAT file (no more than 255 bytes!) */
612
#define CMDLINE_MAXLEN 255
613
 
443 mateuszvis 614
int main(void) {
372 mateuszvis 615
  static struct config cfg;
616
  static unsigned short far *rmod_envseg;
617
  static unsigned short far *lastexitcode;
449 mateuszvis 618
  static struct rmod_props far *rmod;
479 mateuszvis 619
  static char *cmdline;
349 mateuszvis 620
 
479 mateuszvis 621
  rmod = rmod_find(BUFFER_len);
449 mateuszvis 622
  if (rmod == NULL) {
479 mateuszvis 623
    rmod = rmod_install(cfg.envsiz, BUFFER, BUFFER_len);
449 mateuszvis 624
    if (rmod == NULL) {
369 mateuszvis 625
      outputnl("ERROR: rmod_install() failed");
349 mateuszvis 626
      return(1);
627
    }
465 mateuszvis 628
    /* look at command line parameters */
629
    parse_argv(&cfg);
475 mateuszvis 630
    /* copy flags to rmod's storage (and enable ECHO) */
631
    rmod->flags = cfg.flags | FLAG_ECHOFLAG;
465 mateuszvis 632
    /* printf("rmod installed at %Fp\r\n", rmod); */
349 mateuszvis 633
  } else {
465 mateuszvis 634
    /* printf("rmod found at %Fp\r\n", rmod); */
635
    /* if I was spawned by rmod and FLAG_EXEC_AND_QUIT is set, then I should
636
     * die asap, because the command has been executed already, so I no longer
637
     * have a purpose in life */
469 mateuszvis 638
    if (rmod->flags & FLAG_EXEC_AND_QUIT) sayonara(rmod);
349 mateuszvis 639
  }
640
 
465 mateuszvis 641
  rmod_envseg = MK_FP(rmod->rmodseg, RMOD_OFFSET_ENVSEG);
642
  lastexitcode = MK_FP(rmod->rmodseg, RMOD_OFFSET_LEXITCODE);
350 mateuszvis 643
 
367 mateuszvis 644
  /* make COMPSEC point to myself */
645
  set_comspec_to_self(*rmod_envseg);
646
 
369 mateuszvis 647
/*  {
350 mateuszvis 648
    unsigned short envsiz;
649
    unsigned short far *sizptr = MK_FP(*rmod_envseg - 1, 3);
650
    envsiz = *sizptr;
651
    envsiz *= 16;
465 mateuszvis 652
    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 653
  }*/
349 mateuszvis 654
 
483 mateuszvis 655
  /* on /P check for the presence of AUTOEXEC.BAT and execute it if found */
656
  if (cfg.flags & FLAG_PERMANENT) {
657
    if (file_getattr("AUTOEXEC.BAT") >= 0) cfg.execcmd = "AUTOEXEC.BAT";
658
  }
659
 
443 mateuszvis 660
  do {
480 mateuszvis 661
    /* terminate previous command with a CR/LF if ECHO ON (but not during BAT processing) */
483 mateuszvis 662
    if ((rmod->flags & FLAG_ECHOFLAG) && (rmod->batfile[0] == 0)) outputnl("");
474 mateuszvis 663
 
664
    SKIP_NEWLINE:
665
 
666
    /* cancel any redirections that may have been set up before */
667
    redir_revert();
668
 
479 mateuszvis 669
    /* preset cmdline to point at the end of my general-purpose buffer */
484 mateuszvis 670
    cmdline = BUFFER + sizeof(BUFFER) - (CMDLINE_MAXLEN + 1);
474 mateuszvis 671
 
437 mateuszvis 672
    /* (re)load translation strings if needed */
673
    nls_langreload(BUFFER, *rmod_envseg);
674
 
443 mateuszvis 675
    /* skip user input if I have a command to exec (/C or /K) */
676
    if (cfg.execcmd != NULL) {
677
      cmdline = cfg.execcmd;
678
      cfg.execcmd = NULL;
679
      goto EXEC_CMDLINE;
680
    }
681
 
469 mateuszvis 682
    /* if batch file is being executed -> fetch next line */
683
    if (rmod->batfile[0] != 0) {
484 mateuszvis 684
      if (getbatcmd(cmdline, CMDLINE_MAXLEN, rmod) != 0) { /* end of batch */
474 mateuszvis 685
        /* restore echo flag as it was before running the bat file */
475 mateuszvis 686
        rmod->flags &= ~FLAG_ECHOFLAG;
687
        if (rmod->flags & FLAG_ECHO_BEFORE_BAT) rmod->flags |= FLAG_ECHOFLAG;
474 mateuszvis 688
        continue;
689
      }
480 mateuszvis 690
      /* skip any leading spaces */
691
      while (*cmdline == ' ') cmdline++;
474 mateuszvis 692
      /* output prompt and command on screen if echo on and command is not
693
       * inhibiting it with the @ prefix */
479 mateuszvis 694
      if ((rmod->flags & FLAG_ECHOFLAG) && (cmdline[0] != '@')) {
474 mateuszvis 695
        build_and_display_prompt(BUFFER, *rmod_envseg);
479 mateuszvis 696
        outputnl(cmdline);
474 mateuszvis 697
      }
479 mateuszvis 698
      /* skip the @ prefix if present, it is no longer useful */
699
      if (cmdline[0] == '@') cmdline++;
469 mateuszvis 700
    } else {
474 mateuszvis 701
      /* interactive mode: display prompt (if echo enabled) and wait for user
702
       * command line */
475 mateuszvis 703
      if (rmod->flags & FLAG_ECHOFLAG) build_and_display_prompt(BUFFER, *rmod_envseg);
474 mateuszvis 704
      /* collect user input */
469 mateuszvis 705
      cmdline_getinput(FP_SEG(rmod->inputbuf), FP_OFF(rmod->inputbuf));
479 mateuszvis 706
      /* copy it to local cmdline */
707
      if (rmod->inputbuf[1] != 0) _fmemcpy(cmdline, rmod->inputbuf + 2, rmod->inputbuf[1]);
708
      cmdline[(unsigned)(rmod->inputbuf[1])] = 0; /* zero-terminate local buff (oriignal is '\r'-terminated) */
469 mateuszvis 709
    }
349 mateuszvis 710
 
405 mateuszvis 711
    /* if nothing entered, loop again (but without appending an extra CR/LF) */
479 mateuszvis 712
    if (cmdline[0] == 0) goto SKIP_NEWLINE;
349 mateuszvis 713
 
443 mateuszvis 714
    /* I jump here when I need to exec an initial command (/C or /K) */
715
    EXEC_CMDLINE:
716
 
364 mateuszvis 717
    /* move pointer forward to skip over any leading spaces */
718
    while (*cmdline == ' ') cmdline++;
349 mateuszvis 719
 
367 mateuszvis 720
    /* update rmod's ptr to COMPSPEC so it is always up to date */
465 mateuszvis 721
    rmod_updatecomspecptr(rmod->rmodseg, *rmod_envseg);
367 mateuszvis 722
 
402 mateuszvis 723
    /* handle redirections (if any) */
724
    if (redir_parsecmd(cmdline, BUFFER) != 0) {
725
      outputnl("");
726
      continue;
727
    }
728
 
364 mateuszvis 729
    /* try matching (and executing) an internal command */
449 mateuszvis 730
    if (cmd_process(rmod, *rmod_envseg, cmdline, BUFFER) >= -1) {
443 mateuszvis 731
      /* internal command executed */
732
      redir_revert(); /* revert stdout (in case it was redirected) */
733
      continue;
353 mateuszvis 734
    }
352 mateuszvis 735
 
364 mateuszvis 736
    /* if here, then this was not an internal command */
461 mateuszvis 737
    run_as_external(BUFFER, cmdline, *rmod_envseg, rmod);
469 mateuszvis 738
    /* perhaps this is a newly launched BAT file */
739
    if ((rmod->batfile[0] != 0) && (rmod->batnextline == 0)) goto SKIP_NEWLINE;
349 mateuszvis 740
 
474 mateuszvis 741
    /* revert stdout (so the err msg is not redirected) */
402 mateuszvis 742
    redir_revert();
743
 
465 mateuszvis 744
    /* run_as_external() does not return on success, if I am still alive then
745
     * external command failed to execute */
369 mateuszvis 746
    outputnl("Bad command or file name");
349 mateuszvis 747
 
449 mateuszvis 748
  } while ((rmod->flags & FLAG_EXEC_AND_QUIT) == 0);
349 mateuszvis 749
 
449 mateuszvis 750
  sayonara(rmod);
349 mateuszvis 751
  return(0);
752
}