[mirror_edk2.git] / EdkCompatibilityPkg / Foundation / Library / CompilerStub / Ia32 / memset.asm

;------------------------------------------------------------------------------\r
;\r
; Copyright (c) 2007, 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
;   SetMem.asm\r
;\r
; Abstract:\r
;\r
;   memset function\r
;\r
; Notes:\r
;\r
;------------------------------------------------------------------------------\r
\r
    .686\r
    .model  flat,C\r
    .mmx\r
    .code\r
\r
;------------------------------------------------------------------------------\r
;  VOID *\r
;  memset (\r
;    OUT VOID   *Buffer,\r
;    IN  UINT8  Value,\r
;    IN  UINTN  Count\r
;    )\r
;------------------------------------------------------------------------------\r
memset   PROC    USES    edi\r
    mov     al, [esp + 12]\r
    mov     ah, al\r
    shrd    edx, eax, 16\r
    shld    eax, edx, 16\r
    mov     ecx, [esp + 16]             ; ecx <- Count\r
    cmp     ecx, 0                      ; if Count == 0, do nothing\r
    je      @SetDone\r
    mov     edi, [esp + 8]              ; edi <- Buffer\r
    mov     edx, ecx\r
    and     edx, 7\r
    shr     ecx, 3                      ; # of Qwords to set\r
    jz      @SetBytes\r
    add     esp, -10h\r
    movq    [esp], mm0                  ; save mm0\r
    movq    [esp + 8], mm1              ; save mm1\r
    movd    mm0, eax\r
    movd    mm1, eax\r
    psllq   mm0, 32\r
    por     mm0, mm1                    ; fill mm0 with 8 Value's\r
@@:\r
    movq    [edi], mm0\r
    add     edi, 8\r
    loop    @B\r
    movq    mm0, [esp]                  ; restore mm0\r
    movq    mm1, [esp + 8]              ; restore mm1\r
    add     esp, 10h                    ; stack cleanup\r
@SetBytes:\r
    mov     ecx, edx\r
    rep     stosb\r
@SetDone:    \r
    mov     eax, [esp + 8]              ; eax <- Buffer as return value\r
    ret\r
memset   ENDP\r
\r
    END\r
Commit	Line	Data
3eb9473e	1	;------------------------------------------------------------------------------\r
	2	;\r
	3	; Copyright (c) 2007, Intel Corporation\r
	4	; All rights reserved. This program and the accompanying materials\r
	5	; are licensed and made available under the terms and conditions of the BSD License\r
	6	; which accompanies this distribution. The full text of the license may be found at\r
	7	; http://opensource.org/licenses/bsd-license.php\r
	8	;\r
	9	; THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	10	; WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	11	;\r
	12	; Module Name:\r
	13	;\r
	14	; SetMem.asm\r
	15	;\r
	16	; Abstract:\r
	17	;\r
	18	; memset function\r
	19	;\r
	20	; Notes:\r
	21	;\r
	22	;------------------------------------------------------------------------------\r
	23	\r
	24	.686\r
	25	.model flat,C\r
	26	.mmx\r
	27	.code\r
	28	\r
	29	;------------------------------------------------------------------------------\r
	30	; VOID *\r
	31	; memset (\r
	32	; OUT VOID *Buffer,\r
	33	; IN UINT8 Value,\r
	34	; IN UINTN Count\r
	35	; )\r
	36	;------------------------------------------------------------------------------\r
	37	memset PROC USES edi\r
	38	mov al, [esp + 12]\r
	39	mov ah, al\r
	40	shrd edx, eax, 16\r
	41	shld eax, edx, 16\r
	42	mov ecx, [esp + 16] ; ecx <- Count\r
	43	cmp ecx, 0 ; if Count == 0, do nothing\r
	44	je @SetDone\r
	45	mov edi, [esp + 8] ; edi <- Buffer\r
	46	mov edx, ecx\r
	47	and edx, 7\r
	48	shr ecx, 3 ; # of Qwords to set\r
	49	jz @SetBytes\r
	50	add esp, -10h\r
	51	movq [esp], mm0 ; save mm0\r
	52	movq [esp + 8], mm1 ; save mm1\r
	53	movd mm0, eax\r
	54	movd mm1, eax\r
	55	psllq mm0, 32\r
	56	por mm0, mm1 ; fill mm0 with 8 Value's\r
	57	@@:\r
	58	movq [edi], mm0\r
	59	add edi, 8\r
	60	loop @B\r
	61	movq mm0, [esp] ; restore mm0\r
	62	movq mm1, [esp + 8] ; restore mm1\r
	63	add esp, 10h ; stack cleanup\r
	64	@SetBytes:\r
65	mov ecx, edx\r
66	rep stosb\r
67	@SetDone: \r
68	mov eax, [esp + 8] ; eax <- Buffer as return value\r
69	ret\r
70	memset ENDP\r
71	\r
72	END\r