source: xtideuniversalbios/trunk/XTIDE_Universal_BIOS/Src/Initialization/FloppyDrive.asm@ 531

; Project name  :   XTIDE Universal BIOS
; Description   :   Various floppy drive related functions that
;                   Boot Menu uses.

;
; XTIDE Universal BIOS and Associated Tools
; Copyright (C) 2009-2010 by Tomi Tilli, 2011-2013 by XTIDE Universal BIOS Team.
;
; This program is free software; you can redistribute it and/or modify
; it under the terms of the GNU General Public License as published by
; the Free Software Foundation; either version 2 of the License, or
; (at your option) any later version.
;
; This program is distributed in the hope that it will be useful,
; but WITHOUT ANY WARRANTY; without even the implied warranty of
; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
; GNU General Public License for more details.
; Visit http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
;

; Section containing code
SECTION .text

;--------------------------------------------------------------------
; Checks is floppy drive handler installed to interrupt vector 40h.
;
; FloppyDrive_IsInt40hInstalled
;   Parameters:
;       ES:     BDA and Interrupt Vector segment (zero)
;   Returns:
;       CF:     Set if INT 40h is installed
;               Cleared if INT 40h is not installed
;   Corrupts registers:
;       BX, CX, DI
;--------------------------------------------------------------------
FloppyDrive_IsInt40hInstalled:
%ifdef USE_AT
    push    es
    push    dx
    push    ax

    call    LoadInt40hVerifyParameters
    int     BIOS_DISK_INTERRUPT_13h
    jc      SHORT .Int40hIsInstalled    ; Maybe there are not any floppy drives at all
    push    es                          ; Drive Parameter Table segment
    push    di                          ; Drive Parameter Table offset

    call    LoadInt40hVerifyParameters
    int     BIOS_DISKETTE_INTERRUPT_40h

    pop     dx
    pop     cx
    cmp     dx, di                      ; Difference in offsets?
    jne     SHORT .Int40hNotInstalled
    mov     dx, es
    cmp     cx, dx                      ; Difference in segments?
    je      SHORT .Int40hIsInstalled
.Int40hNotInstalled:
    stc
.Int40hIsInstalled:
    pop     ax
    pop     dx
    pop     es

%else ; if XT build
    cmp     BYTE [es:BIOS_DISKETTE_INTERRUPT_40h*4+3], 0C0h ; Any ROM segment? (set CF if not)
%endif ; USE_AT
    cmc
    ret

;--------------------------------------------------------------------
; LoadInt40hVerifyParameters
;   Parameters:
;       Nothing
;   Returns:
;       AH:     08h (Get Drive Parameters)
;       DL:     00h (floppy drive)
;       ES:DI:  0:0h (to guard against BIOS bugs)
;   Corrupts registers:
;       DH
;--------------------------------------------------------------------
%ifdef USE_AT
LoadInt40hVerifyParameters:
    mov     ah, GET_DRIVE_PARAMETERS
    cwd                     ; Floppy drive 0
    mov     di, dx
    mov     es, dx          ; ES:DI = 0000:0000h to guard against BIOS bugs
    ret
%endif


;--------------------------------------------------------------------
; Returns floppy drive type.
; PC/XT system do not support AH=08h but FLOPPY_TYPE_525_OR_35_DD
; is still returned for them.
;
; FloppyDrive_GetType
;   Parameters:
;       DL:     Floppy Drive number
;   Returns:
;       BX:     Floppy Drive Type:
;                   FLOPPY_TYPE_525_OR_35_DD
;                   FLOPPY_TYPE_525_DD
;                   FLOPPY_TYPE_525_HD
;                   FLOPPY_TYPE_35_DD
;                   FLOPPY_TYPE_35_HD
;                   FLOPPY_TYPE_35_ED
;       CF:     Set if AH=08h not supported (XT systems) or error
;               Cleared if type read correctly (AT systems)
;   Corrupts registers:
;       AX, CX, DX, DI, ES
;--------------------------------------------------------------------
%ifdef MODULE_BOOT_MENU
FloppyDrive_GetType:
    mov     ah, GET_DRIVE_PARAMETERS
    xor     bx, bx          ; FLOPPY_TYPE_525_OR_35_DD when function not supported
    int     BIOS_DISKETTE_INTERRUPT_40h
    ret
%endif


;--------------------------------------------------------------------
; Returns number of Floppy Drives in system.
;
; FloppyDrive_GetCountToAX
;   Parameters:
;       DS:     RAMVARS Segment
;   Returns:
;       AX:     Number of Floppy Drives
;--------------------------------------------------------------------
FloppyDrive_GetCountToAX:
%ifdef MODULE_SERIAL_FLOPPY
    call    RamVars_UnpackFlopCntAndFirstToAL
    js      .UseBIOSorBDA               ; We didn't add in any drives, counts here are not valid

    adc     al, 1                       ; adds in the drive count bit, and adds 1 for count vs. 0-index,
    jmp     .FinishCalc                 ; need to clear AH on the way out, and add in minimum drive numbers

.UseBIOSorBDA:
%endif
    call    FloppyDrive_GetCountFromBIOS_or_BDA

.FinishCalc:
    mov     ah, [cs:ROMVARS.bMinFddCnt]
    MAX_U   al, ah
    cbw

    ret


;--------------------------------------------------------------------
; FloppyDrive_GetCountFromBIOS_or_BDA
;   Parameters:
;       Nothing
;   Returns:
;       AL:     Number of Floppy Drives
;       CF:     Cleared if successful
;               Set if BIOS function not supported
;   Corrupts registers:
;       AH, ES
;--------------------------------------------------------------------
FloppyDrive_GetCountFromBIOS_or_BDA:
    push    es

; Reads Floppy Drive Count from BIOS.
; Does not work on most XT systems. Call .GetCountFromBDA
; if this function fails.
%ifdef USE_AT
    push    di
    push    bx
    push    cx
    push    dx

    mov     ah, GET_DRIVE_PARAMETERS
    cwd                             ; Floppy Drive 00h
    int     BIOS_DISKETTE_INTERRUPT_40h
    xchg    dx, ax                  ; Number of Floppy Drives to AL

    pop     dx
    pop     cx
    pop     bx
    pop     di

; Reads Floppy Drive Count (0...4) from BIOS Data Area.
; This function should be used only if .GetCountFromBIOS fails.
%else ; ifndef USE_AT
    LOAD_BDA_SEGMENT_TO es, ax
    mov     al, [es:BDA.wEquipment] ; Load Equipment WORD low byte

%ifdef USE_UNDOC_INTEL
    and     al, 0C1h
    eAAM    64
%else
    mov     ah, al                  ; Copy it to AH
    and     ax, 0C001h              ; Leave bits 15..14 and 0
    eROL_IM ah, 2                   ; EW low byte bits 7..6 to 1..0
%endif ; USE_UNDOC_INTEL

    add     al, ah                  ; AL = Floppy Drive count
%endif ; USE_AT

    pop     es
    ret