3 files changed, 103 insertions, 39 deletions

diff --git a/arch/ia64/kernel/iosapic.c b/arch/ia64/kernel/iosapic.c
index cf27cfb4d165..e647254c2707 100644
--- a/arch/ia64/kernel/iosapic.c
+++ b/arch/ia64/kernel/iosapic.c
@@ -354,6 +354,8 @@ iosapic_set_affinity (unsigned int irq, cpumask_t mask)
 
         irq &= (~IA64_IRQ_REDIRECTED);
 
+        /* IRQ migration across domain is not supported yet */
+        cpus_and(mask, mask, irq_to_domain(irq));
         if (cpus_empty(mask))
                 return;
 
@@ -663,6 +665,7 @@ get_target_cpu (unsigned int gsi, int irq)
 #ifdef CONFIG_SMP
         static int cpu = -1;
         extern int cpe_vector;
+        cpumask_t domain = irq_to_domain(irq);
 
         /*
          * In case of vector shared by multiple RTEs, all RTEs that
@@ -701,7 +704,7 @@ get_target_cpu (unsigned int gsi, int irq)
                         goto skip_numa_setup;
 
                 cpu_mask = node_to_cpumask(iosapic_lists[iosapic_index].node);
-
+                cpus_and(cpu_mask, cpu_mask, domain);
                 for_each_cpu_mask(numa_cpu, cpu_mask) {
                         if (!cpu_online(numa_cpu))
                                 cpu_clear(numa_cpu, cpu_mask);
@@ -731,7 +734,7 @@ skip_numa_setup:
         do {
                 if (++cpu >= NR_CPUS)
                         cpu = 0;
-        } while (!cpu_online(cpu));
+        } while (!cpu_online(cpu) || !cpu_isset(cpu, domain));
 
         return cpu_physical_id(cpu);
 #else  /* CONFIG_SMP */
