Return buffer in rax

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@4032 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/MdePkg/Library/BaseMemoryLibRepStr/x64/SetMem.asm b/MdePkg/Library/BaseMemoryLibRepStr/x64/SetMem.asm
index 5306520169f1ff6a3f08ee0b39db5c1207f70d69..2c9b211516ebc95069aa3861216a51f7a334d5e6 100644 (file)
--- a/MdePkg/Library/BaseMemoryLibRepStr/x64/SetMem.asm
+++ b/MdePkg/Library/BaseMemoryLibRepStr/x64/SetMem.asm
@@ -33,11 +33,12 @@
 ;    )\r
 ;------------------------------------------------------------------------------\r
 InternalMemSetMem   PROC    USES    rdi\r
-    mov     rax, r8\r
-    mov     rdi, rcx\r
-    xchg    rcx, rdx\r
+    push    rcx        ; push Buffer\r
+    mov     rax, r8    ; rax = Value\r
+    mov     rdi, rcx   ; rdi = Buffer\r
+    mov     rcx, rdx   ; rcx = Count\r
     rep     stosb\r
-    mov     rax, rdx\r
+    pop     rax        ; rax = Buffer\r
     ret\r
 InternalMemSetMem   ENDP\r
 \r