litmus: seq-lock: add MODEL_ASSERT() for the important behavior

[c11tester.git] / test / rmwprog.c

diff --git a/test/rmwprog.c b/test/rmwprog.c
index 14929ee23b6502403d12f8dda1194b38c233c59e..57ab54462bab9b03b48a87cf86578d06c8c5e070 100644 (file)
--- a/test/rmwprog.c
+++ b/test/rmwprog.c
@@ -1,26 +1,33 @@
+#include <stdlib.h>
 #include <stdio.h>
+#include <threads.h>
+#include <stdatomic.h>
 
-#include "libthreads.h"
 #include "librace.h"
-#include "stdatomic.h"
 
 atomic_int x;
+static int N = 2;
 
 static void a(void *obj)
 {
-       atomic_fetch_add_explicit(&x, 1, memory_order_relaxed);
-       atomic_fetch_add_explicit(&x, 1, memory_order_relaxed);
+       int i;
+       for (i = 0; i < N; i++)
+               atomic_fetch_add_explicit(&x, 1, memory_order_relaxed);
 }
 
-void user_main()
+int user_main(int argc, char **argv)
 {
        thrd_t t1, t2;
 
-       atomic_init(&x, 0);
+       if (argc > 1)
+               N = atoi(argv[1]);
 
+       atomic_init(&x, 0);
        thrd_create(&t1, (thrd_start_t)&a, NULL);
        thrd_create(&t2, (thrd_start_t)&a, NULL);
 
        thrd_join(t1);
        thrd_join(t2);
+
+       return 0;
 }