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