]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_spanningtree/treeserver.cpp
m_spanningtree Remove split servers from the hash maps in TreeServer::SQuitInternal()
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / treeserver.cpp
index ecde1726148c40f1735c639345ed41f4eb77db6c..5a2d4f8861b74757bdf990b163179beabb770e49 100644 (file)
 TreeServer::TreeServer()
        : Server(ServerInstance->Config->ServerName, ServerInstance->Config->ServerDesc)
        , Parent(NULL), Route(NULL)
-       , VersionString(ServerInstance->GetVersionString()), Socket(NULL), sid(ServerInstance->Config->GetSID()), ServerUser(ServerInstance->FakeClient)
-       , age(ServerInstance->Time()), Warned(false), bursting(false), UserCount(0), OperCount(0), rtt(0), StartBurst(0), Hidden(false)
+       , VersionString(ServerInstance->GetVersionString())
+       , fullversion(ServerInstance->GetVersionString(true))
+       , Socket(NULL), sid(ServerInstance->Config->GetSID()), behind_bursting(0), ServerUser(ServerInstance->FakeClient)
+       , age(ServerInstance->Time()), Warned(false), UserCount(ServerInstance->Users.GetLocalUsers().size())
+       , OperCount(0), rtt(0), StartBurst(0), Hidden(false)
 {
        AddHashEntry();
 }
@@ -47,17 +50,14 @@ TreeServer::TreeServer()
  */
 TreeServer::TreeServer(const std::string& Name, const std::string& Desc, const std::string& id, TreeServer* Above, TreeSocket* Sock, bool Hide)
        : Server(Name, Desc)
-       , Parent(Above), Socket(Sock), sid(id), ServerUser(new FakeUser(id, this))
-       , age(ServerInstance->Time()), Warned(false), bursting(true), UserCount(0), OperCount(0), rtt(0), Hidden(Hide)
+       , Parent(Above), Socket(Sock), sid(id), behind_bursting(Parent->behind_bursting), ServerUser(new FakeUser(id, this))
+       , age(ServerInstance->Time()), Warned(false), UserCount(0), OperCount(0), rtt(0), StartBurst(0), Hidden(Hide)
 {
+       ServerInstance->Logs->Log(MODNAME, LOG_DEBUG, "New server %s behind_bursting %u", GetName().c_str(), behind_bursting);
        CheckULine();
        SetNextPingTime(ServerInstance->Time() + Utils->PingFreq);
        SetPingFlag();
 
-       long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
-       this->StartBurst = ts;
-       ServerInstance->Logs->Log(MODNAME, LOG_DEBUG, "Server %s started bursting at time %lu", sid.c_str(), ts);
-
        /* find the 'route' for this server (e.g. the one directly connected
         * to the local server, which we can use to reach it)
         *
@@ -110,6 +110,19 @@ TreeServer::TreeServer(const std::string& Name, const std::string& Desc, const s
         */
 
        this->AddHashEntry();
+       Parent->AddChild(this);
+}
+
+void TreeServer::BeginBurst(unsigned long startms)
+{
+       behind_bursting++;
+
+       unsigned long now = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
+       // If the start time is in the future (clocks are not synced) then use current time
+       if ((!startms) || (startms > now))
+               startms = now;
+       this->StartBurst = startms;
+       ServerInstance->Logs->Log(MODNAME, LOG_DEBUG, "Server %s started bursting at time %lu behind_bursting %u", sid.c_str(), startms, behind_bursting);
 }
 
 const std::string& TreeServer::GetID()
