2 files changed, 34 insertions, 7 deletions
diff --git a/src/Client/InformationManager.cpp b/src/Client/InformationManager.cpp
index f1627f4..8fde6f7 100644
--- a/src/Client/InformationManager.cpp
+++ b/src/Client/InformationManager.cpp
@@ -45,11 +45,15 @@ void InformationManager::DaemonStateUpdateRequestHandler::handlePacket(const Com
 
   // TODO Require authentication
 
-  std::map<std::string, Common::HostInfo>::iterator host = informationManager.get()->daemons.find(packet["name"]);
-  if(host != informationManager.get()->daemons.end())
-    host->second.setState(packet["state"]);
-  else
-    Common::Logger::log(Common::Logger::WARNING, "Received a state update for an unknown host.");
+  {
+    boost::lock_guard<boost::mutex> lock(informationManager.mutex);
+
+    std::map<std::string, Common::HostInfo>::iterator host = informationManager.daemons.find(packet["name"]);
+    if(host != informationManager.daemons.end())
+      host->second.setState(packet["state"]);
+    else
+      Common::Logger::log(Common::Logger::WARNING, "Received a state update for an unknown host.");
+  }
 
   Common::XmlPacket ret;
   ret.setType("OK");
@@ -68,6 +72,8 @@ void InformationManager::doDeinit() {
 }
 
 void InformationManager::updateDaemonList(Common::Connection *con) {
+  boost::lock_guard<boost::mutex> lock(mutex);
+
   if(updating)
     return;
 
@@ -78,6 +84,8 @@ void InformationManager::updateDaemonList(Common::Connection *con) {
 }
 
 void InformationManager::daemonListRequestFinished(const Common::Request &request) {
+  boost::lock_guard<boost::mutex> lock(mutex);
+
   try {
     const Common::XmlPacket &packet = request.getResult();
 
@@ -99,6 +107,8 @@ void InformationManager::daemonListRequestFinished(const Common::Request &reques
   }
 
   updating = false;
+
+  updateCond.notify_all();
 }
 
 }
diff --git a/src/Client/InformationManager.h b/src/Client/InformationManager.h
index fff26b3..2c39601 100644
--- a/src/Client/InformationManager.h
+++ b/src/Client/InformationManager.h
@@ -27,6 +27,10 @@
 #include <Common/Initializable.h>
 #include <Common/Request.h>
 
+#include <boost/thread/condition_variable.hpp>
+#include <boost/thread/locks.hpp>
+#include <boost/thread/mutex.hpp>
+
 namespace Mad {
 namespace Client {
 
@@ -43,6 +47,9 @@ class InformationManager : public Common::Initializable {
 
     static InformationManager informationManager;
 
+    boost::mutex mutex;
+    boost::condition_variable updateCond;
+
     std::map<std::string, Common::HostInfo> daemons;
 
     bool updating;
@@ -66,11 +73,21 @@ class InformationManager : public Common::Initializable {
 
     void updateDaemonList(Common::Connection *con);
 
-    bool isUpdating() const {
+    bool isUpdating() {
+      boost::lock_guard<boost::mutex> lock(mutex);
       return updating;
     }
 
-    const std::map<std::string, Common::HostInfo>& getDaemons() const {
+    void waitWhileUpdating() {
+      boost::unique_lock<boost::mutex> lock(mutex);
+
+      while(updating)
+        updateCond.wait(lock);
+    }
+
+
+    std::map<std::string, Common::HostInfo> getDaemons() {
+      boost::lock_guard<boost::mutex> lock(mutex);
       return daemons;
     }
 };