1 /* +------------------------------------+
2 * | Inspire Internet Relay Chat Daemon |
3 * +------------------------------------+
5 * InspIRCd: (C) 2002-2007 InspIRCd Development Team
6 * See: http://www.inspircd.org/wiki/index.php/Credits
8 * This program is free but copyrighted software; see
9 * the file COPYING for details.
11 * ---------------------------------------------------
14 /* $ModDesc: Provides a spanning tree server link protocol */
17 #include "configreader.h"
21 #include "commands/cmd_whois.h"
22 #include "commands/cmd_stats.h"
26 #include "transport.h"
28 #include "m_spanningtree/timesynctimer.h"
29 #include "m_spanningtree/resolvers.h"
30 #include "m_spanningtree/main.h"
31 #include "m_spanningtree/utils.h"
32 #include "m_spanningtree/treeserver.h"
33 #include "m_spanningtree/link.h"
34 #include "m_spanningtree/treesocket.h"
35 #include "m_spanningtree/rconnect.h"
37 /* $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 m_spanningtree/rconnect.h */
39 ModuleSpanningTree::ModuleSpanningTree(InspIRCd* Me)
40 : Module(Me), max_local(0), max_global(0)
42 ServerInstance->UseInterface("InspSocketHook");
43 Utils = new SpanningTreeUtilities(Me, this);
44 command_rconnect = new cmd_rconnect(ServerInstance, this, Utils);
45 ServerInstance->AddCommand(command_rconnect);
46 if (Utils->EnableTimeSync)
48 SyncTimer = new TimeSyncTimer(ServerInstance, this);
49 ServerInstance->Timers->AddTimer(SyncTimer);
54 RefreshTimer = new CacheRefreshTimer(ServerInstance, Utils);
55 ServerInstance->Timers->AddTimer(RefreshTimer);
58 void ModuleSpanningTree::ShowLinks(TreeServer* Current, userrec* user, int hops)
60 std::string Parent = Utils->TreeRoot->GetName();
61 if (Current->GetParent())
63 Parent = Current->GetParent()->GetName();
65 for (unsigned int q = 0; q < Current->ChildCount(); q++)
67 if ((Current->GetChild(q)->Hidden) || ((Utils->HideULines) && (ServerInstance->ULine(Current->GetChild(q)->GetName().c_str()))))
71 ShowLinks(Current->GetChild(q),user,hops+1);
76 ShowLinks(Current->GetChild(q),user,hops+1);
79 /* Don't display the line if its a uline, hide ulines is on, and the user isnt an oper */
80 if ((Utils->HideULines) && (ServerInstance->ULine(Current->GetName().c_str())) && (!IS_OPER(user)))
82 /* Or if the server is hidden and they're not an oper */
83 else if ((Current->Hidden) && (!IS_OPER(user)))
86 user->WriteServ("364 %s %s %s :%d %s", user->nick,Current->GetName().c_str(),
87 (Utils->FlatLinks && (!IS_OPER(user))) ? ServerInstance->Config->ServerName : Parent.c_str(),
88 (Utils->FlatLinks && (!IS_OPER(user))) ? 0 : hops,
89 Current->GetDesc().c_str());
92 int ModuleSpanningTree::CountLocalServs()
94 return Utils->TreeRoot->ChildCount();
97 int ModuleSpanningTree::CountServs()
99 return Utils->serverlist.size();
102 void ModuleSpanningTree::HandleLinks(const char** parameters, int pcnt, userrec* user)
104 ShowLinks(Utils->TreeRoot,user,0);
105 user->WriteServ("365 %s * :End of /LINKS list.",user->nick);
109 void ModuleSpanningTree::HandleLusers(const char** parameters, int pcnt, userrec* user)
111 unsigned int n_users = ServerInstance->UserCount();
113 /* Only update these when someone wants to see them, more efficient */
114 if ((unsigned int)ServerInstance->LocalUserCount() > max_local)
115 max_local = ServerInstance->LocalUserCount();
116 if (n_users > max_global)
117 max_global = n_users;
119 unsigned int ulined_count = 0;
120 unsigned int ulined_local_count = 0;
122 /* If ulined are hidden and we're not an oper, count the number of ulined servers hidden,
123 * locally and globally (locally means directly connected to us)
125 if ((Utils->HideULines) && (!*user->oper))
127 for (server_hash::iterator q = Utils->serverlist.begin(); q != Utils->serverlist.end(); q++)
129 if (ServerInstance->ULine(q->second->GetName().c_str()))
132 if (q->second->GetParent() == Utils->TreeRoot)
133 ulined_local_count++;
137 user->WriteServ("251 %s :There are %d users and %d invisible on %d servers",user->nick,n_users-ServerInstance->InvisibleUserCount(),ServerInstance->InvisibleUserCount(),ulined_count ? this->CountServs() - ulined_count : this->CountServs());
138 if (ServerInstance->OperCount())
139 user->WriteServ("252 %s %d :operator(s) online",user->nick,ServerInstance->OperCount());
140 if (ServerInstance->UnregisteredUserCount())
141 user->WriteServ("253 %s %d :unknown connections",user->nick,ServerInstance->UnregisteredUserCount());
142 if (ServerInstance->ChannelCount())
143 user->WriteServ("254 %s %d :channels formed",user->nick,ServerInstance->ChannelCount());
144 user->WriteServ("255 %s :I have %d clients and %d servers",user->nick,ServerInstance->LocalUserCount(),ulined_local_count ? this->CountLocalServs() - ulined_local_count : this->CountLocalServs());
145 user->WriteServ("265 %s :Current Local Users: %d Max: %d",user->nick,ServerInstance->LocalUserCount(),max_local);
146 user->WriteServ("266 %s :Current Global Users: %d Max: %d",user->nick,n_users,max_global);
150 std::string ModuleSpanningTree::TimeToStr(time_t secs)
152 time_t mins_up = secs / 60;
153 time_t hours_up = mins_up / 60;
154 time_t days_up = hours_up / 24;
156 mins_up = mins_up % 60;
157 hours_up = hours_up % 24;
158 return ((days_up ? (ConvToStr(days_up) + "d") : std::string(""))
159 + (hours_up ? (ConvToStr(hours_up) + "h") : std::string(""))
160 + (mins_up ? (ConvToStr(mins_up) + "m") : std::string(""))
161 + ConvToStr(secs) + "s");
164 const std::string ModuleSpanningTree::MapOperInfo(TreeServer* Current)
166 time_t secs_up = ServerInstance->Time() - Current->age;
167 return (" [Up: " + TimeToStr(secs_up) + " Lag: "+ConvToStr(Current->rtt)+"s]");
170 // WARNING: NOT THREAD SAFE - DONT GET ANY SMART IDEAS.
171 void ModuleSpanningTree::ShowMap(TreeServer* Current, userrec* user, int depth, char matrix[128][128], float &totusers, float &totservers)
175 for (int t = 0; t < depth; t++)
177 matrix[line][t] = ' ';
179 // For Aligning, we need to work out exactly how deep this thing is, and produce
180 // a 'Spacer' String to compensate.
182 memset(spacer,' ',40);
183 if ((40 - Current->GetName().length() - depth) > 1) {
184 spacer[40 - Current->GetName().length() - depth] = '\0';
192 /* Neat and tidy default values, as we're dealing with a matrix not a simple string */
193 memset(text, 0, 128);
195 if (ServerInstance->clientlist->size() == 0) {
196 // If there are no users, WHO THE HELL DID THE /MAP?!?!?!
201 percent = ((float)Current->GetUserCount() / (float)ServerInstance->clientlist->size()) * 100;
203 const std::string operdata = IS_OPER(user) ? MapOperInfo(Current) : "";
204 snprintf(text, 126, "%s %s%5d [%5.2f%%]%s", Current->GetName().c_str(), spacer, Current->GetUserCount(), percent, operdata.c_str());
205 totusers += Current->GetUserCount();
207 strlcpy(&matrix[line][depth],text,126);
209 for (unsigned int q = 0; q < Current->ChildCount(); q++)
211 if ((Current->GetChild(q)->Hidden) || ((Utils->HideULines) && (ServerInstance->ULine(Current->GetChild(q)->GetName().c_str()))))
215 ShowMap(Current->GetChild(q),user,(Utils->FlatLinks && (!*user->oper)) ? depth : depth+2,matrix,totusers,totservers);
220 ShowMap(Current->GetChild(q),user,(Utils->FlatLinks && (!*user->oper)) ? depth : depth+2,matrix,totusers,totservers);
226 int ModuleSpanningTree::HandleMotd(const char** parameters, int pcnt, userrec* user)
230 if (match(ServerInstance->Config->ServerName, parameters[0]))
233 /* Remote MOTD, the server is within the 1st parameter */
234 std::deque<std::string> params;
235 params.push_back(parameters[0]);
236 /* Send it out remotely, generate no reply yet */
237 TreeServer* s = Utils->FindServerMask(parameters[0]);
239 Utils->DoOneToOne(user->nick, "MOTD", params, s->GetName());
241 user->WriteServ( "402 %s %s :No such server", user->nick, parameters[0]);
247 int ModuleSpanningTree::HandleAdmin(const char** parameters, int pcnt, userrec* user)
251 if (match(ServerInstance->Config->ServerName, parameters[0]))
254 /* Remote ADMIN, the server is within the 1st parameter */
255 std::deque<std::string> params;
256 params.push_back(parameters[0]);
257 /* Send it out remotely, generate no reply yet */
258 TreeServer* s = Utils->FindServerMask(parameters[0]);
260 Utils->DoOneToOne(user->nick, "ADMIN", params, s->GetName());
262 user->WriteServ( "402 %s %s :No such server", user->nick, parameters[0]);
268 int ModuleSpanningTree::HandleModules(const char** parameters, int pcnt, userrec* user)
270 if (match(ServerInstance->Config->ServerName, parameters[0]))
273 std::deque<std::string> params;
274 params.push_back(parameters[0]);
275 TreeServer* s = Utils->FindServerMask(parameters[0]);
277 Utils->DoOneToOne(user->nick, "MODULES", params, s->GetName());
279 user->WriteServ( "402 %s %s :No such server", user->nick, parameters[0]);
283 int ModuleSpanningTree::HandleStats(const char** parameters, int pcnt, userrec* user)
287 if (match(ServerInstance->Config->ServerName, parameters[1]))
290 /* Remote STATS, the server is within the 2nd parameter */
291 std::deque<std::string> params;
292 params.push_back(parameters[0]);
293 params.push_back(parameters[1]);
294 /* Send it out remotely, generate no reply yet */
296 TreeServer* s = Utils->FindServerMask(parameters[1]);
299 params[1] = s->GetName();
300 Utils->DoOneToOne(user->nick, "STATS", params, s->GetName());
304 user->WriteServ( "402 %s %s :No such server", user->nick, parameters[1]);
311 // Ok, prepare to be confused.
312 // After much mulling over how to approach this, it struck me that
313 // the 'usual' way of doing a /MAP isnt the best way. Instead of
314 // keeping track of a ton of ascii characters, and line by line
315 // under recursion working out where to place them using multiplications
316 // and divisons, we instead render the map onto a backplane of characters
317 // (a character matrix), then draw the branches as a series of "L" shapes
318 // from the nodes. This is not only friendlier on CPU it uses less stack.
319 void ModuleSpanningTree::HandleMap(const char** parameters, int pcnt, userrec* user)
321 // This array represents a virtual screen which we will
322 // "scratch" draw to, as the console device of an irc
323 // client does not provide for a proper terminal.
325 float totservers = 0;
326 char matrix[128][128];
327 for (unsigned int t = 0; t < 128; t++)
332 // The only recursive bit is called here.
333 ShowMap(Utils->TreeRoot,user,0,matrix,totusers,totservers);
334 // Process each line one by one. The algorithm has a limit of
335 // 128 servers (which is far more than a spanning tree should have
336 // anyway, so we're ok). This limit can be raised simply by making
337 // the character matrix deeper, 128 rows taking 10k of memory.
338 for (int l = 1; l < line; l++)
340 // scan across the line looking for the start of the
341 // servername (the recursive part of the algorithm has placed
342 // the servers at indented positions depending on what they
344 int first_nonspace = 0;
345 while (matrix[l][first_nonspace] == ' ')
350 // Draw the `- (corner) section: this may be overwritten by
351 // another L shape passing along the same vertical pane, becoming
352 // a |- (branch) section instead.
353 matrix[l][first_nonspace] = '-';
354 matrix[l][first_nonspace-1] = '`';
356 // Draw upwards until we hit the parent server, causing possibly
357 // other corners (`-) to become branches (|-)
358 while ((matrix[l2][first_nonspace-1] == ' ') || (matrix[l2][first_nonspace-1] == '`'))
360 matrix[l2][first_nonspace-1] = '|';
364 // dump the whole lot to the user. This is the easy bit, honest.
365 for (int t = 0; t < line; t++)
367 user->WriteServ("006 %s :%s",user->nick,&matrix[t][0]);
369 float avg_users = totusers / totservers;
370 user->WriteServ("270 %s :%.0f server%s and %.0f user%s, average %.2f users per server",user->nick,totservers,(totservers > 1 ? "s" : ""),totusers,(totusers > 1 ? "s" : ""),avg_users);
371 user->WriteServ("007 %s :End of /MAP",user->nick);
375 int ModuleSpanningTree::HandleSquit(const char** parameters, int pcnt, userrec* user)
377 TreeServer* s = Utils->FindServerMask(parameters[0]);
380 if (s == Utils->TreeRoot)
382 user->WriteServ("NOTICE %s :*** SQUIT: Foolish mortal, you cannot make a server SQUIT itself! (%s matches local server name)",user->nick,parameters[0]);
385 TreeSocket* sock = s->GetSocket();
388 ServerInstance->SNO->WriteToSnoMask('l',"SQUIT: Server \002%s\002 removed from network by %s",parameters[0],user->nick);
389 sock->Squit(s,std::string("Server quit by ") + user->GetFullRealHost());
390 ServerInstance->SE->DelFd(sock);
397 std::deque<std::string> params;
398 params.push_back(parameters[0]);
399 params.push_back(std::string(":Server quit by ") + user->GetFullRealHost());
400 Utils->DoOneToOne(user->nick, "RSQUIT", params, parameters[0]);
405 user->WriteServ("NOTICE %s :*** SQUIT: The server \002%s\002 does not exist on the network.",user->nick,parameters[0]);
410 int ModuleSpanningTree::HandleTime(const char** parameters, int pcnt, userrec* user)
412 if ((IS_LOCAL(user)) && (pcnt))
414 TreeServer* found = Utils->FindServerMask(parameters[0]);
417 // we dont' override for local server
418 if (found == Utils->TreeRoot)
421 std::deque<std::string> params;
422 params.push_back(found->GetName());
423 params.push_back(user->nick);
424 Utils->DoOneToOne(ServerInstance->Config->ServerName,"TIME",params,found->GetName());
428 user->WriteServ("402 %s %s :No such server",user->nick,parameters[0]);
434 int ModuleSpanningTree::HandleRemoteWhois(const char** parameters, int pcnt, userrec* user)
436 if ((IS_LOCAL(user)) && (pcnt > 1))
438 userrec* remote = ServerInstance->FindNick(parameters[1]);
439 if ((remote) && (remote->GetFd() < 0))
441 std::deque<std::string> params;
442 params.push_back(parameters[1]);
443 Utils->DoOneToOne(user->nick,"IDLE",params,remote->server);
448 user->WriteServ("401 %s %s :No such nick/channel",user->nick, parameters[1]);
449 user->WriteServ("318 %s %s :End of /WHOIS list.",user->nick, parameters[1]);
456 void ModuleSpanningTree::DoPingChecks(time_t curtime)
458 for (unsigned int j = 0; j < Utils->TreeRoot->ChildCount(); j++)
460 TreeServer* serv = Utils->TreeRoot->GetChild(j);
461 TreeSocket* sock = serv->GetSocket();
464 if (curtime >= serv->NextPingTime())
466 if (serv->AnsweredLastPing())
468 sock->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" PING "+serv->GetName());
469 serv->SetNextPingTime(curtime + 60);
470 serv->LastPing = curtime;
471 serv->Warned = false;
475 /* they didnt answer, boot them */
476 sock->SendError("Ping timeout");
477 sock->Squit(serv,"Ping timeout");
478 ServerInstance->SE->DelFd(sock);
484 else if ((Utils->PingWarnTime) && (!serv->Warned) && (curtime >= serv->NextPingTime() - (60 - Utils->PingWarnTime)) && (!serv->AnsweredLastPing()))
486 /* The server hasnt responded, send a warning to opers */
487 ServerInstance->SNO->WriteToSnoMask('l',"Server \002%s\002 has not responded to PING for %d seconds, high latency.", serv->GetName().c_str(), Utils->PingWarnTime);
493 /* Cancel remote burst mode on any servers which still have it enabled due to latency/lack of data.
494 * This prevents lost REMOTECONNECT notices
496 for (server_hash::iterator i = Utils->serverlist.begin(); i != Utils->serverlist.end(); i++)
497 Utils->SetRemoteBursting(i->second, false);
500 void ModuleSpanningTree::ConnectServer(Link* x)
503 QueryType start_type = DNS_QUERY_A;
505 start_type = DNS_QUERY_AAAA;
506 if (strchr(x->IPAddr.c_str(),':'))
509 if (inet_pton(AF_INET6, x->IPAddr.c_str(), &n) < 1)
515 if (inet_aton(x->IPAddr.c_str(),&n) < 1)
520 if (inet_aton(x->IPAddr.c_str(),&n) < 1)
524 /* Do we already have an IP? If so, no need to resolve it. */
527 /* Gave a hook, but it wasnt one we know */
528 if ((!x->Hook.empty()) && (Utils->hooks.find(x->Hook.c_str()) == Utils->hooks.end()))
530 TreeSocket* newsocket = new TreeSocket(Utils, ServerInstance, x->IPAddr,x->Port,false,x->Timeout ? x->Timeout : 10,x->Name.c_str(), x->Bind, x->Hook.empty() ? NULL : Utils->hooks[x->Hook.c_str()]);
531 if (newsocket->GetFd() > -1)
533 /* Handled automatically on success */
537 ServerInstance->SNO->WriteToSnoMask('l',"CONNECT: Error connecting \002%s\002: %s.",x->Name.c_str(),strerror(errno));
539 Utils->DoFailOver(x);
547 ServernameResolver* snr = new ServernameResolver((Module*)this, Utils, ServerInstance,x->IPAddr, *x, cached, start_type);
548 ServerInstance->AddResolver(snr, cached);
550 catch (ModuleException& e)
552 ServerInstance->SNO->WriteToSnoMask('l',"CONNECT: Error connecting \002%s\002: %s.",x->Name.c_str(), e.GetReason());
553 Utils->DoFailOver(x);
558 void ModuleSpanningTree::AutoConnectServers(time_t curtime)
560 for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
562 if ((x->AutoConnect) && (curtime >= x->NextConnectTime))
564 x->NextConnectTime = curtime + x->AutoConnect;
565 TreeServer* CheckDupe = Utils->FindServer(x->Name.c_str());
566 if (x->FailOver.length())
568 TreeServer* CheckFailOver = Utils->FindServer(x->FailOver.c_str());
571 /* The failover for this server is currently a member of the network.
572 * The failover probably succeeded, where the main link did not.
573 * Don't try the main link until the failover is gone again.
580 // an autoconnected server is not connected. Check if its time to connect it
581 ServerInstance->SNO->WriteToSnoMask('l',"AUTOCONNECT: Auto-connecting server \002%s\002 (%lu seconds until next attempt)",x->Name.c_str(),x->AutoConnect);
582 this->ConnectServer(&(*x));
588 int ModuleSpanningTree::HandleVersion(const char** parameters, int pcnt, userrec* user)
590 // we've already checked if pcnt > 0, so this is safe
591 TreeServer* found = Utils->FindServerMask(parameters[0]);
594 std::string Version = found->GetVersion();
595 user->WriteServ("351 %s :%s",user->nick,Version.c_str());
596 if (found == Utils->TreeRoot)
598 ServerInstance->Config->Send005(user);
603 user->WriteServ("402 %s %s :No such server",user->nick,parameters[0]);
608 int ModuleSpanningTree::HandleConnect(const char** parameters, int pcnt, userrec* user)
610 for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
612 if (ServerInstance->MatchText(x->Name.c_str(),parameters[0]))
614 TreeServer* CheckDupe = Utils->FindServer(x->Name.c_str());
617 user->WriteServ("NOTICE %s :*** CONNECT: Connecting to server: \002%s\002 (%s:%d)",user->nick,x->Name.c_str(),(x->HiddenFromStats ? "<hidden>" : x->IPAddr.c_str()),x->Port);
618 ConnectServer(&(*x));
623 user->WriteServ("NOTICE %s :*** CONNECT: Server \002%s\002 already exists on the network and is connected via \002%s\002",user->nick,x->Name.c_str(),CheckDupe->GetParent()->GetName().c_str());
628 user->WriteServ("NOTICE %s :*** CONNECT: No server matching \002%s\002 could be found in the config file.",user->nick,parameters[0]);
632 void ModuleSpanningTree::BroadcastTimeSync()
634 if (Utils->MasterTime)
636 std::deque<std::string> params;
637 params.push_back(ConvToStr(ServerInstance->Time(false)));
638 params.push_back("FORCE");
639 Utils->DoOneToMany(Utils->TreeRoot->GetName(), "TIMESET", params);
643 int ModuleSpanningTree::OnStats(char statschar, userrec* user, string_list &results)
645 if ((statschar == 'c') || (statschar == 'n'))
647 for (unsigned int i = 0; i < Utils->LinkBlocks.size(); i++)
649 results.push_back(std::string(ServerInstance->Config->ServerName)+" 213 "+user->nick+" "+statschar+" *@"+(Utils->LinkBlocks[i].HiddenFromStats ? "<hidden>" : Utils->LinkBlocks[i].IPAddr)+" * "+Utils->LinkBlocks[i].Name.c_str()+" "+ConvToStr(Utils->LinkBlocks[i].Port)+" "+(Utils->LinkBlocks[i].Hook.empty() ? "plaintext" : Utils->LinkBlocks[i].Hook)+" "+(Utils->LinkBlocks[i].AutoConnect ? 'a' : '-')+'s');
650 if (statschar == 'c')
651 results.push_back(std::string(ServerInstance->Config->ServerName)+" 244 "+user->nick+" H * * "+Utils->LinkBlocks[i].Name.c_str());
653 results.push_back(std::string(ServerInstance->Config->ServerName)+" 219 "+user->nick+" "+statschar+" :End of /STATS report");
654 ServerInstance->SNO->WriteToSnoMask('t',"%s '%c' requested by %s (%s@%s)",(!strcmp(user->server,ServerInstance->Config->ServerName) ? "Stats" : "Remote stats"),statschar,user->nick,user->ident,user->host);
658 if (statschar == 'p')
660 /* show all server ports, after showing client ports. -- w00t */
662 for (unsigned int i = 0; i < Utils->Bindings.size(); i++)
664 std::string ip = Utils->Bindings[i]->IP;
668 std::string transport("plaintext");
669 if (Utils->Bindings[i]->GetHook())
670 transport = InspSocketNameRequest(this, Utils->Bindings[i]->GetHook()).Send();
672 results.push_back(ConvToStr(ServerInstance->Config->ServerName) + " 249 "+user->nick+" :" + ip + ":" + ConvToStr(Utils->Bindings[i]->port)+
673 " (server, " + transport + ")");
679 int ModuleSpanningTree::OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated, const std::string &original_line)
681 /* If the command doesnt appear to be valid, we dont want to mess with it. */
685 if (command == "CONNECT")
687 return this->HandleConnect(parameters,pcnt,user);
689 else if (command == "STATS")
691 return this->HandleStats(parameters,pcnt,user);
693 else if (command == "MOTD")
695 return this->HandleMotd(parameters,pcnt,user);
697 else if (command == "ADMIN")
699 return this->HandleAdmin(parameters,pcnt,user);
701 else if (command == "SQUIT")
703 return this->HandleSquit(parameters,pcnt,user);
705 else if (command == "MAP")
707 this->HandleMap(parameters,pcnt,user);
710 else if ((command == "TIME") && (pcnt > 0))
712 return this->HandleTime(parameters,pcnt,user);
714 else if (command == "LUSERS")
716 this->HandleLusers(parameters,pcnt,user);
719 else if (command == "LINKS")
721 this->HandleLinks(parameters,pcnt,user);
724 else if (command == "WHOIS")
729 return this->HandleRemoteWhois(parameters,pcnt,user);
732 else if ((command == "VERSION") && (pcnt > 0))
734 this->HandleVersion(parameters,pcnt,user);
737 else if ((command == "MODULES") && (pcnt > 0))
739 return this->HandleModules(parameters,pcnt,user);
744 void ModuleSpanningTree::OnPostCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, CmdResult result, const std::string &original_line)
746 if ((result == CMD_SUCCESS) && (ServerInstance->IsValidModuleCommand(command, pcnt, user)))
748 // this bit of code cleverly routes all module commands
749 // to all remote severs *automatically* so that modules
750 // can just handle commands locally, without having
751 // to have any special provision in place for remote
752 // commands and linking protocols.
753 std::deque<std::string> params;
755 for (int j = 0; j < pcnt; j++)
757 if (strchr(parameters[j],' '))
759 params.push_back(":" + std::string(parameters[j]));
763 params.push_back(std::string(parameters[j]));
766 Utils->DoOneToMany(user->nick,command,params);
770 void ModuleSpanningTree::OnGetServerDescription(const std::string &servername,std::string &description)
772 TreeServer* s = Utils->FindServer(servername);
775 description = s->GetDesc();
779 void ModuleSpanningTree::OnUserInvite(userrec* source,userrec* dest,chanrec* channel)
781 if (IS_LOCAL(source))
783 std::deque<std::string> params;
784 params.push_back(dest->nick);
785 params.push_back(channel->name);
786 Utils->DoOneToMany(source->nick,"INVITE",params);
790 void ModuleSpanningTree::OnPostLocalTopicChange(userrec* user, chanrec* chan, const std::string &topic)
792 std::deque<std::string> params;
793 params.push_back(chan->name);
794 params.push_back(":"+topic);
795 Utils->DoOneToMany(user->nick,"TOPIC",params);
798 void ModuleSpanningTree::OnWallops(userrec* user, const std::string &text)
802 std::deque<std::string> params;
803 params.push_back(":"+text);
804 Utils->DoOneToMany(user->nick,"WALLOPS",params);
808 void ModuleSpanningTree::OnUserNotice(userrec* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
810 if (target_type == TYPE_USER)
812 userrec* d = (userrec*)dest;
813 if ((d->GetFd() < 0) && (IS_LOCAL(user)))
815 std::deque<std::string> params;
817 params.push_back(d->nick);
818 params.push_back(":"+text);
819 Utils->DoOneToOne(user->nick,"NOTICE",params,d->server);
822 else if (target_type == TYPE_CHANNEL)
826 chanrec *c = (chanrec*)dest;
829 std::string cname = c->name;
831 cname = status + cname;
833 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
834 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
836 TreeSocket* Sock = i->second->GetSocket();
838 Sock->WriteLine(":"+std::string(user->nick)+" NOTICE "+cname+" :"+text);
843 else if (target_type == TYPE_SERVER)
847 char* target = (char*)dest;
848 std::deque<std::string> par;
849 par.push_back(target);
850 par.push_back(":"+text);
851 Utils->DoOneToMany(user->nick,"NOTICE",par);
856 void ModuleSpanningTree::OnUserMessage(userrec* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
858 if (target_type == TYPE_USER)
860 // route private messages which are targetted at clients only to the server
861 // which needs to receive them
862 userrec* d = (userrec*)dest;
863 if ((d->GetFd() < 0) && (IS_LOCAL(user)))
865 std::deque<std::string> params;
867 params.push_back(d->nick);
868 params.push_back(":"+text);
869 Utils->DoOneToOne(user->nick,"PRIVMSG",params,d->server);
872 else if (target_type == TYPE_CHANNEL)
876 chanrec *c = (chanrec*)dest;
879 std::string cname = c->name;
881 cname = status + cname;
883 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
884 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
886 TreeSocket* Sock = i->second->GetSocket();
888 Sock->WriteLine(":"+std::string(user->nick)+" PRIVMSG "+cname+" :"+text);
893 else if (target_type == TYPE_SERVER)
897 char* target = (char*)dest;
898 std::deque<std::string> par;
899 par.push_back(target);
900 par.push_back(":"+text);
901 Utils->DoOneToMany(user->nick,"PRIVMSG",par);
906 void ModuleSpanningTree::OnBackgroundTimer(time_t curtime)
908 AutoConnectServers(curtime);
909 DoPingChecks(curtime);
912 void ModuleSpanningTree::OnUserJoin(userrec* user, chanrec* channel, bool &silent)
914 // Only do this for local users
917 if (channel->GetUserCounter() == 1)
919 std::deque<std::string> params;
920 // set up their permissions and the channel TS with FJOIN.
921 // All users are FJOINed now, because a module may specify
922 // new joining permissions for the user.
923 params.push_back(channel->name);
924 params.push_back(ConvToStr(channel->age));
925 params.push_back(std::string(channel->GetAllPrefixChars(user))+","+std::string(user->nick));
926 Utils->DoOneToMany(ServerInstance->Config->ServerName,"FJOIN",params);
927 /* First user in, sync the modes for the channel */
929 params.push_back(channel->ChanModes(true));
930 Utils->DoOneToMany(ServerInstance->Config->ServerName,"FMODE",params);
934 std::deque<std::string> params;
935 params.push_back(channel->name);
936 params.push_back(ConvToStr(channel->age));
937 Utils->DoOneToMany(user->nick,"JOIN",params);
942 void ModuleSpanningTree::OnChangeHost(userrec* user, const std::string &newhost)
944 // only occurs for local clients
945 if (user->registered != REG_ALL)
947 std::deque<std::string> params;
948 params.push_back(newhost);
949 Utils->DoOneToMany(user->nick,"FHOST",params);
952 void ModuleSpanningTree::OnChangeName(userrec* user, const std::string &gecos)
954 // only occurs for local clients
955 if (user->registered != REG_ALL)
957 std::deque<std::string> params;
958 params.push_back(gecos);
959 Utils->DoOneToMany(user->nick,"FNAME",params);
962 void ModuleSpanningTree::OnUserPart(userrec* user, chanrec* channel, const std::string &partmessage, bool &silent)
966 std::deque<std::string> params;
967 params.push_back(channel->name);
968 if (partmessage != "")
969 params.push_back(":"+partmessage);
970 Utils->DoOneToMany(user->nick,"PART",params);
974 void ModuleSpanningTree::OnUserConnect(userrec* user)
979 std::deque<std::string> params;
980 snprintf(agestr,MAXBUF,"%lu",(unsigned long)user->age);
981 params.push_back(agestr);
982 params.push_back(user->nick);
983 params.push_back(user->host);
984 params.push_back(user->dhost);
985 params.push_back(user->ident);
986 params.push_back("+"+std::string(user->FormatModes()));
987 params.push_back(user->GetIPString());
988 params.push_back(":"+std::string(user->fullname));
989 Utils->DoOneToMany(ServerInstance->Config->ServerName,"NICK",params);
990 // User is Local, change needs to be reflected!
991 TreeServer* SourceServer = Utils->FindServer(user->server);
994 SourceServer->AddUserCount();
999 void ModuleSpanningTree::OnUserQuit(userrec* user, const std::string &reason, const std::string &oper_message)
1001 if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
1003 std::deque<std::string> params;
1005 if (oper_message != reason)
1007 params.push_back(":"+oper_message);
1008 Utils->DoOneToMany(user->nick,"OPERQUIT",params);
1011 params.push_back(":"+reason);
1012 Utils->DoOneToMany(user->nick,"QUIT",params);
1014 // Regardless, We need to modify the user Counts..
1015 TreeServer* SourceServer = Utils->FindServer(user->server);
1018 SourceServer->DelUserCount();
1022 void ModuleSpanningTree::OnUserPostNick(userrec* user, const std::string &oldnick)
1026 std::deque<std::string> params;
1027 params.push_back(user->nick);
1028 Utils->DoOneToMany(oldnick,"NICK",params);
1032 void ModuleSpanningTree::OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason, bool &silent)
1034 if ((source) && (IS_LOCAL(source)))
1036 std::deque<std::string> params;
1037 params.push_back(chan->name);
1038 params.push_back(user->nick);
1039 params.push_back(":"+reason);
1040 Utils->DoOneToMany(source->nick,"KICK",params);
1044 std::deque<std::string> params;
1045 params.push_back(chan->name);
1046 params.push_back(user->nick);
1047 params.push_back(":"+reason);
1048 Utils->DoOneToMany(ServerInstance->Config->ServerName,"KICK",params);
1052 void ModuleSpanningTree::OnRemoteKill(userrec* source, userrec* dest, const std::string &reason)
1054 std::deque<std::string> params;
1055 params.push_back(dest->nick);
1056 params.push_back(":"+reason);
1057 Utils->DoOneToMany(source->nick,"KILL",params);
1060 void ModuleSpanningTree::OnRehash(userrec* user, const std::string ¶meter)
1062 if (parameter != "")
1064 std::deque<std::string> params;
1065 params.push_back(parameter);
1066 Utils->DoOneToMany(user ? user->nick : ServerInstance->Config->ServerName, "REHASH", params);
1068 if (ServerInstance->MatchText(ServerInstance->Config->ServerName,parameter))
1070 ServerInstance->WriteOpers("*** Remote rehash initiated locally by \002%s\002", user ? user->nick : ServerInstance->Config->ServerName);
1071 ServerInstance->RehashServer();
1074 Utils->ReadConfiguration(false);
1075 InitializeDisabledCommands(ServerInstance->Config->DisabledCommands, ServerInstance);
1078 // note: the protocol does not allow direct umode +o except
1079 // via NICK with 8 params. sending OPERTYPE infers +o modechange
1081 void ModuleSpanningTree::OnOper(userrec* user, const std::string &opertype)
1085 std::deque<std::string> params;
1086 params.push_back(opertype);
1087 Utils->DoOneToMany(user->nick,"OPERTYPE",params);
1091 void ModuleSpanningTree::OnLine(userrec* source, const std::string &host, bool adding, char linetype, long duration, const std::string &reason)
1095 /* Server-set lines */
1097 snprintf(data,MAXBUF,"%c %s %s %lu %lu :%s", linetype, host.c_str(), ServerInstance->Config->ServerName, (unsigned long)ServerInstance->Time(false),
1098 (unsigned long)duration, reason.c_str());
1099 std::deque<std::string> params;
1100 params.push_back(data);
1101 Utils->DoOneToMany(ServerInstance->Config->ServerName, "ADDLINE", params);
1105 if (IS_LOCAL(source))
1108 snprintf(type,8,"%cLINE",linetype);
1109 std::string stype = type;
1112 char sduration[MAXBUF];
1113 snprintf(sduration,MAXBUF,"%ld",duration);
1114 std::deque<std::string> params;
1115 params.push_back(host);
1116 params.push_back(sduration);
1117 params.push_back(":"+reason);
1118 Utils->DoOneToMany(source->nick,stype,params);
1122 std::deque<std::string> params;
1123 params.push_back(host);
1124 Utils->DoOneToMany(source->nick,stype,params);
1130 void ModuleSpanningTree::OnAddGLine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
1132 OnLine(source,hostmask,true,'G',duration,reason);
1135 void ModuleSpanningTree::OnAddZLine(long duration, userrec* source, const std::string &reason, const std::string &ipmask)
1137 OnLine(source,ipmask,true,'Z',duration,reason);
1140 void ModuleSpanningTree::OnAddQLine(long duration, userrec* source, const std::string &reason, const std::string &nickmask)
1142 OnLine(source,nickmask,true,'Q',duration,reason);
1145 void ModuleSpanningTree::OnAddELine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
1147 OnLine(source,hostmask,true,'E',duration,reason);
1150 void ModuleSpanningTree::OnDelGLine(userrec* source, const std::string &hostmask)
1152 OnLine(source,hostmask,false,'G',0,"");
1155 void ModuleSpanningTree::OnDelZLine(userrec* source, const std::string &ipmask)
1157 OnLine(source,ipmask,false,'Z',0,"");
1160 void ModuleSpanningTree::OnDelQLine(userrec* source, const std::string &nickmask)
1162 OnLine(source,nickmask,false,'Q',0,"");
1165 void ModuleSpanningTree::OnDelELine(userrec* source, const std::string &hostmask)
1167 OnLine(source,hostmask,false,'E',0,"");
1170 void ModuleSpanningTree::OnMode(userrec* user, void* dest, int target_type, const std::string &text)
1172 if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
1174 std::deque<std::string> params;
1175 std::string command;
1177 if (target_type == TYPE_USER)
1179 userrec* u = (userrec*)dest;
1180 params.push_back(u->nick);
1181 params.push_back(text);
1186 chanrec* c = (chanrec*)dest;
1187 params.push_back(c->name);
1188 params.push_back(ConvToStr(c->age));
1189 params.push_back(text);
1192 Utils->DoOneToMany(user->nick, command, params);
1196 void ModuleSpanningTree::OnSetAway(userrec* user)
1200 std::deque<std::string> params;
1201 params.push_back(":"+std::string(user->awaymsg));
1202 Utils->DoOneToMany(user->nick,"AWAY",params);
1206 void ModuleSpanningTree::OnCancelAway(userrec* user)
1210 std::deque<std::string> params;
1212 Utils->DoOneToMany(user->nick,"AWAY",params);
1216 void ModuleSpanningTree::ProtoSendMode(void* opaque, int target_type, void* target, const std::string &modeline)
1218 TreeSocket* s = (TreeSocket*)opaque;
1221 if (target_type == TYPE_USER)
1223 userrec* u = (userrec*)target;
1224 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" FMODE "+u->nick+" "+ConvToStr(u->age)+" "+modeline);
1228 chanrec* c = (chanrec*)target;
1229 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" FMODE "+c->name+" "+ConvToStr(c->age)+" "+modeline);
1234 void ModuleSpanningTree::ProtoSendMetaData(void* opaque, int target_type, void* target, const std::string &extname, const std::string &extdata)
1236 TreeSocket* s = (TreeSocket*)opaque;
1239 if (target_type == TYPE_USER)
1241 userrec* u = (userrec*)target;
1242 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA "+u->nick+" "+extname+" :"+extdata);
1244 else if (target_type == TYPE_CHANNEL)
1246 chanrec* c = (chanrec*)target;
1247 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA "+c->name+" "+extname+" :"+extdata);
1250 if (target_type == TYPE_OTHER)
1252 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA * "+extname+" :"+extdata);
1256 void ModuleSpanningTree::OnEvent(Event* event)
1258 std::deque<std::string>* params = (std::deque<std::string>*)event->GetData();
1259 if (event->GetEventID() == "send_metadata")
1261 if (params->size() < 3)
1263 (*params)[2] = ":" + (*params)[2];
1264 Utils->DoOneToMany(ServerInstance->Config->ServerName,"METADATA",*params);
1266 else if (event->GetEventID() == "send_topic")
1268 if (params->size() < 2)
1270 (*params)[1] = ":" + (*params)[1];
1271 params->insert(params->begin() + 1,ServerInstance->Config->ServerName);
1272 params->insert(params->begin() + 1,ConvToStr(ServerInstance->Time(true)));
1273 Utils->DoOneToMany(ServerInstance->Config->ServerName,"FTOPIC",*params);
1275 else if (event->GetEventID() == "send_mode")
1277 if (params->size() < 2)
1279 // Insert the TS value of the object, either userrec or chanrec
1281 userrec* a = ServerInstance->FindNick((*params)[0]);
1285 Utils->DoOneToMany(ServerInstance->Config->ServerName,"MODE",*params);
1290 chanrec* a = ServerInstance->FindChan((*params)[0]);
1294 params->insert(params->begin() + 1,ConvToStr(ourTS));
1295 Utils->DoOneToMany(ServerInstance->Config->ServerName,"FMODE",*params);
1299 else if (event->GetEventID() == "send_mode_explicit")
1301 if (params->size() < 2)
1303 Utils->DoOneToMany(ServerInstance->Config->ServerName,"MODE",*params);
1305 else if (event->GetEventID() == "send_opers")
1307 if (params->size() < 1)
1309 (*params)[0] = ":" + (*params)[0];
1310 Utils->DoOneToMany(ServerInstance->Config->ServerName,"OPERNOTICE",*params);
1312 else if (event->GetEventID() == "send_modeset")
1314 if (params->size() < 2)
1316 (*params)[1] = ":" + (*params)[1];
1317 Utils->DoOneToMany(ServerInstance->Config->ServerName,"MODENOTICE",*params);
1319 else if (event->GetEventID() == "send_snoset")
1321 if (params->size() < 2)
1323 (*params)[1] = ":" + (*params)[1];
1324 Utils->DoOneToMany(ServerInstance->Config->ServerName,"SNONOTICE",*params);
1326 else if (event->GetEventID() == "send_push")
1328 if (params->size() < 2)
1331 userrec *a = ServerInstance->FindNick((*params)[0]);
1336 (*params)[1] = ":" + (*params)[1];
1337 Utils->DoOneToOne(ServerInstance->Config->ServerName, "PUSH", *params, a->server);
1341 ModuleSpanningTree::~ModuleSpanningTree()
1343 /* This will also free the listeners */
1346 ServerInstance->Timers->DelTimer(SyncTimer);
1348 ServerInstance->Timers->DelTimer(RefreshTimer);
1350 ServerInstance->DoneWithInterface("InspSocketHook");
1353 Version ModuleSpanningTree::GetVersion()
1355 return Version(1,1,0,2,VF_VENDOR,API_VERSION);
1358 void ModuleSpanningTree::Implements(char* List)
1360 List[I_OnPreCommand] = List[I_OnGetServerDescription] = List[I_OnUserInvite] = List[I_OnPostLocalTopicChange] = 1;
1361 List[I_OnWallops] = List[I_OnUserNotice] = List[I_OnUserMessage] = List[I_OnBackgroundTimer] = 1;
1362 List[I_OnUserJoin] = List[I_OnChangeHost] = List[I_OnChangeName] = List[I_OnUserPart] = List[I_OnUserConnect] = 1;
1363 List[I_OnUserQuit] = List[I_OnUserPostNick] = List[I_OnUserKick] = List[I_OnRemoteKill] = List[I_OnRehash] = 1;
1364 List[I_OnOper] = List[I_OnAddGLine] = List[I_OnAddZLine] = List[I_OnAddQLine] = List[I_OnAddELine] = 1;
1365 List[I_OnDelGLine] = List[I_OnDelZLine] = List[I_OnDelQLine] = List[I_OnDelELine] = List[I_ProtoSendMode] = List[I_OnMode] = 1;
1366 List[I_OnStats] = List[I_ProtoSendMetaData] = List[I_OnEvent] = List[I_OnSetAway] = List[I_OnCancelAway] = List[I_OnPostCommand] = 1;
1369 /* It is IMPORTANT that m_spanningtree is the last module in the chain
1370 * so that any activity it sees is FINAL, e.g. we arent going to send out
1371 * a NICK message before m_cloaking has finished putting the +x on the user,
1373 * Therefore, we return PRIORITY_LAST to make sure we end up at the END of
1374 * the module call queue.
1376 Priority ModuleSpanningTree::Prioritize()
1378 return PRIORITY_LAST;
1381 class ModuleSpanningTreeFactory : public ModuleFactory
1384 ModuleSpanningTreeFactory()
1388 ~ModuleSpanningTreeFactory()
1392 virtual Module * CreateModule(InspIRCd* Me)
1394 return new ModuleSpanningTree(Me);
1400 extern "C" DllExport void * init_module( void )
1402 return new ModuleSpanningTreeFactory;