Merge branch 'parisc-4.9-1' of git://git.kernel.org/pub/scm/linux/kernel/git/deller...
[cascardo/linux.git] / drivers / hv / channel_mgmt.c
index b6c1211..96a85cd 100644 (file)
@@ -21,6 +21,7 @@
 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
 
 #include <linux/kernel.h>
+#include <linux/interrupt.h>
 #include <linux/sched.h>
 #include <linux/wait.h>
 #include <linux/mm.h>
@@ -138,10 +139,32 @@ static const struct vmbus_device vmbus_devs[] = {
        },
 };
 
-static u16 hv_get_dev_type(const uuid_le *guid)
+static const struct {
+       uuid_le guid;
+} vmbus_unsupported_devs[] = {
+       { HV_AVMA1_GUID },
+       { HV_AVMA2_GUID },
+       { HV_RDV_GUID   },
+};
+
+static bool is_unsupported_vmbus_devs(const uuid_le *guid)
+{
+       int i;
+
+       for (i = 0; i < ARRAY_SIZE(vmbus_unsupported_devs); i++)
+               if (!uuid_le_cmp(*guid, vmbus_unsupported_devs[i].guid))
+                       return true;
+       return false;
+}
+
+static u16 hv_get_dev_type(const struct vmbus_channel *channel)
 {
+       const uuid_le *guid = &channel->offermsg.offer.if_type;
        u16 i;
 
+       if (is_hvsock_channel(channel) || is_unsupported_vmbus_devs(guid))
+               return HV_UNKOWN;
+
        for (i = HV_IDE; i < HV_UNKOWN; i++) {
                if (!uuid_le_cmp(*guid, vmbus_devs[i].guid))
                        return i;
@@ -251,14 +274,12 @@ EXPORT_SYMBOL_GPL(vmbus_prep_negotiate_resp);
  */
 static struct vmbus_channel *alloc_channel(void)
 {
-       static atomic_t chan_num = ATOMIC_INIT(0);
        struct vmbus_channel *channel;
 
        channel = kzalloc(sizeof(*channel), GFP_ATOMIC);
        if (!channel)
                return NULL;
 
-       channel->id = atomic_inc_return(&chan_num);
        channel->acquire_ring_lock = true;
        spin_lock_init(&channel->inbound_lock);
        spin_lock_init(&channel->lock);
@@ -303,16 +324,32 @@ static void vmbus_release_relid(u32 relid)
        vmbus_post_msg(&msg, sizeof(struct vmbus_channel_relid_released));
 }
 
+void hv_event_tasklet_disable(struct vmbus_channel *channel)
+{
+       struct tasklet_struct *tasklet;
+       tasklet = hv_context.event_dpc[channel->target_cpu];
+       tasklet_disable(tasklet);
+}
+
+void hv_event_tasklet_enable(struct vmbus_channel *channel)
+{
+       struct tasklet_struct *tasklet;
+       tasklet = hv_context.event_dpc[channel->target_cpu];
+       tasklet_enable(tasklet);
+
+       /* In case there is any pending event */
+       tasklet_schedule(tasklet);
+}
+
 void hv_process_channel_removal(struct vmbus_channel *channel, u32 relid)
 {
        unsigned long flags;
        struct vmbus_channel *primary_channel;
 
-       vmbus_release_relid(relid);
-
        BUG_ON(!channel->rescind);
        BUG_ON(!mutex_is_locked(&vmbus_connection.channel_mutex));
 
+       hv_event_tasklet_disable(channel);
        if (channel->target_cpu != get_cpu()) {
                put_cpu();
                smp_call_function_single(channel->target_cpu,
@@ -321,6 +358,7 @@ void hv_process_channel_removal(struct vmbus_channel *channel, u32 relid)
                percpu_channel_deq(channel);
                put_cpu();
        }
+       hv_event_tasklet_enable(channel);
 
        if (channel->primary_channel == NULL) {
                list_del(&channel->listentry);
@@ -338,8 +376,11 @@ void hv_process_channel_removal(struct vmbus_channel *channel, u32 relid)
         * We need to free the bit for init_vp_index() to work in the case
         * of sub-channel, when we reload drivers like hv_netvsc.
         */
-       cpumask_clear_cpu(channel->target_cpu,
-                         &primary_channel->alloced_cpus_in_node);
+       if (channel->affinity_policy == HV_LOCALIZED)
+               cpumask_clear_cpu(channel->target_cpu,
+                                 &primary_channel->alloced_cpus_in_node);
+
+       vmbus_release_relid(relid);
 
        free_channel(channel);
 }
@@ -405,10 +446,13 @@ static void vmbus_process_offer(struct vmbus_channel *newchannel)
                        goto err_free_chan;
        }
 
-       dev_type = hv_get_dev_type(&newchannel->offermsg.offer.if_type);
+       dev_type = hv_get_dev_type(newchannel);
+       if (dev_type == HV_NIC)
+               set_channel_signal_state(newchannel, HV_SIGNAL_POLICY_EXPLICIT);
 
        init_vp_index(newchannel, dev_type);
 
+       hv_event_tasklet_disable(newchannel);
        if (newchannel->target_cpu != get_cpu()) {
                put_cpu();
                smp_call_function_single(newchannel->target_cpu,
@@ -418,6 +462,7 @@ static void vmbus_process_offer(struct vmbus_channel *newchannel)
                percpu_channel_enq(newchannel);
                put_cpu();
        }
+       hv_event_tasklet_enable(newchannel);
 
        /*
         * This state is used to indicate a successful open
@@ -463,12 +508,11 @@ static void vmbus_process_offer(struct vmbus_channel *newchannel)
        return;
 
 err_deq_chan:
-       vmbus_release_relid(newchannel->offermsg.child_relid);
-
        mutex_lock(&vmbus_connection.channel_mutex);
        list_del(&newchannel->listentry);
        mutex_unlock(&vmbus_connection.channel_mutex);
 
+       hv_event_tasklet_disable(newchannel);
        if (newchannel->target_cpu != get_cpu()) {
                put_cpu();
                smp_call_function_single(newchannel->target_cpu,
@@ -477,6 +521,9 @@ err_deq_chan:
                percpu_channel_deq(newchannel);
                put_cpu();
        }
+       hv_event_tasklet_enable(newchannel);
+
+       vmbus_release_relid(newchannel->offermsg.child_relid);
 
 err_free_chan:
        free_channel(newchannel);
@@ -522,17 +569,17 @@ static void init_vp_index(struct vmbus_channel *channel, u16 dev_type)
        }
 
        /*
-        * We distribute primary channels evenly across all the available
-        * NUMA nodes and within the assigned NUMA node we will assign the
-        * first available CPU to the primary channel.
-        * The sub-channels will be assigned to the CPUs available in the
-        * NUMA node evenly.
+        * Based on the channel affinity policy, we will assign the NUMA
+        * nodes.
         */
-       if (!primary) {
+
+       if ((channel->affinity_policy == HV_BALANCED) || (!primary)) {
                while (true) {
                        next_node = next_numa_node_id++;
-                       if (next_node == nr_node_ids)
+                       if (next_node == nr_node_ids) {
                                next_node = next_numa_node_id = 0;
+                               continue;
+                       }
                        if (cpumask_empty(cpumask_of_node(next_node)))
                                continue;
                        break;
@@ -556,15 +603,17 @@ static void init_vp_index(struct vmbus_channel *channel, u16 dev_type)
 
        cur_cpu = -1;
 
-       /*
-        * Normally Hyper-V host doesn't create more subchannels than there
-        * are VCPUs on the node but it is possible when not all present VCPUs
-        * on the node are initialized by guest. Clear the alloced_cpus_in_node
-        * to start over.
-        */
-       if (cpumask_equal(&primary->alloced_cpus_in_node,
-                         cpumask_of_node(primary->numa_node)))
-               cpumask_clear(&primary->alloced_cpus_in_node);
+       if (primary->affinity_policy == HV_LOCALIZED) {
+               /*
+                * Normally Hyper-V host doesn't create more subchannels
+                * than there are VCPUs on the node but it is possible when not
+                * all present VCPUs on the node are initialized by guest.
+                * Clear the alloced_cpus_in_node to start over.
+                */
+               if (cpumask_equal(&primary->alloced_cpus_in_node,
+                                 cpumask_of_node(primary->numa_node)))
+                       cpumask_clear(&primary->alloced_cpus_in_node);
+       }
 
        while (true) {
                cur_cpu = cpumask_next(cur_cpu, &available_mask);
@@ -575,17 +624,24 @@ static void init_vp_index(struct vmbus_channel *channel, u16 dev_type)
                        continue;
                }
 
-               /*
-                * NOTE: in the case of sub-channel, we clear the sub-channel
-                * related bit(s) in primary->alloced_cpus_in_node in
-                * hv_process_channel_removal(), so when we reload drivers
-                * like hv_netvsc in SMP guest, here we're able to re-allocate
-                * bit from primary->alloced_cpus_in_node.
-                */
-               if (!cpumask_test_cpu(cur_cpu,
-                               &primary->alloced_cpus_in_node)) {
-                       cpumask_set_cpu(cur_cpu,
-                                       &primary->alloced_cpus_in_node);
+               if (primary->affinity_policy == HV_LOCALIZED) {
+                       /*
+                        * NOTE: in the case of sub-channel, we clear the
+                        * sub-channel related bit(s) in
+                        * primary->alloced_cpus_in_node in
+                        * hv_process_channel_removal(), so when we
+                        * reload drivers like hv_netvsc in SMP guest, here
+                        * we're able to re-allocate
+                        * bit from primary->alloced_cpus_in_node.
+                        */
+                       if (!cpumask_test_cpu(cur_cpu,
+                                             &primary->alloced_cpus_in_node)) {
+                               cpumask_set_cpu(cur_cpu,
+                                               &primary->alloced_cpus_in_node);
+                               cpumask_set_cpu(cur_cpu, alloced_mask);
+                               break;
+                       }
+               } else {
                        cpumask_set_cpu(cur_cpu, alloced_mask);
                        break;
                }