3 files changed, 5 insertions, 2 deletions

diff --git a/drivers/usb/wusbcore/cbaf.c b/drivers/usb/wusbcore/cbaf.c
index 1335cbe1191d..25eae405f622 100644
--- a/drivers/usb/wusbcore/cbaf.c
+++ b/drivers/usb/wusbcore/cbaf.c
@@ -638,8 +638,7 @@ static void cbaf_disconnect(struct usb_interface *iface)
         usb_put_intf(iface);
         kfree(cbaf->buffer);
         /* paranoia: clean up crypto keys */
-        memset(cbaf, 0, sizeof(*cbaf));
-        kfree(cbaf);
+        kzfree(cbaf);
 }
 
 static struct usb_device_id cbaf_id_table[] = {
diff --git a/drivers/usb/wusbcore/devconnect.c b/drivers/usb/wusbcore/devconnect.c
index 8e18141bb2e0..f0aac0cf315a 100644
--- a/drivers/usb/wusbcore/devconnect.c
+++ b/drivers/usb/wusbcore/devconnect.c
@@ -889,6 +889,8 @@ static void wusb_dev_add_ncb(struct usb_device *usb_dev)
         if (usb_dev->wusb == 0 || usb_dev->devnum == 1)
                 return;         /* skip non wusb and wusb RHs */
 
+        usb_set_device_state(usb_dev, USB_STATE_UNAUTHENTICATED);
+
         wusbhc = wusbhc_get_by_usb_dev(usb_dev);
         if (wusbhc == NULL)
                 goto error_nodev;
diff --git a/drivers/usb/wusbcore/security.c b/drivers/usb/wusbcore/security.c
index f4aa28eca70d..8118db7f1d8d 100644
--- a/drivers/usb/wusbcore/security.c
+++ b/drivers/usb/wusbcore/security.c
@@ -312,6 +312,7 @@ int wusb_dev_update_address(struct wusbhc *wusbhc, struct wusb_dev *wusb_dev)
         result = wusb_set_dev_addr(wusbhc, wusb_dev, 0);
         if (result < 0)
                 goto error_addr0;
+        usb_set_device_state(usb_dev, USB_STATE_DEFAULT);
         usb_ep0_reinit(usb_dev);
 
         /* Set new (authenticated) address. */
@@ -327,6 +328,7 @@ int wusb_dev_update_address(struct wusbhc *wusbhc, struct wusb_dev *wusb_dev)
         result = wusb_set_dev_addr(wusbhc, wusb_dev, new_address);
         if (result < 0)
                 goto error_addr;
+        usb_set_device_state(usb_dev, USB_STATE_ADDRESS);
         usb_ep0_reinit(usb_dev);
         usb_dev->authenticated = 1;
 error_addr: