Version 0.4.

[loba.git] / synchro.h

diff --git a/synchro.h b/synchro.h
index 7b04f9241f24a2dd91d855c075270b19cfb9b70c..bd49783ccdc943f2710ed76f9e4e60787fb0b5cf 100644 (file)
--- a/synchro.h
+++ b/synchro.h
@@ -9,10 +9,11 @@ public:
     ~mutex_t()                  { xbt_mutex_destroy(mutex); }
     void acquire()              { xbt_mutex_acquire(mutex); }
     void release()              { xbt_mutex_release(mutex); }
-    xbt_mutex_t get()           { return mutex;             }
 
 private:
     xbt_mutex_t mutex;
+
+    friend class condition_t;
 };
 
 class condition_t {
@@ -21,9 +22,13 @@ public:
     ~condition_t()              { xbt_cond_destroy(cond);           }
     void broadcast()            { xbt_cond_broadcast(cond);         }
     void signal()               { xbt_cond_signal(cond);            }
-    void wait(mutex_t& mutex)   { xbt_cond_wait(cond, mutex.get()); }
-    void timedwait(mutex_t& mutex, double delay) {
-        xbt_cond_timedwait(cond, mutex.get(), delay);
+    void wait(mutex_t& mutex)   { xbt_cond_wait(cond, mutex.mutex); }
+    bool timedwait(mutex_t& mutex, double delay) {
+        if (xbt_cond_timedwait(cond, mutex.mutex, delay)) {
+            mutex.acquire();
+            return false;       // got a timeout
+        }
+        return true;
     }
 
 private: