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

; 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		ax, [si+EDRIVE_INFO.dwCylinders]
	mov		dx, [si+EDRIVE_INFO.dwHeads]
	mov		cx, [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