traps_64.c 17.0 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12
/*
 *  Copyright (C) 1991, 1992  Linus Torvalds
 *  Copyright (C) 2000, 2001, 2002 Andi Kleen, SuSE Labs
 *
 *  Pentium III FXSR, SSE support
 *	Gareth Hughes <gareth@valinux.com>, May 2000
 */

/*
 * 'Traps.c' handles hardware traps and faults after we have saved some
 * state in 'entry.S'.
 */
13 14 15 16 17 18 19 20
#include <linux/moduleparam.h>
#include <linux/interrupt.h>
#include <linux/kallsyms.h>
#include <linux/spinlock.h>
#include <linux/kprobes.h>
#include <linux/uaccess.h>
#include <linux/utsname.h>
#include <linux/kdebug.h>
L
Linus Torvalds 已提交
21
#include <linux/kernel.h>
22 23
#include <linux/module.h>
#include <linux/ptrace.h>
L
Linus Torvalds 已提交
24
#include <linux/string.h>
25 26
#include <linux/unwind.h>
#include <linux/delay.h>
L
Linus Torvalds 已提交
27
#include <linux/errno.h>
28 29
#include <linux/kexec.h>
#include <linux/sched.h>
L
Linus Torvalds 已提交
30 31
#include <linux/timer.h>
#include <linux/init.h>
32
#include <linux/bug.h>
33 34
#include <linux/nmi.h>
#include <linux/mm.h>
35 36
#include <linux/smp.h>
#include <linux/io.h>
37

D
Dave Jiang 已提交
38 39 40 41
#if defined(CONFIG_EDAC)
#include <linux/edac.h>
#endif

42 43
#include <asm/stacktrace.h>
#include <asm/processor.h>
L
Linus Torvalds 已提交
44
#include <asm/debugreg.h>
45 46 47
#include <asm/atomic.h>
#include <asm/system.h>
#include <asm/unwind.h>
L
Linus Torvalds 已提交
48 49 50 51
#include <asm/desc.h>
#include <asm/i387.h>
#include <asm/pgalloc.h>
#include <asm/proto.h>
52
#include <asm/pda.h>
53
#include <asm/traps.h>
54 55

#include <mach_traps.h>
L
Linus Torvalds 已提交
56

57
static int ignore_nmis;
58

L
Linus Torvalds 已提交
59 60
static inline void conditional_sti(struct pt_regs *regs)
{
61
	if (regs->flags & X86_EFLAGS_IF)
L
Linus Torvalds 已提交
62 63 64
		local_irq_enable();
}

65 66
static inline void preempt_conditional_sti(struct pt_regs *regs)
{
67
	inc_preempt_count();
68
	if (regs->flags & X86_EFLAGS_IF)
69 70 71 72 73
		local_irq_enable();
}

static inline void preempt_conditional_cli(struct pt_regs *regs)
{
74
	if (regs->flags & X86_EFLAGS_IF)
75
		local_irq_disable();
76 77
	/* Make sure to not schedule here because we could be running
	   on an exception stack. */
78
	dec_preempt_count();
79 80
}

81 82 83
static void __kprobes
do_trap(int trapnr, int signr, char *str, struct pt_regs *regs,
	long error_code, siginfo_t *info)
L
Linus Torvalds 已提交
84
{
85 86
	struct task_struct *tsk = current;

87 88
	if (!user_mode(regs))
		goto kernel_trap;
L
Linus Torvalds 已提交
89

90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109
	/*
	 * We want error_code and trap_no set for userspace faults and
	 * kernelspace faults which result in die(), but not
	 * kernelspace faults which are fixed up.  die() gives the
	 * process no chance to handle the signal and notice the
	 * kernel fault information, so that won't result in polluting
	 * the information about previously queued, but not yet
	 * delivered, faults.  See also do_general_protection below.
	 */
	tsk->thread.error_code = error_code;
	tsk->thread.trap_no = trapnr;

	if (show_unhandled_signals && unhandled_signal(tsk, signr) &&
	    printk_ratelimit()) {
		printk(KERN_INFO
		       "%s[%d] trap %s ip:%lx sp:%lx error:%lx",
		       tsk->comm, tsk->pid, str,
		       regs->ip, regs->sp, error_code);
		print_vma_addr(" in ", regs->ip);
		printk("\n");
L
Linus Torvalds 已提交
110 111
	}

112 113 114 115 116
	if (info)
		force_sig_info(signr, info, tsk);
	else
		force_sig(signr, tsk);
	return;
L
Linus Torvalds 已提交
117

118
kernel_trap:
119 120 121 122
	if (!fixup_exception(regs)) {
		tsk->thread.error_code = error_code;
		tsk->thread.trap_no = trapnr;
		die(str, regs, error_code);
L
Linus Torvalds 已提交
123
	}
124
	return;
L
Linus Torvalds 已提交
125 126
}

