IDR: Expose the XArray lock

Allow users of the IDR to use the XArray lock for their own
synchronisation purposes.  The IDR continues to rely on the caller to
handle locking, but this lets the caller use the lock embedded in the
IDR data structure instead of allocating their own lock.

Signed-off-by: Matthew Wilcox <willy@infradead.org>
Signed-off-by: Jason Gunthorpe <jgg@mellanox.com>

diff --git a/include/linux/idr.h b/include/linux/idr.h
index e856f4e0ab359d085d33266ca4505ea941df01af..3e8215b2c371c27406280fa5c4eccc098cebde7f 100644 (file)
--- a/include/linux/idr.h
+++ b/include/linux/idr.h
@@ -98,6 +98,17 @@ static inline void idr_set_cursor(struct idr *idr, unsigned int val)
  * period).
  */
 
+#define idr_lock(idr)          xa_lock(&(idr)->idr_rt)
+#define idr_unlock(idr)                xa_unlock(&(idr)->idr_rt)
+#define idr_lock_bh(idr)       xa_lock_bh(&(idr)->idr_rt)
+#define idr_unlock_bh(idr)     xa_unlock_bh(&(idr)->idr_rt)
+#define idr_lock_irq(idr)      xa_lock_irq(&(idr)->idr_rt)
+#define idr_unlock_irq(idr)    xa_unlock_irq(&(idr)->idr_rt)
+#define idr_lock_irqsave(idr, flags) \
+                               xa_lock_irqsave(&(idr)->idr_rt, flags)
+#define idr_unlock_irqrestore(idr, flags) \
+                               xa_unlock_irqrestore(&(idr)->idr_rt, flags)
+
 void idr_preload(gfp_t gfp_mask);
 
 int idr_alloc(struct idr *, void *ptr, int start, int end, gfp_t);