Add a locked version of idxd_quiesce() call so that the quiesce can be
called with a lock in situations where the lock is not held by the caller.
In the driver probe/remove path, the lock is already held, so the raw
version can be called w/o locking.
Reviewed-by: Kevin Tian <kevin.tian@intel.com>
Signed-off-by: Dave Jiang <dave.jiang@intel.com>
Link: https://lore.kernel.org/r/163528418980.3925689.5841907054957931211.stgit@djiang5-desk3.ch.intel.com
Signed-off-by: Vinod Koul <vkoul@kernel.org>
        return 0;
 }
 
-void idxd_wq_quiesce(struct idxd_wq *wq)
+void __idxd_wq_quiesce(struct idxd_wq *wq)
 {
+       lockdep_assert_held(&wq->wq_lock);
        percpu_ref_kill(&wq->wq_active);
        wait_for_completion(&wq->wq_dead);
 }
 
+void idxd_wq_quiesce(struct idxd_wq *wq)
+{
+       mutex_lock(&wq->wq_lock);
+       __idxd_wq_quiesce(wq);
+       mutex_unlock(&wq->wq_lock);
+}
+
 /* Device control bits */
 static inline bool idxd_is_enabled(struct idxd_device *idxd)
 {
 
        return 0;
 
 err_dma:
-       idxd_wq_quiesce(wq);
+       __idxd_wq_quiesce(wq);
        percpu_ref_exit(&wq->wq_active);
 err_ref:
        idxd_wq_free_resources(wq);
        struct idxd_wq *wq = idxd_dev_to_wq(idxd_dev);
 
        mutex_lock(&wq->wq_lock);
-       idxd_wq_quiesce(wq);
+       __idxd_wq_quiesce(wq);
        idxd_unregister_dma_channel(wq);
        idxd_wq_free_resources(wq);
        __drv_disable_wq(wq);
 
 void idxd_wq_unmap_portal(struct idxd_wq *wq);
 int idxd_wq_set_pasid(struct idxd_wq *wq, int pasid);
 int idxd_wq_disable_pasid(struct idxd_wq *wq);
+void __idxd_wq_quiesce(struct idxd_wq *wq);
 void idxd_wq_quiesce(struct idxd_wq *wq);
 int idxd_wq_init_percpu_ref(struct idxd_wq *wq);