Signed-off-by: Nikolay Borisov <nbori...@suse.com>
---

Hello Paul, 

Here is the code I used to test the DRW lock via the lock torture 
infrastructure. 
It's rather ugly but got the job done for me. It's definitely not in a mergeable
form. At the very least I think including btrfs headers constitutes a violation 
of separation of different subsystems. Would it be acceptable to guard them 
behind something like "#if BTRFS && BTRFS_DEBUG" ? 

I'm really posting this just for posterity/provenance purposes. I'm fine with 
dropping the patch. 


 fs/btrfs/locking.h           |  1 +
 kernel/locking/locktorture.c | 77 +++++++++++++++++++++++++++++++++++-
 2 files changed, 77 insertions(+), 1 deletion(-)

diff --git a/fs/btrfs/locking.h b/fs/btrfs/locking.h
index 44378c65f843..27627d4fd3a9 100644
--- a/fs/btrfs/locking.h
+++ b/fs/btrfs/locking.h
@@ -9,6 +9,7 @@
 #include <linux/atomic.h>
 #include <linux/wait.h>
 #include <linux/percpu_counter.h>
+#include "extent_io.h"
 
 #define BTRFS_WRITE_LOCK 1
 #define BTRFS_READ_LOCK 2
diff --git a/kernel/locking/locktorture.c b/kernel/locking/locktorture.c
index 80a463d31a8d..774e10a25876 100644
--- a/kernel/locking/locktorture.c
+++ b/kernel/locking/locktorture.c
@@ -30,6 +30,8 @@
 #include <linux/slab.h>
 #include <linux/percpu-rwsem.h>
 #include <linux/torture.h>
+#include "../../fs/btrfs/ctree.h"
+#include "../../fs/btrfs/locking.h"
 
 MODULE_LICENSE("GPL");
 MODULE_AUTHOR("Paul E. McKenney <paul...@linux.ibm.com>");
@@ -85,6 +87,7 @@ struct lock_torture_ops {
 
        unsigned long flags; /* for irq spinlocks */
        const char *name;
+       bool multiple;
 };
 
 struct lock_torture_cxt {
@@ -600,6 +603,7 @@ static void torture_percpu_rwsem_up_read(void) 
__releases(pcpu_rwsem)
        percpu_up_read(&pcpu_rwsem);
 }
 
+
 static struct lock_torture_ops percpu_rwsem_lock_ops = {
        .init           = torture_percpu_rwsem_init,
        .writelock      = torture_percpu_rwsem_down_write,
@@ -612,6 +616,76 @@ static struct lock_torture_ops percpu_rwsem_lock_ops = {
        .name           = "percpu_rwsem_lock"
 };
 
+static struct btrfs_drw_lock torture_drw_lock;
+
+void torture_drw_init(void)
+{
+       BUG_ON(btrfs_drw_lock_init(&torture_drw_lock));
+}
+
+static int torture_drw_write_lock(void) __acquires(torture_drw_lock)
+{
+       btrfs_drw_write_lock(&torture_drw_lock);
+       return 0;
+}
+
+static void torture_drw_write_unlock(void) __releases(torture_drw_lock)
+{
+       btrfs_drw_write_unlock(&torture_drw_lock);
+}
+
+static int torture_drw_read_lock(void) __acquires(torture_drw_lock)
+{
+       btrfs_drw_read_lock(&torture_drw_lock);
+       return 0;
+}
+
+static void torture_drw_read_unlock(void) __releases(torture_drw_lock)
+{
+       btrfs_drw_read_unlock(&torture_drw_lock);
+}
+
+static void torture_drw_write_delay(struct torture_random_state *trsp)
+{
+       const unsigned long longdelay_ms = 100;
+
+       /* We want a long delay occasionally to force massive contention.  */
+       if (!(torture_random(trsp) %
+             (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
+               mdelay(longdelay_ms * 10);
+       else
+               mdelay(longdelay_ms / 10);
+       if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
+               torture_preempt_schedule();  /* Allow test to be preempted. */
+}
+
+static void torture_drw_read_delay(struct torture_random_state *trsp)
+{
+       const unsigned long longdelay_ms = 100;
+
+       /* We want a long delay occasionally to force massive contention.  */
+       if (!(torture_random(trsp) %
+             (cxt.nrealreaders_stress * 2000 * longdelay_ms)))
+               mdelay(longdelay_ms * 2);
+       else
+               mdelay(longdelay_ms / 2);
+       if (!(torture_random(trsp) % (cxt.nrealreaders_stress * 20000)))
+               torture_preempt_schedule();  /* Allow test to be preempted. */
+}
+
+static struct lock_torture_ops btrfs_drw_lock_ops = {
+       .init           = torture_drw_init,
+       .writelock      = torture_drw_write_lock,
+       .write_delay    = torture_drw_write_delay,
+       .task_boost     = torture_boost_dummy,
+       .writeunlock    = torture_drw_write_unlock,
+       .readlock       = torture_drw_read_lock,
+       .read_delay     = torture_drw_read_delay, /* figure what to do with 
this */
+       .readunlock     = torture_drw_read_unlock,
+       .multiple = true,
+       .name           = "btrfs_drw_lock"
+};
+
 /*
  * Lock torture writer kthread.  Repeatedly acquires and releases
  * the lock, checking for duplicate acquisitions.
@@ -630,7 +704,7 @@ static int lock_torture_writer(void *arg)
 
                cxt.cur_ops->task_boost(&rand);
                cxt.cur_ops->writelock();
-               if (WARN_ON_ONCE(lock_is_write_held))
+               if (!cxt.cur_ops->multiple && WARN_ON_ONCE(lock_is_write_held))
                        lwsp->n_lock_fail++;
                lock_is_write_held = 1;
                if (WARN_ON_ONCE(lock_is_read_held))
@@ -852,6 +926,7 @@ static int __init lock_torture_init(void)
 #endif
                &rwsem_lock_ops,
                &percpu_rwsem_lock_ops,
+               &btrfs_drw_lock_ops
        };
 
        if (!torture_init_begin(torture_type, verbose))
-- 
2.17.1

Reply via email to