include cleanup: Update gfp.h and slab.h includes to prepare for breaking implicit...
[safe/jmp/linux-2.6] / arch / x86 / mm / mmio-mod.c
index 2c4baa8..3adff7d 100644 (file)
  *
  * Derived from the read-mod example from relay-examples by Tom Zanussi.
  */
+
+#define pr_fmt(fmt) "mmiotrace: " fmt
+
 #define DEBUG 1
 
 #include <linux/module.h>
 #include <linux/debugfs.h>
+#include <linux/slab.h>
 #include <linux/uaccess.h>
 #include <linux/io.h>
 #include <linux/version.h>
@@ -36,8 +40,6 @@
 
 #include "pf_in.h"
 
-#define NAME "mmiotrace: "
-
 struct trap_reason {
        unsigned long addr;
        unsigned long ip;
@@ -96,17 +98,18 @@ static void print_pte(unsigned long address)
        pte_t *pte = lookup_address(address, &level);
 
        if (!pte) {
-               pr_err(NAME "Error in %s: no pte for page 0x%08lx\n",
-                                                       __func__, address);
+               pr_err("Error in %s: no pte for page 0x%08lx\n",
+                      __func__, address);
                return;
        }
 
        if (level == PG_LEVEL_2M) {
-               pr_emerg(NAME "4MB pages are not currently supported: "
-                                                       "0x%08lx\n", address);
+               pr_emerg("4MB pages are not currently supported: 0x%08lx\n",
+                        address);
                BUG();
        }
-       pr_info(NAME "pte for 0x%lx: 0x%llx 0x%llx\n", address,
+       pr_info("pte for 0x%lx: 0x%llx 0x%llx\n",
+               address,
                (unsigned long long)pte_val(*pte),
                (unsigned long long)pte_val(*pte) & _PAGE_PRESENT);
 }
@@ -118,22 +121,21 @@ static void print_pte(unsigned long address)
 static void die_kmmio_nesting_error(struct pt_regs *regs, unsigned long addr)
 {
        const struct trap_reason *my_reason = &get_cpu_var(pf_reason);
-       pr_emerg(NAME "unexpected fault for address: 0x%08lx, "
-                                       "last fault for address: 0x%08lx\n",
-                                       addr, my_reason->addr);
+       pr_emerg("unexpected fault for address: 0x%08lx, last fault for address: 0x%08lx\n",
+                addr, my_reason->addr);
        print_pte(addr);
        print_symbol(KERN_EMERG "faulting IP is at %s\n", regs->ip);
        print_symbol(KERN_EMERG "last faulting IP was at %s\n", my_reason->ip);
 #ifdef __i386__
        pr_emerg("eax: %08lx   ebx: %08lx   ecx: %08lx   edx: %08lx\n",
-                       regs->ax, regs->bx, regs->cx, regs->dx);
+                regs->ax, regs->bx, regs->cx, regs->dx);
        pr_emerg("esi: %08lx   edi: %08lx   ebp: %08lx   esp: %08lx\n",
-                       regs->si, regs->di, regs->bp, regs->sp);
+                regs->si, regs->di, regs->bp, regs->sp);
 #else
        pr_emerg("rax: %016lx   rcx: %016lx   rdx: %016lx\n",
-                                       regs->ax, regs->cx, regs->dx);
+                regs->ax, regs->cx, regs->dx);
        pr_emerg("rsi: %016lx   rdi: %016lx   rbp: %016lx   rsp: %016lx\n",
-                               regs->si, regs->di, regs->bp, regs->sp);
+                regs->si, regs->di, regs->bp, regs->sp);
 #endif
        put_cpu_var(pf_reason);
        BUG();
