x86/asm/traps: Disable tracing and kprobes in fixup_bad_iret and sync_regs
[pandora-kernel.git] / arch / x86 / kernel / traps.c
1 /*
2  *  Copyright (C) 1991, 1992  Linus Torvalds
3  *  Copyright (C) 2000, 2001, 2002 Andi Kleen, SuSE Labs
4  *
5  *  Pentium III FXSR, SSE support
6  *      Gareth Hughes <gareth@valinux.com>, May 2000
7  */
8
9 /*
10  * Handle hardware traps and faults.
11  */
12 #include <linux/interrupt.h>
13 #include <linux/kallsyms.h>
14 #include <linux/spinlock.h>
15 #include <linux/kprobes.h>
16 #include <linux/uaccess.h>
17 #include <linux/kdebug.h>
18 #include <linux/kgdb.h>
19 #include <linux/kernel.h>
20 #include <linux/module.h>
21 #include <linux/ptrace.h>
22 #include <linux/string.h>
23 #include <linux/delay.h>
24 #include <linux/errno.h>
25 #include <linux/kexec.h>
26 #include <linux/sched.h>
27 #include <linux/timer.h>
28 #include <linux/init.h>
29 #include <linux/bug.h>
30 #include <linux/nmi.h>
31 #include <linux/mm.h>
32 #include <linux/smp.h>
33 #include <linux/io.h>
34
35 #ifdef CONFIG_EISA
36 #include <linux/ioport.h>
37 #include <linux/eisa.h>
38 #endif
39
40 #ifdef CONFIG_MCA
41 #include <linux/mca.h>
42 #endif
43
44 #if defined(CONFIG_EDAC)
45 #include <linux/edac.h>
46 #endif
47
48 #include <asm/kmemcheck.h>
49 #include <asm/stacktrace.h>
50 #include <asm/processor.h>
51 #include <asm/debugreg.h>
52 #include <linux/atomic.h>
53 #include <asm/system.h>
54 #include <asm/traps.h>
55 #include <asm/desc.h>
56 #include <asm/i387.h>
57 #include <asm/mce.h>
58
59 #include <asm/mach_traps.h>
60
61 #ifdef CONFIG_X86_64
62 #include <asm/x86_init.h>
63 #include <asm/pgalloc.h>
64 #include <asm/proto.h>
65 #else
66 #include <asm/processor-flags.h>
67 #include <asm/setup.h>
68
69 asmlinkage int system_call(void);
70
71 /* Do we ignore FPU interrupts ? */
72 char ignore_fpu_irq;
73
74 /*
75  * The IDT has to be page-aligned to simplify the Pentium
76  * F0 0F bug workaround.
77  */
78 gate_desc idt_table[NR_VECTORS] __page_aligned_data = { { { { 0, 0 } } }, };
79 #endif
80
81 DECLARE_BITMAP(used_vectors, NR_VECTORS);
82 EXPORT_SYMBOL_GPL(used_vectors);
83
84 static inline void conditional_sti(struct pt_regs *regs)
85 {
86         if (regs->flags & X86_EFLAGS_IF)
87                 local_irq_enable();
88 }
89
90 static inline void preempt_conditional_sti(struct pt_regs *regs)
91 {
92         inc_preempt_count();
93         if (regs->flags & X86_EFLAGS_IF)
94                 local_irq_enable();
95 }
96
97 static inline void conditional_cli(struct pt_regs *regs)
98 {
99         if (regs->flags & X86_EFLAGS_IF)
100                 local_irq_disable();
101 }
102
103 static inline void preempt_conditional_cli(struct pt_regs *regs)
104 {
105         if (regs->flags & X86_EFLAGS_IF)
106                 local_irq_disable();
107         dec_preempt_count();
108 }
109
110 static void __kprobes
111 do_trap(int trapnr, int signr, char *str, struct pt_regs *regs,
112         long error_code, siginfo_t *info)
113 {
114         struct task_struct *tsk = current;
115
116 #ifdef CONFIG_X86_32
117         if (regs->flags & X86_VM_MASK) {
118                 /*
119                  * traps 0, 1, 3, 4, and 5 should be forwarded to vm86.
120                  * On nmi (interrupt 2), do_trap should not be called.
121                  */
122                 if (trapnr < X86_TRAP_UD)
123                         goto vm86_trap;
124                 goto trap_signal;
125         }
126 #endif
127
128         if (!user_mode(regs))
129                 goto kernel_trap;
130
131 #ifdef CONFIG_X86_32
132 trap_signal:
133 #endif
134         /*
135          * We want error_code and trap_no set for userspace faults and
136          * kernelspace faults which result in die(), but not
137          * kernelspace faults which are fixed up.  die() gives the
138          * process no chance to handle the signal and notice the
139          * kernel fault information, so that won't result in polluting
140          * the information about previously queued, but not yet
141          * delivered, faults.  See also do_general_protection below.
142          */
143         tsk->thread.error_code = error_code;
144         tsk->thread.trap_no = trapnr;
145
146 #ifdef CONFIG_X86_64
147         if (show_unhandled_signals && unhandled_signal(tsk, signr) &&
148             printk_ratelimit()) {
149                 printk(KERN_INFO
150                        "%s[%d] trap %s ip:%lx sp:%lx error:%lx",
151                        tsk->comm, tsk->pid, str,
152                        regs->ip, regs->sp, error_code);
153                 print_vma_addr(" in ", regs->ip);
154                 printk("\n");
155         }
156 #endif
157
158         if (info)
159                 force_sig_info(signr, info, tsk);
160         else
161                 force_sig(signr, tsk);
162         return;
163
164 kernel_trap:
165         if (!fixup_exception(regs)) {
166                 tsk->thread.error_code = error_code;
167                 tsk->thread.trap_no = trapnr;
168                 die(str, regs, error_code);
169         }
170         return;
171
172 #ifdef CONFIG_X86_32
173 vm86_trap:
174         if (handle_vm86_trap((struct kernel_vm86_regs *) regs,
175                                                 error_code, trapnr))
176                 goto trap_signal;
177         return;
178 #endif
179 }
180
181 #define DO_ERROR(trapnr, signr, str, name)                              \
182 dotraplinkage void do_##name(struct pt_regs *regs, long error_code)     \
183 {                                                                       \
184         if (notify_die(DIE_TRAP, str, regs, error_code, trapnr, signr)  \
185                                                         == NOTIFY_STOP) \
186                 return;                                                 \
187         conditional_sti(regs);                                          \
188         do_trap(trapnr, signr, str, regs, error_code, NULL);            \
189 }
190
191 #define DO_ERROR_INFO(trapnr, signr, str, name, sicode, siaddr)         \
192 dotraplinkage void do_##name(struct pt_regs *regs, long error_code)     \
193 {                                                                       \
194         siginfo_t info;                                                 \
195         info.si_signo = signr;                                          \
196         info.si_errno = 0;                                              \
197         info.si_code = sicode;                                          \
198         info.si_addr = (void __user *)siaddr;                           \
199         if (notify_die(DIE_TRAP, str, regs, error_code, trapnr, signr)  \
200                                                         == NOTIFY_STOP) \
201                 return;                                                 \
202         conditional_sti(regs);                                          \
203         do_trap(trapnr, signr, str, regs, error_code, &info);           \
204 }
205
206 DO_ERROR_INFO(X86_TRAP_DE, SIGFPE, "divide error", divide_error, FPE_INTDIV,
207                 regs->ip)
208 DO_ERROR(X86_TRAP_OF, SIGSEGV, "overflow", overflow)
209 DO_ERROR(X86_TRAP_BR, SIGSEGV, "bounds", bounds)
210 DO_ERROR_INFO(X86_TRAP_UD, SIGILL, "invalid opcode", invalid_op, ILL_ILLOPN,
211                 regs->ip)
212 DO_ERROR(X86_TRAP_OLD_MF, SIGFPE, "coprocessor segment overrun",
213                 coprocessor_segment_overrun)
214 DO_ERROR(X86_TRAP_TS, SIGSEGV, "invalid TSS", invalid_TSS)
215 DO_ERROR(X86_TRAP_NP, SIGBUS, "segment not present", segment_not_present)
216 DO_ERROR(X86_TRAP_SS, SIGBUS, "stack segment", stack_segment)
217 DO_ERROR_INFO(X86_TRAP_AC, SIGBUS, "alignment check", alignment_check,
218                 BUS_ADRALN, 0)
219
220 #ifdef CONFIG_X86_64
221 /* Runs on IST stack */
222 dotraplinkage void do_double_fault(struct pt_regs *regs, long error_code)
223 {
224         static const char str[] = "double fault";
225         struct task_struct *tsk = current;
226
227 #ifdef CONFIG_X86_ESPFIX64
228         extern unsigned char native_irq_return_iret[];
229
230         /*
231          * If IRET takes a non-IST fault on the espfix64 stack, then we
232          * end up promoting it to a doublefault.  In that case, modify
233          * the stack to make it look like we just entered the #GP
234          * handler from user space, similar to bad_iret.
235          */
236         if (((long)regs->sp >> PGDIR_SHIFT) == ESPFIX_PGD_ENTRY &&
237                 regs->cs == __KERNEL_CS &&
238                 regs->ip == (unsigned long)native_irq_return_iret)
239         {
240                 struct pt_regs *normal_regs = task_pt_regs(current);
241
242                 /* Fake a #GP(0) from userspace. */
243                 memmove(&normal_regs->ip, (void *)regs->sp, 5*8);
244                 normal_regs->orig_ax = 0;  /* Missing (lost) #GP error code */
245                 regs->ip = (unsigned long)general_protection;
246                 regs->sp = (unsigned long)&normal_regs->orig_ax;
247                 return;
248         }
249 #endif
250
251         /* Return not checked because double check cannot be ignored */
252         notify_die(DIE_TRAP, str, regs, error_code, X86_TRAP_DF, SIGSEGV);
253
254         tsk->thread.error_code = error_code;
255         tsk->thread.trap_no = X86_TRAP_DF;
256
257         /*
258          * This is always a kernel trap and never fixable (and thus must
259          * never return).
260          */
261         for (;;)
262                 die(str, regs, error_code);
263 }
264 #endif
265
266 dotraplinkage void __kprobes
267 do_general_protection(struct pt_regs *regs, long error_code)
268 {
269         struct task_struct *tsk;
270
271         conditional_sti(regs);
272
273 #ifdef CONFIG_X86_32
274         if (regs->flags & X86_VM_MASK)
275                 goto gp_in_vm86;
276 #endif
277
278         tsk = current;
279         if (!user_mode(regs))
280                 goto gp_in_kernel;
281
282         tsk->thread.error_code = error_code;
283         tsk->thread.trap_no = X86_TRAP_GP;
284
285         if (show_unhandled_signals && unhandled_signal(tsk, SIGSEGV) &&
286                         printk_ratelimit()) {
287                 printk(KERN_INFO
288                         "%s[%d] general protection ip:%lx sp:%lx error:%lx",
289                         tsk->comm, task_pid_nr(tsk),
290                         regs->ip, regs->sp, error_code);
291                 print_vma_addr(" in ", regs->ip);
292                 printk("\n");
293         }
294
295         force_sig(SIGSEGV, tsk);
296         return;
297
298 #ifdef CONFIG_X86_32
299 gp_in_vm86:
300         local_irq_enable();
301         handle_vm86_fault((struct kernel_vm86_regs *) regs, error_code);
302         return;
303 #endif
304
305 gp_in_kernel:
306         if (fixup_exception(regs))
307                 return;
308
309         tsk->thread.error_code = error_code;
310         tsk->thread.trap_no = X86_TRAP_GP;
311         if (notify_die(DIE_GPF, "general protection fault", regs, error_code,
312                         X86_TRAP_GP, SIGSEGV) == NOTIFY_STOP)
313                 return;
314         die("general protection fault", regs, error_code);
315 }
316
317 /* May run on IST stack. */
318 dotraplinkage void __kprobes do_int3(struct pt_regs *regs, long error_code)
319 {
320 #ifdef CONFIG_KGDB_LOW_LEVEL_TRAP
321         if (kgdb_ll_trap(DIE_INT3, "int3", regs, error_code, X86_TRAP_BP,
322                                 SIGTRAP) == NOTIFY_STOP)
323                 return;
324 #endif /* CONFIG_KGDB_LOW_LEVEL_TRAP */
325 #ifdef CONFIG_KPROBES
326
327         if (notify_die(DIE_INT3, "int3", regs, error_code, X86_TRAP_BP,
328                         SIGTRAP) == NOTIFY_STOP)
329                 return;
330 #else
331         if (notify_die(DIE_TRAP, "int3", regs, error_code, 3, SIGTRAP)
332                         == NOTIFY_STOP)
333                 return;
334 #endif
335
336         preempt_conditional_sti(regs);
337         do_trap(X86_TRAP_BP, SIGTRAP, "int3", regs, error_code, NULL);
338         preempt_conditional_cli(regs);
339 }
340
341 #ifdef CONFIG_X86_64
342 /*
343  * Help handler running on IST stack to switch back to user stack
344  * for scheduling or signal handling. The actual stack switch is done in
345  * entry.S
346  */
347 asmlinkage notrace __kprobes struct pt_regs *sync_regs(struct pt_regs *eregs)
348 {
349         struct pt_regs *regs = eregs;
350         /* Did already sync */
351         if (eregs == (struct pt_regs *)eregs->sp)
352                 ;
353         /* Exception from user space */
354         else if (user_mode(eregs))
355                 regs = task_pt_regs(current);
356         /*
357          * Exception from kernel and interrupts are enabled. Move to
358          * kernel process stack.
359          */
360         else if (eregs->flags & X86_EFLAGS_IF)
361                 regs = (struct pt_regs *)(eregs->sp -= sizeof(struct pt_regs));
362         if (eregs != regs)
363                 *regs = *eregs;
364         return regs;
365 }
366
367 struct bad_iret_stack {
368         void *error_entry_ret;
369         struct pt_regs regs;
370 };
371
372 asmlinkage notrace __kprobes
373 struct bad_iret_stack *fixup_bad_iret(struct bad_iret_stack *s)
374 {
375         /*
376          * This is called from entry_64.S early in handling a fault
377          * caused by a bad iret to user mode.  To handle the fault
378          * correctly, we want move our stack frame to task_pt_regs
379          * and we want to pretend that the exception came from the
380          * iret target.
381          */
382         struct bad_iret_stack *new_stack =
383                 container_of(task_pt_regs(current),
384                              struct bad_iret_stack, regs);
385
386         /* Copy the IRET target to the new stack. */
387         memmove(&new_stack->regs.ip, (void *)s->regs.sp, 5*8);
388
389         /* Copy the remainder of the stack from the current stack. */
390         memmove(new_stack, s, offsetof(struct bad_iret_stack, regs.ip));
391
392         BUG_ON(!user_mode_vm(&new_stack->regs));
393         return new_stack;
394 }
395 #endif
396
397 /*
398  * Our handling of the processor debug registers is non-trivial.
399  * We do not clear them on entry and exit from the kernel. Therefore
400  * it is possible to get a watchpoint trap here from inside the kernel.
401  * However, the code in ./ptrace.c has ensured that the user can
402  * only set watchpoints on userspace addresses. Therefore the in-kernel
403  * watchpoint trap can only occur in code which is reading/writing
404  * from user space. Such code must not hold kernel locks (since it
405  * can equally take a page fault), therefore it is safe to call
406  * force_sig_info even though that claims and releases locks.
407  *
408  * Code in ./signal.c ensures that the debug control register
409  * is restored before we deliver any signal, and therefore that
410  * user code runs with the correct debug control register even though
411  * we clear it here.
412  *
413  * Being careful here means that we don't have to be as careful in a
414  * lot of more complicated places (task switching can be a bit lazy
415  * about restoring all the debug state, and ptrace doesn't have to
416  * find every occurrence of the TF bit that could be saved away even
417  * by user code)
418  *
419  * May run on IST stack.
420  */
421 dotraplinkage void __kprobes do_debug(struct pt_regs *regs, long error_code)
422 {
423         struct task_struct *tsk = current;
424         int user_icebp = 0;
425         unsigned long dr6;
426         int si_code;
427
428         get_debugreg(dr6, 6);
429
430         /* Filter out all the reserved bits which are preset to 1 */
431         dr6 &= ~DR6_RESERVED;
432
433         /*
434          * If dr6 has no reason to give us about the origin of this trap,
435          * then it's very likely the result of an icebp/int01 trap.
436          * User wants a sigtrap for that.
437          */
438         if (!dr6 && user_mode(regs))
439                 user_icebp = 1;
440
441         /* Catch kmemcheck conditions first of all! */
442         if ((dr6 & DR_STEP) && kmemcheck_trap(regs))
443                 return;
444
445         /* DR6 may or may not be cleared by the CPU */
446         set_debugreg(0, 6);
447
448         /*
449          * The processor cleared BTF, so don't mark that we need it set.
450          */
451         clear_tsk_thread_flag(tsk, TIF_BLOCKSTEP);
452
453         /* Store the virtualized DR6 value */
454         tsk->thread.debugreg6 = dr6;
455
456         if (notify_die(DIE_DEBUG, "debug", regs, PTR_ERR(&dr6), error_code,
457                                                         SIGTRAP) == NOTIFY_STOP)
458                 return;
459
460         /* It's safe to allow irq's after DR6 has been saved */
461         preempt_conditional_sti(regs);
462
463         if (regs->flags & X86_VM_MASK) {
464                 handle_vm86_trap((struct kernel_vm86_regs *) regs, error_code,
465                                         X86_TRAP_DB);
466                 preempt_conditional_cli(regs);
467                 return;
468         }
469
470         /*
471          * Single-stepping through system calls: ignore any exceptions in
472          * kernel space, but re-enable TF when returning to user mode.
473          *
474          * We already checked v86 mode above, so we can check for kernel mode
475          * by just checking the CPL of CS.
476          */
477         if ((dr6 & DR_STEP) && !user_mode(regs)) {
478                 tsk->thread.debugreg6 &= ~DR_STEP;
479                 set_tsk_thread_flag(tsk, TIF_SINGLESTEP);
480                 regs->flags &= ~X86_EFLAGS_TF;
481         }
482         si_code = get_si_code(tsk->thread.debugreg6);
483         if (tsk->thread.debugreg6 & (DR_STEP | DR_TRAP_BITS) || user_icebp)
484                 send_sigtrap(tsk, regs, error_code, si_code);
485         preempt_conditional_cli(regs);
486
487         return;
488 }
489
490 /*
491  * Note that we play around with the 'TS' bit in an attempt to get
492  * the correct behaviour even in the presence of the asynchronous
493  * IRQ13 behaviour
494  */
495 void math_error(struct pt_regs *regs, int error_code, int trapnr)
496 {
497         struct task_struct *task = current;
498         siginfo_t info;
499         unsigned short err;
500         char *str = (trapnr == X86_TRAP_MF) ? "fpu exception" :
501                                                 "simd exception";
502
503         if (notify_die(DIE_TRAP, str, regs, error_code, trapnr, SIGFPE) == NOTIFY_STOP)
504                 return;
505         conditional_sti(regs);
506
507         if (!user_mode_vm(regs))
508         {
509                 if (!fixup_exception(regs)) {
510                         task->thread.error_code = error_code;
511                         task->thread.trap_no = trapnr;
512                         die(str, regs, error_code);
513                 }
514                 return;
515         }
516
517         /*
518          * Save the info for the exception handler and clear the error.
519          */
520         save_init_fpu(task);
521         task->thread.trap_no = trapnr;
522         task->thread.error_code = error_code;
523         info.si_signo = SIGFPE;
524         info.si_errno = 0;
525         info.si_addr = (void __user *)regs->ip;
526         if (trapnr == X86_TRAP_MF) {
527                 unsigned short cwd, swd;
528                 /*
529                  * (~cwd & swd) will mask out exceptions that are not set to unmasked
530                  * status.  0x3f is the exception bits in these regs, 0x200 is the
531                  * C1 reg you need in case of a stack fault, 0x040 is the stack
532                  * fault bit.  We should only be taking one exception at a time,
533                  * so if this combination doesn't produce any single exception,
534                  * then we have a bad program that isn't synchronizing its FPU usage
535                  * and it will suffer the consequences since we won't be able to
536                  * fully reproduce the context of the exception
537                  */
538                 cwd = get_fpu_cwd(task);
539                 swd = get_fpu_swd(task);
540
541                 err = swd & ~cwd;
542         } else {
543                 /*
544                  * The SIMD FPU exceptions are handled a little differently, as there
545                  * is only a single status/control register.  Thus, to determine which
546                  * unmasked exception was caught we must mask the exception mask bits
547                  * at 0x1f80, and then use these to mask the exception bits at 0x3f.
548                  */
549                 unsigned short mxcsr = get_fpu_mxcsr(task);
550                 err = ~(mxcsr >> 7) & mxcsr;
551         }
552
553         if (err & 0x001) {      /* Invalid op */
554                 /*
555                  * swd & 0x240 == 0x040: Stack Underflow
556                  * swd & 0x240 == 0x240: Stack Overflow
557                  * User must clear the SF bit (0x40) if set
558                  */
559                 info.si_code = FPE_FLTINV;
560         } else if (err & 0x004) { /* Divide by Zero */
561                 info.si_code = FPE_FLTDIV;
562         } else if (err & 0x008) { /* Overflow */
563                 info.si_code = FPE_FLTOVF;
564         } else if (err & 0x012) { /* Denormal, Underflow */
565                 info.si_code = FPE_FLTUND;
566         } else if (err & 0x020) { /* Precision */
567                 info.si_code = FPE_FLTRES;
568         } else {
569                 /*
570                  * If we're using IRQ 13, or supposedly even some trap
571                  * X86_TRAP_MF implementations, it's possible
572                  * we get a spurious trap, which is not an error.
573                  */
574                 return;
575         }
576         force_sig_info(SIGFPE, &info, task);
577 }
578
579 dotraplinkage void do_coprocessor_error(struct pt_regs *regs, long error_code)
580 {
581 #ifdef CONFIG_X86_32
582         ignore_fpu_irq = 1;
583 #endif
584
585         math_error(regs, error_code, X86_TRAP_MF);
586 }
587
588 dotraplinkage void
589 do_simd_coprocessor_error(struct pt_regs *regs, long error_code)
590 {
591         math_error(regs, error_code, X86_TRAP_XF);
592 }
593
594 dotraplinkage void
595 do_spurious_interrupt_bug(struct pt_regs *regs, long error_code)
596 {
597         conditional_sti(regs);
598 #if 0
599         /* No need to warn about this any longer. */
600         printk(KERN_INFO "Ignoring P6 Local APIC Spurious Interrupt Bug...\n");
601 #endif
602 }
603
604 asmlinkage void __attribute__((weak)) smp_thermal_interrupt(void)
605 {
606 }
607
608 asmlinkage void __attribute__((weak)) smp_threshold_interrupt(void)
609 {
610 }
611
612 /*
613  * This gets called with the process already owning the
614  * FPU state, and with CR0.TS cleared. It just needs to
615  * restore the FPU register state.
616  */
617 void __math_state_restore(struct task_struct *tsk)
618 {
619         /* We need a safe address that is cheap to find and that is already
620            in L1. We've just brought in "tsk->thread.has_fpu", so use that */
621 #define safe_address (tsk->thread.has_fpu)
622
623         /* AMD K7/K8 CPUs don't save/restore FDP/FIP/FOP unless an exception
624            is pending.  Clear the x87 state here by setting it to fixed
625            values. safe_address is a random variable that should be in L1 */
626         if (unlikely(static_cpu_has(X86_FEATURE_FXSAVE_LEAK))) {
627                 asm volatile(
628                         "fnclex\n\t"
629                         "emms\n\t"
630                         "fildl %P[addr]"        /* set F?P to defined value */
631                         : : [addr] "m" (safe_address));
632         }
633
634         /*
635          * Paranoid restore. send a SIGSEGV if we fail to restore the state.
636          */
637         if (unlikely(restore_fpu_checking(tsk))) {
638                 __thread_fpu_end(tsk);
639                 force_sig(SIGSEGV, tsk);
640                 return;
641         }
642 }
643
644 /*
645  * 'math_state_restore()' saves the current math information in the
646  * old math state array, and gets the new ones from the current task
647  *
648  * Careful.. There are problems with IBM-designed IRQ13 behaviour.
649  * Don't touch unless you *really* know how it works.
650  *
651  * Must be called with kernel preemption disabled (eg with local
652  * local interrupts as in the case of do_device_not_available).
653  */
654 void math_state_restore(void)
655 {
656         struct task_struct *tsk = current;
657
658         if (!tsk_used_math(tsk)) {
659                 local_irq_enable();
660                 /*
661                  * does a slab alloc which can sleep
662                  */
663                 if (init_fpu(tsk)) {
664                         /*
665                          * ran out of memory!
666                          */
667                         do_group_exit(SIGKILL);
668                         return;
669                 }
670                 local_irq_disable();
671         }
672
673         __thread_fpu_begin(tsk);
674         __math_state_restore(tsk);
675
676         tsk->fpu_counter++;
677 }
678 EXPORT_SYMBOL_GPL(math_state_restore);
679
680 dotraplinkage void __kprobes
681 do_device_not_available(struct pt_regs *regs, long error_code)
682 {
683 #ifdef CONFIG_MATH_EMULATION
684         if (read_cr0() & X86_CR0_EM) {
685                 struct math_emu_info info = { };
686
687                 conditional_sti(regs);
688
689                 info.regs = regs;
690                 math_emulate(&info);
691                 return;
692         }
693 #endif
694         math_state_restore(); /* interrupts still off */
695 #ifdef CONFIG_X86_32
696         conditional_sti(regs);
697 #endif
698 }
699
700 #ifdef CONFIG_X86_32
701 dotraplinkage void do_iret_error(struct pt_regs *regs, long error_code)
702 {
703         siginfo_t info;
704         local_irq_enable();
705
706         info.si_signo = SIGILL;
707         info.si_errno = 0;
708         info.si_code = ILL_BADSTK;
709         info.si_addr = NULL;
710         if (notify_die(DIE_TRAP, "iret exception", regs, error_code,
711                         X86_TRAP_IRET, SIGILL) == NOTIFY_STOP)
712                 return;
713         do_trap(X86_TRAP_IRET, SIGILL, "iret exception", regs, error_code,
714                 &info);
715 }
716 #endif
717
718 /* Set of traps needed for early debugging. */
719 void __init early_trap_init(void)
720 {
721         set_intr_gate_ist(X86_TRAP_DB, &debug, DEBUG_STACK);
722         /* int3 can be called from all */
723         set_system_intr_gate_ist(X86_TRAP_BP, &int3, DEBUG_STACK);
724         set_intr_gate(X86_TRAP_PF, &page_fault);
725         load_idt(&idt_descr);
726 }
727
728 void __init trap_init(void)
729 {
730         int i;
731
732 #ifdef CONFIG_EISA
733         void __iomem *p = early_ioremap(0x0FFFD9, 4);
734
735         if (readl(p) == 'E' + ('I'<<8) + ('S'<<16) + ('A'<<24))
736                 EISA_bus = 1;
737         early_iounmap(p, 4);
738 #endif
739
740         set_intr_gate(X86_TRAP_DE, &divide_error);
741         set_intr_gate_ist(X86_TRAP_NMI, &nmi, NMI_STACK);
742         /* int4 can be called from all */
743         set_system_intr_gate(X86_TRAP_OF, &overflow);
744         set_intr_gate(X86_TRAP_BR, &bounds);
745         set_intr_gate(X86_TRAP_UD, &invalid_op);
746         set_intr_gate(X86_TRAP_NM, &device_not_available);
747 #ifdef CONFIG_X86_32
748         set_task_gate(X86_TRAP_DF, GDT_ENTRY_DOUBLEFAULT_TSS);
749 #else
750         set_intr_gate_ist(X86_TRAP_DF, &double_fault, DOUBLEFAULT_STACK);
751 #endif
752         set_intr_gate(X86_TRAP_OLD_MF, &coprocessor_segment_overrun);
753         set_intr_gate(X86_TRAP_TS, &invalid_TSS);
754         set_intr_gate(X86_TRAP_NP, &segment_not_present);
755         set_intr_gate(X86_TRAP_SS, stack_segment);
756         set_intr_gate(X86_TRAP_GP, &general_protection);
757         set_intr_gate(X86_TRAP_SPURIOUS, &spurious_interrupt_bug);
758         set_intr_gate(X86_TRAP_MF, &coprocessor_error);
759         set_intr_gate(X86_TRAP_AC, &alignment_check);
760 #ifdef CONFIG_X86_MCE
761         set_intr_gate_ist(X86_TRAP_MC, &machine_check, MCE_STACK);
762 #endif
763         set_intr_gate(X86_TRAP_XF, &simd_coprocessor_error);
764
765         /* Reserve all the builtin and the syscall vector: */
766         for (i = 0; i < FIRST_EXTERNAL_VECTOR; i++)
767                 set_bit(i, used_vectors);
768
769 #ifdef CONFIG_IA32_EMULATION
770         set_system_intr_gate(IA32_SYSCALL_VECTOR, ia32_syscall);
771         set_bit(IA32_SYSCALL_VECTOR, used_vectors);
772 #endif
773
774 #ifdef CONFIG_X86_32
775         set_system_trap_gate(SYSCALL_VECTOR, &system_call);
776         set_bit(SYSCALL_VECTOR, used_vectors);
777 #endif
778
779         /*
780          * Should be a barrier for any external CPU state:
781          */
782         cpu_init();
783
784         x86_init.irqs.trap_init();
785 }