--- /dev/null
+/*
+ *  AMD CPU Microcode Update Driver for Linux
+ *  Copyright (C) 2008 Advanced Micro Devices Inc.
+ *
+ *  Author: Peter Oruba <peter.oruba@amd.com>
+ *
+ *  Based on work by:
+ *  Tigran Aivazian <tigran@aivazian.fsnet.co.uk>
+ *
+ *  This driver allows to upgrade microcode on AMD
+ *  family 0x10 and 0x11 processors.
+ *
+ *  Licensed unter the terms of the GNU General Public
+ *  License version 2. See file COPYING for details.
+*/
+
+#include <linux/capability.h>
+#include <linux/kernel.h>
+#include <linux/init.h>
+#include <linux/sched.h>
+#include <linux/cpumask.h>
+#include <linux/module.h>
+#include <linux/slab.h>
+#include <linux/vmalloc.h>
+#include <linux/miscdevice.h>
+#include <linux/spinlock.h>
+#include <linux/mm.h>
+#include <linux/fs.h>
+#include <linux/mutex.h>
+#include <linux/cpu.h>
+#include <linux/firmware.h>
+#include <linux/platform_device.h>
+#include <linux/pci.h>
+#include <linux/pci_ids.h>
+
+#include <asm/msr.h>
+#include <asm/uaccess.h>
+#include <asm/processor.h>
+#include <asm/microcode.h>
+
+MODULE_DESCRIPTION("AMD Microcode Update Driver");
+MODULE_AUTHOR("Peter Oruba <peter.oruba@amd.com>");
+MODULE_LICENSE("GPLv2");
+
+#define UCODE_MAGIC                0x00414d44
+#define UCODE_EQUIV_CPU_TABLE_TYPE 0x00000000
+#define UCODE_UCODE_TYPE           0x00000001
+
+#define UCODE_MAX_SIZE          (2048)
+#define DEFAULT_UCODE_DATASIZE (896)     /* 896 bytes */
+#define MC_HEADER_SIZE         (sizeof(struct microcode_header_amd))     /* 64 bytes */
+#define DEFAULT_UCODE_TOTALSIZE (DEFAULT_UCODE_DATASIZE + MC_HEADER_SIZE) /* 960 bytes */
+#define DWSIZE                 (sizeof(u32))
+/* For now we support a fixed ucode total size only */
+#define get_totalsize(mc) \
+       ((((struct microcode_amd *)mc)->hdr.mc_patch_data_len * 28) \
+        + MC_HEADER_SIZE)
+
+extern int microcode_init(void *opaque, struct module *module);
+extern void microcode_exit(void);
+
+/* serialize access to the physical write */
+static DEFINE_SPINLOCK(microcode_update_lock);
+
+/* no concurrent ->write()s are allowed on /dev/cpu/microcode */
+extern struct mutex (microcode_mutex);
+
+struct equiv_cpu_entry *equiv_cpu_table;
+
+extern struct ucode_cpu_info ucode_cpu_info[NR_CPUS];
+
+static void collect_cpu_info_amd(int cpu)
+{
+       struct cpuinfo_x86 *c = &cpu_data(cpu);
+       struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
+
+       /* We should bind the task to the CPU */
+       BUG_ON(raw_smp_processor_id() != cpu);
+       uci->rev = 0;
+       uci->pf = 0;
+       uci->mc.mc_amd = NULL;
+       uci->valid = 1;
+
+       if (c->x86_vendor != X86_VENDOR_AMD || c->x86 < 0x10) {
+               printk(KERN_ERR "microcode: CPU%d not a capable AMD processor\n",
+                      cpu);
+               uci->valid = 0;
+               return;
+       }
+
+       asm volatile("movl %1, %%ecx; rdmsr"
+                    : "=a" (uci->rev)
+                    : "i" (0x0000008B) : "ecx");
+
+       printk(KERN_INFO "microcode: collect_cpu_info_amd : patch_id=0x%x\n",
+              uci->rev);
+}
+
+static int get_matching_microcode_amd(void *mc, int cpu)
+{
+       struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
+       struct microcode_header_amd *mc_header = mc;
+       unsigned long total_size = get_totalsize(mc_header);
+       void *new_mc;
+       struct pci_dev *nb_pci_dev, *sb_pci_dev;
+       unsigned int current_cpu_id;
+       unsigned int equiv_cpu_id = 0x00;
+       unsigned int i = 0;
+
+       /* We should bind the task to the CPU */
+       BUG_ON(cpu != raw_smp_processor_id());
+
+       /* This is a tricky part. We might be called from a write operation */
+       /* to the device file instead of the usual process of firmware */
+       /* loading. This routine needs to be able to distinguish both */
+/* cases. This is done by checking if there alread is a equivalent */
+       /* CPU table installed. If not, we're written through */
+       /* /dev/cpu/microcode. */
+/* Since we ignore all checks. The error case in which going through */
+/* firmware loading and that table is not loaded has already been */
+       /* checked earlier. */
+       if (equiv_cpu_table == NULL) {
+               printk(KERN_INFO "microcode: CPU%d microcode update with "
+                      "version 0x%x (current=0x%x)\n",
+                      cpu, mc_header->patch_id, uci->rev);
+               goto out;
+       }
+
+       current_cpu_id = cpuid_eax(0x00000001);
+
+       while (equiv_cpu_table[i].installed_cpu != 0) {
+               if (current_cpu_id == equiv_cpu_table[i].installed_cpu) {
+                       equiv_cpu_id = equiv_cpu_table[i].equiv_cpu;
+                       break;
+               }
+               i++;
+       }
+
+       if (!equiv_cpu_id) {
+               printk(KERN_ERR "microcode: CPU%d cpu_id "
+                      "not found in equivalent cpu table \n", cpu);
+               return 0;
+       }
+
+       if ((mc_header->processor_rev_id[0]) != (equiv_cpu_id & 0xff)) {
+               printk(KERN_ERR
+                       "microcode: CPU%d patch does not match "
+                       "(patch is %x, cpu extended is %x) \n",
+                       cpu, mc_header->processor_rev_id[0],
+                       (equiv_cpu_id & 0xff));
+               return 0;
+       }
+
+       if ((mc_header->processor_rev_id[1]) != ((equiv_cpu_id >> 16) & 0xff)) {
+               printk(KERN_ERR "microcode: CPU%d patch does not match "
+                       "(patch is %x, cpu base id is %x) \n",
+                       cpu, mc_header->processor_rev_id[1],
+                       ((equiv_cpu_id >> 16) & 0xff));
+
+               return 0;
+       }
+
+       /* ucode may be northbridge specific */
+       if (mc_header->nb_dev_id) {
+               nb_pci_dev = pci_get_device(PCI_VENDOR_ID_AMD,
+                                           (mc_header->nb_dev_id & 0xff),
+                                           NULL);
+               if ((!nb_pci_dev) ||
+                   (mc_header->nb_rev_id != nb_pci_dev->revision)) {
+                       printk(KERN_ERR "microcode: CPU%d NB mismatch \n", cpu);
+                       pci_dev_put(nb_pci_dev);
+                       return 0;
+               }
+               pci_dev_put(nb_pci_dev);
+       }
+
+       /* ucode may be southbridge specific */
+       if (mc_header->sb_dev_id) {
+               sb_pci_dev = pci_get_device(PCI_VENDOR_ID_AMD,
+                                           (mc_header->sb_dev_id & 0xff),
+                                           NULL);
+               if ((!sb_pci_dev) ||
+                   (mc_header->sb_rev_id != sb_pci_dev->revision)) {
+                       printk(KERN_ERR "microcode: CPU%d SB mismatch \n", cpu);
+                       pci_dev_put(sb_pci_dev);
+                       return 0;
+               }
+               pci_dev_put(sb_pci_dev);
+       }
+
+       if (mc_header->patch_id <= uci->rev)
+               return 0;
+
+       printk(KERN_INFO "microcode: CPU%d found a matching microcode "
+              "update with version 0x%x (current=0x%x)\n",
+              cpu, mc_header->patch_id, uci->rev);
+
+out:
+       new_mc = vmalloc(UCODE_MAX_SIZE);
+       if (!new_mc) {
+               printk(KERN_ERR "microcode: error, can't allocate memory\n");
+               return -ENOMEM;
+       }
+       memset(new_mc, 0, UCODE_MAX_SIZE);
+
+       /* free previous update file */
+       vfree(uci->mc.mc_amd);
+
+       memcpy(new_mc, mc, total_size);
+
+       uci->mc.mc_amd = new_mc;
+       return 1;
+}
+
+static void apply_microcode_amd(int cpu)
+{
+       unsigned long flags;
+       unsigned int eax, edx;
+       unsigned int rev;
+       int cpu_num = raw_smp_processor_id();
+       struct ucode_cpu_info *uci = ucode_cpu_info + cpu_num;
+
+       /* We should bind the task to the CPU */
+       BUG_ON(cpu_num != cpu);
+
+       if (uci->mc.mc_amd == NULL)
+               return;
+
+       spin_lock_irqsave(µcode_update_lock, flags);
+
+       edx = (unsigned int)(((unsigned long)
+                             &(uci->mc.mc_amd->hdr.data_code)) >> 32);
+       eax = (unsigned int)(((unsigned long)
+                             &(uci->mc.mc_amd->hdr.data_code)) & 0xffffffffL);
+
+       asm volatile("movl %0, %%ecx; wrmsr" :
+                    : "i" (0xc0010020), "a" (eax), "d" (edx) : "ecx");
+
+       /* get patch id after patching */
+       asm volatile("movl %1, %%ecx; rdmsr"
+                    : "=a" (rev)
+                    : "i" (0x0000008B) : "ecx");
+
+       spin_unlock_irqrestore(µcode_update_lock, flags);
+
+       /* check current patch id and patch's id for match */
+       if (rev != uci->mc.mc_amd->hdr.patch_id) {
+               printk(KERN_ERR "microcode: CPU%d update from revision "
+                      "0x%x to 0x%x failed\n", cpu_num,
+                      uci->mc.mc_amd->hdr.patch_id, rev);
+               return;
+       }
+
+       printk(KERN_INFO "microcode: CPU%d updated from revision "
+              "0x%x to 0x%x \n",
+              cpu_num, uci->rev, uci->mc.mc_amd->hdr.patch_id);
+
+       uci->rev = rev;
+}
+
+#ifdef CONFIG_MICROCODE_OLD_INTERFACE
+extern void __user *user_buffer;        /* user area microcode data buffer */
+extern unsigned int user_buffer_size;   /* it's size */
+
+static long get_next_ucode_amd(void **mc, long offset)
+{
+       struct microcode_header_amd mc_header;
+       unsigned long total_size;
+
+       /* No more data */
+       if (offset >= user_buffer_size)
+               return 0;
+       if (copy_from_user(&mc_header, user_buffer + offset, MC_HEADER_SIZE)) {
+               printk(KERN_ERR "microcode: error! Can not read user data\n");
+               return -EFAULT;
+       }
+       total_size = get_totalsize(&mc_header);
+       if (offset + total_size > user_buffer_size) {
+               printk(KERN_ERR "microcode: error! Bad total size in microcode "
+                      "data file\n");
+               return -EINVAL;
+       }
+       *mc = vmalloc(UCODE_MAX_SIZE);
+       if (!*mc)
+               return -ENOMEM;
+       memset(*mc, 0, UCODE_MAX_SIZE);
+
+       if (copy_from_user(*mc, user_buffer + offset, total_size)) {
+               printk(KERN_ERR "microcode: error! Can not read user data\n");
+               vfree(*mc);
+               return -EFAULT;
+       }
+       return offset + total_size;
+}
+#else
+#define get_next_ucode_amd() NULL
+#endif
+
+static long get_next_ucode_from_buffer_amd(void **mc, void *buf,
+                                      unsigned long size, long offset)
+{
+       struct microcode_header_amd *mc_header;
+       unsigned long total_size;
+       unsigned char *buf_pos = buf;
+
+       /* No more data */
+       if (offset >= size)
+               return 0;
+
+       if (buf_pos[offset] != UCODE_UCODE_TYPE) {
+               printk(KERN_ERR "microcode: error! "
+                      "Wrong microcode payload type field\n");
+               return -EINVAL;
+       }
+
+       mc_header = (struct microcode_header_amd *)(&buf_pos[offset+8]);
+
+       total_size = (unsigned long) (buf_pos[offset+4] +
+                                     (buf_pos[offset+5] << 8));
+
+       printk(KERN_INFO "microcode: size %lu, total_size %lu, offset %ld\n",
+               size, total_size, offset);
+
+       if (offset + total_size > size) {
+               printk(KERN_ERR "microcode: error! Bad data in microcode data file\n");
+               return -EINVAL;
+       }
+
+       *mc = vmalloc(UCODE_MAX_SIZE);
+       if (!*mc) {
+               printk(KERN_ERR "microcode: error! "
+                      "Can not allocate memory for microcode patch\n");
+               return -ENOMEM;
+       }
+
+       memset(*mc, 0, UCODE_MAX_SIZE);
+       memcpy(*mc, buf + offset + 8, total_size);
+
+       return offset + total_size + 8;
+}
+
+static long install_equiv_cpu_table(void *buf, unsigned long size, long offset)
+{
+       unsigned int *buf_pos = buf;
+
+       /* No more data */
+       if (offset >= size)
+               return 0;
+
+       if (buf_pos[1] != UCODE_EQUIV_CPU_TABLE_TYPE) {
+               printk(KERN_ERR "microcode: error! "
+                      "Wrong microcode equivalnet cpu table type field\n");
+               return 0;
+       }
+
+       if (size == 0) {
+               printk(KERN_ERR "microcode: error! "
+                      "Wrong microcode equivalnet cpu table length\n");
+               return 0;
+       }
+
+       equiv_cpu_table = (struct equiv_cpu_entry *) vmalloc(size);
+       if (!equiv_cpu_table) {
+               printk(KERN_ERR "microcode: error, can't allocate memory for equiv CPU table\n");
+               return 0;
+       }
+
+       memset(equiv_cpu_table, 0, size);
+       memcpy(equiv_cpu_table, &buf_pos[3], size);
+
+       return size + 12; /* add header length */
+}
+
+/* fake device for request_firmware */
+extern struct platform_device *microcode_pdev;
+
+static int cpu_request_microcode_amd(int cpu)
+{
+       char name[30];
+       const struct firmware *firmware;
+       void *buf;
+       unsigned int *buf_pos;
+       unsigned long size;
+       long offset = 0;
+       int error;
+       void *mc;
+
+       /* We should bind the task to the CPU */
+       BUG_ON(cpu != raw_smp_processor_id());
+
+       sprintf(name, "amd-ucode/microcode_amd.bin");
+       error = request_firmware(&firmware, "amd-ucode/microcode_amd.bin",
+                                µcode_pdev->dev);
+       if (error) {
+               printk(KERN_ERR "microcode: ucode data file %s load failed\n",
+                      name);
+               return error;
+       }
+
+       buf_pos = buf = firmware->data;
+       size = firmware->size;
+
+       if (buf_pos[0] != UCODE_MAGIC) {
+               printk(KERN_ERR "microcode: error! Wrong microcode patch file magic\n");
+               return -EINVAL;
+       }
+
+       offset = install_equiv_cpu_table(buf, buf_pos[2], offset);
+
+       if (!offset) {
+               printk(KERN_ERR "microcode: installing equivalent cpu table failed\n");
+               return -EINVAL;
+       }
+
+       while ((offset =
+               get_next_ucode_from_buffer_amd(&mc, buf, size, offset)) > 0) {
+               error = get_matching_microcode_amd(mc, cpu);
+               if (error < 0)
+                       break;
+               /*
+                * It's possible the data file has multiple matching ucode,
+                * lets keep searching till the latest version
+                */
+               if (error == 1) {
+                       apply_microcode_amd(cpu);
+                       error = 0;
+               }
+               vfree(mc);
+       }
+       if (offset > 0) {
+               vfree(mc);
+               vfree(equiv_cpu_table);
+               equiv_cpu_table = NULL;
+       }
+       if (offset < 0)
+               error = offset;
+       release_firmware(firmware);
+
+       return error;
+}
+
+static int apply_microcode_check_cpu_amd(int cpu)
+{
+       struct cpuinfo_x86 *c = &cpu_data(cpu);
+       struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
+       unsigned int rev;
+       cpumask_t old;
+       cpumask_of_cpu_ptr(newmask, cpu);
+       int err = 0;
+
+       /* Check if the microcode is available */
+       if (!uci->mc.mc_amd)
+               return 0;
+
+       old = current->cpus_allowed;
+       set_cpus_allowed(current, newmask);
+
+       /* Check if the microcode we have in memory matches the CPU */
+       if (c->x86_vendor != X86_VENDOR_AMD || c->x86 < 16)
+               err = -EINVAL;
+
+       if (!err) {
+               asm volatile("movl %1, %%ecx; rdmsr"
+                    : "=a" (rev)
+                    : "i" (0x0000008B) : "ecx");
+
+               if (uci->rev != rev)
+                       err = -EINVAL;
+       }
+
+       if (!err)
+               apply_microcode_amd(cpu);
+       else
+               printk(KERN_ERR "microcode: Could not apply microcode to CPU%d:"
+                      " rev=0x%x\n",
+                      cpu,  uci->rev);
+
+       set_cpus_allowed(current, old);
+       return err;
+}
+
+static void microcode_fini_cpu_amd(int cpu)
+{
+       struct ucode_cpu_info *uci = ucode_cpu_info + cpu;
+
+       mutex_lock(µcode_mutex);
+       uci->valid = 0;
+       vfree(uci->mc.mc_amd);
+       uci->mc.mc_amd = NULL;
+       mutex_unlock(µcode_mutex);
+}
+
+static struct microcode_ops microcode_amd_ops = {
+       .get_next_ucode                   = get_next_ucode_amd,
+       .get_matching_microcode           = get_matching_microcode_amd,
+       .microcode_sanity_check           = NULL,
+       .apply_microcode_check_cpu        = apply_microcode_check_cpu_amd,
+       .cpu_request_microcode            = cpu_request_microcode_amd,
+       .collect_cpu_info                 = collect_cpu_info_amd,
+       .apply_microcode                  = apply_microcode_amd,
+       .microcode_fini_cpu               = microcode_fini_cpu_amd,
+};
+
+static int __init microcode_amd_module_init(void)
+{
+       struct cpuinfo_x86 *c = &cpu_data(get_cpu());
+
+       equiv_cpu_table = NULL;
+       if (c->x86_vendor == X86_VENDOR_AMD)
+               return microcode_init(µcode_amd_ops, THIS_MODULE);
+       else
+               return -ENODEV;
+}
+
+static void __exit microcode_amd_module_exit(void)
+{
+       microcode_exit();
+}
+
+module_init(microcode_amd_module_init)
+module_exit(microcode_amd_module_exit)