Need use 32-bit register to hold 32-bit stack address.

[mirror_edk2.git] / MdePkg / Library / BaseLib / X64 / DisablePaging64.asm

diff --git a/MdePkg/Library/BaseLib/X64/DisablePaging64.asm b/MdePkg/Library/BaseLib/X64/DisablePaging64.asm
index 502a54c4e5657e7919000c5ee26e45546ea73885..a2995daa33305cdf40752d4277be2e3978c0be13 100644 (file)
--- a/MdePkg/Library/BaseLib/X64/DisablePaging64.asm
+++ b/MdePkg/Library/BaseLib/X64/DisablePaging64.asm
@@ -37,9 +37,9 @@
 InternalX86DisablePaging64    PROC\r
     cli\r
     lea     rsi, @F                     ; rsi <- The start address of transition code\r
-    mov     rdi, [rsp + 28h]            ; rdi <- New stack\r
-    sub     rdi, 64                     ; rdi <- use 64 byte in stack to hold transition code  \r
-    mov     r10, rdi                    ; r10 <- The start address of transicition code below 4G\r
+    mov     edi, [rsp + 28h]            ; rdi <- New stack\r
+    sub     edi, 64                     ; rdi <- use 64 byte in stack to hold transition code  \r
+    mov     r10d, edi                   ; r10 <- The start address of transicition code below 4G\r
     lea     rax, mTransitionEnd         ; rax <- end of transition code\r
     sub     rax, rsi                    ; rax <- The size of transition piece code \r
     push    rcx                         ; save rcx to stack\r