arch/x86/kernel/cpu/microcode/core.c

   1 /*
   2  * CPU Microcode Update Driver for Linux
   3  *
   4  * Copyright (C) 2000-2006 Tigran Aivazian <aivazian.tigran@gmail.com>
   5  *            2006      Shaohua Li <shaohua.li@intel.com>
   6  *            2013-2016 Borislav Petkov <bp@alien8.de>
   7  *
   8  * X86 CPU microcode early update for Linux:
   9  *
  10  *      Copyright (C) 2012 Fenghua Yu <fenghua.yu@intel.com>
  11  *                         H Peter Anvin" <hpa@zytor.com>
  12  *                (C) 2015 Borislav Petkov <bp@alien8.de>
  13  *
  14  * This driver allows to upgrade microcode on x86 processors.
  15  *
  16  * This program is free software; you can redistribute it and/or
  17  * modify it under the terms of the GNU General Public License
  18  * as published by the Free Software Foundation; either version
  19  * 2 of the License, or (at your option) any later version.
  20  */
  21
  22 #define pr_fmt(fmt) "microcode: " fmt
  23
  24 #include <linux/platform_device.h>
  25 #include <linux/stop_machine.h>
  26 #include <linux/syscore_ops.h>
  27 #include <linux/miscdevice.h>
  28 #include <linux/capability.h>
  29 #include <linux/firmware.h>
  30 #include <linux/kernel.h>
  31 #include <linux/delay.h>
  32 #include <linux/mutex.h>
  33 #include <linux/cpu.h>
  34 #include <linux/nmi.h>
  35 #include <linux/fs.h>
  36 #include <linux/mm.h>
  37
  38 #include <asm/microcode_intel.h>
  39 #include <asm/cpu_device_id.h>
  40 #include <asm/microcode_amd.h>
  41 #include <asm/perf_event.h>
  42 #include <asm/microcode.h>
  43 #include <asm/processor.h>
  44 #include <asm/cmdline.h>
  45 #include <asm/setup.h>
  46
  47 #define DRIVER_VERSION  "2.2"
  48
  49 static struct microcode_ops     *microcode_ops;
  50 static bool dis_ucode_ldr = true;
  51
  52 bool initrd_gone;
  53
  54 LIST_HEAD(microcode_cache);
  55
  56 /*
  57  * Synchronization.
  58  *
  59  * All non cpu-hotplug-callback call sites use:
  60  *
  61  * - microcode_mutex to synchronize with each other;
  62  * - get/put_online_cpus() to synchronize with
  63  *   the cpu-hotplug-callback call sites.
  64  *
  65  * We guarantee that only a single cpu is being
  66  * updated at any particular moment of time.
  67  */
  68 static DEFINE_MUTEX(microcode_mutex);
  69
  70 /*
  71  * Serialize late loading so that CPUs get updated one-by-one.
  72  */
  73 static DEFINE_RAW_SPINLOCK(update_lock);
  74
  75 struct ucode_cpu_info           ucode_cpu_info[NR_CPUS];
  76
  77 struct cpu_info_ctx {
  78         struct cpu_signature    *cpu_sig;
  79         int                     err;
  80 };
  81
  82 /*
  83  * Those patch levels cannot be updated to newer ones and thus should be final.
  84  */
  85 static u32 final_levels[] = {
  86         0x01000098,
  87         0x0100009f,
  88         0x010000af,
  89         0, /* T-101 terminator */
  90 };
  91
  92 /*
  93  * Check the current patch level on this CPU.
  94  *
  95  * Returns:
  96  *  - true: if update should stop
  97  *  - false: otherwise
  98  */
  99 static bool amd_check_current_patch_level(void)
 100 {
 101         u32 lvl, dummy, i;
 102         u32 *levels;
 103
 104         native_rdmsr(MSR_AMD64_PATCH_LEVEL, lvl, dummy);
 105
 106         if (IS_ENABLED(CONFIG_X86_32))
 107                 levels = (u32 *)__pa_nodebug(&final_levels);
 108         else
 109                 levels = final_levels;
 110
 111         for (i = 0; levels[i]; i++) {
 112                 if (lvl == levels[i])
 113                         return true;
 114         }
 115         return false;
 116 }
 117
 118 static bool __init check_loader_disabled_bsp(void)
 119 {
 120         static const char *__dis_opt_str = "dis_ucode_ldr";
 121
 122 #ifdef CONFIG_X86_32
 123         const char *cmdline = (const char *)__pa_nodebug(boot_command_line);
 124         const char *option  = (const char *)__pa_nodebug(__dis_opt_str);
 125         bool *res = (bool *)__pa_nodebug(&dis_ucode_ldr);
 126
 127 #else /* CONFIG_X86_64 */
 128         const char *cmdline = boot_command_line;
 129         const char *option  = __dis_opt_str;
 130         bool *res = &dis_ucode_ldr;
 131 #endif
 132
 133         /*
 134          * CPUID(1).ECX[31]: reserved for hypervisor use. This is still not
 135          * completely accurate as xen pv guests don't see that CPUID bit set but
 136          * that's good enough as they don't land on the BSP path anyway.
 137          */
 138         if (native_cpuid_ecx(1) & BIT(31))
 139                 return *res;
 140
 141         if (x86_cpuid_vendor() == X86_VENDOR_AMD) {
 142                 if (amd_check_current_patch_level())
 143                         return *res;
 144         }
 145
 146         if (cmdline_find_option_bool(cmdline, option) <= 0)
 147                 *res = false;
 148
 149         return *res;
 150 }
 151
 152 extern struct builtin_fw __start_builtin_fw[];
 153 extern struct builtin_fw __end_builtin_fw[];
 154
 155 bool get_builtin_firmware(struct cpio_data *cd, const char *name)
 156 {
 157 #ifdef CONFIG_FW_LOADER
 158         struct builtin_fw *b_fw;
 159
 160         for (b_fw = __start_builtin_fw; b_fw != __end_builtin_fw; b_fw++) {
 161                 if (!strcmp(name, b_fw->name)) {
 162                         cd->size = b_fw->size;
 163                         cd->data = b_fw->data;
 164                         return true;
 165                 }
 166         }
 167 #endif
 168         return false;
 169 }
 170
 171 void __init load_ucode_bsp(void)
 172 {
 173         unsigned int cpuid_1_eax;
 174         bool intel = true;
 175
 176         if (!have_cpuid_p())
 177                 return;
 178
 179         cpuid_1_eax = native_cpuid_eax(1);
 180
 181         switch (x86_cpuid_vendor()) {
 182         case X86_VENDOR_INTEL:
 183                 if (x86_family(cpuid_1_eax) < 6)
 184                         return;
 185                 break;
 186
 187         case X86_VENDOR_AMD:
 188                 if (x86_family(cpuid_1_eax) < 0x10)
 189                         return;
 190                 intel = false;
 191                 break;
 192
 193         default:
 194                 return;
 195         }
 196
 197         if (check_loader_disabled_bsp())
 198                 return;
 199
 200         if (intel)
 201                 load_ucode_intel_bsp();
 202         else
 203                 load_ucode_amd_bsp(cpuid_1_eax);
 204 }
 205
 206 static bool check_loader_disabled_ap(void)
 207 {
 208 #ifdef CONFIG_X86_32
 209         return *((bool *)__pa_nodebug(&dis_ucode_ldr));
 210 #else
 211         return dis_ucode_ldr;
 212 #endif
 213 }
 214
 215 void load_ucode_ap(void)
 216 {
 217         unsigned int cpuid_1_eax;
 218
 219         if (check_loader_disabled_ap())
 220                 return;
 221
 222         cpuid_1_eax = native_cpuid_eax(1);
 223
 224         switch (x86_cpuid_vendor()) {
 225         case X86_VENDOR_INTEL:
 226                 if (x86_family(cpuid_1_eax) >= 6)
 227                         load_ucode_intel_ap();
 228                 break;
 229         case X86_VENDOR_AMD:
 230                 if (x86_family(cpuid_1_eax) >= 0x10)
 231                         load_ucode_amd_ap(cpuid_1_eax);
 232                 break;
 233         default:
 234                 break;
 235         }
 236 }
 237
 238 static int __init save_microcode_in_initrd(void)
 239 {
 240         struct cpuinfo_x86 *c = &boot_cpu_data;
 241         int ret = -EINVAL;
 242
 243         switch (c->x86_vendor) {
 244         case X86_VENDOR_INTEL:
 245                 if (c->x86 >= 6)
 246                         ret = save_microcode_in_initrd_intel();
 247                 break;
 248         case X86_VENDOR_AMD:
 249                 if (c->x86 >= 0x10)
 250                         ret = save_microcode_in_initrd_amd(cpuid_eax(1));
 251                 break;
 252         default:
 253                 break;
 254         }
 255
 256         initrd_gone = true;
 257
 258         return ret;
 259 }
 260
 261 struct cpio_data find_microcode_in_initrd(const char *path, bool use_pa)
 262 {
 263 #ifdef CONFIG_BLK_DEV_INITRD
 264         unsigned long start = 0;
 265         size_t size;
 266
 267 #ifdef CONFIG_X86_32
 268         struct boot_params *params;
 269
 270         if (use_pa)
 271                 params = (struct boot_params *)__pa_nodebug(&boot_params);
 272         else
 273                 params = &boot_params;
 274
 275         size = params->hdr.ramdisk_size;
 276
 277         /*
 278          * Set start only if we have an initrd image. We cannot use initrd_start
 279          * because it is not set that early yet.
 280          */
 281         if (size)
 282                 start = params->hdr.ramdisk_image;
 283
 284 # else /* CONFIG_X86_64 */
 285         size  = (unsigned long)boot_params.ext_ramdisk_size << 32;
 286         size |= boot_params.hdr.ramdisk_size;
 287
 288         if (size) {
 289                 start  = (unsigned long)boot_params.ext_ramdisk_image << 32;
 290                 start |= boot_params.hdr.ramdisk_image;
 291
 292                 start += PAGE_OFFSET;
 293         }
 294 # endif
 295
 296         /*
 297          * Fixup the start address: after reserve_initrd() runs, initrd_start
 298          * has the virtual address of the beginning of the initrd. It also
 299          * possibly relocates the ramdisk. In either case, initrd_start contains
 300          * the updated address so use that instead.
 301          *
 302          * initrd_gone is for the hotplug case where we've thrown out initrd
 303          * already.
 304          */
 305         if (!use_pa) {
 306                 if (initrd_gone)
 307                         return (struct cpio_data){ NULL, 0, "" };
 308                 if (initrd_start)
 309                         start = initrd_start;
 310         } else {
 311                 /*
 312                  * The picture with physical addresses is a bit different: we
 313                  * need to get the *physical* address to which the ramdisk was
 314                  * relocated, i.e., relocated_ramdisk (not initrd_start) and
 315                  * since we're running from physical addresses, we need to access
 316                  * relocated_ramdisk through its *physical* address too.
 317                  */
 318                 u64 *rr = (u64 *)__pa_nodebug(&relocated_ramdisk);
 319                 if (*rr)
 320                         start = *rr;
 321         }
 322
 323         return find_cpio_data(path, (void *)start, size, NULL);
 324 #else /* !CONFIG_BLK_DEV_INITRD */
 325         return (struct cpio_data){ NULL, 0, "" };
 326 #endif
 327 }
 328
 329 void reload_early_microcode(void)
 330 {
 331         int vendor, family;
 332
 333         vendor = x86_cpuid_vendor();
 334         family = x86_cpuid_family();
 335
 336         switch (vendor) {
 337         case X86_VENDOR_INTEL:
 338                 if (family >= 6)
 339                         reload_ucode_intel();
 340                 break;
 341         case X86_VENDOR_AMD:
 342                 if (family >= 0x10)
 343                         reload_ucode_amd();
 344                 break;
 345         default:
 346                 break;
 347         }
 348 }
 349
 350 static void collect_cpu_info_local(void *arg)
 351 {
 352         struct cpu_info_ctx *ctx = arg;
 353
 354         ctx->err = microcode_ops->collect_cpu_info(smp_processor_id(),
 355                                                    ctx->cpu_sig);
 356 }
 357
 358 static int collect_cpu_info_on_target(int cpu, struct cpu_signature *cpu_sig)
 359 {
 360         struct cpu_info_ctx ctx = { .cpu_sig = cpu_sig, .err = 0 };
 361         int ret;
 362
 363         ret = smp_call_function_single(cpu, collect_cpu_info_local, &ctx, 1);
 364         if (!ret)
 365                 ret = ctx.err;
 366
 367         return ret;
 368 }
 369
 370 static int collect_cpu_info(int cpu)
 371 {
 372         struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
 373         int ret;
 374
 375         memset(uci, 0, sizeof(*uci));
 376
 377         ret = collect_cpu_info_on_target(cpu, &uci->cpu_sig);
 378         if (!ret)
 379                 uci->valid = 1;
 380
 381         return ret;
 382 }
 383
 384 static void apply_microcode_local(void *arg)
 385 {
 386         enum ucode_state *err = arg;
 387
 388         *err = microcode_ops->apply_microcode(smp_processor_id());
 389 }
 390
 391 static int apply_microcode_on_target(int cpu)
 392 {
 393         enum ucode_state err;
 394         int ret;
 395
 396         ret = smp_call_function_single(cpu, apply_microcode_local, &err, 1);
 397         if (!ret) {
 398                 if (err == UCODE_ERROR)
 399                         ret = 1;
 400         }
 401         return ret;
 402 }
 403
 404 #ifdef CONFIG_MICROCODE_OLD_INTERFACE
 405 static int do_microcode_update(const void __user *buf, size_t size)
 406 {
 407         int error = 0;
 408         int cpu;
 409
 410         for_each_online_cpu(cpu) {
 411                 struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
 412                 enum ucode_state ustate;
 413
 414                 if (!uci->valid)
 415                         continue;
 416
 417                 ustate = microcode_ops->request_microcode_user(cpu, buf, size);
 418                 if (ustate == UCODE_ERROR) {
 419                         error = -1;
 420                         break;
 421                 } else if (ustate == UCODE_NEW) {
 422                         apply_microcode_on_target(cpu);
 423                 }
 424         }
 425
 426         return error;
 427 }
 428
 429 static int microcode_open(struct inode *inode, struct file *file)
 430 {
 431         return capable(CAP_SYS_RAWIO) ? stream_open(inode, file) : -EPERM;
 432 }
 433
 434 static ssize_t microcode_write(struct file *file, const char __user *buf,
 435                                size_t len, loff_t *ppos)
 436 {
 437         ssize_t ret = -EINVAL;
 438         unsigned long nr_pages = totalram_pages();
 439
 440         if ((len >> PAGE_SHIFT) > nr_pages) {
 441                 pr_err("too much data (max %ld pages)\n", nr_pages);
 442                 return ret;
 443         }
 444
 445         get_online_cpus();
 446         mutex_lock(&microcode_mutex);
 447
 448         if (do_microcode_update(buf, len) == 0)
 449                 ret = (ssize_t)len;
 450
 451         if (ret > 0)
 452                 perf_check_microcode();
 453
 454         mutex_unlock(&microcode_mutex);
 455         put_online_cpus();
 456
 457         return ret;
 458 }
 459
 460 static const struct file_operations microcode_fops = {
 461         .owner                  = THIS_MODULE,
 462         .write                  = microcode_write,
 463         .open                   = microcode_open,
 464         .llseek         = no_llseek,
 465 };
 466
 467 static struct miscdevice microcode_dev = {
 468         .minor                  = MICROCODE_MINOR,
 469         .name                   = "microcode",
 470         .nodename               = "cpu/microcode",
 471         .fops                   = &microcode_fops,
 472 };
 473
 474 static int __init microcode_dev_init(void)
 475 {
 476         int error;
 477
 478         error = misc_register(&microcode_dev);
 479         if (error) {
 480                 pr_err("can't misc_register on minor=%d\n", MICROCODE_MINOR);
 481                 return error;
 482         }
 483
 484         return 0;
 485 }
 486
 487 static void __exit microcode_dev_exit(void)
 488 {
 489         misc_deregister(&microcode_dev);
 490 }
 491 #else
 492 #define microcode_dev_init()    0
 493 #define microcode_dev_exit()    do { } while (0)
 494 #endif
 495
 496 /* fake device for request_firmware */
 497 static struct platform_device   *microcode_pdev;
 498
 499 /*
 500  * Late loading dance. Why the heavy-handed stomp_machine effort?
 501  *
 502  * - HT siblings must be idle and not execute other code while the other sibling
 503  *   is loading microcode in order to avoid any negative interactions caused by
 504  *   the loading.
 505  *
 506  * - In addition, microcode update on the cores must be serialized until this
 507  *   requirement can be relaxed in the future. Right now, this is conservative
 508  *   and good.
 509  */
 510 #define SPINUNIT 100 /* 100 nsec */
 511
 512 static int check_online_cpus(void)
 513 {
 514         unsigned int cpu;
 515
 516         /*
 517          * Make sure all CPUs are online.  It's fine for SMT to be disabled if
 518          * all the primary threads are still online.
 519          */
 520         for_each_present_cpu(cpu) {
 521                 if (topology_is_primary_thread(cpu) && !cpu_online(cpu)) {
 522                         pr_err("Not all CPUs online, aborting microcode update.\n");
 523                         return -EINVAL;
 524                 }
 525         }
 526
 527         return 0;
 528 }
 529
 530 static atomic_t late_cpus_in;
 531 static atomic_t late_cpus_out;
 532
 533 static int __wait_for_cpus(atomic_t *t, long long timeout)
 534 {
 535         int all_cpus = num_online_cpus();
 536
 537         atomic_inc(t);
 538
 539         while (atomic_read(t) < all_cpus) {
 540                 if (timeout < SPINUNIT) {
 541                         pr_err("Timeout while waiting for CPUs rendezvous, remaining: %d\n",
 542                                 all_cpus - atomic_read(t));
 543                         return 1;
 544                 }
 545
 546                 ndelay(SPINUNIT);
 547                 timeout -= SPINUNIT;
 548
 549                 touch_nmi_watchdog();
 550         }
 551         return 0;
 552 }
 553
 554 /*
 555  * Returns:
 556  * < 0 - on error
 557  *   0 - no update done
 558  *   1 - microcode was updated
 559  */
 560 static int __reload_late(void *info)
 561 {
 562         int cpu = smp_processor_id();
 563         enum ucode_state err;
 564         int ret = 0;
 565
 566         /*
 567          * Wait for all CPUs to arrive. A load will not be attempted unless all
 568          * CPUs show up.
 569          * */
 570         if (__wait_for_cpus(&late_cpus_in, NSEC_PER_SEC))
 571                 return -1;
 572
 573         raw_spin_lock(&update_lock);
 574         apply_microcode_local(&err);
 575         raw_spin_unlock(&update_lock);
 576
 577         /* siblings return UCODE_OK because their engine got updated already */
 578         if (err > UCODE_NFOUND) {
 579                 pr_warn("Error reloading microcode on CPU %d\n", cpu);
 580                 ret = -1;
 581         } else if (err == UCODE_UPDATED || err == UCODE_OK) {
 582                 ret = 1;
 583         }
 584
 585         /*
 586          * Increase the wait timeout to a safe value here since we're
 587          * serializing the microcode update and that could take a while on a
 588          * large number of CPUs. And that is fine as the *actual* timeout will
 589          * be determined by the last CPU finished updating and thus cut short.
 590          */
 591         if (__wait_for_cpus(&late_cpus_out, NSEC_PER_SEC * num_online_cpus()))
 592                 panic("Timeout during microcode update!\n");
 593
 594         return ret;
 595 }
 596
 597 /*
 598  * Reload microcode late on all CPUs. Wait for a sec until they
 599  * all gather together.
 600  */
 601 static int microcode_reload_late(void)
 602 {
 603         int ret;
 604
 605         atomic_set(&late_cpus_in,  0);
 606         atomic_set(&late_cpus_out, 0);
 607
 608         ret = stop_machine_cpuslocked(__reload_late, NULL, cpu_online_mask);
 609         if (ret > 0)
 610                 microcode_check();
 611
 612         pr_info("Reload completed, microcode revision: 0x%x\n", boot_cpu_data.microcode);
 613
 614         return ret;
 615 }
 616
 617 static ssize_t reload_store(struct device *dev,
 618                             struct device_attribute *attr,
 619                             const char *buf, size_t size)
 620 {
 621         enum ucode_state tmp_ret = UCODE_OK;
 622         int bsp = boot_cpu_data.cpu_index;
 623         unsigned long val;
 624         ssize_t ret = 0;
 625
 626         ret = kstrtoul(buf, 0, &val);
 627         if (ret)
 628                 return ret;
 629
 630         if (val != 1)
 631                 return size;
 632
 633         tmp_ret = microcode_ops->request_microcode_fw(bsp, &microcode_pdev->dev, true);
 634         if (tmp_ret != UCODE_NEW)
 635                 return size;
 636
 637         get_online_cpus();
 638
 639         ret = check_online_cpus();
 640         if (ret)
 641                 goto put;
 642
 643         mutex_lock(&microcode_mutex);
 644         ret = microcode_reload_late();
 645         mutex_unlock(&microcode_mutex);
 646
 647 put:
 648         put_online_cpus();
 649
 650         if (ret >= 0)
 651                 ret = size;
 652
 653         return ret;
 654 }
 655
 656 static ssize_t version_show(struct device *dev,
 657                         struct device_attribute *attr, char *buf)
 658 {
 659         struct ucode_cpu_info *uci = ucode_cpu_info + dev->id;
 660
 661         return sprintf(buf, "0x%x\n", uci->cpu_sig.rev);
 662 }
 663
 664 static ssize_t pf_show(struct device *dev,
 665                         struct device_attribute *attr, char *buf)
 666 {
 667         struct ucode_cpu_info *uci = ucode_cpu_info + dev->id;
 668
 669         return sprintf(buf, "0x%x\n", uci->cpu_sig.pf);
 670 }
 671
 672 static DEVICE_ATTR_WO(reload);
 673 static DEVICE_ATTR(version, 0444, version_show, NULL);
 674 static DEVICE_ATTR(processor_flags, 0444, pf_show, NULL);
 675
 676 static struct attribute *mc_default_attrs[] = {
 677         &dev_attr_version.attr,
 678         &dev_attr_processor_flags.attr,
 679         NULL
 680 };
 681
 682 static const struct attribute_group mc_attr_group = {
 683         .attrs                  = mc_default_attrs,
 684         .name                   = "microcode",
 685 };
 686
 687 static void microcode_fini_cpu(int cpu)
 688 {
 689         if (microcode_ops->microcode_fini_cpu)
 690                 microcode_ops->microcode_fini_cpu(cpu);
 691 }
 692
 693 static enum ucode_state microcode_resume_cpu(int cpu)
 694 {
 695         if (apply_microcode_on_target(cpu))
 696                 return UCODE_ERROR;
 697
 698         pr_debug("CPU%d updated upon resume\n", cpu);
 699
 700         return UCODE_OK;
 701 }
 702
 703 static enum ucode_state microcode_init_cpu(int cpu, bool refresh_fw)
 704 {
 705         enum ucode_state ustate;
 706         struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
 707
 708         if (uci->valid)
 709                 return UCODE_OK;
 710
 711         if (collect_cpu_info(cpu))
 712                 return UCODE_ERROR;
 713
 714         /* --dimm. Trigger a delayed update? */
 715         if (system_state != SYSTEM_RUNNING)
 716                 return UCODE_NFOUND;
 717
 718         ustate = microcode_ops->request_microcode_fw(cpu, &microcode_pdev->dev, refresh_fw);
 719         if (ustate == UCODE_NEW) {
 720                 pr_debug("CPU%d updated upon init\n", cpu);
 721                 apply_microcode_on_target(cpu);
 722         }
 723
 724         return ustate;
 725 }
 726
 727 static enum ucode_state microcode_update_cpu(int cpu)
 728 {
 729         struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
 730
 731         /* Refresh CPU microcode revision after resume. */
 732         collect_cpu_info(cpu);
 733
 734         if (uci->valid)
 735                 return microcode_resume_cpu(cpu);
 736
 737         return microcode_init_cpu(cpu, false);
 738 }
 739
 740 static int mc_device_add(struct device *dev, struct subsys_interface *sif)
 741 {
 742         int err, cpu = dev->id;
 743
 744         if (!cpu_online(cpu))
 745                 return 0;
 746
 747         pr_debug("CPU%d added\n", cpu);
 748
 749         err = sysfs_create_group(&dev->kobj, &mc_attr_group);
 750         if (err)
 751                 return err;
 752
 753         if (microcode_init_cpu(cpu, true) == UCODE_ERROR)
 754                 return -EINVAL;
 755
 756         return err;
 757 }
 758
 759 static void mc_device_remove(struct device *dev, struct subsys_interface *sif)
 760 {
 761         int cpu = dev->id;
 762
 763         if (!cpu_online(cpu))
 764                 return;
 765
 766         pr_debug("CPU%d removed\n", cpu);
 767         microcode_fini_cpu(cpu);
 768         sysfs_remove_group(&dev->kobj, &mc_attr_group);
 769 }
 770
 771 static struct subsys_interface mc_cpu_interface = {
 772         .name                   = "microcode",
 773         .subsys                 = &cpu_subsys,
 774         .add_dev                = mc_device_add,
 775         .remove_dev             = mc_device_remove,
 776 };
 777
 778 /**
 779  * mc_bp_resume - Update boot CPU microcode during resume.
 780  */
 781 static void mc_bp_resume(void)
 782 {
 783         int cpu = smp_processor_id();
 784         struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
 785
 786         if (uci->valid && uci->mc)
 787                 microcode_ops->apply_microcode(cpu);
 788         else if (!uci->mc)
 789                 reload_early_microcode();
 790 }
 791
 792 static struct syscore_ops mc_syscore_ops = {
 793         .resume                 = mc_bp_resume,
 794 };
 795
 796 static int mc_cpu_online(unsigned int cpu)
 797 {
 798         struct device *dev;
 799
 800         dev = get_cpu_device(cpu);
 801         microcode_update_cpu(cpu);
 802         pr_debug("CPU%d added\n", cpu);
 803
 804         if (sysfs_create_group(&dev->kobj, &mc_attr_group))
 805                 pr_err("Failed to create group for CPU%d\n", cpu);
 806         return 0;
 807 }
 808
 809 static int mc_cpu_down_prep(unsigned int cpu)
 810 {
 811         struct device *dev;
 812
 813         dev = get_cpu_device(cpu);
 814         /* Suspend is in progress, only remove the interface */
 815         sysfs_remove_group(&dev->kobj, &mc_attr_group);
 816         pr_debug("CPU%d removed\n", cpu);
 817
 818         return 0;
 819 }
 820
 821 static struct attribute *cpu_root_microcode_attrs[] = {
 822         &dev_attr_reload.attr,
 823         NULL
 824 };
 825
 826 static const struct attribute_group cpu_root_microcode_group = {
 827         .name  = "microcode",
 828         .attrs = cpu_root_microcode_attrs,
 829 };
 830
 831 int __init microcode_init(void)
 832 {
 833         struct cpuinfo_x86 *c = &boot_cpu_data;
 834         int error;
 835
 836         if (dis_ucode_ldr)
 837                 return -EINVAL;
 838
 839         if (c->x86_vendor == X86_VENDOR_INTEL)
 840                 microcode_ops = init_intel_microcode();
 841         else if (c->x86_vendor == X86_VENDOR_AMD)
 842                 microcode_ops = init_amd_microcode();
 843         else
 844                 pr_err("no support for this CPU vendor\n");
 845
 846         if (!microcode_ops)
 847                 return -ENODEV;
 848
 849         microcode_pdev = platform_device_register_simple("microcode", -1,
 850                                                          NULL, 0);
 851         if (IS_ERR(microcode_pdev))
 852                 return PTR_ERR(microcode_pdev);
 853
 854         get_online_cpus();
 855         mutex_lock(&microcode_mutex);
 856
 857         error = subsys_interface_register(&mc_cpu_interface);
 858         if (!error)
 859                 perf_check_microcode();
 860         mutex_unlock(&microcode_mutex);
 861         put_online_cpus();
 862
 863         if (error)
 864                 goto out_pdev;
 865
 866         error = sysfs_create_group(&cpu_subsys.dev_root->kobj,
 867                                    &cpu_root_microcode_group);
 868
 869         if (error) {
 870                 pr_err("Error creating microcode group!\n");
 871                 goto out_driver;
 872         }
 873
 874         error = microcode_dev_init();
 875         if (error)
 876                 goto out_ucode_group;
 877
 878         register_syscore_ops(&mc_syscore_ops);
 879         cpuhp_setup_state_nocalls(CPUHP_AP_ONLINE_DYN, "x86/microcode:online",
 880                                   mc_cpu_online, mc_cpu_down_prep);
 881
 882         pr_info("Microcode Update Driver: v%s.", DRIVER_VERSION);
 883
 884         return 0;
 885
 886  out_ucode_group:
 887         sysfs_remove_group(&cpu_subsys.dev_root->kobj,
 888                            &cpu_root_microcode_group);
 889
 890  out_driver:
 891         get_online_cpus();
 892         mutex_lock(&microcode_mutex);
 893
 894         subsys_interface_unregister(&mc_cpu_interface);
 895
 896         mutex_unlock(&microcode_mutex);
 897         put_online_cpus();
 898
 899  out_pdev:
 900         platform_device_unregister(microcode_pdev);
 901         return error;
 902
 903 }
 904 fs_initcall(save_microcode_in_initrd);
 905 late_initcall(microcode_init);