-
Notifications
You must be signed in to change notification settings - Fork 4.6k
/
unwinder.cpp
220 lines (180 loc) · 6.91 KB
/
unwinder.cpp
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
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
//
#include "stdafx.h"
#include "unwinder.h"
#ifdef FEATURE_EH_FUNCLETS
BOOL OOPStackUnwinderX86::Unwind(T_CONTEXT* pContextRecord, T_KNONVOLATILE_CONTEXT_POINTERS* pContextPointers)
{
REGDISPLAY rd;
FillRegDisplay(&rd, pContextRecord);
rd.SP = pContextRecord->Esp;
rd.PCTAddr = (UINT_PTR)&(pContextRecord->Eip);
if (pContextPointers)
{
rd.pCurrentContextPointers = pContextPointers;
}
CodeManState codeManState;
codeManState.dwIsSet = 0;
DWORD ControlPc = pContextRecord->Eip;
EECodeInfo codeInfo;
codeInfo.Init((PCODE) ControlPc);
if (!UnwindStackFrame(&rd, &codeInfo, UpdateAllRegs, &codeManState, NULL))
{
return FALSE;
}
pContextRecord->ContextFlags |= CONTEXT_UNWOUND_TO_CALL;
#define ARGUMENT_AND_SCRATCH_REGISTER(reg) if (rd.pCurrentContextPointers->reg) pContextRecord->reg = *rd.pCurrentContextPointers->reg;
ENUM_ARGUMENT_AND_SCRATCH_REGISTERS();
#undef ARGUMENT_AND_SCRATCH_REGISTER
#define CALLEE_SAVED_REGISTER(reg) if (rd.pCurrentContextPointers->reg) pContextRecord->reg = *rd.pCurrentContextPointers->reg;
ENUM_CALLEE_SAVED_REGISTERS();
#undef CALLEE_SAVED_REGISTER
pContextRecord->Esp = rd.SP - codeInfo.GetCodeManager()->GetStackParameterSize(&codeInfo);
pContextRecord->Eip = rd.ControlPC;
return TRUE;
}
/*++
Routine Description:
This function virtually unwinds the specified function by executing its
prologue code backward or its epilogue code forward.
If a context pointers record is specified, then the address where each
nonvolatile registers is restored from is recorded in the appropriate
element of the context pointers record.
Arguments:
HandlerType - Supplies the handler type expected for the virtual unwind.
This may be either an exception or an unwind handler. A flag may
optionally be supplied to avoid epilogue detection if it is known
the specified control PC is not located inside a function epilogue.
ImageBase - Supplies the base address of the image that contains the
function being unwound.
ControlPc - Supplies the address where control left the specified
function.
FunctionEntry - Supplies the address of the function table entry for the
specified function.
ContextRecord - Supplies the address of a context record.
HandlerData - Supplies a pointer to a variable that receives a pointer
the language handler data.
EstablisherFrame - Supplies a pointer to a variable that receives the
the establisher frame pointer value.
ContextPointers - Supplies an optional pointer to a context pointers
record.
HandlerRoutine - Supplies an optional pointer to a variable that receives
the handler routine address. If control did not leave the specified
function in either the prologue or an epilogue and a handler of the
proper type is associated with the function, then the address of the
language specific exception handler is returned. Otherwise, NULL is
returned.
--*/
HRESULT
OOPStackUnwinderX86::VirtualUnwind(
_In_ DWORD HandlerType,
_In_ DWORD ImageBase,
_In_ DWORD ControlPc,
_In_ _PIMAGE_RUNTIME_FUNCTION_ENTRY FunctionEntry,
__inout PCONTEXT ContextRecord,
_Out_ PVOID *HandlerData,
_Out_ PDWORD EstablisherFrame,
__inout_opt PKNONVOLATILE_CONTEXT_POINTERS ContextPointers,
_Outptr_opt_result_maybenull_ PEXCEPTION_ROUTINE *HandlerRoutine
)
{
if (HandlerRoutine != NULL)
{
*HandlerRoutine = NULL;
}
_ASSERTE(ContextRecord->Eip == ControlPc);
if (!OOPStackUnwinderX86::Unwind(ContextRecord, ContextPointers))
{
return HRESULT_FROM_WIN32(ERROR_READ_FAULT);
}
// For x86, the value of Establisher Frame Pointer is Caller SP
//
// (Please refers to CLR ABI for details)
*EstablisherFrame = ContextRecord->Esp;
return S_OK;
}
BOOL DacUnwindStackFrame(T_CONTEXT* pContextRecord, T_KNONVOLATILE_CONTEXT_POINTERS* pContextPointers)
{
BOOL res = OOPStackUnwinderX86::Unwind(pContextRecord, NULL);
if (res && pContextPointers)
{
FillContextPointers(pContextPointers, pContextRecord);
}
return res;
}
//---------------------------------------------------------------------------------------
//
// This function behaves like the RtlVirtualUnwind in Windows.
// It virtually unwinds the specified function by executing its
// prologue code backward or its epilogue code forward.
//
// If a context pointers record is specified, then the address where each
// nonvolatile registers is restored from is recorded in the appropriate
// element of the context pointers record.
//
// Arguments:
//
// HandlerType - Supplies the handler type expected for the virtual unwind.
// This may be either an exception or an unwind handler. A flag may
// optionally be supplied to avoid epilogue detection if it is known
// the specified control PC is not located inside a function epilogue.
//
// ImageBase - Supplies the base address of the image that contains the
// function being unwound.
//
// ControlPc - Supplies the address where control left the specified
// function.
//
// FunctionEntry - Supplies the address of the function table entry for the
// specified function.
//
// ContextRecord - Supplies the address of a context record.
//
// HandlerData - Supplies a pointer to a variable that receives a pointer
// the language handler data.
//
// EstablisherFrame - Supplies a pointer to a variable that receives the
// the establisher frame pointer value.
//
// ContextPointers - Supplies an optional pointer to a context pointers
// record.
//
// Return value:
//
// The handler routine address. If control did not leave the specified
// function in either the prologue or an epilogue and a handler of the
// proper type is associated with the function, then the address of the
// language specific exception handler is returned. Otherwise, NULL is
// returned.
//
EXTERN_C
NTSYSAPI
PEXCEPTION_ROUTINE
NTAPI
RtlVirtualUnwind (
_In_ DWORD HandlerType,
_In_ DWORD ImageBase,
_In_ DWORD ControlPc,
_In_ PRUNTIME_FUNCTION FunctionEntry,
__inout PT_CONTEXT ContextRecord,
_Out_ PVOID *HandlerData,
_Out_ PDWORD EstablisherFrame,
__inout_opt PT_KNONVOLATILE_CONTEXT_POINTERS ContextPointers
)
{
PEXCEPTION_ROUTINE handlerRoutine;
HRESULT res = OOPStackUnwinderX86::VirtualUnwind(
HandlerType,
ImageBase,
ControlPc,
(_PIMAGE_RUNTIME_FUNCTION_ENTRY)FunctionEntry,
ContextRecord,
HandlerData,
EstablisherFrame,
ContextPointers,
&handlerRoutine);
_ASSERTE(SUCCEEDED(res));
return handlerRoutine;
}
#endif // FEATURE_EH_FUNCLETS