1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
| ;------------------------------------------------------------------------------
| ;
| ; Copyright (c) 2006, Intel Corporation. All rights reserved.<BR>
| ; SPDX-License-Identifier: BSD-2-Clause-Patent
| ;
| ; Module Name:
| ;
| ; SetMem.nasm
| ;
| ; Abstract:
| ;
| ; SetMem function
| ;
| ; Notes:
| ;
| ;------------------------------------------------------------------------------
|
| SECTION .text
|
| ;------------------------------------------------------------------------------
| ; VOID *
| ; EFIAPI
| ; InternalMemSetMem (
| ; IN VOID *Buffer,
| ; IN UINTN Count,
| ; IN UINT8 Value
| ; );
| ;------------------------------------------------------------------------------
| global ASM_PFX(InternalMemSetMem)
| ASM_PFX(InternalMemSetMem):
| push edi
| mov edx, [esp + 12] ; edx <- Count
| mov edi, [esp + 8] ; edi <- Buffer
| mov al, [esp + 16] ; al <- Value
| xor ecx, ecx
| sub ecx, edi
| and ecx, 63 ; ecx + edi aligns on 16-byte boundary
| jz .0
| cmp ecx, edx
| cmova ecx, edx
| sub edx, ecx
| rep stosb
| .0:
| mov ecx, edx
| and edx, 63
| shr ecx, 6 ; ecx <- # of DQwords to set
| jz @SetBytes
| mov ah, al ; ax <- Value | (Value << 8)
| add esp, -16
| movdqu [esp], xmm0 ; save xmm0
| movd xmm0, eax
| pshuflw xmm0, xmm0, 0 ; xmm0[0..63] <- Value repeats 8 times
| movlhps xmm0, xmm0 ; xmm0 <- Value repeats 16 times
| .1:
| movntdq [edi], xmm0 ; edi should be 16-byte aligned
| movntdq [edi + 16], xmm0
| movntdq [edi + 32], xmm0
| movntdq [edi + 48], xmm0
| add edi, 64
| loop .1
| mfence
| movdqu xmm0, [esp] ; restore xmm0
| add esp, 16 ; stack cleanup
| @SetBytes:
| mov ecx, edx
| rep stosb
| mov eax, [esp + 8] ; eax <- Buffer as return value
| pop edi
| ret
|
|
|
