]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_spanningtree/utils.cpp
Use it here, too
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / utils.cpp
index 0ba9069171b0607228a3d79cc248caf9d5b569b7..1e0788623e5d10f4c61493c05c68ea5a079bff71 100644 (file)
@@ -1,11 +1,20 @@
-#include "configreader.h"
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
+ *
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
+ * This program is free but copyrighted software; see
+ *            the file COPYING for details.
+ *
+ * ---------------------------------------------------
+ */
+
+#include "inspircd.h"
 #include "commands/cmd_whois.h"
 #include "commands/cmd_stats.h"
 #include "socket.h"
-#include "inspircd.h"
 #include "wildcard.h"
 #include "xline.h"
 #include "transport.h"
 #include "m_spanningtree/treesocket.h"
 #include "m_spanningtree/resolvers.h"
 
+/* $ModDep: m_spanningtree/timesynctimer.h m_spanningtree/resolvers.h m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/link.h m_spanningtree/treesocket.h */
+
+bool SpanningTreeUtilities::IsSID(const std::string &str)
+{
+       /* Returns true if the string given is exactly 3 characters long,
+        * starts with a digit, and has no '.' in the other 2
+        */
+       return ((str.length() == 3) && isdigit(str[0]) && (str[1] != '.' && str[2] != '.'));
+}
+
 /** Yay for fast searches!
  * This is hundreds of times faster than recursion
  * or even scanning a linked list, especially when
  */
 TreeServer* SpanningTreeUtilities::FindServer(const std::string &ServerName)
 {
-       server_hash::iterator iter;
-       iter = serverlist.find(ServerName.c_str());
+       if (IsSID(ServerName))
+               return this->FindServerID(ServerName);
+
+       server_hash::iterator iter = serverlist.find(ServerName.c_str());
        if (iter != serverlist.end())
        {
                return iter->second;
@@ -38,6 +59,66 @@ TreeServer* SpanningTreeUtilities::FindServer(const std::string &ServerName)
        }
 }
 
+TreeServer* SpanningTreeUtilities::FindRemoteBurstServer(TreeServer* Server)
+{
+       server_hash::iterator iter = RemoteServersBursting.find(Server->GetName().c_str());
+       if (iter != RemoteServersBursting.end())
+               return iter->second;
+       else
+               return NULL;
+}
+
+TreeSocket* SpanningTreeUtilities::FindBurstingServer(const std::string &ServerName)
+{
+       std::map<irc::string,TreeSocket*>::iterator iter;
+       iter = burstingserverlist.find(ServerName.c_str());
+       if (iter != burstingserverlist.end())
+       {
+               return iter->second;
+       }
+       else
+       {
+               return NULL;
+       }
+}
+
+void SpanningTreeUtilities::SetRemoteBursting(TreeServer* Server, bool bursting)
+{
+       server_hash::iterator iter = RemoteServersBursting.find(Server->GetName().c_str());
+       if (bursting)
+       {
+               if (iter == RemoteServersBursting.end())
+                       RemoteServersBursting.insert(make_pair(Server->GetName(), Server));
+               else return;
+       }
+       else
+       {
+               if (iter != RemoteServersBursting.end())
+                       RemoteServersBursting.erase(iter);
+               else return;
+       }
+       ServerInstance->Log(DEBUG,"Server %s is %sbursting nicknames", Server->GetName().c_str(), bursting ? "" : "no longer ");
+}
+
+void SpanningTreeUtilities::AddBurstingServer(const std::string &ServerName, TreeSocket* s)
+{
+       std::map<irc::string,TreeSocket*>::iterator iter = burstingserverlist.find(ServerName.c_str());
+       if (iter == burstingserverlist.end())
+               burstingserverlist[ServerName.c_str()] = s;
+}
+
+void SpanningTreeUtilities::DelBurstingServer(TreeSocket* s)
+{
+        for (std::map<irc::string,TreeSocket*>::iterator iter = burstingserverlist.begin(); iter != burstingserverlist.end(); iter++)
+        {
+                if (iter->second == s)
+                {
+                        burstingserverlist.erase(iter);
+                        return;
+                }
+        }
+}
+
 /** Returns the locally connected server we must route a
  * message through to reach server 'ServerName'. This
  * only applies to one-to-one and not one-to-many routing.
@@ -75,6 +156,16 @@ TreeServer* SpanningTreeUtilities::FindServerMask(const std::string &ServerName)
        return NULL;
 }
 
+TreeServer* SpanningTreeUtilities::FindServerID(const std::string &id)
+{
+       ServerInstance->Log(DEBUG,"Looking for id: %s", id.c_str());
+       server_hash::iterator iter = sidlist.find(id);
+       if (iter != sidlist.end())
+               return iter->second;
+       else
+               return NULL;
+}
+
 /* A convenient wrapper that returns true if a server exists */
 bool SpanningTreeUtilities::IsServer(const std::string &ServerName)
 {
@@ -87,7 +178,9 @@ SpanningTreeUtilities::SpanningTreeUtilities(InspIRCd* Instance, ModuleSpanningT
 
        lines_to_apply = 0;
 
-       this->TreeRoot = new TreeServer(this, ServerInstance, ServerInstance->Config->ServerName, ServerInstance->Config->ServerDesc);
+       ServerInstance->Log(DEBUG, "SpanningTreeUtilities: SID: %s", OurSID.c_str());
+
+       this->TreeRoot = new TreeServer(this, ServerInstance, ServerInstance->Config->ServerName, ServerInstance->Config->ServerDesc, ServerInstance->Config->GetSID());
 
        modulelist* ml = ServerInstance->FindInterface("InspSocketHook");
 
@@ -116,7 +209,6 @@ SpanningTreeUtilities::~SpanningTreeUtilities()
        {
                ServerInstance->SE->DelFd(Bindings[i]);
                Bindings[i]->Close();
-               DELETE(Bindings[i]);
        }
        while (TreeRoot->ChildCount())
        {
@@ -126,10 +218,10 @@ SpanningTreeUtilities::~SpanningTreeUtilities()
                        TreeSocket* sock = child_server->GetSocket();
                        ServerInstance->SE->DelFd(sock);
                        sock->Close();
-                       DELETE(sock);
                }
        }
        delete TreeRoot;
+       ServerInstance->InspSocketCull();
 }
 
 void SpanningTreeUtilities::AddThisServer(TreeServer* server, TreeServerList &list)
