Makes Dept able to track dependencies by rwsem.

Signed-off-by: Byungchul Park <byungchul.p...@lge.com>
---
 include/linux/lockdep.h      | 24 ++++++++++++++++++++----
 include/linux/percpu-rwsem.h | 10 +++++++++-
 include/linux/rwsem.h        | 32 ++++++++++++++++++++++++++++++++
 3 files changed, 61 insertions(+), 5 deletions(-)

diff --git a/include/linux/lockdep.h b/include/linux/lockdep.h
index 91f8783..fb354c4 100644
--- a/include/linux/lockdep.h
+++ b/include/linux/lockdep.h
@@ -645,10 +645,26 @@ static inline void print_irqtrace_events(struct 
task_struct *curr)
        dept_mutex_unlock(&(l)->dmap, i);                               \
 } while (0)
 
-#define rwsem_acquire(l, s, t, i)              lock_acquire_exclusive(l, s, t, 
NULL, i)
-#define rwsem_acquire_nest(l, s, t, n, i)      lock_acquire_exclusive(l, s, t, 
n, i)
-#define rwsem_acquire_read(l, s, t, i)         lock_acquire_shared(l, s, t, 
NULL, i)
-#define rwsem_release(l, i)                    lock_release(l, i)
+#define rwsem_acquire(l, s, t, i)                                      \
+do {                                                                   \
+       lock_acquire_exclusive(l, s, t, NULL, i);                       \
+       dept_rwsem_lock(&(l)->dmap, s, t, NULL, "up_write", i);         \
+} while (0)
+#define rwsem_acquire_nest(l, s, t, n, i)                              \
+do {                                                                   \
+       lock_acquire_exclusive(l, s, t, n, i);                          \
+       dept_rwsem_lock(&(l)->dmap, s, t, (n) ? &(n)->dmap : NULL, "up_write", 
i);\
+} while (0)
+#define rwsem_acquire_read(l, s, t, i)                                 \
+do {                                                                   \
+       lock_acquire_shared(l, s, t, NULL, i);                          \
+       dept_rwsem_lock(&(l)->dmap, s, t, NULL, "up_read", i);          \
+} while (0)
+#define rwsem_release(l, i)                                            \
+do {                                                                   \
+       lock_release(l, i);                                             \
+       dept_rwsem_unlock(&(l)->dmap, i);                               \
+} while (0)
 
 #define lock_map_acquire(l)                    lock_acquire_exclusive(l, 0, 0, 
NULL, _THIS_IP_)
 #define lock_map_acquire_read(l)               
lock_acquire_shared_recursive(l, 0, 0, NULL, _THIS_IP_)
diff --git a/include/linux/percpu-rwsem.h b/include/linux/percpu-rwsem.h
index 5fda40f..7ec5625 100644
--- a/include/linux/percpu-rwsem.h
+++ b/include/linux/percpu-rwsem.h
@@ -20,8 +20,16 @@ struct percpu_rw_semaphore {
 #endif
 };
 
+#ifdef CONFIG_DEPT
+#define __PERCPU_RWSEM_DMAP_INIT(lockname) .dmap = { .name = #lockname }
+#else
+#define __PERCPU_RWSEM_DMAP_INIT(lockname)
+#endif
+
 #ifdef CONFIG_DEBUG_LOCK_ALLOC
-#define __PERCPU_RWSEM_DEP_MAP_INIT(lockname)  .dep_map = { .name = #lockname 
},
+#define __PERCPU_RWSEM_DEP_MAP_INIT(lockname)  .dep_map = {    \
+       .name = #lockname,                                      \
+       __PERCPU_RWSEM_DMAP_INIT(lockname) },
 #else
 #define __PERCPU_RWSEM_DEP_MAP_INIT(lockname)
 #endif
diff --git a/include/linux/rwsem.h b/include/linux/rwsem.h
index f934876..f0772c2 100644
--- a/include/linux/rwsem.h
+++ b/include/linux/rwsem.h
@@ -16,11 +16,18 @@
 #include <linux/atomic.h>
 #include <linux/err.h>
 
+#ifdef CONFIG_DEPT
+# define RWSEM_DMAP_INIT(lockname)     .dmap = { .name = #lockname },
+#else
+# define RWSEM_DMAP_INIT(lockname)
+#endif
+
 #ifdef CONFIG_DEBUG_LOCK_ALLOC
 # define __RWSEM_DEP_MAP_INIT(lockname)                        \
        .dep_map = {                                    \
                .name = #lockname,                      \
                .wait_type_inner = LD_WAIT_SLEEP,       \
+               RWSEM_DMAP_INIT(lockname)               \
        },
 #else
 # define __RWSEM_DEP_MAP_INIT(lockname)
@@ -32,6 +39,31 @@
 #include <linux/osq_lock.h>
 #endif
 
+#ifdef CONFIG_DEPT
+#define dept_rwsem_lock(m, ne, t, n, e_fn, ip)                         \
+do {                                                                   \
+       if (t) {                                                        \
+               dept_ecxt_enter(m, 1UL, ip, __func__, e_fn, ne);        \
+               dept_ask_event(m);                                      \
+       } else if (n) {                                                 \
+               dept_ecxt_enter_nokeep(m);                              \
+               dept_ask_event(m);                                      \
+       } else {                                                        \
+               dept_wait(m, 1UL, ip, __func__, ne);                    \
+               dept_ecxt_enter(m, 1UL, ip, __func__, e_fn, ne);        \
+               dept_ask_event(m);                                      \
+       }                                                               \
+} while (0)
+#define dept_rwsem_unlock(m, ip)                                       \
+do {                                                                   \
+       dept_event(m, 1UL, ip, __func__);                               \
+       dept_ecxt_exit(m, 1UL, ip);                                     \
+} while (0)
+#else
+#define dept_rwsem_lock(m, ne, t, n, e_fn, ip) do { } while (0)
+#define dept_rwsem_unlock(m, ip)               do { } while (0)
+#endif
+
 /*
  * For an uncontended rwsem, count and owner are the only fields a task
  * needs to touch when acquiring the rwsem. So they are put next to each
-- 
1.9.1

Reply via email to