/******************************************************************************
  * GADGET block
  *****************************************************************************/
+/**
+ * ci_hdrc_gadget_connect: caller makes sure gadget driver is binded
+ */
+static void ci_hdrc_gadget_connect(struct usb_gadget *_gadget, int is_active)
+{
+       struct ci_hdrc *ci = container_of(_gadget, struct ci_hdrc, gadget);
+
+       if (is_active) {
+               pm_runtime_get_sync(&_gadget->dev);
+               hw_device_reset(ci);
+               hw_device_state(ci, ci->ep0out->qh.dma);
+               usb_gadget_set_state(_gadget, USB_STATE_POWERED);
+               usb_udc_vbus_handler(_gadget, true);
+       } else {
+               usb_udc_vbus_handler(_gadget, false);
+               if (ci->driver)
+                       ci->driver->disconnect(&ci->gadget);
+               hw_device_state(ci, 0);
+               if (ci->platdata->notify_event)
+                       ci->platdata->notify_event(ci,
+                       CI_HDRC_CONTROLLER_STOPPED_EVENT);
+               _gadget_stop_activity(&ci->gadget);
+               pm_runtime_put_sync(&_gadget->dev);
+               usb_gadget_set_state(_gadget, USB_STATE_NOTATTACHED);
+       }
+}
+
 static int ci_udc_vbus_session(struct usb_gadget *_gadget, int is_active)
 {
        struct ci_hdrc *ci = container_of(_gadget, struct ci_hdrc, gadget);
                usb_phy_set_charger_state(ci->usb_phy, is_active ?
                        USB_CHARGER_PRESENT : USB_CHARGER_ABSENT);
 
-       if (gadget_ready) {
-               if (is_active) {
-                       pm_runtime_get_sync(&_gadget->dev);
-                       hw_device_reset(ci);
-                       hw_device_state(ci, ci->ep0out->qh.dma);
-                       usb_gadget_set_state(_gadget, USB_STATE_POWERED);
-                       usb_udc_vbus_handler(_gadget, true);
-               } else {
-                       usb_udc_vbus_handler(_gadget, false);
-                       if (ci->driver)
-                               ci->driver->disconnect(&ci->gadget);
-                       hw_device_state(ci, 0);
-                       if (ci->platdata->notify_event)
-                               ci->platdata->notify_event(ci,
-                               CI_HDRC_CONTROLLER_STOPPED_EVENT);
-                       _gadget_stop_activity(&ci->gadget);
-                       pm_runtime_put_sync(&_gadget->dev);
-                       usb_gadget_set_state(_gadget, USB_STATE_NOTATTACHED);
-               }
-       }
+       if (gadget_ready)
+               ci_hdrc_gadget_connect(_gadget, is_active);
 
        return 0;
 }
                return retval;
        }
 
-       pm_runtime_get_sync(&ci->gadget.dev);
-       if (ci->vbus_active) {
-               hw_device_reset(ci);
-       } else {
+       if (ci->vbus_active)
+               ci_hdrc_gadget_connect(gadget, 1);
+       else
                usb_udc_vbus_handler(&ci->gadget, false);
-               pm_runtime_put_sync(&ci->gadget.dev);
-               return retval;
-       }
-
-       retval = hw_device_state(ci, ci->ep0out->qh.dma);
-       if (retval)
-               pm_runtime_put_sync(&ci->gadget.dev);
 
        return retval;
 }