ntdll: Make it possible to use RtlGetNtGlobalFlags before the TEB is initialized.
[wine] / dlls / ntdll / signal_i386.c
1 /*
2  * i386 signal handling routines
3  *
4  * Copyright 1999 Alexandre Julliard
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #ifdef __i386__
22
23 #include "config.h"
24 #include "wine/port.h"
25
26 #include <errno.h>
27 #include <signal.h>
28 #include <stdlib.h>
29 #include <stdarg.h>
30 #include <stdio.h>
31 #include <sys/types.h>
32 #ifdef HAVE_UNISTD_H
33 # include <unistd.h>
34 #endif
35
36 #ifdef HAVE_SYS_PARAM_H
37 # include <sys/param.h>
38 #endif
39 #ifdef HAVE_SYSCALL_H
40 # include <syscall.h>
41 #else
42 # ifdef HAVE_SYS_SYSCALL_H
43 #  include <sys/syscall.h>
44 # endif
45 #endif
46
47 #ifdef HAVE_SYS_VM86_H
48 # include <sys/vm86.h>
49 #endif
50
51 #ifdef HAVE_SYS_SIGNAL_H
52 # include <sys/signal.h>
53 #endif
54 #ifdef HAVE_SYS_SYSCTL_H
55 # include <sys/sysctl.h>
56 #endif
57
58 #include "ntstatus.h"
59 #define WIN32_NO_STATUS
60 #include "windef.h"
61 #include "wine/library.h"
62 #include "ntdll_misc.h"
63 #include "wine/exception.h"
64 #include "wine/debug.h"
65
66 #ifdef HAVE_VALGRIND_MEMCHECK_H
67 #include <valgrind/memcheck.h>
68 #endif
69
70 #undef ERR  /* Solaris needs to define this */
71
72 /* not defined for x86, so copy the x86_64 definition */
73 typedef struct DECLSPEC_ALIGN(16) _M128A
74 {
75     ULONGLONG Low;
76     LONGLONG High;
77 } M128A;
78
79 typedef struct
80 {
81     WORD ControlWord;
82     WORD StatusWord;
83     BYTE TagWord;
84     BYTE Reserved1;
85     WORD ErrorOpcode;
86     DWORD ErrorOffset;
87     WORD ErrorSelector;
88     WORD Reserved2;
89     DWORD DataOffset;
90     WORD DataSelector;
91     WORD Reserved3;
92     DWORD MxCsr;
93     DWORD MxCsr_Mask;
94     M128A FloatRegisters[8];
95     M128A XmmRegisters[16];
96     BYTE Reserved4[96];
97 } XMM_SAVE_AREA32;
98
99 /***********************************************************************
100  * signal context platform-specific definitions
101  */
102
103 #ifdef linux
104
105 typedef ucontext_t SIGCONTEXT;
106
107 #define EAX_sig(context)     ((context)->uc_mcontext.gregs[REG_EAX])
108 #define EBX_sig(context)     ((context)->uc_mcontext.gregs[REG_EBX])
109 #define ECX_sig(context)     ((context)->uc_mcontext.gregs[REG_ECX])
110 #define EDX_sig(context)     ((context)->uc_mcontext.gregs[REG_EDX])
111 #define ESI_sig(context)     ((context)->uc_mcontext.gregs[REG_ESI])
112 #define EDI_sig(context)     ((context)->uc_mcontext.gregs[REG_EDI])
113 #define EBP_sig(context)     ((context)->uc_mcontext.gregs[REG_EBP])
114 #define ESP_sig(context)     ((context)->uc_mcontext.gregs[REG_ESP])
115
116 #define CS_sig(context)      ((context)->uc_mcontext.gregs[REG_CS])
117 #define DS_sig(context)      ((context)->uc_mcontext.gregs[REG_DS])
118 #define ES_sig(context)      ((context)->uc_mcontext.gregs[REG_ES])
119 #define SS_sig(context)      ((context)->uc_mcontext.gregs[REG_SS])
120 #define FS_sig(context)      ((context)->uc_mcontext.gregs[REG_FS])
121 #define GS_sig(context)      ((context)->uc_mcontext.gregs[REG_GS])
122
123 #define EFL_sig(context)     ((context)->uc_mcontext.gregs[REG_EFL])
124 #define EIP_sig(context)     ((context)->uc_mcontext.gregs[REG_EIP])
125 #define TRAP_sig(context)    ((context)->uc_mcontext.gregs[REG_TRAPNO])
126 #define ERROR_sig(context)   ((context)->uc_mcontext.gregs[REG_ERR])
127
128 #define FPU_sig(context)     ((FLOATING_SAVE_AREA*)((context)->uc_mcontext.fpregs))
129 #define FPUX_sig(context)    (FPU_sig(context) && !((context)->uc_mcontext.fpregs->status >> 16) ? (XMM_SAVE_AREA32 *)(FPU_sig(context) + 1) : NULL)
130
131 #define VM86_EAX 0 /* the %eax value while vm86_enter is executing */
132 #define VIF_FLAG 0x00080000
133 #define VIP_FLAG 0x00100000
134
135 int vm86_enter( void **vm86_ptr );
136 void vm86_return(void);
137 void vm86_return_end(void);
138 __ASM_GLOBAL_FUNC(vm86_enter,
139                   "pushl %ebp\n\t"
140                   __ASM_CFI(".cfi_adjust_cfa_offset 4\n\t")
141                   __ASM_CFI(".cfi_rel_offset %ebp,0\n\t")
142                   "movl %esp,%ebp\n\t"
143                   __ASM_CFI(".cfi_def_cfa_register %ebp\n\t")
144                   "pushl %ebx\n\t"
145                   __ASM_CFI(".cfi_rel_offset %ebx,-4\n\t")
146                   "movl $166,%eax\n\t"  /*SYS_vm86*/
147                   "movl 8(%ebp),%ecx\n\t" /* vm86_ptr */
148                   "movl (%ecx),%ecx\n\t"
149                   "movl $1,%ebx\n\t"    /*VM86_ENTER*/
150                   "pushl %ecx\n\t"      /* put vm86plus_struct ptr somewhere we can find it */
151                   "pushl %fs\n\t"
152                   "pushl %gs\n\t"
153                   "int $0x80\n"
154                   ".globl " __ASM_NAME("vm86_return") "\n\t"
155                   __ASM_FUNC("vm86_return") "\n"
156                   __ASM_NAME("vm86_return") ":\n\t"
157                   "popl %gs\n\t"
158                   "popl %fs\n\t"
159                   "popl %ecx\n\t"
160                   "popl %ebx\n\t"
161                   __ASM_CFI(".cfi_same_value %ebx\n\t")
162                   "popl %ebp\n\t"
163                   __ASM_CFI(".cfi_def_cfa %esp,4\n\t")
164                   __ASM_CFI(".cfi_same_value %ebp\n\t")
165                   "testl %eax,%eax\n\t"
166                   "jl 0f\n\t"
167                   "cmpb $0,%al\n\t" /* VM86_SIGNAL */
168                   "je " __ASM_NAME("vm86_enter") "\n\t"
169                   "0:\n\t"
170                   "movl 4(%esp),%ecx\n\t"  /* vm86_ptr */
171                   "movl $0,(%ecx)\n\t"
172                   ".globl " __ASM_NAME("vm86_return_end") "\n\t"
173                   __ASM_FUNC("vm86_return_end") "\n"
174                   __ASM_NAME("vm86_return_end") ":\n\t"
175                   "ret" )
176
177 #ifdef HAVE_SYS_VM86_H
178 # define __HAVE_VM86
179 #endif
180
181 #endif  /* linux */
182
183 #ifdef BSDI
184
185 #include <machine/frame.h>
186 typedef struct trapframe SIGCONTEXT;
187
188 #define EAX_sig(context)     ((context)->tf_eax)
189 #define EBX_sig(context)     ((context)->tf_ebx)
190 #define ECX_sig(context)     ((context)->tf_ecx)
191 #define EDX_sig(context)     ((context)->tf_edx)
192 #define ESI_sig(context)     ((context)->tf_esi)
193 #define EDI_sig(context)     ((context)->tf_edi)
194 #define EBP_sig(context)     ((context)->tf_ebp)
195
196 #define CS_sig(context)      ((context)->tf_cs)
197 #define DS_sig(context)      ((context)->tf_ds)
198 #define ES_sig(context)      ((context)->tf_es)
199 #define SS_sig(context)      ((context)->tf_ss)
200
201 #define EFL_sig(context)     ((context)->tf_eflags)
202
203 #define EIP_sig(context)     (*((unsigned long*)&(context)->tf_eip))
204 #define ESP_sig(context)     (*((unsigned long*)&(context)->tf_esp))
205
206 #define FPU_sig(context)     NULL  /* FIXME */
207 #define FPUX_sig(context)    NULL  /* FIXME */
208
209 #endif /* bsdi */
210
211 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
212
213 typedef struct sigcontext SIGCONTEXT;
214
215 #define EAX_sig(context)     ((context)->sc_eax)
216 #define EBX_sig(context)     ((context)->sc_ebx)
217 #define ECX_sig(context)     ((context)->sc_ecx)
218 #define EDX_sig(context)     ((context)->sc_edx)
219 #define ESI_sig(context)     ((context)->sc_esi)
220 #define EDI_sig(context)     ((context)->sc_edi)
221 #define EBP_sig(context)     ((context)->sc_ebp)
222
223 #define CS_sig(context)      ((context)->sc_cs)
224 #define DS_sig(context)      ((context)->sc_ds)
225 #define ES_sig(context)      ((context)->sc_es)
226 #define FS_sig(context)      ((context)->sc_fs)
227 #define GS_sig(context)      ((context)->sc_gs)
228 #define SS_sig(context)      ((context)->sc_ss)
229
230 #define TRAP_sig(context)    ((context)->sc_trapno)
231 #define ERROR_sig(context)   ((context)->sc_err)
232 #define EFL_sig(context)     ((context)->sc_eflags)
233
234 #define EIP_sig(context)     ((context)->sc_eip)
235 #define ESP_sig(context)     ((context)->sc_esp)
236
237 #define FPU_sig(context)     NULL  /* FIXME */
238 #define FPUX_sig(context)    NULL  /* FIXME */
239
240 #endif  /* __FreeBSD__ */
241
242 #ifdef __OpenBSD__
243
244 typedef struct sigcontext SIGCONTEXT;
245
246 #define EAX_sig(context)     ((context)->sc_eax)
247 #define EBX_sig(context)     ((context)->sc_ebx)
248 #define ECX_sig(context)     ((context)->sc_ecx)
249 #define EDX_sig(context)     ((context)->sc_edx)
250 #define ESI_sig(context)     ((context)->sc_esi)
251 #define EDI_sig(context)     ((context)->sc_edi)
252 #define EBP_sig(context)     ((context)->sc_ebp)
253
254 #define CS_sig(context)      ((context)->sc_cs)
255 #define DS_sig(context)      ((context)->sc_ds)
256 #define ES_sig(context)      ((context)->sc_es)
257 #define FS_sig(context)      ((context)->sc_fs)
258 #define GS_sig(context)      ((context)->sc_gs)
259 #define SS_sig(context)      ((context)->sc_ss)
260
261 #define TRAP_sig(context)    ((context)->sc_trapno)
262 #define ERROR_sig(context)   ((context)->sc_err)
263 #define EFL_sig(context)     ((context)->sc_eflags)
264
265 #define EIP_sig(context)     ((context)->sc_eip)
266 #define ESP_sig(context)     ((context)->sc_esp)
267
268 #define FPU_sig(context)     NULL  /* FIXME */
269 #define FPUX_sig(context)    NULL  /* FIXME */
270
271 #define T_MCHK T_MACHK
272 #define T_XMMFLT T_XFTRAP
273
274 #endif  /* __OpenBSD__ */
275
276 #if defined(__svr4__) || defined(_SCO_DS) || defined(__sun)
277
278 #ifdef _SCO_DS
279 #include <sys/regset.h>
280 #endif
281 #include <sys/ucontext.h>
282 typedef struct ucontext SIGCONTEXT;
283
284 #ifdef _SCO_DS
285 #define gregs regs
286 #endif
287
288 #define EAX_sig(context)     ((context)->uc_mcontext.gregs[EAX])
289 #define EBX_sig(context)     ((context)->uc_mcontext.gregs[EBX])
290 #define ECX_sig(context)     ((context)->uc_mcontext.gregs[ECX])
291 #define EDX_sig(context)     ((context)->uc_mcontext.gregs[EDX])
292 #define ESI_sig(context)     ((context)->uc_mcontext.gregs[ESI])
293 #define EDI_sig(context)     ((context)->uc_mcontext.gregs[EDI])
294 #define EBP_sig(context)     ((context)->uc_mcontext.gregs[EBP])
295
296 #define CS_sig(context)      ((context)->uc_mcontext.gregs[CS])
297 #define DS_sig(context)      ((context)->uc_mcontext.gregs[DS])
298 #define ES_sig(context)      ((context)->uc_mcontext.gregs[ES])
299 #define SS_sig(context)      ((context)->uc_mcontext.gregs[SS])
300
301 #define FS_sig(context)      ((context)->uc_mcontext.gregs[FS])
302 #define GS_sig(context)      ((context)->uc_mcontext.gregs[GS])
303
304 #define EFL_sig(context)     ((context)->uc_mcontext.gregs[EFL])
305
306 #define EIP_sig(context)     ((context)->uc_mcontext.gregs[EIP])
307 #ifdef UESP
308 #define ESP_sig(context)     ((context)->uc_mcontext.gregs[UESP])
309 #elif defined(R_ESP)
310 #define ESP_sig(context)     ((context)->uc_mcontext.gregs[R_ESP])
311 #else
312 #define ESP_sig(context)     ((context)->uc_mcontext.gregs[ESP])
313 #endif
314 #ifdef ERR
315 #define ERROR_sig(context)   ((context)->uc_mcontext.gregs[ERR])
316 #endif
317 #ifdef TRAPNO
318 #define TRAP_sig(context)     ((context)->uc_mcontext.gregs[TRAPNO])
319 #endif
320
321 #define FPU_sig(context)     NULL  /* FIXME */
322 #define FPUX_sig(context)    NULL  /* FIXME */
323
324 #endif  /* svr4 || SCO_DS */
325
326 #ifdef __APPLE__
327 # include <sys/ucontext.h>
328
329 typedef ucontext_t SIGCONTEXT;
330
331 /* work around silly renaming of struct members in OS X 10.5 */
332 #if __DARWIN_UNIX03 && defined(_STRUCT_X86_EXCEPTION_STATE32)
333 #define EAX_sig(context)     ((context)->uc_mcontext->__ss.__eax)
334 #define EBX_sig(context)     ((context)->uc_mcontext->__ss.__ebx)
335 #define ECX_sig(context)     ((context)->uc_mcontext->__ss.__ecx)
336 #define EDX_sig(context)     ((context)->uc_mcontext->__ss.__edx)
337 #define ESI_sig(context)     ((context)->uc_mcontext->__ss.__esi)
338 #define EDI_sig(context)     ((context)->uc_mcontext->__ss.__edi)
339 #define EBP_sig(context)     ((context)->uc_mcontext->__ss.__ebp)
340 #define CS_sig(context)      ((context)->uc_mcontext->__ss.__cs)
341 #define DS_sig(context)      ((context)->uc_mcontext->__ss.__ds)
342 #define ES_sig(context)      ((context)->uc_mcontext->__ss.__es)
343 #define FS_sig(context)      ((context)->uc_mcontext->__ss.__fs)
344 #define GS_sig(context)      ((context)->uc_mcontext->__ss.__gs)
345 #define SS_sig(context)      ((context)->uc_mcontext->__ss.__ss)
346 #define EFL_sig(context)     ((context)->uc_mcontext->__ss.__eflags)
347 #define EIP_sig(context)     (*((unsigned long*)&(context)->uc_mcontext->__ss.__eip))
348 #define ESP_sig(context)     (*((unsigned long*)&(context)->uc_mcontext->__ss.__esp))
349 #define TRAP_sig(context)    ((context)->uc_mcontext->__es.__trapno)
350 #define ERROR_sig(context)   ((context)->uc_mcontext->__es.__err)
351 #define FPU_sig(context)     NULL
352 #define FPUX_sig(context)    ((XMM_SAVE_AREA32 *)&(context)->uc_mcontext->__fs.__fpu_fcw)
353 #else
354 #define EAX_sig(context)     ((context)->uc_mcontext->ss.eax)
355 #define EBX_sig(context)     ((context)->uc_mcontext->ss.ebx)
356 #define ECX_sig(context)     ((context)->uc_mcontext->ss.ecx)
357 #define EDX_sig(context)     ((context)->uc_mcontext->ss.edx)
358 #define ESI_sig(context)     ((context)->uc_mcontext->ss.esi)
359 #define EDI_sig(context)     ((context)->uc_mcontext->ss.edi)
360 #define EBP_sig(context)     ((context)->uc_mcontext->ss.ebp)
361 #define CS_sig(context)      ((context)->uc_mcontext->ss.cs)
362 #define DS_sig(context)      ((context)->uc_mcontext->ss.ds)
363 #define ES_sig(context)      ((context)->uc_mcontext->ss.es)
364 #define FS_sig(context)      ((context)->uc_mcontext->ss.fs)
365 #define GS_sig(context)      ((context)->uc_mcontext->ss.gs)
366 #define SS_sig(context)      ((context)->uc_mcontext->ss.ss)
367 #define EFL_sig(context)     ((context)->uc_mcontext->ss.eflags)
368 #define EIP_sig(context)     (*((unsigned long*)&(context)->uc_mcontext->ss.eip))
369 #define ESP_sig(context)     (*((unsigned long*)&(context)->uc_mcontext->ss.esp))
370 #define TRAP_sig(context)    ((context)->uc_mcontext->es.trapno)
371 #define ERROR_sig(context)   ((context)->uc_mcontext->es.err)
372 #define FPU_sig(context)     NULL
373 #define FPUX_sig(context)    ((XMM_SAVE_AREA32 *)&(context)->uc_mcontext->fs.fpu_fcw)
374 #endif
375
376 #endif /* __APPLE__ */
377
378 #if defined(__NetBSD__)
379 # include <sys/ucontext.h>
380 # include <sys/types.h>
381 # include <signal.h>
382
383 typedef ucontext_t SIGCONTEXT;
384
385 #define EAX_sig(context)       ((context)->uc_mcontext.__gregs[_REG_EAX])
386 #define EBX_sig(context)       ((context)->uc_mcontext.__gregs[_REG_EBX])
387 #define ECX_sig(context)       ((context)->uc_mcontext.__gregs[_REG_ECX])
388 #define EDX_sig(context)       ((context)->uc_mcontext.__gregs[_REG_EDX])
389 #define ESI_sig(context)       ((context)->uc_mcontext.__gregs[_REG_ESI])
390 #define EDI_sig(context)       ((context)->uc_mcontext.__gregs[_REG_EDI])
391 #define EBP_sig(context)       ((context)->uc_mcontext.__gregs[_REG_EBP])
392 #define ESP_sig(context)       _UC_MACHINE_SP(context)
393
394 #define CS_sig(context)        ((context)->uc_mcontext.__gregs[_REG_CS])
395 #define DS_sig(context)        ((context)->uc_mcontext.__gregs[_REG_DS])
396 #define ES_sig(context)        ((context)->uc_mcontext.__gregs[_REG_ES])
397 #define SS_sig(context)        ((context)->uc_mcontext.__gregs[_REG_SS])
398 #define FS_sig(context)        ((context)->uc_mcontext.__gregs[_REG_FS])
399 #define GS_sig(context)        ((context)->uc_mcontext.__gregs[_REG_GS])
400
401 #define EFL_sig(context)       ((context)->uc_mcontext.__gregs[_REG_EFL])
402 #define EIP_sig(context)       _UC_MACHINE_PC(context)
403 #define TRAP_sig(context)      ((context)->uc_mcontext.__gregs[_REG_TRAPNO])
404 #define ERROR_sig(context)     ((context)->uc_mcontext.__gregs[_REG_ERR])
405
406 #define FPU_sig(context)     NULL
407 #define FPUX_sig(context)    ((XMM_SAVE_AREA32 *)&((context)->uc_mcontext.__fpregs))
408
409 #define T_MCHK T_MCA
410 #define T_XMMFLT T_XMM
411
412 #endif /* __NetBSD__ */
413
414 WINE_DEFAULT_DEBUG_CHANNEL(seh);
415
416 typedef int (*wine_signal_handler)(unsigned int sig);
417
418 static const size_t teb_size = 4096;  /* we reserve one page for the TEB */
419 static size_t signal_stack_mask;
420 static size_t signal_stack_size;
421
422 static wine_signal_handler handlers[256];
423
424 static int fpux_support;  /* whether the CPU support extended fpu context */
425
426 extern void DECLSPEC_NORETURN __wine_restore_regs( const CONTEXT *context );
427
428 enum i386_trap_code
429 {
430     TRAP_x86_UNKNOWN    = -1,  /* Unknown fault (TRAP_sig not defined) */
431 #if defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__)
432     TRAP_x86_DIVIDE     = T_DIVIDE,     /* Division by zero exception */
433     TRAP_x86_TRCTRAP    = T_TRCTRAP,    /* Single-step exception */
434     TRAP_x86_NMI        = T_NMI,        /* NMI interrupt */
435     TRAP_x86_BPTFLT     = T_BPTFLT,     /* Breakpoint exception */
436     TRAP_x86_OFLOW      = T_OFLOW,      /* Overflow exception */
437     TRAP_x86_BOUND      = T_BOUND,      /* Bound range exception */
438     TRAP_x86_PRIVINFLT  = T_PRIVINFLT,  /* Invalid opcode exception */
439     TRAP_x86_DNA        = T_DNA,        /* Device not available exception */
440     TRAP_x86_DOUBLEFLT  = T_DOUBLEFLT,  /* Double fault exception */
441     TRAP_x86_FPOPFLT    = T_FPOPFLT,    /* Coprocessor segment overrun */
442     TRAP_x86_TSSFLT     = T_TSSFLT,     /* Invalid TSS exception */
443     TRAP_x86_SEGNPFLT   = T_SEGNPFLT,   /* Segment not present exception */
444     TRAP_x86_STKFLT     = T_STKFLT,     /* Stack fault */
445     TRAP_x86_PROTFLT    = T_PROTFLT,    /* General protection fault */
446     TRAP_x86_PAGEFLT    = T_PAGEFLT,    /* Page fault */
447     TRAP_x86_ARITHTRAP  = T_ARITHTRAP,  /* Floating point exception */
448     TRAP_x86_ALIGNFLT   = T_ALIGNFLT,   /* Alignment check exception */
449     TRAP_x86_MCHK       = T_MCHK,       /* Machine check exception */
450     TRAP_x86_CACHEFLT   = T_XMMFLT      /* Cache flush exception */
451 #else
452     TRAP_x86_DIVIDE     = 0,   /* Division by zero exception */
453     TRAP_x86_TRCTRAP    = 1,   /* Single-step exception */
454     TRAP_x86_NMI        = 2,   /* NMI interrupt */
455     TRAP_x86_BPTFLT     = 3,   /* Breakpoint exception */
456     TRAP_x86_OFLOW      = 4,   /* Overflow exception */
457     TRAP_x86_BOUND      = 5,   /* Bound range exception */
458     TRAP_x86_PRIVINFLT  = 6,   /* Invalid opcode exception */
459     TRAP_x86_DNA        = 7,   /* Device not available exception */
460     TRAP_x86_DOUBLEFLT  = 8,   /* Double fault exception */
461     TRAP_x86_FPOPFLT    = 9,   /* Coprocessor segment overrun */
462     TRAP_x86_TSSFLT     = 10,  /* Invalid TSS exception */
463     TRAP_x86_SEGNPFLT   = 11,  /* Segment not present exception */
464     TRAP_x86_STKFLT     = 12,  /* Stack fault */
465     TRAP_x86_PROTFLT    = 13,  /* General protection fault */
466     TRAP_x86_PAGEFLT    = 14,  /* Page fault */
467     TRAP_x86_ARITHTRAP  = 16,  /* Floating point exception */
468     TRAP_x86_ALIGNFLT   = 17,  /* Alignment check exception */
469     TRAP_x86_MCHK       = 18,  /* Machine check exception */
470     TRAP_x86_CACHEFLT   = 19   /* SIMD exception (via SIGFPE) if CPU is SSE capable
471                                   otherwise Cache flush exception (via SIGSEV) */
472 #endif
473 };
474
475 /* Exception record for handling exceptions happening inside exception handlers */
476 typedef struct
477 {
478     EXCEPTION_REGISTRATION_RECORD frame;
479     EXCEPTION_REGISTRATION_RECORD *prevFrame;
480 } EXC_NESTED_FRAME;
481
482 extern DWORD EXC_CallHandler( EXCEPTION_RECORD *record, EXCEPTION_REGISTRATION_RECORD *frame,
483                               CONTEXT *context, EXCEPTION_REGISTRATION_RECORD **dispatcher,
484                               PEXCEPTION_HANDLER handler, PEXCEPTION_HANDLER nested_handler );
485
486 /***********************************************************************
487  *           dispatch_signal
488  */
489 static inline int dispatch_signal(unsigned int sig)
490 {
491     if (handlers[sig] == NULL) return 0;
492     return handlers[sig](sig);
493 }
494
495
496 /***********************************************************************
497  *           get_trap_code
498  *
499  * Get the trap code for a signal.
500  */
501 static inline enum i386_trap_code get_trap_code( const SIGCONTEXT *sigcontext )
502 {
503 #ifdef TRAP_sig
504     return TRAP_sig(sigcontext);
505 #else
506     return TRAP_x86_UNKNOWN;  /* unknown trap code */
507 #endif
508 }
509
510 /***********************************************************************
511  *           get_error_code
512  *
513  * Get the error code for a signal.
514  */
515 static inline WORD get_error_code( const SIGCONTEXT *sigcontext )
516 {
517 #ifdef ERROR_sig
518     return ERROR_sig(sigcontext);
519 #else
520     return 0;
521 #endif
522 }
523
524 /***********************************************************************
525  *           get_signal_stack
526  *
527  * Get the base of the signal stack for the current thread.
528  */
529 static inline void *get_signal_stack(void)
530 {
531     return (char *)NtCurrentTeb() + 4096;
532 }
533
534
535 /***********************************************************************
536  *           get_current_teb
537  *
538  * Get the current teb based on the stack pointer.
539  */
540 static inline TEB *get_current_teb(void)
541 {
542     unsigned long esp;
543     __asm__("movl %%esp,%0" : "=g" (esp) );
544     return (TEB *)(esp & ~signal_stack_mask);
545 }
546
547
548 /*******************************************************************
549  *         raise_handler
550  *
551  * Handler for exceptions happening inside a handler.
552  */
553 static DWORD raise_handler( EXCEPTION_RECORD *rec, EXCEPTION_REGISTRATION_RECORD *frame,
554                             CONTEXT *context, EXCEPTION_REGISTRATION_RECORD **dispatcher )
555 {
556     if (rec->ExceptionFlags & (EH_UNWINDING | EH_EXIT_UNWIND))
557         return ExceptionContinueSearch;
558     /* We shouldn't get here so we store faulty frame in dispatcher */
559     *dispatcher = ((EXC_NESTED_FRAME*)frame)->prevFrame;
560     return ExceptionNestedException;
561 }
562
563
564 /*******************************************************************
565  *         unwind_handler
566  *
567  * Handler for exceptions happening inside an unwind handler.
568  */
569 static DWORD unwind_handler( EXCEPTION_RECORD *rec, EXCEPTION_REGISTRATION_RECORD *frame,
570                              CONTEXT *context, EXCEPTION_REGISTRATION_RECORD **dispatcher )
571 {
572     if (!(rec->ExceptionFlags & (EH_UNWINDING | EH_EXIT_UNWIND)))
573         return ExceptionContinueSearch;
574     /* We shouldn't get here so we store faulty frame in dispatcher */
575     *dispatcher = ((EXC_NESTED_FRAME*)frame)->prevFrame;
576     return ExceptionCollidedUnwind;
577 }
578
579
580 /**********************************************************************
581  *           call_stack_handlers
582  *
583  * Call the stack handlers chain.
584  */
585 static NTSTATUS call_stack_handlers( EXCEPTION_RECORD *rec, CONTEXT *context )
586 {
587     EXCEPTION_REGISTRATION_RECORD *frame, *dispatch, *nested_frame;
588     DWORD res;
589
590     frame = NtCurrentTeb()->Tib.ExceptionList;
591     nested_frame = NULL;
592     while (frame != (EXCEPTION_REGISTRATION_RECORD*)~0UL)
593     {
594         /* Check frame address */
595         if (((void*)frame < NtCurrentTeb()->Tib.StackLimit) ||
596             ((void*)(frame+1) > NtCurrentTeb()->Tib.StackBase) ||
597             (ULONG_PTR)frame & 3)
598         {
599             rec->ExceptionFlags |= EH_STACK_INVALID;
600             break;
601         }
602
603         /* Call handler */
604         TRACE( "calling handler at %p code=%x flags=%x\n",
605                frame->Handler, rec->ExceptionCode, rec->ExceptionFlags );
606         res = EXC_CallHandler( rec, frame, context, &dispatch, frame->Handler, raise_handler );
607         TRACE( "handler at %p returned %x\n", frame->Handler, res );
608
609         if (frame == nested_frame)
610         {
611             /* no longer nested */
612             nested_frame = NULL;
613             rec->ExceptionFlags &= ~EH_NESTED_CALL;
614         }
615
616         switch(res)
617         {
618         case ExceptionContinueExecution:
619             if (!(rec->ExceptionFlags & EH_NONCONTINUABLE)) return STATUS_SUCCESS;
620             return STATUS_NONCONTINUABLE_EXCEPTION;
621         case ExceptionContinueSearch:
622             break;
623         case ExceptionNestedException:
624             if (nested_frame < dispatch) nested_frame = dispatch;
625             rec->ExceptionFlags |= EH_NESTED_CALL;
626             break;
627         default:
628             return STATUS_INVALID_DISPOSITION;
629         }
630         frame = frame->Prev;
631     }
632     return STATUS_UNHANDLED_EXCEPTION;
633 }
634
635
636 /*******************************************************************
637  *              raise_exception
638  *
639  * Implementation of NtRaiseException.
640  */
641 static NTSTATUS raise_exception( EXCEPTION_RECORD *rec, CONTEXT *context, BOOL first_chance )
642 {
643     NTSTATUS status;
644
645     if (first_chance)
646     {
647         DWORD c;
648
649         TRACE( "code=%x flags=%x addr=%p ip=%08x tid=%04x\n",
650                rec->ExceptionCode, rec->ExceptionFlags, rec->ExceptionAddress,
651                context->Eip, GetCurrentThreadId() );
652         for (c = 0; c < rec->NumberParameters; c++)
653             TRACE( " info[%d]=%08lx\n", c, rec->ExceptionInformation[c] );
654         if (rec->ExceptionCode == EXCEPTION_WINE_STUB)
655         {
656             if (rec->ExceptionInformation[1] >> 16)
657                 MESSAGE( "wine: Call from %p to unimplemented function %s.%s, aborting\n",
658                          rec->ExceptionAddress,
659                          (char*)rec->ExceptionInformation[0], (char*)rec->ExceptionInformation[1] );
660             else
661                 MESSAGE( "wine: Call from %p to unimplemented function %s.%ld, aborting\n",
662                          rec->ExceptionAddress,
663                          (char*)rec->ExceptionInformation[0], rec->ExceptionInformation[1] );
664         }
665         else
666         {
667             TRACE(" eax=%08x ebx=%08x ecx=%08x edx=%08x esi=%08x edi=%08x\n",
668                   context->Eax, context->Ebx, context->Ecx,
669                   context->Edx, context->Esi, context->Edi );
670             TRACE(" ebp=%08x esp=%08x cs=%04x ds=%04x es=%04x fs=%04x gs=%04x flags=%08x\n",
671                   context->Ebp, context->Esp, context->SegCs, context->SegDs,
672                   context->SegEs, context->SegFs, context->SegGs, context->EFlags );
673         }
674         status = send_debug_event( rec, TRUE, context );
675         if (status == DBG_CONTINUE || status == DBG_EXCEPTION_HANDLED)
676             return STATUS_SUCCESS;
677
678         /* fix up instruction pointer in context for EXCEPTION_BREAKPOINT */
679         if (rec->ExceptionCode == EXCEPTION_BREAKPOINT) context->Eip--;
680
681         if (call_vectored_handlers( rec, context ) == EXCEPTION_CONTINUE_EXECUTION)
682             return STATUS_SUCCESS;
683
684         if ((status = call_stack_handlers( rec, context )) != STATUS_UNHANDLED_EXCEPTION)
685             return status;
686     }
687
688     /* last chance exception */
689
690     status = send_debug_event( rec, FALSE, context );
691     if (status != DBG_CONTINUE)
692     {
693         if (rec->ExceptionFlags & EH_STACK_INVALID)
694             WINE_ERR("Exception frame is not in stack limits => unable to dispatch exception.\n");
695         else if (rec->ExceptionCode == STATUS_NONCONTINUABLE_EXCEPTION)
696             WINE_ERR("Process attempted to continue execution after noncontinuable exception.\n");
697         else
698             WINE_ERR("Unhandled exception code %x flags %x addr %p\n",
699                      rec->ExceptionCode, rec->ExceptionFlags, rec->ExceptionAddress );
700         NtTerminateProcess( NtCurrentProcess(), rec->ExceptionCode );
701     }
702     return STATUS_SUCCESS;
703 }
704
705
706 #ifdef __HAVE_VM86
707 /***********************************************************************
708  *           save_vm86_context
709  *
710  * Set the register values from a vm86 structure.
711  */
712 static void save_vm86_context( CONTEXT *context, const struct vm86plus_struct *vm86 )
713 {
714     context->ContextFlags = CONTEXT_FULL;
715     context->Eax    = vm86->regs.eax;
716     context->Ebx    = vm86->regs.ebx;
717     context->Ecx    = vm86->regs.ecx;
718     context->Edx    = vm86->regs.edx;
719     context->Esi    = vm86->regs.esi;
720     context->Edi    = vm86->regs.edi;
721     context->Esp    = vm86->regs.esp;
722     context->Ebp    = vm86->regs.ebp;
723     context->Eip    = vm86->regs.eip;
724     context->SegCs  = vm86->regs.cs;
725     context->SegDs  = vm86->regs.ds;
726     context->SegEs  = vm86->regs.es;
727     context->SegFs  = vm86->regs.fs;
728     context->SegGs  = vm86->regs.gs;
729     context->SegSs  = vm86->regs.ss;
730     context->EFlags = vm86->regs.eflags;
731 }
732
733
734 /***********************************************************************
735  *           restore_vm86_context
736  *
737  * Build a vm86 structure from the register values.
738  */
739 static void restore_vm86_context( const CONTEXT *context, struct vm86plus_struct *vm86 )
740 {
741     vm86->regs.eax    = context->Eax;
742     vm86->regs.ebx    = context->Ebx;
743     vm86->regs.ecx    = context->Ecx;
744     vm86->regs.edx    = context->Edx;
745     vm86->regs.esi    = context->Esi;
746     vm86->regs.edi    = context->Edi;
747     vm86->regs.esp    = context->Esp;
748     vm86->regs.ebp    = context->Ebp;
749     vm86->regs.eip    = context->Eip;
750     vm86->regs.cs     = context->SegCs;
751     vm86->regs.ds     = context->SegDs;
752     vm86->regs.es     = context->SegEs;
753     vm86->regs.fs     = context->SegFs;
754     vm86->regs.gs     = context->SegGs;
755     vm86->regs.ss     = context->SegSs;
756     vm86->regs.eflags = context->EFlags;
757 }
758
759
760 /**********************************************************************
761  *              merge_vm86_pending_flags
762  *
763  * Merges TEB.vm86_ptr and TEB.vm86_pending VIP flags and
764  * raises exception if there are pending events and VIF flag
765  * has been turned on.
766  *
767  * Called from __wine_enter_vm86 because vm86_enter
768  * doesn't check for pending events. 
769  *
770  * Called from raise_vm86_sti_exception to check for
771  * pending events in a signal safe way.
772  */
773 static void merge_vm86_pending_flags( EXCEPTION_RECORD *rec )
774 {
775     BOOL check_pending = TRUE;
776     struct vm86plus_struct *vm86 =
777         (struct vm86plus_struct*)(ntdll_get_thread_data()->vm86_ptr);
778
779     /*
780      * In order to prevent a race when SIGUSR2 occurs while
781      * we are returning from exception handler, pending events
782      * will be rechecked after each raised exception.
783      */
784     while (check_pending && get_vm86_teb_info()->vm86_pending)
785     {
786         check_pending = FALSE;
787         ntdll_get_thread_data()->vm86_ptr = NULL;
788             
789         /*
790          * If VIF is set, throw exception.
791          * Note that SIGUSR2 may turn VIF flag off so
792          * VIF check must occur only when TEB.vm86_ptr is NULL.
793          */
794         if (vm86->regs.eflags & VIF_FLAG)
795         {
796             CONTEXT vcontext;
797             save_vm86_context( &vcontext, vm86 );
798             
799             rec->ExceptionCode    = EXCEPTION_VM86_STI;
800             rec->ExceptionFlags   = EXCEPTION_CONTINUABLE;
801             rec->ExceptionRecord  = NULL;
802             rec->NumberParameters = 0;
803             rec->ExceptionAddress = (LPVOID)vcontext.Eip;
804
805             vcontext.EFlags &= ~VIP_FLAG;
806             get_vm86_teb_info()->vm86_pending = 0;
807             raise_exception( rec, &vcontext, TRUE );
808
809             restore_vm86_context( &vcontext, vm86 );
810             check_pending = TRUE;
811         }
812
813         ntdll_get_thread_data()->vm86_ptr = vm86;
814     }
815
816     /*
817      * Merge VIP flags in a signal safe way. This requires
818      * that the following operation compiles into atomic
819      * instruction.
820      */
821     vm86->regs.eflags |= get_vm86_teb_info()->vm86_pending;
822 }
823 #endif /* __HAVE_VM86 */
824
825
826 #ifdef __sun
827
828 /* We have to workaround two Solaris breakages:
829  * - Solaris doesn't restore %ds and %es before calling the signal handler so exceptions in 16-bit
830  *   code crash badly.
831  * - Solaris inserts a libc trampoline to call our handler, but the trampoline expects that registers
832  *   are setup correctly. So we need to insert our own trampoline below the libc trampoline to set %gs.
833  */
834
835 extern int sigaction_syscall( int sig, const struct sigaction *new, struct sigaction *old );
836 __ASM_GLOBAL_FUNC( sigaction_syscall,
837                   "movl $0x62,%eax\n\t"
838                   "int $0x91\n\t"
839                   "ret" )
840
841 /* assume the same libc handler is used for all signals */
842 static void (*libc_sigacthandler)( int signal, siginfo_t *siginfo, void *context );
843
844 static void wine_sigacthandler( int signal, siginfo_t *siginfo, void *sigcontext )
845 {
846     struct ntdll_thread_data *thread_data;
847
848     __asm__ __volatile__("mov %ss,%ax; mov %ax,%ds; mov %ax,%es");
849
850     thread_data = (struct ntdll_thread_data *)get_current_teb()->SpareBytes1;
851     wine_set_fs( thread_data->fs );
852     wine_set_gs( thread_data->gs );
853
854     libc_sigacthandler( signal, siginfo, sigcontext );
855 }
856
857 static int solaris_sigaction( int sig, const struct sigaction *new, struct sigaction *old )
858 {
859     struct sigaction real_act;
860
861     if (sigaction( sig, new, old ) == -1) return -1;
862
863     /* retrieve the real handler and flags with a direct syscall */
864     sigaction_syscall( sig, NULL, &real_act );
865     libc_sigacthandler = real_act.sa_sigaction;
866     real_act.sa_sigaction = wine_sigacthandler;
867     sigaction_syscall( sig, &real_act, NULL );
868     return 0;
869 }
870 #define sigaction(sig,new,old) solaris_sigaction(sig,new,old)
871
872 #endif
873
874 typedef void (WINAPI *raise_func)( EXCEPTION_RECORD *rec, CONTEXT *context );
875
876
877 /***********************************************************************
878  *           init_handler
879  *
880  * Handler initialization when the full context is not needed.
881  * Return the stack pointer to use for pushing the exception data.
882  */
883 static inline void *init_handler( const SIGCONTEXT *sigcontext, WORD *fs, WORD *gs )
884 {
885     TEB *teb = get_current_teb();
886
887     /* get %fs and %gs at time of the fault */
888 #ifdef FS_sig
889     *fs = LOWORD(FS_sig(sigcontext));
890 #else
891     *fs = wine_get_fs();
892 #endif
893 #ifdef GS_sig
894     *gs = LOWORD(GS_sig(sigcontext));
895 #else
896     *gs = wine_get_gs();
897 #endif
898
899 #ifndef __sun  /* see above for Solaris handling */
900     {
901         struct ntdll_thread_data *thread_data = (struct ntdll_thread_data *)teb->SpareBytes1;
902         wine_set_fs( thread_data->fs );
903         wine_set_gs( thread_data->gs );
904     }
905 #endif
906
907     if (!wine_ldt_is_system(CS_sig(sigcontext)) ||
908         !wine_ldt_is_system(SS_sig(sigcontext)))  /* 16-bit mode */
909     {
910         /*
911          * Win16 or DOS protected mode. Note that during switch
912          * from 16-bit mode to linear mode, CS may be set to system
913          * segment before FS is restored. Fortunately, in this case
914          * SS is still non-system segment. This is why both CS and SS
915          * are checked.
916          */
917         return teb->WOW32Reserved;
918     }
919     return (void *)(ESP_sig(sigcontext) & ~3);
920 }
921
922
923 /***********************************************************************
924  *           save_fpu
925  *
926  * Save the thread FPU context.
927  */
928 static inline void save_fpu( CONTEXT *context )
929 {
930 #ifdef __GNUC__
931     context->ContextFlags |= CONTEXT_FLOATING_POINT;
932     __asm__ __volatile__( "fnsave %0; fwait" : "=m" (context->FloatSave) );
933 #endif
934 }
935
936
937 /***********************************************************************
938  *           save_fpux
939  *
940  * Save the thread FPU extended context.
941  */
942 static inline void save_fpux( CONTEXT *context )
943 {
944 #ifdef __GNUC__
945     /* we have to enforce alignment by hand */
946     char buffer[sizeof(XMM_SAVE_AREA32) + 16];
947     XMM_SAVE_AREA32 *state = (XMM_SAVE_AREA32 *)(((ULONG_PTR)buffer + 15) & ~15);
948
949     __asm__ __volatile__( "fxsave %0" : "=m" (*state) );
950     context->ContextFlags |= CONTEXT_EXTENDED_REGISTERS;
951     memcpy( context->ExtendedRegisters, state, sizeof(*state) );
952 #endif
953 }
954
955
956 /***********************************************************************
957  *           restore_fpu
958  *
959  * Restore the FPU context to a sigcontext.
960  */
961 static inline void restore_fpu( const CONTEXT *context )
962 {
963     FLOATING_SAVE_AREA float_status = context->FloatSave;
964     /* reset the current interrupt status */
965     float_status.StatusWord &= float_status.ControlWord | 0xffffff80;
966 #ifdef __GNUC__
967     __asm__ __volatile__( "frstor %0; fwait" : : "m" (float_status) );
968 #endif  /* __GNUC__ */
969 }
970
971
972 /***********************************************************************
973  *           restore_fpux
974  *
975  * Restore the FPU extended context to a sigcontext.
976  */
977 static inline void restore_fpux( const CONTEXT *context )
978 {
979 #ifdef __GNUC__
980     /* we have to enforce alignment by hand */
981     char buffer[sizeof(XMM_SAVE_AREA32) + 16];
982     XMM_SAVE_AREA32 *state = (XMM_SAVE_AREA32 *)(((ULONG_PTR)buffer + 15) & ~15);
983
984     memcpy( state, context->ExtendedRegisters, sizeof(*state) );
985     /* reset the current interrupt status */
986     state->StatusWord &= state->ControlWord | 0xff80;
987     __asm__ __volatile__( "fxrstor %0" : : "m" (*state) );
988 #endif
989 }
990
991
992 /***********************************************************************
993  *           fpux_to_fpu
994  *
995  * Build a standard FPU context from an extended one.
996  */
997 static void fpux_to_fpu( FLOATING_SAVE_AREA *fpu, const XMM_SAVE_AREA32 *fpux )
998 {
999     unsigned int i, tag, stack_top;
1000
1001     fpu->ControlWord   = fpux->ControlWord | 0xffff0000;
1002     fpu->StatusWord    = fpux->StatusWord | 0xffff0000;
1003     fpu->ErrorOffset   = fpux->ErrorOffset;
1004     fpu->ErrorSelector = fpux->ErrorSelector | (fpux->ErrorOpcode << 16);
1005     fpu->DataOffset    = fpux->DataOffset;
1006     fpu->DataSelector  = fpux->DataSelector;
1007     fpu->Cr0NpxState   = fpux->StatusWord | 0xffff0000;
1008
1009     stack_top = (fpux->StatusWord >> 11) & 7;
1010     fpu->TagWord = 0xffff0000;
1011     for (i = 0; i < 8; i++)
1012     {
1013         memcpy( &fpu->RegisterArea[10 * i], &fpux->FloatRegisters[i], 10 );
1014         if (!(fpux->TagWord & (1 << i))) tag = 3;  /* empty */
1015         else
1016         {
1017             const M128A *reg = &fpux->FloatRegisters[(i - stack_top) & 7];
1018             if ((reg->High & 0x7fff) == 0x7fff)  /* exponent all ones */
1019             {
1020                 tag = 2;  /* special */
1021             }
1022             else if (!(reg->High & 0x7fff))  /* exponent all zeroes */
1023             {
1024                 if (reg->Low) tag = 2;  /* special */
1025                 else tag = 1;  /* zero */
1026             }
1027             else
1028             {
1029                 if (reg->Low >> 63) tag = 0;  /* valid */
1030                 else tag = 2;  /* special */
1031             }
1032         }
1033         fpu->TagWord |= tag << (2 * i);
1034     }
1035 }
1036
1037
1038 /***********************************************************************
1039  *           save_context
1040  *
1041  * Build a context structure from the signal info.
1042  */
1043 static inline void save_context( CONTEXT *context, const SIGCONTEXT *sigcontext, WORD fs, WORD gs )
1044 {
1045     struct ntdll_thread_data * const regs = ntdll_get_thread_data();
1046     FLOATING_SAVE_AREA *fpu = FPU_sig(sigcontext);
1047     XMM_SAVE_AREA32 *fpux = FPUX_sig(sigcontext);
1048
1049     memset(context, 0, sizeof(*context));
1050     context->ContextFlags = CONTEXT_FULL | CONTEXT_DEBUG_REGISTERS;
1051     context->Eax          = EAX_sig(sigcontext);
1052     context->Ebx          = EBX_sig(sigcontext);
1053     context->Ecx          = ECX_sig(sigcontext);
1054     context->Edx          = EDX_sig(sigcontext);
1055     context->Esi          = ESI_sig(sigcontext);
1056     context->Edi          = EDI_sig(sigcontext);
1057     context->Ebp          = EBP_sig(sigcontext);
1058     context->EFlags       = EFL_sig(sigcontext);
1059     context->Eip          = EIP_sig(sigcontext);
1060     context->Esp          = ESP_sig(sigcontext);
1061     context->SegCs        = LOWORD(CS_sig(sigcontext));
1062     context->SegDs        = LOWORD(DS_sig(sigcontext));
1063     context->SegEs        = LOWORD(ES_sig(sigcontext));
1064     context->SegFs        = fs;
1065     context->SegGs        = gs;
1066     context->SegSs        = LOWORD(SS_sig(sigcontext));
1067     context->Dr0          = regs->dr0;
1068     context->Dr1          = regs->dr1;
1069     context->Dr2          = regs->dr2;
1070     context->Dr3          = regs->dr3;
1071     context->Dr6          = regs->dr6;
1072     context->Dr7          = regs->dr7;
1073
1074     if (fpu)
1075     {
1076         context->ContextFlags |= CONTEXT_FLOATING_POINT;
1077         context->FloatSave = *fpu;
1078     }
1079     if (fpux)
1080     {
1081         context->ContextFlags |= CONTEXT_FLOATING_POINT | CONTEXT_EXTENDED_REGISTERS;
1082         memcpy( context->ExtendedRegisters, fpux, sizeof(*fpux) );
1083         fpux_support = 1;
1084         if (!fpu) fpux_to_fpu( &context->FloatSave, fpux );
1085     }
1086     if (!fpu && !fpux) save_fpu( context );
1087 }
1088
1089
1090 /***********************************************************************
1091  *           restore_context
1092  *
1093  * Restore the signal info from the context.
1094  */
1095 static inline void restore_context( const CONTEXT *context, SIGCONTEXT *sigcontext )
1096 {
1097     struct ntdll_thread_data * const regs = ntdll_get_thread_data();
1098     FLOATING_SAVE_AREA *fpu = FPU_sig(sigcontext);
1099     XMM_SAVE_AREA32 *fpux = FPUX_sig(sigcontext);
1100
1101     regs->dr0 = context->Dr0;
1102     regs->dr1 = context->Dr1;
1103     regs->dr2 = context->Dr2;
1104     regs->dr3 = context->Dr3;
1105     regs->dr6 = context->Dr6;
1106     regs->dr7 = context->Dr7;
1107     EAX_sig(sigcontext) = context->Eax;
1108     EBX_sig(sigcontext) = context->Ebx;
1109     ECX_sig(sigcontext) = context->Ecx;
1110     EDX_sig(sigcontext) = context->Edx;
1111     ESI_sig(sigcontext) = context->Esi;
1112     EDI_sig(sigcontext) = context->Edi;
1113     EBP_sig(sigcontext) = context->Ebp;
1114     EFL_sig(sigcontext) = context->EFlags;
1115     EIP_sig(sigcontext) = context->Eip;
1116     ESP_sig(sigcontext) = context->Esp;
1117     CS_sig(sigcontext)  = context->SegCs;
1118     DS_sig(sigcontext)  = context->SegDs;
1119     ES_sig(sigcontext)  = context->SegEs;
1120     SS_sig(sigcontext)  = context->SegSs;
1121 #ifdef GS_sig
1122     GS_sig(sigcontext)  = context->SegGs;
1123 #else
1124     wine_set_gs( context->SegGs );
1125 #endif
1126 #ifdef FS_sig
1127     FS_sig(sigcontext)  = context->SegFs;
1128 #else
1129     wine_set_fs( context->SegFs );
1130 #endif
1131
1132     if (fpu) *fpu = context->FloatSave;
1133     if (fpux) memcpy( fpux, context->ExtendedRegisters, sizeof(*fpux) );
1134     if (!fpu && !fpux) restore_fpu( context );
1135 }
1136
1137
1138 /***********************************************************************
1139  *              RtlCaptureContext (NTDLL.@)
1140  */
1141 void WINAPI __regs_RtlCaptureContext( CONTEXT *context, CONTEXT *regs )
1142 {
1143     *context = *regs;
1144     if (fpux_support) save_fpux( context );
1145     else save_fpu( context );
1146 }
1147 DEFINE_REGS_ENTRYPOINT( RtlCaptureContext, 1 )
1148
1149
1150 /***********************************************************************
1151  *           set_cpu_context
1152  *
1153  * Set the new CPU context. Used by NtSetContextThread.
1154  */
1155 void set_cpu_context( const CONTEXT *context )
1156 {
1157     DWORD flags = context->ContextFlags & ~CONTEXT_i386;
1158
1159     if ((flags & CONTEXT_EXTENDED_REGISTERS) && fpux_support) restore_fpux( context );
1160     else if (flags & CONTEXT_FLOATING_POINT) restore_fpu( context );
1161
1162     if (flags & CONTEXT_DEBUG_REGISTERS)
1163     {
1164         ntdll_get_thread_data()->dr0 = context->Dr0;
1165         ntdll_get_thread_data()->dr1 = context->Dr1;
1166         ntdll_get_thread_data()->dr2 = context->Dr2;
1167         ntdll_get_thread_data()->dr3 = context->Dr3;
1168         ntdll_get_thread_data()->dr6 = context->Dr6;
1169         ntdll_get_thread_data()->dr7 = context->Dr7;
1170     }
1171     if (flags & CONTEXT_FULL)
1172     {
1173         if (!(flags & CONTEXT_CONTROL))
1174             FIXME( "setting partial context (%x) not supported\n", flags );
1175         else if (flags & CONTEXT_SEGMENTS)
1176             __wine_restore_regs( context );
1177         else
1178         {
1179             CONTEXT newcontext = *context;
1180             newcontext.SegDs = wine_get_ds();
1181             newcontext.SegEs = wine_get_es();
1182             newcontext.SegFs = wine_get_fs();
1183             newcontext.SegGs = wine_get_gs();
1184             __wine_restore_regs( &newcontext );
1185         }
1186     }
1187 }
1188
1189
1190 /***********************************************************************
1191  *           copy_context
1192  *
1193  * Copy a register context according to the flags.
1194  */
1195 void copy_context( CONTEXT *to, const CONTEXT *from, DWORD flags )
1196 {
1197     flags &= ~CONTEXT_i386;  /* get rid of CPU id */
1198     if (flags & CONTEXT_INTEGER)
1199     {
1200         to->Eax = from->Eax;
1201         to->Ebx = from->Ebx;
1202         to->Ecx = from->Ecx;
1203         to->Edx = from->Edx;
1204         to->Esi = from->Esi;
1205         to->Edi = from->Edi;
1206     }
1207     if (flags & CONTEXT_CONTROL)
1208     {
1209         to->Ebp    = from->Ebp;
1210         to->Esp    = from->Esp;
1211         to->Eip    = from->Eip;
1212         to->SegCs  = from->SegCs;
1213         to->SegSs  = from->SegSs;
1214         to->EFlags = from->EFlags;
1215     }
1216     if (flags & CONTEXT_SEGMENTS)
1217     {
1218         to->SegDs = from->SegDs;
1219         to->SegEs = from->SegEs;
1220         to->SegFs = from->SegFs;
1221         to->SegGs = from->SegGs;
1222     }
1223     if (flags & CONTEXT_DEBUG_REGISTERS)
1224     {
1225         to->Dr0 = from->Dr0;
1226         to->Dr1 = from->Dr1;
1227         to->Dr2 = from->Dr2;
1228         to->Dr3 = from->Dr3;
1229         to->Dr6 = from->Dr6;
1230         to->Dr7 = from->Dr7;
1231     }
1232     if (flags & CONTEXT_FLOATING_POINT)
1233     {
1234         to->FloatSave = from->FloatSave;
1235     }
1236     if (flags & CONTEXT_EXTENDED_REGISTERS)
1237     {
1238         memcpy( to->ExtendedRegisters, from->ExtendedRegisters, sizeof(to->ExtendedRegisters) );
1239     }
1240 }
1241
1242
1243 /***********************************************************************
1244  *           context_to_server
1245  *
1246  * Convert a register context to the server format.
1247  */
1248 NTSTATUS context_to_server( context_t *to, const CONTEXT *from )
1249 {
1250     DWORD flags = from->ContextFlags & ~CONTEXT_i386;  /* get rid of CPU id */
1251
1252     memset( to, 0, sizeof(*to) );
1253     to->cpu = CPU_x86;
1254
1255     if (flags & CONTEXT_CONTROL)
1256     {
1257         to->flags |= SERVER_CTX_CONTROL;
1258         to->ctl.i386_regs.ebp    = from->Ebp;
1259         to->ctl.i386_regs.esp    = from->Esp;
1260         to->ctl.i386_regs.eip    = from->Eip;
1261         to->ctl.i386_regs.cs     = from->SegCs;
1262         to->ctl.i386_regs.ss     = from->SegSs;
1263         to->ctl.i386_regs.eflags = from->EFlags;
1264     }
1265     if (flags & CONTEXT_INTEGER)
1266     {
1267         to->flags |= SERVER_CTX_INTEGER;
1268         to->integer.i386_regs.eax = from->Eax;
1269         to->integer.i386_regs.ebx = from->Ebx;
1270         to->integer.i386_regs.ecx = from->Ecx;
1271         to->integer.i386_regs.edx = from->Edx;
1272         to->integer.i386_regs.esi = from->Esi;
1273         to->integer.i386_regs.edi = from->Edi;
1274     }
1275     if (flags & CONTEXT_SEGMENTS)
1276     {
1277         to->flags |= SERVER_CTX_SEGMENTS;
1278         to->seg.i386_regs.ds = from->SegDs;
1279         to->seg.i386_regs.es = from->SegEs;
1280         to->seg.i386_regs.fs = from->SegFs;
1281         to->seg.i386_regs.gs = from->SegGs;
1282     }
1283     if (flags & CONTEXT_FLOATING_POINT)
1284     {
1285         to->flags |= SERVER_CTX_FLOATING_POINT;
1286         to->fp.i386_regs.ctrl     = from->FloatSave.ControlWord;
1287         to->fp.i386_regs.status   = from->FloatSave.StatusWord;
1288         to->fp.i386_regs.tag      = from->FloatSave.TagWord;
1289         to->fp.i386_regs.err_off  = from->FloatSave.ErrorOffset;
1290         to->fp.i386_regs.err_sel  = from->FloatSave.ErrorSelector;
1291         to->fp.i386_regs.data_off = from->FloatSave.DataOffset;
1292         to->fp.i386_regs.data_sel = from->FloatSave.DataSelector;
1293         to->fp.i386_regs.cr0npx   = from->FloatSave.Cr0NpxState;
1294         memcpy( to->fp.i386_regs.regs, from->FloatSave.RegisterArea, sizeof(to->fp.i386_regs.regs) );
1295     }
1296     if (flags & CONTEXT_DEBUG_REGISTERS)
1297     {
1298         to->flags |= SERVER_CTX_DEBUG_REGISTERS;
1299         to->debug.i386_regs.dr0 = from->Dr0;
1300         to->debug.i386_regs.dr1 = from->Dr1;
1301         to->debug.i386_regs.dr2 = from->Dr2;
1302         to->debug.i386_regs.dr3 = from->Dr3;
1303         to->debug.i386_regs.dr6 = from->Dr6;
1304         to->debug.i386_regs.dr7 = from->Dr7;
1305     }
1306     if (flags & CONTEXT_EXTENDED_REGISTERS)
1307     {
1308         to->flags |= SERVER_CTX_EXTENDED_REGISTERS;
1309         memcpy( to->ext.i386_regs, from->ExtendedRegisters, sizeof(to->ext.i386_regs) );
1310     }
1311     return STATUS_SUCCESS;
1312 }
1313
1314
1315 /***********************************************************************
1316  *           context_from_server
1317  *
1318  * Convert a register context from the server format.
1319  */
1320 NTSTATUS context_from_server( CONTEXT *to, const context_t *from )
1321 {
1322     if (from->cpu != CPU_x86) return STATUS_INVALID_PARAMETER;
1323
1324     to->ContextFlags = CONTEXT_i386;
1325     if (from->flags & SERVER_CTX_CONTROL)
1326     {
1327         to->ContextFlags |= CONTEXT_CONTROL;
1328         to->Ebp    = from->ctl.i386_regs.ebp;
1329         to->Esp    = from->ctl.i386_regs.esp;
1330         to->Eip    = from->ctl.i386_regs.eip;
1331         to->SegCs  = from->ctl.i386_regs.cs;
1332         to->SegSs  = from->ctl.i386_regs.ss;
1333         to->EFlags = from->ctl.i386_regs.eflags;
1334     }
1335     if (from->flags & SERVER_CTX_INTEGER)
1336     {
1337         to->ContextFlags |= CONTEXT_INTEGER;
1338         to->Eax = from->integer.i386_regs.eax;
1339         to->Ebx = from->integer.i386_regs.ebx;
1340         to->Ecx = from->integer.i386_regs.ecx;
1341         to->Edx = from->integer.i386_regs.edx;
1342         to->Esi = from->integer.i386_regs.esi;
1343         to->Edi = from->integer.i386_regs.edi;
1344     }
1345     if (from->flags & SERVER_CTX_SEGMENTS)
1346     {
1347         to->ContextFlags |= CONTEXT_SEGMENTS;
1348         to->SegDs = from->seg.i386_regs.ds;
1349         to->SegEs = from->seg.i386_regs.es;
1350         to->SegFs = from->seg.i386_regs.fs;
1351         to->SegGs = from->seg.i386_regs.gs;
1352     }
1353     if (from->flags & SERVER_CTX_FLOATING_POINT)
1354     {
1355         to->ContextFlags |= CONTEXT_FLOATING_POINT;
1356         to->FloatSave.ControlWord   = from->fp.i386_regs.ctrl;
1357         to->FloatSave.StatusWord    = from->fp.i386_regs.status;
1358         to->FloatSave.TagWord       = from->fp.i386_regs.tag;
1359         to->FloatSave.ErrorOffset   = from->fp.i386_regs.err_off;
1360         to->FloatSave.ErrorSelector = from->fp.i386_regs.err_sel;
1361         to->FloatSave.DataOffset    = from->fp.i386_regs.data_off;
1362         to->FloatSave.DataSelector  = from->fp.i386_regs.data_sel;
1363         to->FloatSave.Cr0NpxState   = from->fp.i386_regs.cr0npx;
1364         memcpy( to->FloatSave.RegisterArea, from->fp.i386_regs.regs, sizeof(to->FloatSave.RegisterArea) );
1365     }
1366     if (from->flags & SERVER_CTX_DEBUG_REGISTERS)
1367     {
1368         to->ContextFlags |= CONTEXT_DEBUG_REGISTERS;
1369         to->Dr0 = from->debug.i386_regs.dr0;
1370         to->Dr1 = from->debug.i386_regs.dr1;
1371         to->Dr2 = from->debug.i386_regs.dr2;
1372         to->Dr3 = from->debug.i386_regs.dr3;
1373         to->Dr6 = from->debug.i386_regs.dr6;
1374         to->Dr7 = from->debug.i386_regs.dr7;
1375     }
1376     if (from->flags & SERVER_CTX_EXTENDED_REGISTERS)
1377     {
1378         to->ContextFlags |= CONTEXT_EXTENDED_REGISTERS;
1379         memcpy( to->ExtendedRegisters, from->ext.i386_regs, sizeof(to->ExtendedRegisters) );
1380     }
1381     return STATUS_SUCCESS;
1382 }
1383
1384
1385 /***********************************************************************
1386  *           is_privileged_instr
1387  *
1388  * Check if the fault location is a privileged instruction.
1389  * Based on the instruction emulation code in dlls/kernel/instr.c.
1390  */
1391 static inline DWORD is_privileged_instr( CONTEXT86 *context )
1392 {
1393     const BYTE *instr;
1394     unsigned int prefix_count = 0;
1395
1396     if (!wine_ldt_is_system( context->SegCs )) return 0;
1397     instr = (BYTE *)context->Eip;
1398
1399     for (;;) switch(*instr)
1400     {
1401     /* instruction prefixes */
1402     case 0x2e:  /* %cs: */
1403     case 0x36:  /* %ss: */
1404     case 0x3e:  /* %ds: */
1405     case 0x26:  /* %es: */
1406     case 0x64:  /* %fs: */
1407     case 0x65:  /* %gs: */
1408     case 0x66:  /* opcode size */
1409     case 0x67:  /* addr size */
1410     case 0xf0:  /* lock */
1411     case 0xf2:  /* repne */
1412     case 0xf3:  /* repe */
1413         if (++prefix_count >= 15) return EXCEPTION_ILLEGAL_INSTRUCTION;
1414         instr++;
1415         continue;
1416
1417     case 0x0f: /* extended instruction */
1418         switch(instr[1])
1419         {
1420         case 0x20: /* mov crX, reg */
1421         case 0x21: /* mov drX, reg */
1422         case 0x22: /* mov reg, crX */
1423         case 0x23: /* mov reg drX */
1424             return EXCEPTION_PRIV_INSTRUCTION;
1425         }
1426         return 0;
1427     case 0x6c: /* insb (%dx) */
1428     case 0x6d: /* insl (%dx) */
1429     case 0x6e: /* outsb (%dx) */
1430     case 0x6f: /* outsl (%dx) */
1431     case 0xcd: /* int $xx */
1432     case 0xe4: /* inb al,XX */
1433     case 0xe5: /* in (e)ax,XX */
1434     case 0xe6: /* outb XX,al */
1435     case 0xe7: /* out XX,(e)ax */
1436     case 0xec: /* inb (%dx),%al */
1437     case 0xed: /* inl (%dx),%eax */
1438     case 0xee: /* outb %al,(%dx) */
1439     case 0xef: /* outl %eax,(%dx) */
1440     case 0xf4: /* hlt */
1441     case 0xfa: /* cli */
1442     case 0xfb: /* sti */
1443         return EXCEPTION_PRIV_INSTRUCTION;
1444     default:
1445         return 0;
1446     }
1447 }
1448
1449
1450 #include "pshpack1.h"
1451 struct atl_thunk
1452 {
1453     DWORD movl;  /* movl this,4(%esp) */
1454     DWORD this;
1455     BYTE  jmp;   /* jmp func */
1456     int   func;
1457 };
1458 #include "poppack.h"
1459
1460 /**********************************************************************
1461  *              check_atl_thunk
1462  *
1463  * Check if code destination is an ATL thunk, and emulate it if so.
1464  */
1465 static BOOL check_atl_thunk( EXCEPTION_RECORD *rec, CONTEXT *context )
1466 {
1467     const struct atl_thunk *thunk = (const struct atl_thunk *)rec->ExceptionInformation[1];
1468     BOOL ret = FALSE;
1469
1470     __TRY
1471     {
1472         if (thunk->movl == 0x042444c7 && thunk->jmp == 0xe9)
1473         {
1474             *((DWORD *)context->Esp + 1) = thunk->this;
1475             context->Eip = (DWORD_PTR)(&thunk->func + 1) + thunk->func;
1476             TRACE( "emulating ATL thunk at %p, func=%08x arg=%08x\n",
1477                    thunk, context->Eip, *((DWORD *)context->Esp + 1) );
1478             ret = TRUE;
1479         }
1480     }
1481     __EXCEPT_PAGE_FAULT
1482     {
1483         return FALSE;
1484     }
1485     __ENDTRY
1486     return ret;
1487 }
1488
1489
1490 /***********************************************************************
1491  *           setup_exception_record
1492  *
1493  * Setup the exception record and context on the thread stack.
1494  */
1495 static EXCEPTION_RECORD *setup_exception_record( SIGCONTEXT *sigcontext, void *stack_ptr,
1496                                                  WORD fs, WORD gs, raise_func func )
1497 {
1498     struct stack_layout
1499     {
1500         void             *ret_addr;      /* return address from raise_func */
1501         EXCEPTION_RECORD *rec_ptr;       /* first arg for raise_func */
1502         CONTEXT          *context_ptr;   /* second arg for raise_func */
1503         CONTEXT           context;
1504         EXCEPTION_RECORD  rec;
1505         DWORD             ebp;
1506         DWORD             eip;
1507     } *stack = stack_ptr;
1508     DWORD exception_code = 0;
1509
1510     /* stack sanity checks */
1511
1512     if ((char *)stack >= (char *)get_signal_stack() &&
1513         (char *)stack < (char *)get_signal_stack() + signal_stack_size)
1514     {
1515         WINE_ERR( "nested exception on signal stack in thread %04x eip %08x esp %08x stack %p-%p\n",
1516                   GetCurrentThreadId(), (unsigned int) EIP_sig(sigcontext),
1517                   (unsigned int) ESP_sig(sigcontext), NtCurrentTeb()->Tib.StackLimit,
1518                   NtCurrentTeb()->Tib.StackBase );
1519         abort_thread(1);
1520     }
1521
1522     if (stack - 1 > stack || /* check for overflow in subtraction */
1523         (char *)stack <= (char *)NtCurrentTeb()->DeallocationStack ||
1524         (char *)stack > (char *)NtCurrentTeb()->Tib.StackBase)
1525     {
1526         WARN( "exception outside of stack limits in thread %04x eip %08x esp %08x stack %p-%p\n",
1527               GetCurrentThreadId(), (unsigned int) EIP_sig(sigcontext),
1528               (unsigned int) ESP_sig(sigcontext), NtCurrentTeb()->Tib.StackLimit,
1529               NtCurrentTeb()->Tib.StackBase );
1530     }
1531     else if ((char *)(stack - 1) < (char *)NtCurrentTeb()->DeallocationStack + 4096)
1532     {
1533         /* stack overflow on last page, unrecoverable */
1534         UINT diff = (char *)NtCurrentTeb()->DeallocationStack + 4096 - (char *)(stack - 1);
1535         WINE_ERR( "stack overflow %u bytes in thread %04x eip %08x esp %08x stack %p-%p-%p\n",
1536                   diff, GetCurrentThreadId(), (unsigned int) EIP_sig(sigcontext),
1537                   (unsigned int) ESP_sig(sigcontext), NtCurrentTeb()->DeallocationStack,
1538                   NtCurrentTeb()->Tib.StackLimit, NtCurrentTeb()->Tib.StackBase );
1539         abort_thread(1);
1540     }
1541     else if ((char *)(stack - 1) < (char *)NtCurrentTeb()->Tib.StackLimit)
1542     {
1543         /* stack access below stack limit, may be recoverable */
1544         if (virtual_handle_stack_fault( stack - 1 )) exception_code = EXCEPTION_STACK_OVERFLOW;
1545         else
1546         {
1547             UINT diff = (char *)NtCurrentTeb()->Tib.StackLimit - (char *)(stack - 1);
1548             WINE_ERR( "stack overflow %u bytes in thread %04x eip %08x esp %08x stack %p-%p-%p\n",
1549                       diff, GetCurrentThreadId(), (unsigned int) EIP_sig(sigcontext),
1550                       (unsigned int) ESP_sig(sigcontext), NtCurrentTeb()->DeallocationStack,
1551                       NtCurrentTeb()->Tib.StackLimit, NtCurrentTeb()->Tib.StackBase );
1552             abort_thread(1);
1553         }
1554     }
1555
1556     stack--;  /* push the stack_layout structure */
1557 #if defined(VALGRIND_MAKE_MEM_UNDEFINED)
1558     VALGRIND_MAKE_MEM_UNDEFINED(stack, sizeof(*stack));
1559 #elif defined(VALGRIND_MAKE_WRITABLE)
1560     VALGRIND_MAKE_WRITABLE(stack, sizeof(*stack));
1561 #endif
1562     stack->ret_addr     = (void *)0xdeadbabe;  /* raise_func must not return */
1563     stack->rec_ptr      = &stack->rec;
1564     stack->context_ptr  = &stack->context;
1565
1566     stack->rec.ExceptionRecord  = NULL;
1567     stack->rec.ExceptionCode    = exception_code;
1568     stack->rec.ExceptionFlags   = EXCEPTION_CONTINUABLE;
1569     stack->rec.ExceptionAddress = (LPVOID)EIP_sig(sigcontext);
1570     stack->rec.NumberParameters = 0;
1571
1572     save_context( &stack->context, sigcontext, fs, gs );
1573
1574     /* now modify the sigcontext to return to the raise function */
1575     ESP_sig(sigcontext) = (DWORD)stack;
1576     EIP_sig(sigcontext) = (DWORD)func;
1577     /* clear single-step, direction, and align check flag */
1578     EFL_sig(sigcontext) &= ~(0x100|0x400|0x40000);
1579     CS_sig(sigcontext)  = wine_get_cs();
1580     DS_sig(sigcontext)  = wine_get_ds();
1581     ES_sig(sigcontext)  = wine_get_es();
1582     FS_sig(sigcontext)  = wine_get_fs();
1583     GS_sig(sigcontext)  = wine_get_gs();
1584     SS_sig(sigcontext)  = wine_get_ss();
1585
1586     return stack->rec_ptr;
1587 }
1588
1589
1590 /***********************************************************************
1591  *           setup_exception
1592  *
1593  * Setup a proper stack frame for the raise function, and modify the
1594  * sigcontext so that the return from the signal handler will call
1595  * the raise function.
1596  */
1597 static EXCEPTION_RECORD *setup_exception( SIGCONTEXT *sigcontext, raise_func func )
1598 {
1599     WORD fs, gs;
1600     void *stack = init_handler( sigcontext, &fs, &gs );
1601
1602     return setup_exception_record( sigcontext, stack, fs, gs, func );
1603 }
1604
1605
1606 /***********************************************************************
1607  *           get_exception_context
1608  *
1609  * Get a pointer to the context built by setup_exception.
1610  */
1611 static inline CONTEXT *get_exception_context( EXCEPTION_RECORD *rec )
1612 {
1613     return (CONTEXT *)rec - 1;  /* cf. stack_layout structure */
1614 }
1615
1616
1617 /**********************************************************************
1618  *              get_fpu_code
1619  *
1620  * Get the FPU exception code from the FPU status.
1621  */
1622 static inline DWORD get_fpu_code( const CONTEXT *context )
1623 {
1624     DWORD status = context->FloatSave.StatusWord & ~(context->FloatSave.ControlWord & 0x3f);
1625
1626     if (status & 0x01)  /* IE */
1627     {
1628         if (status & 0x40)  /* SF */
1629             return EXCEPTION_FLT_STACK_CHECK;
1630         else
1631             return EXCEPTION_FLT_INVALID_OPERATION;
1632     }
1633     if (status & 0x02) return EXCEPTION_FLT_DENORMAL_OPERAND;  /* DE flag */
1634     if (status & 0x04) return EXCEPTION_FLT_DIVIDE_BY_ZERO;    /* ZE flag */
1635     if (status & 0x08) return EXCEPTION_FLT_OVERFLOW;          /* OE flag */
1636     if (status & 0x10) return EXCEPTION_FLT_UNDERFLOW;         /* UE flag */
1637     if (status & 0x20) return EXCEPTION_FLT_INEXACT_RESULT;    /* PE flag */
1638     return EXCEPTION_FLT_INVALID_OPERATION;  /* generic error */
1639 }
1640
1641
1642 /**********************************************************************
1643  *              raise_segv_exception
1644  */
1645 static void WINAPI raise_segv_exception( EXCEPTION_RECORD *rec, CONTEXT *context )
1646 {
1647     NTSTATUS status;
1648
1649     switch(rec->ExceptionCode)
1650     {
1651     case EXCEPTION_ACCESS_VIOLATION:
1652         if (rec->NumberParameters == 2)
1653         {
1654             if (rec->ExceptionInformation[0] == EXCEPTION_EXECUTE_FAULT && check_atl_thunk( rec, context ))
1655                 goto done;
1656             if (!(rec->ExceptionCode = virtual_handle_fault( (void *)rec->ExceptionInformation[1],
1657                                                              rec->ExceptionInformation[0] )))
1658                 goto done;
1659             /* send EXCEPTION_EXECUTE_FAULT only if data execution prevention is enabled */
1660             if (rec->ExceptionInformation[0] == EXCEPTION_EXECUTE_FAULT)
1661             {
1662                 ULONG flags;
1663                 NtQueryInformationProcess( GetCurrentProcess(), ProcessExecuteFlags,
1664                                            &flags, sizeof(flags), NULL );
1665                 if (!(flags & MEM_EXECUTE_OPTION_DISABLE))
1666                     rec->ExceptionInformation[0] = EXCEPTION_READ_FAULT;
1667             }
1668         }
1669         break;
1670     case EXCEPTION_DATATYPE_MISALIGNMENT:
1671         /* FIXME: pass through exception handler first? */
1672         if (context->EFlags & 0x00040000)
1673         {
1674             /* Disable AC flag, return */
1675             context->EFlags &= ~0x00040000;
1676             goto done;
1677         }
1678         break;
1679     }
1680     status = NtRaiseException( rec, context, TRUE );
1681     if (status) raise_status( status, rec );
1682 done:
1683     NtSetContextThread( GetCurrentThread(), context );
1684 }
1685
1686
1687 /**********************************************************************
1688  *              raise_trap_exception
1689  */
1690 static void WINAPI raise_trap_exception( EXCEPTION_RECORD *rec, CONTEXT *context )
1691 {
1692     NTSTATUS status;
1693
1694     if (rec->ExceptionCode == EXCEPTION_SINGLE_STEP)
1695     {
1696         /* when single stepping can't tell whether this is a hw bp or a
1697          * single step interrupt. try to avoid as much overhead as possible
1698          * and only do a server call if there is any hw bp enabled. */
1699
1700         if( !(context->EFlags & 0x100) || (ntdll_get_thread_data()->dr7 & 0xff) )
1701         {
1702             /* (possible) hardware breakpoint, fetch the debug registers */
1703             context->ContextFlags = CONTEXT_DEBUG_REGISTERS;
1704             NtGetContextThread(GetCurrentThread(), context);
1705             context->ContextFlags |= CONTEXT_FULL;  /* restore flags */
1706         }
1707
1708         context->EFlags &= ~0x100;  /* clear single-step flag */
1709     }
1710
1711     status = NtRaiseException( rec, context, TRUE );
1712     if (status) raise_status( status, rec );
1713 }
1714
1715
1716 /**********************************************************************
1717  *              raise_generic_exception
1718  *
1719  * Generic raise function for exceptions that don't need special treatment.
1720  */
1721 static void WINAPI raise_generic_exception( EXCEPTION_RECORD *rec, CONTEXT *context )
1722 {
1723     NTSTATUS status;
1724
1725     status = NtRaiseException( rec, context, TRUE );
1726     if (status) raise_status( status, rec );
1727 }
1728
1729
1730 #ifdef __HAVE_VM86
1731 /**********************************************************************
1732  *              raise_vm86_sti_exception
1733  */
1734 static void WINAPI raise_vm86_sti_exception( EXCEPTION_RECORD *rec, CONTEXT *context )
1735 {
1736     /* merge_vm86_pending_flags merges the vm86_pending flag in safely */
1737     get_vm86_teb_info()->vm86_pending |= VIP_FLAG;
1738
1739     if (ntdll_get_thread_data()->vm86_ptr)
1740     {
1741         if (((char*)context->Eip >= (char*)vm86_return) &&
1742             ((char*)context->Eip <= (char*)vm86_return_end) &&
1743             (VM86_TYPE(context->Eax) != VM86_SIGNAL)) {
1744             /* exiting from VM86, can't throw */
1745             goto done;
1746         }
1747         merge_vm86_pending_flags( rec );
1748     }
1749     else if (get_vm86_teb_info()->dpmi_vif &&
1750              !wine_ldt_is_system(context->SegCs) &&
1751              !wine_ldt_is_system(context->SegSs))
1752     {
1753         /* Executing DPMI code and virtual interrupts are enabled. */
1754         get_vm86_teb_info()->vm86_pending = 0;
1755         NtRaiseException( rec, context, TRUE );
1756     }
1757 done:
1758     NtSetContextThread( GetCurrentThread(), context );
1759 }
1760
1761
1762 /**********************************************************************
1763  *              usr2_handler
1764  *
1765  * Handler for SIGUSR2.
1766  * We use it to signal that the running __wine_enter_vm86() should
1767  * immediately set VIP_FLAG, causing pending events to be handled
1768  * as early as possible.
1769  */
1770 static void usr2_handler( int signal, siginfo_t *siginfo, void *sigcontext )
1771 {
1772     EXCEPTION_RECORD *rec = setup_exception( sigcontext, raise_vm86_sti_exception );
1773     rec->ExceptionCode = EXCEPTION_VM86_STI;
1774 }
1775 #endif /* __HAVE_VM86 */
1776
1777
1778 /**********************************************************************
1779  *              segv_handler
1780  *
1781  * Handler for SIGSEGV and related errors.
1782  */
1783 static void segv_handler( int signal, siginfo_t *siginfo, void *sigcontext )
1784 {
1785     WORD fs, gs;
1786     EXCEPTION_RECORD *rec;
1787     SIGCONTEXT *context = sigcontext;
1788     void *stack = init_handler( sigcontext, &fs, &gs );
1789
1790     /* check for page fault inside the thread stack */
1791     if (get_trap_code(context) == TRAP_x86_PAGEFLT &&
1792         (char *)siginfo->si_addr >= (char *)NtCurrentTeb()->DeallocationStack &&
1793         (char *)siginfo->si_addr < (char *)NtCurrentTeb()->Tib.StackBase &&
1794         virtual_handle_stack_fault( siginfo->si_addr ))
1795     {
1796         /* check if this was the last guard page */
1797         if ((char *)siginfo->si_addr < (char *)NtCurrentTeb()->DeallocationStack + 2*4096)
1798         {
1799             rec = setup_exception_record( context, stack, fs, gs, raise_segv_exception );
1800             rec->ExceptionCode = EXCEPTION_STACK_OVERFLOW;
1801         }
1802         return;
1803     }
1804
1805     rec = setup_exception_record( context, stack, fs, gs, raise_segv_exception );
1806     if (rec->ExceptionCode == EXCEPTION_STACK_OVERFLOW) return;
1807
1808     switch(get_trap_code(context))
1809     {
1810     case TRAP_x86_OFLOW:   /* Overflow exception */
1811         rec->ExceptionCode = EXCEPTION_INT_OVERFLOW;
1812         break;
1813     case TRAP_x86_BOUND:   /* Bound range exception */
1814         rec->ExceptionCode = EXCEPTION_ARRAY_BOUNDS_EXCEEDED;
1815         break;
1816     case TRAP_x86_PRIVINFLT:   /* Invalid opcode exception */
1817         rec->ExceptionCode = EXCEPTION_ILLEGAL_INSTRUCTION;
1818         break;
1819     case TRAP_x86_STKFLT:  /* Stack fault */
1820         rec->ExceptionCode = EXCEPTION_STACK_OVERFLOW;
1821         break;
1822     case TRAP_x86_SEGNPFLT:  /* Segment not present exception */
1823     case TRAP_x86_PROTFLT:   /* General protection fault */
1824     case TRAP_x86_UNKNOWN:   /* Unknown fault code */
1825         {
1826             WORD err = get_error_code(context);
1827             if (!err && (rec->ExceptionCode = is_privileged_instr( get_exception_context(rec) ))) break;
1828             rec->ExceptionCode = EXCEPTION_ACCESS_VIOLATION;
1829             rec->NumberParameters = 2;
1830             rec->ExceptionInformation[0] = 0;
1831             /* if error contains a LDT selector, use that as fault address */
1832             if ((err & 7) == 4 && !wine_ldt_is_system( err | 7 ))
1833                 rec->ExceptionInformation[1] = err & ~7;
1834             else
1835                 rec->ExceptionInformation[1] = 0xffffffff;
1836         }
1837         break;
1838     case TRAP_x86_PAGEFLT:  /* Page fault */
1839         rec->ExceptionCode = EXCEPTION_ACCESS_VIOLATION;
1840         rec->NumberParameters = 2;
1841         rec->ExceptionInformation[0] = (get_error_code(context) >> 1) & 0x09;
1842         rec->ExceptionInformation[1] = (ULONG_PTR)siginfo->si_addr;
1843         break;
1844     case TRAP_x86_ALIGNFLT:  /* Alignment check exception */
1845         rec->ExceptionCode = EXCEPTION_DATATYPE_MISALIGNMENT;
1846         break;
1847     default:
1848         WINE_ERR( "Got unexpected trap %d\n", get_trap_code(context) );
1849         /* fall through */
1850     case TRAP_x86_NMI:       /* NMI interrupt */
1851     case TRAP_x86_DNA:       /* Device not available exception */
1852     case TRAP_x86_DOUBLEFLT: /* Double fault exception */
1853     case TRAP_x86_TSSFLT:    /* Invalid TSS exception */
1854     case TRAP_x86_MCHK:      /* Machine check exception */
1855     case TRAP_x86_CACHEFLT:  /* Cache flush exception */
1856         rec->ExceptionCode = EXCEPTION_ILLEGAL_INSTRUCTION;
1857         break;
1858     }
1859 }
1860
1861
1862 /**********************************************************************
1863  *              trap_handler
1864  *
1865  * Handler for SIGTRAP.
1866  */
1867 static void trap_handler( int signal, siginfo_t *siginfo, void *sigcontext )
1868 {
1869     SIGCONTEXT *context = sigcontext;
1870     EXCEPTION_RECORD *rec = setup_exception( context, raise_trap_exception );
1871
1872     switch(get_trap_code(context))
1873     {
1874     case TRAP_x86_TRCTRAP:  /* Single-step exception */
1875         rec->ExceptionCode = EXCEPTION_SINGLE_STEP;
1876         break;
1877     case TRAP_x86_BPTFLT:   /* Breakpoint exception */
1878         rec->ExceptionAddress = (char *)rec->ExceptionAddress - 1;  /* back up over the int3 instruction */
1879         /* fall through */
1880     default:
1881         rec->ExceptionCode = EXCEPTION_BREAKPOINT;
1882         break;
1883     }
1884 }
1885
1886
1887 /**********************************************************************
1888  *              fpe_handler
1889  *
1890  * Handler for SIGFPE.
1891  */
1892 static void fpe_handler( int signal, siginfo_t *siginfo, void *sigcontext )
1893 {
1894     CONTEXT *win_context;
1895     SIGCONTEXT *context = sigcontext;
1896     EXCEPTION_RECORD *rec = setup_exception( context, raise_generic_exception );
1897
1898     win_context = get_exception_context( rec );
1899
1900     switch(get_trap_code(context))
1901     {
1902     case TRAP_x86_DIVIDE:   /* Division by zero exception */
1903         rec->ExceptionCode = EXCEPTION_INT_DIVIDE_BY_ZERO;
1904         break;
1905     case TRAP_x86_FPOPFLT:   /* Coprocessor segment overrun */
1906         rec->ExceptionCode = EXCEPTION_FLT_INVALID_OPERATION;
1907         break;
1908     case TRAP_x86_ARITHTRAP:  /* Floating point exception */
1909     case TRAP_x86_UNKNOWN:    /* Unknown fault code */
1910         rec->ExceptionCode = get_fpu_code( win_context );
1911         rec->ExceptionAddress = (LPVOID)win_context->FloatSave.ErrorOffset;
1912         break;
1913     case TRAP_x86_CACHEFLT:  /* SIMD exception */
1914         /* TODO:
1915          * Behaviour only tested for divide-by-zero exceptions
1916          * Check for other SIMD exceptions as well */
1917         if(siginfo->si_code != FPE_FLTDIV)
1918             FIXME("untested SIMD exception: %#x. Might not work correctly\n",
1919                   siginfo->si_code);
1920
1921         rec->ExceptionCode = STATUS_FLOAT_MULTIPLE_TRAPS;
1922         rec->NumberParameters = 1;
1923         /* no idea what meaning is actually behind this but that's what native does */
1924         rec->ExceptionInformation[0] = 0;
1925         break;
1926     default:
1927         WINE_ERR( "Got unexpected trap %d\n", get_trap_code(context) );
1928         rec->ExceptionCode = EXCEPTION_FLT_INVALID_OPERATION;
1929         break;
1930     }
1931 }
1932
1933
1934 /**********************************************************************
1935  *              int_handler
1936  *
1937  * Handler for SIGINT.
1938  *
1939  * FIXME: should not be calling external functions on the signal stack.
1940  */
1941 static void int_handler( int signal, siginfo_t *siginfo, void *sigcontext )
1942 {
1943     WORD fs, gs;
1944     init_handler( sigcontext, &fs, &gs );
1945     if (!dispatch_signal(SIGINT))
1946     {
1947         EXCEPTION_RECORD *rec = setup_exception( sigcontext, raise_generic_exception );
1948         rec->ExceptionCode = CONTROL_C_EXIT;
1949     }
1950 }
1951
1952 /**********************************************************************
1953  *              abrt_handler
1954  *
1955  * Handler for SIGABRT.
1956  */
1957 static void abrt_handler( int signal, siginfo_t *siginfo, void *sigcontext )
1958 {
1959     EXCEPTION_RECORD *rec = setup_exception( sigcontext, raise_generic_exception );
1960     rec->ExceptionCode  = EXCEPTION_WINE_ASSERTION;
1961     rec->ExceptionFlags = EH_NONCONTINUABLE;
1962 }
1963
1964
1965 /**********************************************************************
1966  *              quit_handler
1967  *
1968  * Handler for SIGQUIT.
1969  */
1970 static void quit_handler( int signal, siginfo_t *siginfo, void *sigcontext )
1971 {
1972     WORD fs, gs;
1973     init_handler( sigcontext, &fs, &gs );
1974     abort_thread(0);
1975 }
1976
1977
1978 /**********************************************************************
1979  *              usr1_handler
1980  *
1981  * Handler for SIGUSR1, used to signal a thread that it got suspended.
1982  */
1983 static void usr1_handler( int signal, siginfo_t *siginfo, void *sigcontext )
1984 {
1985     CONTEXT context;
1986     WORD fs, gs;
1987
1988     init_handler( sigcontext, &fs, &gs );
1989     save_context( &context, sigcontext, fs, gs );
1990     wait_suspend( &context );
1991     restore_context( &context, sigcontext );
1992 }
1993
1994
1995 /***********************************************************************
1996  *           __wine_set_signal_handler   (NTDLL.@)
1997  */
1998 int CDECL __wine_set_signal_handler(unsigned int sig, wine_signal_handler wsh)
1999 {
2000     if (sig >= sizeof(handlers) / sizeof(handlers[0])) return -1;
2001     if (handlers[sig] != NULL) return -2;
2002     handlers[sig] = wsh;
2003     return 0;
2004 }
2005
2006
2007 /***********************************************************************
2008  *           locking for LDT routines
2009  */
2010 static RTL_CRITICAL_SECTION ldt_section;
2011 static RTL_CRITICAL_SECTION_DEBUG critsect_debug =
2012 {
2013     0, 0, &ldt_section,
2014     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
2015       0, 0, { (DWORD_PTR)(__FILE__ ": ldt_section") }
2016 };
2017 static RTL_CRITICAL_SECTION ldt_section = { &critsect_debug, -1, 0, 0, 0, 0 };
2018 static sigset_t ldt_sigset;
2019
2020 static void ldt_lock(void)
2021 {
2022     sigset_t sigset;
2023
2024     pthread_sigmask( SIG_BLOCK, &server_block_set, &sigset );
2025     RtlEnterCriticalSection( &ldt_section );
2026     if (ldt_section.RecursionCount == 1) ldt_sigset = sigset;
2027 }
2028
2029 static void ldt_unlock(void)
2030 {
2031     if (ldt_section.RecursionCount == 1)
2032     {
2033         sigset_t sigset = ldt_sigset;
2034         RtlLeaveCriticalSection( &ldt_section );
2035         pthread_sigmask( SIG_SETMASK, &sigset, NULL );
2036     }
2037     else RtlLeaveCriticalSection( &ldt_section );
2038 }
2039
2040
2041 /**********************************************************************
2042  *              signal_alloc_thread
2043  */
2044 NTSTATUS signal_alloc_thread( TEB **teb )
2045 {
2046     static size_t sigstack_zero_bits;
2047     struct ntdll_thread_data *thread_data;
2048     struct ntdll_thread_data *parent_data = NULL;
2049     SIZE_T size;
2050     void *addr = NULL;
2051     NTSTATUS status;
2052
2053     if (!sigstack_zero_bits)
2054     {
2055         size_t min_size = teb_size + max( MINSIGSTKSZ, 8192 );
2056         /* find the first power of two not smaller than min_size */
2057         sigstack_zero_bits = 12;
2058         while ((1u << sigstack_zero_bits) < min_size) sigstack_zero_bits++;
2059         signal_stack_mask = (1 << sigstack_zero_bits) - 1;
2060         signal_stack_size = (1 << sigstack_zero_bits) - teb_size;
2061     }
2062     else parent_data = ntdll_get_thread_data();
2063
2064     size = signal_stack_mask + 1;
2065     if (!(status = NtAllocateVirtualMemory( NtCurrentProcess(), &addr, sigstack_zero_bits,
2066                                             &size, MEM_COMMIT | MEM_TOP_DOWN, PAGE_READWRITE )))
2067     {
2068         *teb = addr;
2069         (*teb)->Tib.Self = &(*teb)->Tib;
2070         (*teb)->Tib.ExceptionList = (void *)~0UL;
2071         thread_data = (struct ntdll_thread_data *)(*teb)->SpareBytes1;
2072         if (!(thread_data->fs = wine_ldt_alloc_fs()))
2073         {
2074             size = 0;
2075             NtFreeVirtualMemory( NtCurrentProcess(), &addr, &size, MEM_RELEASE );
2076             status = STATUS_TOO_MANY_THREADS;
2077         }
2078         if (parent_data)
2079         {
2080             /* inherit debug registers from parent thread */
2081             thread_data->dr0 = parent_data->dr0;
2082             thread_data->dr1 = parent_data->dr1;
2083             thread_data->dr2 = parent_data->dr2;
2084             thread_data->dr3 = parent_data->dr3;
2085             thread_data->dr6 = parent_data->dr6;
2086             thread_data->dr7 = parent_data->dr7;
2087         }
2088
2089     }
2090     return status;
2091 }
2092
2093
2094 /**********************************************************************
2095  *              signal_free_thread
2096  */
2097 void signal_free_thread( TEB *teb )
2098 {
2099     SIZE_T size;
2100     struct ntdll_thread_data *thread_data = (struct ntdll_thread_data *)teb->SpareBytes1;
2101
2102     if (thread_data) wine_ldt_free_fs( thread_data->fs );
2103     if (teb->DeallocationStack)
2104     {
2105         size = 0;
2106         NtFreeVirtualMemory( GetCurrentProcess(), &teb->DeallocationStack, &size, MEM_RELEASE );
2107     }
2108     size = 0;
2109     NtFreeVirtualMemory( NtCurrentProcess(), (void **)&teb, &size, MEM_RELEASE );
2110 }
2111
2112
2113 /**********************************************************************
2114  *              signal_init_thread
2115  */
2116 void signal_init_thread( TEB *teb )
2117 {
2118     struct ntdll_thread_data *thread_data = (struct ntdll_thread_data *)teb->SpareBytes1;
2119     LDT_ENTRY fs_entry;
2120     stack_t ss;
2121
2122 #ifdef __APPLE__
2123     int mib[2], val = 1;
2124
2125     mib[0] = CTL_KERN;
2126     mib[1] = KERN_THALTSTACK;
2127     sysctl( mib, 2, NULL, NULL, &val, sizeof(val) );
2128 #endif
2129
2130     ss.ss_sp    = (char *)teb + teb_size;
2131     ss.ss_size  = signal_stack_size;
2132     ss.ss_flags = 0;
2133     if (sigaltstack(&ss, NULL) == -1) perror( "sigaltstack" );
2134
2135     wine_ldt_set_base( &fs_entry, teb );
2136     wine_ldt_set_limit( &fs_entry, teb_size - 1 );
2137     wine_ldt_set_flags( &fs_entry, WINE_LDT_FLAGS_DATA|WINE_LDT_FLAGS_32BIT );
2138     wine_ldt_init_fs( thread_data->fs, &fs_entry );
2139     thread_data->gs = wine_get_gs();
2140 }
2141
2142 /**********************************************************************
2143  *              signal_init_process
2144  */
2145 void signal_init_process(void)
2146 {
2147     struct sigaction sig_act;
2148
2149     sig_act.sa_mask = server_block_set;
2150     sig_act.sa_flags = SA_SIGINFO | SA_RESTART;
2151 #ifdef SA_ONSTACK
2152     sig_act.sa_flags |= SA_ONSTACK;
2153 #endif
2154
2155     sig_act.sa_sigaction = int_handler;
2156     if (sigaction( SIGINT, &sig_act, NULL ) == -1) goto error;
2157     sig_act.sa_sigaction = fpe_handler;
2158     if (sigaction( SIGFPE, &sig_act, NULL ) == -1) goto error;
2159     sig_act.sa_sigaction = abrt_handler;
2160     if (sigaction( SIGABRT, &sig_act, NULL ) == -1) goto error;
2161     sig_act.sa_sigaction = quit_handler;
2162     if (sigaction( SIGQUIT, &sig_act, NULL ) == -1) goto error;
2163     sig_act.sa_sigaction = usr1_handler;
2164     if (sigaction( SIGUSR1, &sig_act, NULL ) == -1) goto error;
2165
2166     sig_act.sa_sigaction = segv_handler;
2167     if (sigaction( SIGSEGV, &sig_act, NULL ) == -1) goto error;
2168     if (sigaction( SIGILL, &sig_act, NULL ) == -1) goto error;
2169 #ifdef SIGBUS
2170     if (sigaction( SIGBUS, &sig_act, NULL ) == -1) goto error;
2171 #endif
2172
2173 #ifdef SIGTRAP
2174     sig_act.sa_sigaction = trap_handler;
2175     if (sigaction( SIGTRAP, &sig_act, NULL ) == -1) goto error;
2176 #endif
2177
2178 #ifdef __HAVE_VM86
2179     sig_act.sa_sigaction = usr2_handler;
2180     if (sigaction( SIGUSR2, &sig_act, NULL ) == -1) goto error;
2181 #endif
2182
2183     wine_ldt_init_locking( ldt_lock, ldt_unlock );
2184     return;
2185
2186  error:
2187     perror("sigaction");
2188     exit(1);
2189 }
2190
2191
2192 #ifdef __HAVE_VM86
2193 /**********************************************************************
2194  *              __wine_enter_vm86   (NTDLL.@)
2195  *
2196  * Enter vm86 mode with the specified register context.
2197  */
2198 void __wine_enter_vm86( CONTEXT *context )
2199 {
2200     EXCEPTION_RECORD rec;
2201     int res;
2202     struct vm86plus_struct vm86;
2203
2204     memset( &vm86, 0, sizeof(vm86) );
2205     for (;;)
2206     {
2207         restore_vm86_context( context, &vm86 );
2208
2209         ntdll_get_thread_data()->vm86_ptr = &vm86;
2210         merge_vm86_pending_flags( &rec );
2211
2212         res = vm86_enter( &ntdll_get_thread_data()->vm86_ptr ); /* uses and clears teb->vm86_ptr */
2213         if (res < 0)
2214         {
2215             errno = -res;
2216             return;
2217         }
2218
2219         save_vm86_context( context, &vm86 );
2220
2221         rec.ExceptionFlags   = EXCEPTION_CONTINUABLE;
2222         rec.ExceptionRecord  = NULL;
2223         rec.ExceptionAddress = (LPVOID)context->Eip;
2224         rec.NumberParameters = 0;
2225
2226         switch(VM86_TYPE(res))
2227         {
2228         case VM86_UNKNOWN: /* unhandled GP fault - IO-instruction or similar */
2229             rec.ExceptionCode = EXCEPTION_PRIV_INSTRUCTION;
2230             break;
2231         case VM86_TRAP: /* return due to DOS-debugger request */
2232             switch(VM86_ARG(res))
2233             {
2234             case TRAP_x86_TRCTRAP:  /* Single-step exception */
2235                 rec.ExceptionCode = EXCEPTION_SINGLE_STEP;
2236                 break;
2237             case TRAP_x86_BPTFLT:   /* Breakpoint exception */
2238                 rec.ExceptionAddress = (char *)rec.ExceptionAddress - 1;  /* back up over the int3 instruction */
2239                 /* fall through */
2240             default:
2241                 rec.ExceptionCode = EXCEPTION_BREAKPOINT;
2242                 break;
2243             }
2244             break;
2245         case VM86_INTx: /* int3/int x instruction (ARG = x) */
2246             rec.ExceptionCode = EXCEPTION_VM86_INTx;
2247             rec.NumberParameters = 1;
2248             rec.ExceptionInformation[0] = VM86_ARG(res);
2249             break;
2250         case VM86_STI: /* sti/popf/iret instruction enabled virtual interrupts */
2251             context->EFlags |= VIF_FLAG;
2252             context->EFlags &= ~VIP_FLAG;
2253             get_vm86_teb_info()->vm86_pending = 0;
2254             rec.ExceptionCode = EXCEPTION_VM86_STI;
2255             break;
2256         case VM86_PICRETURN: /* return due to pending PIC request */
2257             rec.ExceptionCode = EXCEPTION_VM86_PICRETURN;
2258             break;
2259         case VM86_SIGNAL: /* cannot happen because vm86_enter handles this case */
2260         default:
2261             WINE_ERR( "unhandled result from vm86 mode %x\n", res );
2262             continue;
2263         }
2264         raise_exception( &rec, context, TRUE );
2265     }
2266 }
2267
2268 #else /* __HAVE_VM86 */
2269 /**********************************************************************
2270  *              __wine_enter_vm86   (NTDLL.@)
2271  */
2272 void __wine_enter_vm86( CONTEXT *context )
2273 {
2274     MESSAGE("vm86 mode not supported on this platform\n");
2275 }
2276 #endif /* __HAVE_VM86 */
2277
2278
2279 /*******************************************************************
2280  *              RtlUnwind (NTDLL.@)
2281  */
2282 void WINAPI __regs_RtlUnwind( EXCEPTION_REGISTRATION_RECORD* pEndFrame, PVOID targetIp,
2283                               PEXCEPTION_RECORD pRecord, PVOID retval, CONTEXT *context )
2284 {
2285     EXCEPTION_RECORD record;
2286     EXCEPTION_REGISTRATION_RECORD *frame, *dispatch;
2287     DWORD res;
2288
2289     context->Eax = (DWORD)retval;
2290
2291     /* build an exception record, if we do not have one */
2292     if (!pRecord)
2293     {
2294         record.ExceptionCode    = STATUS_UNWIND;
2295         record.ExceptionFlags   = 0;
2296         record.ExceptionRecord  = NULL;
2297         record.ExceptionAddress = (void *)context->Eip;
2298         record.NumberParameters = 0;
2299         pRecord = &record;
2300     }
2301
2302     pRecord->ExceptionFlags |= EH_UNWINDING | (pEndFrame ? 0 : EH_EXIT_UNWIND);
2303
2304     TRACE( "code=%x flags=%x\n", pRecord->ExceptionCode, pRecord->ExceptionFlags );
2305
2306     /* get chain of exception frames */
2307     frame = NtCurrentTeb()->Tib.ExceptionList;
2308     while ((frame != (EXCEPTION_REGISTRATION_RECORD*)~0UL) && (frame != pEndFrame))
2309     {
2310         /* Check frame address */
2311         if (pEndFrame && (frame > pEndFrame))
2312             raise_status( STATUS_INVALID_UNWIND_TARGET, pRecord );
2313
2314         if (((void*)frame < NtCurrentTeb()->Tib.StackLimit) ||
2315             ((void*)(frame+1) > NtCurrentTeb()->Tib.StackBase) ||
2316             (UINT_PTR)frame & 3)
2317             raise_status( STATUS_BAD_STACK, pRecord );
2318
2319         /* Call handler */
2320         TRACE( "calling handler at %p code=%x flags=%x\n",
2321                frame->Handler, pRecord->ExceptionCode, pRecord->ExceptionFlags );
2322         res = EXC_CallHandler( pRecord, frame, context, &dispatch, frame->Handler, unwind_handler );
2323         TRACE( "handler at %p returned %x\n", frame->Handler, res );
2324
2325         switch(res)
2326         {
2327         case ExceptionContinueSearch:
2328             break;
2329         case ExceptionCollidedUnwind:
2330             frame = dispatch;
2331             break;
2332         default:
2333             raise_status( STATUS_INVALID_DISPOSITION, pRecord );
2334             break;
2335         }
2336         frame = __wine_pop_frame( frame );
2337     }
2338 }
2339 DEFINE_REGS_ENTRYPOINT( RtlUnwind, 4 )
2340
2341
2342 /*******************************************************************
2343  *              NtRaiseException (NTDLL.@)
2344  */
2345 NTSTATUS WINAPI NtRaiseException( EXCEPTION_RECORD *rec, CONTEXT *context, BOOL first_chance )
2346 {
2347     NTSTATUS status = raise_exception( rec, context, first_chance );
2348     if (status == STATUS_SUCCESS) NtSetContextThread( GetCurrentThread(), context );
2349     return status;
2350 }
2351
2352
2353 /***********************************************************************
2354  *              RtlRaiseException (NTDLL.@)
2355  */
2356 void WINAPI __regs_RtlRaiseException( EXCEPTION_RECORD *rec, CONTEXT *context )
2357 {
2358     NTSTATUS status;
2359
2360     rec->ExceptionAddress = (void *)context->Eip;
2361     status = raise_exception( rec, context, TRUE );
2362     if (status != STATUS_SUCCESS) raise_status( status, rec );
2363 }
2364 DEFINE_REGS_ENTRYPOINT( RtlRaiseException, 1 )
2365
2366
2367 /*************************************************************************
2368  *              RtlCaptureStackBackTrace (NTDLL.@)
2369  */
2370 USHORT WINAPI RtlCaptureStackBackTrace( ULONG skip, ULONG count, PVOID *buffer, ULONG *hash )
2371 {
2372     CONTEXT context;
2373     ULONG i;
2374     ULONG *frame;
2375
2376     RtlCaptureContext( &context );
2377     if (hash) *hash = 0;
2378     frame = (ULONG *)context.Ebp;
2379
2380     while (skip--)
2381     {
2382         if (((void *)frame < NtCurrentTeb()->Tib.StackLimit) ||
2383             ((void *)(frame + 1) >= NtCurrentTeb()->Tib.StackBase)) return 0;
2384         frame = (ULONG *)*frame;
2385     }
2386
2387     for (i = 0; i < count; i++)
2388     {
2389         if (((void *)frame < NtCurrentTeb()->Tib.StackLimit) ||
2390             ((void *)(frame + 1) >= NtCurrentTeb()->Tib.StackBase)) break;
2391         buffer[i] = (void *)frame[1];
2392         if (hash) *hash += frame[1];
2393         frame = (ULONG *)*frame;
2394     }
2395     return i;
2396 }
2397
2398
2399 /* wrapper for apps that don't declare the thread function correctly */
2400 extern void DECLSPEC_NORETURN call_thread_func( LPTHREAD_START_ROUTINE entry, void *arg );
2401 __ASM_GLOBAL_FUNC(call_thread_func,
2402                   "pushl %ebp\n\t"
2403                   __ASM_CFI(".cfi_adjust_cfa_offset 4\n\t")
2404                   __ASM_CFI(".cfi_rel_offset %ebp,0\n\t")
2405                   "movl %esp,%ebp\n\t"
2406                   __ASM_CFI(".cfi_def_cfa_register %ebp\n\t")
2407                   "subl $4,%esp\n\t"
2408                   "pushl 12(%ebp)\n\t"
2409                   "call *8(%ebp)\n\t"
2410                   "leal -4(%ebp),%esp\n\t"
2411                   "pushl %eax\n\t"
2412                   "call " __ASM_NAME("exit_thread") "\n\t"
2413                   "int $3" )
2414
2415 /***********************************************************************
2416  *           call_thread_entry_point
2417  */
2418 void call_thread_entry_point( LPTHREAD_START_ROUTINE entry, void *arg )
2419 {
2420     __TRY
2421     {
2422         call_thread_func( entry, arg );
2423     }
2424     __EXCEPT(unhandled_exception_filter)
2425     {
2426         NtTerminateThread( GetCurrentThread(), GetExceptionCode() );
2427     }
2428     __ENDTRY
2429     abort();  /* should not be reached */
2430 }
2431
2432 /***********************************************************************
2433  *           RtlExitUserThread  (NTDLL.@)
2434  */
2435 void WINAPI RtlExitUserThread( ULONG status )
2436 {
2437     exit_thread( status );
2438 }
2439
2440 /***********************************************************************
2441  *           abort_thread
2442  */
2443 void abort_thread( int status )
2444 {
2445     terminate_thread( status );
2446 }
2447
2448 /**********************************************************************
2449  *              DbgBreakPoint   (NTDLL.@)
2450  */
2451 __ASM_STDCALL_FUNC( DbgBreakPoint, 0, "int $3; ret")
2452
2453 /**********************************************************************
2454  *              DbgUserBreakPoint   (NTDLL.@)
2455  */
2456 __ASM_STDCALL_FUNC( DbgUserBreakPoint, 0, "int $3; ret")
2457
2458 /**********************************************************************
2459  *           NtCurrentTeb   (NTDLL.@)
2460  */
2461 __ASM_STDCALL_FUNC( NtCurrentTeb, 0, ".byte 0x64\n\tmovl 0x18,%eax\n\tret" )
2462
2463
2464 /**********************************************************************
2465  *              EXC_CallHandler   (internal)
2466  *
2467  * Some exception handlers depend on EBP to have a fixed position relative to
2468  * the exception frame.
2469  * Shrinker depends on (*1) doing what it does,
2470  * (*2) being the exact instruction it is and (*3) beginning with 0x64
2471  * (i.e. the %fs prefix to the movl instruction). It also depends on the
2472  * function calling the handler having only 5 parameters (*4).
2473  */
2474 __ASM_GLOBAL_FUNC( EXC_CallHandler,
2475                   "pushl %ebp\n\t"
2476                   __ASM_CFI(".cfi_adjust_cfa_offset 4\n\t")
2477                   __ASM_CFI(".cfi_rel_offset %ebp,0\n\t")
2478                   "movl %esp,%ebp\n\t"
2479                   __ASM_CFI(".cfi_def_cfa_register %ebp\n\t")
2480                    "pushl %ebx\n\t"
2481                    __ASM_CFI(".cfi_rel_offset %ebx,-4\n\t")
2482                    "movl 28(%ebp), %edx\n\t" /* ugly hack to pass the 6th param needed because of Shrinker */
2483                    "pushl 24(%ebp)\n\t"
2484                    "pushl 20(%ebp)\n\t"
2485                    "pushl 16(%ebp)\n\t"
2486                    "pushl 12(%ebp)\n\t"
2487                    "pushl 8(%ebp)\n\t"
2488                    "call " __ASM_NAME("call_exception_handler") "\n\t"
2489                    "popl %ebx\n\t"
2490                    __ASM_CFI(".cfi_same_value %ebx\n\t")
2491                    "leave\n"
2492                    __ASM_CFI(".cfi_def_cfa %esp,4\n\t")
2493                    __ASM_CFI(".cfi_same_value %ebp\n\t")
2494                    "ret" )
2495 __ASM_GLOBAL_FUNC(call_exception_handler,
2496                   "pushl %ebp\n\t"
2497                   __ASM_CFI(".cfi_adjust_cfa_offset 4\n\t")
2498                   __ASM_CFI(".cfi_rel_offset %ebp,0\n\t")
2499                   "movl %esp,%ebp\n\t"
2500                   __ASM_CFI(".cfi_def_cfa_register %ebp\n\t")
2501                   "subl $12,%esp\n\t"
2502                   "pushl 12(%ebp)\n\t"      /* make any exceptions in this... */
2503                   "pushl %edx\n\t"          /* handler be handled by... */
2504                   ".byte 0x64\n\t"
2505                   "pushl (0)\n\t"           /* nested_handler (passed in edx). */
2506                   ".byte 0x64\n\t"
2507                   "movl %esp,(0)\n\t"       /* push the new exception frame onto the exception stack. */
2508                   "pushl 20(%ebp)\n\t"
2509                   "pushl 16(%ebp)\n\t"
2510                   "pushl 12(%ebp)\n\t"
2511                   "pushl 8(%ebp)\n\t"
2512                   "movl 24(%ebp), %ecx\n\t" /* (*1) */
2513                   "call *%ecx\n\t"          /* call handler. (*2) */
2514                   ".byte 0x64\n\t"
2515                   "movl (0), %esp\n\t"      /* restore previous... (*3) */
2516                   ".byte 0x64\n\t"
2517                   "popl (0)\n\t"            /* exception frame. */
2518                   "movl %ebp, %esp\n\t"     /* restore saved stack, in case it was corrupted */
2519                   "popl %ebp\n\t"
2520                    __ASM_CFI(".cfi_def_cfa %esp,4\n\t")
2521                    __ASM_CFI(".cfi_same_value %ebp\n\t")
2522                   "ret $20" )            /* (*4) */
2523
2524 #endif  /* __i386__ */