1 files changed, 6 insertions, 7 deletions

diff --git a/drivers/net/virtio_net.c b/drivers/net/virtio_net.c
index 113ee93dbb2e..656a02e28e26 100644
--- a/drivers/net/virtio_net.c
+++ b/drivers/net/virtio_net.c
@@ -1160,11 +1160,6 @@ static int virtnet_cpu_callback(struct notifier_block *nfb,
 {
         struct virtnet_info *vi = container_of(nfb, struct virtnet_info, nb);
 
-        mutex_lock(&vi->config_lock);
-
-        if (!vi->config_enable)
-                goto done;
-
         switch(action & ~CPU_TASKS_FROZEN) {
         case CPU_ONLINE:
         case CPU_DOWN_FAILED:
@@ -1178,8 +1173,6 @@ static int virtnet_cpu_callback(struct notifier_block *nfb,
                 break;
         }
 
-done:
-        mutex_unlock(&vi->config_lock);
         return NOTIFY_OK;
 }
 
@@ -1747,6 +1740,8 @@ static int virtnet_freeze(struct virtio_device *vdev)
         struct virtnet_info *vi = vdev->priv;
         int i;
 
+        unregister_hotcpu_notifier(&vi->nb);
+
         /* Prevent config work handler from accessing the device */
         mutex_lock(&vi->config_lock);
         vi->config_enable = false;
@@ -1795,6 +1790,10 @@ static int virtnet_restore(struct virtio_device *vdev)
         virtnet_set_queues(vi, vi->curr_queue_pairs);
         rtnl_unlock();
 
+        err = register_hotcpu_notifier(&vi->nb);
+        if (err)
+                return err;
+
         return 0;
 }
 #endif