MdePkg/Library/BaseMemoryLibSse2/X64/SetMem32.nasm

   1 ;------------------------------------------------------------------------------
   2 ;
   3 ; Copyright (c) 2006, Intel Corporation. All rights reserved.<BR>
   4 ; This program and the accompanying materials
   5 ; are licensed and made available under the terms and conditions of the BSD License
   6 ; which accompanies this distribution.  The full text of the license may be found at
   7 ; http://opensource.org/licenses/bsd-license.php.
   8 ;
   9 ; THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  10 ; WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  11 ;
  12 ; Module Name:
  13 ;
  14 ;   SetMem32.nasm
  15 ;
  16 ; Abstract:
  17 ;
  18 ;   SetMem32 function
  19 ;
  20 ; Notes:
  21 ;
  22 ;------------------------------------------------------------------------------
  23
  24     DEFAULT REL
  25     SECTION .text
  26
  27 ;------------------------------------------------------------------------------
  28 ;  VOID *
  29 ;  InternalMemSetMem32 (
  30 ;    IN VOID   *Buffer,
  31 ;    IN UINTN  Count,
  32 ;    IN UINT8  Value
  33 ;    )
  34 ;------------------------------------------------------------------------------
  35 global ASM_PFX(InternalMemSetMem32)
  36 ASM_PFX(InternalMemSetMem32):
  37     push    rdi
  38     mov     rdi, rcx
  39     mov     r9, rdi
  40     xor     rcx, rcx
  41     sub     rcx, rdi
  42     and     rcx, 15
  43     mov     rax, r8
  44     jz      .0
  45     shr     rcx, 2
  46     cmp     rcx, rdx
  47     cmova   rcx, rdx
  48     sub     rdx, rcx
  49     rep     stosd
  50 .0:
  51     mov     rcx, rdx
  52     and     edx, 3
  53     shr     rcx, 2
  54     jz      @SetDwords
  55     movd    xmm0, eax
  56     pshufd  xmm0, xmm0, 0
  57 .1:
  58     movntdq [rdi], xmm0
  59     add     rdi, 16
  60     loop    .1
  61     mfence
  62 @SetDwords:
  63     mov     ecx, edx
  64     rep     stosd
  65     mov     rax, r9
  66     pop     rdi
  67     ret
  68