-
Notifications
You must be signed in to change notification settings - Fork 4.6k
/
ThunkPoolThunks.asm
291 lines (233 loc) · 9.21 KB
/
ThunkPoolThunks.asm
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
;; Licensed to the .NET Foundation under one or more agreements.
;; The .NET Foundation licenses this file to you under the MIT license.
;; -----------------------------------------------------------------------------------------------------------
;;#include "asmmacros.inc"
;; -----------------------------------------------------------------------------------------------------------
LEAF_ENTRY macro Name, Section
Section segment para 'CODE'
align 16
public Name
Name proc
endm
NAMED_LEAF_ENTRY macro Name, Section, SectionAlias
Section segment para alias(SectionAlias) 'CODE'
align 16
public Name
Name proc
endm
LEAF_END macro Name, Section
Name endp
Section ends
endm
NAMED_READONLY_DATA_SECTION macro Section, SectionAlias
Section segment alias(SectionAlias) read 'DATA'
align 16
DQ 0
Section ends
endm
NAMED_READWRITE_DATA_SECTION macro Section, SectionAlias
Section segment alias(SectionAlias) read write 'DATA'
align 16
DQ 0
Section ends
endm
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; STUBS & DATA SECTIONS ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
THUNK_CODESIZE equ 10h ;; 7-byte lea, 6-byte jmp, 3 bytes of nops
THUNK_DATASIZE equ 010h ;; 2 qwords
THUNK_POOL_NUM_THUNKS_PER_PAGE equ 0FAh ;; 250 thunks per page
PAGE_SIZE equ 01000h ;; 4K
POINTER_SIZE equ 08h
LOAD_DATA_ADDRESS macro groupIndex, index, thunkPool
ALIGN 10h ;; make sure we align to 16-byte boundary for CFG table
;; set r10 to beginning of data page : r10 <- [thunkPool] + PAGE_SIZE
;; fix offset of the data : r10 <- r10 + (THUNK_DATASIZE * current thunk's index)
lea r10, [thunkPool + PAGE_SIZE + (groupIndex * THUNK_DATASIZE * 10 + THUNK_DATASIZE * index)]
endm
JUMP_TO_COMMON macro groupIndex, index, thunkPool
;; jump to the location pointed at by the last qword in the data page
jmp qword ptr[thunkPool + PAGE_SIZE + PAGE_SIZE - POINTER_SIZE]
endm
TenThunks macro groupIndex, thunkPool
;; Each thunk will load the address of its corresponding data (from the page that immediately follows)
;; and call a common stub. The address of the common stub is setup by the caller (first qword
;; in the thunks data section, hence the +8's below) depending on the 'kind' of thunks needed (interop,
;; fat function pointers, etc...)
;; Each data block used by a thunk consists of two qword values:
;; - Context: some value given to the thunk as context (passed in r10). Example for fat-fptrs: context = generic dictionary
;; - Target : target code that the thunk eventually jumps to.
LOAD_DATA_ADDRESS groupIndex,0,thunkPool
JUMP_TO_COMMON groupIndex,0,thunkPool
LOAD_DATA_ADDRESS groupIndex,1,thunkPool
JUMP_TO_COMMON groupIndex,1,thunkPool
LOAD_DATA_ADDRESS groupIndex,2,thunkPool
JUMP_TO_COMMON groupIndex,2,thunkPool
LOAD_DATA_ADDRESS groupIndex,3,thunkPool
JUMP_TO_COMMON groupIndex,3,thunkPool
LOAD_DATA_ADDRESS groupIndex,4,thunkPool
JUMP_TO_COMMON groupIndex,4,thunkPool
LOAD_DATA_ADDRESS groupIndex,5,thunkPool
JUMP_TO_COMMON groupIndex,5,thunkPool
LOAD_DATA_ADDRESS groupIndex,6,thunkPool
JUMP_TO_COMMON groupIndex,6,thunkPool
LOAD_DATA_ADDRESS groupIndex,7,thunkPool
JUMP_TO_COMMON groupIndex,7,thunkPool
LOAD_DATA_ADDRESS groupIndex,8,thunkPool
JUMP_TO_COMMON groupIndex,8,thunkPool
LOAD_DATA_ADDRESS groupIndex,9,thunkPool
JUMP_TO_COMMON groupIndex,9,thunkPool
endm
THUNKS_PAGE_BLOCK macro thunkPool
TenThunks 0,thunkPool
TenThunks 1,thunkPool
TenThunks 2,thunkPool
TenThunks 3,thunkPool
TenThunks 4,thunkPool
TenThunks 5,thunkPool
TenThunks 6,thunkPool
TenThunks 7,thunkPool
TenThunks 8,thunkPool
TenThunks 9,thunkPool
TenThunks 10,thunkPool
TenThunks 11,thunkPool
TenThunks 12,thunkPool
TenThunks 13,thunkPool
TenThunks 14,thunkPool
TenThunks 15,thunkPool
TenThunks 16,thunkPool
TenThunks 17,thunkPool
TenThunks 18,thunkPool
TenThunks 19,thunkPool
TenThunks 20,thunkPool
TenThunks 21,thunkPool
TenThunks 22,thunkPool
TenThunks 23,thunkPool
TenThunks 24,thunkPool
endm
;;
;; The first thunks section should be 64K aligned because it can get
;; mapped multiple times in memory, and mapping works on allocation
;; granularity boundaries (we don't want to map more than what we need)
;;
;; The easiest way to do so is by having the thunks section at the
;; first 64K aligned virtual address in the binary. We provide a section
;; layout file to the linker to tell it how to layout the thunks sections
;; that we care about. (ndp\rh\src\runtime\DLLs\app\mrt100_app_sectionlayout.txt)
;;
;; The PE spec says images cannot have gaps between sections (other
;; than what is required by the section alignment value in the header),
;; therefore we need a couple of padding data sections (otherwise the
;; OS will not load the image).
;;
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment0, ".pad0"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment1, ".pad1"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment2, ".pad2"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment3, ".pad3"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment4, ".pad4"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment5, ".pad5"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment6, ".pad6"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment7, ".pad7"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment8, ".pad8"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment9, ".pad9"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment10, ".pad10"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment11, ".pad11"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment12, ".pad12"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment13, ".pad13"
NAMED_READONLY_DATA_SECTION PaddingFor64KAlignment14, ".pad14"
;;
;; Thunk Stubs
;; NOTE: Keep number of blocks in sync with macro/constant named 'NUM_THUNK_BLOCKS' in:
;; - ndp\FxCore\src\System.Private.CoreLib\System\Runtime\InteropServices\ThunkPool.cs
;; - ndp\rh\src\tools\rhbind\zapimage.h
;;
NAMED_LEAF_ENTRY ThunkPool, TKS0, ".tks0"
THUNKS_PAGE_BLOCK ThunkPool
LEAF_END ThunkPool, TKS0
NAMED_READWRITE_DATA_SECTION ThunkData0, ".tkd0"
NAMED_LEAF_ENTRY ThunkPool1, TKS1, ".tks1"
THUNKS_PAGE_BLOCK ThunkPool1
LEAF_END ThunkPool1, TKS1
NAMED_READWRITE_DATA_SECTION ThunkData1, ".tkd1"
NAMED_LEAF_ENTRY ThunkPool2, TKS2, ".tks2"
THUNKS_PAGE_BLOCK ThunkPool2
LEAF_END ThunkPool2, TKS2
NAMED_READWRITE_DATA_SECTION ThunkData2, ".tkd2"
NAMED_LEAF_ENTRY ThunkPool3, TKS3, ".tks3"
THUNKS_PAGE_BLOCK ThunkPool3
LEAF_END ThunkPool3, TKS3
NAMED_READWRITE_DATA_SECTION ThunkData3, ".tkd3"
NAMED_LEAF_ENTRY ThunkPool4, TKS4, ".tks4"
THUNKS_PAGE_BLOCK ThunkPool4
LEAF_END ThunkPool4, TKS4
NAMED_READWRITE_DATA_SECTION ThunkData4, ".tkd4"
NAMED_LEAF_ENTRY ThunkPool5, TKS5, ".tks5"
THUNKS_PAGE_BLOCK ThunkPool5
LEAF_END ThunkPool5, TKS5
NAMED_READWRITE_DATA_SECTION ThunkData5, ".tkd5"
NAMED_LEAF_ENTRY ThunkPool6, TKS6, ".tks6"
THUNKS_PAGE_BLOCK ThunkPool6
LEAF_END ThunkPool6, TKS6
NAMED_READWRITE_DATA_SECTION ThunkData6, ".tkd6"
NAMED_LEAF_ENTRY ThunkPool7, TKS7, ".tks7"
THUNKS_PAGE_BLOCK ThunkPool7
LEAF_END ThunkPool7, TKS7
NAMED_READWRITE_DATA_SECTION ThunkData7, ".tkd7"
;;
;; IntPtr RhpGetThunksBase()
;;
LEAF_ENTRY RhpGetThunksBase, _TEXT
;; Return the address of the first thunk pool to the caller (this is really the base address)
lea rax, [ThunkPool]
ret
LEAF_END RhpGetThunksBase, _TEXT
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; General Helpers ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;
;; int RhpGetNumThunksPerBlock()
;;
LEAF_ENTRY RhpGetNumThunksPerBlock, _TEXT
mov rax, THUNK_POOL_NUM_THUNKS_PER_PAGE
ret
LEAF_END RhpGetNumThunksPerBlock, _TEXT
;;
;; int RhpGetThunkSize()
;;
LEAF_ENTRY RhpGetThunkSize, _TEXT
mov rax, THUNK_CODESIZE
ret
LEAF_END RhpGetThunkSize, _TEXT
;;
;; int RhpGetNumThunkBlocksPerMapping()
;;
LEAF_ENTRY RhpGetNumThunkBlocksPerMapping, _TEXT
mov rax, 8
ret
LEAF_END RhpGetNumThunkBlocksPerMapping, _TEXT
;;
;; int RhpGetThunkBlockSize
;;
LEAF_ENTRY RhpGetThunkBlockSize, _TEXT
mov rax, PAGE_SIZE * 2
ret
LEAF_END RhpGetThunkBlockSize, _TEXT
;;
;; IntPtr RhpGetThunkDataBlockAddress(IntPtr thunkStubAddress)
;;
LEAF_ENTRY RhpGetThunkDataBlockAddress, _TEXT
mov rax, rcx
mov rcx, PAGE_SIZE - 1
not rcx
and rax, rcx
add rax, PAGE_SIZE
ret
LEAF_END RhpGetThunkDataBlockAddress, _TEXT
;;
;; IntPtr RhpGetThunkStubsBlockAddress(IntPtr thunkDataAddress)
;;
LEAF_ENTRY RhpGetThunkStubsBlockAddress, _TEXT
mov rax, rcx
mov rcx, PAGE_SIZE - 1
not rcx
and rax, rcx
sub rax, PAGE_SIZE
ret
LEAF_END RhpGetThunkStubsBlockAddress, _TEXT
end