Fix an issue in page allocation.

[mirror_edk2.git] / MdePkg / Library / BaseMemoryLibMmx / X64 / SetMem16.asm

;------------------------------------------------------------------------------\r
;\r
; Copyright (c) 2006, Intel Corporation\r
; All rights reserved. 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
;   SetMem16.asm\r
;\r
; Abstract:\r
;\r
;   SetMem16 function\r
;\r
; Notes:\r
;\r
;------------------------------------------------------------------------------\r
\r
    .code\r
\r
;------------------------------------------------------------------------------\r
; VOID *\r
; EFIAPI\r
; InternalMemSetMem16 (\r
;   OUT     VOID                      *Buffer,\r
;   IN      UINTN                     Length,\r
;   IN      UINT16                    Value\r
;   );\r
;------------------------------------------------------------------------------\r
InternalMemSetMem16 PROC    USES    rdi\r
    mov     rax, r8\r
    DB      48h, 0fh, 6eh, 0c0h         ; movd mm0, rax\r
    mov     r8, rcx\r
    mov     rdi, r8\r
    mov     rcx, rdx\r
    and     edx, 3\r
    shr     rcx, 2\r
    jz      @SetWords\r
    DB      0fh, 70h, 0C0h, 00h         ; pshufw mm0, mm0, 0h\r
@@:\r
    DB      0fh, 0e7h, 07h              ; movntq [rdi], mm0\r
    add     rdi, 8\r
    loop    @B\r
    mfence\r
@SetWords:\r
    mov     ecx, edx\r
    rep     stosw\r
    mov     rax, r8\r
    ret\r
InternalMemSetMem16 ENDP\r
\r
    END\r