@@ -119,9 +132,17 @@ const std::string& TreeServer::GetID()
 
 void TreeServer::FinishBurstInternal()
 {
-       this->bursting = false;
-       SetNextPingTime(ServerInstance->Time() + Utils->PingFreq);
-       SetPingFlag();
+       // Check is needed because 1202 protocol servers don't send the bursting state of a server, so servers
+       // introduced during a netburst may later send ENDBURST which would normally decrease this counter
+       if (behind_bursting > 0)
+               behind_bursting--;
+       ServerInstance->Logs->Log(MODNAME, LOG_DEBUG, "FinishBurstInternal() %s behind_bursting %u", GetName().c_str(), behind_bursting);
+
+       if (!IsBehindBursting())
+       {
+               SetNextPingTime(ServerInstance->Time() + Utils->PingFreq);
+               SetPingFlag();
+       }
        for (ChildServers::const_iterator i = Children.begin(); i != Children.end(); ++i)
        {
                TreeServer* child = *i;
@@ -131,20 +152,75 @@ void TreeServer::FinishBurstInternal()
 
 void TreeServer::FinishBurst()
 {
-       FinishBurstInternal();
        ServerInstance->XLines->ApplyLines();
        long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
        unsigned long bursttime = ts - this->StartBurst;
        ServerInstance->SNO->WriteToSnoMask(Parent == Utils->TreeRoot ? 'l' : 'L', "Received end of netburst from \2%s\2 (burst time: %lu %s)",
                GetName().c_str(), (bursttime > 10000 ? bursttime / 1000 : bursttime), (bursttime > 10000 ? "secs" : "msecs"));
        AddServerEvent(Utils->Creator, GetName());
+
+       StartBurst = 0;
+       FinishBurstInternal();
+}
+
+void TreeServer::SQuitChild(TreeServer* server, const std::string& reason)
+{
+       DelServerEvent(Utils->Creator, server->GetName());
+       DelChild(server);
+
+       if (IsRoot())
+       {
+               // Server split from us, generate a SQUIT message and broadcast it
+               ServerInstance->SNO->WriteGlobalSno('l', "Server \002" + server->GetName() + "\002 split: " + reason);
+               CmdBuilder("SQUIT").push(server->GetID()).push_last(reason).Broadcast();
+       }
+       else
+       {
+               ServerInstance->SNO->WriteToSnoMask('L', "Server \002" + server->GetName() + "\002 split from server \002" + GetName() + "\002 with reason: " + reason);
+       }
+
+       int num_lost_servers = 0;
+       int num_lost_users = 0;
+       const std::string quitreason = GetName() + " " + server->GetName();
+
+       ModuleSpanningTree* st = Utils->Creator;
+       st->SplitInProgress = true;
+       server->SQuitInternal(quitreason, num_lost_servers, num_lost_users);
+       st->SplitInProgress = false;
+
+       ServerInstance->SNO->WriteToSnoMask(IsRoot() ? 'l' : 'L', "Netsplit complete, lost \002%d\002 user%s on \002%d\002 server%s.",
+               num_lost_users, num_lost_users != 1 ? "s" : "", num_lost_servers, num_lost_servers != 1 ? "s" : "");
+
+       server->Tidy();
+
+       // No-op if the socket is already closed (i.e. it called us)
+       if (server->IsLocal())
+               server->GetSocket()->Close();
+
+       server->cull();
+       delete server;
+}
+
+void TreeServer::SQuitInternal(const std::string& reason, int& num_lost_servers, int& num_lost_users)
+{
+       ServerInstance->Logs->Log(MODNAME, LOG_DEBUG, "Server %s lost in split", GetName().c_str());
+
+       for (ChildServers::const_iterator i = Children.begin(); i != Children.end(); ++i)
+       {
+               TreeServer* server = *i;
+               server->SQuitInternal(reason, num_lost_servers, num_lost_users);
+       }
+
+       num_lost_servers++;
+       num_lost_users += QuitUsers(reason);
+       RemoveHash();
 }
 
 int TreeServer::QuitUsers(const std::string &reason)
 {
        std::string publicreason = ServerInstance->Config->HideSplits ? "*.net *.split" : reason;
 
-       const user_hash& users = *ServerInstance->Users->clientlist;
+       const user_hash& users = ServerInstance->Users->GetUsers();
        unsigned int original_size = users.size();
        for (user_hash::const_iterator i = users.begin(); i != users.end(); )
        {
@@ -160,11 +236,24 @@ int TreeServer::QuitUsers(const std::string &reason)
 void TreeServer::CheckULine()
 {
        uline = silentuline = false;
-       std::map<irc::string, bool>::iterator it = ServerInstance->Config->ulines.find(GetName().c_str());
-       if (it != ServerInstance->Config->ulines.end())
+
+       ConfigTagList tags = ServerInstance->Config->ConfTags("uline");
+       for (ConfigIter i = tags.first; i != tags.second; ++i)
        {
-               uline = true;
-               silentuline = it->second;
+               ConfigTag* tag = i->second;
+               std::string server = tag->getString("server");
+               if (!strcasecmp(server.c_str(), GetName().c_str()))
+               {
+                       if (this->IsRoot())
+                       {
+                               ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "Servers should not uline themselves (at " + tag->getTagLocation() + ")");
+                               return;
+                       }
+
+                       uline = true;
+                       silentuline = tag->getBool("silent");
+                       break;
+               }
        }
 }
 
@@ -274,7 +363,10 @@ TreeServer::~TreeServer()
        /* We'd better tidy up after ourselves, eh? */
        if (!IsRoot())
                delete ServerUser;
+}
 
+void TreeServer::RemoveHash()
+{
        Utils->sidlist.erase(sid);
        Utils->serverlist.erase(GetName());
 }