MdePkg BaseMemoryLibSse2: Convert X64/SetMem32.asm to NASM

[mirror_edk2.git] / MdePkg / Library / BaseMemoryLibSse2 / X64 / SetMem32.nasm

diff --git a/MdePkg/Library/BaseMemoryLibSse2/X64/SetMem32.nasm b/MdePkg/Library/BaseMemoryLibSse2/X64/SetMem32.nasm
new file mode 100644 (file)
index 0000000..a9bb43a
--- /dev/null
+++ b/MdePkg/Library/BaseMemoryLibSse2/X64/SetMem32.nasm
@@ -0,0 +1,68 @@
+;------------------------------------------------------------------------------\r
+;\r
+; Copyright (c) 2006, Intel Corporation. All rights reserved.<BR>\r
+; This program and the accompanying materials\r
+; are licensed and made available under the terms and conditions of the BSD License\r
+; which accompanies this distribution.  The full text of the license may be found at\r
+; http://opensource.org/licenses/bsd-license.php.\r
+;\r
+; THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
+; WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+;\r
+; Module Name:\r
+;\r
+;   SetMem32.nasm\r
+;\r
+; Abstract:\r
+;\r
+;   SetMem32 function\r
+;\r
+; Notes:\r
+;\r
+;------------------------------------------------------------------------------\r
+\r
+    DEFAULT REL\r
+    SECTION .text\r
+\r
+;------------------------------------------------------------------------------\r
+;  VOID *\r
+;  InternalMemSetMem32 (\r
+;    IN VOID   *Buffer,\r
+;    IN UINTN  Count,\r
+;    IN UINT8  Value\r
+;    )\r
+;------------------------------------------------------------------------------\r
+global ASM_PFX(InternalMemSetMem32)\r
+ASM_PFX(InternalMemSetMem32):\r
+    push    rdi\r
+    mov     rdi, rcx\r
+    mov     r9, rdi\r
+    xor     rcx, rcx\r
+    sub     rcx, rdi\r
+    and     rcx, 15\r
+    mov     rax, r8\r
+    jz      .0\r
+    shr     rcx, 2\r
+    cmp     rcx, rdx\r
+    cmova   rcx, rdx\r
+    sub     rdx, rcx\r
+    rep     stosd\r
+.0:\r
+    mov     rcx, rdx\r
+    and     edx, 3\r
+    shr     rcx, 2\r
+    jz      @SetDwords\r
+    movd    xmm0, eax\r
+    pshufd  xmm0, xmm0, 0\r
+.1:\r
+    movntdq [rdi], xmm0\r
+    add     rdi, 16\r
+    loop    .1\r
+    mfence\r
+@SetDwords:\r
+    mov     ecx, edx\r
+    rep     stosd\r
+    mov     rax, r9\r
+    pop     rdi\r
+    ret\r
+\r