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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
|
; Licensed to the .NET Foundation under one or more agreements.
; The .NET Foundation licenses this file to you under the MIT license.
; See the LICENSE file in the project root for more information.
;; ==++==
;;
;;
;; ==--==
#include "ksarm64.h"
TEXTAREA
; Calls to JIT_MemSet is emitted by jit for initialization of large structs.
; We need to provide our own implementation of memset instead of using the ones in crt because crt implementation does not gurantee
; that aligned 8/4/2 - byte memory will be written atomically. This is required because members in a struct can be read atomically
; and their values should be written atomically.
;
;
;void JIT_MemSet(void *dst, int val, SIZE_T count)
;{
; uintptr_t valEx = (char)val;
; valEx = valEx | valEx << 8;
; valEx = valEx | valEx << 16;
; valEx = valEx | valEx << 32;
;
; // If not aligned then make it 8-byte aligned
; if(((uintptr_t)dst&0x7) != 0)
; {
; if(((uintptr_t)dst&0x3) == 0)
; {
; *(UINT*)dst = (UINT)valEx;
; dst = (UINT*)dst + 1;
; count-=4;
; }
; else if(((uintptr_t)dst&0x1) == 0)
; {
; while(count > 0 && ((uintptr_t)dst&0x7) != 0)
; {
; *(short*)dst = (short)valEx;
; dst = (short*)dst + 1;
; count-=2;
; }
; }
; else
; {
; while(count > 0 && ((uintptr_t)dst&0x7) != 0)
; {
; *(char*)dst = (char)valEx;
; dst = (char*)dst + 1;
; count--;
; }
; }
; }
;
; while(count >= 8)
; {
; *(uintptr_t*)dst = valEx;
; dst = (uintptr_t*)dst + 1;
; count-=8;
; }
;
; if(count & 4)
; {
; *(UINT*)dst = (UINT)valEx;
; dst = (UINT*)dst + 1;
; }
;
; if(count & 2)
; {
; *(short*)dst = (short)valEx;
; dst = (short*)dst + 1;
; }
;
; if(count & 1)
; {
; *(char*)dst = (char)valEx;
; }
;}
;
; Assembly code corresponding to above C++ method. JIT_MemSet can AV and clr exception personality routine needs to
; determine if the exception has taken place inside JIT_Memset in order to throw corresponding managed exception.
; Determining this is slow if the method were implemented as C++ method (using unwind info). In .asm file by adding JIT_MemSet_End
; marker it can be easily determined if exception happened in JIT_MemSet. Therefore, JIT_MemSet has been written in assembly instead of
; as C++ method.
LEAF_ENTRY JIT_MemSet
sxtb w8,w1
sxtw x8,w8
orr x8,x8,x8 lsl #8
orr x8,x8,x8 lsl #0x10
orr x9,x8,x8 lsl #0x20
and x8,x0,#7
cbz x8,JIT_MemSet_0x7c
and x8,x0,#3
cbnz x8,JIT_MemSet_0x38
str w9,[x0]
add x0,x0,#4
mov x8,#-4
add x2,x2,x8
b JIT_MemSet_0x7c
JIT_MemSet_0x38
cbz x2,JIT_MemSet_0x7c
tbnz x0,#0,JIT_MemSet_0x60
JIT_MemSet_0x40
and x8,x0,#7
cbz x8,JIT_MemSet_0x7c
strh w9,[x0]
add x0,x0,#2
mov x8,#-2
add x2,x2,x8
cbnz x2,JIT_MemSet_0x40
b JIT_MemSet_0x7c
JIT_MemSet_0x60
and x8,x0,#7
cbz x8,JIT_MemSet_0x7c
strb w9,[x0]
add x0,x0,#1
mov x8,#-1
add x2,x2,x8
cbnz x2,JIT_MemSet_0x60
JIT_MemSet_0x7c
cmp x2,#8
blo JIT_MemSet_0xb8
lsr x8,x2,#3
mov x11,x8
mov x10,x0
add x8,x10,x11 lsl #3
JIT_MemSet_0x9c
cmp x10,x8
beq JIT_MemSet_0xac
str x9,[x10],#8
b JIT_MemSet_0x9c
JIT_MemSet_0xac
mov x8,#-8
madd x2,x11,x8,x2
add x0,x0,x11 lsl #3
JIT_MemSet_0xb8
tbz x2,#2,JIT_MemSet_0xc4
str w9,[x0]
add x0,x0,#4
JIT_MemSet_0xc4
tbz x2,#1,JIT_MemSet_0xd0
strh w9,[x0]
add x0,x0,#2
JIT_MemSet_0xd0
tbz x2,#0,JIT_MemSet_0xd8
strb w9,[x0]
JIT_MemSet_0xd8
ret lr
LEAF_END
LEAF_ENTRY JIT_MemSet_End
LEAF_END
; See comments above for JIT_MemSet
;void JIT_MemCpy(void *dst, const void *src, SIZE_T count)
;{
; // If not aligned then make it 8-byte aligned
; if(((uintptr_t)dst&0x7) != 0)
; {
; if(((uintptr_t)dst&0x3) == 0)
; {
; *(UINT*)dst = *(UINT*)src;
; dst = (UINT*)dst + 1;
; src = (UINT*)src + 1;
; count-=4;
; }
; else if(((uintptr_t)dst&0x1) == 0)
; {
; while(count > 0 && ((uintptr_t)dst&0x7) != 0)
; {
; *(short*)dst = *(short*)src;
; dst = (short*)dst + 1;
; src = (short*)src + 1;
; count-=2;
; }
; }
; else
; {
; while(count > 0 && ((uintptr_t)dst&0x7) != 0)
; {
; *(char*)dst = *(char*)src;
; dst = (char*)dst + 1;
; src = (char*)src + 1;
; count--;
; }
; }
; }
;
; while(count >= 8)
; {
; *(uintptr_t*)dst = *(uintptr_t*)src;
; dst = (uintptr_t*)dst + 1;
; src = (uintptr_t*)src + 1;
; count-=8;
; }
;
; if(count & 4)
; {
; *(UINT*)dst = *(UINT*)src;
; dst = (UINT*)dst + 1;
; src = (UINT*)src + 1;
; }
;
; if(count & 2)
; {
; *(short*)dst = *(short*)src;
; dst = (short*)dst + 1;
; src = (short*)src + 1;
; }
;
; if(count & 1)
; {
; *(char*)dst = *(char*)src;
; }
;}
;
; Assembly code corresponding to above C++ method.
; See comments above for JIT_MemSet method
LEAF_ENTRY JIT_MemCpy
and x8,x0,#7
cbz x8,JIT_MemCpy_0x80
and x8,x0,#3
cbnz x8,JIT_MemCpy_0x2c
ldr w8,[x1]
str w8,[x0]
add x0,x0,#4
add x1,x1,#4
mov x8,#-4
add x2,x2,x8
b JIT_MemCpy_0x80
JIT_MemCpy_0x2c
cbz x2,JIT_MemCpy_0x80
tbnz x0,#0,JIT_MemCpy_0x5c
JIT_MemCpy_0x34
and x8,x0,#7
cbz x8,JIT_MemCpy_0x80
ldrsh w8,[x1]
strh w8,[x0]
add x0,x0,#2
add x1,x1,#2
mov x8,#-2
add x2,x2,x8
cbnz x2,JIT_MemCpy_0x34
b JIT_MemCpy_0x80
JIT_MemCpy_0x5c
and x8,x0,#7
cbz x8,JIT_MemCpy_0x80
ldrsb w8,[x1]
strb w8,[x0]
add x0,x0,#1
add x1,x1,#1
mov x8,#-1
add x2,x2,x8
cbnz x2,JIT_MemCpy_0x5c
JIT_MemCpy_0x80
cmp x2,#8
blo JIT_MemCpy_0xb4
lsr x9,x2,#3
mov x8,#-8
madd x2,x9,x8,x2
JIT_MemCpy_0xa0
ldr x8,[x1],#8
str x8,[x0],#8
mov x8,#-1
add x9,x9,x8
cbnz x9,JIT_MemCpy_0xa0
JIT_MemCpy_0xb4
tbz x2,#2,JIT_MemCpy_0xc8
ldr w8,[x1]
str w8,[x0]
add x0,x0,#4
add x1,x1,#4
JIT_MemCpy_0xc8
tbz x2,#1,JIT_MemCpy_0xdc
ldrsh w8,[x1]
strh w8,[x0]
add x0,x0,#2
add x1,x1,#2
JIT_MemCpy_0xdc
tbz x2,#0,JIT_MemCpy_0xe8
ldrsb w8,[x1]
strb w8,[x0]
JIT_MemCpy_0xe8
ret lr
LEAF_END
LEAF_ENTRY JIT_MemCpy_End
LEAF_END
; Must be at very end of file
END
|