@@ -159,9 +251,9 @@ void SpanningTreeUtilities::GetListOfServersForChannel(chanrec* c, TreeServerLis
        }
        for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
        {
-               if ((i->second->GetFd() < 0) && (exempt_list.find(i->second) == exempt_list.end()))
+               if ((i->first->GetFd() < 0) && (exempt_list.find(i->first) == exempt_list.end()))
                {
-                       TreeServer* best = this->BestRouteTo(i->second->server);
+                       TreeServer* best = this->BestRouteTo(i->first->server);
                        if (best)
                                AddThisServer(best,list);
                }
@@ -330,12 +422,58 @@ bool SpanningTreeUtilities::DoOneToOne(const std::string &prefix, const std::str
        }
 }
 
+void SpanningTreeUtilities::RefreshIPCache()
+{
+       ValidIPs.clear();
+       for (std::vector<Link>::iterator L = LinkBlocks.begin(); L != LinkBlocks.end(); L++)
+       {
+               if ((!L->IPAddr.empty()) && (!L->RecvPass.empty()) && (!L->SendPass.empty()) && (!L->Name.empty()) && (L->Port))
+               {
+                       ValidIPs.push_back(L->IPAddr);
+
+                       if (L->AllowMask.length())
+                               ValidIPs.push_back(L->AllowMask);
+
+                       /* Needs resolving */
+                       bool ipvalid = true;
+                       QueryType start_type = DNS_QUERY_A;
+#ifdef IPV6
+                       start_type = DNS_QUERY_AAAA;
+                       if (strchr(L->IPAddr.c_str(),':'))
+                       {
+                               in6_addr n;
+                               if (inet_pton(AF_INET6, L->IPAddr.c_str(), &n) < 1)
+                                       ipvalid = false;
+                       }
+                       else
+#endif
+                       {
+                               in_addr n;
+                               if (inet_aton(L->IPAddr.c_str(),&n) < 1)
+                                       ipvalid = false;
+                       }
+                       if (!ipvalid)
+                       {
+                               try
+                               {
+                                       bool cached;
+                                       SecurityIPResolver* sr = new SecurityIPResolver((Module*)this->Creator, this, ServerInstance, L->IPAddr, *L, cached, start_type);
+                                       ServerInstance->AddResolver(sr, cached);
+                               }
+                               catch (...)
+                               {
+                               }
+                       }
+               }
+       }
+}
+
 void SpanningTreeUtilities::ReadConfiguration(bool rebind)
 {
        ConfigReader* Conf = new ConfigReader(ServerInstance);
        if (rebind)
        {
-               for (int j =0; j < Conf->Enumerate("bind"); j++)
+               for (int j = 0; j < Conf->Enumerate("bind"); j++)
                {
                        std::string Type = Conf->ReadValue("bind","type",j);
                        std::string IP = Conf->ReadValue("bind","address",j);
@@ -348,7 +486,7 @@ void SpanningTreeUtilities::ReadConfiguration(bool rebind)
                                while ((portno = portrange.GetToken()))
                                {
                                        if (IP == "*")
-                                               IP = "";
+                                               IP.clear();
 
                                        if ((!transport.empty()) && (hooks.find(transport.c_str()) ==  hooks.end()))
                                        {
@@ -364,9 +502,8 @@ void SpanningTreeUtilities::ReadConfiguration(bool rebind)
                                        }
                                        else
                                        {
-                                               ServerInstance->Log(DEFAULT,"m_spanningtree: Warning: Failed to bind server port %s:%d",IP.c_str(), portno);
+                                               ServerInstance->Log(DEFAULT,"m_spanningtree: Warning: Failed to bind server port: %s:%d: %s",IP.c_str(), portno, strerror(errno));
                                                listener->Close();
-                                               DELETE(listener);
                                        }
                                }
                        }
@@ -375,23 +512,38 @@ void SpanningTreeUtilities::ReadConfiguration(bool rebind)
        FlatLinks = Conf->ReadFlag("options","flatlinks",0);
        HideULines = Conf->ReadFlag("options","hideulines",0);
        AnnounceTSChange = Conf->ReadFlag("options","announcets",0);
-       EnableTimeSync = !(Conf->ReadFlag("options","notimesync",0));
+       EnableTimeSync = Conf->ReadFlag("timesync","enable",0);
+       MasterTime = Conf->ReadFlag("timesync", "master", 0);
+       ChallengeResponse = !Conf->ReadFlag("options", "disablehmac", 0);
+       quiet_bursts = Conf->ReadFlag("options", "quietbursts", 0);
+       PingWarnTime = Conf->ReadInteger("options", "pingwarning", 0, true);
+       PingFreq = Conf->ReadInteger("options", "serverpingfreq", 0, true);
+
+       if (PingFreq == 0)
+               PingFreq = 60;
+
+       if (PingWarnTime < 0 || PingWarnTime > PingFreq - 1)
+               PingWarnTime = 0;
+
        LinkBlocks.clear();
        ValidIPs.clear();
-       for (int j =0; j < Conf->Enumerate("link"); j++)
+       for (int j = 0; j < Conf->Enumerate("link"); j++)
        {
                Link L;
-               std::string Allow = Conf->ReadValue("link","allowmask",j);
-               L.Name = (Conf->ReadValue("link","name",j)).c_str();
-               L.IPAddr = Conf->ReadValue("link","ipaddr",j);
-               L.FailOver = Conf->ReadValue("link","failover",j).c_str();
-               L.Port = Conf->ReadInteger("link","port",j,true);
-               L.SendPass = Conf->ReadValue("link","sendpass",j);
-               L.RecvPass = Conf->ReadValue("link","recvpass",j);
-               L.AutoConnect = Conf->ReadInteger("link","autoconnect",j,true);
-               L.HiddenFromStats = Conf->ReadFlag("link","hidden",j);
-               L.Timeout = Conf->ReadInteger("link","timeout",j,true);
+               std::string Allow = Conf->ReadValue("link", "allowmask", j);
+               L.Name = (Conf->ReadValue("link", "name", j)).c_str();
+               L.AllowMask = Allow;
+               L.IPAddr = Conf->ReadValue("link", "ipaddr", j);
+               L.FailOver = Conf->ReadValue("link", "failover", j).c_str();
+               L.Port = Conf->ReadInteger("link", "port", j, true);
+               L.SendPass = Conf->ReadValue("link", "sendpass", j);
+               L.RecvPass = Conf->ReadValue("link", "recvpass", j);
+               L.AutoConnect = Conf->ReadInteger("link", "autoconnect", j, true);
+               L.HiddenFromStats = Conf->ReadFlag("link", "statshidden", j);
+               L.Timeout = Conf->ReadInteger("link", "timeout", j, true);
                L.Hook = Conf->ReadValue("link", "transport", j);
+               L.Bind = Conf->ReadValue("link", "bind", j);
+               L.Hidden = Conf->ReadFlag("link", "hidden", j);
 
                if ((!L.Hook.empty()) && (hooks.find(L.Hook.c_str()) ==  hooks.end()))
                {
@@ -405,7 +557,7 @@ void SpanningTreeUtilities::ReadConfiguration(bool rebind)
                /* Bugfix by brain, do not allow people to enter bad configurations */
                if (L.Name != ServerInstance->Config->ServerName)
                {
-                       if ((L.IPAddr != "") && (L.RecvPass != "") && (L.SendPass != "") && (L.Name != "") && (L.Port))
+                       if ((!L.IPAddr.empty()) && (!L.RecvPass.empty()) && (!L.SendPass.empty()) && (!L.Name.empty()) && (L.Port))
                        {
                                ValidIPs.push_back(L.IPAddr);
 
@@ -413,16 +565,37 @@ void SpanningTreeUtilities::ReadConfiguration(bool rebind)
                                        ValidIPs.push_back(Allow);
 
                                /* Needs resolving */
-                               insp_inaddr binip;
-                               if (insp_aton(L.IPAddr.c_str(), &binip) < 1)
+                               bool ipvalid = true;
+                               QueryType start_type = DNS_QUERY_A;
+#ifdef IPV6
+                               start_type = DNS_QUERY_AAAA;
+                               if (strchr(L.IPAddr.c_str(),':'))
+                               {
+                                       in6_addr n;
+                                       if (inet_pton(AF_INET6, L.IPAddr.c_str(), &n) < 1)
+                                               ipvalid = false;
+                               }
+                               else
+                               {
+                                       in_addr n;
+                                       if (inet_aton(L.IPAddr.c_str(),&n) < 1)
+                                               ipvalid = false;
+                               }
+#else
+                               in_addr n;
+                               if (inet_aton(L.IPAddr.c_str(),&n) < 1)
+                                       ipvalid = false;
+#endif
+
+                               if (!ipvalid)
                                {
                                        try
                                        {
                                                bool cached;
-                                               SecurityIPResolver* sr = new SecurityIPResolver((Module*)this->Creator, this, ServerInstance, L.IPAddr, L, cached);
+                                               SecurityIPResolver* sr = new SecurityIPResolver((Module*)this->Creator, this, ServerInstance, L.IPAddr, L, cached, start_type);
                                                ServerInstance->AddResolver(sr, cached);
                                        }
-                                       catch (ModuleException& e)
+                                       catch (...)
                                        {
                                        }
                                }
@@ -431,19 +604,19 @@ void SpanningTreeUtilities::ReadConfiguration(bool rebind)
                        }
                        else
                        {
-                               if (L.IPAddr == "")
+                               if (L.IPAddr.empty())
                                {
                                        ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', IP address not defined!",L.Name.c_str());
                                }
-                               else if (L.RecvPass == "")
+                               else if (L.RecvPass.empty())
                                {
                                        ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', recvpass not defined!",L.Name.c_str());
                                }
-                               else if (L.SendPass == "")
+                               else if (L.SendPass.empty())
                                {
                                        ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', sendpass not defined!",L.Name.c_str());
                                }
-                               else if (L.Name == "")
+                               else if (L.Name.empty())
                                {
                                        ServerInstance->Log(DEFAULT,"Invalid configuration, link tag without a name!");
                                }
@@ -467,18 +640,18 @@ void SpanningTreeUtilities::DoFailOver(Link* x)
        {
                if (x->FailOver == x->Name)
                {
-                       ServerInstance->SNO->WriteToSnoMask('l',"FAILOVER: Some muppet configured the failover for server \002%s\002 to point at itself. Not following it!", x->Name.c_str());
+                       Creator->RemoteMessage(NULL,"FAILOVER: Some muppet configured the failover for server \002%s\002 to point at itself. Not following it!", x->Name.c_str());
                        return;
                }
                Link* TryThisOne = this->FindLink(x->FailOver.c_str());
                if (TryThisOne)
                {
-                       ServerInstance->SNO->WriteToSnoMask('l',"FAILOVER: Trying failover link for \002%s\002: \002%s\002...", x->Name.c_str(), TryThisOne->Name.c_str());
+                       Creator->RemoteMessage(NULL,"FAILOVER: Trying failover link for \002%s\002: \002%s\002...", x->Name.c_str(), TryThisOne->Name.c_str());
                        Creator->ConnectServer(TryThisOne);
                }
                else
                {
-                       ServerInstance->SNO->WriteToSnoMask('l',"FAILOVER: Invalid failover server specified for server \002%s\002, will not follow!", x->Name.c_str());
+                       Creator->RemoteMessage(NULL,"FAILOVER: Invalid failover server specified for server \002%s\002, will not follow!", x->Name.c_str());
                }
        }
 }