Introduce state to manage the lifecycle of net device.

Signed-off-by: K. Y. Srinivasan <k...@microsoft.com>
Signed-off-by: Haiyang Zhang <haiya...@microsoft.com>
---
 drivers/staging/hv/hyperv_net.h |    1 +
 drivers/staging/hv/netvsc.c     |    3 +++
 2 files changed, 4 insertions(+), 0 deletions(-)

diff --git a/drivers/staging/hv/hyperv_net.h b/drivers/staging/hv/hyperv_net.h
index 5782fea..0b347c1 100644
--- a/drivers/staging/hv/hyperv_net.h
+++ b/drivers/staging/hv/hyperv_net.h
@@ -371,6 +371,7 @@ struct netvsc_device {
 
        atomic_t refcnt;
        atomic_t num_outstanding_sends;
+       bool destroy;
        /*
         * List of free preallocated hv_netvsc_packet to represent receive
         * packet
diff --git a/drivers/staging/hv/netvsc.c b/drivers/staging/hv/netvsc.c
index 7722102..67cf7fb 100644
--- a/drivers/staging/hv/netvsc.c
+++ b/drivers/staging/hv/netvsc.c
@@ -43,6 +43,7 @@ static struct netvsc_device *alloc_net_device(struct 
hv_device *device)
        /* Set to 2 to allow both inbound and outbound traffic */
        atomic_set(&net_device->refcnt, 2);
 
+       net_device->destroy = false;
        net_device->dev = device;
        device->ext = net_device;
 
@@ -404,6 +405,8 @@ int netvsc_device_remove(struct hv_device *device)
                return -ENODEV;
        }
 
+       net_device->destroy = true;
+
        /* Wait for all send completions */
        while (atomic_read(&net_device->num_outstanding_sends)) {
                dev_err(&device->device,
-- 
1.7.4.1

_______________________________________________
Virtualization mailing list
Virtualization@lists.linux-foundation.org
https://lists.linux-foundation.org/mailman/listinfo/virtualization

Reply via email to