@@ -900,7 +903,7 @@ iosapic_register_platform_intr (u32 int_type, unsigned int gsi,
         switch (int_type) {
               case ACPI_INTERRUPT_PMI:
                 irq = vector = iosapic_vector;
-                bind_irq_vector(irq, vector);
+                bind_irq_vector(irq, vector, CPU_MASK_ALL);
                 /*
                  * since PMI vector is alloc'd by FW(ACPI) not by kernel,
                  * we need to make sure the vector is available
@@ -917,7 +920,7 @@ iosapic_register_platform_intr (u32 int_type, unsigned int gsi,
                 break;
               case ACPI_INTERRUPT_CPEI:
                 irq = vector = IA64_CPE_VECTOR;
-                BUG_ON(bind_irq_vector(irq, vector));
+                BUG_ON(bind_irq_vector(irq, vector, CPU_MASK_ALL));
                 delivery = IOSAPIC_LOWEST_PRIORITY;
                 mask = 1;
                 break;
@@ -953,7 +956,7 @@ iosapic_override_isa_irq (unsigned int isa_irq, unsigned int gsi,
         unsigned int dest = cpu_physical_id(smp_processor_id());
 
         irq = vector = isa_irq_to_vector(isa_irq);
-        BUG_ON(bind_irq_vector(irq, vector));
+        BUG_ON(bind_irq_vector(irq, vector, CPU_MASK_ALL));
         register_intr(gsi, irq, IOSAPIC_LOWEST_PRIORITY, polarity, trigger);
 
         DBG("ISA: IRQ %u -> GSI %u (%s,%s) -> CPU %d (0x%04x) vector %d\n",
diff --git a/arch/ia64/kernel/irq_ia64.c b/arch/ia64/kernel/irq_ia64.c
index 072427c2c3f6..a3667631ed80 100644
--- a/arch/ia64/kernel/irq_ia64.c
+++ b/arch/ia64/kernel/irq_ia64.c
@@ -60,6 +60,8 @@ int ia64_last_device_vector = IA64_DEF_LAST_DEVICE_VECTOR;
 void __iomem *ipi_base_addr = ((void __iomem *)
                                (__IA64_UNCACHED_OFFSET | IA64_IPI_DEFAULT_BASE_ADDR));
 
+static cpumask_t vector_allocation_domain(int cpu);
+
 /*
  * Legacy IRQ to IA-64 vector translation table.
  */
@@ -73,13 +75,20 @@ EXPORT_SYMBOL(isa_irq_to_vector_map);
 DEFINE_SPINLOCK(vector_lock);
 
 struct irq_cfg irq_cfg[NR_IRQS] __read_mostly = {
-        [0 ... NR_IRQS - 1] = { .vector = IRQ_VECTOR_UNASSIGNED }
+        [0 ... NR_IRQS - 1] = {
+                .vector = IRQ_VECTOR_UNASSIGNED,
+                .domain = CPU_MASK_NONE
+        }
 };
 
 DEFINE_PER_CPU(int[IA64_NUM_VECTORS], vector_irq) = {
         [0 ... IA64_NUM_VECTORS - 1] = IA64_SPURIOUS_INT_VECTOR
 };
 
+static cpumask_t vector_table[IA64_MAX_DEVICE_VECTORS] = {
+        [0 ... IA64_MAX_DEVICE_VECTORS - 1] = CPU_MASK_NONE
+};
+
 static int irq_status[NR_IRQS] = {
         [0 ... NR_IRQS -1] = IRQ_UNUSED
 };
@@ -111,39 +120,54 @@ static inline int find_unassigned_irq(void)
         return -ENOSPC;
 }
 
-static inline int find_unassigned_vector(void)
+static inline int find_unassigned_vector(cpumask_t domain)
 {
-        int vector;
+        cpumask_t mask;
+        int pos;
+
+        cpus_and(mask, domain, cpu_online_map);
+        if (cpus_empty(mask))
+                return -EINVAL;
 
-        for (vector = IA64_FIRST_DEVICE_VECTOR;
-             vector <= IA64_LAST_DEVICE_VECTOR; vector++)
-                if (__get_cpu_var(vector_irq[vector]) == IA64_SPURIOUS_INT_VECTOR)
-                        return vector;
+        for (pos = 0; pos < IA64_NUM_DEVICE_VECTORS; pos++) {
+                cpus_and(mask, domain, vector_table[pos]);
+                if (!cpus_empty(mask))
+                        continue;
+                return IA64_FIRST_DEVICE_VECTOR + pos;
+        }
         return -ENOSPC;
 }
 
-static int __bind_irq_vector(int irq, int vector)
+static int __bind_irq_vector(int irq, int vector, cpumask_t domain)
 {
-        int cpu;
+        cpumask_t mask;
+        int cpu, pos;
+        struct irq_cfg *cfg = &irq_cfg[irq];
 
-        if (irq_to_vector(irq) == vector)
+        cpus_and(mask, domain, cpu_online_map);
+        if (cpus_empty(mask))
+                return -EINVAL;
+        if ((cfg->vector == vector) && cpus_equal(cfg->domain, domain))
                 return 0;
-        if (irq_to_vector(irq) != IRQ_VECTOR_UNASSIGNED)
+        if (cfg->vector != IRQ_VECTOR_UNASSIGNED)
                 return -EBUSY;
-        for_each_online_cpu(cpu)
+        for_each_cpu_mask(cpu, mask)
                 per_cpu(vector_irq, cpu)[vector] = irq;
-        irq_cfg[irq].vector = vector;
+        cfg->vector = vector;
+        cfg->domain = domain;
         irq_status[irq] = IRQ_USED;
+        pos = vector - IA64_FIRST_DEVICE_VECTOR;
+        cpus_or(vector_table[pos], vector_table[pos], domain);
         return 0;
 }
 
-int bind_irq_vector(int irq, int vector)
+int bind_irq_vector(int irq, int vector, cpumask_t domain)
 {
         unsigned long flags;
         int ret;
 
         spin_lock_irqsave(&vector_lock, flags);
-        ret = __bind_irq_vector(irq, vector);
+        ret = __bind_irq_vector(irq, vector, domain);
         spin_unlock_irqrestore(&vector_lock, flags);
         return ret;
 }
@@ -151,16 +175,24 @@ int bind_irq_vector(int irq, int vector)
 static void clear_irq_vector(int irq)
 {
         unsigned long flags;
-        int vector, cpu;
+        int vector, cpu, pos;
+        cpumask_t mask;
+        cpumask_t domain;
+        struct irq_cfg *cfg = &irq_cfg[irq];
 
         spin_lock_irqsave(&vector_lock, flags);
         BUG_ON((unsigned)irq >= NR_IRQS);
-        BUG_ON(irq_cfg[irq].vector == IRQ_VECTOR_UNASSIGNED);
-        vector = irq_cfg[irq].vector;
-        for_each_online_cpu(cpu)
+        BUG_ON(cfg->vector == IRQ_VECTOR_UNASSIGNED);
+        vector = cfg->vector;
+        domain = cfg->domain;
+        cpus_and(mask, cfg->domain, cpu_online_map);
+        for_each_cpu_mask(cpu, mask)
                 per_cpu(vector_irq, cpu)[vector] = IA64_SPURIOUS_INT_VECTOR;
-        irq_cfg[irq].vector = IRQ_VECTOR_UNASSIGNED;
+        cfg->vector = IRQ_VECTOR_UNASSIGNED;
+        cfg->domain = CPU_MASK_NONE;
         irq_status[irq] = IRQ_UNUSED;
+        pos = vector - IA64_FIRST_DEVICE_VECTOR;
+        cpus_andnot(vector_table[pos], vector_table[pos], domain);
         spin_unlock_irqrestore(&vector_lock, flags);
 }
 
@@ -168,18 +200,26 @@ int
 assign_irq_vector (int irq)
 {
         unsigned long flags;
-        int vector = -ENOSPC;
+        int vector, cpu;
+        cpumask_t domain;
+
+        vector = -ENOSPC;
 
+        spin_lock_irqsave(&vector_lock, flags);
         if (irq < 0) {
                 goto out;
         }
-        spin_lock_irqsave(&vector_lock, flags);
-        vector = find_unassigned_vector();
+        for_each_online_cpu(cpu) {
+                domain = vector_allocation_domain(cpu);
+                vector = find_unassigned_vector(domain);
+                if (vector >= 0)
+                        break;
+        }
         if (vector < 0)
                 goto out;
-        BUG_ON(__bind_irq_vector(irq, vector));
-        spin_unlock_irqrestore(&vector_lock, flags);
+        BUG_ON(__bind_irq_vector(irq, vector, domain));
  out:
+        spin_unlock_irqrestore(&vector_lock, flags);
         return vector;
 }
 
@@ -198,7 +238,7 @@ reserve_irq_vector (int vector)
         if (vector < IA64_FIRST_DEVICE_VECTOR ||
             vector > IA64_LAST_DEVICE_VECTOR)
                 return -EINVAL;
-        return !!bind_irq_vector(vector, vector);
+        return !!bind_irq_vector(vector, vector, CPU_MASK_ALL);
 }
 
 /*
@@ -214,11 +254,19 @@ void __setup_vector_irq(int cpu)
                 per_cpu(vector_irq, cpu)[vector] = IA64_SPURIOUS_INT_VECTOR;
         /* Mark the inuse vectors */
         for (irq = 0; irq < NR_IRQS; ++irq) {
-                if ((vector = irq_to_vector(irq)) != IRQ_VECTOR_UNASSIGNED)
-                        per_cpu(vector_irq, cpu)[vector] = irq;
+                if (!cpu_isset(cpu, irq_cfg[irq].domain))
+                        continue;
+                vector = irq_to_vector(irq);
+                per_cpu(vector_irq, cpu)[vector] = irq;
         }
 }
 
