Subversion Repositories SvarDOS

/* This file is part of the SvarCOM project and is published under the terms

 * of the MIT license.

 *

 * Copyright (C) 2021 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 <string.h>

#include "env.h"

#include "helpers.h"

#include "rmod.h"

#include "rmodinit.h"

/* returns segment where rmod is installed */

unsigned short rmod_install(unsigned short envsize) {

  char far *myptr, far *mcb;

  unsigned short far *owner;

  unsigned int rmodseg = 0xffff;

  unsigned int envseg = 0;

  /* read my current env segment from PSP */

  _asm {

    push ax

    push bx

    mov bx, 0x2c

    mov ax, [bx]

    mov envseg, ax

    pop bx

    pop ax

  }

  /* printf("original (PSP) env buffer at %04X\r\n", envseg); */

  /* if custom envsize requested, convert it to number of paragraphs */

  if (envsize != 0) {

    envsize += 15;

    envsize /= 16;

  }

  _asm {

    /* link in the UMB memory chain for enabling high-memory allocation (and save initial status on stack) */

    mov ax, 0x5802  /* GET UMB LINK STATE */

    int 0x21

    xor ah, ah

    push ax         /* save link state on stack */

    mov ax, 0x5803  /* SET UMB LINK STATE */

    mov bx, 1

    int 0x21

    /* get current allocation strategy and save it in DX */

    mov ax, 0x5800

    int 0x21

    push ax

    pop dx

    /* set strategy to 'last fit, try high then low memory' */

    mov ax, 0x5801

    mov bx, 0x0082

    int 0x21

    /* ask for a memory block and save the given segment to rmodseg */

    mov ah, 0x48

    mov bx, (rmod_len + 15) / 16

    int 0x21

    jc ALLOC_FAIL

    mov rmodseg, ax

    /* ask for a memory block for the environment and save it to envseg (only if custom size requested) */

    mov bx, envsize

    test bx, bx

    jz ALLOC_FAIL

    mov ah, 0x48

    int 0x21

    jc ALLOC_FAIL

    mov envseg, ax

    ALLOC_FAIL:

    /* restore initial allocation strategy */

    mov ax, 0x5801

    mov bx, dx

    int 0x21

    /* restore initial UMB memory link state */

    mov ax, 0x5803

    pop bx       /* pop initial UMB link state from stack */

    int 0x21

  }

  if (rmodseg == 0xffff) {

    outputnl("malloc error");

    return(0xffff);

  }

  /* copy rmod to its destination */

  myptr = MK_FP(rmodseg, 0);

  _fmemcpy(myptr, rmod, rmod_len);

  /* mark rmod memory as "self owned" */

  mcb = MK_FP(rmodseg - 1, 0);

  owner = (void far *)(mcb + 1);

  *owner = rmodseg;

  _fmemcpy(mcb + 8, "SVARCOM", 8);

  /* mark env memory as "owned by rmod" */

  mcb = MK_FP(envseg - 1, 0);

  owner = (void far *)(mcb + 1);

  *owner = rmodseg;

  _fmemcpy(mcb + 8, "SVARENV", 8);

  /* write env segment to rmod buffer */

  owner = MK_FP(rmodseg, RMOD_OFFSET_ENVSEG);

  *owner = envseg;

  /* write boot drive to rmod bootdrive field */

  _asm {

    push ax

    push bx

    push dx

    push ds

    mov ax, 0x3305 /* DOS 4.0+ - GET BOOT DRIVE */

    int 0x21 /* boot drive is in DL now (1=A:, 2=B:, etc) */

    add dl, 'A'-1 /* convert to a proper ASCII letter */

    /* set DS to rmodseg */

    mov ax, rmodseg

    mov ds, ax

    /* write boot drive to rmod bootdrive field */

    mov bx, RMOD_OFFSET_BOOTDRIVE

    mov [bx], dl

    pop ds

    pop dx

    pop bx

    pop ax

  }

  /* set the int22 handler in my PSP to rmod so DOS jumps to rmod after I

   * terminate and save the original handler in rmod's memory */

  _asm {

    push ax

    push bx

    push si

    push di

    push es

    /* save my original parent in rmod's memory */

    mov es, [rmodseg]

    mov si, 0x0a

    mov di, RMOD_OFFSET_ORIGPARENT

    cld

    movsw  /* mov ES:[DI], DS:[SI] and SI += 2 and DI += 2 */

    movsw

    mov bx, 0x0a                   /* int22 handler is at 0x0A of the PSP */

    mov ax, RMOD_OFFSET_ROUTINE

    mov [bx], ax                   /* int handler offset */

    mov ax, rmodseg

    mov [bx+2], ax                 /* int handler segment */

    pop es

    pop di

    pop si

    pop bx

    pop ax

  }

  return(rmodseg);

}

/* look up my parent: if it's rmod then return its segment,

 * otherwise return 0xffff */

unsigned short rmod_find(void) {

  unsigned short *parent = (void *)0x0C; /* parent's seg in PSP[Ch] ("prev. int22 handler") */

  unsigned short far *ptr;

  const unsigned short sig[] = {0x1983, 0x1985, 0x2017, 0x2019};

  unsigned char i;

  /* is it rmod? */

  ptr = MK_FP(*parent, 0);

  for (i = 0; i < 4; i++) if (ptr[i] != sig[i]) return(0xffff);

  /* match successfull (rmod is my parent) */

  return(*parent);

}

/* update rmod's pointer to comspec */

void rmod_updatecomspecptr(unsigned short rmod_seg, unsigned short env_seg) {

  unsigned short far *comspecptr = MK_FP(rmod_seg, RMOD_OFFSET_COMSPECPTR);

  char far *comspecfp = env_lookup_val(env_seg, "COMSPEC");

  if (comspecfp != NULL) {

    *comspecptr = FP_OFF(comspecfp);

  } else {

    *comspecptr = 0;

  }

}