X-Git-Url: http://plrg.eecs.uci.edu/git/?p=c11tester.git;a=blobdiff_plain;f=threads.cc;h=f417b3ff353fa3a59a23328c44dfb6b908f4727c;hp=762bbff34c2463a3dbaf6179e3af3e1eb977ca8b;hb=0a650d68e478aff11bf8b72ee04ed7cc6a102ce8;hpb=d3f4056998c5b286c3102be270a863fa88a1bfd5

diff --git a/threads.cc b/threads.cc
index 762bbff3..f417b3ff 100644
--- a/threads.cc
+++ b/threads.cc
@@ -5,6 +5,7 @@
 #include <string.h>
 
 #include <threads.h>
+#include <mutex>
 #include "common.h"
 #include "threads-model.h"
 #include "action.h"
@@ -197,3 +198,19 @@ void Thread::set_state(thread_state s)
 	ASSERT(s == THREAD_COMPLETED || state != THREAD_COMPLETED);
 	state = s;
 }
+
+/**
+ * Get the Thread that this Thread is waiting on
+ * @return The thread we are waiting on, if any; otherwise NULL
+ */
+Thread * Thread::waiting_on() const
+{
+	if (!pending)
+		return NULL;
+
+	if (pending->get_type() == THREAD_JOIN)
+		return pending->get_thread_operand();
+	else if (pending->is_lock())
+		return (Thread *)pending->get_mutex()->get_state()->locked;
+	return NULL;
+}