3 files changed, 10 insertions, 8 deletions

diff --git a/drivers/hv/channel_mgmt.c b/drivers/hv/channel_mgmt.c
index 53a8600162a5..ff1be167eb04 100644
--- a/drivers/hv/channel_mgmt.c
+++ b/drivers/hv/channel_mgmt.c
@@ -318,7 +318,7 @@ static u32 get_vp_index(uuid_le *type_guid)
                 return 0;
         }
         cur_cpu = (++next_vp % max_cpus);
-        return 0;
+        return cur_cpu;
 }
 
 /*
diff --git a/drivers/hv/hv.c b/drivers/hv/hv.c
index 1c5481da6e4a..731158910c1e 100644
--- a/drivers/hv/hv.c
+++ b/drivers/hv/hv.c
@@ -272,7 +272,7 @@ u16 hv_signal_event(void *con_id)
  * retrieve the initialized message and event pages.  Otherwise, we create and
  * initialize the message and event pages.
  */
-void hv_synic_init(void *irqarg)
+void hv_synic_init(void *arg)
 {
         u64 version;
         union hv_synic_simp simp;
@@ -281,7 +281,6 @@ void hv_synic_init(void *irqarg)
         union hv_synic_scontrol sctrl;
         u64 vp_index;
 
-        u32 irq_vector = *((u32 *)(irqarg));
         int cpu = smp_processor_id();
 
         if (!hv_context.hypercall_page)
@@ -335,7 +334,7 @@ void hv_synic_init(void *irqarg)
         rdmsrl(HV_X64_MSR_SINT0 + VMBUS_MESSAGE_SINT, shared_sint.as_uint64);
 
         shared_sint.as_uint64 = 0;
-        shared_sint.vector = irq_vector; /* HV_SHARED_SINT_IDT_VECTOR + 0x20; */
+        shared_sint.vector = HYPERVISOR_CALLBACK_VECTOR;
         shared_sint.masked = false;
         shared_sint.auto_eoi = true;
 
diff --git a/drivers/hv/vmbus_drv.c b/drivers/hv/vmbus_drv.c
index cf19dfa5ead1..bf421e0efa1e 100644
--- a/drivers/hv/vmbus_drv.c
+++ b/drivers/hv/vmbus_drv.c
@@ -36,6 +36,7 @@
 #include <linux/kernel_stat.h>
 #include <asm/hyperv.h>
 #include <asm/hypervisor.h>
+#include <asm/mshyperv.h>
 #include "hyperv_vmbus.h"
 
 
@@ -528,7 +529,6 @@ static void vmbus_flow_handler(unsigned int irq, struct irq_desc *desc)
 static int vmbus_bus_init(int irq)
 {
         int ret;
-        unsigned int vector;
 
         /* Hypervisor initialization...setup hypercall page..etc */
         ret = hv_init();
@@ -558,13 +558,16 @@ static int vmbus_bus_init(int irq)
          */
         irq_set_handler(irq, vmbus_flow_handler);
 
-        vector = IRQ0_VECTOR + irq;
+        /*
+         * Register our interrupt handler.
+         */
+        hv_register_vmbus_handler(irq, vmbus_isr);
 
         /*
-         * Notify the hypervisor of our irq and
+         * Initialize the per-cpu interrupt state and
          * connect to the host.
          */
-        on_each_cpu(hv_synic_init, (void *)&vector, 1);
+        on_each_cpu(hv_synic_init, NULL, 1);
         ret = vmbus_connect();
         if (ret)
                 goto err_irq;