127
#define DO_ERROR(trapnr, signr, str, name)				\
128
dotraplinkage void do_##name(struct pt_regs *regs, long error_code)	\
129 130 131 132
{									\
	if (notify_die(DIE_TRAP, str, regs, error_code, trapnr, signr)	\
							== NOTIFY_STOP)	\
		return;							\
133
	conditional_sti(regs);						\
134
	do_trap(trapnr, signr, str, regs, error_code, NULL);		\
L
Linus Torvalds 已提交
135 136
}

137
#define DO_ERROR_INFO(trapnr, signr, str, name, sicode, siaddr)		\
138
dotraplinkage void do_##name(struct pt_regs *regs, long error_code)	\
139 140 141 142 143 144 145 146 147
{									\
	siginfo_t info;							\
	info.si_signo = signr;						\
	info.si_errno = 0;						\
	info.si_code = sicode;						\
	info.si_addr = (void __user *)siaddr;				\
	if (notify_die(DIE_TRAP, str, regs, error_code, trapnr, signr)	\
							== NOTIFY_STOP)	\
		return;							\
148
	conditional_sti(regs);						\
149
	do_trap(trapnr, signr, str, regs, error_code, &info);		\
L
Linus Torvalds 已提交
150 151
}

152 153 154 155 156
DO_ERROR_INFO(0, SIGFPE, "divide error", divide_error, FPE_INTDIV, regs->ip)
DO_ERROR(4, SIGSEGV, "overflow", overflow)
DO_ERROR(5, SIGSEGV, "bounds", bounds)
DO_ERROR_INFO(6, SIGILL, "invalid opcode", invalid_op, ILL_ILLOPN, regs->ip)
DO_ERROR(9, SIGFPE, "coprocessor segment overrun", coprocessor_segment_overrun)
L
Linus Torvalds 已提交
157
DO_ERROR(10, SIGSEGV, "invalid TSS", invalid_TSS)
158
DO_ERROR(11, SIGBUS, "segment not present", segment_not_present)
L
Linus Torvalds 已提交
159
DO_ERROR_INFO(17, SIGBUS, "alignment check", alignment_check, BUS_ADRALN, 0)
160 161

/* Runs on IST stack */
162
dotraplinkage void do_stack_segment(struct pt_regs *regs, long error_code)
163 164 165 166 167 168 169 170
{
	if (notify_die(DIE_TRAP, "stack segment", regs, error_code,
			12, SIGBUS) == NOTIFY_STOP)
		return;
	preempt_conditional_sti(regs);
	do_trap(12, SIGBUS, "stack segment", regs, error_code, NULL);
	preempt_conditional_cli(regs);
}
171

172
dotraplinkage void do_double_fault(struct pt_regs *regs, long error_code)
173 174 175 176 177 178 179 180 181 182 183 184 185 186 187
{
	static const char str[] = "double fault";
	struct task_struct *tsk = current;

	/* Return not checked because double check cannot be ignored */
	notify_die(DIE_TRAP, str, regs, error_code, 8, SIGSEGV);

	tsk->thread.error_code = error_code;
	tsk->thread.trap_no = 8;

	/* This is always a kernel trap and never fixable (and thus must
	   never return). */
	for (;;)
		die(str, regs, error_code);
}
L
Linus Torvalds 已提交
188

