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