+static cpumask_t vector_allocation_domain(int cpu)
+{
+        return CPU_MASK_ALL;
+}
+
+
 void destroy_and_reserve_irq(unsigned int irq)
 {
         dynamic_irq_cleanup(irq);
@@ -233,17 +281,23 @@ void destroy_and_reserve_irq(unsigned int irq)
 int create_irq(void)
 {
         unsigned long flags;
-        int irq, vector;
+        int irq, vector, cpu;
+        cpumask_t domain;
 
-        irq = -ENOSPC;
+        irq = vector = -ENOSPC;
         spin_lock_irqsave(&vector_lock, flags);
-        vector = find_unassigned_vector();
+        for_each_online_cpu(cpu) {
+                domain = vector_allocation_domain(cpu);
+                vector = find_unassigned_vector(domain);
+                if (vector >= 0)
+                        break;
+        }
         if (vector < 0)
                 goto out;
         irq = find_unassigned_irq();
         if (irq < 0)
                 goto out;
-        BUG_ON(__bind_irq_vector(irq, vector));
+        BUG_ON(__bind_irq_vector(irq, vector, domain));
  out:
         spin_unlock_irqrestore(&vector_lock, flags);
         if (irq >= 0)
@@ -434,7 +488,7 @@ register_percpu_irq (ia64_vector vec, struct irqaction *action)
         unsigned int irq;
 
         irq = vec;
-        BUG_ON(bind_irq_vector(irq, vec));
+        BUG_ON(bind_irq_vector(irq, vec, CPU_MASK_ALL));
         desc = irq_desc + irq;
         desc->status |= IRQ_PER_CPU;
         desc->chip = &irq_type_ia64_lsapic;
diff --git a/arch/ia64/kernel/msi_ia64.c b/arch/ia64/kernel/msi_ia64.c
index c81080df70df..1d22670cc88b 100644
--- a/arch/ia64/kernel/msi_ia64.c
+++ b/arch/ia64/kernel/msi_ia64.c
@@ -52,6 +52,11 @@ static void ia64_set_msi_irq_affinity(unsigned int irq, cpumask_t cpu_mask)
         struct msi_msg msg;
         u32 addr;
 
+        /* IRQ migration across domain is not supported yet */
+        cpus_and(cpu_mask, cpu_mask, irq_to_domain(irq));
+        if (cpus_empty(cpu_mask))
+                return;
+
         read_msi_msg(irq, &msg);
 
         addr = msg.address_lo;
@@ -69,13 +74,15 @@ int ia64_setup_msi_irq(struct pci_dev *pdev, struct msi_desc *desc)
         struct msi_msg  msg;
         unsigned long   dest_phys_id;
         int     irq, vector;
+        cpumask_t mask;
 
         irq = create_irq();
         if (irq < 0)
                 return irq;
 
         set_irq_msi(irq, desc);
-        dest_phys_id = cpu_physical_id(first_cpu(cpu_online_map));
+        cpus_and(mask, irq_to_domain(irq), cpu_online_map);
+        dest_phys_id = cpu_physical_id(first_cpu(mask));
         vector = irq_to_vector(irq);
 
         msg.address_hi = 0;