freebsd-nq/sys/boot/i386/boot0/boot0ext.S
2004-04-20 13:34:18 +00:00

494 lines
14 KiB
ArmAsm

#
# Copyright (c) 1998 Robert Nordier
# All rights reserved.
#
# Redistribution and use in source and binary forms are freely
# permitted provided that the above copyright notice and this
# paragraph and the following disclaimer are duplicated in all
# such forms.
#
# This software is provided "AS IS" and without any express or
# implied warranties, including, without limitation, the implied
# warranties of merchantability and fitness for a particular
# purpose.
#
# $FreeBSD$
# A 1024-byte boot manager.
.set NHRDRV,0x475 # Number of hard drives
.set ORIGIN,0x600 # Execution address
.set SECTOR_SIZE,0x200 # Length of a sector
.set NUM_SECTORS,2 # Total length in sectors
.set FAKE,ORIGIN+(SECTOR_SIZE*NUM_SECTORS) # Partition entry
.set LOAD,0x7c00 # Load address
.set PRT_OFF,0x1be # Partition table
.set TBL0SZ,table0_end-table0 # Table 0 size
.set TBL1SZ,table1_end-table1 # Table 1 size
.set MAGIC,0xaa55 # Magic: bootable
.set KEY_ENTER,0x1c # Enter key scan code
.set KEY_F1,0x3b # F1 key scan code
.set KEY_1,0x02 # #1 key scan code
#
# Flag bits
#
.set FL_PACKET,0x80 # Packet mode
.set FL_NOUPDATE,0x40 # Don't save selection
.set FL_SETDRV,0x20 # Override drive number
#
# Addresses in the sector of embedded data values.
# Accessed with negative offsets from the end of the relocated sectors (%bp).
#
.set _PRT_END,(FAKE-(ORIGIN+SECTOR_SIZE*(NUM_SECTORS-1)))
.set _NXTDRV,-(_PRT_END+0x48) # Next drive
.set _OPT,-(_PRT_END+0x47) # Default option
.set _SETDRV,-(_PRT_END+0x46) # Drive to force
.set _FLAGS,-(_PRT_END+0x45) # Flags
.set _TICKS,-(_PRT_END+0x44) # Timeout ticks
.set _FAKE,0x0 # Fake partition entry
.set _MNUOPT,0xc # Menu options
.globl start # Entry point
.code16 # This runs in real mode
#
# Initialise segments and registers to known values.
# segments start at 0.
# The stack is immediately below the address we were loaded to.
#
# Note that this section of code is used as the first signature check in
# boot0cfg(8).
#
start: cld # String ops inc
xorw %ax,%ax # Zero
movw %ax,%es # Address
movw %ax,%ds # data
movw %ax,%ss # Set up
movw $LOAD,%sp # stack
#
# End signature code
#
#
# Set address for variable space beyond code, and clear it.
# Notice that this is also used to point to the values embedded in the block,
# by using negative offsets.
#
movw $fake,%bp # Address variables
movw %bp,%di # %di used in stosw
movw $0x8,%cx # Words to clear
rep # Zero
stosw # them
incb -0xe(%di) # Sector number 1
#
# Check what flags were loaded with us; specifically, use a predefined Drive.
# If what the bios gives us is bad, use the '0' in the block instead, as well.
#
testb $FL_SETDRV,LOAD+flags-start # Set number drive?
jnz boot.1 # Yes
testb %dl,%dl # Drive number valid?
js boot.2 # Possibly (0x80 set)
boot.1: movb LOAD+setdrv-start,%dl # Drive number to use
boot.2:
#
# Reload all of boot0 (including the extra sectors) into memory at the
# relocation address.
#
push %dx # Save drive number
movw $start,%bx # Origin we were linked for
movw %bp,%si # Fake PTE
movw $0x200+NUM_SECTORS,%ax # Read in all
callw intx13 # of boot0
pop %dx # Restore
#
# Relocate to the new copy of the code.
#
jmp main+ORIGIN-LOAD # To relocated code
#
# Whatever we decided to use, now store it into the fake
# partition entry that lives in the data space above us.
#
main: movb %dl,_FAKE(%bp) # Save drive number
callw putn # To new line
pushw %dx # Save drive number
#
# Start out with a pointer to the 4th byte of the first table entry
# so that after 4 iterations it's beyond the end of the sector.
# and beyond a 256 byte boundary and has overflowed 8 bits (see next comment).
# (remember that the table starts 2 bytes earlier than you would expect
# as the bootable flag is after it in the block)
#
movw $(partbl+0x4),%bx # Partition table (+4)
xorw %dx,%dx # Item number
#
# Loop around on the partition table, printing values until we
# pass a 256 byte boundary. The end of loop test is at main.5.
#
main.3: movb %ch,-0x4(%bx) # Zero active flag (ch == 0)
btw %dx,_FLAGS(%bp) # Entry enabled?
jnc main.5 # No
#
# If any of the entries in the table are
# the same as the 'type' in the slice table entry,
# then this is an empty or non bootable partition. Skip it.
#
movb (%bx),%al # Load type
movw $tables,%di # Lookup tables
movb $TBL0SZ,%cl # Number of entries
repne # Exclude
scasb # partition?
je main.5 # Yes
#
# Now scan the table of known types
#
movb $TBL1SZ,%cl # Number of entries
repne # Known
scasb # type?
jne main.4 # No
#
# If it matches get the matching element in the
# next array. If it doesn't, we are already
# pointing at its first element which points to a "?".
#
addw $TBL1SZ,%di # Adjust
main.4: movb (%di),%cl # Partition
addw %cx,%di # description
callw putx # Display it
main.5: incw %dx # Next item
addb $0x10,%bl # Next entry
jnc main.3 # Till done
#
# Passed a 256 byte boundary..
# table is finished.
# Add one to the drive number and check it is valid,
#
popw %ax # Drive number
subb $0x80-0x1,%al # Does next
cmpb NHRDRV,%al # drive exist? (from BIOS?)
jb main.6 # Yes
# If not then if there is only one drive,
# Don't display drive as an option.
#
decw %ax # Already drive 0?
jz main.7 # Yes
# If it was illegal or we cycled through them,
# then go back to drive 0.
#
xorb %al,%al # Drive 0
#
# Whatever drive we selected, make it an ascii digit and save it back
# to the "next drive" location in the loaded block in case we
# want to save it for next time.
# This also is part of the printed drive string so add 0x80 to indicate
# end of string.
#
main.6: addb $'0'|0x80,%al # Save next
movb %al,_NXTDRV(%bp) # drive number
movw $drive,%di # Display
callw putx # item
#
# Now that we've printed the drive (if we needed to), display a prompt.
# Get ready for the input by noting the time.
#
main.7: movw $prompt,%si # Display
callw putstr # prompt
movb _OPT(%bp),%dl # Display
decw %si # default
callw putkey # key
xorb %ah,%ah # BIOS: Get
int $0x1a # system time
movw %dx,%di # Ticks when
addw _TICKS(%bp),%di # timeout
#
# Busy loop, looking for keystrokes but
# keeping one eye on the time.
#
main.8: movb $0x1,%ah # BIOS: Check
int $0x16 # for keypress
jnz main.11 # Have one
xorb %ah,%ah # BIOS: Get
int $0x1a # system time
cmpw %di,%dx # Timeout?
jb main.8 # No
#
# If timed out or defaulting, come here.
#
main.9: movb _OPT(%bp),%al # Load default
jmp main.12 # Join common code
#
# User's last try was bad, beep in displeasure.
# Since nothing was printed, just continue on as if the user
# hadn't done anything. This gives the effect of the user getting a beep
# for all bad keystrokes but no action until either the timeout
# occurs or the user hits a good key.
#
main.10: movb $0x7,%al # Signal
callw putchr # error
#
# Get the keystroke.
#
main.11: xorb %ah,%ah # BIOS: Get
int $0x16 # keypress
movb %ah,%al # Scan code
#
# If it's CR act as if timed out.
#
cmpb $KEY_ENTER,%al # Enter pressed?
je main.9 # Yes
#
# Otherwise check if legal
# If not ask again.
#
subb $KEY_F1,%al # Less F1 scan code
cmpb $0x4,%al # F1..F5?
jna main.12 # Yes
subb $(KEY_1 - KEY_F1),%al # Less #1 scan code
cmpb $0x4,%al # #1..#5?
ja main.10 # No
#
# We have a selection.
# but if it's a bad selection go back to complain.
# The bits in MNUOPT were set when the options were printed.
# Anything not printed is not an option.
#
main.12: cbtw # Option
btw %ax,_MNUOPT(%bp) # enabled?
jnc main.10 # No
#
# Save the info in the original tables
# for rewriting to the disk.
#
movb %al,_OPT(%bp) # Save option
movw $fake,%si # Partition for write
movb (%si),%dl # Drive number
movw %si,%bx # Partition for read
cmpb $0x4,%al # F5/#5 pressed?
pushf # Save
je main.13 # Yes
shlb $0x4,%al # Point to
addw $partbl,%ax # selected
xchgw %bx,%ax # partition
movb $0x80,(%bx) # Flag active
#
# If not asked to do a write-back (flags 0x40) don't do one.
#
main.13: pushw %bx # Save
testb $FL_NOUPDATE,_FLAGS(%bp) # Skip update?
jnz main.14 # Yes
movw $start,%bx # Data to write
movw $0x301,%ax # Write 1 sector
callw intx13 # to disk
main.14: popw %si # Restore
popf # Restore
#
# If going to next drive, replace drive with selected one.
# Remember to un-ascii it. Hey 0x80 is already set, cool!
#
jne main.15 # If not F5/#5
movb _NXTDRV(%bp),%dl # Next drive
subb $'0',%dl # number
#
# load selected bootsector to the LOAD location in RAM.
# If it fails to read or isn't marked bootable, treat it
# as a bad selection.
#
main.15: movw $LOAD,%bx # Address for read
movw $0x201,%ax # Read 1 sector
callw intx13 # from disk
jc main.10 # If error
cmpw $MAGIC,0x1fe(%bx) # Bootable?
jne main.10 # No
callw putn # Leave some space
jmp *%bx # Invoke bootstrap
#
# Display routines
#
putkey: movb $'F',%al # Display
callw putchr # 'F'
movb $'1',%al # Prepare
addb %dl,%al # digit
jmp putstr.1 # Display the rest
#
# Display the option and note that it is a valid option.
# That last point is a bit tricky..
#
putx: btsw %dx,_MNUOPT(%bp) # Enable menu option
movw $item,%si # Display
callw putkey # key
movw %di,%si # Display the rest
puts: callw putstr # Display string
putn: movw $crlf,%si # To next line
putstr: lodsb # Get byte
testb $0x80,%al # End of string?
jnz putstr.2 # Yes
putstr.1: callw putchr # Display char
jmp putstr # Continue
putstr.2: andb $~0x80,%al # Clear MSB
putchr: pushw %bx # Save
movw $0x7,%bx # Page:attribute
movb $0xe,%ah # BIOS: Display
int $0x10 # character
popw %bx # Restore
retw # To caller
# One-sector disk I/O routine
#
# Calling conventions: (assumes %si -> partition table entry)
#
# 0x1(%si) - byte - head
# 0x2(%si) - word - cylinder/sector
# 0x8(%si) - long - LBA to use if needed
# %ah - byte - operation, 2 = read, 3 = write
# %al - byte - sector count
# %dl - byte - drive number
# %es:(%bx) - void - buffer to use for transfer
#
# If the head == 0xff, and cylinder/sector == 0xffff, then try
# to use the EDD stuff with the LBA offset instead of CHS. However,
# use CHS if at all possible.
intx13: movb 0x1(%si),%dh # Load head
movw 0x2(%si),%cx # Load cylinder:sector
pushw %si # Save
movw %sp,%di # Save
cmpb $0xff,%dh # Might we need LBA?
jne intx13.2 # No, just use CHS
cmpw $0xffff,%cx # Do we need LBA?
jne intx13.2 # No
testb $FL_PACKET,_FLAGS(%bp) # Try the packet interface?
jz intx13.2 # No
pushw %cx # Save
pushw %bx # Save
movw $0x55aa,%bx # Magic
pushw %ax # Save
movb $0x41,%ah # BIOS: EDD extensions
int $0x13 # present?
popw %ax # Restore
jc intx13.1 # Not present, use CHS
cmpw $0xaa55,%bx # Magic?
jne intx13.1 # Not present, use CHS
testb $0x1,%cl # Packet mode available?
jz intx13.1 # No, use CHS
orb $0x40,%ah # Use disk packet
intx13.1: popw %bx # Restore
popw %cx # Restore
testb $0x40,%ah # Using packet mode?
jz intx13.2 # No, so skip the rest
pushl $0x0 # Set the
pushl 0x8(%si) # LBA address
pushw %es # Set the transfer
pushw %bx # buffer address
push $0x0 # Punch a hole in the stack
push $0x10 # Packet size
movw %sp,%si # Packet pointer
xchgb %al,0x2(%si) # Set the block count in the
# packet and zero %al,
# turning verify off for writes
intx13.2: int $0x13 # BIOS: Disk I/O
movw %di,%sp # Restore
popw %si # Restore
retw # To caller
.org PRT_OFF-0xe,0x90
#
# These values are sometimes changed before writing back to the drive
# Be especially careful that nxtdrv: must come after drive:, as it
# is part of the same string.
#
# Note that the 'drive' string variable is used as the second signature
# check in boot0cfg(8).
#
version_minor: .byte 0x1 # minor version
version_major: .byte 0x1 # major version
drive: .ascii "Drive "
nxtdrv: .byte 0x0 # Next drive number
opt: .byte 0x0 # Option
setdrv: .byte 0x80 # Drive to force
flags: .byte FLAGS # Flags
ticks: .word TICKS # Delay
#
# here is the 64 byte partition table that fdisk would fiddle with.
#
partbl: .fill 0x40,0x1,0x0 # Partition table
.word MAGIC # Magic number
#
# start of sector two.. ugh
#
.org SECTOR_SIZE,0x90
# Menu strings
item: .ascii " "; .byte ' '|0x80
prompt: .ascii "\nDefault:"; .byte ' '|0x80
crlf: .ascii "\r"; .byte '\n'|0x80
# Partition type tables
tables:
#
# These entries identify invalid or NON BOOT types and partitions.
#
table0: .byte 0x0, 0x5, 0xf
table0_end:
#
# These values indicate bootable types we know the names of
#
table1: .byte 0x1, 0x4, 0x6, 0x7, 0xb, 0xc, 0xe, 0x42, 0x63, 0x83
.byte 0x9f, 0xa5, 0xa6, 0xa9
table1_end:
#
# These are offsets that match the known names above and point to the strings
# that will be printed.
#
.byte os_misc-. # Unknown
.byte os_dos-. # DOS
.byte os_dos-. # DOS
.byte os_dos-. # DOS
.byte os_nt-. # NT/XP or OS/2
.byte os_windows-. # Windows 32-bit FAT
.byte os_windows-. # Windows 32-bit FAT ext int 13
.byte os_windows-. # Windows
.byte os_windows-. # Windows 2000 dyn ext
.byte os_unix-. # UNIX
.byte os_linux-. # Linux
.byte os_bsdos-. # BSD/OS
.byte os_freebsd-. # FreeBSD
.byte os_openbsd-. # OpenBSD
.byte os_netbsd-. # NetBSD
#
# And here are the strings themselves. 0x80 or'd into a byte indicates
# the end of the string. (not so great for Russians but...)
#
os_misc: .ascii "Unknow"; .byte 'n'|0x80
os_dos: .ascii "DO"; .byte 'S'|0x80
os_nt: .ascii "Windows NT/X"; .byte 'P'|0x80
os_windows: .ascii "Window"; .byte 's'|0x80
os_unix: .ascii "UNI"; .byte 'X'|0x80
os_linux: .ascii "Linu"; .byte 'x'|0x80
os_freebsd: .ascii "FreeBS"; .byte 'D'|0x80
os_openbsd: .ascii "OpenBS"; .byte 'D'|0x80
os_netbsd: .ascii "NetBS"; .byte 'D'|0x80
os_bsdos: .ascii "BSD/O"; .byte 'S'|0x80
#
# Fake partition entry created at the end of the table used when loading
# boot0 at the very beginning and when loading an MBR from another disk when
# F5 is pressed.
#
.org SECTOR_SIZE*NUM_SECTORS, 0x0
fake: