From 998d51f9e8de8cc7a927eb64da2a5971e3689c0d Mon Sep 17 00:00:00 2001
From: Abdur-Rahman Mansoor <armansoor@missingno.dev>
Date: Sat, 10 Aug 2024 19:27:58 -0400
Subject: [PATCH] fix: shutdown freeze (#5)

---
 drivers/nvme/nvme.asm | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/drivers/nvme/nvme.asm b/drivers/nvme/nvme.asm
index b93b8d0..b095c37 100644
--- a/drivers/nvme/nvme.asm
+++ b/drivers/nvme/nvme.asm
@@ -39,9 +39,10 @@ struct DISKMEDIAINFO
 	capacity 	dq ?
 ends
 
-proc START stdcall, reason:dword
+proc START c, reason:dword, cmdline:dword
 local   AnythingLoadedSuccessfully db 0
 
+	push 	esi edi
 	cmp     [reason], DRV_ENTRY
 	jne     .err
 
@@ -81,11 +82,12 @@ local   AnythingLoadedSuccessfully db 0
 	cmp 	[AnythingLoadedSuccessfully], 0
 	jz 	.err
 	invoke  RegService, my_service, service_proc
+	pop 	edi esi
 	ret
 
 .err:
-	call    nvme_cleanup
-      	xor     eax, eax
+	;call    nvme_cleanup
+	pop 	edi esi
       	ret
 
 endp