Drivers: hv: vmbus: Fix a bug in rescind processing in vmbus_close_internal()

[firefly-linux-kernel-4.4.55.git] / drivers / hv / channel.c

diff --git a/drivers/hv/channel.c b/drivers/hv/channel.c
index da53180f5eb43203dca3c15b4dc9fff7b7a8b7fe..2c8206d820f2e8c075fff41421af01391a106b6e 100644 (file)
--- a/drivers/hv/channel.c
+++ b/drivers/hv/channel.c
@@ -501,15 +501,6 @@ static int vmbus_close_internal(struct vmbus_channel *channel)
                put_cpu();
        }
 
-       /*
-        * If the channel has been rescinded; process device removal.
-        */
-       if (channel->rescind) {
-               hv_process_channel_removal(channel,
-                                          channel->offermsg.child_relid);
-               return 0;
-       }
-
        /* Send a closing message */
 
        msg = &channel->close_msg.msg;
@@ -549,6 +540,12 @@ static int vmbus_close_internal(struct vmbus_channel *channel)
        free_pages((unsigned long)channel->ringbuffer_pages,
                get_order(channel->ringbuffer_pagecount * PAGE_SIZE));
 
+       /*
+        * If the channel has been rescinded; process device removal.
+        */
+       if (channel->rescind)
+               hv_process_channel_removal(channel,
+                                          channel->offermsg.child_relid);
        return ret;
 }