source: xtideuniversalbios/trunk/BIOS_Drive_Information_Tool/Src/Main.asm@ 421

; Project name  :   BIOS Drive Information Tool
; Description   :   BIOS Drive Information Tool reads and displays
;                   drive information from BIOS.

;
; XTIDE Universal BIOS and Associated Tools 
; Copyright (C) 2009-2010 by Tomi Tilli, 2011-2012 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
;       

; Include .inc files
%define INCLUDE_DISPLAY_LIBRARY
%define INCLUDE_KEYBOARD_LIBRARY
%include "AssemblyLibrary.inc"  ; Assembly Library. Must be included first!
%include "Version.inc"          ; From XTIDE Universal BIOS
%include "ATA_ID.inc"           ; From XTIDE Universal BIOS
%include "Int13h.inc"           ; From XTIDE Universal BIOS
%include "EBIOS.inc"            ; From XTIDE Universal BIOS
FLG_DRVNHEAD_DRV    EQU (1<<4)  ; Required by CustomDPT.inc
%include "CustomDPT.inc"        ; From XTIDE Universal BIOS


; Section containing code
SECTION .text

; Program first instruction.
ORG 100h                        ; Code starts at offset 100h (DOS .COM)
Start:
    jmp     StartBiosDriveInformationTool

; Include library and other sources
%include "AssemblyLibrary.asm"
%include "LbaAssist.asm"        ; From XTIDE Universal BIOS
%include "Strings.asm"
%include "Bios.asm"
%include "Print.asm"


;--------------------------------------------------------------------
; Program start
;--------------------------------------------------------------------
ALIGN JUMP_ALIGN
StartBiosDriveInformationTool:
    CALL_DISPLAY_LIBRARY    InitializeDisplayContext
    call    Print_SetCharacterOutputToSTDOUT

    mov     si, g_szProgramName
    call    Print_NullTerminatedStringFromSI

    call    ReadAndDisplayAllHardDrives

    ; Exit to DOS
    mov     ax, 4C00h           ; Exit to DOS
    int     21h



ReadAndDisplayAllHardDrives:
    call    Bios_GetNumberOfHardDrivesToDX
    jc      SHORT .NoDrivesAvailable
    mov     cx, dx
    mov     dl, 80h             ; First hard drive
    jmp     SHORT .DisplayFirstDrive

ALIGN JUMP_ALIGN
.DisplayNextDriveFromDL:
    mov     si, g_szPressAnyKey
    call    Print_NullTerminatedStringFromSI
    call    Keyboard_GetKeystrokeToAXandWaitIfNecessary

.DisplayFirstDrive:
    mov     si, g_szHeaderDrive
    call    Print_DriveNumberFromDLusingFormatStringInSI

    mov     si, g_szAtaInfoHeader
    call    Print_NullTerminatedStringFromSI
    call    DisplayAtaInformationForDriveDL

    mov     si, g_szOldInfoHeader
    call    Print_NullTerminatedStringFromSI
    call    DisplayOldInt13hInformationForDriveDL

    mov     si, g_szNewInfoHeader
    call    Print_NullTerminatedStringFromSI
    call    DisplayNewInt13hInformationFromDriveDL

    inc     dx
    loop    .DisplayNextDriveFromDL
.NoDrivesAvailable:
    ret


ALIGN JUMP_ALIGN
DisplayAtaInformationForDriveDL:
    push    cx
    push    dx

    call    Bios_ReadAtaInfoFromDriveDLtoBX
    call    Print_ErrorMessageFromAHifError
    jc      SHORT .SkipAtaInfoSinceError

    call    Print_NameFromAtaInfoInBX

    mov     cx, [bx+ATA1.wCylCnt]
    mov     dx, [bx+ATA1.wHeadCnt]
    mov     ax, [bx+ATA1.wSPT]
    call    Print_CHSfromCXDXAX

    test    WORD [bx+ATA1.wFields], A1_wFields_54to58
    jz      SHORT .SkipChsSectors
    mov     si, g_szChsSectors
    call    Print_NullTerminatedStringFromSI
    mov     si, bx
    mov     ax, [si+ATA1.dwCurSCnt]
    mov     dx, [si+ATA1.dwCurSCnt+2]
    xor     bx, bx
    call    Print_TotalSectorsFromBXDXAX
    mov     bx, si
.SkipChsSectors:

    test    WORD [bx+ATA1.wCaps], A1_wCaps_LBA
    jz      SHORT .SkipLBA28
    mov     si, g_szLBA28
    call    Print_NullTerminatedStringFromSI
    mov     si, bx
    mov     ax, [si+ATA1.dwLBACnt]
    mov     dx, [si+ATA1.dwLBACnt+2]
    xor     bx, bx
    call    Print_TotalSectorsFromBXDXAX
    mov     bx, si
