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