small changes

[c11tester.git] / threads.cc

diff --git a/threads.cc b/threads.cc
index f83433bc1a0da8696fd3f0d8497f76fcc1943c81..e22ffa3221f83ae1f95abb5b7b6c63da04c72183 100644 (file)
--- a/threads.cc
+++ b/threads.cc
@@ -508,13 +508,15 @@ Thread * Thread::waiting_on() const
        if (!pending)
                return NULL;
 
-       if (pending->get_type() == THREAD_JOIN)
-               return pending->get_thread_operand();
-       else if (pending->get_type() == PTHREAD_JOIN)
-               return pending->get_thread_operand();
-       else if (pending->is_lock())
-               return (Thread *)pending->get_mutex()->get_state()->locked;
-       return NULL;
+       switch (pending->get_type()) {
+               case THREAD_JOIN:
+               case PTHREAD_JOIN:
+                       return pending->get_thread_operand();
+               case ATOMIC_LOCK:
+                       return (Thread *)pending->get_mutex()->get_state()->locked;
+               default:
+                       return NULL;
+       }
 }
 
 /**