[CODEREVIEW]

Co-authored-by: Hermès BÉLUSCA - MAÏTO <[email protected]>
DarkFire01 · Oct 11, 2023 · dbd3356 · dbd3356
1 parent e04cbd3
commit dbd3356
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 15 deletions.
diff --git a/boot/freeldr/freeldr/arch/uefi/amd64/uefiasm.S b/boot/freeldr/freeldr/arch/uefi/amd64/uefiasm.S
@@ -25,7 +25,6 @@ _exituefi:
     push rdi
     push rbx
 
-
     /* Save the old stack */
     mov rbx, rsp
 
@@ -46,6 +45,12 @@ _exituefi:
     pop rsi
     pop rbp
 
+#ifdef _USE_ML
+    lgdt fword ptr [_gdtptr]
+#else
+    lgdt cs:[_gdtptr][rip] /* GAS isn't my friend - avoid letting it generate absolute addressing */
+#endif
+
     /* All done */
     ret
 
@@ -58,16 +63,6 @@ _changestack:
     call ExecuteLoaderCleanly[rip]
     ret
 
-// void _changegdt(VOID)
-PUBLIC _changegdt
-_changegdt:
-#ifdef _USE_ML
-    lgdt fword ptr [_gdtptr]
-#else
-    lgdt cs:[_gdtptr][rip] /* GAS isn't my friend - avoid letting it generate absolute addressing */
-#endif
-    ret
-
 .align 8
 gdt:
     .word HEX(0000), HEX(0000), HEX(0000), HEX(0000) /* 00: NULL descriptor */

diff --git a/boot/freeldr/freeldr/arch/uefi/i386/uefiasm.S b/boot/freeldr/freeldr/arch/uefi/i386/uefiasm.S
@@ -47,6 +47,13 @@ __exituefi:
     pop esi
     pop ebp
 
+#ifdef _USE_ML
+    lidt fword ptr ds:[_i386idtptr]
+    lgdt fword ptr [_gdtptr]
+#else
+    lgdt cs:[_gdtptr]
+    lidt _i386idtptr
+#endif
 
     /* All done */
     ret

diff --git a/boot/freeldr/freeldr/arch/uefi/uefildr.c b/boot/freeldr/freeldr/arch/uefi/uefildr.c
@@ -81,7 +81,7 @@ VOID __cdecl Reboot(VOID)
     WARN("Stopping freeloader something has gone wrong - Freezing");
     for(;;)
     {
-
+        NOTHING;
     }
 }
 #endif
diff --git a/boot/freeldr/freeldr/arch/uefi/uefimem.c b/boot/freeldr/freeldr/arch/uefi/uefimem.c
@@ -40,7 +40,6 @@ EFI_HANDLE PublicBootHandle;
 PVOID ExitStack;
 PVOID EndofExitStack;
 
-void _changegdt(VOID);
 void _exituefi(VOID);
 
 /* FUNCTIONS *****************************************************************/
@@ -89,7 +88,7 @@ VOID
 UefiSetMemory(
     _Inout_ PFREELDR_MEMORY_DESCRIPTOR MemoryMap,
     _In_ ULONG_PTR BaseAddress,
-    _In_ PFN_COUNT Size,
+    _In_ PFN_COUNT SizeInPages,
     _In_ TYPE_OF_MEMORY MemoryType)
 {
     ULONG_PTR BasePage, PageCount;
@@ -327,5 +326,4 @@ VOID
 UefiPrepareForReactOS(VOID)
 {
     _exituefi();
-    _changegdt();
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -81,7 +81,7 @@ VOID __cdecl Reboot(VOID) @@
         WARN("Stopping freeloader something has gone wrong - Freezing");
         for(;;)
         {
+            NOTHING;
         }
     }
     #endif