Remove races from DPMI async event handling.
[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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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 #ifdef HAVE_UNISTD_H
32 # include <unistd.h>
33 #endif
34
35 #ifdef HAVE_SYS_PARAM_H
36 # include <sys/param.h>
37 #endif
38 #ifdef HAVE_SYSCALL_H
39 # include <syscall.h>
40 #else
41 # ifdef HAVE_SYS_SYSCALL_H
42 #  include <sys/syscall.h>
43 # endif
44 #endif
45
46 #ifdef HAVE_SYS_VM86_H
47 # include <sys/vm86.h>
48 #endif
49
50 #ifdef HAVE_SYS_SIGNAL_H
51 # include <sys/signal.h>
52 #endif
53
54 #include "windef.h"
55 #include "winbase.h"
56 #include "winreg.h"
57 #include "winternl.h"
58 #include "wine/library.h"
59 #include "ntdll_misc.h"
60 #include "selectors.h"
61
62 /***********************************************************************
63  * signal context platform-specific definitions
64  */
65
66 #ifdef linux
67 typedef struct
68 {
69     unsigned short sc_gs, __gsh;
70     unsigned short sc_fs, __fsh;
71     unsigned short sc_es, __esh;
72     unsigned short sc_ds, __dsh;
73     unsigned long sc_edi;
74     unsigned long sc_esi;
75     unsigned long sc_ebp;
76     unsigned long sc_esp;
77     unsigned long sc_ebx;
78     unsigned long sc_edx;
79     unsigned long sc_ecx;
80     unsigned long sc_eax;
81     unsigned long sc_trapno;
82     unsigned long sc_err;
83     unsigned long sc_eip;
84     unsigned short sc_cs, __csh;
85     unsigned long sc_eflags;
86     unsigned long esp_at_signal;
87     unsigned short sc_ss, __ssh;
88     unsigned long i387;
89     unsigned long oldmask;
90     unsigned long cr2;
91 } SIGCONTEXT;
92
93 #define HANDLER_DEF(name) void name( int __signal, SIGCONTEXT __context )
94 #define HANDLER_CONTEXT (&__context)
95
96 /* this is the sigaction structure from the Linux 2.1.20 kernel.  */
97 struct kernel_sigaction
98 {
99     void (*ksa_handler)();
100     unsigned long ksa_mask;
101     unsigned long ksa_flags;
102     void *ksa_restorer;
103 };
104
105 /* Similar to the sigaction function in libc, except it leaves alone the
106    restorer field, which is used to specify the signal stack address */
107 static inline int wine_sigaction( int sig, struct kernel_sigaction *new,
108                                   struct kernel_sigaction *old )
109 {
110     __asm__ __volatile__( "pushl %%ebx\n\t"
111                           "movl %2,%%ebx\n\t"
112                           "int $0x80\n\t"
113                           "popl %%ebx"
114                           : "=a" (sig)
115                           : "0" (SYS_sigaction), "r" (sig), "c" (new), "d" (old) );
116     if (sig>=0) return 0;
117     errno = -sig;
118     return -1;
119 }
120
121 #ifdef HAVE_SIGALTSTACK
122 /* direct syscall for sigaltstack to work around glibc 2.0 brain-damage */
123 static inline int wine_sigaltstack( const struct sigaltstack *new,
124                                     struct sigaltstack *old )
125 {
126     int ret;
127     __asm__ __volatile__( "pushl %%ebx\n\t"
128                           "movl %2,%%ebx\n\t"
129                           "int $0x80\n\t"
130                           "popl %%ebx"
131                           : "=a" (ret)
132                           : "0" (SYS_sigaltstack), "r" (new), "c" (old) );
133     if (ret >= 0) return 0;
134     errno = -ret;
135     return -1;
136 }
137 #endif
138
139 #define VM86_EAX 0 /* the %eax value while vm86_enter is executing */
140
141 int vm86_enter( void **vm86_ptr );
142 void vm86_return(void);
143 void vm86_return_end(void);
144 __ASM_GLOBAL_FUNC(vm86_enter,
145                   "pushl %ebp\n\t"
146                   "movl %esp, %ebp\n\t"
147                   "movl $166,%eax\n\t"  /*SYS_vm86*/
148                   "movl 8(%ebp),%ecx\n\t" /* vm86_ptr */
149                   "movl (%ecx),%ecx\n\t"
150                   "pushl %ebx\n\t"
151                   "movl $1,%ebx\n\t"    /*VM86_ENTER*/
152                   "pushl %ecx\n\t"      /* put vm86plus_struct ptr somewhere we can find it */
153                   "pushl %gs\n\t"
154                   "pushl %fs\n\t"
155                   "int $0x80\n"
156                   ".globl " __ASM_NAME("vm86_return") "\n\t"
157                   __ASM_FUNC("vm86_return") "\n"
158                   __ASM_NAME("vm86_return") ":\n\t"
159                   "popl %fs\n\t"
160                   "popl %gs\n\t"
161                   "popl %ecx\n\t"
162                   "popl %ebx\n\t"
163                   "popl %ebp\n\t"
164                   "testl %eax,%eax\n\t"
165                   "jl 0f\n\t"
166                   "cmpb $0,%al\n\t" /* VM86_SIGNAL */
167                   "je " __ASM_NAME("vm86_enter") "\n\t"
168                   "0:\n\t"
169                   "movl 4(%esp),%ecx\n\t"  /* vm86_ptr */
170                   "movl $0,(%ecx)\n\t"
171                   ".globl " __ASM_NAME("vm86_return_end") "\n\t"
172                   __ASM_FUNC("vm86_return_end") "\n"
173                   __ASM_NAME("vm86_return_end") ":\n\t"
174                   "ret" );
175
176 #ifdef HAVE_SYS_VM86_H
177 # define __HAVE_VM86
178 #endif
179
180 #endif  /* linux */
181
182 #ifdef BSDI
183
184 #define EAX_sig(context)     ((context)->tf_eax)
185 #define EBX_sig(context)     ((context)->tf_ebx)
186 #define ECX_sig(context)     ((context)->tf_ecx)
187 #define EDX_sig(context)     ((context)->tf_edx)
188 #define ESI_sig(context)     ((context)->tf_esi)
189 #define EDI_sig(context)     ((context)->tf_edi)
190 #define EBP_sig(context)     ((context)->tf_ebp)
191
192 #define CS_sig(context)      ((context)->tf_cs)
193 #define DS_sig(context)      ((context)->tf_ds)
194 #define ES_sig(context)      ((context)->tf_es)
195 #define SS_sig(context)      ((context)->tf_ss)
196
197 #include <machine/frame.h>
198 typedef struct trapframe SIGCONTEXT;
199
200 #define HANDLER_DEF(name) void name( int __signal, int code, SIGCONTEXT *__context )
201 #define HANDLER_CONTEXT __context
202
203 #define EFL_sig(context)     ((context)->tf_eflags)
204
205 #define EIP_sig(context)     (*((unsigned long*)&(context)->tf_eip))
206 #define ESP_sig(context)     (*((unsigned long*)&(context)->tf_esp))
207
208 #endif /* bsdi */
209
210 #if defined(__NetBSD__) || defined(__FreeBSD__) || defined(__OpenBSD__)
211
212 typedef struct sigcontext SIGCONTEXT;
213
214 #define HANDLER_DEF(name) void name( int __signal, int code, SIGCONTEXT *__context )
215 #define HANDLER_CONTEXT __context
216
217 #endif  /* FreeBSD */
218
219 #if defined(__svr4__) || defined(_SCO_DS) || defined(__sun)
220
221 #ifdef _SCO_DS
222 #include <sys/regset.h>
223 #endif
224 /* Solaris kludge */
225 #undef ERR
226 #include <sys/ucontext.h>
227 #undef ERR
228 typedef struct ucontext SIGCONTEXT;
229
230 #define HANDLER_DEF(name) void name( int __signal, void *__siginfo, SIGCONTEXT *__context )
231 #define HANDLER_CONTEXT __context
232
233 #endif  /* svr4 || SCO_DS */
234
235 #ifdef __EMX__
236
237 typedef struct
238 {
239     unsigned long ContextFlags;
240     FLOATING_SAVE_AREA sc_float;
241     unsigned long sc_gs;
242     unsigned long sc_fs;
243     unsigned long sc_es;
244     unsigned long sc_ds;
245     unsigned long sc_edi;
246     unsigned long sc_esi;
247     unsigned long sc_eax;
248     unsigned long sc_ebx;
249     unsigned long sc_ecx;
250     unsigned long sc_edx;
251     unsigned long sc_ebp;
252     unsigned long sc_eip;
253     unsigned long sc_cs;
254     unsigned long sc_eflags;
255     unsigned long sc_esp;
256     unsigned long sc_ss;
257 } SIGCONTEXT;
258
259 #endif  /* __EMX__ */
260
261 #ifdef __CYGWIN__
262
263 /* FIXME: This section is just here so it can compile, it's most likely
264  * completely wrong. */
265
266 typedef struct
267 {
268     unsigned short sc_gs, __gsh;
269     unsigned short sc_fs, __fsh;
270     unsigned short sc_es, __esh;
271     unsigned short sc_ds, __dsh;
272     unsigned long sc_edi;
273     unsigned long sc_esi;
274     unsigned long sc_ebp;
275     unsigned long sc_esp;
276     unsigned long sc_ebx;
277     unsigned long sc_edx;
278     unsigned long sc_ecx;
279     unsigned long sc_eax;
280     unsigned long sc_trapno;
281     unsigned long sc_err;
282     unsigned long sc_eip;
283     unsigned short sc_cs, __csh;
284     unsigned long sc_eflags;
285     unsigned long esp_at_signal;
286     unsigned short sc_ss, __ssh;
287     unsigned long i387;
288     unsigned long oldmask;
289     unsigned long cr2;
290 } SIGCONTEXT;
291
292 #define HANDLER_DEF(name) void name( int __signal, SIGCONTEXT __context )
293 #define HANDLER_CONTEXT (&__context)
294
295 #endif /* __CYGWIN__ */
296
297 #if defined(linux) || defined(__NetBSD__) || defined(__FreeBSD__) ||\
298     defined(__OpenBSD__) || defined(__EMX__) || defined(__CYGWIN__)
299
300 #define EAX_sig(context)     ((context)->sc_eax)
301 #define EBX_sig(context)     ((context)->sc_ebx)
302 #define ECX_sig(context)     ((context)->sc_ecx)
303 #define EDX_sig(context)     ((context)->sc_edx)
304 #define ESI_sig(context)     ((context)->sc_esi)
305 #define EDI_sig(context)     ((context)->sc_edi)
306 #define EBP_sig(context)     ((context)->sc_ebp)
307
308 #define CS_sig(context)      ((context)->sc_cs)
309 #define DS_sig(context)      ((context)->sc_ds)
310 #define ES_sig(context)      ((context)->sc_es)
311 #define FS_sig(context)      ((context)->sc_fs)
312 #define GS_sig(context)      ((context)->sc_gs)
313 #define SS_sig(context)      ((context)->sc_ss)
314
315 #define TRAP_sig(context)    ((context)->sc_trapno)
316
317 #ifdef __NetBSD__
318 #define ERROR_sig(context)   ((context)->sc_err)
319 #endif
320
321 #ifdef linux
322 #define ERROR_sig(context)   ((context)->sc_err)
323 #define FPU_sig(context)     ((FLOATING_SAVE_AREA*)((context)->i387))
324 #define FAULT_ADDRESS        ((void *)HANDLER_CONTEXT->cr2)
325 #endif
326
327 #ifdef __FreeBSD__
328 #define EFL_sig(context)     ((context)->sc_efl)
329 /* FreeBSD, see i386/i386/traps.c::trap_pfault va->err kludge  */
330 #define FAULT_ADDRESS        ((void *)HANDLER_CONTEXT->sc_err)
331 #else
332 #define EFL_sig(context)     ((context)->sc_eflags)
333 #endif
334
335 #define EIP_sig(context)     (*((unsigned long*)&(context)->sc_eip))
336 #define ESP_sig(context)     (*((unsigned long*)&(context)->sc_esp))
337
338 #endif  /* linux || __NetBSD__ || __FreeBSD__ || __OpenBSD__ */
339
340 #if defined(__svr4__) || defined(_SCO_DS) || defined(__sun)
341
342 #ifdef _SCO_DS
343 #define gregs regs
344 #endif
345
346 #define EAX_sig(context)     ((context)->uc_mcontext.gregs[EAX])
347 #define EBX_sig(context)     ((context)->uc_mcontext.gregs[EBX])
348 #define ECX_sig(context)     ((context)->uc_mcontext.gregs[ECX])
349 #define EDX_sig(context)     ((context)->uc_mcontext.gregs[EDX])
350 #define ESI_sig(context)     ((context)->uc_mcontext.gregs[ESI])
351 #define EDI_sig(context)     ((context)->uc_mcontext.gregs[EDI])
352 #define EBP_sig(context)     ((context)->uc_mcontext.gregs[EBP])
353
354 #define CS_sig(context)      ((context)->uc_mcontext.gregs[CS])
355 #define DS_sig(context)      ((context)->uc_mcontext.gregs[DS])
356 #define ES_sig(context)      ((context)->uc_mcontext.gregs[ES])
357 #define SS_sig(context)      ((context)->uc_mcontext.gregs[SS])
358
359 #define FS_sig(context)      ((context)->uc_mcontext.gregs[FS])
360 #define GS_sig(context)      ((context)->uc_mcontext.gregs[GS])
361
362 #define EFL_sig(context)     ((context)->uc_mcontext.gregs[EFL])
363
364 #define EIP_sig(context)     ((context)->uc_mcontext.gregs[EIP])
365 #ifdef R_ESP
366 #define ESP_sig(context)     ((context)->uc_mcontext.gregs[R_ESP])
367 #else
368 #define ESP_sig(context)     ((context)->uc_mcontext.gregs[ESP])
369 #endif
370 #ifdef TRAPNO
371 #define TRAP_sig(context)     ((context)->uc_mcontext.gregs[TRAPNO])
372 #endif
373
374 #define FAULT_ADDRESS         (__siginfo->si_addr)
375
376 #endif  /* svr4 || SCO_DS */
377
378
379 /* exception code definitions (already defined by FreeBSD/NetBSD) */
380 #if !defined(__FreeBSD__) && !defined(__NetBSD__) /* FIXME: other BSDs? */
381 #define T_DIVIDE        0   /* Division by zero exception */
382 #define T_TRCTRAP       1   /* Single-step exception */
383 #define T_NMI           2   /* NMI interrupt */
384 #define T_BPTFLT        3   /* Breakpoint exception */
385 #define T_OFLOW         4   /* Overflow exception */
386 #define T_BOUND         5   /* Bound range exception */
387 #define T_PRIVINFLT     6   /* Invalid opcode exception */
388 #define T_DNA           7   /* Device not available exception */
389 #define T_DOUBLEFLT     8   /* Double fault exception */
390 #define T_FPOPFLT       9   /* Coprocessor segment overrun */
391 #define T_TSSFLT        10  /* Invalid TSS exception */
392 #define T_SEGNPFLT      11  /* Segment not present exception */
393 #define T_STKFLT        12  /* Stack fault */
394 #define T_PROTFLT       13  /* General protection fault */
395 #define T_PAGEFLT       14  /* Page fault */
396 #define T_RESERVED      15  /* Unknown exception */
397 #define T_ARITHTRAP     16  /* Floating point exception */
398 #define T_ALIGNFLT      17  /* Alignment check exception */
399 #define T_MCHK          18  /* Machine check exception */
400 #define T_CACHEFLT      19  /* Cache flush exception */
401 #endif
402 #if defined(__NetBSD__)
403 #define T_MCHK          19  /* Machine check exception */
404 #endif
405
406 #define T_UNKNOWN     (-1)  /* Unknown fault (TRAP_sig not defined) */
407
408 #include "wine/exception.h"
409 #include "global.h"
410 #include "wine/debug.h"
411
412 WINE_DEFAULT_DEBUG_CHANNEL(seh);
413
414 typedef int (*wine_signal_handler)(unsigned int sig);
415
416 static wine_signal_handler handlers[256];
417
418 extern void WINAPI EXC_RtlRaiseException( PEXCEPTION_RECORD, PCONTEXT );
419
420 /* Global variable to save the thread %fs register while in 16-bit code (FIXME) */
421 static unsigned int signal_fs;
422
423 /***********************************************************************
424  *           dispatch_signal
425  */
426 inline static int dispatch_signal(unsigned int sig)
427 {
428     if (handlers[sig] == NULL) return 0;
429     return handlers[sig](sig);
430 }
431
432
433 /***********************************************************************
434  *           get_trap_code
435  *
436  * Get the trap code for a signal.
437  */
438 static inline int get_trap_code( const SIGCONTEXT *sigcontext )
439 {
440 #ifdef TRAP_sig
441     return TRAP_sig(sigcontext);
442 #else
443     return T_UNKNOWN;  /* unknown trap code */
444 #endif
445 }
446
447 /***********************************************************************
448  *           get_error_code
449  *
450  * Get the error code for a signal.
451  */
452 static inline int get_error_code( const SIGCONTEXT *sigcontext )
453 {
454 #ifdef ERROR_sig
455     return ERROR_sig(sigcontext);
456 #else
457     return 0;
458 #endif
459 }
460
461 #ifdef __HAVE_VM86
462 /***********************************************************************
463  *           save_vm86_context
464  *
465  * Set the register values from a vm86 structure.
466  */
467 static void save_vm86_context( CONTEXT *context, const struct vm86plus_struct *vm86 )
468 {
469     context->Eax    = vm86->regs.eax;
470     context->Ebx    = vm86->regs.ebx;
471     context->Ecx    = vm86->regs.ecx;
472     context->Edx    = vm86->regs.edx;
473     context->Esi    = vm86->regs.esi;
474     context->Edi    = vm86->regs.edi;
475     context->Esp    = vm86->regs.esp;
476     context->Ebp    = vm86->regs.ebp;
477     context->Eip    = vm86->regs.eip;
478     context->SegCs  = vm86->regs.cs;
479     context->SegDs  = vm86->regs.ds;
480     context->SegEs  = vm86->regs.es;
481     context->SegFs  = vm86->regs.fs;
482     context->SegGs  = vm86->regs.gs;
483     context->SegSs  = vm86->regs.ss;
484     context->EFlags = vm86->regs.eflags;
485 }
486
487
488 /***********************************************************************
489  *           restore_vm86_context
490  *
491  * Build a vm86 structure from the register values.
492  */
493 static void restore_vm86_context( const CONTEXT *context, struct vm86plus_struct *vm86 )
494 {
495     vm86->regs.eax    = context->Eax;
496     vm86->regs.ebx    = context->Ebx;
497     vm86->regs.ecx    = context->Ecx;
498     vm86->regs.edx    = context->Edx;
499     vm86->regs.esi    = context->Esi;
500     vm86->regs.edi    = context->Edi;
501     vm86->regs.esp    = context->Esp;
502     vm86->regs.ebp    = context->Ebp;
503     vm86->regs.eip    = context->Eip;
504     vm86->regs.cs     = context->SegCs;
505     vm86->regs.ds     = context->SegDs;
506     vm86->regs.es     = context->SegEs;
507     vm86->regs.fs     = context->SegFs;
508     vm86->regs.gs     = context->SegGs;
509     vm86->regs.ss     = context->SegSs;
510     vm86->regs.eflags = context->EFlags;
511 }
512 #endif /* __HAVE_VM86 */
513
514
515 /***********************************************************************
516  *           save_context
517  *
518  * Set the register values from a sigcontext.
519  */
520 static void save_context( CONTEXT *context, const SIGCONTEXT *sigcontext )
521 {
522     /* get %fs and %gs at time of the fault */
523 #ifdef FS_sig
524     context->SegFs = LOWORD(FS_sig(sigcontext));
525 #else
526     context->SegFs = wine_get_fs();
527 #endif
528 #ifdef GS_sig
529     context->SegGs = LOWORD(GS_sig(sigcontext));
530 #else
531     context->SegGs = wine_get_gs();
532 #endif
533
534     /* now restore a proper %fs for the fault handler */
535     if (!IS_SELECTOR_SYSTEM(CS_sig(sigcontext)) ||
536         !IS_SELECTOR_SYSTEM(SS_sig(sigcontext)))  /* 16-bit mode */
537     {
538         /*
539          * Win16 or DOS protected mode. Note that during switch 
540          * from 16-bit mode to linear mode, CS may be set to system 
541          * segment before FS is restored. Fortunately, in this case 
542          * SS is still non-system segment. This is why both CS and SS
543          * are checked.
544          */
545         wine_set_fs( signal_fs );
546         wine_set_gs( NtCurrentTeb()->gs_sel );
547     }
548 #ifdef __HAVE_VM86
549     else if ((void *)EIP_sig(sigcontext) == vm86_return)  /* vm86 mode */
550     {
551         unsigned int *stack = (unsigned int *)ESP_sig(sigcontext);
552
553         /* fetch the saved %fs on the stack */
554         wine_set_fs( stack[0] );
555         wine_set_gs( stack[1] );
556         if (EAX_sig(sigcontext) == VM86_EAX) {
557             /* retrieve pointer to vm86plus struct that was stored in vm86_enter
558              * (but we could also get if from teb->vm86_ptr) */
559             struct vm86plus_struct *vm86 = (struct vm86plus_struct *)stack[2];
560             /* get context from vm86 struct */
561             save_vm86_context( context, vm86 );
562             return;
563         }
564     }
565 #endif  /* __HAVE_VM86 */
566     else  /* 32-bit mode */
567     {
568 #ifdef FS_sig
569         wine_set_fs( FS_sig(sigcontext) );
570 #endif
571 #ifdef GS_sig
572         wine_set_gs( GS_sig(sigcontext) );
573 #endif
574     }
575
576     context->Eax    = EAX_sig(sigcontext);
577     context->Ebx    = EBX_sig(sigcontext);
578     context->Ecx    = ECX_sig(sigcontext);
579     context->Edx    = EDX_sig(sigcontext);
580     context->Esi    = ESI_sig(sigcontext);
581     context->Edi    = EDI_sig(sigcontext);
582     context->Ebp    = EBP_sig(sigcontext);
583     context->EFlags = EFL_sig(sigcontext);
584     context->Eip    = EIP_sig(sigcontext);
585     context->Esp    = ESP_sig(sigcontext);
586     context->SegCs  = LOWORD(CS_sig(sigcontext));
587     context->SegDs  = LOWORD(DS_sig(sigcontext));
588     context->SegEs  = LOWORD(ES_sig(sigcontext));
589     context->SegSs  = LOWORD(SS_sig(sigcontext));
590 }
591
592
593 /***********************************************************************
594  *           restore_context
595  *
596  * Build a sigcontext from the register values.
597  */
598 static void restore_context( CONTEXT *context, SIGCONTEXT *sigcontext )
599 {
600 #ifdef __HAVE_VM86
601     BOOL check_pending = TRUE;
602
603     /* check if exception occurred in vm86 mode */
604     if ((void *)EIP_sig(sigcontext) == vm86_return &&
605         IS_SELECTOR_SYSTEM(CS_sig(sigcontext)) &&
606         EAX_sig(sigcontext) == VM86_EAX)
607     {
608         unsigned int *stack = (unsigned int *)ESP_sig(sigcontext);
609         /* retrieve pointer to vm86plus struct that was stored in vm86_enter
610          * (but we could also get it from teb->vm86_ptr) */
611         struct vm86plus_struct *vm86 = (struct vm86plus_struct *)stack[2];
612         restore_vm86_context( context, vm86 );
613         return;
614     }
615
616     while (NtCurrentTeb()->dpmi_vif &&
617            !IS_SELECTOR_SYSTEM(context->SegCs) &&
618            !IS_SELECTOR_SYSTEM(context->SegSs) &&
619            check_pending)
620     {
621         /*
622          * Executing DPMI code and virtual interrupts are enabled. 
623          * We must block signals so that we can be safely check
624          * for pending asynchronous events. Return from signal handler
625          * will unblock signals again so it is is safe to do so.
626          */
627         SIGNAL_Block();
628         check_pending = FALSE;
629
630         if (NtCurrentTeb()->vm86_pending)
631         {
632             EXCEPTION_RECORD rec;
633
634             rec.ExceptionAddress        = (LPVOID)context->Eip;
635             rec.ExceptionCode           = EXCEPTION_VM86_STI;
636             rec.ExceptionFlags          = EXCEPTION_CONTINUABLE;
637             rec.ExceptionRecord         = NULL;
638             rec.NumberParameters        = 1;
639             rec.ExceptionInformation[0] = 0;
640
641             NtCurrentTeb()->vm86_pending = 0;
642             EXC_RtlRaiseException( &rec, context );
643
644             /* 
645              * EXC_RtlRaiseException has unblocked all signals
646              * and we must retry check for pending asynchronous 
647              * events in order to prevent races.
648              */
649             check_pending = TRUE;
650         }
651     }
652 #endif /* __HAVE_VM86 */
653
654     EAX_sig(sigcontext) = context->Eax;
655     EBX_sig(sigcontext) = context->Ebx;
656     ECX_sig(sigcontext) = context->Ecx;
657     EDX_sig(sigcontext) = context->Edx;
658     ESI_sig(sigcontext) = context->Esi;
659     EDI_sig(sigcontext) = context->Edi;
660     EBP_sig(sigcontext) = context->Ebp;
661     EFL_sig(sigcontext) = context->EFlags;
662     EIP_sig(sigcontext) = context->Eip;
663     ESP_sig(sigcontext) = context->Esp;
664     CS_sig(sigcontext)  = context->SegCs;
665     DS_sig(sigcontext)  = context->SegDs;
666     ES_sig(sigcontext)  = context->SegEs;
667     SS_sig(sigcontext)  = context->SegSs;
668 #ifdef FS_sig
669     FS_sig(sigcontext)  = context->SegFs;
670 #else
671     wine_set_fs( context->SegFs );
672 #endif
673 #ifdef GS_sig
674     GS_sig(sigcontext)  = context->SegGs;
675 #else
676     wine_set_gs( context->SegGs );
677 #endif
678 }
679
680
681 /***********************************************************************
682  *           init_handler
683  *
684  * Handler initialization when the full context is not needed.
685  */
686 static void init_handler( const SIGCONTEXT *sigcontext )
687 {
688     /* restore a proper %fs for the fault handler */
689     if (!IS_SELECTOR_SYSTEM(CS_sig(sigcontext)) ||
690         !IS_SELECTOR_SYSTEM(SS_sig(sigcontext)))  /* 16-bit mode */
691     {
692         wine_set_fs( signal_fs );
693         wine_set_gs( NtCurrentTeb()->gs_sel );
694     }
695 #ifdef __HAVE_VM86
696     else if ((void *)EIP_sig(sigcontext) == vm86_return)  /* vm86 mode */
697     {
698         /* fetch the saved %fs on the stack */
699         wine_set_fs( *(unsigned int *)ESP_sig(sigcontext) );
700         wine_set_gs( NtCurrentTeb()->gs_sel );
701     }
702 #endif  /* __HAVE_VM86 */
703     else  /* 32-bit mode, get %fs at time of the fault */
704     {
705 #ifdef FS_sig
706         wine_set_fs( FS_sig(sigcontext) );
707 #endif
708 #ifdef GS_sig
709         wine_set_gs( GS_sig(sigcontext) );
710 #endif
711     }
712 }
713
714
715 /***********************************************************************
716  *           save_fpu
717  *
718  * Set the FPU context from a sigcontext.
719  */
720 inline static void save_fpu( CONTEXT *context, const SIGCONTEXT *sigcontext )
721 {
722 #ifdef FPU_sig
723     if (FPU_sig(sigcontext))
724     {
725         context->FloatSave = *FPU_sig(sigcontext);
726         return;
727     }
728 #endif  /* FPU_sig */
729 #ifdef __GNUC__
730     __asm__ __volatile__( "fnsave %0; fwait" : "=m" (context->FloatSave) );
731 #endif  /* __GNUC__ */
732 }
733
734
735 /***********************************************************************
736  *           restore_fpu
737  *
738  * Restore the FPU context to a sigcontext.
739  */
740 inline static void restore_fpu( CONTEXT *context, const SIGCONTEXT *sigcontext )
741 {
742     /* reset the current interrupt status */
743     context->FloatSave.StatusWord &= context->FloatSave.ControlWord | 0xffffff80;
744 #ifdef FPU_sig
745     if (FPU_sig(sigcontext))
746     {
747         *FPU_sig(sigcontext) = context->FloatSave;
748         return;
749     }
750 #endif  /* FPU_sig */
751 #ifdef __GNUC__
752     /* avoid nested exceptions */
753     __asm__ __volatile__( "frstor %0; fwait" : : "m" (context->FloatSave) );
754 #endif  /* __GNUC__ */
755 }
756
757
758 /**********************************************************************
759  *              get_fpu_code
760  *
761  * Get the FPU exception code from the FPU status.
762  */
763 static inline DWORD get_fpu_code( const CONTEXT *context )
764 {
765     DWORD status = context->FloatSave.StatusWord;
766
767     if (status & 0x01)  /* IE */
768     {
769         if (status & 0x40)  /* SF */
770             return EXCEPTION_FLT_STACK_CHECK;
771         else
772             return EXCEPTION_FLT_INVALID_OPERATION;
773     }
774     if (status & 0x02) return EXCEPTION_FLT_DENORMAL_OPERAND;  /* DE flag */
775     if (status & 0x04) return EXCEPTION_FLT_DIVIDE_BY_ZERO;    /* ZE flag */
776     if (status & 0x08) return EXCEPTION_FLT_OVERFLOW;          /* OE flag */
777     if (status & 0x10) return EXCEPTION_FLT_UNDERFLOW;         /* UE flag */
778     if (status & 0x20) return EXCEPTION_FLT_INEXACT_RESULT;    /* PE flag */
779     return EXCEPTION_FLT_INVALID_OPERATION;  /* generic error */
780 }
781
782
783 /**********************************************************************
784  *              do_segv
785  *
786  * Implementation of SIGSEGV handler.
787  */
788 static void do_segv( CONTEXT *context, int trap_code, void *cr2, int err_code )
789 {
790     EXCEPTION_RECORD rec;
791
792     rec.ExceptionRecord  = NULL;
793     rec.ExceptionFlags   = EXCEPTION_CONTINUABLE;
794     rec.ExceptionAddress = (LPVOID)context->Eip;
795     rec.NumberParameters = 0;
796
797     switch(trap_code)
798     {
799     case T_OFLOW:   /* Overflow exception */
800         rec.ExceptionCode = EXCEPTION_INT_OVERFLOW;
801         break;
802     case T_BOUND:   /* Bound range exception */
803         rec.ExceptionCode = EXCEPTION_ARRAY_BOUNDS_EXCEEDED;
804         break;
805     case T_PRIVINFLT:   /* Invalid opcode exception */
806         rec.ExceptionCode = EXCEPTION_ILLEGAL_INSTRUCTION;
807         break;
808     case T_STKFLT:  /* Stack fault */
809         rec.ExceptionCode = EXCEPTION_STACK_OVERFLOW;
810         break;
811     case T_SEGNPFLT:  /* Segment not present exception */
812     case T_PROTFLT:   /* General protection fault */
813     case T_UNKNOWN:   /* Unknown fault code */
814         rec.ExceptionCode = err_code ? EXCEPTION_ACCESS_VIOLATION : EXCEPTION_PRIV_INSTRUCTION;
815         break;
816     case T_PAGEFLT:  /* Page fault */
817 #ifdef FAULT_ADDRESS
818         if (!(rec.ExceptionCode = VIRTUAL_HandleFault( cr2 ))) return;
819         rec.NumberParameters = 2;
820         rec.ExceptionInformation[0] = (err_code & 2) != 0;
821         rec.ExceptionInformation[1] = (DWORD)cr2;
822 #else
823         rec.ExceptionCode = EXCEPTION_ACCESS_VIOLATION;
824 #endif
825         break;
826     case T_ALIGNFLT:  /* Alignment check exception */
827         /* FIXME: pass through exception handler first? */
828         if (context->EFlags & 0x00040000)
829         {
830             /* Disable AC flag, return */
831             context->EFlags &= ~0x00040000;
832             return;
833         }
834         rec.ExceptionCode = EXCEPTION_DATATYPE_MISALIGNMENT;
835         break;
836     default:
837         ERR( "Got unexpected trap %d\n", trap_code );
838         /* fall through */
839     case T_NMI:       /* NMI interrupt */
840     case T_DNA:       /* Device not available exception */
841     case T_DOUBLEFLT: /* Double fault exception */
842     case T_TSSFLT:    /* Invalid TSS exception */
843     case T_RESERVED:  /* Unknown exception */
844     case T_MCHK:      /* Machine check exception */
845 #ifdef T_CACHEFLT
846     case T_CACHEFLT:  /* Cache flush exception */
847 #endif
848         rec.ExceptionCode = EXCEPTION_ILLEGAL_INSTRUCTION;
849         break;
850     }
851     EXC_RtlRaiseException( &rec, context );
852 }
853
854
855 /**********************************************************************
856  *              do_trap
857  *
858  * Implementation of SIGTRAP handler.
859  */
860 static void do_trap( CONTEXT *context, int trap_code )
861 {
862     EXCEPTION_RECORD rec;
863     DWORD dr0, dr1, dr2, dr3, dr6, dr7;
864
865     rec.ExceptionFlags   = EXCEPTION_CONTINUABLE;
866     rec.ExceptionRecord  = NULL;
867     rec.ExceptionAddress = (LPVOID)context->Eip;
868     rec.NumberParameters = 0;
869
870     switch(trap_code)
871     {
872     case T_TRCTRAP:  /* Single-step exception */
873         rec.ExceptionCode = EXCEPTION_SINGLE_STEP;
874         if (context->EFlags & 0x100)
875         {
876             context->EFlags &= ~0x100;  /* clear single-step flag */
877         }
878         else  /* hardware breakpoint, fetch the debug registers */
879         {
880             context->ContextFlags = CONTEXT_DEBUG_REGISTERS;
881             NtGetContextThread(GetCurrentThread(), context);
882             /* do we really have a bp from a debug register ?
883              * if not, then someone did a kill(SIGTRAP) on us, and we
884              * shall return a breakpoint, not a single step exception
885              */
886             if (!(context->Dr6 & 0xf))
887                 rec.ExceptionCode = EXCEPTION_BREAKPOINT;
888         }
889         break;
890     case T_BPTFLT:   /* Breakpoint exception */
891         rec.ExceptionAddress = (char *)rec.ExceptionAddress - 1;  /* back up over the int3 instruction */
892         /* fall through */
893     default:
894         rec.ExceptionCode = EXCEPTION_BREAKPOINT;
895         break;
896     }
897     dr0 = context->Dr0;
898     dr1 = context->Dr1;
899     dr2 = context->Dr2;
900     dr3 = context->Dr3;
901     dr6 = context->Dr6;
902     dr7 = context->Dr7;
903
904     EXC_RtlRaiseException( &rec, context );
905
906     if (dr0 != context->Dr0 || dr1 != context->Dr1 || dr2 != context->Dr2 ||
907         dr3 != context->Dr3 || dr6 != context->Dr6 || dr7 != context->Dr7)
908     {
909         /* the debug registers have changed, set the new values */
910         context->ContextFlags = CONTEXT_DEBUG_REGISTERS;
911         NtSetContextThread(GetCurrentThread(), context);
912     }
913 }
914
915
916 /**********************************************************************
917  *              do_fpe
918  *
919  * Implementation of SIGFPE handler
920  */
921 static void do_fpe( CONTEXT *context, int trap_code )
922 {
923     EXCEPTION_RECORD rec;
924
925     switch(trap_code)
926     {
927     case T_DIVIDE:   /* Division by zero exception */
928         rec.ExceptionCode = EXCEPTION_INT_DIVIDE_BY_ZERO;
929         break;
930     case T_FPOPFLT:   /* Coprocessor segment overrun */
931         rec.ExceptionCode = EXCEPTION_FLT_INVALID_OPERATION;
932         break;
933     case T_ARITHTRAP:  /* Floating point exception */
934     case T_UNKNOWN:    /* Unknown fault code */
935         rec.ExceptionCode = get_fpu_code( context );
936         break;
937     default:
938         ERR( "Got unexpected trap %d\n", trap_code );
939         rec.ExceptionCode = EXCEPTION_FLT_INVALID_OPERATION;
940         break;
941     }
942     rec.ExceptionFlags   = EXCEPTION_CONTINUABLE;
943     rec.ExceptionRecord  = NULL;
944     rec.ExceptionAddress = (LPVOID)context->Eip;
945     rec.NumberParameters = 0;
946     EXC_RtlRaiseException( &rec, context );
947 }
948
949
950 #ifdef __HAVE_VM86
951 /**********************************************************************
952  *              set_vm86_pend
953  *
954  * Handler for SIGUSR2, which we use to set the vm86 pending flag.
955  */
956 static void set_vm86_pend( CONTEXT *context )
957 {
958     EXCEPTION_RECORD rec;
959     TEB *teb = NtCurrentTeb();
960     struct vm86plus_struct *vm86 = (struct vm86plus_struct*)(teb->vm86_ptr);
961
962     rec.ExceptionCode           = EXCEPTION_VM86_STI;
963     rec.ExceptionFlags          = EXCEPTION_CONTINUABLE;
964     rec.ExceptionRecord         = NULL;
965     rec.NumberParameters        = 1;
966     rec.ExceptionInformation[0] = 0;
967
968     /* __wine_enter_vm86() merges the vm86_pending flag in safely */
969     teb->vm86_pending |= VIP_MASK;
970     /* see if we were in VM86 mode */
971     if (context->EFlags & 0x00020000)
972     {
973         /* seems so, also set flag in signal context */
974         if (context->EFlags & VIP_MASK) return;
975         context->EFlags |= VIP_MASK;
976         vm86->regs.eflags |= VIP_MASK; /* no exception recursion */
977         if (context->EFlags & VIF_MASK) {
978             /* VIF is set, throw exception */
979             teb->vm86_pending = 0;
980             teb->vm86_ptr = NULL;
981             rec.ExceptionAddress = (LPVOID)context->Eip;
982             EXC_RtlRaiseException( &rec, context );
983             /*
984              * FIXME: EXC_RtlRaiseException has unblocked all signals. 
985              *        If we receive nested SIGUSR2 here, VM86 event 
986              *        handling may lock up!
987              */
988             teb->vm86_ptr = vm86;
989         }
990     }
991     else if (vm86)
992     {
993         /* not in VM86, but possibly setting up for it */
994         if (vm86->regs.eflags & VIP_MASK) return;
995         vm86->regs.eflags |= VIP_MASK;
996         if (((char*)context->Eip >= (char*)vm86_return) &&
997             ((char*)context->Eip <= (char*)vm86_return_end) &&
998             (VM86_TYPE(context->Eax) != VM86_SIGNAL)) {
999             /* exiting from VM86, can't throw */
1000             return;
1001         }
1002         if (vm86->regs.eflags & VIF_MASK) {
1003             /* VIF is set, throw exception */
1004             CONTEXT vcontext;
1005             teb->vm86_pending = 0;
1006             teb->vm86_ptr = NULL;
1007             save_vm86_context( &vcontext, vm86 );
1008             rec.ExceptionAddress = (LPVOID)vcontext.Eip;
1009             EXC_RtlRaiseException( &rec, &vcontext );
1010             /*
1011              * FIXME: EXC_RtlRaiseException has unblocked all signals. 
1012              *        If we receive nested SIGUSR2 here, VM86 event 
1013              *        handling may lock up!
1014              */
1015             teb->vm86_ptr = vm86;
1016             restore_vm86_context( &vcontext, vm86 );
1017         }
1018     }
1019 }
1020
1021
1022 /**********************************************************************
1023  *              usr2_handler
1024  *
1025  * Handler for SIGUSR2.
1026  * We use it to signal that the running __wine_enter_vm86() should
1027  * immediately set VIP_MASK, causing pending events to be handled
1028  * as early as possible.
1029  */
1030 static HANDLER_DEF(usr2_handler)
1031 {
1032     CONTEXT context;
1033
1034     save_context( &context, HANDLER_CONTEXT );
1035     set_vm86_pend( &context );
1036     restore_context( &context, HANDLER_CONTEXT );
1037 }
1038 #endif /* __HAVE_VM86 */
1039
1040
1041 /**********************************************************************
1042  *              segv_handler
1043  *
1044  * Handler for SIGSEGV and related errors.
1045  */
1046 static HANDLER_DEF(segv_handler)
1047 {
1048     CONTEXT context;
1049     void *cr2;
1050
1051     save_context( &context, HANDLER_CONTEXT );
1052 #ifdef FAULT_ADDRESS
1053     cr2 = FAULT_ADDRESS;
1054 #else
1055     cr2 = NULL;
1056 #endif
1057     do_segv( &context, get_trap_code(HANDLER_CONTEXT), cr2, get_error_code(HANDLER_CONTEXT) );
1058     restore_context( &context, HANDLER_CONTEXT );
1059 }
1060
1061
1062 /**********************************************************************
1063  *              trap_handler
1064  *
1065  * Handler for SIGTRAP.
1066  */
1067 static HANDLER_DEF(trap_handler)
1068 {
1069     CONTEXT context;
1070     save_context( &context, HANDLER_CONTEXT );
1071     do_trap( &context, get_trap_code(HANDLER_CONTEXT) );
1072     restore_context( &context, HANDLER_CONTEXT );
1073 }
1074
1075
1076 /**********************************************************************
1077  *              fpe_handler
1078  *
1079  * Handler for SIGFPE.
1080  */
1081 static HANDLER_DEF(fpe_handler)
1082 {
1083     CONTEXT context;
1084     save_fpu( &context, HANDLER_CONTEXT );
1085     save_context( &context, HANDLER_CONTEXT );
1086     do_fpe( &context, get_trap_code(HANDLER_CONTEXT) );
1087     restore_context( &context, HANDLER_CONTEXT );
1088     restore_fpu( &context, HANDLER_CONTEXT );
1089 }
1090
1091
1092 /**********************************************************************
1093  *              int_handler
1094  *
1095  * Handler for SIGINT.
1096  */
1097 static HANDLER_DEF(int_handler)
1098 {
1099     init_handler( HANDLER_CONTEXT );
1100     if (!dispatch_signal(SIGINT))
1101     {
1102         EXCEPTION_RECORD rec;
1103         CONTEXT context;
1104
1105         save_context( &context, HANDLER_CONTEXT );
1106         rec.ExceptionCode    = CONTROL_C_EXIT;
1107         rec.ExceptionFlags   = EXCEPTION_CONTINUABLE;
1108         rec.ExceptionRecord  = NULL;
1109         rec.ExceptionAddress = (LPVOID)context.Eip;
1110         rec.NumberParameters = 0;
1111         EXC_RtlRaiseException( &rec, &context );
1112         restore_context( &context, HANDLER_CONTEXT );
1113     }
1114 }
1115
1116 /**********************************************************************
1117  *              abrt_handler
1118  *
1119  * Handler for SIGABRT.
1120  */
1121 static HANDLER_DEF(abrt_handler)
1122 {
1123     EXCEPTION_RECORD rec;
1124     CONTEXT context;
1125
1126     save_context( &context, HANDLER_CONTEXT );
1127     rec.ExceptionCode    = EXCEPTION_WINE_ASSERTION;
1128     rec.ExceptionFlags   = EH_NONCONTINUABLE;
1129     rec.ExceptionRecord  = NULL;
1130     rec.ExceptionAddress = (LPVOID)context.Eip;
1131     rec.NumberParameters = 0;
1132     EXC_RtlRaiseException( &rec, &context ); /* Should never return.. */
1133     restore_context( &context, HANDLER_CONTEXT );
1134 }
1135
1136
1137 /**********************************************************************
1138  *              term_handler
1139  *
1140  * Handler for SIGTERM.
1141  */
1142 static HANDLER_DEF(term_handler)
1143 {
1144     init_handler( HANDLER_CONTEXT );
1145     SYSDEPS_AbortThread(0);
1146 }
1147
1148
1149 /**********************************************************************
1150  *              usr1_handler
1151  *
1152  * Handler for SIGUSR1, used to signal a thread that it got suspended.
1153  */
1154 static HANDLER_DEF(usr1_handler)
1155 {
1156     LARGE_INTEGER timeout;
1157
1158     init_handler( HANDLER_CONTEXT );
1159     /* wait with 0 timeout, will only return once the thread is no longer suspended */
1160     timeout.QuadPart = 0;
1161     NTDLL_wait_for_multiple_objects( 0, NULL, 0, &timeout );
1162 }
1163
1164
1165 /***********************************************************************
1166  *           set_handler
1167  *
1168  * Set a signal handler
1169  */
1170 static int set_handler( int sig, int have_sigaltstack, void (*func)() )
1171 {
1172     struct sigaction sig_act;
1173
1174 #ifdef linux
1175     if (!have_sigaltstack && NtCurrentTeb()->signal_stack)
1176     {
1177         struct kernel_sigaction sig_act;
1178         sig_act.ksa_handler = func;
1179         sig_act.ksa_flags   = SA_RESTART;
1180         sig_act.ksa_mask    = (1 << (SIGINT-1)) |
1181                               (1 << (SIGUSR2-1));
1182         /* point to the top of the stack */
1183         sig_act.ksa_restorer = (char *)NtCurrentTeb()->signal_stack + SIGNAL_STACK_SIZE;
1184         return wine_sigaction( sig, &sig_act, NULL );
1185     }
1186 #endif  /* linux */
1187     sig_act.sa_handler = func;
1188     sigemptyset( &sig_act.sa_mask );
1189     sigaddset( &sig_act.sa_mask, SIGINT );
1190     sigaddset( &sig_act.sa_mask, SIGUSR2 );
1191
1192 #if defined(linux) || defined(__NetBSD__)
1193     sig_act.sa_flags = SA_RESTART;
1194 #elif defined (__svr4__) || defined(_SCO_DS)
1195     sig_act.sa_flags = SA_SIGINFO | SA_RESTART;
1196 #else
1197     sig_act.sa_flags = 0;
1198 #endif
1199
1200 #ifdef SA_ONSTACK
1201     if (have_sigaltstack) sig_act.sa_flags |= SA_ONSTACK;
1202 #endif
1203     return sigaction( sig, &sig_act, NULL );
1204 }
1205
1206
1207 /***********************************************************************
1208  *           __wine_set_signal_handler   (NTDLL.@)
1209  */
1210 int __wine_set_signal_handler(unsigned int sig, wine_signal_handler wsh)
1211 {
1212     if (sig > sizeof(handlers) / sizeof(handlers[0])) return -1;
1213     if (handlers[sig] != NULL) return -2;
1214     handlers[sig] = wsh;
1215     return 0;
1216 }
1217
1218
1219 /***********************************************************************
1220  *           __wine_set_signal_fs  (NTDLL.@)
1221  */
1222 void __wine_set_signal_fs( unsigned int fs )
1223 {
1224     signal_fs = fs;
1225 }
1226
1227
1228 /**********************************************************************
1229  *              SIGNAL_Init
1230  */
1231 BOOL SIGNAL_Init(void)
1232 {
1233     int have_sigaltstack = 0;
1234
1235 #ifdef HAVE_SIGALTSTACK
1236     struct sigaltstack ss;
1237     if ((ss.ss_sp = NtCurrentTeb()->signal_stack))
1238     {
1239         ss.ss_size  = SIGNAL_STACK_SIZE;
1240         ss.ss_flags = 0;
1241         if (!sigaltstack(&ss, NULL)) have_sigaltstack = 1;
1242 #ifdef linux
1243         /* sigaltstack may fail because the kernel is too old, or
1244            because glibc is brain-dead. In the latter case a
1245            direct system call should succeed. */
1246         else if (!wine_sigaltstack(&ss, NULL)) have_sigaltstack = 1;
1247 #endif  /* linux */
1248     }
1249 #endif  /* HAVE_SIGALTSTACK */
1250
1251     if (set_handler( SIGINT,  have_sigaltstack, (void (*)())int_handler ) == -1) goto error;
1252     if (set_handler( SIGFPE,  have_sigaltstack, (void (*)())fpe_handler ) == -1) goto error;
1253     if (set_handler( SIGSEGV, have_sigaltstack, (void (*)())segv_handler ) == -1) goto error;
1254     if (set_handler( SIGILL,  have_sigaltstack, (void (*)())segv_handler ) == -1) goto error;
1255     if (set_handler( SIGABRT, have_sigaltstack, (void (*)())abrt_handler ) == -1) goto error;
1256     if (set_handler( SIGTERM, have_sigaltstack, (void (*)())term_handler ) == -1) goto error;
1257     if (set_handler( SIGUSR1, have_sigaltstack, (void (*)())usr1_handler ) == -1) goto error;
1258 #ifdef SIGBUS
1259     if (set_handler( SIGBUS,  have_sigaltstack, (void (*)())segv_handler ) == -1) goto error;
1260 #endif
1261 #ifdef SIGTRAP
1262     if (set_handler( SIGTRAP, have_sigaltstack, (void (*)())trap_handler ) == -1) goto error;
1263 #endif
1264
1265 #ifdef __HAVE_VM86
1266     if (set_handler( SIGUSR2, have_sigaltstack, (void (*)())usr2_handler ) == -1) goto error;
1267 #endif
1268
1269     return TRUE;
1270
1271  error:
1272     perror("sigaction");
1273     return FALSE;
1274 }
1275
1276
1277 /**********************************************************************
1278  *              SIGNAL_Block
1279  *
1280  * Block the async signals.
1281  */
1282 void SIGNAL_Block(void)
1283 {
1284     sigset_t block_set;
1285
1286     sigemptyset( &block_set );
1287     sigaddset( &block_set, SIGIO );
1288     sigaddset( &block_set, SIGHUP );
1289     sigaddset( &block_set, SIGUSR1 );
1290     sigaddset( &block_set, SIGUSR2 );
1291     sigprocmask( SIG_BLOCK, &block_set, NULL );
1292 }
1293
1294
1295 /***********************************************************************
1296  *           SIGNAL_Unblock
1297  *
1298  * Unblock signals. Called from EXC_RtlRaiseException.
1299  */
1300 void SIGNAL_Unblock(void)
1301 {
1302     sigset_t all_sigs;
1303
1304     sigfillset( &all_sigs );
1305     sigprocmask( SIG_UNBLOCK, &all_sigs, NULL );
1306 }
1307
1308
1309 /**********************************************************************
1310  *              SIGNAL_Reset
1311  *
1312  * Restore the default handlers.
1313  */
1314 void SIGNAL_Reset(void)
1315 {
1316     signal( SIGINT, SIG_DFL );
1317     signal( SIGFPE, SIG_DFL );
1318     signal( SIGSEGV, SIG_DFL );
1319     signal( SIGILL, SIG_DFL );
1320     signal( SIGABRT, SIG_DFL );
1321     signal( SIGTERM, SIG_DFL );
1322 #ifdef SIGBUS
1323     signal( SIGBUS, SIG_DFL );
1324 #endif
1325 #ifdef SIGTRAP
1326     signal( SIGTRAP, SIG_DFL );
1327 #endif
1328 }
1329
1330
1331 #ifdef __HAVE_VM86
1332 /**********************************************************************
1333  *              __wine_enter_vm86   (NTDLL.@)
1334  *
1335  * Enter vm86 mode with the specified register context.
1336  */
1337 void __wine_enter_vm86( CONTEXT *context )
1338 {
1339     EXCEPTION_RECORD rec;
1340     TEB *teb = NtCurrentTeb();
1341     int res;
1342     struct vm86plus_struct vm86;
1343
1344     memset( &vm86, 0, sizeof(vm86) );
1345     for (;;)
1346     {
1347         restore_vm86_context( context, &vm86 );
1348         /* Linux doesn't preserve pending flag (VIP_MASK) on return,
1349          * so save it on entry, just in case */
1350         teb->vm86_pending |= (context->EFlags & VIP_MASK);
1351         /* Work around race conditions with signal handler
1352          * (avoiding sigprocmask for performance reasons) */
1353         teb->vm86_ptr = &vm86;
1354         vm86.regs.eflags |= teb->vm86_pending;
1355         /* Check for VIF|VIP here, since vm86_enter doesn't */
1356         if ((vm86.regs.eflags & (VIF_MASK|VIP_MASK)) == (VIF_MASK|VIP_MASK)) {
1357             teb->vm86_ptr = NULL;
1358             teb->vm86_pending = 0;
1359             context->EFlags |= VIP_MASK;
1360             rec.ExceptionCode = EXCEPTION_VM86_STI;
1361             rec.ExceptionInformation[0] = 0;
1362             goto cancel_vm86;
1363         }
1364
1365         do
1366         {
1367             res = vm86_enter( &teb->vm86_ptr ); /* uses and clears teb->vm86_ptr */
1368             if (res < 0)
1369             {
1370                 errno = -res;
1371                 return;
1372             }
1373         } while (VM86_TYPE(res) == VM86_SIGNAL);
1374
1375         save_vm86_context( context, &vm86 );
1376         context->EFlags |= teb->vm86_pending;
1377
1378         switch(VM86_TYPE(res))
1379         {
1380         case VM86_UNKNOWN: /* unhandled GP fault - IO-instruction or similar */
1381             do_segv( context, T_PROTFLT, 0, 0 );
1382             continue;
1383         case VM86_TRAP: /* return due to DOS-debugger request */
1384             do_trap( context, VM86_ARG(res) );
1385             continue;
1386         case VM86_INTx: /* int3/int x instruction (ARG = x) */
1387             rec.ExceptionCode = EXCEPTION_VM86_INTx;
1388             break;
1389         case VM86_STI: /* sti/popf/iret instruction enabled virtual interrupts */
1390             teb->vm86_pending = 0;
1391             rec.ExceptionCode = EXCEPTION_VM86_STI;
1392             break;
1393         case VM86_PICRETURN: /* return due to pending PIC request */
1394             rec.ExceptionCode = EXCEPTION_VM86_PICRETURN;
1395             break;
1396         default:
1397             ERR( "unhandled result from vm86 mode %x\n", res );
1398             continue;
1399         }
1400         rec.ExceptionInformation[0] = VM86_ARG(res);
1401 cancel_vm86:
1402         rec.ExceptionFlags          = EXCEPTION_CONTINUABLE;
1403         rec.ExceptionRecord         = NULL;
1404         rec.ExceptionAddress        = (LPVOID)context->Eip;
1405         rec.NumberParameters        = 1;
1406         EXC_RtlRaiseException( &rec, context );
1407     }
1408 }
1409
1410 #else /* __HAVE_VM86 */
1411 /**********************************************************************
1412  *              __wine_enter_vm86   (NTDLL.@)
1413  */
1414 void __wine_enter_vm86( CONTEXT *context )
1415 {
1416     MESSAGE("vm86 mode not supported on this platform\n");
1417 }
1418 #endif /* __HAVE_VM86 */
1419
1420 /**********************************************************************
1421  *              DbgBreakPoint   (NTDLL.@)
1422  */
1423 __ASM_GLOBAL_FUNC( DbgBreakPoint, "int $3; ret");
1424
1425 /**********************************************************************
1426  *              DbgUserBreakPoint   (NTDLL.@)
1427  */
1428 __ASM_GLOBAL_FUNC( DbgUserBreakPoint, "int $3; ret");
1429
1430 #endif  /* __i386__ */