@@ -213,7 +215,7 @@ static void post(struct kmmio_probe *p, unsigned long condition,
        /* this should always return the active_trace count to 0 */
        my_reason->active_traces--;
        if (my_reason->active_traces) {
-               pr_emerg(NAME "unexpected post handler");
+               pr_emerg("unexpected post handler");
                BUG();
        }
 
@@ -244,7 +246,7 @@ static void ioremap_trace_core(resource_size_t offset, unsigned long size,
        };
 
        if (!trace) {
-               pr_err(NAME "kmalloc failed in ioremap\n");
+               pr_err("kmalloc failed in ioremap\n");
                return;
        }
 
@@ -282,8 +284,8 @@ void mmiotrace_ioremap(resource_size_t offset, unsigned long size,
        if (!is_enabled()) /* recheck and proper locking in *_core() */
                return;
 
-       pr_debug(NAME "ioremap_*(0x%llx, 0x%lx) = %p\n",
-                               (unsigned long long)offset, size, addr);
+       pr_debug("ioremap_*(0x%llx, 0x%lx) = %p\n",
+                (unsigned long long)offset, size, addr);
        if ((filter_offset) && (offset != filter_offset))
                return;
        ioremap_trace_core(offset, size, addr);
@@ -301,7 +303,7 @@ static void iounmap_trace_core(volatile void __iomem *addr)
        struct remap_trace *tmp;
        struct remap_trace *found_trace = NULL;
 
-       pr_debug(NAME "Unmapping %p.\n", addr);
+       pr_debug("Unmapping %p.\n", addr);
 
        spin_lock_irq(&trace_lock);
        if (!is_enabled())
@@ -363,9 +365,8 @@ static void clear_trace_list(void)
         * Caller also ensures is_enabled() cannot change.
         */
        list_for_each_entry(trace, &trace_list, list) {
-               pr_notice(NAME "purging non-iounmapped "
-                                       "trace @0x%08lx, size 0x%lx.\n",
-                                       trace->probe.addr, trace->probe.len);
+               pr_notice("purging non-iounmapped trace @0x%08lx, size 0x%lx.\n",
+                         trace->probe.addr, trace->probe.len);
                if (!nommiotrace)
                        unregister_kmmio_probe(&trace->probe);
        }
@@ -378,30 +379,36 @@ static void clear_trace_list(void)
 }
 
 #ifdef CONFIG_HOTPLUG_CPU
-static cpumask_t downed_cpus;
+static cpumask_var_t downed_cpus;
 
 static void enter_uniprocessor(void)
 {
        int cpu;
        int err;
 
+       if (downed_cpus == NULL &&
+           !alloc_cpumask_var(&downed_cpus, GFP_KERNEL)) {
+               pr_notice("Failed to allocate mask\n");
+               goto out;
+       }
+
        get_online_cpus();
-       downed_cpus = cpu_online_map;
-       cpu_clear(first_cpu(cpu_online_map), downed_cpus);
+       cpumask_copy(downed_cpus, cpu_online_mask);
+       cpumask_clear_cpu(cpumask_first(cpu_online_mask), downed_cpus);
        if (num_online_cpus() > 1)
-               pr_notice(NAME "Disabling non-boot CPUs...\n");
+               pr_notice("Disabling non-boot CPUs...\n");
        put_online_cpus();
 
-       for_each_cpu_mask(cpu, downed_cpus) {
+       for_each_cpu(cpu, downed_cpus) {
                err = cpu_down(cpu);
                if (!err)
-                       pr_info(NAME "CPU%d is down.\n", cpu);
+                       pr_info("CPU%d is down.\n", cpu);
                else
-                       pr_err(NAME "Error taking CPU%d down: %d\n", cpu, err);
+                       pr_err("Error taking CPU%d down: %d\n", cpu, err);
        }
+out:
        if (num_online_cpus() > 1)
-               pr_warning(NAME "multiple CPUs still online, "
-                                               "may miss events.\n");
+               pr_warning("multiple CPUs still online, may miss events.\n");
 }
 
 /* __ref because leave_uniprocessor calls cpu_up which is __cpuinit,
@@ -411,15 +418,15 @@ static void __ref leave_uniprocessor(void)
        int cpu;
        int err;
 
-       if (cpus_weight(downed_cpus) == 0)
+       if (downed_cpus == NULL || cpumask_weight(downed_cpus) == 0)
                return;
-       pr_notice(NAME "Re-enabling CPUs...\n");
-       for_each_cpu_mask(cpu, downed_cpus) {
+       pr_notice("Re-enabling CPUs...\n");
+       for_each_cpu(cpu, downed_cpus) {
                err = cpu_up(cpu);
                if (!err)
-                       pr_info(NAME "enabled CPU%d.\n", cpu);
+                       pr_info("enabled CPU%d.\n", cpu);
                else
-                       pr_err(NAME "cannot re-enable CPU%d: %d\n", cpu, err);
+                       pr_err("cannot re-enable CPU%d: %d\n", cpu, err);
        }
 }
 
@@ -427,8 +434,8 @@ static void __ref leave_uniprocessor(void)
 static void enter_uniprocessor(void)
 {
        if (num_online_cpus() > 1)
-               pr_warning(NAME "multiple CPUs are online, may miss events. "
-                       "Suggest booting with maxcpus=1 kernel argument.\n");
+               pr_warning("multiple CPUs are online, may miss events. "
+                          "Suggest booting with maxcpus=1 kernel argument.\n");
 }
 
 static void leave_uniprocessor(void)
@@ -443,12 +450,13 @@ void enable_mmiotrace(void)
                goto out;
 
        if (nommiotrace)
-               pr_info(NAME "MMIO tracing disabled.\n");
+               pr_info("MMIO tracing disabled.\n");
+       kmmio_init();
        enter_uniprocessor();
        spin_lock_irq(&trace_lock);
        atomic_inc(&mmiotrace_enabled);
        spin_unlock_irq(&trace_lock);
-       pr_info(NAME "enabled.\n");
+       pr_info("enabled.\n");
 out:
        mutex_unlock(&mmiotrace_mutex);
 }
@@ -466,7 +474,8 @@ void disable_mmiotrace(void)
 
        clear_trace_list(); /* guarantees: no more kmmio callbacks */
        leave_uniprocessor();
-       pr_info(NAME "disabled.\n");
+       kmmio_cleanup();
+       pr_info("disabled.\n");
 out:
        mutex_unlock(&mmiotrace_mutex);
 }