Unregister maps on game server disconnection.

1 files changed, 15 insertions, 0 deletions

diff --git a/src/account-server/serverhandler.cpp b/src/account-server/serverhandler.cpp
index 5300a066..99d872f8 100644
--- a/src/account-server/serverhandler.cpp
+++ b/src/account-server/serverhandler.cpp
@@ -43,6 +43,19 @@ NetComputer *ServerHandler::computerConnected(ENetPeer *peer)
 
 void ServerHandler::computerDisconnected(NetComputer *comp)
 {
+    Servers::iterator i = servers.begin();
+    while (i != servers.end())
+    {
+        if (i->second.server == comp)
+        {
+            LOG_INFO("Unregistering map " << i->first << '.', 0);
+            servers.erase(i++);
+        }
+        else
+        {
+            ++i;
+        }
+    }
     delete comp;
 }
 
@@ -93,9 +106,11 @@ void ServerHandler::processMessage(NetComputer *comp, MessageIn &msg)
             LOG_INFO("Game server " << address << ':' << port
                      << " wants to register " << (msg.getUnreadLength() / 2)
                      << " maps.", 0);
+            
             while (msg.getUnreadLength())
             {
                 unsigned id = msg.readShort();
+                LOG_INFO("Registering map " << id << '.', 0);
                 if (!servers.insert(std::make_pair(id, s)).second)
                 {
                     LOG_ERROR("Server Handler: map is already registered.", 0);