deque: add MODEL_ASSERT(), fixup #include's

[model-checker-benchmarks.git] / chase-lev-deque / main.c

diff --git a/chase-lev-deque/main.c b/chase-lev-deque/main.c
index 77ea5a8..c744b47 100644 (file)
--- a/chase-lev-deque/main.c
+++ b/chase-lev-deque/main.c
@@ -3,6 +3,9 @@
 #include <stdio.h>
 #include <threads.h>
 #include <stdatomic.h>
+#include <stdio.h>
+
+#include "model-assert.h"
 
 #include "deque.h"
 
@@ -11,7 +14,9 @@ int a;
 int b;
 
 static void task(void * param) {
-       a=steal(q);
+       do {
+               a=steal(q);
+       } while(a==EMPTY);
 }
 
 int user_main(int argc, char **argv)
@@ -21,11 +26,11 @@ int user_main(int argc, char **argv)
        thrd_create(&t, task, 0);
        push(q, 1);
        push(q, 2);
-       do {
-               b=take(q);
-       }       while(b==EMPTY);
+       b=take(q);
        thrd_join(t);
-       if (a+b!=3)
-               printf("a=%d b=%d\n",a,b);
+
+       printf("a=%d b=%d\n",a,b);
+       MODEL_ASSERT(a + b == 3);
+
        return 0;
 }