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