MdePkg BaseMemoryLibMmx: Convert X64/SetMem.asm to NASM

[mirror_edk2.git] / MdePkg / Library / BaseMemoryLibMmx / X64 / SetMem.nasm

;------------------------------------------------------------------------------\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
;   SetMem.nasm\r
;\r
; Abstract:\r
;\r
;   SetMem function\r
;\r
; Notes:\r
;\r
;------------------------------------------------------------------------------\r
\r
    DEFAULT REL\r
    SECTION .text\r
\r
;------------------------------------------------------------------------------\r
; VOID *\r
; EFIAPI\r
; InternalMemSetMem (\r
;   OUT     VOID                      *Buffer,\r
;   IN      UINTN                     Length,\r
;   IN      UINT8                     Value\r
;   );\r
;------------------------------------------------------------------------------\r
global ASM_PFX(InternalMemSetMem)\r
ASM_PFX(InternalMemSetMem):\r
    push    rdi\r
    mov     rax, r8\r
    mov     ah, al\r
    DB      0x48, 0xf, 0x6e, 0xc0         ; movd mm0, rax\r
    mov     r8, rcx\r
    mov     rdi, r8                     ; rdi <- Buffer\r
    mov     rcx, rdx\r
    and     edx, 7\r
    shr     rcx, 3\r
    jz      @SetBytes\r
    DB      0xf, 0x70, 0xC0, 0x0         ; pshufw mm0, mm0, 0h\r
.0:\r
    DB      0xf, 0xe7, 0x7              ; movntq [rdi], mm0\r
    add     rdi, 8\r
    loop    .0\r
    mfence\r
@SetBytes:\r
    mov     ecx, edx\r
    rep     stosb\r
    mov     rax, r8\r
    pop     rdi\r
    ret\r
\r