From 92231d488633e10efddb84b8423fc368f21516e0 Mon Sep 17 00:00:00 2001
From: Andrew Stitcher <astitcher@apache.org>
Date: Tue, 23 Jun 2009 20:02:29 +0000
Subject: Add blocking to broker::Timer/TimerTask so cancel and fire() cannot
 be interleaved.

git-svn-id: https://svn.apache.org/repos/asf/qpid/trunk/qpid@787812 13f79535-47bb-0310-9956-ffa450edef68
---
 cpp/src/qpid/broker/Timer.cpp | 15 +++++++++++++--
 cpp/src/qpid/broker/Timer.h   |  3 ++-
 2 files changed, 15 insertions(+), 3 deletions(-)

(limited to 'cpp')

diff --git a/cpp/src/qpid/broker/Timer.cpp b/cpp/src/qpid/broker/Timer.cpp
index 53a1597deb..7832204e2a 100644
--- a/cpp/src/qpid/broker/Timer.cpp
+++ b/cpp/src/qpid/broker/Timer.cpp
@@ -26,6 +26,8 @@ using qpid::sys::AbsTime;
 using qpid::sys::Duration;
 using qpid::sys::Monitor;
 using qpid::sys::Thread;
+using qpid::sys::Mutex;
+using qpid::sys::ScopedLock;
 using namespace qpid::broker;
 
 TimerTask::TimerTask(Duration timeout) :
@@ -38,7 +40,11 @@ TimerTask::~TimerTask(){}
 
 void TimerTask::reset() { time = AbsTime(AbsTime::now(), duration); }
 
-void TimerTask::cancel() { cancelled = true; }
+void TimerTask::cancel() {
+    ScopedLock<Mutex> l(cancelLock);
+    cancelled = true;
+}
+
 bool TimerTask::isCancelled() const { return cancelled; }
 
 Timer::Timer() : active(false)
@@ -60,16 +66,21 @@ void Timer::run()
         } else {
             intrusive_ptr<TimerTask> t = tasks.top();
             tasks.pop();
+            {
+            ScopedLock<Mutex> l(t->cancelLock);
             if (t->isCancelled()) {
+                continue;
             } else if(t->time < AbsTime::now()) {
                 Monitor::ScopedUnlock u(monitor);
                 t->fire();
+                continue;
             } else {
                 // If the timer was adjusted into the future it might no longer
                 // be the next event, so push and then get top to make sure
                 tasks.push(t);
-                monitor.wait(tasks.top()->time);
             }
+            }
+            monitor.wait(tasks.top()->time);
         }
     }
 }
diff --git a/cpp/src/qpid/broker/Timer.h b/cpp/src/qpid/broker/Timer.h
index 564fec5804..7218fb0e79 100644
--- a/cpp/src/qpid/broker/Timer.h
+++ b/cpp/src/qpid/broker/Timer.h
@@ -43,6 +43,7 @@ struct TimerTask : public RefCounted {
     const qpid::sys::Duration duration;
     qpid::sys::AbsTime time;
     volatile bool cancelled;
+    qpid::sys::Mutex cancelLock;
 
     QPID_BROKER_EXTERN TimerTask(qpid::sys::Duration timeout);
     TimerTask(qpid::sys::AbsTime time);
@@ -60,7 +61,7 @@ struct Later {
 
 class Timer : private qpid::sys::Runnable {
   protected:
-    qpid::sys::Monitor monitor;            
+    qpid::sys::Monitor monitor;
     std::priority_queue<boost::intrusive_ptr<TimerTask>,
                         std::vector<boost::intrusive_ptr<TimerTask> >,
                         Later> tasks;
-- 
cgit v1.2.1