}
 EXPORT_SYMBOL_GPL(ufshcd_delay_us);
 
-/*
+/**
  * ufshcd_wait_for_register - wait for register value to change
- * @hba - per-adapter interface
- * @reg - mmio register offset
- * @mask - mask to apply to read register value
- * @val - wait condition
- * @interval_us - polling interval in microsecs
- * @timeout_ms - timeout in millisecs
- * @can_sleep - perform sleep or just spin
+ * @hba: per-adapter interface
+ * @reg: mmio register offset
+ * @mask: mask to apply to the read register value
+ * @val: value to wait for
+ * @interval_us: polling interval in microseconds
+ * @timeout_ms: timeout in milliseconds
  *
- * Returns -ETIMEDOUT on error, zero on success
+ * Return:
+ * -ETIMEDOUT on error, zero on success.
  */
 int ufshcd_wait_for_register(struct ufs_hba *hba, u32 reg, u32 mask,
                                u32 val, unsigned long interval_us,
-                               unsigned long timeout_ms, bool can_sleep)
+                               unsigned long timeout_ms)
 {
        int err = 0;
        unsigned long timeout = jiffies + msecs_to_jiffies(timeout_ms);
        val = val & mask;
 
        while ((ufshcd_readl(hba, reg) & mask) != val) {
-               if (can_sleep)
-                       usleep_range(interval_us, interval_us + 50);
-               else
-                       udelay(interval_us);
+               usleep_range(interval_us, interval_us + 50);
                if (time_after(jiffies, timeout)) {
                        if ((ufshcd_readl(hba, reg) & mask) != val)
                                err = -ETIMEDOUT;
         */
        err = ufshcd_wait_for_register(hba,
                        REG_UTP_TRANSFER_REQ_DOOR_BELL,
-                       mask, ~mask, 1000, 1000, true);
+                       mask, ~mask, 1000, 1000);
 
        return err;
 }
 /**
  * ufshcd_hba_stop - Send controller to reset state
  * @hba: per adapter instance
- * @can_sleep: perform sleep or just spin
  */
-static inline void ufshcd_hba_stop(struct ufs_hba *hba, bool can_sleep)
+static inline void ufshcd_hba_stop(struct ufs_hba *hba)
 {
+       unsigned long flags;
        int err;
 
+       /*
+        * Obtain the host lock to prevent that the controller is disabled
+        * while the UFS interrupt handler is active on another CPU.
+        */
+       spin_lock_irqsave(hba->host->host_lock, flags);
        ufshcd_writel(hba, CONTROLLER_DISABLE,  REG_CONTROLLER_ENABLE);
+       spin_unlock_irqrestore(hba->host->host_lock, flags);
+
        err = ufshcd_wait_for_register(hba, REG_CONTROLLER_ENABLE,
                                        CONTROLLER_ENABLE, CONTROLLER_DISABLE,
-                                       10, 1, can_sleep);
+                                       10, 1);
        if (err)
                dev_err(hba->dev, "%s: Controller disable failed\n", __func__);
 }
 
        if (!ufshcd_is_hba_active(hba))
                /* change controller state to "reset state" */
-               ufshcd_hba_stop(hba, true);
+               ufshcd_hba_stop(hba);
 
        /* UniPro link is disabled at this point */
        ufshcd_set_link_off(hba);
        /* poll for max. 1 sec to clear door bell register by h/w */
        err = ufshcd_wait_for_register(hba,
                        REG_UTP_TASK_REQ_DOOR_BELL,
-                       mask, 0, 1000, 1000, true);
+                       mask, 0, 1000, 1000);
 out:
        return err;
 }
         * Stop the host controller and complete the requests
         * cleared by h/w
         */
+       ufshcd_hba_stop(hba);
+
        spin_lock_irqsave(hba->host->host_lock, flags);
-       ufshcd_hba_stop(hba, false);
        hba->silence_err_logs = true;
        ufshcd_complete_requests(hba);
        hba->silence_err_logs = false;
                 * Change controller state to "reset state" which
                 * should also put the link in off/reset state
                 */
-               ufshcd_hba_stop(hba, true);
+               ufshcd_hba_stop(hba);
                /*
                 * TODO: Check if we need any delay to make sure that
                 * controller is reset
        scsi_remove_host(hba->host);
        /* disable interrupts */
        ufshcd_disable_intr(hba, hba->intr_mask);
-       ufshcd_hba_stop(hba, true);
+       ufshcd_hba_stop(hba);
 
        ufshcd_exit_clk_scaling(hba);
        ufshcd_exit_clk_gating(hba);