189
dotraplinkage void __kprobes
190
do_general_protection(struct pt_regs *regs, long error_code)
L
Linus Torvalds 已提交
191
{
192
	struct task_struct *tsk;
193

L
Linus Torvalds 已提交
194 195
	conditional_sti(regs);

196 197 198
	tsk = current;
	if (!user_mode(regs))
		goto gp_in_kernel;
L
Linus Torvalds 已提交
199

200 201 202 203 204 205 206 207 208 209 210 211 212 213 214
	tsk->thread.error_code = error_code;
	tsk->thread.trap_no = 13;

	if (show_unhandled_signals && unhandled_signal(tsk, SIGSEGV) &&
			printk_ratelimit()) {
		printk(KERN_INFO
			"%s[%d] general protection ip:%lx sp:%lx error:%lx",
			tsk->comm, tsk->pid,
			regs->ip, regs->sp, error_code);
		print_vma_addr(" in ", regs->ip);
		printk("\n");
	}

	force_sig(SIGSEGV, tsk);
	return;
L
Linus Torvalds 已提交
215

216
gp_in_kernel:
217 218
	if (fixup_exception(regs))
		return;
219

220 221 222 223 224 225
	tsk->thread.error_code = error_code;
	tsk->thread.trap_no = 13;
	if (notify_die(DIE_GPF, "general protection fault", regs,
				error_code, 13, SIGSEGV) == NOTIFY_STOP)
		return;
	die("general protection fault", regs, error_code);
L
Linus Torvalds 已提交
226 227
}

228
static notrace __kprobes void
229
mem_parity_error(unsigned char reason, struct pt_regs *regs)
L
Linus Torvalds 已提交
230
{
231 232
	printk(KERN_EMERG "Uhhuh. NMI received for unknown reason %02x.\n",
		reason);
233
	printk(KERN_EMERG "You have some hardware problem, likely on the PCI bus.\n");
234

D
Dave Jiang 已提交
235
#if defined(CONFIG_EDAC)
236
	if (edac_handler_set()) {
D
Dave Jiang 已提交
237 238 239 240 241
		edac_atomic_assert_error();
		return;
	}
#endif

242
	if (panic_on_unrecovered_nmi)
243 244 245
		panic("NMI: Not continuing");

	printk(KERN_EMERG "Dazed and confused, but trying to continue\n");
L
Linus Torvalds 已提交
246 247 248 249 250 251

	/* Clear and disable the memory parity error line. */
	reason = (reason & 0xf) | 4;
	outb(reason, 0x61);
}

252
static notrace __kprobes void
253
io_check_error(unsigned char reason, struct pt_regs *regs)
L
Linus Torvalds 已提交
254 255 256 257 258 259 260 261 262 263 264 265
{
	printk("NMI: IOCK error (debug interrupt?)\n");
	show_registers(regs);

	/* Re-enable the IOCK line, wait for a few seconds */
	reason = (reason & 0xf) | 8;
	outb(reason, 0x61);
	mdelay(2000);
	reason &= ~8;
	outb(reason, 0x61);
}

266
static notrace __kprobes void
267
unknown_nmi_error(unsigned char reason, struct pt_regs *regs)
268
{
269 270
	if (notify_die(DIE_NMIUNKNOWN, "nmi", regs, reason, 2, SIGINT) ==
			NOTIFY_STOP)
J
Jason Wessel 已提交
271
		return;
272 273 274
	printk(KERN_EMERG "Uhhuh. NMI received for unknown reason %02x.\n",
		reason);
	printk(KERN_EMERG "Do you have a strange power saving mode enabled?\n");
275 276

	if (panic_on_unrecovered_nmi)
277
		panic("NMI: Not continuing");
278

279
	printk(KERN_EMERG "Dazed and confused, but trying to continue\n");
L
Linus Torvalds 已提交
280 281
}

282 283
/* Runs on IST stack. This code must keep interrupts off all the time.
   Nested NMIs are prevented by the CPU. */
