[NTOS]: Implement MmDeleteTeb, VADs are now deleted/freed on thread exit as well...
[reactos.git] / reactos / ntoskrnl / ke / amd64 / context.c
1 /*
2 * PROJECT: ReactOS Kernel
3 * LICENSE: GPL - See COPYING in the top level directory
4 * PURPOSE: CONTEXT related functions
5 * PROGRAMMERS: Timo Kreuzer (timo.kreuzer@reactos.org)
6 */
7
8 /* INCLUDES ******************************************************************/
9
10 #include <ntoskrnl.h>
11
12 #define NDEBUG
13 #include <debug.h>
14
15 /* FUNCTIONS *****************************************************************/
16
17 VOID
18 NTAPI
19 KeContextToTrapFrame(IN PCONTEXT Context,
20 IN OUT PKEXCEPTION_FRAME ExceptionFrame,
21 IN OUT PKTRAP_FRAME TrapFrame,
22 IN ULONG ContextFlags,
23 IN KPROCESSOR_MODE PreviousMode)
24 {
25 KIRQL OldIrql;
26
27 /* Do this at APC_LEVEL */
28 OldIrql = KeGetCurrentIrql();
29 if (OldIrql < APC_LEVEL) KeRaiseIrql(APC_LEVEL, &OldIrql);
30
31 /* Handle integer registers */
32 if ((Context->ContextFlags & CONTEXT_INTEGER) == CONTEXT_INTEGER)
33 {
34 TrapFrame->Rax = Context->Rax;
35 TrapFrame->Rbx = Context->Rbx;
36 TrapFrame->Rcx = Context->Rcx;
37 TrapFrame->Rdx = Context->Rdx;
38 TrapFrame->Rsi = Context->Rsi;
39 TrapFrame->Rdi = Context->Rdi;
40 TrapFrame->Rbp = Context->Rbp;
41 TrapFrame->R8 = Context->R8;
42 TrapFrame->R9 = Context->R9;
43 TrapFrame->R10 = Context->R10;
44 TrapFrame->R11 = Context->R11;
45 ExceptionFrame->R12 = Context->R12;
46 ExceptionFrame->R13 = Context->R13;
47 ExceptionFrame->R14 = Context->R14;
48 ExceptionFrame->R15 = Context->R15;
49 }
50
51 /* Handle floating point registers */
52 if (((Context->ContextFlags & CONTEXT_FLOATING_POINT) ==
53 CONTEXT_FLOATING_POINT) && (Context->SegCs & MODE_MASK))
54 {
55 TrapFrame->Xmm0 = Context->Xmm0;
56 TrapFrame->Xmm1 = Context->Xmm1;
57 TrapFrame->Xmm2 = Context->Xmm2;
58 TrapFrame->Xmm3 = Context->Xmm3;
59 TrapFrame->Xmm4 = Context->Xmm4;
60 TrapFrame->Xmm5 = Context->Xmm5;
61 ExceptionFrame->Xmm6 = Context->Xmm6;
62 ExceptionFrame->Xmm7 = Context->Xmm7;
63 ExceptionFrame->Xmm8 = Context->Xmm8;
64 ExceptionFrame->Xmm9 = Context->Xmm9;
65 ExceptionFrame->Xmm10 = Context->Xmm10;
66 ExceptionFrame->Xmm11 = Context->Xmm11;
67 ExceptionFrame->Xmm12 = Context->Xmm12;
68 ExceptionFrame->Xmm13 = Context->Xmm13;
69 ExceptionFrame->Xmm14 = Context->Xmm14;
70 ExceptionFrame->Xmm15 = Context->Xmm15;
71 }
72
73 /* Handle control registers */
74 if ((Context->ContextFlags & CONTEXT_CONTROL) == CONTEXT_CONTROL)
75 {
76 /* Check if this was a Kernel Trap */
77 if (Context->SegCs == KGDT64_R0_CODE)
78 {
79 /* Set valid selectors */
80 TrapFrame->SegCs = KGDT64_R0_CODE;
81 TrapFrame->SegSs = KGDT64_R0_DATA;
82 }
83 else
84 {
85 /* Copy selectors */
86 TrapFrame->SegCs = Context->SegCs;
87 TrapFrame->SegSs = Context->SegSs;
88 }
89
90 /* RIP, RSP, EFLAGS */
91 TrapFrame->Rip = Context->Rip;
92 TrapFrame->Rsp = Context->Rsp;
93 TrapFrame->EFlags = Context->EFlags;
94 }
95
96 /* Handle segment selectors */
97 if ((Context->ContextFlags & CONTEXT_SEGMENTS) == CONTEXT_SEGMENTS)
98 {
99 /* Check if this was a Kernel Trap */
100 if (Context->SegCs == KGDT64_R0_CODE)
101 {
102 /* Set valid selectors */
103 TrapFrame->SegDs = KGDT64_R3_DATA | RPL_MASK;
104 TrapFrame->SegEs = KGDT64_R3_DATA | RPL_MASK;
105 TrapFrame->SegFs = KGDT64_R3_CMTEB | RPL_MASK;
106 TrapFrame->SegGs = KGDT64_R3_DATA | RPL_MASK;
107 }
108 else
109 {
110 /* Copy selectors */
111 TrapFrame->SegDs = Context->SegDs;
112 TrapFrame->SegEs = Context->SegEs;
113 TrapFrame->SegFs = Context->SegFs;
114 TrapFrame->SegGs = Context->SegGs;
115 }
116 }
117
118 /* Handle debug registers */
119 if ((Context->ContextFlags & CONTEXT_DEBUG_REGISTERS) ==
120 CONTEXT_DEBUG_REGISTERS)
121 {
122 /* Copy the debug registers */
123 TrapFrame->Dr0 = Context->Dr0;
124 TrapFrame->Dr1 = Context->Dr1;
125 TrapFrame->Dr2 = Context->Dr2;
126 TrapFrame->Dr3 = Context->Dr3;
127 TrapFrame->Dr6 = Context->Dr6;
128 TrapFrame->Dr7 = Context->Dr7;
129 }
130
131 /* Restore IRQL */
132 if (OldIrql < APC_LEVEL) KeLowerIrql(OldIrql);
133 }
134
135 VOID
136 NTAPI
137 KeTrapFrameToContext(IN PKTRAP_FRAME TrapFrame,
138 IN PKEXCEPTION_FRAME ExceptionFrame,
139 IN OUT PCONTEXT Context)
140 {
141 KIRQL OldIrql;
142
143 /* Do this at APC_LEVEL */
144 OldIrql = KeGetCurrentIrql();
145 if (OldIrql < APC_LEVEL) KeRaiseIrql(APC_LEVEL, &OldIrql);
146
147 /* Handle integer registers */
148 if ((Context->ContextFlags & CONTEXT_INTEGER) == CONTEXT_INTEGER)
149 {
150 Context->Rax = TrapFrame->Rax;
151 Context->Rbx = TrapFrame->Rbx;
152 Context->Rcx = TrapFrame->Rcx;
153 Context->Rdx = TrapFrame->Rdx;
154 Context->Rsi = TrapFrame->Rsi;
155 Context->Rdi = TrapFrame->Rdi;
156 Context->Rbp = TrapFrame->Rbp;
157 Context->R8 = TrapFrame->R8;
158 Context->R9 = TrapFrame->R9;
159 Context->R10 = TrapFrame->R10;
160 Context->R11 = TrapFrame->R11;
161 Context->R12 = ExceptionFrame->R12;
162 Context->R13 = ExceptionFrame->R13;
163 Context->R14 = ExceptionFrame->R14;
164 Context->R15 = ExceptionFrame->R15;
165 }
166
167 /* Handle floating point registers */
168 if (((Context->ContextFlags & CONTEXT_FLOATING_POINT) ==
169 CONTEXT_FLOATING_POINT) && (TrapFrame->SegCs & MODE_MASK))
170 {
171 Context->Xmm0 = TrapFrame->Xmm0;
172 Context->Xmm1 = TrapFrame->Xmm1;
173 Context->Xmm2 = TrapFrame->Xmm2;
174 Context->Xmm3 = TrapFrame->Xmm3;
175 Context->Xmm4 = TrapFrame->Xmm4;
176 Context->Xmm5 = TrapFrame->Xmm5;
177 Context->Xmm6 = ExceptionFrame->Xmm6;
178 Context->Xmm7 = ExceptionFrame->Xmm7;
179 Context->Xmm8 = ExceptionFrame->Xmm8;
180 Context->Xmm9 = ExceptionFrame->Xmm9;
181 Context->Xmm10 = ExceptionFrame->Xmm10;
182 Context->Xmm11 = ExceptionFrame->Xmm11;
183 Context->Xmm12 = ExceptionFrame->Xmm12;
184 Context->Xmm13 = ExceptionFrame->Xmm13;
185 Context->Xmm14 = ExceptionFrame->Xmm14;
186 Context->Xmm15 = ExceptionFrame->Xmm15;
187 }
188
189 /* Handle control registers */
190 if ((Context->ContextFlags & CONTEXT_CONTROL) == CONTEXT_CONTROL)
191 {
192 /* Check if this was a Kernel Trap */
193 if (TrapFrame->SegCs == KGDT64_R0_CODE)
194 {
195 /* Set valid selectors */
196 Context->SegCs = KGDT64_R0_CODE;
197 Context->SegSs = KGDT64_R0_DATA;
198 }
199 else
200 {
201 /* Copy selectors */
202 Context->SegCs = TrapFrame->SegCs;
203 Context->SegSs = TrapFrame->SegSs;
204 }
205
206 /* Copy RIP, RSP, EFLAGS */
207 Context->Rip = TrapFrame->Rip;
208 Context->Rsp = TrapFrame->Rsp;
209 Context->EFlags = TrapFrame->EFlags;
210 }
211
212 /* Handle segment selectors */
213 if ((Context->ContextFlags & CONTEXT_SEGMENTS) == CONTEXT_SEGMENTS)
214 {
215 /* Check if this was a Kernel Trap */
216 if (TrapFrame->SegCs == KGDT64_R0_CODE)
217 {
218 /* Set valid selectors */
219 Context->SegDs = KGDT64_R3_DATA | RPL_MASK;
220 Context->SegEs = KGDT64_R3_DATA | RPL_MASK;
221 Context->SegFs = KGDT64_R3_CMTEB | RPL_MASK;
222 Context->SegGs = KGDT64_R3_DATA | RPL_MASK;
223 }
224 else
225 {
226 /* Copy selectors */
227 Context->SegDs = TrapFrame->SegDs;
228 Context->SegEs = TrapFrame->SegEs;
229 Context->SegFs = TrapFrame->SegFs;
230 Context->SegGs = TrapFrame->SegGs;
231 }
232 }
233
234 /* Handle debug registers */
235 if ((Context->ContextFlags & CONTEXT_DEBUG_REGISTERS) ==
236 CONTEXT_DEBUG_REGISTERS)
237 {
238 /* Copy the debug registers */
239 Context->Dr0 = TrapFrame->Dr0;
240 Context->Dr1 = TrapFrame->Dr1;
241 Context->Dr2 = TrapFrame->Dr2;
242 Context->Dr3 = TrapFrame->Dr3;
243 Context->Dr6 = TrapFrame->Dr6;
244 Context->Dr7 = TrapFrame->Dr7;
245 }
246
247 /* Restore IRQL */
248 if (OldIrql < APC_LEVEL) KeLowerIrql(OldIrql);
249 }
250