]>
Commit | Line | Data |
---|---|---|
878ddf1f | 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 | .686\r | |
25 | .model flat,C\r | |
26 | .xmm\r | |
27 | .code\r | |
28 | \r | |
29 | ;------------------------------------------------------------------------------\r | |
30 | ; VOID *\r | |
31 | ; _mem_SetMem (\r | |
32 | ; IN VOID *Buffer,\r | |
33 | ; IN UINTN Count,\r | |
34 | ; IN UINT8 Value\r | |
35 | ; )\r | |
36 | ;------------------------------------------------------------------------------\r | |
37 | InternalMemSetMem PROC USES edi\r | |
38 | mov edx, [esp + 12] ; edx <- Count\r | |
39 | mov edi, [esp + 8] ; edi <- Buffer\r | |
40 | mov al, [esp + 16] ; al <- Value\r | |
41 | xor ecx, ecx\r | |
42 | sub ecx, edi\r | |
43 | and ecx, 15 ; ecx + edi aligns on 16-byte boundary\r | |
44 | jz @F\r | |
45 | cmp ecx, edx\r | |
46 | cmova ecx, edx\r | |
47 | sub edx, ecx\r | |
48 | rep stosb\r | |
49 | @@:\r | |
50 | mov ecx, edx\r | |
51 | and edx, 15\r | |
52 | shr ecx, 4 ; ecx <- # of DQwords to set\r | |
53 | jz @SetBytes\r | |
54 | mov ah, al ; ax <- Value | (Value << 8)\r | |
55 | add esp, -16\r | |
56 | movdqu [esp], xmm0 ; save xmm0\r | |
57 | movd xmm0, eax\r | |
58 | pshuflw xmm0, xmm0, 0 ; xmm0[0..63] <- Value repeats 8 times\r | |
59 | movlhps xmm0, xmm0 ; xmm0 <- Value repeats 16 times\r | |
60 | @@:\r | |
61 | movntdq [edi], xmm0 ; edi should be 16-byte aligned\r | |
62 | add edi, 16\r | |
63 | loop @B\r | |
64 | mfence\r | |
65 | movdqu xmm0, [esp] ; restore xmm0\r | |
66 | add esp, 16 ; stack cleanup\r | |
67 | @SetBytes:\r | |
68 | mov ecx, edx\r | |
69 | rep stosb\r | |
70 | mov eax, [esp + 8] ; eax <- Buffer as return value\r | |
71 | ret\r | |
72 | InternalMemSetMem ENDP\r | |
73 | \r | |
74 | END\r |