284
asmlinkage notrace __kprobes void default_do_nmi(struct pt_regs *regs)
L
Linus Torvalds 已提交
285 286
{
	unsigned char reason = 0;
A
Ashok Raj 已提交
287 288 289
	int cpu;

	cpu = smp_processor_id();
L
Linus Torvalds 已提交
290

291
	/* Only the BSP gets external NMIs from the system. */
A
Ashok Raj 已提交
292
	if (!cpu)
L
Linus Torvalds 已提交
293 294 295
		reason = get_nmi_reason();

	if (!(reason & 0xc0)) {
296
		if (notify_die(DIE_NMI_IPI, "nmi_ipi", regs, reason, 2, SIGINT)
L
Linus Torvalds 已提交
297 298 299 300 301 302
								== NOTIFY_STOP)
			return;
		/*
		 * Ok, so this is none of the documented NMI sources,
		 * so it must be the NMI watchdog.
		 */
303
		if (nmi_watchdog_tick(regs, reason))
L
Linus Torvalds 已提交
304
			return;
305
		if (!do_nmi_callback(regs, cpu))
306 307
			unknown_nmi_error(reason, regs);

L
Linus Torvalds 已提交
308 309
		return;
	}
310
	if (notify_die(DIE_NMI, "nmi", regs, reason, 2, SIGINT) == NOTIFY_STOP)
311
		return;
L
Linus Torvalds 已提交
312 313 314 315 316 317 318 319

	/* AK: following checks seem to be broken on modern chipsets. FIXME */
	if (reason & 0x80)
		mem_parity_error(reason, regs);
	if (reason & 0x40)
		io_check_error(reason, regs);
}

320
dotraplinkage notrace __kprobes void
321 322 323
do_nmi(struct pt_regs *regs, long error_code)
{
	nmi_enter();
324

325
	add_pda(__nmi_count, 1);
326

327 328
	if (!ignore_nmis)
		default_do_nmi(regs);
329

330 331 332 333 334 335 336 337 338 339 340 341 342 343 344
	nmi_exit();
}

void stop_nmi(void)
{
	acpi_nmi_disable();
	ignore_nmis++;
}

void restart_nmi(void)
{
	ignore_nmis--;
	acpi_nmi_enable();
}

345
/* runs on IST stack. */
346
dotraplinkage void __kprobes do_int3(struct pt_regs *regs, long error_code)
L
Linus Torvalds 已提交
347
{
348 349
	if (notify_die(DIE_INT3, "int3", regs, error_code, 3, SIGTRAP)
			== NOTIFY_STOP)
L
Linus Torvalds 已提交
350
		return;
351

352
	preempt_conditional_sti(regs);
L
Linus Torvalds 已提交
353
	do_trap(3, SIGTRAP, "int3", regs, error_code, NULL);
354
	preempt_conditional_cli(regs);
L
Linus Torvalds 已提交
355 356
}

357 358 359
/* Help handler running on IST stack to switch back to user stack
   for scheduling or signal handling. The actual stack switch is done in
   entry.S */
360
asmlinkage __kprobes struct pt_regs *sync_regs(struct pt_regs *eregs)
361 362 363
{
	struct pt_regs *regs = eregs;
	/* Did already sync */
364
	if (eregs == (struct pt_regs *)eregs->sp)
365 366
		;
	/* Exception from user space */
367
	else if (user_mode(eregs))
A
Al Viro 已提交
368
		regs = task_pt_regs(current);
369
	/* Exception from kernel and interrupts are enabled. Move to
370
	   kernel process stack. */
371 372
	else if (eregs->flags & X86_EFLAGS_IF)
		regs = (struct pt_regs *)(eregs->sp -= sizeof(struct pt_regs));
373 374 375 376 377
	if (eregs != regs)
		*regs = *eregs;
	return regs;
}

