source: xtideuniversalbios/trunk/XTIDE_Universal_BIOS/Src/Boot/BootMenuInfo.asm @ 363

; Project name  :   XTIDE Universal BIOS
; Description   :   Functions for generating and accessing drive
;                   information to be displayed on boot menu.

; Section containing code
SECTION .text

;--------------------------------------------------------------------
; Creates new BOOTMENUINFO struct for detected hard disk.
;
; BootMenuInfo_CreateForHardDisk
;   Parameters:
;       DL:     Drive number
;       DS:DI:  Ptr to Disk Parameter Table
;       ES:SI:  Ptr to 512-byte ATA information read from the drive
;   Returns:
;       ES:BX:  Ptr to BOOTMENUINFO (if successful)
;   Corrupts registers:
;       AX, CX, DX, DI
;--------------------------------------------------------------------
BootMenuInfo_CreateForHardDisk:
    call    BootMenuInfo_ConvertDPTtoBX         ; ES:BX now points to new BOOTMENUINFO
    push    ds                                  ; Preserve RAMVARS...
    push    si                                  ; ...and SI

    push    es                                  ; ES to be copied to DS

%ifdef MODULE_ADVANCED_ATA
    ; Copy DPT_ADVANCED_ATA to BOOTMENUINFO to keep DPTs small.
    ; DPT_ADVANCED_ATA has variables that are only needed during initialization.
    mov     ax, [di+DPT_ADVANCED_ATA.wIdeBasePort]
    mov     [es:bx+BOOTMENUINFO.wIdeBasePort], ax
    mov     dx, [di+DPT_ADVANCED_ATA.wMinPioActiveTimeNs]
    mov     [es:bx+BOOTMENUINFO.wMinPioActiveTimeNs], dx

    mov     ax, [di+DPT_ADVANCED_ATA.wMinPioRecoveryTimeNs]
    mov     cx, [di+DPT_ADVANCED_ATA.wControllerID]
    mov     dx, [di+DPT_ADVANCED_ATA.wControllerBasePort]
    pop     ds                                  ; ES copied to DS
    mov     [bx+BOOTMENUINFO.wMinPioRecoveryTimeNs], ax
    mov     [bx+BOOTMENUINFO.wControllerID], cx
    mov     [bx+BOOTMENUINFO.wControllerBasePort], dx

%else
    pop     ds                                  ; ES copied to DS
%endif

    ; Store Drive Name
    add     si, BYTE ATA1.strModel              ; DS:SI now points drive name
    lea     di, [bx+BOOTMENUINFO.szDrvName]     ; ES:DI now points to name destination
    mov     cx, MAX_HARD_DISK_NAME_LENGTH / 2   ; Max number of WORDs allowed
.CopyNextWord:
    lodsw
    xchg    al, ah                              ; Change endianness
    stosw
    loop    .CopyNextWord
    xor     ax, ax                              ; Zero AX and clear CF
    stosw                                       ; Terminate with NULL

    pop     si
    pop     ds
    ret


;--------------------------------------------------------------------
; BootMenuInfo_GetTotalSectorCount
;   Parameters:
;       DS:DI:      DPT Pointer
;   Returns:
;       BX:DX:AX:   48-bit sector count
;   Corrupts registers:
;       CX
;--------------------------------------------------------------------
BootMenuInfo_GetTotalSectorCount:
    test    BYTE [di+DPT.bFlagsLow], FLG_DRVNHEAD_LBA
    jnz     SHORT .ReturnFullCapacity
    jmp     AH15h_GetSectorCountToBXDXAX
.ReturnFullCapacity:
    jmp     AccessDPT_GetLbaSectorCountToBXDXAX


;--------------------------------------------------------------------
; BootMenuInfo_IsAvailable
;   Parameters:
;       Nothing
;   Returns:
;       ES:     Segment to BOOTVARS with BOOTMENUINFOs
;       ZF:     Set if BOOTVARS with BOOTMENUINFOs is available
;               Cleared if not available (no longer initializing)
;   Corrupts registers:
;       BX
;--------------------------------------------------------------------
BootMenuInfo_IsAvailable:
    LOAD_BDA_SEGMENT_TO es, bx
    cmp     WORD [es:BOOTVARS.wMagicWord], BOOTVARS_MAGIC_WORD
    ret


;--------------------------------------------------------------------
; Returns offset to BOOTMENUINFO based on DPT pointer.
;
; BootMenuInfo_ConvertDPTtoBX
;   Parameters:
;       DS:DI:  DPT Pointer
;   Returns:
;       BX:     Offset to BOOTMENUINFO struct
;   Corrupts registers:
;       Nothing
;--------------------------------------------------------------------
BootMenuInfo_ConvertDPTtoBX:
    push    ax
    mov     ax, di
    sub     ax, BYTE RAMVARS_size                   ; subtract off base of DPTs
    mov     bl, DPT_BOOTMENUINFO_SIZE_MULTIPLIER    ; BOOTMENUINFO's are a whole number multiple of DPT size
    mul     bl                              
    add     ax, BOOTVARS.rgBootNfo                  ; add base of BOOTMENUINFO
    xchg    ax, bx
    pop     ax
    ret