fixed minor bugs
[cdsspec-compiler.git] / benchmark / ms-queue / my_queue.c
index 634a051c56ad79622fcea50c1937fb2c2ab61b65..3b1784bd500a1bf69106f031927f769db12f60d5 100644 (file)
@@ -1,9 +1,63 @@
 #include <threads.h>
 #include <stdlib.h>
+#include <stdatomic.h>
 #include "librace.h"
 #include "model-assert.h"
 
-#include "my_queue.h"
+
+typedef unsigned long long pointer;
+typedef atomic_ullong pointer_t;
+
+#define MAKE_POINTER(ptr, count)       ((((pointer)count) << 32) | ptr)
+#define PTR_MASK 0xffffffffLL
+#define COUNT_MASK (0xffffffffLL << 32)
+
+static inline void set_count(pointer *p, unsigned int val) { *p = (*p & ~COUNT_MASK) | ((pointer)val << 32); }
+static inline void set_ptr(pointer *p, unsigned int val) { *p = (*p & ~PTR_MASK) | val; }
+static inline unsigned int get_count(pointer p) { return (p & COUNT_MASK) >> 32; }
+static inline unsigned int get_ptr(pointer p) { return p & PTR_MASK; }
+
+typedef struct node {
+       unsigned int value;
+       pointer_t next;
+} node_t;
+
+typedef struct {
+       pointer_t head;
+       pointer_t tail;
+       node_t nodes[MAX_NODES + 1];
+} queue_t;
+
+void init_queue(queue_t *q, int num_threads);
+
+#include <list>
+using namespace std;
+/**
+       @Begin
+       @Global_define:
+               @DeclareStruct:
+               typedef struct tag_elem {
+                       Tag id;
+                       unsigned int data;
+               } tag_elem_t;
+               
+               @DeclareVar:
+               list<tag_elem_t> __queue;
+               Tag tag;
+               @InitVar:
+                       __queue = list<tag_elem_t>();
+                       tag = 1; // Beginning of available id
+       @Happens_before:
+               # Only check the happens-before relationship according to the id of the
+               # commit_point_set. For commit_point_set that has same ID, A -> B means
+               # B happens after the previous A.
+               Enqueue -> Dequeue
+       @End
+*/
+
+
+int get_thread_num();
+
 
 #define relaxed memory_order_relaxed
 #define release memory_order_release
@@ -68,7 +122,7 @@ void init_queue(queue_t *q, int num_threads)
 
        /* Initialize each thread's free list with INITIAL_FREE pointers */
        /* The actual nodes are initialized with poison indexes */
-       free_lists = malloc(num_threads * sizeof(*free_lists));
+       free_lists = (unsigned int**) malloc(num_threads * sizeof(*free_lists));
        for (i = 0; i < num_threads; i++) {
                for (j = 0; j < INITIAL_FREE; j++) {
                        free_lists[i][j] = 2 + i * MAX_FREELIST + j;
@@ -84,7 +138,16 @@ void init_queue(queue_t *q, int num_threads)
 
 /**
        @Begin
-       @Interface_define: Enqueue
+       @Interface: Enqueue
+       @Commit_point_set: Enqueue_Success_Point
+       @ID: tag++
+       @Action:
+               # __ID__ is an internal macro that refers to the id of the current
+               # interface call
+               tag_elem_t elem;
+               elem.id = __ID__;
+               elem.data = val;
+               __queue.push_back(elem);
        @End
 */
 void enqueue(queue_t *q, unsigned int val)
@@ -123,7 +186,7 @@ void enqueue(queue_t *q, unsigned int val)
                                                &tail, value, release, release);
                                /**
                                        @Begin
-                                       @Commit_point_define_check: __ATOMIC_RET__ == true
+                                       @Commit_point_define_check: commit_success == true
                                        @Label: Enqueue_Success_Point
                                        @End
                                */
@@ -137,10 +200,16 @@ void enqueue(queue_t *q, unsigned int val)
                        release, release);
 }
 
-
 /**
        @Begin
-       @Interface_define: Dequeue
+       @Interface: Dequeue
+       @Commit_point_set: Dequeue_Success_Point
+       @ID: __queue.back().id
+       @Action:
+               unsigned int _Old_Val = __queue.front().data;
+               __queue.pop_front();
+       @Post_check:
+               _Old_Val == __RET__
        @End
 */
 unsigned int dequeue(queue_t *q)
@@ -177,7 +246,7 @@ unsigned int dequeue(queue_t *q)
                                                release, release);
                                /**
                                        @Begin
-                                       @Commit_point_define_check: __ATOMIC_RET__ == true
+                                       @Commit_point_define_check: success == true
                                        @Label: Dequeue_Success_Point
                                        @End
                                */
@@ -189,3 +258,86 @@ unsigned int dequeue(queue_t *q)
        reclaim(get_ptr(head));
        return value;
 }
+
+
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <threads.h>
+
+#include "my_queue.h"
+#include "model-assert.h"
+
+static int procs = 2;
+static queue_t *queue;
+static thrd_t *threads;
+static unsigned int *input;
+static unsigned int *output;
+static int num_threads;
+
+int get_thread_num()
+{
+       thrd_t curr = thrd_current();
+       int i;
+       for (i = 0; i < num_threads; i++)
+               if (curr.priv == threads[i].priv)
+                       return i;
+       MODEL_ASSERT(0);
+       return -1;
+}
+
+static void main_task(void *param)
+{
+       unsigned int val;
+       int pid = *((int *)param);
+
+       if (!pid) {
+               input[0] = 17;
+               enqueue(queue, input[0]);
+               output[0] = dequeue(queue);
+       } else {
+               input[1] = 37;
+               enqueue(queue, input[1]);
+               output[1] = dequeue(queue);
+       }
+}
+
+int user_main(int argc, char **argv)
+{
+       int i;
+       int *param;
+       unsigned int in_sum = 0, out_sum = 0;
+
+       queue = (queue_t*) calloc(1, sizeof(*queue));
+       MODEL_ASSERT(queue);
+
+       num_threads = procs;
+       threads = (thrd_t*) malloc(num_threads * sizeof(thrd_t));
+       param = (int*) malloc(num_threads * sizeof(*param));
+       input = (unsigned int*) calloc(num_threads, sizeof(*input));
+       output = (unsigned int*) calloc(num_threads, sizeof(*output));
+
+       init_queue(queue, num_threads);
+       for (i = 0; i < num_threads; i++) {
+               param[i] = i;
+               thrd_create(&threads[i], main_task, &param[i]);
+       }
+       for (i = 0; i < num_threads; i++)
+               thrd_join(threads[i]);
+
+       for (i = 0; i < num_threads; i++) {
+               in_sum += input[i];
+               out_sum += output[i];
+       }
+       for (i = 0; i < num_threads; i++)
+               printf("input[%d] = %u\n", i, input[i]);
+       for (i = 0; i < num_threads; i++)
+               printf("output[%d] = %u\n", i, output[i]);
+       MODEL_ASSERT(in_sum == out_sum);
+
+       free(param);
+       free(threads);
+       free(queue);
+
+       return 0;
+}