]> Pileus Git - ~andy/linux/blob - arch/x86/kernel/reboot.c
10ae9be07b47359d7eff37807668f16dc4cfc026
[~andy/linux] / arch / x86 / kernel / reboot.c
1 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
2
3 #include <linux/module.h>
4 #include <linux/reboot.h>
5 #include <linux/init.h>
6 #include <linux/pm.h>
7 #include <linux/efi.h>
8 #include <linux/dmi.h>
9 #include <linux/sched.h>
10 #include <linux/tboot.h>
11 #include <linux/delay.h>
12 #include <acpi/reboot.h>
13 #include <asm/io.h>
14 #include <asm/apic.h>
15 #include <asm/desc.h>
16 #include <asm/hpet.h>
17 #include <asm/pgtable.h>
18 #include <asm/proto.h>
19 #include <asm/reboot_fixups.h>
20 #include <asm/reboot.h>
21 #include <asm/pci_x86.h>
22 #include <asm/virtext.h>
23 #include <asm/cpu.h>
24 #include <asm/nmi.h>
25
26 #ifdef CONFIG_X86_32
27 # include <linux/ctype.h>
28 # include <linux/mc146818rtc.h>
29 # include <asm/realmode.h>
30 #else
31 # include <asm/x86_init.h>
32 #endif
33
34 /*
35  * Power off function, if any
36  */
37 void (*pm_power_off)(void);
38 EXPORT_SYMBOL(pm_power_off);
39
40 static const struct desc_ptr no_idt = {};
41 static int reboot_mode;
42 enum reboot_type reboot_type = BOOT_ACPI;
43 int reboot_force;
44
45 /*
46  * This variable is used privately to keep track of whether or not
47  * reboot_type is still set to its default value (i.e., reboot= hasn't
48  * been set on the command line).  This is needed so that we can
49  * suppress DMI scanning for reboot quirks.  Without it, it's
50  * impossible to override a faulty reboot quirk without recompiling.
51  */
52 static int reboot_default = 1;
53
54 #if defined(CONFIG_X86_32) && defined(CONFIG_SMP)
55 static int reboot_cpu = -1;
56 #endif
57
58 /*
59  * This is set if we need to go through the 'emergency' path.
60  * When machine_emergency_restart() is called, we may be on
61  * an inconsistent state and won't be able to do a clean cleanup
62  */
63 static int reboot_emergency;
64
65 /* This is set by the PCI code if either type 1 or type 2 PCI is detected */
66 bool port_cf9_safe = false;
67
68 /*
69  * reboot=b[ios] | s[mp] | t[riple] | k[bd] | e[fi] [, [w]arm | [c]old] | p[ci]
70  * warm   Don't set the cold reboot flag
71  * cold   Set the cold reboot flag
72  * bios   Reboot by jumping through the BIOS (only for X86_32)
73  * smp    Reboot by executing reset on BSP or other CPU (only for X86_32)
74  * triple Force a triple fault (init)
75  * kbd    Use the keyboard controller. cold reset (default)
76  * acpi   Use the RESET_REG in the FADT
77  * efi    Use efi reset_system runtime service
78  * pci    Use the so-called "PCI reset register", CF9
79  * force  Avoid anything that could hang.
80  */
81 static int __init reboot_setup(char *str)
82 {
83         for (;;) {
84                 /*
85                  * Having anything passed on the command line via
86                  * reboot= will cause us to disable DMI checking
87                  * below.
88                  */
89                 reboot_default = 0;
90
91                 switch (*str) {
92                 case 'w':
93                         reboot_mode = 0x1234;
94                         break;
95
96                 case 'c':
97                         reboot_mode = 0;
98                         break;
99
100 #ifdef CONFIG_X86_32
101 #ifdef CONFIG_SMP
102                 case 's':
103                         if (isdigit(*(str+1))) {
104                                 reboot_cpu = (int) (*(str+1) - '0');
105                                 if (isdigit(*(str+2)))
106                                         reboot_cpu = reboot_cpu*10 + (int)(*(str+2) - '0');
107                         }
108                         /*
109                          * We will leave sorting out the final value
110                          * when we are ready to reboot, since we might not
111                          * have detected BSP APIC ID or smp_num_cpu
112                          */
113                         break;
114 #endif /* CONFIG_SMP */
115
116                 case 'b':
117 #endif
118                 case 'a':
119                 case 'k':
120                 case 't':
121                 case 'e':
122                 case 'p':
123                         reboot_type = *str;
124                         break;
125
126                 case 'f':
127                         reboot_force = 1;
128                         break;
129                 }
130
131                 str = strchr(str, ',');
132                 if (str)
133                         str++;
134                 else
135                         break;
136         }
137         return 1;
138 }
139
140 __setup("reboot=", reboot_setup);
141
142
143 #ifdef CONFIG_X86_32
144 /*
145  * Reboot options and system auto-detection code provided by
146  * Dell Inc. so their systems "just work". :-)
147  */
148
149 /*
150  * Some machines require the "reboot=b" or "reboot=k"  commandline options,
151  * this quirk makes that automatic.
152  */
153 static int __init set_bios_reboot(const struct dmi_system_id *d)
154 {
155         if (reboot_type != BOOT_BIOS) {
156                 reboot_type = BOOT_BIOS;
157                 pr_info("%s series board detected. Selecting %s-method for reboots.\n",
158                         "BIOS", d->ident);
159         }
160         return 0;
161 }
162
163 void machine_real_restart(unsigned int type)
164 {
165         void (*restart_lowmem)(unsigned int) = (void (*)(unsigned int))
166                 real_mode_header->machine_real_restart_asm;
167
168         local_irq_disable();
169
170         /*
171          * Write zero to CMOS register number 0x0f, which the BIOS POST
172          * routine will recognize as telling it to do a proper reboot.  (Well
173          * that's what this book in front of me says -- it may only apply to
174          * the Phoenix BIOS though, it's not clear).  At the same time,
175          * disable NMIs by setting the top bit in the CMOS address register,
176          * as we're about to do peculiar things to the CPU.  I'm not sure if
177          * `outb_p' is needed instead of just `outb'.  Use it to be on the
178          * safe side.  (Yes, CMOS_WRITE does outb_p's. -  Paul G.)
179          */
180         spin_lock(&rtc_lock);
181         CMOS_WRITE(0x00, 0x8f);
182         spin_unlock(&rtc_lock);
183
184         /*
185          * Switch back to the initial page table.
186          */
187         load_cr3(initial_page_table);
188
189         /*
190          * Write 0x1234 to absolute memory location 0x472.  The BIOS reads
191          * this on booting to tell it to "Bypass memory test (also warm
192          * boot)".  This seems like a fairly standard thing that gets set by
193          * REBOOT.COM programs, and the previous reset routine did this
194          * too. */
195         *((unsigned short *)0x472) = reboot_mode;
196
197         /* Jump to the identity-mapped low memory code */
198         restart_lowmem(type);
199 }
200 #ifdef CONFIG_APM_MODULE
201 EXPORT_SYMBOL(machine_real_restart);
202 #endif
203
204 #endif /* CONFIG_X86_32 */
205
206 /*
207  * Some Apple MacBook and MacBookPro's needs reboot=p to be able to reboot
208  */
209 static int __init set_pci_reboot(const struct dmi_system_id *d)
210 {
211         if (reboot_type != BOOT_CF9) {
212                 reboot_type = BOOT_CF9;
213                 pr_info("%s series board detected. Selecting %s-method for reboots.\n",
214                         "PCI", d->ident);
215         }
216         return 0;
217 }
218
219 static int __init set_kbd_reboot(const struct dmi_system_id *d)
220 {
221         if (reboot_type != BOOT_KBD) {
222                 reboot_type = BOOT_KBD;
223                 pr_info("%s series board detected. Selecting %s-method for reboot.\n",
224                         "KBD", d->ident);
225         }
226         return 0;
227 }
228
229 /*
230  * This is a single dmi_table handling all reboot quirks.  Note that
231  * REBOOT_BIOS is only available for 32bit
232  */
233 static struct dmi_system_id __initdata reboot_dmi_table[] = {
234 #ifdef CONFIG_X86_32
235         {       /* Handle problems with rebooting on Dell E520's */
236                 .callback = set_bios_reboot,
237                 .ident = "Dell E520",
238                 .matches = {
239                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
240                         DMI_MATCH(DMI_PRODUCT_NAME, "Dell DM061"),
241                 },
242         },
243         {       /* Handle problems with rebooting on Dell 1300's */
244                 .callback = set_bios_reboot,
245                 .ident = "Dell PowerEdge 1300",
246                 .matches = {
247                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Computer Corporation"),
248                         DMI_MATCH(DMI_PRODUCT_NAME, "PowerEdge 1300/"),
249                 },
250         },
251         {       /* Handle problems with rebooting on Dell 300's */
252                 .callback = set_bios_reboot,
253                 .ident = "Dell PowerEdge 300",
254                 .matches = {
255                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Computer Corporation"),
256                         DMI_MATCH(DMI_PRODUCT_NAME, "PowerEdge 300/"),
257                 },
258         },
259         {       /* Handle problems with rebooting on Dell Optiplex 745's SFF */
260                 .callback = set_bios_reboot,
261                 .ident = "Dell OptiPlex 745",
262                 .matches = {
263                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
264                         DMI_MATCH(DMI_PRODUCT_NAME, "OptiPlex 745"),
265                 },
266         },
267         {       /* Handle problems with rebooting on Dell Optiplex 745's DFF */
268                 .callback = set_bios_reboot,
269                 .ident = "Dell OptiPlex 745",
270                 .matches = {
271                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
272                         DMI_MATCH(DMI_PRODUCT_NAME, "OptiPlex 745"),
273                         DMI_MATCH(DMI_BOARD_NAME, "0MM599"),
274                 },
275         },
276         {       /* Handle problems with rebooting on Dell Optiplex 745 with 0KW626 */
277                 .callback = set_bios_reboot,
278                 .ident = "Dell OptiPlex 745",
279                 .matches = {
280                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
281                         DMI_MATCH(DMI_PRODUCT_NAME, "OptiPlex 745"),
282                         DMI_MATCH(DMI_BOARD_NAME, "0KW626"),
283                 },
284         },
285         {       /* Handle problems with rebooting on Dell Optiplex 330 with 0KP561 */
286                 .callback = set_bios_reboot,
287                 .ident = "Dell OptiPlex 330",
288                 .matches = {
289                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
290                         DMI_MATCH(DMI_PRODUCT_NAME, "OptiPlex 330"),
291                         DMI_MATCH(DMI_BOARD_NAME, "0KP561"),
292                 },
293         },
294         {       /* Handle problems with rebooting on Dell Optiplex 360 with 0T656F */
295                 .callback = set_bios_reboot,
296                 .ident = "Dell OptiPlex 360",
297                 .matches = {
298                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
299                         DMI_MATCH(DMI_PRODUCT_NAME, "OptiPlex 360"),
300                         DMI_MATCH(DMI_BOARD_NAME, "0T656F"),
301                 },
302         },
303         {       /* Handle problems with rebooting on Dell OptiPlex 760 with 0G919G */
304                 .callback = set_bios_reboot,
305                 .ident = "Dell OptiPlex 760",
306                 .matches = {
307                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
308                         DMI_MATCH(DMI_PRODUCT_NAME, "OptiPlex 760"),
309                         DMI_MATCH(DMI_BOARD_NAME, "0G919G"),
310                 },
311         },
312         {       /* Handle problems with rebooting on Dell 2400's */
313                 .callback = set_bios_reboot,
314                 .ident = "Dell PowerEdge 2400",
315                 .matches = {
316                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Computer Corporation"),
317                         DMI_MATCH(DMI_PRODUCT_NAME, "PowerEdge 2400"),
318                 },
319         },
320         {       /* Handle problems with rebooting on Dell T5400's */
321                 .callback = set_bios_reboot,
322                 .ident = "Dell Precision T5400",
323                 .matches = {
324                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
325                         DMI_MATCH(DMI_PRODUCT_NAME, "Precision WorkStation T5400"),
326                 },
327         },
328         {       /* Handle problems with rebooting on Dell T7400's */
329                 .callback = set_bios_reboot,
330                 .ident = "Dell Precision T7400",
331                 .matches = {
332                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
333                         DMI_MATCH(DMI_PRODUCT_NAME, "Precision WorkStation T7400"),
334                 },
335         },
336         {       /* Handle problems with rebooting on HP laptops */
337                 .callback = set_bios_reboot,
338                 .ident = "HP Compaq Laptop",
339                 .matches = {
340                         DMI_MATCH(DMI_SYS_VENDOR, "Hewlett-Packard"),
341                         DMI_MATCH(DMI_PRODUCT_NAME, "HP Compaq"),
342                 },
343         },
344         {       /* Handle problems with rebooting on Dell XPS710 */
345                 .callback = set_bios_reboot,
346                 .ident = "Dell XPS710",
347                 .matches = {
348                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
349                         DMI_MATCH(DMI_PRODUCT_NAME, "Dell XPS710"),
350                 },
351         },
352         {       /* Handle problems with rebooting on Dell DXP061 */
353                 .callback = set_bios_reboot,
354                 .ident = "Dell DXP061",
355                 .matches = {
356                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
357                         DMI_MATCH(DMI_PRODUCT_NAME, "Dell DXP061"),
358                 },
359         },
360         {       /* Handle problems with rebooting on Sony VGN-Z540N */
361                 .callback = set_bios_reboot,
362                 .ident = "Sony VGN-Z540N",
363                 .matches = {
364                         DMI_MATCH(DMI_SYS_VENDOR, "Sony Corporation"),
365                         DMI_MATCH(DMI_PRODUCT_NAME, "VGN-Z540N"),
366                 },
367         },
368         {       /* Handle problems with rebooting on CompuLab SBC-FITPC2 */
369                 .callback = set_bios_reboot,
370                 .ident = "CompuLab SBC-FITPC2",
371                 .matches = {
372                         DMI_MATCH(DMI_SYS_VENDOR, "CompuLab"),
373                         DMI_MATCH(DMI_PRODUCT_NAME, "SBC-FITPC2"),
374                 },
375         },
376         {       /* Handle problems with rebooting on ASUS P4S800 */
377                 .callback = set_bios_reboot,
378                 .ident = "ASUS P4S800",
379                 .matches = {
380                         DMI_MATCH(DMI_BOARD_VENDOR, "ASUSTeK Computer INC."),
381                         DMI_MATCH(DMI_BOARD_NAME, "P4S800"),
382                 },
383         },
384 #endif /* CONFIG_X86_32 */
385
386         {       /* Handle reboot issue on Acer Aspire one */
387                 .callback = set_kbd_reboot,
388                 .ident = "Acer Aspire One A110",
389                 .matches = {
390                         DMI_MATCH(DMI_SYS_VENDOR, "Acer"),
391                         DMI_MATCH(DMI_PRODUCT_NAME, "AOA110"),
392                 },
393         },
394         {       /* Handle problems with rebooting on Apple MacBook5 */
395                 .callback = set_pci_reboot,
396                 .ident = "Apple MacBook5",
397                 .matches = {
398                         DMI_MATCH(DMI_SYS_VENDOR, "Apple Inc."),
399                         DMI_MATCH(DMI_PRODUCT_NAME, "MacBook5"),
400                 },
401         },
402         {       /* Handle problems with rebooting on Apple MacBookPro5 */
403                 .callback = set_pci_reboot,
404                 .ident = "Apple MacBookPro5",
405                 .matches = {
406                         DMI_MATCH(DMI_SYS_VENDOR, "Apple Inc."),
407                         DMI_MATCH(DMI_PRODUCT_NAME, "MacBookPro5"),
408                 },
409         },
410         {       /* Handle problems with rebooting on Apple Macmini3,1 */
411                 .callback = set_pci_reboot,
412                 .ident = "Apple Macmini3,1",
413                 .matches = {
414                         DMI_MATCH(DMI_SYS_VENDOR, "Apple Inc."),
415                         DMI_MATCH(DMI_PRODUCT_NAME, "Macmini3,1"),
416                 },
417         },
418         {       /* Handle problems with rebooting on the iMac9,1. */
419                 .callback = set_pci_reboot,
420                 .ident = "Apple iMac9,1",
421                 .matches = {
422                         DMI_MATCH(DMI_SYS_VENDOR, "Apple Inc."),
423                         DMI_MATCH(DMI_PRODUCT_NAME, "iMac9,1"),
424                 },
425         },
426         {       /* Handle problems with rebooting on the Latitude E6320. */
427                 .callback = set_pci_reboot,
428                 .ident = "Dell Latitude E6320",
429                 .matches = {
430                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
431                         DMI_MATCH(DMI_PRODUCT_NAME, "Latitude E6320"),
432                 },
433         },
434         {       /* Handle problems with rebooting on the Latitude E5420. */
435                 .callback = set_pci_reboot,
436                 .ident = "Dell Latitude E5420",
437                 .matches = {
438                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
439                         DMI_MATCH(DMI_PRODUCT_NAME, "Latitude E5420"),
440                 },
441         },
442         {       /* Handle problems with rebooting on the Latitude E6420. */
443                 .callback = set_pci_reboot,
444                 .ident = "Dell Latitude E6420",
445                 .matches = {
446                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
447                         DMI_MATCH(DMI_PRODUCT_NAME, "Latitude E6420"),
448                 },
449         },
450         {       /* Handle problems with rebooting on the OptiPlex 990. */
451                 .callback = set_pci_reboot,
452                 .ident = "Dell OptiPlex 990",
453                 .matches = {
454                         DMI_MATCH(DMI_SYS_VENDOR, "Dell Inc."),
455                         DMI_MATCH(DMI_PRODUCT_NAME, "OptiPlex 990"),
456                 },
457         },
458         { }
459 };
460
461 static int __init reboot_init(void)
462 {
463         /*
464          * Only do the DMI check if reboot_type hasn't been overridden
465          * on the command line
466          */
467         if (reboot_default)
468                 dmi_check_system(reboot_dmi_table);
469         return 0;
470 }
471 core_initcall(reboot_init);
472
473 static inline void kb_wait(void)
474 {
475         int i;
476
477         for (i = 0; i < 0x10000; i++) {
478                 if ((inb(0x64) & 0x02) == 0)
479                         break;
480                 udelay(2);
481         }
482 }
483
484 static void vmxoff_nmi(int cpu, struct pt_regs *regs)
485 {
486         cpu_emergency_vmxoff();
487 }
488
489 /* Use NMIs as IPIs to tell all CPUs to disable virtualization */
490 static void emergency_vmx_disable_all(void)
491 {
492         /* Just make sure we won't change CPUs while doing this */
493         local_irq_disable();
494
495         /*
496          * We need to disable VMX on all CPUs before rebooting, otherwise
497          * we risk hanging up the machine, because the CPU ignore INIT
498          * signals when VMX is enabled.
499          *
500          * We can't take any locks and we may be on an inconsistent
501          * state, so we use NMIs as IPIs to tell the other CPUs to disable
502          * VMX and halt.
503          *
504          * For safety, we will avoid running the nmi_shootdown_cpus()
505          * stuff unnecessarily, but we don't have a way to check
506          * if other CPUs have VMX enabled. So we will call it only if the
507          * CPU we are running on has VMX enabled.
508          *
509          * We will miss cases where VMX is not enabled on all CPUs. This
510          * shouldn't do much harm because KVM always enable VMX on all
511          * CPUs anyway. But we can miss it on the small window where KVM
512          * is still enabling VMX.
513          */
514         if (cpu_has_vmx() && cpu_vmx_enabled()) {
515                 /* Disable VMX on this CPU. */
516                 cpu_vmxoff();
517
518                 /* Halt and disable VMX on the other CPUs */
519                 nmi_shootdown_cpus(vmxoff_nmi);
520
521         }
522 }
523
524
525 void __attribute__((weak)) mach_reboot_fixups(void)
526 {
527 }
528
529 /*
530  * Windows compatible x86 hardware expects the following on reboot:
531  *
532  * 1) If the FADT has the ACPI reboot register flag set, try it
533  * 2) If still alive, write to the keyboard controller
534  * 3) If still alive, write to the ACPI reboot register again
535  * 4) If still alive, write to the keyboard controller again
536  *
537  * If the machine is still alive at this stage, it gives up. We default to
538  * following the same pattern, except that if we're still alive after (4) we'll
539  * try to force a triple fault and then cycle between hitting the keyboard
540  * controller and doing that
541  */
542 static void native_machine_emergency_restart(void)
543 {
544         int i;
545         int attempt = 0;
546         int orig_reboot_type = reboot_type;
547
548         if (reboot_emergency)
549                 emergency_vmx_disable_all();
550
551         tboot_shutdown(TB_SHUTDOWN_REBOOT);
552
553         /* Tell the BIOS if we want cold or warm reboot */
554         *((unsigned short *)__va(0x472)) = reboot_mode;
555
556         for (;;) {
557                 /* Could also try the reset bit in the Hammer NB */
558                 switch (reboot_type) {
559                 case BOOT_KBD:
560                         mach_reboot_fixups(); /* For board specific fixups */
561
562                         for (i = 0; i < 10; i++) {
563                                 kb_wait();
564                                 udelay(50);
565                                 outb(0xfe, 0x64); /* Pulse reset low */
566                                 udelay(50);
567                         }
568                         if (attempt == 0 && orig_reboot_type == BOOT_ACPI) {
569                                 attempt = 1;
570                                 reboot_type = BOOT_ACPI;
571                         } else {
572                                 reboot_type = BOOT_TRIPLE;
573                         }
574                         break;
575
576                 case BOOT_TRIPLE:
577                         load_idt(&no_idt);
578                         __asm__ __volatile__("int3");
579
580                         reboot_type = BOOT_KBD;
581                         break;
582
583 #ifdef CONFIG_X86_32
584                 case BOOT_BIOS:
585                         machine_real_restart(MRR_BIOS);
586
587                         reboot_type = BOOT_KBD;
588                         break;
589 #endif
590
591                 case BOOT_ACPI:
592                         acpi_reboot();
593                         reboot_type = BOOT_KBD;
594                         break;
595
596                 case BOOT_EFI:
597                         if (efi_enabled)
598                                 efi.reset_system(reboot_mode ?
599                                                  EFI_RESET_WARM :
600                                                  EFI_RESET_COLD,
601                                                  EFI_SUCCESS, 0, NULL);
602                         reboot_type = BOOT_KBD;
603                         break;
604
605                 case BOOT_CF9:
606                         port_cf9_safe = true;
607                         /* Fall through */
608
609                 case BOOT_CF9_COND:
610                         if (port_cf9_safe) {
611                                 u8 cf9 = inb(0xcf9) & ~6;
612                                 outb(cf9|2, 0xcf9); /* Request hard reset */
613                                 udelay(50);
614                                 outb(cf9|6, 0xcf9); /* Actually do the reset */
615                                 udelay(50);
616                         }
617                         reboot_type = BOOT_KBD;
618                         break;
619                 }
620         }
621 }
622
623 void native_machine_shutdown(void)
624 {
625         /* Stop the cpus and apics */
626 #ifdef CONFIG_SMP
627
628         /* The boot cpu is always logical cpu 0 */
629         int reboot_cpu_id = 0;
630
631 #ifdef CONFIG_X86_32
632         /* See if there has been given a command line override */
633         if ((reboot_cpu != -1) && (reboot_cpu < nr_cpu_ids) &&
634                 cpu_online(reboot_cpu))
635                 reboot_cpu_id = reboot_cpu;
636 #endif
637
638         /* Make certain the cpu I'm about to reboot on is online */
639         if (!cpu_online(reboot_cpu_id))
640                 reboot_cpu_id = smp_processor_id();
641
642         /* Make certain I only run on the appropriate processor */
643         set_cpus_allowed_ptr(current, cpumask_of(reboot_cpu_id));
644
645         /*
646          * O.K Now that I'm on the appropriate processor, stop all of the
647          * others. Also disable the local irq to not receive the per-cpu
648          * timer interrupt which may trigger scheduler's load balance.
649          */
650         local_irq_disable();
651         stop_other_cpus();
652 #endif
653
654         lapic_shutdown();
655
656 #ifdef CONFIG_X86_IO_APIC
657         disable_IO_APIC();
658 #endif
659
660 #ifdef CONFIG_HPET_TIMER
661         hpet_disable();
662 #endif
663
664 #ifdef CONFIG_X86_64
665         x86_platform.iommu_shutdown();
666 #endif
667 }
668
669 static void __machine_emergency_restart(int emergency)
670 {
671         reboot_emergency = emergency;
672         machine_ops.emergency_restart();
673 }
674
675 static void native_machine_restart(char *__unused)
676 {
677         pr_notice("machine restart\n");
678
679         if (!reboot_force)
680                 machine_shutdown();
681         __machine_emergency_restart(0);
682 }
683
684 static void native_machine_halt(void)
685 {
686         /* Stop other cpus and apics */
687         machine_shutdown();
688
689         tboot_shutdown(TB_SHUTDOWN_HALT);
690
691         stop_this_cpu(NULL);
692 }
693
694 static void native_machine_power_off(void)
695 {
696         if (pm_power_off) {
697                 if (!reboot_force)
698                         machine_shutdown();
699                 pm_power_off();
700         }
701         /* A fallback in case there is no PM info available */
702         tboot_shutdown(TB_SHUTDOWN_HALT);
703 }
704
705 struct machine_ops machine_ops = {
706         .power_off = native_machine_power_off,
707         .shutdown = native_machine_shutdown,
708         .emergency_restart = native_machine_emergency_restart,
709         .restart = native_machine_restart,
710         .halt = native_machine_halt,
711 #ifdef CONFIG_KEXEC
712         .crash_shutdown = native_machine_crash_shutdown,
713 #endif
714 };
715
716 void machine_power_off(void)
717 {
718         machine_ops.power_off();
719 }
720
721 void machine_shutdown(void)
722 {
723         machine_ops.shutdown();
724 }
725
726 void machine_emergency_restart(void)
727 {
728         __machine_emergency_restart(1);
729 }
730
731 void machine_restart(char *cmd)
732 {
733         machine_ops.restart(cmd);
734 }
735
736 void machine_halt(void)
737 {
738         machine_ops.halt();
739 }
740
741 #ifdef CONFIG_KEXEC
742 void machine_crash_shutdown(struct pt_regs *regs)
743 {
744         machine_ops.crash_shutdown(regs);
745 }
746 #endif
747
748
749 #if defined(CONFIG_SMP)
750
751 /* This keeps a track of which one is crashing cpu. */
752 static int crashing_cpu;
753 static nmi_shootdown_cb shootdown_callback;
754
755 static atomic_t waiting_for_crash_ipi;
756
757 static int crash_nmi_callback(unsigned int val, struct pt_regs *regs)
758 {
759         int cpu;
760
761         cpu = raw_smp_processor_id();
762
763         /*
764          * Don't do anything if this handler is invoked on crashing cpu.
765          * Otherwise, system will completely hang. Crashing cpu can get
766          * an NMI if system was initially booted with nmi_watchdog parameter.
767          */
768         if (cpu == crashing_cpu)
769                 return NMI_HANDLED;
770         local_irq_disable();
771
772         shootdown_callback(cpu, regs);
773
774         atomic_dec(&waiting_for_crash_ipi);
775         /* Assume hlt works */
776         halt();
777         for (;;)
778                 cpu_relax();
779
780         return NMI_HANDLED;
781 }
782
783 static void smp_send_nmi_allbutself(void)
784 {
785         apic->send_IPI_allbutself(NMI_VECTOR);
786 }
787
788 /*
789  * Halt all other CPUs, calling the specified function on each of them
790  *
791  * This function can be used to halt all other CPUs on crash
792  * or emergency reboot time. The function passed as parameter
793  * will be called inside a NMI handler on all CPUs.
794  */
795 void nmi_shootdown_cpus(nmi_shootdown_cb callback)
796 {
797         unsigned long msecs;
798         local_irq_disable();
799
800         /* Make a note of crashing cpu. Will be used in NMI callback. */
801         crashing_cpu = safe_smp_processor_id();
802
803         shootdown_callback = callback;
804
805         atomic_set(&waiting_for_crash_ipi, num_online_cpus() - 1);
806         /* Would it be better to replace the trap vector here? */
807         if (register_nmi_handler(NMI_LOCAL, crash_nmi_callback,
808                                  NMI_FLAG_FIRST, "crash"))
809                 return;         /* Return what? */
810         /*
811          * Ensure the new callback function is set before sending
812          * out the NMI
813          */
814         wmb();
815
816         smp_send_nmi_allbutself();
817
818         msecs = 1000; /* Wait at most a second for the other cpus to stop */
819         while ((atomic_read(&waiting_for_crash_ipi) > 0) && msecs) {
820                 mdelay(1);
821                 msecs--;
822         }
823
824         /* Leave the nmi callback set */
825 }
826 #else /* !CONFIG_SMP */
827 void nmi_shootdown_cpus(nmi_shootdown_cb callback)
828 {
829         /* No other CPUs to shoot down */
830 }
831 #endif