#include <linux/dma-mapping.h>
 #include <linux/mm.h>
 #include <linux/debugfs.h>
+#include <linux/wait.h>
 
 #include <linux/usb/ch9.h>
 #include <linux/usb/gadget.h>
  * @endpoint: usb endpoint
  * @pending_list: list of pending requests for this endpoint
  * @started_list: list of started requests on this endpoint
+ * @wait_end_transfer: wait_queue_head_t for waiting on End Transfer complete
  * @lock: spinlock for endpoint request queue traversal
  * @regs: pointer to first endpoint register
  * @trb_pool: array of transaction buffers
        struct list_head        pending_list;
        struct list_head        started_list;
 
+       wait_queue_head_t       wait_end_transfer;
+
        spinlock_t              lock;
        void __iomem            *regs;
 
 #define DWC3_EP_BUSY           (1 << 4)
 #define DWC3_EP_PENDING_REQUEST        (1 << 5)
 #define DWC3_EP_MISSED_ISOC    (1 << 6)
+#define DWC3_EP_END_TRANSFER_PENDING   (1 << 7)
 
        /* This last one is specific to EP0 */
 #define DWC3_EP0_DIR_IN                (1 << 31)
 #define DEPEVT_TRANSFER_BUS_EXPIRY     2
 
        u32     parameters:16;
+
+/* For Command Complete Events */
+#define DEPEVT_PARAMETER_CMD(n)        (((n) & (0xf << 8)) >> 8)
 } __packed;
 
 /**
 
                dep->comp_desc = comp_desc;
                dep->type = usb_endpoint_type(desc);
                dep->flags |= DWC3_EP_ENABLED;
+               dep->flags &= ~DWC3_EP_END_TRANSFER_PENDING;
 
                reg = dwc3_readl(dwc->regs, DWC3_DALEPENA);
                reg |= DWC3_DALEPENA_EP(dep->number);
                dwc3_writel(dwc->regs, DWC3_DALEPENA, reg);
 
+               init_waitqueue_head(&dep->wait_end_transfer);
+
                if (usb_endpoint_xfer_control(desc))
                        return 0;
 
        dep->endpoint.desc = NULL;
        dep->comp_desc = NULL;
        dep->type = 0;
-       dep->flags = 0;
+       dep->flags &= DWC3_EP_END_TRANSFER_PENDING;
 
        return 0;
 }
 
 static void __dwc3_gadget_stop(struct dwc3 *dwc)
 {
-       if (pm_runtime_suspended(dwc->dev))
-               return;
-
        dwc3_gadget_disable_irq(dwc);
        __dwc3_gadget_ep_disable(dwc->eps[0]);
        __dwc3_gadget_ep_disable(dwc->eps[1]);
 {
        struct dwc3             *dwc = gadget_to_dwc(g);
        unsigned long           flags;
+       int                     epnum;
 
        spin_lock_irqsave(&dwc->lock, flags);
+
+       if (pm_runtime_suspended(dwc->dev))
+               goto out;
+
        __dwc3_gadget_stop(dwc);
+
+       for (epnum = 2; epnum < DWC3_ENDPOINTS_NUM; epnum++) {
+               struct dwc3_ep  *dep = dwc->eps[epnum];
+
+               if (!dep)
+                       continue;
+
+               if (!(dep->flags & DWC3_EP_END_TRANSFER_PENDING))
+                       continue;
+
+               wait_event_lock_irq(dep->wait_end_transfer,
+                                   !(dep->flags & DWC3_EP_END_TRANSFER_PENDING),
+                                   dwc->lock);
+       }
+
+out:
        dwc->gadget_driver      = NULL;
        spin_unlock_irqrestore(&dwc->lock, flags);
 
 {
        struct dwc3_ep          *dep;
        u8                      epnum = event->endpoint_number;
+       u8                      cmd;
 
        dep = dwc->eps[epnum];
 
-       if (!(dep->flags & DWC3_EP_ENABLED))
+       if (!(dep->flags & DWC3_EP_ENABLED) &&
+           !(dep->flags & DWC3_EP_END_TRANSFER_PENDING))
                return;
 
        if (epnum == 0 || epnum == 1) {
                        return;
                }
                break;
-       case DWC3_DEPEVT_RXTXFIFOEVT:
        case DWC3_DEPEVT_EPCMDCMPLT:
+               cmd = DEPEVT_PARAMETER_CMD(event->parameters);
+
+               if (cmd == DWC3_DEPCMD_ENDTRANSFER) {
+                       dep->flags &= ~DWC3_EP_END_TRANSFER_PENDING;
+                       wake_up(&dep->wait_end_transfer);
+               }
+               break;
+       case DWC3_DEPEVT_RXTXFIFOEVT:
                break;
        }
 }
 
        dep = dwc->eps[epnum];
 
-       if (!dep->resource_index)
+       if ((dep->flags & DWC3_EP_END_TRANSFER_PENDING) ||
+           !dep->resource_index)
                return;
 
        /*
        dep->resource_index = 0;
        dep->flags &= ~DWC3_EP_BUSY;
 
-       if (dwc3_is_usb31(dwc) || dwc->revision < DWC3_REVISION_310A)
+       if (dwc3_is_usb31(dwc) || dwc->revision < DWC3_REVISION_310A) {
+               dep->flags |= DWC3_EP_END_TRANSFER_PENDING;
                udelay(100);
+       }
 }
 
 static void dwc3_clear_stall_all_ep(struct dwc3 *dwc)