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_OK)
 422                         apply_microcode_on_target(cpu);
 423         }
 424
 425         return error;
 426 }
 427
 428 static int microcode_open(struct inode *inode, struct file *file)
 429 {
 430         return capable(CAP_SYS_RAWIO) ? nonseekable_open(inode, file) : -EPERM;
 431 }
 432
 433 static ssize_t microcode_write(struct file *file, const char __user *buf,
 434                                size_t len, loff_t *ppos)
 435 {
 436         ssize_t ret = -EINVAL;
 437         unsigned long nr_pages = totalram_pages();
 438
 439         if ((len >> PAGE_SHIFT) > nr_pages) {
 440                 pr_err("too much data (max %ld pages)\n", nr_pages);
 441                 return ret;
 442         }
 443
 444         get_online_cpus();
 445         mutex_lock(&microcode_mutex);
 446
 447         if (do_microcode_update(buf, len) == 0)
 448                 ret = (ssize_t)len;
 449
 450         if (ret > 0)
 451                 perf_check_microcode();
 452
 453         mutex_unlock(&microcode_mutex);
 454         put_online_cpus();
 455
 456         return ret;
 457 }
 458
 459 static const struct file_operations microcode_fops = {
 460         .owner                  = THIS_MODULE,
 461         .write                  = microcode_write,
 462         .open                   = microcode_open,
 463         .llseek         = no_llseek,
 464 };
 465
 466 static struct miscdevice microcode_dev = {
 467         .minor                  = MICROCODE_MINOR,
 468         .name                   = "microcode",
 469         .nodename               = "cpu/microcode",
 470         .fops                   = &microcode_fops,
 471 };
 472
 473 static int __init microcode_dev_init(void)
 474 {
 475         int error;
 476
 477         error = misc_register(&microcode_dev);
 478         if (error) {
 479                 pr_err("can't misc_register on minor=%d\n", MICROCODE_MINOR);
 480                 return error;
 481         }
 482
 483         return 0;
 484 }
 485
 486 static void __exit microcode_dev_exit(void)
 487 {
 488         misc_deregister(&microcode_dev);
 489 }
 490 #else
 491 #define microcode_dev_init()    0
 492 #define microcode_dev_exit()    do { } while (0)
 493 #endif
 494
 495 /* fake device for request_firmware */
 496 static struct platform_device   *microcode_pdev;
 497
 498 /*
 499  * Late loading dance. Why the heavy-handed stomp_machine effort?
 500  *
 501  * - HT siblings must be idle and not execute other code while the other sibling
 502  *   is loading microcode in order to avoid any negative interactions caused by
 503  *   the loading.
 504  *
 505  * - In addition, microcode update on the cores must be serialized until this
 506  *   requirement can be relaxed in the future. Right now, this is conservative
 507  *   and good.
 508  */
 509 #define SPINUNIT 100 /* 100 nsec */
 510
 511 static int check_online_cpus(void)
 512 {
 513         unsigned int cpu;
 514
 515         /*
 516          * Make sure all CPUs are online.  It's fine for SMT to be disabled if
 517          * all the primary threads are still online.
 518          */
 519         for_each_present_cpu(cpu) {
 520                 if (topology_is_primary_thread(cpu) && !cpu_online(cpu)) {
 521                         pr_err("Not all CPUs online, aborting microcode update.\n");
 522                         return -EINVAL;
 523                 }
 524         }
 525
 526         return 0;
 527 }
 528
 529 static atomic_t late_cpus_in;
 530 static atomic_t late_cpus_out;
 531
 532 static int __wait_for_cpus(atomic_t *t, long long timeout)
 533 {
 534         int all_cpus = num_online_cpus();
 535
 536         atomic_inc(t);
 537
 538         while (atomic_read(t) < all_cpus) {
 539                 if (timeout < SPINUNIT) {
 540                         pr_err("Timeout while waiting for CPUs rendezvous, remaining: %d\n",
 541                                 all_cpus - atomic_read(t));
 542                         return 1;
 543                 }
 544
 545                 ndelay(SPINUNIT);
 546                 timeout -= SPINUNIT;
 547
 548                 touch_nmi_watchdog();
 549         }
 550         return 0;
 551 }
 552
 553 /*
 554  * Returns:
 555  * < 0 - on error
 556  *   0 - no update done
 557  *   1 - microcode was updated
 558  */
 559 static int __reload_late(void *info)
 560 {
 561         int cpu = smp_processor_id();
 562         enum ucode_state err;
 563         int ret = 0;
 564
 565         /*
 566          * Wait for all CPUs to arrive. A load will not be attempted unless all
 567          * CPUs show up.
 568          * */
 569         if (__wait_for_cpus(&late_cpus_in, NSEC_PER_SEC))
 570                 return -1;
 571
 572         raw_spin_lock(&update_lock);
 573         apply_microcode_local(&err);
 574         raw_spin_unlock(&update_lock);
 575
 576         /* siblings return UCODE_OK because their engine got updated already */
 577         if (err > UCODE_NFOUND) {
 578                 pr_warn("Error reloading microcode on CPU %d\n", cpu);
 579                 ret = -1;
 580         } else if (err == UCODE_UPDATED || err == UCODE_OK) {
 581                 ret = 1;
 582         }
 583
 584         /*
 585          * Increase the wait timeout to a safe value here since we're
 586          * serializing the microcode update and that could take a while on a
 587          * large number of CPUs. And that is fine as the *actual* timeout will
 588          * be determined by the last CPU finished updating and thus cut short.
 589          */
 590         if (__wait_for_cpus(&late_cpus_out, NSEC_PER_SEC * num_online_cpus()))
 591                 panic("Timeout during microcode update!\n");
 592
 593         return ret;
 594 }
 595
 596 /*
 597  * Reload microcode late on all CPUs. Wait for a sec until they
 598  * all gather together.
 599  */
 600 static int microcode_reload_late(void)
 601 {
 602         int ret;
 603
 604         atomic_set(&late_cpus_in,  0);
 605         atomic_set(&late_cpus_out, 0);
 606
 607         ret = stop_machine_cpuslocked(__reload_late, NULL, cpu_online_mask);
 608         if (ret > 0)
 609                 microcode_check();
 610
 611         pr_info("Reload completed, microcode revision: 0x%x\n", boot_cpu_data.microcode);
 612
 613         return ret;
 614 }
 615
 616 static ssize_t reload_store(struct device *dev,
 617                             struct device_attribute *attr,
 618                             const char *buf, size_t size)
 619 {
 620         enum ucode_state tmp_ret = UCODE_OK;
 621         int bsp = boot_cpu_data.cpu_index;
 622         unsigned long val;
 623         ssize_t ret = 0;
 624
 625         ret = kstrtoul(buf, 0, &val);
 626         if (ret)
 627                 return ret;
 628
 629         if (val != 1)
 630                 return size;
 631
 632         tmp_ret = microcode_ops->request_microcode_fw(bsp, &microcode_pdev->dev, true);
 633         if (tmp_ret != UCODE_NEW)
 634                 return size;
 635
 636         get_online_cpus();
 637
 638         ret = check_online_cpus();
 639         if (ret)
 640                 goto put;
 641
 642         mutex_lock(&microcode_mutex);
 643         ret = microcode_reload_late();
 644         mutex_unlock(&microcode_mutex);
 645
 646 put:
 647         put_online_cpus();
 648
 649         if (ret >= 0)
 650                 ret = size;
 651
 652         return ret;
 653 }
 654
 655 static ssize_t version_show(struct device *dev,
 656                         struct device_attribute *attr, char *buf)
 657 {
 658         struct ucode_cpu_info *uci = ucode_cpu_info + dev->id;
 659
 660         return sprintf(buf, "0x%x\n", uci->cpu_sig.rev);
 661 }
 662
 663 static ssize_t pf_show(struct device *dev,
 664                         struct device_attribute *attr, char *buf)
 665 {
 666         struct ucode_cpu_info *uci = ucode_cpu_info + dev->id;
 667
 668         return sprintf(buf, "0x%x\n", uci->cpu_sig.pf);
 669 }
 670
 671 static DEVICE_ATTR_WO(reload);
 672 static DEVICE_ATTR(version, 0444, version_show, NULL);
 673 static DEVICE_ATTR(processor_flags, 0444, pf_show, NULL);
 674
 675 static struct attribute *mc_default_attrs[] = {
 676         &dev_attr_version.attr,
 677         &dev_attr_processor_flags.attr,
 678         NULL
 679 };
 680
 681 static const struct attribute_group mc_attr_group = {
 682         .attrs                  = mc_default_attrs,
 683         .name                   = "microcode",
 684 };
 685
 686 static void microcode_fini_cpu(int cpu)
 687 {
 688         if (microcode_ops->microcode_fini_cpu)
 689                 microcode_ops->microcode_fini_cpu(cpu);
 690 }
 691
 692 static enum ucode_state microcode_resume_cpu(int cpu)
 693 {
 694         if (apply_microcode_on_target(cpu))
 695                 return UCODE_ERROR;
 696
 697         pr_debug("CPU%d updated upon resume\n", cpu);
 698
 699         return UCODE_OK;
 700 }
 701
 702 static enum ucode_state microcode_init_cpu(int cpu, bool refresh_fw)
 703 {
 704         enum ucode_state ustate;
 705         struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
 706
 707         if (uci->valid)
 708                 return UCODE_OK;
 709
 710         if (collect_cpu_info(cpu))
 711                 return UCODE_ERROR;
 712
 713         /* --dimm. Trigger a delayed update? */
 714         if (system_state != SYSTEM_RUNNING)
 715                 return UCODE_NFOUND;
 716
 717         ustate = microcode_ops->request_microcode_fw(cpu, &microcode_pdev->dev, refresh_fw);
 718         if (ustate == UCODE_NEW) {
 719                 pr_debug("CPU%d updated upon init\n", cpu);
 720                 apply_microcode_on_target(cpu);
 721         }
 722
 723         return ustate;
 724 }
 725
 726 static enum ucode_state microcode_update_cpu(int cpu)
 727 {
 728         struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
 729
 730         /* Refresh CPU microcode revision after resume. */
 731         collect_cpu_info(cpu);
 732
 733         if (uci->valid)
 734                 return microcode_resume_cpu(cpu);
 735
 736         return microcode_init_cpu(cpu, false);
 737 }
 738
 739 static int mc_device_add(struct device *dev, struct subsys_interface *sif)
 740 {
 741         int err, cpu = dev->id;
 742
 743         if (!cpu_online(cpu))
 744                 return 0;
 745
 746         pr_debug("CPU%d added\n", cpu);
 747
 748         err = sysfs_create_group(&dev->kobj, &mc_attr_group);
 749         if (err)
 750                 return err;
 751
 752         if (microcode_init_cpu(cpu, true) == UCODE_ERROR)
 753                 return -EINVAL;
 754
 755         return err;
 756 }
 757
 758 static void mc_device_remove(struct device *dev, struct subsys_interface *sif)
 759 {
 760         int cpu = dev->id;
 761
 762         if (!cpu_online(cpu))
 763                 return;
 764
 765         pr_debug("CPU%d removed\n", cpu);
 766         microcode_fini_cpu(cpu);
 767         sysfs_remove_group(&dev->kobj, &mc_attr_group);
 768 }
 769
 770 static struct subsys_interface mc_cpu_interface = {
 771         .name                   = "microcode",
 772         .subsys                 = &cpu_subsys,
 773         .add_dev                = mc_device_add,
 774         .remove_dev             = mc_device_remove,
 775 };
 776
 777 /**
 778  * mc_bp_resume - Update boot CPU microcode during resume.
 779  */
 780 static void mc_bp_resume(void)
 781 {
 782         int cpu = smp_processor_id();
 783         struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
 784
 785         if (uci->valid && uci->mc)
 786                 microcode_ops->apply_microcode(cpu);
 787         else if (!uci->mc)
 788                 reload_early_microcode();
 789 }
 790
 791 static struct syscore_ops mc_syscore_ops = {
 792         .resume                 = mc_bp_resume,
 793 };
 794
 795 static int mc_cpu_online(unsigned int cpu)
 796 {
 797         struct device *dev;
 798
 799         dev = get_cpu_device(cpu);
 800         microcode_update_cpu(cpu);
 801         pr_debug("CPU%d added\n", cpu);
 802
 803         if (sysfs_create_group(&dev->kobj, &mc_attr_group))
 804                 pr_err("Failed to create group for CPU%d\n", cpu);
 805         return 0;
 806 }
 807
 808 static int mc_cpu_down_prep(unsigned int cpu)
 809 {
 810         struct device *dev;
 811
 812         dev = get_cpu_device(cpu);
 813         /* Suspend is in progress, only remove the interface */
 814         sysfs_remove_group(&dev->kobj, &mc_attr_group);
 815         pr_debug("CPU%d removed\n", cpu);
 816
 817         return 0;
 818 }
 819
 820 static struct attribute *cpu_root_microcode_attrs[] = {
 821         &dev_attr_reload.attr,
 822         NULL
 823 };
 824
 825 static const struct attribute_group cpu_root_microcode_group = {
 826         .name  = "microcode",
 827         .attrs = cpu_root_microcode_attrs,
 828 };
 829
 830 int __init microcode_init(void)
 831 {
 832         struct cpuinfo_x86 *c = &boot_cpu_data;
 833         int error;
 834
 835         if (dis_ucode_ldr)
 836                 return -EINVAL;
 837
 838         if (c->x86_vendor == X86_VENDOR_INTEL)
 839                 microcode_ops = init_intel_microcode();
 840         else if (c->x86_vendor == X86_VENDOR_AMD)
 841                 microcode_ops = init_amd_microcode();
 842         else
 843                 pr_err("no support for this CPU vendor\n");
 844
 845         if (!microcode_ops)
 846                 return -ENODEV;
 847
 848         microcode_pdev = platform_device_register_simple("microcode", -1,
 849                                                          NULL, 0);
 850         if (IS_ERR(microcode_pdev))
 851                 return PTR_ERR(microcode_pdev);
 852
 853         get_online_cpus();
 854         mutex_lock(&microcode_mutex);
 855
 856         error = subsys_interface_register(&mc_cpu_interface);
 857         if (!error)
 858                 perf_check_microcode();
 859         mutex_unlock(&microcode_mutex);
 860         put_online_cpus();
 861
 862         if (error)
 863                 goto out_pdev;
 864
 865         error = sysfs_create_group(&cpu_subsys.dev_root->kobj,
 866                                    &cpu_root_microcode_group);
 867
 868         if (error) {
 869                 pr_err("Error creating microcode group!\n");
 870                 goto out_driver;
 871         }
 872
 873         error = microcode_dev_init();
 874         if (error)
 875                 goto out_ucode_group;
 876
 877         register_syscore_ops(&mc_syscore_ops);
 878         cpuhp_setup_state_nocalls(CPUHP_AP_ONLINE_DYN, "x86/microcode:online",
 879                                   mc_cpu_online, mc_cpu_down_prep);
 880
 881         pr_info("Microcode Update Driver: v%s.", DRIVER_VERSION);
 882
 883         return 0;
 884
 885  out_ucode_group:
 886         sysfs_remove_group(&cpu_subsys.dev_root->kobj,
 887                            &cpu_root_microcode_group);
 888
 889  out_driver:
 890         get_online_cpus();
 891         mutex_lock(&microcode_mutex);
 892
 893         subsys_interface_unregister(&mc_cpu_interface);
 894
 895         mutex_unlock(&microcode_mutex);
 896         put_online_cpus();
 897
 898  out_pdev:
 899         platform_device_unregister(microcode_pdev);
 900         return error;
 901
 902 }
 903 fs_initcall(save_microcode_in_initrd);
 904 late_initcall(microcode_init);