L
Linus Torvalds 已提交
378
/* runs on IST stack. */
379
dotraplinkage void __kprobes do_debug(struct pt_regs *regs, long error_code)
L
Linus Torvalds 已提交
380 381
{
	struct task_struct *tsk = current;
382
	unsigned long condition;
383
	int si_code;
L
Linus Torvalds 已提交
384

385
	get_debugreg(condition, 6);
L
Linus Torvalds 已提交
386

387 388 389 390 391 392
	/*
	 * The processor cleared BTF, so don't mark that we need it set.
	 */
	clear_tsk_thread_flag(tsk, TIF_DEBUGCTLMSR);
	tsk->thread.debugctlmsr = 0;

L
Linus Torvalds 已提交
393
	if (notify_die(DIE_DEBUG, "debug", regs, condition, error_code,
394
						SIGTRAP) == NOTIFY_STOP)
395
		return;
396

397
	/* It's safe to allow irq's after DR6 has been saved */
398
	preempt_conditional_sti(regs);
L
Linus Torvalds 已提交
399 400 401

	/* Mask out spurious debug traps due to lazy DR7 setting */
	if (condition & (DR_TRAP0|DR_TRAP1|DR_TRAP2|DR_TRAP3)) {
402
		if (!tsk->thread.debugreg7)
L
Linus Torvalds 已提交
403 404 405
			goto clear_dr7;
	}

406
	/* Save debug status register where ptrace can see it */
L
Linus Torvalds 已提交
407 408
	tsk->thread.debugreg6 = condition;

R
Roland McGrath 已提交
409 410 411 412
	/*
	 * Single-stepping through TF: make sure we ignore any events in
	 * kernel space (but re-enable TF when returning to user mode).
	 */
413
	if (condition & DR_STEP) {
414 415
		if (!user_mode(regs))
			goto clear_TF_reenable;
L
Linus Torvalds 已提交
416 417
	}

418
	si_code = get_si_code(condition);
L
Linus Torvalds 已提交
419
	/* Ok, finally something we can handle */
420
	send_sigtrap(tsk, regs, error_code, si_code);
L
Linus Torvalds 已提交
421

422 423 424 425
	/*
	 * Disable additional traps. They'll be re-enabled when
	 * the signal is delivered.
	 */
L
Linus Torvalds 已提交
426
clear_dr7:
427
	set_debugreg(0, 7);
428
	preempt_conditional_cli(regs);
429
	return;
L
Linus Torvalds 已提交
430 431 432

clear_TF_reenable:
	set_tsk_thread_flag(tsk, TIF_SINGLESTEP);
433
	regs->flags &= ~X86_EFLAGS_TF;
434
	preempt_conditional_cli(regs);
435
	return;
L
Linus Torvalds 已提交
436 437
}

438
static int kernel_math_error(struct pt_regs *regs, const char *str, int trapnr)
L
Linus Torvalds 已提交
439
{
440
	if (fixup_exception(regs))
L
Linus Torvalds 已提交
441
		return 1;
442

443
	notify_die(DIE_GPF, str, regs, 0, trapnr, SIGFPE);
444
	/* Illegal floating point operation in the kernel */
445
	current->thread.trap_no = trapnr;
L
Linus Torvalds 已提交
446 447 448 449 450 451 452 453 454
	die(str, regs, 0);
	return 0;
}

/*
 * Note that we play around with the 'TS' bit in an attempt to get
 * the correct behaviour even in the presence of the asynchronous
 * IRQ13 behaviour
 */
