Added timed_wait to condition variables.

3 files changed, 26 insertions, 5 deletions

diff --git a/src/rt/globals.h b/src/rt/globals.h
index f8025a40..85aa5860 100644
--- a/src/rt/globals.h
+++ b/src/rt/globals.h
@@ -20,12 +20,14 @@ extern "C" {
 }
 #elif defined(__GNUC__)
 #include <unistd.h>
+#include <sys/time.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <dlfcn.h>
 #include <pthread.h>
 #include <errno.h>
+#include <time.h>
 #else
 #error "Platform not supported."
 #endif
diff --git a/src/rt/sync/condition_variable.cpp b/src/rt/sync/condition_variable.cpp
index d4032572..c34ab7f4 100644
--- a/src/rt/sync/condition_variable.cpp
+++ b/src/rt/sync/condition_variable.cpp
@@ -9,14 +9,16 @@
 
 // #define TRACE
 
-condition_variable::condition_variable() {
 #if defined(__WIN32__)
+condition_variable::condition_variable() {
     _event = CreateEvent(NULL, FALSE, FALSE, NULL);
+}
 #else
+condition_variable::condition_variable() {
     pthread_cond_init(&_cond, NULL);
     pthread_mutex_init(&_mutex, NULL);
-#endif
 }
+#endif
 
 condition_variable::~condition_variable() {
 #if defined(__WIN32__)
@@ -31,15 +33,31 @@ condition_variable::~condition_variable() {
  * Wait indefinitely until condition is signaled.
  */
 void condition_variable::wait() {
+    timed_wait(0);
+}
+
+void condition_variable::timed_wait(size_t timeout_in_ns) {
 #ifdef TRACE
-    printf("waiting on condition_variable: 0x%" PRIxPTR "\n",
-           (uintptr_t)this);
+    printf("waiting on condition_variable: 0x%" PRIxPTR " for %d ns. \n",
+           (uintptr_t) this, (int) timeout_in_ns);
 #endif
 #if defined(__WIN32__)
     WaitForSingleObject(_event, INFINITE);
 #else
     pthread_mutex_lock(&_mutex);
-    pthread_cond_wait(&_cond, &_mutex);
+    // wait() automatically releases the mutex while it waits, and acquires
+    // it right before exiting. This allows signal() to acquire the mutex
+    // when signaling.)
+    if (timeout_in_ns == 0) {
+        pthread_cond_wait(&_cond, &_mutex);
+    } else {
+        timeval time_val;
+        gettimeofday(&time_val, NULL);
+        timespec time_spec;
+        time_spec.tv_sec = time_val.tv_sec + 0;
+        time_spec.tv_nsec = time_val.tv_usec * 1000 + timeout_in_ns;
+        pthread_cond_timedwait(&_cond, &_mutex, &time_spec);
+    }
     pthread_mutex_unlock(&_mutex);
 #endif
 #ifdef TRACE
diff --git a/src/rt/sync/condition_variable.h b/src/rt/sync/condition_variable.h
index f847ef99..f336a7f2 100644
--- a/src/rt/sync/condition_variable.h
+++ b/src/rt/sync/condition_variable.h
@@ -13,6 +13,7 @@ public:
     virtual ~condition_variable();
 
     void wait();
+    void timed_wait(size_t timeout_in_ns);
     void signal();
 };