X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=ms-queue%2Fqueue.c;h=c7b5de58dd3369af21e85c98feb1aa0461176eb9;hb=b684f62b7411fea476b2e1f6a8bbf920ac4c7216;hp=0a5f3bd38a9a77c218a9be970dc67fb95fff4854;hpb=1438eb7c0715e53611a717e593bfa3fe1bd30588;p=model-checker-benchmarks.git

diff --git a/ms-queue/queue.c b/ms-queue/queue.c
index 0a5f3bd..c7b5de5 100644
--- a/ms-queue/queue.c
+++ b/ms-queue/queue.c
@@ -33,6 +33,13 @@ static unsigned int new_node()
 	return 0;
 }
 
+/* Simulate the fact that when a node got recycled, it will get assigned to the
+ * same queue or for other usage */
+void simulateRecycledNodeUpdate(queue_t *q, unsigned int node) {
+	atomic_store_explicit(&q->nodes[node].next, -1, memory_order_release);
+}
+
+
 /* Place this node index back on this thread's free list */
 static void reclaim(unsigned int node)
 {
@@ -76,7 +83,7 @@ void init_queue(queue_t *q, int num_threads)
 	atomic_init(&q->nodes[1].next, MAKE_POINTER(0, 0));
 }
 
-void enqueue(queue_t *q, unsigned int val)
+void enqueue(queue_t *q, unsigned int val, bool yield)
 {
 	int success = 0;
 	unsigned int node;
@@ -120,7 +127,7 @@ void enqueue(queue_t *q, unsigned int val)
 			release, release);
 }
 
-bool dequeue(queue_t *q, unsigned int *retVal)
+bool dequeue(queue_t *q, unsigned int *retVal, unsigned int *reclaimNode)
 {
 	int success = 0;
 	pointer head;
@@ -156,6 +163,7 @@ bool dequeue(queue_t *q, unsigned int *retVal)
 			}
 		}
 	}
+	reclaimNode = get_ptr(head);
 	reclaim(get_ptr(head));
 	return true;
 }