455
void math_error(void __user *ip)
L
Linus Torvalds 已提交
456
{
457
	struct task_struct *task;
L
Linus Torvalds 已提交
458 459 460 461 462 463 464 465 466 467 468 469 470
	siginfo_t info;
	unsigned short cwd, swd;

	/*
	 * Save the info for the exception handler and clear the error.
	 */
	task = current;
	save_init_fpu(task);
	task->thread.trap_no = 16;
	task->thread.error_code = 0;
	info.si_signo = SIGFPE;
	info.si_errno = 0;
	info.si_code = __SI_FAULT;
471
	info.si_addr = ip;
L
Linus Torvalds 已提交
472 473 474 475 476 477 478 479 480 481 482 483
	/*
	 * (~cwd & swd) will mask out exceptions that are not set to unmasked
	 * status.  0x3f is the exception bits in these regs, 0x200 is the
	 * C1 reg you need in case of a stack fault, 0x040 is the stack
	 * fault bit.  We should only be taking one exception at a time,
	 * so if this combination doesn't produce any single exception,
	 * then we have a bad program that isn't synchronizing its FPU usage
	 * and it will suffer the consequences since we won't be able to
	 * fully reproduce the context of the exception
	 */
	cwd = get_fpu_cwd(task);
	swd = get_fpu_swd(task);
484
	switch (swd & ~cwd & 0x3f) {
485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508
	case 0x000: /* No unmasked exception */
	default: /* Multiple exceptions */
		break;
	case 0x001: /* Invalid Op */
		/*
		 * swd & 0x240 == 0x040: Stack Underflow
		 * swd & 0x240 == 0x240: Stack Overflow
		 * User must clear the SF bit (0x40) if set
		 */
		info.si_code = FPE_FLTINV;
		break;
	case 0x002: /* Denormalize */
	case 0x010: /* Underflow */
		info.si_code = FPE_FLTUND;
		break;
	case 0x004: /* Zero Divide */
		info.si_code = FPE_FLTDIV;
		break;
	case 0x008: /* Overflow */
		info.si_code = FPE_FLTOVF;
		break;
	case 0x020: /* Precision */
		info.si_code = FPE_FLTRES;
		break;
L
Linus Torvalds 已提交
509 510 511 512
	}
	force_sig_info(SIGFPE, &info, task);
}

513
dotraplinkage void do_coprocessor_error(struct pt_regs *regs, long error_code)
514 515 516 517 518 519 520 521
{
	conditional_sti(regs);
	if (!user_mode(regs) &&
	    kernel_math_error(regs, "kernel x87 math error", 16))
		return;
	math_error((void __user *)regs->ip);
}

L
Linus Torvalds 已提交
522 523
asmlinkage void bad_intr(void)
{
524
	printk("bad interrupt");
L
Linus Torvalds 已提交
525 526
}

527
static void simd_math_error(void __user *ip)
L
Linus Torvalds 已提交
528
{
529
	struct task_struct *task;
L
Linus Torvalds 已提交
530 531 532 533 534 535 536 537 538 539 540 541 542
	siginfo_t info;
	unsigned short mxcsr;

	/*
	 * Save the info for the exception handler and clear the error.
	 */
	task = current;
	save_init_fpu(task);
	task->thread.trap_no = 19;
	task->thread.error_code = 0;
	info.si_signo = SIGFPE;
	info.si_errno = 0;
	info.si_code = __SI_FAULT;
543
	info.si_addr = ip;
L
Linus Torvalds 已提交
544 545 546 547 548 549 550 551
	/*
	 * The SIMD FPU exceptions are handled a little differently, as there
	 * is only a single status/control register.  Thus, to determine which
	 * unmasked exception was caught we must mask the exception mask bits
	 * at 0x1f80, and then use these to mask the exception bits at 0x3f.
	 */
	mxcsr = get_fpu_mxcsr(task);
	switch (~((mxcsr & 0x1f80) >> 7) & (mxcsr & 0x3f)) {
552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570
	case 0x000:
	default:
		break;
	case 0x001: /* Invalid Op */
		info.si_code = FPE_FLTINV;
		break;
	case 0x002: /* Denormalize */
	case 0x010: /* Underflow */
		info.si_code = FPE_FLTUND;
		break;
	case 0x004: /* Zero Divide */
		info.si_code = FPE_FLTDIV;
		break;
	case 0x008: /* Overflow */
		info.si_code = FPE_FLTOVF;
		break;
	case 0x020: /* Precision */
		info.si_code = FPE_FLTRES;
		break;
L
Linus Torvalds 已提交
571 572 573 574
	}
	force_sig_info(SIGFPE, &info, task);
}

575 576
dotraplinkage void
do_simd_coprocessor_error(struct pt_regs *regs, long error_code)
577 578 579 580 581 582 583 584
{
	conditional_sti(regs);
	if (!user_mode(regs) &&
			kernel_math_error(regs, "kernel simd math error", 19))
		return;
	simd_math_error((void __user *)regs->ip);
}

