2 files changed, 70 insertions, 0 deletions
diff --git a/MdePkg/Library/BaseMemoryLibSse2/BaseMemoryLibSse2.inf b/MdePkg/Library/BaseMemoryLibSse2/BaseMemoryLibSse2.inf
index 8ab21350b4..bfa40b066b 100644
--- a/MdePkg/Library/BaseMemoryLibSse2/BaseMemoryLibSse2.inf
+++ b/MdePkg/Library/BaseMemoryLibSse2/BaseMemoryLibSse2.inf
@@ -59,6 +59,7 @@
   Ia32/ZeroMem.S
   Ia32/SetMem64.nasm
   Ia32/SetMem64.S
+  Ia32/SetMem32.nasm
   Ia32/SetMem32.S
   Ia32/SetMem16.S
   Ia32/SetMem.S
@@ -77,6 +78,7 @@
   Ia32/ZeroMem.asm
   Ia32/SetMem64.nasm
   Ia32/SetMem64.asm
+  Ia32/SetMem32.nasm
   Ia32/SetMem32.asm
   Ia32/SetMem16.asm
   Ia32/SetMem.asm
diff --git a/MdePkg/Library/BaseMemoryLibSse2/Ia32/SetMem32.nasm b/MdePkg/Library/BaseMemoryLibSse2/Ia32/SetMem32.nasm
new file mode 100644
index 0000000000..e0848f2041
--- /dev/null
+++ b/MdePkg/Library/BaseMemoryLibSse2/Ia32/SetMem32.nasm
@@ -0,0 +1,68 @@
+;------------------------------------------------------------------------------
+;
+; Copyright (c) 2006, Intel Corporation. All rights reserved.<BR>
+; This program and the accompanying materials
+; are licensed and made available under the terms and conditions of the BSD License
+; which accompanies this distribution.  The full text of the license may be found at
+; http://opensource.org/licenses/bsd-license.php.
+;
+; THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+; WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+;
+; Module Name:
+;
+;   SetMem32.nasm
+;
+; Abstract:
+;
+;   SetMem32 function
+;
+; Notes:
+;
+;------------------------------------------------------------------------------
+
+    SECTION .text
+
+;------------------------------------------------------------------------------
+;  VOID *
+;  EFIAPI
+;  InternalMemSetMem32 (
+;    IN VOID   *Buffer,
+;    IN UINTN  Count,
+;    IN UINT32 Value
+;    );
+;------------------------------------------------------------------------------
+global ASM_PFX(InternalMemSetMem32)
+ASM_PFX(InternalMemSetMem32):
+    push    edi
+    mov     edx, [esp + 12]
+    mov     edi, [esp + 8]
+    xor     ecx, ecx
+    sub     ecx, edi
+    and     ecx, 15                     ; ecx + edi aligns on 16-byte boundary
+    mov     eax, [esp + 16]
+    jz      .0
+    shr     ecx, 2
+    cmp     ecx, edx
+    cmova   ecx, edx
+    sub     edx, ecx
+    rep     stosd
+.0:
+    mov     ecx, edx
+    and     edx, 3
+    shr     ecx, 2
+    jz      @SetDwords
+    movd    xmm0, eax
+    pshufd  xmm0, xmm0, 0
+.1:
+    movntdq [edi], xmm0
+    add     edi, 16
+    loop    .1
+    mfence
+@SetDwords:
+    mov     ecx, edx
+    rep     stosd
+    mov     eax, [esp + 8]
+    pop     edi
+    ret
+