]>
Commit | Line | Data |
---|---|---|
f1baef62 | 1 | ;------------------------------------------------------------------------------\r |
2 | ;\r | |
3 | ; Copyright (c) 2006, 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 | ; SetMem function\r | |
19 | ;\r | |
20 | ; Notes:\r | |
21 | ;\r | |
22 | ;------------------------------------------------------------------------------\r | |
23 | \r | |
24 | .code\r | |
25 | \r | |
26 | ;------------------------------------------------------------------------------\r | |
27 | ; VOID *\r | |
28 | ; InternalMemSetMem (\r | |
29 | ; IN VOID *Buffer,\r | |
30 | ; IN UINTN Count,\r | |
31 | ; IN UINT8 Value\r | |
32 | ; )\r | |
33 | ;------------------------------------------------------------------------------\r | |
34 | InternalMemSetMem PROC USES rdi\r | |
35 | mov rdi, rcx ; rdi <- Buffer\r | |
36 | mov al, r8b ; al <- Value\r | |
37 | mov r9, rdi ; r9 <- Buffer as return value\r | |
38 | xor rcx, rcx\r | |
39 | sub rcx, rdi\r | |
40 | and rcx, 15 ; rcx + rdi aligns on 16-byte boundary\r | |
41 | jz @F\r | |
42 | cmp rcx, rdx\r | |
43 | cmova rcx, rdx\r | |
44 | sub rdx, rcx\r | |
45 | rep stosb\r | |
46 | @@:\r | |
47 | mov rcx, rdx\r | |
48 | and rdx, 15\r | |
49 | shr rcx, 4\r | |
50 | jz @SetBytes\r | |
51 | mov ah, al ; ax <- Value repeats twice\r | |
52 | movdqa [rsp + 10h], xmm0 ; save xmm0\r | |
53 | movd xmm0, eax ; xmm0[0..16] <- Value repeats twice\r | |
54 | pshuflw xmm0, xmm0, 0 ; xmm0[0..63] <- Value repeats 8 times\r | |
55 | movlhps xmm0, xmm0 ; xmm0 <- Value repeats 16 times\r | |
56 | @@:\r | |
57 | movntdq [rdi], xmm0 ; rdi should be 16-byte aligned\r | |
58 | add rdi, 16\r | |
59 | loop @B\r | |
60 | mfence\r | |
61 | movdqa xmm0, [rsp + 10h] ; restore xmm0\r | |
62 | @SetBytes:\r | |
63 | mov ecx, edx ; high 32 bits of rcx are always zero\r | |
64 | rep stosb\r | |
65 | mov rax, r9 ; rax <- Return value\r | |
66 | ret\r | |
67 | InternalMemSetMem ENDP\r | |
68 | \r | |
69 | END\r |