585 586
dotraplinkage void
do_spurious_interrupt_bug(struct pt_regs *regs, long error_code)
L
Linus Torvalds 已提交
587 588 589 590
{
}

asmlinkage void __attribute__((weak)) smp_thermal_interrupt(void)
591 592 593 594
{
}

asmlinkage void __attribute__((weak)) mce_threshold_interrupt(void)
L
Linus Torvalds 已提交
595 596 597 598
{
}

/*
599
 * 'math_state_restore()' saves the current math information in the
L
Linus Torvalds 已提交
600 601 602 603 604 605 606 607 608
 * old math state array, and gets the new ones from the current task
 *
 * Careful.. There are problems with IBM-designed IRQ13 behaviour.
 * Don't touch unless you *really* know how it works.
 */
asmlinkage void math_state_restore(void)
{
	struct task_struct *me = current;

609 610 611 612 613 614 615 616 617 618 619 620 621 622 623
	if (!used_math()) {
		local_irq_enable();
		/*
		 * does a slab alloc which can sleep
		 */
		if (init_fpu(me)) {
			/*
			 * ran out of memory!
			 */
			do_group_exit(SIGKILL);
			return;
		}
		local_irq_disable();
	}

624
	clts();				/* Allow maths ops (or we recurse) */
625 626 627
	/*
	 * Paranoid restore. send a SIGSEGV if we fail to restore the state.
	 */
628
	if (unlikely(restore_fpu_checking(me))) {
629 630 631 632
		stts();
		force_sig(SIGSEGV, me);
		return;
	}
A
Al Viro 已提交
633
	task_thread_info(me)->status |= TS_USEDFPU;
634
	me->fpu_counter++;
L
Linus Torvalds 已提交
635
}
636
EXPORT_SYMBOL_GPL(math_state_restore);
L
Linus Torvalds 已提交
637

638 639 640 641 642 643
dotraplinkage void __kprobes
do_device_not_available(struct pt_regs *regs, long error)
{
	math_state_restore();
}

L
Linus Torvalds 已提交
644 645
void __init trap_init(void)
{
646 647 648
	set_intr_gate(0, &divide_error);
	set_intr_gate_ist(1, &debug, DEBUG_STACK);
	set_intr_gate_ist(2, &nmi, NMI_STACK);
649 650 651 652
	/* int3 can be called from all */
	set_system_gate_ist(3, &int3, DEBUG_STACK);
	/* int4 can be called from all */
	set_system_gate(4, &overflow);
653 654 655 656 657 658 659 660 661 662 663 664 665
	set_intr_gate(5, &bounds);
	set_intr_gate(6, &invalid_op);
	set_intr_gate(7, &device_not_available);
	set_intr_gate_ist(8, &double_fault, DOUBLEFAULT_STACK);
	set_intr_gate(9, &coprocessor_segment_overrun);
	set_intr_gate(10, &invalid_TSS);
	set_intr_gate(11, &segment_not_present);
	set_intr_gate_ist(12, &stack_segment, STACKFAULT_STACK);
	set_intr_gate(13, &general_protection);
	set_intr_gate(14, &page_fault);
	set_intr_gate(15, &spurious_interrupt_bug);
	set_intr_gate(16, &coprocessor_error);
	set_intr_gate(17, &alignment_check);
L
Linus Torvalds 已提交
666
#ifdef CONFIG_X86_MCE
667
	set_intr_gate_ist(18, &machine_check, MCE_STACK);
L
Linus Torvalds 已提交
668
#endif
669
	set_intr_gate(19, &simd_coprocessor_error);
L
Linus Torvalds 已提交
670 671 672 673 674

#ifdef CONFIG_IA32_EMULATION
	set_system_gate(IA32_SYSCALL_VECTOR, ia32_syscall);
#endif
	/*
675
	 * Should be a barrier for any external CPU state:
L
Linus Torvalds 已提交
676 677 678
	 */
	cpu_init();
}