Gleb fixed bitmap ops usage in kvm_ioapic_get_delivery_bitmask.

Sheng merged two functions, as well as fix several issues in
kvm_get_intr_delivery_bitmask:
1. deliver_bitmask is a bitmap rather than a unsigned long intereger.
2. Lowest priority target bitmap wrong calculated by mistake.
3. Prevent potential NULL reference.
4. Declaration in include/kvm_host.h caused powerpc compilation warning.

Signed-off-by: Gleb Natapov <[email protected]>
Signed-off-by: Sheng Yang <[email protected]>
---
 include/linux/kvm_host.h |    3 ---
 virt/kvm/ioapic.c        |   39 ---------------------------------------
 virt/kvm/ioapic.h        |    5 +++--
 virt/kvm/irq_comm.c      |   42 ++++++++++++++++++++++++++++++++++++++----
 4 files changed, 41 insertions(+), 48 deletions(-)

diff --git a/include/linux/kvm_host.h b/include/linux/kvm_host.h
index 3832243..fb60f31 100644
--- a/include/linux/kvm_host.h
+++ b/include/linux/kvm_host.h
@@ -363,9 +363,6 @@ void kvm_unregister_irq_mask_notifier(struct kvm *kvm, int 
irq,
                                      struct kvm_irq_mask_notifier *kimn);
 void kvm_fire_mask_notifiers(struct kvm *kvm, int irq, bool mask);
 
-void kvm_get_intr_delivery_bitmask(struct kvm_ioapic *ioapic,
-                                  union kvm_ioapic_redirect_entry *entry,
-                                  unsigned long *deliver_bitmask);
 int kvm_set_irq(struct kvm *kvm, int irq_source_id, int irq, int level);
 void kvm_notify_acked_irq(struct kvm *kvm, unsigned irqchip, unsigned pin);
 void kvm_register_irq_ack_notifier(struct kvm *kvm,
diff --git a/virt/kvm/ioapic.c b/virt/kvm/ioapic.c
index 7c2cb2b..2c40ff8 100644
--- a/virt/kvm/ioapic.c
+++ b/virt/kvm/ioapic.c
@@ -161,45 +161,6 @@ static void ioapic_inj_nmi(struct kvm_vcpu *vcpu)
        kvm_vcpu_kick(vcpu);
 }
 
