urcu tests: add writer duration parameter

[urcu.git] / tests / test_urcu_assign.c

diff --git a/tests/test_urcu_assign.c b/tests/test_urcu_assign.c
index d38361f313d3c43ab5a0c1250b21cc53fc9c4bc3..663a4e87258ad91774e5cf7dfcbf386eb33d5e7a 100644 (file)
--- a/tests/test_urcu_assign.c
+++ b/tests/test_urcu_assign.c
@@ -77,6 +77,9 @@ static unsigned long duration;
 /* read-side C.S. duration, in loops */
 static unsigned long rduration;
 
+/* write-side C.S. duration, in loops */
+static unsigned long wduration;
+
 static inline void loop_sleep(unsigned long l)
 {
        while(l-- != 0)
@@ -274,6 +277,8 @@ void *thr_writer(void *_count)
                rcu_copy_mutex_lock();
                old = test_rcu_pointer;
                rcu_assign_pointer(test_rcu_pointer, new);
+               if (unlikely(wduration))
+                       loop_sleep(wduration);
                rcu_copy_mutex_unlock();
                synchronize_rcu();
                if (old)
@@ -300,6 +305,7 @@ void show_usage(int argc, char **argv)
 #endif
        printf(" [-d delay] (writer period (us))");
        printf(" [-c duration] (reader C.S. duration (in loops))");
+       printf(" [-e duration] (writer C.S. duration (in loops))");
        printf(" [-v] (verbose output)");
        printf(" [-a cpu#] [-a cpu#]... (affinity)");
        printf("\n");
@@ -373,6 +379,13 @@ int main(int argc, char **argv)
                        }
                        wdelay = atol(argv[++i]);
                        break;
+               case 'e':
+                       if (argc < i + 2) {
+                               show_usage(argc, argv);
+                               return -1;
+                       }
+                       wduration = atol(argv[++i]);
+                       break;
                case 'v':
                        verbose_mode = 1;
                        break;