In preparation for cleaning up reference counts, introduce state to manage the lifecycle of net device. Signed-off-by: K. Y. Srinivasan <kys@xxxxxxxxxxxxx> Signed-off-by: Haiyang Zhang <haiyangz@xxxxxxxxxxxxx> Signed-off-by: Abhishek Kane <v-abkane@xxxxxxxxxxxxx> Signed-off-by: Hank Janssen <hjanssen@xxxxxxxxxxxxx> --- drivers/staging/hv/hyperv_net.h | 1 + drivers/staging/hv/netvsc.c | 2 ++ 2 files changed, 3 insertions(+), 0 deletions(-) diff --git a/drivers/staging/hv/hyperv_net.h b/drivers/staging/hv/hyperv_net.h index b5bee9e..c6836be 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; bool drain_notify; wait_queue_head_t waiting_to_drain; /* diff --git a/drivers/staging/hv/netvsc.c b/drivers/staging/hv/netvsc.c index 56749ca..f03018c 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->drain_notify = false; + net_device->destroy = false; init_waitqueue_head(&net_device->waiting_to_drain); net_device->dev = device; @@ -409,6 +410,7 @@ int netvsc_device_remove(struct hv_device *device) netdev_err(ndev, "No net device present!!\n"); return -ENODEV; } + net_device->destroy = true; /* Wait for all send completions */ while (atomic_read(&net_device->num_outstanding_sends)) { -- 1.7.4.1 _______________________________________________ Virtualization mailing list Virtualization@xxxxxxxxxxxxxxxxxxxxxxxxxx https://lists.linux-foundation.org/mailman/listinfo/virtualization