-void kvm_ioapic_get_delivery_bitmask(struct kvm_ioapic *ioapic, u8 dest,
-                                    u8 dest_mode, unsigned long *mask)
-{
-       int i;
-       struct kvm *kvm = ioapic->kvm;
-       struct kvm_vcpu *vcpu;
-
-       ioapic_debug("dest %d dest_mode %d\n", dest, dest_mode);
-
-       *mask = 0;
-       if (dest_mode == 0) {   /* Physical mode. */
-               if (dest == 0xFF) {     /* Broadcast. */
-                       for (i = 0; i < KVM_MAX_VCPUS; ++i)
-                               if (kvm->vcpus[i] && kvm->vcpus[i]->arch.apic)
-                                       *mask |= 1 << i;
-                       return;
-               }
-               for (i = 0; i < KVM_MAX_VCPUS; ++i) {
-                       vcpu = kvm->vcpus[i];
-                       if (!vcpu)
-                               continue;
-                       if (kvm_apic_match_physical_addr(vcpu->arch.apic, 
dest)) {
-                               if (vcpu->arch.apic)
-                                       *mask = 1 << i;
-                               break;
-                       }
-               }
-       } else if (dest != 0)   /* Logical mode, MDA non-zero. */
-               for (i = 0; i < KVM_MAX_VCPUS; ++i) {
-                       vcpu = kvm->vcpus[i];
-                       if (!vcpu)
-                               continue;
-                       if (vcpu->arch.apic &&
-                           kvm_apic_match_logical_addr(vcpu->arch.apic, dest))
-                               *mask |= 1 << vcpu->vcpu_id;
-               }
-       ioapic_debug("mask %x\n", *mask);
-}
-
 static int ioapic_deliver(struct kvm_ioapic *ioapic, int irq)
 {
        union kvm_ioapic_redirect_entry entry = ioapic->redirtbl[irq];
diff --git a/virt/kvm/ioapic.h b/virt/kvm/ioapic.h
index 7275f87..c8032ab 100644
--- a/virt/kvm/ioapic.h
+++ b/virt/kvm/ioapic.h
@@ -70,7 +70,8 @@ void kvm_ioapic_update_eoi(struct kvm *kvm, int vector, int 
trigger_mode);
 int kvm_ioapic_init(struct kvm *kvm);
 int kvm_ioapic_set_irq(struct kvm_ioapic *ioapic, int irq, int level);
 void kvm_ioapic_reset(struct kvm_ioapic *ioapic);
-void kvm_ioapic_get_delivery_bitmask(struct kvm_ioapic *ioapic, u8 dest,
-                                    u8 dest_mode, unsigned long *mask);
+void kvm_get_intr_delivery_bitmask(struct kvm_ioapic *ioapic,
+                                  union kvm_ioapic_redirect_entry *entry,
+                                  unsigned long *deliver_bitmask);
 
 #endif
diff --git a/virt/kvm/irq_comm.c b/virt/kvm/irq_comm.c
index d165e05..bb6c5d0 100644
--- a/virt/kvm/irq_comm.c
+++ b/virt/kvm/irq_comm.c
@@ -47,15 +47,49 @@ void kvm_get_intr_delivery_bitmask(struct kvm_ioapic 
*ioapic,
                                   union kvm_ioapic_redirect_entry *entry,
                                   unsigned long *deliver_bitmask)
 {
+       int i;
+       struct kvm *kvm = ioapic->kvm;
        struct kvm_vcpu *vcpu;
 
-       kvm_ioapic_get_delivery_bitmask(ioapic, entry->fields.dest_id,
-                                       entry->fields.dest_mode,
-                                       deliver_bitmask);
+       bitmap_zero(deliver_bitmask, KVM_MAX_VCPUS);
+
+       if (entry->fields.dest_mode == 0) {     /* Physical mode. */
+               if (entry->fields.dest_id == 0xFF) {    /* Broadcast. */
+                       for (i = 0; i < KVM_MAX_VCPUS; ++i)
+                               if (kvm->vcpus[i] && kvm->vcpus[i]->arch.apic)
+                                       __set_bit(i, deliver_bitmask);
+                       return;
+               }
+               for (i = 0; i < KVM_MAX_VCPUS; ++i) {
+                       vcpu = kvm->vcpus[i];
+                       if (!vcpu)
+                               continue;
+                       if (kvm_apic_match_physical_addr(vcpu->arch.apic,
+                                       entry->fields.dest_id)) {
+                               if (vcpu->arch.apic)
+                                       __set_bit(i, deliver_bitmask);
+                               break;
+                       }
+               }
+       } else if (entry->fields.dest_id != 0) /* Logical mode, MDA non-zero. */
+               for (i = 0; i < KVM_MAX_VCPUS; ++i) {
+                       vcpu = kvm->vcpus[i];
+                       if (!vcpu)
+                               continue;
+                       if (vcpu->arch.apic &&
+                           kvm_apic_match_logical_addr(vcpu->arch.apic,
+                                       entry->fields.dest_id))
+                               __set_bit(i, deliver_bitmask);
+               }
+
        switch (entry->fields.delivery_mode) {
        case IOAPIC_LOWEST_PRIORITY:
+               /* Select one in deliver_bitmask */
                vcpu = kvm_get_lowest_prio_vcpu(ioapic->kvm,
                                entry->fields.vector, deliver_bitmask);
+               bitmap_zero(deliver_bitmask, KVM_MAX_VCPUS);
+               if (!vcpu)
+                       return;
                __set_bit(vcpu->vcpu_id, deliver_bitmask);
                break;
        case IOAPIC_FIXED:
@@ -65,7 +99,7 @@ void kvm_get_intr_delivery_bitmask(struct kvm_ioapic *ioapic,
                if (printk_ratelimit())
                        printk(KERN_INFO "kvm: unsupported delivery mode %d\n",
                                entry->fields.delivery_mode);
-               *deliver_bitmask = 0;
+               bitmap_zero(deliver_bitmask, KVM_MAX_VCPUS);
        }
 }
 
-- 
1.5.4.5

--
To unsubscribe from this list: send the line "unsubscribe kvm" in
the body of a message to [email protected]
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to