source: xtideuniversalbios/trunk/XTIDE_Universal_BIOS/Src/Boot/BootInfo.asm @ 252

; 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 BOOTNFO struct for detected hard disk.
;
; BootInfo_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 BOOTNFO (if successful)
;   Corrupts registers:
;       AX, BX, CX, DX, DI, SI
;--------------------------------------------------------------------
BootInfo_CreateForHardDisk:
    call    BootInfo_ConvertDPTtoBX     ; ES:BX now points to new BOOTNFO
    push    bx                          ; Preserve for return

    mov     di, bx                      ; Starting pointer at beginning of structure

;
; Store Drive Name
;       
    push    ds                          ; Preserve RAMVARS
    push    si                          ; Preserve SI for call to GetTotalSectorCount...

    push    es                          ; ES copied to DS
    pop     ds

    add     si, BYTE ATA1.strModel      ; DS:SI now points drive name
    lea     di, [bx+BOOTNFO.szDrvName]  ; ES:DI now points to name destination
    mov     cx, LEN_BOOTNFO_DRV / 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

;
; Store Sector Count
;
    call    AtaID_GetTotalSectorCountToBXDXAXfromAtaInfoInESSI

    stosw
    xchg    ax, dx
    stosw
    xchg    ax, bx
    stosw

    pop     bx
        
    ret


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


;--------------------------------------------------------------------
; Returns offset to BOOTNFO based on DPT pointer.
;
; BootInfo_ConvertDPTtoBX
;   Parameters:
;       DS:DI:  DPT Pointer
;   Returns:
;       BX:     Offset to BOOTNFO struct
;   Corrupts registers:
;       AX
;--------------------------------------------------------------------
ALIGN JUMP_ALIGN
BootInfo_ConvertDPTtoBX:
    mov     ax, di
    sub     ax, RAMVARS_size                ; subtract off base of DPTs
    mov     bl, DPT_BOOTNFO_SIZE_MULTIPLIER ; BOOTNFO's are a whole number multiple of DPT size
    mul     bl                              
    add     ax, BOOTVARS.rgBootNfo          ; add base of BOOTNFO
    xchg    ax, bx
    ret