call_rcu(&nvdev->rcu, free_netvsc_device);
 }
 
-static void netvsc_destroy_buf(struct hv_device *device)
+static void netvsc_revoke_buf(struct hv_device *device,
+                             struct netvsc_device *net_device)
 {
        struct nvsp_message *revoke_packet;
        struct net_device *ndev = hv_get_drvdata(device);
-       struct net_device_context *ndc = netdev_priv(ndev);
-       struct netvsc_device *net_device = rtnl_dereference(ndc->nvdev);
        int ret;
 
        /*
                net_device->recv_section_cnt = 0;
        }
 
-       /* Teardown the gpadl on the vsp end */
-       if (net_device->recv_buf_gpadl_handle) {
-               ret = vmbus_teardown_gpadl(device->channel,
-                                          net_device->recv_buf_gpadl_handle);
-
-               /* If we failed here, we might as well return and have a leak
-                * rather than continue and a bugchk
-                */
-               if (ret != 0) {
-                       netdev_err(ndev,
-                                  "unable to teardown receive buffer's gpadl\n");
-                       return;
-               }
-               net_device->recv_buf_gpadl_handle = 0;
-       }
-
-       if (net_device->recv_buf) {
-               /* Free up the receive buffer */
-               vfree(net_device->recv_buf);
-               net_device->recv_buf = NULL;
-       }
-
        /* Deal with the send buffer we may have setup.
         * If we got a  send section size, it means we received a
         * NVSP_MSG1_TYPE_SEND_SEND_BUF_COMPLETE msg (ie sent
                }
                net_device->send_section_cnt = 0;
        }
-       /* Teardown the gpadl on the vsp end */
+}
+
+static void netvsc_teardown_gpadl(struct hv_device *device,
+                                 struct netvsc_device *net_device)
+{
+       struct net_device *ndev = hv_get_drvdata(device);
+       int ret;
+
+       if (net_device->recv_buf_gpadl_handle) {
+               ret = vmbus_teardown_gpadl(device->channel,
+                                          net_device->recv_buf_gpadl_handle);
+
+               /* If we failed here, we might as well return and have a leak
+                * rather than continue and a bugchk
+                */
+               if (ret != 0) {
+                       netdev_err(ndev,
+                                  "unable to teardown receive buffer's gpadl\n");
+                       return;
+               }
+               net_device->recv_buf_gpadl_handle = 0;
+       }
+
+       if (net_device->recv_buf) {
+               /* Free up the receive buffer */
+               vfree(net_device->recv_buf);
+               net_device->recv_buf = NULL;
+       }
+
        if (net_device->send_buf_gpadl_handle) {
                ret = vmbus_teardown_gpadl(device->channel,
                                           net_device->send_buf_gpadl_handle);
        goto exit;
 
 cleanup:
-       netvsc_destroy_buf(device);
+       netvsc_revoke_buf(device, net_device);
+       netvsc_teardown_gpadl(device, net_device);
 
 exit:
        return ret;
        return ret;
 }
 
-static void netvsc_disconnect_vsp(struct hv_device *device)
-{
-       netvsc_destroy_buf(device);
-}
-
 /*
  * netvsc_device_remove - Callback when the root bus device is removed
  */
 
        cancel_work_sync(&net_device->subchan_work);
 
-       netvsc_disconnect_vsp(device);
+       netvsc_revoke_buf(device, net_device);
 
        RCU_INIT_POINTER(net_device_ctx->nvdev, NULL);
 
        /* Now, we can close the channel safely */
        vmbus_close(device->channel);
 
+       netvsc_teardown_gpadl(device, net_device);
+
        /* And dissassociate NAPI context from device */
        for (i = 0; i < net_device->num_chn; i++)
                netif_napi_del(&net_device->chan_table[i].napi);