-/* +------------------------------------+\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> ¶ms)\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> ¶ms, 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> ¶ms)\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> ¶ms)\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> ¶ms, 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> ¶ms, 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> ¶ms)
+{
+ 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> ¶ms, 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> ¶ms)
+{
+ 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> ¶ms)
+{
+ 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> ¶ms, 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> ¶ms, 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;
+}
+