Several use case for getting pointer to spinlock protecting a directory.

Signed-off-by: Jérôme Glisse <[email protected]>
---
 include/linux/hmm_pt.h | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/include/linux/hmm_pt.h b/include/linux/hmm_pt.h
index f745d6c..22100a6 100644
--- a/include/linux/hmm_pt.h
+++ b/include/linux/hmm_pt.h
@@ -255,6 +255,16 @@ static inline void hmm_pt_directory_lock(struct hmm_pt *pt,
                spin_lock(&pt->lock);
 }
 
+static inline spinlock_t *hmm_pt_directory_lock_ptr(struct hmm_pt *pt,
+                                                   struct page *ptd,
+                                                   unsigned level)
+{
+       if (level)
+               return &ptd->ptl;
+       else
+               return &pt->lock;
+}
+
 static inline void hmm_pt_directory_unlock(struct hmm_pt *pt,
                                           struct page *ptd,
                                           unsigned level)
@@ -272,6 +282,13 @@ static inline void hmm_pt_directory_lock(struct hmm_pt *pt,
        spin_lock(&pt->lock);
 }
 
+static inline spinlock_t *hmm_pt_directory_lock_ptr(struct hmm_pt *pt,
+                                                   struct page *ptd,
+                                                   unsigned level)
+{
+       return &pt->lock;
+}
+
 static inline void hmm_pt_directory_unlock(struct hmm_pt *pt,
                                           struct page *ptd,
                                           unsigned level)
@@ -358,6 +375,14 @@ static inline void hmm_pt_iter_directory_lock(struct 
hmm_pt_iter *iter)
        hmm_pt_directory_lock(pt, iter->ptd[pt->llevel - 1], pt->llevel);
 }
 
+static inline spinlock_t *hmm_pt_iter_directory_lock_ptr(struct hmm_pt_iter *i)
+{
+       struct hmm_pt *pt = i->pt;
+
+       return hmm_pt_directory_lock_ptr(pt, i->ptd[pt->llevel - 1],
+                                        pt->llevel);
+}
+
 static inline void hmm_pt_iter_directory_unlock(struct hmm_pt_iter *iter)
 {
        struct hmm_pt *pt = iter->pt;
-- 
2.4.3

Reply via email to