mt7601u: don't cleanup device second time after .resume()

Make sure .disconnect() doesn't cleanup the device if
.resume() failed.  This may happen when device is removed
during suspend.

Signed-off-by: Jakub Kicinski <kubakici@wp.pl>
Signed-off-by: Kalle Valo <kvalo@codeaurora.org>

diff --git a/drivers/net/wireless/mediatek/mt7601u/init.c b/drivers/net/wireless/mediatek/mt7601u/init.c
index 1fc86e8..45eb079 100644 (file)
--- a/drivers/net/wireless/mediatek/mt7601u/init.c
+++ b/drivers/net/wireless/mediatek/mt7601u/init.c
@@ -427,6 +427,9 @@ err:
 
 void mt7601u_cleanup(struct mt7601u_dev *dev)
 {
+       if (!test_and_clear_bit(MT7601U_STATE_INITIALIZED, &dev->state))
+               return;
+
        mt7601u_stop_hardware(dev);
        mt7601u_dma_cleanup(dev);
        mt7601u_mcu_cmd_deinit(dev);

diff --git a/drivers/net/wireless/mediatek/mt7601u/usb.c b/drivers/net/wireless/mediatek/mt7601u/usb.c
index 99e2b39..54dba40 100644 (file)
--- a/drivers/net/wireless/mediatek/mt7601u/usb.c
+++ b/drivers/net/wireless/mediatek/mt7601u/usb.c
@@ -338,8 +338,15 @@ static int mt7601u_suspend(struct usb_interface *usb_intf, pm_message_t state)
 static int mt7601u_resume(struct usb_interface *usb_intf)
 {
        struct mt7601u_dev *dev = usb_get_intfdata(usb_intf);
+       int ret;
+
+       ret = mt7601u_init_hardware(dev);
+       if (ret)
+               return ret;
+
+       set_bit(MT7601U_STATE_INITIALIZED, &dev->state);
 
-       return mt7601u_init_hardware(dev);
+       return 0;
 }
 
 MODULE_DEVICE_TABLE(usb, mt7601u_device_table);