-/* +------------------------------------+\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
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
+ *
+ * Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ * Copyright (C) 2007-2009 Craig Edwards <craigedwards@brainbox.cc>
+ * Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
+ * Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ *
+ * This file is part of InspIRCd. InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+
+#include "inspircd.h"
+#include "socket.h"
+#include "xline.h"
+#include "socketengine.h"
+
+#include "main.h"
+#include "utils.h"
+#include "treeserver.h"
+#include "link.h"
+#include "treesocket.h"
+#include "resolvers.h"
+
+/* Create server sockets off a listener. */
+ModResult ModuleSpanningTree::OnAcceptConnection(int newsock, ListenSocket* from, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server)
+{
+ if (from->bind_tag->getString("type") != "servers")
+ return MOD_RES_PASSTHRU;
+
+ std::string incomingip = client->addr();
+
+ for (std::vector<std::string>::iterator i = Utils->ValidIPs.begin(); i != Utils->ValidIPs.end(); i++)
+ {
+ if (*i == "*" || *i == incomingip || irc::sockets::cidr_mask(*i).match(*client))
+ {
+ /* we don't need to do anything with the pointer, creating it stores it in the necessary places */
+ new TreeSocket(Utils, newsock, from, client, server);
+ return MOD_RES_ALLOW;
+ }
+ }
+ ServerInstance->SNO->WriteToSnoMask('l', "Server connection from %s denied (no link blocks with that IP address)", incomingip.c_str());
+ return MOD_RES_DENY;
+}
+
+/** 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)
+{
+ if (ServerInstance->IsSID(ServerName))
+ return this->FindServerID(ServerName);
+
+ server_hash::iterator iter = serverlist.find(ServerName.c_str());
+ if (iter != serverlist.end())
+ {
+ return iter->second;
+ }
+ else
+ {
+ return NULL;
+ }
+}
+
+/** 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() || ServerName == ServerInstance->Config->GetSID())
+ return NULL;
+ TreeServer* Found = FindServer(ServerName);
+ if (Found)
+ {
+ return Found->GetRoute();
+ }
+ else
+ {
+ // Cheat a bit. This allows for (better) working versions of routing commands with nick based prefixes, without hassle
+ User *u = ServerInstance->FindNick(ServerName);
+ if (u)
+ {
+ Found = FindServer(u->server);
+ if (Found)
+ return Found->GetRoute();
+ }
+
+ 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 (InspIRCd::Match(i->first,ServerName))
+ return i->second;
+ }
+ return NULL;
+}
+
+TreeServer* SpanningTreeUtilities::FindServerID(const std::string &id)
+{
+ server_hash::iterator iter = sidlist.find(id);
+ if (iter != sidlist.end())
+ return iter->second;
+ else
+ return NULL;
+}
+
+/* A convenient wrapper that returns true if a server exists */
+bool SpanningTreeUtilities::IsServer(const std::string &ServerName)
+{
+ return (FindServer(ServerName) != NULL);
+}
+
+SpanningTreeUtilities::SpanningTreeUtilities(ModuleSpanningTree* C) : Creator(C)
+{
+ ServerInstance->Logs->Log("m_spanningtree",DEBUG,"***** Using SID for hash: %s *****", ServerInstance->Config->GetSID().c_str());
+
+ this->TreeRoot = new TreeServer(this, ServerInstance->Config->ServerName, ServerInstance->Config->ServerDesc, ServerInstance->Config->GetSID());
+ this->ReadConfiguration();
+}
+
+CullResult SpanningTreeUtilities::cull()
+{
+ while (TreeRoot->ChildCount())
+ {
+ TreeServer* child_server = TreeRoot->GetChild(0);
+ if (child_server)
+ {
+ TreeSocket* sock = child_server->GetSocket();
+ sock->Close();
+ }
+ }
+
+ for(std::map<TreeSocket*, std::pair<std::string, int> >::iterator i = timeoutlist.begin(); i != timeoutlist.end(); ++i)
+ {
+ TreeSocket* s = i->first;
+ s->Close();
+ }
+ TreeRoot->cull();
+
+ return classbase::cull();
+}
+
+SpanningTreeUtilities::~SpanningTreeUtilities()
+{
+ 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(Channel* c, TreeServerList &list, char status, const CUList &exempt_list)
+{
+ unsigned int minrank = 0;
+ if (status)
+ {
+ ModeHandler* mh = ServerInstance->Modes->FindPrefix(status);
+ if (mh)
+ minrank = mh->GetPrefixRank();
+ }
+
+ const UserMembList *ulist = c->GetUsers();
+
+ for (UserMembCIter i = ulist->begin(); i != ulist->end(); i++)
+ {
+ if (IS_LOCAL(i->first))
+ continue;
+
+ if (minrank && i->second->getRank() < minrank)
+ continue;
+
+ if (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, const parameterlist ¶ms)
+{
+ TreeServer* omitroute = this->BestRouteTo(omit);
+ 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, const parameterlist ¶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, const parameterlist ¶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, const parameterlist ¶ms)
+{
+ std::string spfx = prefix;
+ std::string scmd = command;
+ return this->DoOneToMany(spfx, scmd, params);
+}
+
+bool SpanningTreeUtilities::DoOneToAllButSender(const char* prefix, const char* command, const parameterlist ¶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, const parameterlist ¶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<reference<Link> >::iterator i = LinkBlocks.begin(); i != LinkBlocks.end(); ++i)
+ {
+ Link* L = *i;
+ if (!L->Port)
+ {
+ ServerInstance->Logs->Log("m_spanningtree",DEFAULT,"m_spanningtree: Ignoring a link block without a port.");
+ /* Invalid link block */
+ continue;
+ }
+
+ if (L->AllowMask.length())
+ ValidIPs.push_back(L->AllowMask);
+
+ irc::sockets::sockaddrs dummy;
+ bool ipvalid = irc::sockets::aptosa(L->IPAddr, L->Port, dummy);
+ if ((L->IPAddr == "*") || (ipvalid))
+ ValidIPs.push_back(L->IPAddr);
+ else
+ {
+ try
+ {
+ bool cached;
+ SecurityIPResolver* sr = new SecurityIPResolver(Creator, this, L->IPAddr, L, cached, DNS_QUERY_AAAA);
+ ServerInstance->AddResolver(sr, cached);
+ }
+ catch (...)
+ {
+ }
+ }
+ }
+}
+
+void SpanningTreeUtilities::ReadConfiguration()
+{
+ ConfigReader Conf;
+
+ FlatLinks = Conf.ReadFlag("security","flatlinks",0);
+ HideULines = Conf.ReadFlag("security","hideulines",0);
+ AnnounceTSChange = Conf.ReadFlag("options","announcets",0);
+ AllowOptCommon = Conf.ReadFlag("options", "allowmismatch", 0);
+ ChallengeResponse = !Conf.ReadFlag("security", "disablehmac", 0);
+ quiet_bursts = Conf.ReadFlag("performance", "quietbursts", 0);
+ PingWarnTime = Conf.ReadInteger("options", "pingwarning", 0, true);
+ PingFreq = Conf.ReadInteger("options", "serverpingfreq", 0, true);
+
+ if (PingFreq == 0)
+ PingFreq = 60;
+
+ if (PingWarnTime < 0 || PingWarnTime > PingFreq - 1)
+ PingWarnTime = 0;
+
+ AutoconnectBlocks.clear();
+ LinkBlocks.clear();
+ ConfigTagList tags = ServerInstance->Config->ConfTags("link");
+ for(ConfigIter i = tags.first; i != tags.second; ++i)
+ {
+ ConfigTag* tag = i->second;
+ reference<Link> L = new Link(tag);
+ L->Name = tag->getString("name").c_str();
+ L->AllowMask = tag->getString("allowmask");
+ L->IPAddr = tag->getString("ipaddr");
+ L->Port = tag->getInt("port");
+ L->SendPass = tag->getString("sendpass", tag->getString("password"));
+ L->RecvPass = tag->getString("recvpass", tag->getString("password"));
+ L->Fingerprint = tag->getString("fingerprint");
+ L->HiddenFromStats = tag->getBool("statshidden");
+ L->Timeout = tag->getInt("timeout", 30);
+ L->Hook = tag->getString("ssl");
+ L->Bind = tag->getString("bind");
+ L->Hidden = tag->getBool("hidden");
+
+ if (L->Name.empty())
+ throw ModuleException("Invalid configuration, found a link tag without a name!" + (!L->IPAddr.empty() ? " IP address: "+L->IPAddr : ""));
+
+ if (L->Name.find('.') == std::string::npos)
+ throw ModuleException("The link name '"+assign(L->Name)+"' is invalid as it must contain at least one '.' character");
+
+ if (L->Name.length() > 64)
+ throw ModuleException("The link name '"+assign(L->Name)+"' is invalid as it is longer than 64 characters");
+
+ if (L->RecvPass.empty())
+ throw ModuleException("Invalid configuration for server '"+assign(L->Name)+"', recvpass not defined");
+
+ if (L->SendPass.empty())
+ throw ModuleException("Invalid configuration for server '"+assign(L->Name)+"', sendpass not defined");
+
+ if ((L->SendPass.find(' ') != std::string::npos) || (L->RecvPass.find(' ') != std::string::npos))
+ throw ModuleException("Link block '" + assign(L->Name) + "' has a password set that contains a space character which is invalid");
+
+ if ((L->SendPass[0] == ':') || (L->RecvPass[0] == ':'))
+ throw ModuleException("Link block '" + assign(L->Name) + "' has a password set that begins with a colon (:) which is invalid");
+
+ if (L->IPAddr.empty())
+ {
+ L->IPAddr = "*";
+ ServerInstance->Logs->Log("m_spanningtree",DEFAULT,"Configuration warning: Link block '" + assign(L->Name) + "' has no IP defined! This will allow any IP to connect as this server, and MAY not be what you want.");
+ }
+
+ if (!L->Port)
+ ServerInstance->Logs->Log("m_spanningtree",DEFAULT,"Configuration warning: Link block '" + assign(L->Name) + "' has no port defined, you will not be able to /connect it.");
+
+ L->Fingerprint.erase(std::remove(L->Fingerprint.begin(), L->Fingerprint.end(), ':'), L->Fingerprint.end());
+ LinkBlocks.push_back(L);
+ }
+
+ tags = ServerInstance->Config->ConfTags("autoconnect");
+ for(ConfigIter i = tags.first; i != tags.second; ++i)
+ {
+ ConfigTag* tag = i->second;
+ reference<Autoconnect> A = new Autoconnect(tag);
+ A->Period = tag->getInt("period");
+ A->NextConnectTime = ServerInstance->Time() + A->Period;
+ A->position = -1;
+ irc::spacesepstream ss(tag->getString("server"));
+ std::string server;
+ while (ss.GetToken(server))
+ {
+ A->servers.push_back(server);
+ }
+
+ if (A->Period <= 0)
+ {
+ throw ModuleException("Invalid configuration for autoconnect, period not a positive integer!");
+ }
+
+ if (A->servers.empty())
+ {
+ throw ModuleException("Invalid configuration for autoconnect, server cannot be empty!");
+ }
+
+ AutoconnectBlocks.push_back(A);
+ }
+
+ RefreshIPCache();
+}
+
+Link* SpanningTreeUtilities::FindLink(const std::string& name)
+{
+ for (std::vector<reference<Link> >::iterator i = LinkBlocks.begin(); i != LinkBlocks.end(); ++i)
+ {
+ Link* x = *i;
+ if (InspIRCd::Match(x->Name.c_str(), name.c_str()))
+ {
+ return x;
+ }
+ }
+ return NULL;
+}