GSoC/kolibrios-nvme-driver
2
0
Fork 0
mirror of https://git.missingno.dev/kolibrios-nvme-driver/ synced 2025-01-03 19:35:56 +01:00
Code Issues 2 Actions Activity

separate determine_active_nsids and is_active_namespace

Browse Source
This commit is contained in:
Abdur-Rahman Mansoor 2024-06-22 15:18:43 -04:00
parent 1c69e25c2b
commit 0b16efacd3
1 changed files with 58 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

108
drivers/nvme/nvme.asm
View File

@ -165,73 +165,81 @@ proc create_namespace stdcall, pci:dword, cid:word
endp endp
; See page 248 of the NVMe 1.4 specification for reference proc is_active_namespace stdcall, pci:dword, nsid:dword
proc determine_active_nsids stdcall, pci:dword
push esi edi
push esi edi ebx
xor esi, esi
xor edi, edi
invoke KernelAlloc, 0x1000 invoke KernelAlloc, 0x1000
test eax, eax test eax, eax
jz .fail jnz @f
pop edi esi
ret
@@:
mov esi, eax mov esi, eax
invoke GetPhysAddr invoke GetPhysAddr
stdcall nvme_identify, [pci], 0, eax, CNS_IDCS stdcall nvme_identify, [pci], [nsid], eax, CNS_IDNS
mov eax, dword [esi + IDENTC.nn]
mov edx, [pci]
mov dword [edx + pcidev.nn], eax
test eax, eax
jz .fail
invoke KernelAlloc, 0x1000
test eax, eax
jz .fail
mov edi, eax
invoke GetPhysAddr
mov ebx, eax
xor ecx, ecx xor ecx, ecx
.loop: @@:
cmp ecx, dword [esi + IDENTC.nn] mov eax, dword [esi + ecx * 4]
jg .success
stdcall nvme_identify, [pci], ecx, ebx, CNS_IDNS ; identify N'th namespace
push ecx
xor ecx, ecx
; We want to check if entire struct is zeroed out, if so, it is not an active NSID,
; otherwise it is
.inner_loop:
mov eax, dword [edi + ecx * 4]
test eax, eax test eax, eax
jnz .is_active_nsid jnz .is_active_nsid
inc ecx inc ecx
cmp ecx, 0x1000 / 4 cmp ecx, 0x1000 / 4
jne .inner_loop jne @b
pop ecx
inc ecx .not_active_nsid:
jmp .loop invoke KernelFree, esi
pop edi esi
.is_active_nsid:
pop ecx ; pop active NSID value into ecx
mov eax, [pci]
mov eax, dword [eax + pcidev.nsids]
mov dword [eax + ecx * 4], ecx
inc ecx
jmp .loop
.fail:
invoke KernelFree, esi ; assumes KernelFree can check if its nullptr (hopefully)
invoke KernelFree, edi
pop ebx edi esi
xor eax, eax xor eax, eax
ret ret
.success: .is_active_nsid:
invoke KernelFree, esi invoke KernelFree, esi
invoke KernelFree, edi pop edi esi
pop ebx edi esi
xor eax, eax xor eax, eax
inc eax inc eax
ret ret
endp
; See page 248 of the NVMe 1.4 specification for reference
proc determine_active_nsids stdcall, pci:dword
push esi edi
invoke KernelAlloc, 0x1000
test eax, eax
jnz @f
pop edi esi
ret
@@:
mov esi, eax
invoke GetPhysAddr
stdcall nvme_identify, [pci], 0, eax, CNS_IDCS
mov edi, [pci]
mov eax, dword [esi + IDENTC.nn]
mov dword [edi + pcidev.nn], eax
mov edi, [edi + pcidev.nsids]
xor ecx, ecx
.loop:
cmp ecx, dword [esi + IDENTC.nn]
je .ret
stdcall is_active_namespace, [pci], ecx
test eax, eax
jz .not_active_namespace
mov dword [edi + ecx * 4], ecx
.not_active_namespace:
inc ecx
jmp .loop
.ret:
invoke KernelFree, esi
pop edi esi
xor eax, eax
ret
endp endp