.SkipLBA28:

    test    WORD [bx+ATA6.wSetSup83], A6_wSetSup83_LBA48
    jz      SHORT .SkipLBA48
    mov     si, g_szLBA48
    call    Print_NullTerminatedStringFromSI
    mov     si, bx
    mov     ax, [bx+ATA6.qwLBACnt]
    mov     dx, [bx+ATA6.qwLBACnt+2]
    mov     bx, [bx+ATA6.qwLBACnt+4]
    call    Print_TotalSectorsFromBXDXAX
    mov     bx, si
.SkipLBA48:

    ; Print L-CHS generated by XTIDE Universal BIOS
    mov     ax, g_szXTUBversion
    mov     si, g_szXTUB
    call    Print_VersionStringFromAXusingFormatStringInSI
    call    DisplayXTUBcompatibilityInfoUsingAtaInfoFromDSBX

.SkipAtaInfoSinceError:
    pop     dx
    pop     cx
    ret


ALIGN JUMP_ALIGN
DisplayXTUBcompatibilityInfoUsingAtaInfoFromDSBX:
    test    WORD [bx+ATA1.wCaps], A1_wCaps_LBA
    jz      SHORT .LbaNotSupported
    test    WORD [bx+ATA6.wSetSup83], A6_wSetSup83_LBA48
    jz      SHORT .LoadLba28SectorCount

    ; Load LBA48 Sector Count
    mov     ax, [bx+ATA6.qwLBACnt]
    mov     dx, [bx+ATA6.qwLBACnt+2]
    mov     bx, [bx+ATA6.qwLBACnt+4]
    jmp     SHORT .ConvertLbaToLCHS
.LoadLba28SectorCount:
    mov     ax, [bx+ATA1.dwLBACnt]
    mov     dx, [bx+ATA1.dwLBACnt+2]
    xor     bx, bx

.ConvertLbaToLCHS:
    call    LbaAssist_ConvertSectorCountFromBXDXAXtoLbaAssistedCHSinDXAXBLBH
    LIMIT_LBA_CYLINDERS_IN_DXAX_TO_LCHS_CYLINDERS
    xchg    cx, ax
    eMOVZX  dx, bl
    eMOVZX  ax, bh
    call    Print_CHSfromCXDXAX
.LbaNotSupported:
    ret


ALIGN JUMP_ALIGN
DisplayOldInt13hInformationForDriveDL:
    push    cx
    push    dx

    call    Bios_ReadOldInt13hParametersFromDriveDL
    call    Print_ErrorMessageFromAHifError
    jc      SHORT .SkipOldInt13hSinceError
    call    Print_CHSfromCXDXAX

    mov     si, g_szSectors
    call    Print_NullTerminatedStringFromSI
    pop     dx
    push    dx
    call    Bios_ReadOldInt13hCapacityFromDriveDL
    call    Print_ErrorMessageFromAHifError
    jc      SHORT .SkipOldInt13hSinceError
    xchg    ax, dx
    mov     dx, cx
    xor     bx, bx
    call    Print_TotalSectorsFromBXDXAX
.SkipOldInt13hSinceError:
    pop     dx
    pop     cx
    ret


ALIGN JUMP_ALIGN
DisplayNewInt13hInformationFromDriveDL:
    push    cx
    push    dx

    call    Bios_ReadEbiosVersionFromDriveDL
    call    Print_ErrorMessageFromAHifError
    jc      SHORT .SkipNewInt13hSinceError
    call    Print_EbiosVersionFromBXandExtensionsFromCX

    call    Bios_ReadEbiosInfoFromDriveDLtoDSSI
    call    Print_ErrorMessageFromAHifError
    jc      SHORT .SkipNewInt13hSinceError

    test    WORD [si+EDRIVE_INFO.wFlags], FLG_CHS_INFORMATION_IS_VALID
    jz      SHORT .SkipEbiosCHS
    mov     cx, [si+EDRIVE_INFO.dwCylinders]
    mov     dx, [si+EDRIVE_INFO.dwHeads]
    mov     ax, [si+EDRIVE_INFO.dwSectorsPerTrack]
    call    Print_CHSfromCXDXAX
.SkipEbiosCHS:

    push    si
    mov     si, g_szSectors
    call    Print_NullTerminatedStringFromSI
    pop     si
    mov     ax, [si+EDRIVE_INFO.qwTotalSectors]
    mov     dx, [si+EDRIVE_INFO.qwTotalSectors+2]
    mov     bx, [si+EDRIVE_INFO.qwTotalSectors+4]
    call    Print_TotalSectorsFromBXDXAX

    mov     ax, [si+EDRIVE_INFO.wSectorSize]
    mov     si, g_szNewSectorSize
    call    Print_SectorSizeFromAXusingFormatStringInSI

.SkipNewInt13hSinceError:
    pop     dx
    pop     cx
    ret