]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_spanningtree/utils.cpp
OOPS! We try again, since I'm smoking craq. LF is 0x0a NOT CR.
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / utils.cpp
index 4d0256fa2662428b7f2fbf95b7a5577499f131ac..9675a6ac8588f35dbf2ca71a1d1c41de0a3a1713 100644 (file)
@@ -1 +1,649 @@
-/*       +------------------------------------+\r *       | Inspire Internet Relay Chat Daemon |\r *       +------------------------------------+\r *\r *  InspIRCd: (C) 2002-2007 InspIRCd Development Team\r * See: http://www.inspircd.org/wiki/index.php/Credits\r *\r * This program is free but copyrighted software; see\r *            the file COPYING for details.\r *\r * ---------------------------------------------------\r */\r\r#include "inspircd.h"\r#include "configreader.h"\r#include "users.h"\r#include "channels.h"\r#include "modules.h"\r#include "commands/cmd_whois.h"\r#include "commands/cmd_stats.h"\r#include "socket.h"\r#include "wildcard.h"\r#include "xline.h"\r#include "transport.h"\r#include "socketengine.h"\r\r#include "m_spanningtree/main.h"\r#include "m_spanningtree/utils.h"\r#include "m_spanningtree/treeserver.h"\r#include "m_spanningtree/link.h"\r#include "m_spanningtree/treesocket.h"\r#include "m_spanningtree/resolvers.h"\r\r/* $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 */\r\r/** Yay for fast searches!\r * This is hundreds of times faster than recursion\r * or even scanning a linked list, especially when\r * there are more than a few servers to deal with.\r * (read as: lots).\r */\rTreeServer* SpanningTreeUtilities::FindServer(const std::string &ServerName)\r{\r     server_hash::iterator iter = serverlist.find(ServerName.c_str());\r      if (iter != serverlist.end())\r  {\r              return iter->second;\r   }\r      else\r   {\r              return NULL;\r   }\r}\r\rTreeServer* SpanningTreeUtilities::FindRemoteBurstServer(TreeServer* Server)\r{\r    server_hash::iterator iter = RemoteServersBursting.find(Server->GetName().c_str());\r    if (iter != RemoteServersBursting.end())\r               return iter->second;\r   else\r           return NULL;\r}\r\rTreeSocket* SpanningTreeUtilities::FindBurstingServer(const std::string &ServerName)\r{\r std::map<irc::string,TreeSocket*>::iterator iter;\r      iter = burstingserverlist.find(ServerName.c_str());\r    if (iter != burstingserverlist.end())\r  {\r              return iter->second;\r   }\r      else\r   {\r              return NULL;\r   }\r}\r\rvoid SpanningTreeUtilities::SetRemoteBursting(TreeServer* Server, bool bursting)\r{\r        server_hash::iterator iter = RemoteServersBursting.find(Server->GetName().c_str());\r    if (bursting)\r  {\r              if (iter == RemoteServersBursting.end())\r                       RemoteServersBursting.insert(make_pair(Server->GetName(), Server));\r            else return;\r   }\r      else\r   {\r              if (iter != RemoteServersBursting.end())\r                       RemoteServersBursting.erase(iter);\r             else return;\r   }\r      ServerInstance->Log(DEBUG,"Server %s is %sbursting nicknames", Server->GetName().c_str(), bursting ? "" : "no longer ");\r}\r\rvoid SpanningTreeUtilities::AddBurstingServer(const std::string &ServerName, TreeSocket* s)\r{\r      std::map<irc::string,TreeSocket*>::iterator iter = burstingserverlist.find(ServerName.c_str());\r        if (iter == burstingserverlist.end())\r          burstingserverlist[ServerName.c_str()] = s;\r}\r\rvoid SpanningTreeUtilities::DelBurstingServer(TreeSocket* s)\r{\r   for (std::map<irc::string,TreeSocket*>::iterator iter = burstingserverlist.begin(); iter != burstingserverlist.end(); iter++)\r  {\r              if (iter->second == s)\r                 {\r                      burstingserverlist.erase(iter);\r                        return;\r                }\r      }\r}\r\r/** Returns the locally connected server we must route a\r * message through to reach server 'ServerName'. This\r * only applies to one-to-one and not one-to-many routing.\r * See the comments for the constructor of TreeServer\r * for more details.\r */\rTreeServer* SpanningTreeUtilities::BestRouteTo(const std::string &ServerName)\r{\r if (ServerName.c_str() == TreeRoot->GetName())\r         return NULL;\r   TreeServer* Found = FindServer(ServerName);\r    if (Found)\r     {\r              return Found->GetRoute();\r      }\r      else\r   {\r              return NULL;\r   }\r}\r\r/** Find the first server matching a given glob mask.\r * Theres no find-using-glob method of hash_map [awwww :-(]\r * so instead, we iterate over the list using an iterator\r * and match each one until we get a hit. Yes its slow,\r * deal with it.\r */\rTreeServer* SpanningTreeUtilities::FindServerMask(const std::string &ServerName)\r{\r       for (server_hash::iterator i = serverlist.begin(); i != serverlist.end(); i++)\r {\r              if (match(i->first.c_str(),ServerName.c_str()))\r                        return i->second;\r      }\r      return NULL;\r}\r\r/* A convenient wrapper that returns true if a server exists */\rbool SpanningTreeUtilities::IsServer(const std::string &ServerName)\r{\r  return (FindServer(ServerName) != NULL);\r}\r\rSpanningTreeUtilities::SpanningTreeUtilities(InspIRCd* Instance, ModuleSpanningTree* C) : ServerInstance(Instance), Creator(C)\r{\r   Bindings.clear();\r\r     lines_to_apply = 0;\r\r   this->TreeRoot = new TreeServer(this, ServerInstance, ServerInstance->Config->ServerName, ServerInstance->Config->ServerDesc);\r\r        modulelist* ml = ServerInstance->FindInterface("InspSocketHook");\r\r     /* Did we find any modules? */\r if (ml)\r        {\r              /* Yes, enumerate them all to find out the hook name */\r                for (modulelist::iterator m = ml->begin(); m != ml->end(); m++)\r                {\r                      /* Make a request to it for its name, its implementing\r                  * InspSocketHook so we know its safe to do this\r                        */\r                    std::string name = InspSocketNameRequest((Module*)Creator, *m).Send();\r                 /* Build a map of them */\r                      hooks[name.c_str()] = *m;\r                      hooknames.push_back(name);\r             }\r      }\r\r     this->ReadConfiguration(true);\r}\r\rSpanningTreeUtilities::~SpanningTreeUtilities()\r{\r    for (unsigned int i = 0; i < Bindings.size(); i++)\r     {\r              ServerInstance->SE->DelFd(Bindings[i]);\r                Bindings[i]->Close();\r          DELETE(Bindings[i]);\r   }\r      while (TreeRoot->ChildCount())\r {\r              TreeServer* child_server = TreeRoot->GetChild(0);\r              if (child_server)\r              {\r                      TreeSocket* sock = child_server->GetSocket();\r                  ServerInstance->SE->DelFd(sock);\r                       sock->Close();\r                 DELETE(sock);\r          }\r      }\r      delete TreeRoot;\r}\r\rvoid SpanningTreeUtilities::AddThisServer(TreeServer* server, TreeServerList &list)\r{\r      if (list.find(server) == list.end())\r           list[server] = server;\r}\r\r/* returns a list of DIRECT servernames for a specific channel */\rvoid SpanningTreeUtilities::GetListOfServersForChannel(chanrec* c, TreeServerList &list, char status, const CUList &exempt_list)\r{\r CUList *ulist;\r switch (status)\r        {\r              case '@':\r                      ulist = c->GetOppedUsers();\r            break;\r         case '%':\r                      ulist = c->GetHalfoppedUsers();\r                break;\r         case '+':\r                      ulist = c->GetVoicedUsers();\r           break;\r         default:\r                       ulist = c->GetUsers();\r         break;\r }\r      for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)\r      {\r              if ((i->first->GetFd() < 0) && (exempt_list.find(i->first) == exempt_list.end()))\r              {\r                      TreeServer* best = this->BestRouteTo(i->first->server);\r                        if (best)\r                              AddThisServer(best,list);\r              }\r      }\r      return;\r}\r\rbool SpanningTreeUtilities::DoOneToAllButSenderRaw(const std::string &data, const std::string &omit, const std::string &prefix, const irc::string &command, std::deque<std::string> &params)\r{\r      char pfx = 0;\r  TreeServer* omitroute = this->BestRouteTo(omit);\r       if ((command == "NOTICE") || (command == "PRIVMSG"))\r   {\r              if (params.size() >= 2)\r                {\r                      /* Prefixes */\r                 if ((*(params[0].c_str()) == '@') || (*(params[0].c_str()) == '%') || (*(params[0].c_str()) == '+'))\r                   {\r                              pfx = params[0][0];\r                            params[0] = params[0].substr(1, params[0].length()-1);\r                 }\r                      if ((*(params[0].c_str()) != '#') && (*(params[0].c_str()) != '$'))\r                    {\r                              // special routing for private messages/notices\r                                userrec* d = ServerInstance->FindNick(params[0]);\r                              if (d)\r                         {\r                                      std::deque<std::string> par;\r                                   par.push_back(params[0]);\r                                      par.push_back(":"+params[1]);\r                                  this->DoOneToOne(prefix,command.c_str(),par,d->server);\r                                        return true;\r                           }\r                      }\r                      else if (*(params[0].c_str()) == '$')\r                  {\r                              std::deque<std::string> par;\r                           par.push_back(params[0]);\r                              par.push_back(":"+params[1]);\r                          this->DoOneToAllButSender(prefix,command.c_str(),par,omitroute->GetName());\r                            return true;\r                   }\r                      else\r                   {\r                              chanrec* c = ServerInstance->FindChan(params[0]);\r                              userrec* u = ServerInstance->FindNick(prefix);\r                         if (c && u)\r                            {\r                                      CUList elist;\r                                  TreeServerList list;\r                                   FOREACH_MOD(I_OnBuildExemptList, OnBuildExemptList((command == "PRIVMSG" ? MSG_PRIVMSG : MSG_NOTICE), c, u, pfx, elist));\r                                      GetListOfServersForChannel(c,list,pfx,elist);\r\r                                 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)\r                                  {\r                                              TreeSocket* Sock = i->second->GetSocket();\r                                             if ((Sock) && (i->second->GetName() != omit) && (omitroute != i->second))\r                                              {\r                                                      Sock->WriteLine(data);\r                                         }\r                                      }\r                                      return true;\r                           }\r                      }\r              }\r      }\r      unsigned int items =this->TreeRoot->ChildCount();\r      for (unsigned int x = 0; x < items; x++)\r       {\r              TreeServer* Route = this->TreeRoot->GetChild(x);\r               if ((Route) && (Route->GetSocket()) && (Route->GetName() != omit) && (omitroute != Route))\r             {\r                      TreeSocket* Sock = Route->GetSocket();\r                 if (Sock)\r                              Sock->WriteLine(data);\r         }\r      }\r      return true;\r}\r\rbool SpanningTreeUtilities::DoOneToAllButSender(const std::string &prefix, const std::string &command, std::deque<std::string> &params, std::string omit)\r{\r    TreeServer* omitroute = this->BestRouteTo(omit);\r       std::string FullLine = ":" + prefix + " " + command;\r   unsigned int words = params.size();\r    for (unsigned int x = 0; x < words; x++)\r       {\r              FullLine = FullLine + " " + params[x];\r }\r      unsigned int items = this->TreeRoot->ChildCount();\r     for (unsigned int x = 0; x < items; x++)\r       {\r              TreeServer* Route = this->TreeRoot->GetChild(x);\r               // Send the line IF:\r           // The route has a socket (its a direct connection)\r            // The route isnt the one to be omitted\r                // The route isnt the path to the one to be omitted\r            if ((Route) && (Route->GetSocket()) && (Route->GetName() != omit) && (omitroute != Route))\r             {\r                      TreeSocket* Sock = Route->GetSocket();\r                 if (Sock)\r                              Sock->WriteLine(FullLine);\r             }\r      }\r      return true;\r}\r\rbool SpanningTreeUtilities::DoOneToMany(const std::string &prefix, const std::string &command, std::deque<std::string> &params)\r{\r      std::string FullLine = ":" + prefix + " " + command;\r   unsigned int words = params.size();\r    for (unsigned int x = 0; x < words; x++)\r       {\r              FullLine = FullLine + " " + params[x];\r }\r      unsigned int items = this->TreeRoot->ChildCount();\r     for (unsigned int x = 0; x < items; x++)\r       {\r              TreeServer* Route = this->TreeRoot->GetChild(x);\r               if (Route && Route->GetSocket())\r               {\r                      TreeSocket* Sock = Route->GetSocket();\r                 if (Sock)\r                              Sock->WriteLine(FullLine);\r             }\r      }\r      return true;\r}\r\rbool SpanningTreeUtilities::DoOneToMany(const char* prefix, const char* command, std::deque<std::string> &params)\r{\r    std::string spfx = prefix;\r     std::string scmd = command;\r    return this->DoOneToMany(spfx, scmd, params);\r}\r\rbool SpanningTreeUtilities::DoOneToAllButSender(const char* prefix, const char* command, std::deque<std::string> &params, std::string omit)\r{\r std::string spfx = prefix;\r     std::string scmd = command;\r    return this->DoOneToAllButSender(spfx, scmd, params, omit);\r}\r\rbool SpanningTreeUtilities::DoOneToOne(const std::string &prefix, const std::string &command, std::deque<std::string> &params, std::string target)\r{\r    TreeServer* Route = this->BestRouteTo(target);\r if (Route)\r     {\r              std::string FullLine = ":" + prefix + " " + command;\r           unsigned int words = params.size();\r            for (unsigned int x = 0; x < words; x++)\r               {\r                      FullLine = FullLine + " " + params[x];\r         }\r              if (Route && Route->GetSocket())\r               {\r                      TreeSocket* Sock = Route->GetSocket();\r                 if (Sock)\r                              Sock->WriteLine(FullLine);\r             }\r              return true;\r   }\r      else\r   {\r              return false;\r  }\r}\r\rvoid SpanningTreeUtilities::RefreshIPCache()\r{\r    ValidIPs.clear();\r      for (std::vector<Link>::iterator L = LinkBlocks.begin(); L != LinkBlocks.end(); L++)\r   {\r              if ((!L->IPAddr.empty()) && (!L->RecvPass.empty()) && (!L->SendPass.empty()) && (!L->Name.empty()) && (L->Port))\r               {\r                      ValidIPs.push_back(L->IPAddr);\r\r                        if (L->AllowMask.length())\r                             ValidIPs.push_back(L->AllowMask);\r\r                     /* Needs resolving */\r                  bool ipvalid = true;\r                   QueryType start_type = DNS_QUERY_A;\r#ifdef IPV6\r                        start_type = DNS_QUERY_AAAA;\r                   if (strchr(L->IPAddr.c_str(),':'))\r                     {\r                              in6_addr n;\r                            if (inet_pton(AF_INET6, L->IPAddr.c_str(), &n) < 1)\r                                    ipvalid = false;\r                       }\r                      else\r#endif\r                    {\r                              in_addr n;\r                             if (inet_aton(L->IPAddr.c_str(),&n) < 1)\r                                       ipvalid = false;\r                       }\r                      if (!ipvalid)\r                  {\r                              try\r                            {\r                                      bool cached;\r                                   SecurityIPResolver* sr = new SecurityIPResolver((Module*)this->Creator, this, ServerInstance, L->IPAddr, *L, cached, start_type);\r                                      ServerInstance->AddResolver(sr, cached);\r                               }\r                              catch (...)\r                            {\r                              }\r                      }\r              }\r      }\r}\r\rvoid SpanningTreeUtilities::ReadConfiguration(bool rebind)\r{\r      ConfigReader* Conf = new ConfigReader(ServerInstance);\r if (rebind)\r    {\r              for (int j = 0; j < Conf->Enumerate("bind"); j++)\r              {\r                      std::string Type = Conf->ReadValue("bind","type",j);\r                   std::string IP = Conf->ReadValue("bind","address",j);\r                  std::string Port = Conf->ReadValue("bind","port",j);\r                   std::string transport = Conf->ReadValue("bind","transport",j);\r                 if (Type == "servers")\r                 {\r                              irc::portparser portrange(Port, false);\r                                int portno = -1;\r                               while ((portno = portrange.GetToken()))\r                                {\r                                      if (IP == "*")\r                                         IP.clear();\r\r                                   if ((!transport.empty()) && (hooks.find(transport.c_str()) ==  hooks.end()))\r                                   {\r                                              ServerInstance->Log(DEFAULT,"m_spanningtree: WARNING: Can't find transport type '%s' for port %s:%s - maybe you forgot to load it BEFORE m_spanningtree in your config file? - Skipping this port binding", transport.c_str(), IP.c_str(), Port.c_str());\r                                              break;\r                                 }\r\r                                     TreeSocket* listener = new TreeSocket(this, ServerInstance, IP.c_str(), portno, true, 10, transport.empty() ? NULL : hooks[transport.c_str()]);\r                                        if (listener->GetState() == I_LISTENING)\r                                       {\r                                              ServerInstance->Log(DEFAULT,"m_spanningtree: Binding server port %s:%d successful!", IP.c_str(), portno);\r                                              Bindings.push_back(listener);\r                                  }\r                                      else\r                                   {\r                                              ServerInstance->Log(DEFAULT,"m_spanningtree: Warning: Failed to bind server port: %s:%d: %s",IP.c_str(), portno, strerror(errno));\r                                             listener->Close();\r                                             DELETE(listener);\r                                      }\r                              }\r                      }\r              }\r      }\r      FlatLinks = Conf->ReadFlag("options","flatlinks",0);\r   HideULines = Conf->ReadFlag("options","hideulines",0);\r AnnounceTSChange = Conf->ReadFlag("options","announcets",0);\r   EnableTimeSync = Conf->ReadFlag("timesync","enable",0);\r        MasterTime = Conf->ReadFlag("timesync", "master", 0);\r  ChallengeResponse = !Conf->ReadFlag("options", "disablehmac", 0);\r      quiet_bursts = Conf->ReadFlag("options", "quietbursts", 0);\r    PingWarnTime = Conf->ReadInteger("options", "pingwarning", 0, true);\r\r  if (PingWarnTime < 0 || PingWarnTime > 59)\r             PingWarnTime = 0;\r\r     LinkBlocks.clear();\r    ValidIPs.clear();\r      for (int j = 0; j < Conf->Enumerate("link"); j++)\r      {\r              Link L;\r                std::string Allow = Conf->ReadValue("link", "allowmask", j);\r           L.Name = (Conf->ReadValue("link", "name", j)).c_str();\r         L.AllowMask = Allow;\r           L.IPAddr = Conf->ReadValue("link", "ipaddr", j);\r               L.FailOver = Conf->ReadValue("link", "failover", j).c_str();\r           L.Port = Conf->ReadInteger("link", "port", j, true);\r           L.SendPass = Conf->ReadValue("link", "sendpass", j);\r           L.RecvPass = Conf->ReadValue("link", "recvpass", j);\r           L.AutoConnect = Conf->ReadInteger("link", "autoconnect", j, true);\r             L.HiddenFromStats = Conf->ReadFlag("link", "statshidden", j);\r          L.Timeout = Conf->ReadInteger("link", "timeout", j, true);\r             L.Hook = Conf->ReadValue("link", "transport", j);\r              L.Bind = Conf->ReadValue("link", "bind", j);\r           L.Hidden = Conf->ReadFlag("link", "hidden", j);\r\r               if ((!L.Hook.empty()) && (hooks.find(L.Hook.c_str()) ==  hooks.end()))\r         {\r                      ServerInstance->Log(DEFAULT,"m_spanningtree: WARNING: Can't find transport type '%s' for link '%s' - maybe you forgot to load it BEFORE m_spanningtree in your config file? Skipping <link> tag completely.",\r                  L.Hook.c_str(), L.Name.c_str());\r                       continue;\r\r             }\r\r             L.NextConnectTime = time(NULL) + L.AutoConnect;\r                /* Bugfix by brain, do not allow people to enter bad configurations */\r         if (L.Name != ServerInstance->Config->ServerName)\r              {\r                      if ((!L.IPAddr.empty()) && (!L.RecvPass.empty()) && (!L.SendPass.empty()) && (!L.Name.empty()) && (L.Port))\r                    {\r                              ValidIPs.push_back(L.IPAddr);\r\r                         if (Allow.length())\r                                    ValidIPs.push_back(Allow);\r\r                            /* Needs resolving */\r                          bool ipvalid = true;\r                           QueryType start_type = DNS_QUERY_A;\r#ifdef IPV6\r                                start_type = DNS_QUERY_AAAA;\r                           if (strchr(L.IPAddr.c_str(),':'))\r                              {\r                                      in6_addr n;\r                                    if (inet_pton(AF_INET6, L.IPAddr.c_str(), &n) < 1)\r                                             ipvalid = false;\r                               }\r                              else\r                           {\r                                      in_addr n;\r                                     if (inet_aton(L.IPAddr.c_str(),&n) < 1)\r                                                ipvalid = false;\r                               }\r#else\r                                in_addr n;\r                             if (inet_aton(L.IPAddr.c_str(),&n) < 1)\r                                        ipvalid = false;\r#endif\r\r                               if (!ipvalid)\r                          {\r                                      try\r                                    {\r                                              bool cached;\r                                           SecurityIPResolver* sr = new SecurityIPResolver((Module*)this->Creator, this, ServerInstance, L.IPAddr, L, cached, start_type);\r                                                ServerInstance->AddResolver(sr, cached);\r                                       }\r                                      catch (...)\r                                    {\r                                      }\r                              }\r\r                             LinkBlocks.push_back(L);\r                       }\r                      else\r                   {\r                              if (L.IPAddr.empty())\r                          {\r                                      ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', IP address not defined!",L.Name.c_str());\r                          }\r                              else if (L.RecvPass.empty())\r                           {\r                                      ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', recvpass not defined!",L.Name.c_str());\r                            }\r                              else if (L.SendPass.empty())\r                           {\r                                      ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', sendpass not defined!",L.Name.c_str());\r                            }\r                              else if (L.Name.empty())\r                               {\r                                      ServerInstance->Log(DEFAULT,"Invalid configuration, link tag without a name!");\r                                }\r                              else if (!L.Port)\r                              {\r                                      ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', no port specified!",L.Name.c_str());\r                               }\r                      }\r              }\r              else\r           {\r                      ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', link tag has the same server name as the local server!",L.Name.c_str());\r           }\r      }\r      DELETE(Conf);\r}\r\rvoid SpanningTreeUtilities::DoFailOver(Link* x)\r{\r     if (x->FailOver.length())\r      {\r              if (x->FailOver == x->Name)\r            {\r                      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());\r                 return;\r                }\r              Link* TryThisOne = this->FindLink(x->FailOver.c_str());\r                if (TryThisOne)\r                {\r                      ServerInstance->SNO->WriteToSnoMask('l',"FAILOVER: Trying failover link for \002%s\002: \002%s\002...", x->Name.c_str(), TryThisOne->Name.c_str());\r                    Creator->ConnectServer(TryThisOne);\r            }\r              else\r           {\r                      ServerInstance->SNO->WriteToSnoMask('l',"FAILOVER: Invalid failover server specified for server \002%s\002, will not follow!", x->Name.c_str());\r               }\r      }\r}\r\rLink* SpanningTreeUtilities::FindLink(const std::string& name)\r{\r  for (std::vector<Link>::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++)\r    {\r              if (ServerInstance->MatchText(x->Name.c_str(), name.c_str()))\r          {\r                      return &(*x);\r          }\r      }\r      return NULL;\r}\r\r
\ No newline at end of file
+/*       +------------------------------------+
+ *       | 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 "configreader.h"
+#include "users.h"
+#include "channels.h"
+#include "modules.h"
+#include "commands/cmd_whois.h"
+#include "commands/cmd_stats.h"
+#include "socket.h"
+#include "wildcard.h"
+#include "xline.h"
+#include "transport.h"
+#include "socketengine.h"
+
+#include "m_spanningtree/main.h"
+#include "m_spanningtree/utils.h"
+#include "m_spanningtree/treeserver.h"
+#include "m_spanningtree/link.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 */
+
+/** Yay for fast searches!
+ * This is hundreds of times faster than recursion
+ * or even scanning a linked list, especially when
+ * there are more than a few servers to deal with.
+ * (read as: lots).
+ */
+TreeServer* SpanningTreeUtilities::FindServer(const std::string &ServerName)
+{
+       server_hash::iterator iter = serverlist.find(ServerName.c_str());
+       if (iter != serverlist.end())
+       {
+               return iter->second;
+       }
+       else
+       {
+               return NULL;
+       }
+}
+
+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.
+ * See the comments for the constructor of TreeServer
+ * for more details.
+ */
+TreeServer* SpanningTreeUtilities::BestRouteTo(const std::string &ServerName)
+{
+       if (ServerName.c_str() == TreeRoot->GetName())
+               return NULL;
+       TreeServer* Found = FindServer(ServerName);
+       if (Found)
+       {
+               return Found->GetRoute();
+       }
+       else
+       {
+               return NULL;
+       }
+}
+
+/** Find the first server matching a given glob mask.
+ * Theres no find-using-glob method of hash_map [awwww :-(]
+ * so instead, we iterate over the list using an iterator
+ * and match each one until we get a hit. Yes its slow,
+ * deal with it.
+ */
+TreeServer* SpanningTreeUtilities::FindServerMask(const std::string &ServerName)
+{
+       for (server_hash::iterator i = serverlist.begin(); i != serverlist.end(); i++)
+       {
+               if (match(i->first.c_str(),ServerName.c_str()))
+                       return i->second;
+       }
+       return NULL;
+}
+
+/* A convenient wrapper that returns true if a server exists */
+bool SpanningTreeUtilities::IsServer(const std::string &ServerName)
+{
+       return (FindServer(ServerName) != NULL);
+}
+
+SpanningTreeUtilities::SpanningTreeUtilities(InspIRCd* Instance, ModuleSpanningTree* C) : ServerInstance(Instance), Creator(C)
+{
+       Bindings.clear();
+
+       lines_to_apply = 0;
+
+       this->TreeRoot = new TreeServer(this, ServerInstance, ServerInstance->Config->ServerName, ServerInstance->Config->ServerDesc);
+
+       modulelist* ml = ServerInstance->FindInterface("InspSocketHook");
+
+       /* Did we find any modules? */
+       if (ml)
+       {
+               /* Yes, enumerate them all to find out the hook name */
+               for (modulelist::iterator m = ml->begin(); m != ml->end(); m++)
+               {
+                       /* Make a request to it for its name, its implementing
+                        * InspSocketHook so we know its safe to do this
+                        */
+                       std::string name = InspSocketNameRequest((Module*)Creator, *m).Send();
+                       /* Build a map of them */
+                       hooks[name.c_str()] = *m;
+                       hooknames.push_back(name);
+               }
+       }
+
+       this->ReadConfiguration(true);
+}
+
+SpanningTreeUtilities::~SpanningTreeUtilities()
+{
+       for (unsigned int i = 0; i < Bindings.size(); i++)
+       {
+               ServerInstance->SE->DelFd(Bindings[i]);
+               Bindings[i]->Close();
+               DELETE(Bindings[i]);
+       }
+       while (TreeRoot->ChildCount())
+       {
+               TreeServer* child_server = TreeRoot->GetChild(0);
+               if (child_server)
+               {
+                       TreeSocket* sock = child_server->GetSocket();
+                       ServerInstance->SE->DelFd(sock);
+                       sock->Close();
+                       DELETE(sock);
+               }
+       }
+       delete TreeRoot;
+}
+
+void SpanningTreeUtilities::AddThisServer(TreeServer* server, TreeServerList &list)
+{
+       if (list.find(server) == list.end())
+               list[server] = server;
+}
+
+/* returns a list of DIRECT servernames for a specific channel */
+void SpanningTreeUtilities::GetListOfServersForChannel(chanrec* c, TreeServerList &list, char status, const CUList &exempt_list)
+{
+       CUList *ulist;
+       switch (status)
+       {
+               case '@':
+                       ulist = c->GetOppedUsers();
+               break;
+               case '%':
+                       ulist = c->GetHalfoppedUsers();
+               break;
+               case '+':
+                       ulist = c->GetVoicedUsers();
+               break;
+               default:
+                       ulist = c->GetUsers();
+               break;
+       }
+       for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
+       {
+               if ((i->first->GetFd() < 0) && (exempt_list.find(i->first) == exempt_list.end()))
+               {
+                       TreeServer* best = this->BestRouteTo(i->first->server);
+                       if (best)
+                               AddThisServer(best,list);
+               }
+       }
+       return;
+}
+
+bool SpanningTreeUtilities::DoOneToAllButSenderRaw(const std::string &data, const std::string &omit, const std::string &prefix, const irc::string &command, std::deque<std::string> &params)
+{
+       char pfx = 0;
+       TreeServer* omitroute = this->BestRouteTo(omit);
+       if ((command == "NOTICE") || (command == "PRIVMSG"))
+       {
+               if (params.size() >= 2)
+               {
+                       /* Prefixes */
+                       if ((*(params[0].c_str()) == '@') || (*(params[0].c_str()) == '%') || (*(params[0].c_str()) == '+'))
+                       {
+                               pfx = params[0][0];
+                               params[0] = params[0].substr(1, params[0].length()-1);
+                       }
+                       if ((*(params[0].c_str()) != '#') && (*(params[0].c_str()) != '$'))
+                       {
+                               // special routing for private messages/notices
+                               userrec* d = ServerInstance->FindNick(params[0]);
+                               if (d)
+                               {
+                                       std::deque<std::string> par;
+                                       par.push_back(params[0]);
+                                       par.push_back(":"+params[1]);
+                                       this->DoOneToOne(prefix,command.c_str(),par,d->server);
+                                       return true;
+                               }
+                       }
+                       else if (*(params[0].c_str()) == '$')
+                       {
+                               std::deque<std::string> par;
+                               par.push_back(params[0]);
+                               par.push_back(":"+params[1]);
+                               this->DoOneToAllButSender(prefix,command.c_str(),par,omitroute->GetName());
+                               return true;
+                       }
+                       else
+                       {
+                               chanrec* c = ServerInstance->FindChan(params[0]);
+                               userrec* u = ServerInstance->FindNick(prefix);
+                               if (c && u)
+                               {
+                                       CUList elist;
+                                       TreeServerList list;
+                                       FOREACH_MOD(I_OnBuildExemptList, OnBuildExemptList((command == "PRIVMSG" ? MSG_PRIVMSG : MSG_NOTICE), c, u, pfx, elist));
+                                       GetListOfServersForChannel(c,list,pfx,elist);
+
+                                       for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
+                                       {
+                                               TreeSocket* Sock = i->second->GetSocket();
+                                               if ((Sock) && (i->second->GetName() != omit) && (omitroute != i->second))
+                                               {
+                                                       Sock->WriteLine(data);
+                                               }
+                                       }
+                                       return true;
+                               }
+                       }
+               }
+       }
+       unsigned int items =this->TreeRoot->ChildCount();
+       for (unsigned int x = 0; x < items; x++)
+       {
+               TreeServer* Route = this->TreeRoot->GetChild(x);
+               if ((Route) && (Route->GetSocket()) && (Route->GetName() != omit) && (omitroute != Route))
+               {
+                       TreeSocket* Sock = Route->GetSocket();
+                       if (Sock)
+                               Sock->WriteLine(data);
+               }
+       }
+       return true;
+}
+
+bool SpanningTreeUtilities::DoOneToAllButSender(const std::string &prefix, const std::string &command, std::deque<std::string> &params, std::string omit)
+{
+       TreeServer* omitroute = this->BestRouteTo(omit);
+       std::string FullLine = ":" + prefix + " " + command;
+       unsigned int words = params.size();
+       for (unsigned int x = 0; x < words; x++)
+       {
+               FullLine = FullLine + " " + params[x];
+       }
+       unsigned int items = this->TreeRoot->ChildCount();
+       for (unsigned int x = 0; x < items; x++)
+       {
+               TreeServer* Route = this->TreeRoot->GetChild(x);
+               // Send the line IF:
+               // The route has a socket (its a direct connection)
+               // The route isnt the one to be omitted
+               // The route isnt the path to the one to be omitted
+               if ((Route) && (Route->GetSocket()) && (Route->GetName() != omit) && (omitroute != Route))
+               {
+                       TreeSocket* Sock = Route->GetSocket();
+                       if (Sock)
+                               Sock->WriteLine(FullLine);
+               }
+       }
+       return true;
+}
+
+bool SpanningTreeUtilities::DoOneToMany(const std::string &prefix, const std::string &command, std::deque<std::string> &params)
+{
+       std::string FullLine = ":" + prefix + " " + command;
+       unsigned int words = params.size();
+       for (unsigned int x = 0; x < words; x++)
+       {
+               FullLine = FullLine + " " + params[x];
+       }
+       unsigned int items = this->TreeRoot->ChildCount();
+       for (unsigned int x = 0; x < items; x++)
+       {
+               TreeServer* Route = this->TreeRoot->GetChild(x);
+               if (Route && Route->GetSocket())
+               {
+                       TreeSocket* Sock = Route->GetSocket();
+                       if (Sock)
+                               Sock->WriteLine(FullLine);
+               }
+       }
+       return true;
+}
+
+bool SpanningTreeUtilities::DoOneToMany(const char* prefix, const char* command, std::deque<std::string> &params)
+{
+       std::string spfx = prefix;
+       std::string scmd = command;
+       return this->DoOneToMany(spfx, scmd, params);
+}
+
+bool SpanningTreeUtilities::DoOneToAllButSender(const char* prefix, const char* command, std::deque<std::string> &params, std::string omit)
+{
+       std::string spfx = prefix;
+       std::string scmd = command;
+       return this->DoOneToAllButSender(spfx, scmd, params, omit);
+}
+
+bool SpanningTreeUtilities::DoOneToOne(const std::string &prefix, const std::string &command, std::deque<std::string> &params, std::string target)
+{
+       TreeServer* Route = this->BestRouteTo(target);
+       if (Route)
+       {
+               std::string FullLine = ":" + prefix + " " + command;
+               unsigned int words = params.size();
+               for (unsigned int x = 0; x < words; x++)
+               {
+                       FullLine = FullLine + " " + params[x];
+               }
+               if (Route && Route->GetSocket())
+               {
+                       TreeSocket* Sock = Route->GetSocket();
+                       if (Sock)
+                               Sock->WriteLine(FullLine);
+               }
+               return true;
+       }
+       else
+       {
+               return false;
+       }
+}
+
+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++)
+               {
+                       std::string Type = Conf->ReadValue("bind","type",j);
+                       std::string IP = Conf->ReadValue("bind","address",j);
+                       std::string Port = Conf->ReadValue("bind","port",j);
+                       std::string transport = Conf->ReadValue("bind","transport",j);
+                       if (Type == "servers")
+                       {
+                               irc::portparser portrange(Port, false);
+                               int portno = -1;
+                               while ((portno = portrange.GetToken()))
+                               {
+                                       if (IP == "*")
+                                               IP.clear();
+
+                                       if ((!transport.empty()) && (hooks.find(transport.c_str()) ==  hooks.end()))
+                                       {
+                                               ServerInstance->Log(DEFAULT,"m_spanningtree: WARNING: Can't find transport type '%s' for port %s:%s - maybe you forgot to load it BEFORE m_spanningtree in your config file? - Skipping this port binding", transport.c_str(), IP.c_str(), Port.c_str());
+                                               break;
+                                       }
+
+                                       TreeSocket* listener = new TreeSocket(this, ServerInstance, IP.c_str(), portno, true, 10, transport.empty() ? NULL : hooks[transport.c_str()]);
+                                       if (listener->GetState() == I_LISTENING)
+                                       {
+                                               ServerInstance->Log(DEFAULT,"m_spanningtree: Binding server port %s:%d successful!", IP.c_str(), portno);
+                                               Bindings.push_back(listener);
+                                       }
+                                       else
+                                       {
+                                               ServerInstance->Log(DEFAULT,"m_spanningtree: Warning: Failed to bind server port: %s:%d: %s",IP.c_str(), portno, strerror(errno));
+                                               listener->Close();
+                                               DELETE(listener);
+                                       }
+                               }
+                       }
+               }
+       }
+       FlatLinks = Conf->ReadFlag("options","flatlinks",0);
+       HideULines = Conf->ReadFlag("options","hideulines",0);
+       AnnounceTSChange = Conf->ReadFlag("options","announcets",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);
+
+       if (PingWarnTime < 0 || PingWarnTime > 59)
+               PingWarnTime = 0;
+
+       LinkBlocks.clear();
+       ValidIPs.clear();
+       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.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()))
+               {
+                       ServerInstance->Log(DEFAULT,"m_spanningtree: WARNING: Can't find transport type '%s' for link '%s' - maybe you forgot to load it BEFORE m_spanningtree in your config file? Skipping <link> tag completely.",
+                       L.Hook.c_str(), L.Name.c_str());
+                       continue;
+
+               }
+
+               L.NextConnectTime = time(NULL) + L.AutoConnect;
+               /* Bugfix by brain, do not allow people to enter bad configurations */
+               if (L.Name != ServerInstance->Config->ServerName)
+               {
+                       if ((!L.IPAddr.empty()) && (!L.RecvPass.empty()) && (!L.SendPass.empty()) && (!L.Name.empty()) && (L.Port))
+                       {
+                               ValidIPs.push_back(L.IPAddr);
+
+                               if (Allow.length())
+                                       ValidIPs.push_back(Allow);
+
+                               /* 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
+                               {
+                                       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, start_type);
+                                               ServerInstance->AddResolver(sr, cached);
+                                       }
+                                       catch (...)
+                                       {
+                                       }
+                               }
+
+                               LinkBlocks.push_back(L);
+                       }
+                       else
+                       {
+                               if (L.IPAddr.empty())
+                               {
+                                       ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', IP address not defined!",L.Name.c_str());
+                               }
+                               else if (L.RecvPass.empty())
+                               {
+                                       ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', recvpass not defined!",L.Name.c_str());
+                               }
+                               else if (L.SendPass.empty())
+                               {
+                                       ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', sendpass not defined!",L.Name.c_str());
+                               }
+                               else if (L.Name.empty())
+                               {
+                                       ServerInstance->Log(DEFAULT,"Invalid configuration, link tag without a name!");
+                               }
+                               else if (!L.Port)
+                               {
+                                       ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', no port specified!",L.Name.c_str());
+                               }
+                       }
+               }
+               else
+               {
+                       ServerInstance->Log(DEFAULT,"Invalid configuration for server '%s', link tag has the same server name as the local server!",L.Name.c_str());
+               }
+       }
+       DELETE(Conf);
+}
+
+void SpanningTreeUtilities::DoFailOver(Link* x)
+{
+       if (x->FailOver.length())
+       {
+               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());
+                       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->ConnectServer(TryThisOne);
+               }
+               else
+               {
+                       ServerInstance->SNO->WriteToSnoMask('l',"FAILOVER: Invalid failover server specified for server \002%s\002, will not follow!", x->Name.c_str());
+               }
+       }
+}
+
+Link* SpanningTreeUtilities::FindLink(const std::string& name)
+{
+       for (std::vector<Link>::iterator x = LinkBlocks.begin(); x < LinkBlocks.end(); x++)
+       {
+               if (ServerInstance->MatchText(x->Name.c_str(), name.c_str()))
+               {
+                       return &(*x);
+               }
+       }
+       return NULL;
+}
+