#include <linux/spinlock.h>
 #include <linux/types.h>
 
-#define VGIC_NR_IRQS           256
+#define VGIC_NR_IRQS_LEGACY    256
 #define VGIC_NR_SGIS           16
 #define VGIC_NR_PPIS           16
 #define VGIC_NR_PRIVATE_IRQS   (VGIC_NR_SGIS + VGIC_NR_PPIS)
 #error Invalid number of CPU interfaces
 #endif
 
-#if (VGIC_NR_IRQS & 31)
+#if (VGIC_NR_IRQS_LEGACY & 31)
 #error "VGIC_NR_IRQS must be a multiple of 32"
 #endif
 
-#if (VGIC_NR_IRQS > VGIC_MAX_IRQS)
+#if (VGIC_NR_IRQS_LEGACY > VGIC_MAX_IRQS)
 #error "VGIC_NR_IRQS must be <= 1024"
 #endif
 
 
 
        case 4:                 /* GICD_TYPER */
                reg  = (atomic_read(&vcpu->kvm->online_vcpus) - 1) << 5;
-               reg |= (VGIC_NR_IRQS >> 5) - 1;
+               reg |= (vcpu->kvm->arch.vgic.nr_irqs >> 5) - 1;
                vgic_reg_access(mmio, ®, word_offset,
                                ACCESS_READ_VALUE | ACCESS_WRITE_IGNORED);
                break;
 static bool vgic_queue_irq(struct kvm_vcpu *vcpu, u8 sgi_source_id, int irq)
 {
        struct vgic_cpu *vgic_cpu = &vcpu->arch.vgic_cpu;
+       struct vgic_dist *dist = &vcpu->kvm->arch.vgic;
        struct vgic_lr vlr;
        int lr;
 
        /* Sanitize the input... */
        BUG_ON(sgi_source_id & ~7);
        BUG_ON(sgi_source_id && irq >= VGIC_NR_SGIS);
-       BUG_ON(irq >= VGIC_NR_IRQS);
+       BUG_ON(irq >= dist->nr_irqs);
 
        kvm_debug("Queue IRQ%d\n", irq);
 
 
                vlr = vgic_get_lr(vcpu, lr);
 
-               BUG_ON(vlr.irq >= VGIC_NR_IRQS);
+               BUG_ON(vlr.irq >= dist->nr_irqs);
                vgic_cpu->vgic_irq_lr_map[vlr.irq] = LR_EMPTY;
        }
 
        if (vcpu->vcpu_id >= dist->nr_cpus)
                return -EBUSY;
 
-       for (i = 0; i < VGIC_NR_IRQS; i++) {
+       for (i = 0; i < dist->nr_irqs; i++) {
                if (i < VGIC_NR_PPIS)
                        vgic_bitmap_set_irq_val(&dist->irq_enabled,
                                                vcpu->vcpu_id, i, 1);
        int ret, i;
 
        nr_cpus = dist->nr_cpus = KVM_MAX_VCPUS;
-       nr_irqs = dist->nr_irqs = VGIC_NR_IRQS;
+
+       if (!dist->nr_irqs)
+               dist->nr_irqs = VGIC_NR_IRQS_LEGACY;
+
+       nr_irqs = dist->nr_irqs;
 
        ret  = vgic_init_bitmap(&dist->irq_enabled, nr_cpus, nr_irqs);
        ret |= vgic_init_bitmap(&dist->irq_level, nr_cpus, nr_irqs);
                goto out;
        }
 
-       for (i = VGIC_NR_PRIVATE_IRQS; i < VGIC_NR_IRQS; i += 4)
+       for (i = VGIC_NR_PRIVATE_IRQS; i < kvm->arch.vgic.nr_irqs; i += 4)
                vgic_set_target_reg(kvm, 0, i);
 
        kvm->arch.vgic.ready = true;