forked from KolibriOS/kolibrios
kolibri-ahci:
- added disk registration in system - added ahci_read (DOES NOT WORK, TODO), ahci_querymedia - other changes git-svn-id: svn://kolibrios.org@9140 a494cfbc-eb01-0410-851d-a64ba20cac60
This commit is contained in:
parent
b436ed9f25
commit
8dfac86e67
@ -24,7 +24,8 @@ AHCI_DEV_PM = 3
|
|||||||
AHCI_DEV_SATAPI = 4
|
AHCI_DEV_SATAPI = 4
|
||||||
|
|
||||||
; ATA commands
|
; ATA commands
|
||||||
ATA_IDENTIFY = 0xEC
|
ATA_IDENTIFY = 0xEC
|
||||||
|
ATA_CMD_READ_DMA_EX = 0x25
|
||||||
|
|
||||||
; ATA constants
|
; ATA constants
|
||||||
ATA_DEV_BUSY = 0x80
|
ATA_DEV_BUSY = 0x80
|
||||||
@ -163,6 +164,7 @@ struct PORT_DATA
|
|||||||
port dd ? ; address of correspoding HBA_PORT structure
|
port dd ? ; address of correspoding HBA_PORT structure
|
||||||
portno dd ? ; port index, 0..31
|
portno dd ? ; port index, 0..31
|
||||||
drive_type db ? ; drive type
|
drive_type db ? ; drive type
|
||||||
|
sector_count dq ? ; number of sectors
|
||||||
ends
|
ends
|
||||||
|
|
||||||
; Register FIS – Host to Device
|
; Register FIS – Host to Device
|
||||||
@ -310,12 +312,32 @@ uglobal
|
|||||||
align 4
|
align 4
|
||||||
ahci_controller AHCI_DATA
|
ahci_controller AHCI_DATA
|
||||||
port_data_arr rb (sizeof.PORT_DATA*AHCI_MAX_PORTS)
|
port_data_arr rb (sizeof.PORT_DATA*AHCI_MAX_PORTS)
|
||||||
|
ahci_mutex MUTEX
|
||||||
|
endg
|
||||||
|
|
||||||
|
iglobal
|
||||||
|
align 4
|
||||||
|
ahci_callbacks:
|
||||||
|
dd ahci_callbacks.end - ahci_callbacks
|
||||||
|
dd 0 ; no close function
|
||||||
|
dd 0 ; no closemedia function
|
||||||
|
dd ahci_querymedia
|
||||||
|
dd ahci_read
|
||||||
|
dd 0;ahci_write
|
||||||
|
dd 0 ; no flush function
|
||||||
|
dd 0 ; use default cache size
|
||||||
|
.end:
|
||||||
|
hd_name db 'hd', 0, 0, 0
|
||||||
|
hd_counter dd 0
|
||||||
endg
|
endg
|
||||||
|
|
||||||
; -----------------------------------------------------------------------
|
; -----------------------------------------------------------------------
|
||||||
; detect ahci controller and initialize
|
; detect ahci controller and initialize
|
||||||
align 4
|
align 4
|
||||||
ahci_init:
|
ahci_init:
|
||||||
|
mov ecx, ahci_mutex
|
||||||
|
call mutex_init
|
||||||
|
|
||||||
mov ecx, ahci_controller
|
mov ecx, ahci_controller
|
||||||
mov esi, pcidev_list
|
mov esi, pcidev_list
|
||||||
.find_ahci_ctr:
|
.find_ahci_ctr:
|
||||||
@ -433,6 +455,7 @@ ahci_init:
|
|||||||
; IDT::RegisterInterruptHandler(irq, InterruptHandler);
|
; IDT::RegisterInterruptHandler(irq, InterruptHandler);
|
||||||
; ahciHBA->is = 0xffffffff;
|
; ahciHBA->is = 0xffffffff;
|
||||||
|
|
||||||
|
mov [hd_counter], 0
|
||||||
xor ebx, ebx
|
xor ebx, ebx
|
||||||
.detect_drives:
|
.detect_drives:
|
||||||
cmp ebx, AHCI_MAX_PORTS
|
cmp ebx, AHCI_MAX_PORTS
|
||||||
@ -538,6 +561,39 @@ ahci_init:
|
|||||||
|
|
||||||
stdcall ahci_port_identify, ecx
|
stdcall ahci_port_identify, ecx
|
||||||
|
|
||||||
|
cmp [ecx + PORT_DATA.drive_type], AHCI_DEV_SATA
|
||||||
|
jne .after_add_disk ; skip adding disk code
|
||||||
|
; register disk in system:
|
||||||
|
push ecx edx
|
||||||
|
mov eax, [hd_counter]
|
||||||
|
xor edx, edx
|
||||||
|
mov ecx, 10
|
||||||
|
div ecx ; eax = hd_counter / 10, edx = hd_counter % 10
|
||||||
|
test eax, eax
|
||||||
|
jz .concat_one
|
||||||
|
add al, '0'
|
||||||
|
mov byte [hd_name + 2], al
|
||||||
|
add dl, '0'
|
||||||
|
mov byte [hd_name + 3], dl
|
||||||
|
jmp .endif1
|
||||||
|
.concat_one:
|
||||||
|
add dl, '0'
|
||||||
|
mov byte [hd_name + 2], dl
|
||||||
|
.endif1:
|
||||||
|
pop edx ecx
|
||||||
|
|
||||||
|
DEBUGF 1, "adding '%s'\n", hd_name
|
||||||
|
|
||||||
|
stdcall disk_add, ahci_callbacks, hd_name, ecx, 0
|
||||||
|
test eax, eax
|
||||||
|
jz .disk_add_fail
|
||||||
|
stdcall disk_media_changed, eax, 1 ; system will scan for partitions on disk
|
||||||
|
jmp .after_add_disk
|
||||||
|
|
||||||
|
.disk_add_fail:
|
||||||
|
DEBUGF 1, "Failed to add disk\n"
|
||||||
|
.after_add_disk:
|
||||||
|
|
||||||
.continue_detect_drives:
|
.continue_detect_drives:
|
||||||
inc ebx
|
inc ebx
|
||||||
jmp .detect_drives
|
jmp .detect_drives
|
||||||
@ -574,6 +630,7 @@ proc ahci_port_identify stdcall, pdata: dword
|
|||||||
jne .cmdslot_found
|
jne .cmdslot_found
|
||||||
|
|
||||||
DEBUGF 1, "No free cmdslot on port %u\n", [esi + PORT_DATA.portno]
|
DEBUGF 1, "No free cmdslot on port %u\n", [esi + PORT_DATA.portno]
|
||||||
|
jmp .ret
|
||||||
|
|
||||||
.cmdslot_found:
|
.cmdslot_found:
|
||||||
mov [cmdslot], eax
|
mov [cmdslot], eax
|
||||||
@ -666,6 +723,225 @@ proc ahci_port_identify stdcall, pdata: dword
|
|||||||
ret
|
ret
|
||||||
endp
|
endp
|
||||||
|
|
||||||
|
proc ahci_querymedia stdcall, pdata, mediainfo
|
||||||
|
push ecx edx
|
||||||
|
mov eax, [mediainfo]
|
||||||
|
mov edx, [pdata]
|
||||||
|
mov [eax + DISKMEDIAINFO.Flags], 0
|
||||||
|
mov [eax + DISKMEDIAINFO.SectorSize], 512
|
||||||
|
mov ecx, dword[edx + PORT_DATA.sector_count]
|
||||||
|
mov dword [eax + DISKMEDIAINFO.Capacity], ecx
|
||||||
|
mov ecx, dword[edx + PORT_DATA.sector_count + 4]
|
||||||
|
mov dword [eax + DISKMEDIAINFO.Capacity + 4], ecx
|
||||||
|
pop edx ecx
|
||||||
|
xor eax, eax
|
||||||
|
ret
|
||||||
|
endp
|
||||||
|
|
||||||
|
; Read sectors
|
||||||
|
; return value: 0 = success, otherwise = error
|
||||||
|
proc ahci_read stdcall pdata: dword, buffer: dword, startsector: qword, numsectors_ptr:dword
|
||||||
|
locals
|
||||||
|
cmdslot dd ?
|
||||||
|
cmdheader dd ?
|
||||||
|
cmdtable dd ?
|
||||||
|
numsectors dd ?
|
||||||
|
buffer_pos dd ?
|
||||||
|
buffer_length dd ?
|
||||||
|
endl
|
||||||
|
|
||||||
|
pushad
|
||||||
|
|
||||||
|
mov ecx, ahci_mutex
|
||||||
|
call mutex_lock
|
||||||
|
|
||||||
|
; xor ecx, ecx
|
||||||
|
; mov esi, [buffer]
|
||||||
|
; .print_data:
|
||||||
|
; cmp ecx, 512
|
||||||
|
; jae .end_print_data
|
||||||
|
|
||||||
|
; mov al, byte [esi + ecx]
|
||||||
|
; mov byte [tmpstr], al
|
||||||
|
; mov byte [tmpstr + 1], 0
|
||||||
|
; DEBUGF 1, "0x%x(%s) ", al:1, tmpstr
|
||||||
|
|
||||||
|
; inc ecx
|
||||||
|
; jmp .print_data
|
||||||
|
; .end_print_data:
|
||||||
|
; DEBUGF 1, "\n"
|
||||||
|
|
||||||
|
mov eax, [numsectors_ptr]
|
||||||
|
mov eax, [eax]
|
||||||
|
mov [numsectors], eax
|
||||||
|
|
||||||
|
DEBUGF 1, " ahci_read: buffer = 0x%x, startsector = 0x%x:%x, numsectors = %u\n", [buffer], [startsector], [startsector + 4], eax
|
||||||
|
|
||||||
|
mov esi, [pdata] ; esi - address of PORT_DATA struct of port
|
||||||
|
mov edi, [esi + PORT_DATA.port] ; edi - address of HBA_PORT struct of port
|
||||||
|
mov eax, edi
|
||||||
|
call ahci_find_cmdslot
|
||||||
|
cmp eax, -1
|
||||||
|
jne .cmdslot_found
|
||||||
|
|
||||||
|
DEBUGF 1, "No free cmdslot on port %u\n", [esi + PORT_DATA.portno]
|
||||||
|
jmp .ret
|
||||||
|
|
||||||
|
.cmdslot_found:
|
||||||
|
mov [cmdslot], eax
|
||||||
|
; DEBUGF 1, "Found free cmdslot %u on port %u\n", [cmdslot], [esi + PORT_DATA.portno]
|
||||||
|
|
||||||
|
shl eax, BSF sizeof.HBA_CMD_HDR
|
||||||
|
add eax, [esi + PORT_DATA.clb]
|
||||||
|
mov [cmdheader], eax ; address of virtual mapping of command header
|
||||||
|
mov eax, [cmdslot]
|
||||||
|
mov eax, [esi + eax*4 + PORT_DATA.ctba_arr]
|
||||||
|
mov [cmdtable], eax ; address of virtual mapping of command table of command header
|
||||||
|
|
||||||
|
mov eax, [cmdheader]
|
||||||
|
and [eax + HBA_CMD_HDR.flags1], not 0x1F ; zero out lower 5 bits, they will be used for cfl
|
||||||
|
or [eax + HBA_CMD_HDR.flags1], (sizeof.FIS_REG_H2D / 4) ; set command fis length in dwords
|
||||||
|
movzx bx, [eax + HBA_CMD_HDR.flags1]
|
||||||
|
btr bx, 6 ; flag W = 0
|
||||||
|
mov [eax + HBA_CMD_HDR.flags1], bl
|
||||||
|
movzx bx, [eax + HBA_CMD_HDR.flags2]
|
||||||
|
btr bx, 2 ; flag C = 0
|
||||||
|
mov [eax + HBA_CMD_HDR.flags2], bl
|
||||||
|
|
||||||
|
mov ebx, [numsectors]
|
||||||
|
shl ebx, 9 ; *= 512
|
||||||
|
mov [buffer_length], ebx
|
||||||
|
dec ebx
|
||||||
|
shr ebx, 12 ; /= 4096
|
||||||
|
inc ebx
|
||||||
|
mov [eax + HBA_CMD_HDR.prdtl], bx
|
||||||
|
;DEBUGF 1, " prdtl = %u\n", [eax + HBA_CMD_HDR.prdtl]:2
|
||||||
|
|
||||||
|
; zero out the command table with its prdt entries
|
||||||
|
dec ebx
|
||||||
|
shl ebx, BSF sizeof.HBA_PRDT_ENTRY
|
||||||
|
add ebx, sizeof.HBA_CMD_TBL
|
||||||
|
stdcall _memset, [cmdtable], 0, ebx
|
||||||
|
|
||||||
|
DEBUGF 1, " prdtl = %u\n", [eax + HBA_CMD_HDR.prdtl]:2
|
||||||
|
;jmp .ret
|
||||||
|
|
||||||
|
xor ecx, ecx
|
||||||
|
movzx edx, [eax + HBA_CMD_HDR.prdtl]
|
||||||
|
dec edx
|
||||||
|
mov eax, [buffer]
|
||||||
|
mov [buffer_pos], eax
|
||||||
|
|
||||||
|
.prdt_fill:
|
||||||
|
cmp ecx, edx
|
||||||
|
jae .prdt_fill_end
|
||||||
|
|
||||||
|
mov ebx, [buffer_pos]
|
||||||
|
and ebx, 0xFFF
|
||||||
|
call get_pg_addr ; eax = phys addr
|
||||||
|
add eax, ebx
|
||||||
|
DEBUGF 1, " PHYS = 0x%x\n", eax
|
||||||
|
mov ebx, ecx
|
||||||
|
shl ebx, BSF sizeof.HBA_PRDT_ENTRY
|
||||||
|
add ebx, [cmdtable]
|
||||||
|
add ebx, HBA_CMD_TBL.prdt_entry ; now ebx - address of ecx'th prdt_entry
|
||||||
|
|
||||||
|
mov [ebx + HBA_PRDT_ENTRY.dba], eax
|
||||||
|
mov [ebx + HBA_PRDT_ENTRY.dbau], 0
|
||||||
|
and [ebx + HBA_PRDT_ENTRY.flags], not 0x3FFFFF ; zero out lower 22 bits, they used for byte count
|
||||||
|
or [ebx + HBA_PRDT_ENTRY.flags], 4096 - 1 ; reason why -1 see in spec on this field
|
||||||
|
; or [eax + HBA_CMD_TBL.prdt_entry + HBA_PRDT_ENTRY.flags], 1 shl 31 ; enable interrupt on completion
|
||||||
|
|
||||||
|
add [buffer_pos], 4096
|
||||||
|
sub [buffer_length], 4096
|
||||||
|
|
||||||
|
inc ecx
|
||||||
|
jmp .prdt_fill
|
||||||
|
.prdt_fill_end:
|
||||||
|
|
||||||
|
mov ebx, [buffer_pos]
|
||||||
|
and ebx, 0xFFF
|
||||||
|
call get_pg_addr ; eax = phys addr
|
||||||
|
add eax, ebx
|
||||||
|
DEBUGF 1, " PHYS. = 0x%x\n", eax
|
||||||
|
mov ebx, ecx
|
||||||
|
shl ebx, BSF sizeof.HBA_PRDT_ENTRY
|
||||||
|
add ebx, [cmdtable]
|
||||||
|
add ebx, HBA_CMD_TBL.prdt_entry ; now ebx - address of ecx'th prdt_entry
|
||||||
|
mov [ebx + HBA_PRDT_ENTRY.dba], eax
|
||||||
|
mov [ebx + HBA_PRDT_ENTRY.dbau], 0
|
||||||
|
and [ebx + HBA_PRDT_ENTRY.flags], not 0x3FFFFF ; zero out lower 22 bits, they used for byte count
|
||||||
|
mov eax, [buffer_length]
|
||||||
|
dec eax
|
||||||
|
DEBUGF 1, " DBC = %u\n", eax
|
||||||
|
or [ebx + HBA_PRDT_ENTRY.flags], eax ; reason why -1 see in spec on this field
|
||||||
|
; or [eax + HBA_CMD_TBL.prdt_entry + HBA_PRDT_ENTRY.flags], 1 shl 31 ; enable interrupt on completion
|
||||||
|
|
||||||
|
mov eax, [cmdtable]
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.fis_type], FIS_TYPE_REG_H2D
|
||||||
|
movzx ebx, byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.flags]
|
||||||
|
bts ebx, bit_AHCI_H2D_FLAG_CMD ; Set Command bit in H2D FIS.
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.flags], bl
|
||||||
|
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.command], ATA_CMD_READ_DMA_EX
|
||||||
|
|
||||||
|
mov ebx, dword [startsector]
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.lba0], bl
|
||||||
|
shr ebx, 8
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.lba1], bl
|
||||||
|
shr ebx, 8
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.lba2], bl
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.device], 1 shl 6 ; LBA mode
|
||||||
|
shr ebx, 8
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.lba3], bl
|
||||||
|
mov ebx, dword [startsector + 4]
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.lba4], bl
|
||||||
|
shr ebx, 8
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.lba5], bl
|
||||||
|
|
||||||
|
mov ebx, [numsectors]
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.countl], bl
|
||||||
|
shr ebx, 8
|
||||||
|
mov byte [eax + HBA_CMD_TBL.cfis + FIS_REG_H2D.counth], bl
|
||||||
|
|
||||||
|
; Wait on previous command to complete, before issuing new command.
|
||||||
|
stdcall ahci_port_wait, edi, AHCI_PORT_TIMEOUT
|
||||||
|
|
||||||
|
mov eax, [cmdslot]
|
||||||
|
bts [edi + HBA_PORT.command_issue], eax ; Issue the command
|
||||||
|
|
||||||
|
; Wait for command completion
|
||||||
|
stdcall ahci_port_cmd_wait, edi, eax;, AHCI_PORT_CMD_TIMEOUT
|
||||||
|
|
||||||
|
DEBUGF 1, "sata_error register = 0x%x\n", [edi + HBA_PORT.sata_error]
|
||||||
|
|
||||||
|
DEBUGF 1, "reading completed\n"
|
||||||
|
|
||||||
|
xor ecx, ecx
|
||||||
|
mov esi, [buffer]
|
||||||
|
.print_data:
|
||||||
|
cmp ecx, 512
|
||||||
|
jae .end_print_data
|
||||||
|
|
||||||
|
mov al, byte [esi + ecx]
|
||||||
|
mov byte [tmpstr], al
|
||||||
|
mov byte [tmpstr + 1], 0
|
||||||
|
DEBUGF 1, "0x%x(%s) ", al:1, tmpstr
|
||||||
|
|
||||||
|
inc ecx
|
||||||
|
jmp .print_data
|
||||||
|
.end_print_data:
|
||||||
|
DEBUGF 1, "\n"
|
||||||
|
|
||||||
|
.ret:
|
||||||
|
mov ecx, ahci_mutex
|
||||||
|
call mutex_unlock
|
||||||
|
|
||||||
|
popad
|
||||||
|
xor eax, eax
|
||||||
|
ret
|
||||||
|
endp
|
||||||
|
tmpstr rb 16
|
||||||
|
|
||||||
; Start command engine
|
; Start command engine
|
||||||
; in: eax - address of HBA_PORT structure
|
; in: eax - address of HBA_PORT structure
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user