]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/m_spanningtree/main.cpp
bebb69012bb0c0fadb842269bb3be018c8c51350
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / main.cpp
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
6  * See: http://www.inspircd.org/wiki/index.php/Credits
7  *
8  * This program is free but copyrighted software; see
9  *          the file COPYING for details.
10  *
11  * ---------------------------------------------------
12  */
13
14 /* $ModDesc: Provides a spanning tree server link protocol */
15
16 #include "inspircd.h"
17 #include "configreader.h"
18 #include "users.h"
19 #include "channels.h"
20 #include "modules.h"
21 #include "commands/cmd_whois.h"
22 #include "commands/cmd_stats.h"
23 #include "socket.h"
24 #include "wildcard.h"
25 #include "xline.h"
26 #include "transport.h"
27
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"
36
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 */
38
39 ModuleSpanningTree::ModuleSpanningTree(InspIRCd* Me)
40         : Module(Me), max_local(0), max_global(0)
41 {
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)
47         {
48                 SyncTimer = new TimeSyncTimer(ServerInstance, this);
49                 ServerInstance->Timers->AddTimer(SyncTimer);
50         }
51         else
52                 SyncTimer = NULL;
53
54         RefreshTimer = new CacheRefreshTimer(ServerInstance, Utils);
55         ServerInstance->Timers->AddTimer(RefreshTimer);
56 }
57
58 void ModuleSpanningTree::ShowLinks(TreeServer* Current, userrec* user, int hops)
59 {
60         std::string Parent = Utils->TreeRoot->GetName();
61         if (Current->GetParent())
62         {
63                 Parent = Current->GetParent()->GetName();
64         }
65         for (unsigned int q = 0; q < Current->ChildCount(); q++)
66         {
67                 if ((Current->GetChild(q)->Hidden) || ((Utils->HideULines) && (ServerInstance->ULine(Current->GetChild(q)->GetName().c_str()))))
68                 {
69                         if (*user->oper)
70                         {
71                                  ShowLinks(Current->GetChild(q),user,hops+1);
72                         }
73                 }
74                 else
75                 {
76                         ShowLinks(Current->GetChild(q),user,hops+1);
77                 }
78         }
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)))
81                 return;
82         /* Or if the server is hidden and they're not an oper */
83         else if ((Current->Hidden) && (!IS_OPER(user)))
84                 return;
85
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());
90 }
91
92 int ModuleSpanningTree::CountLocalServs()
93 {
94         return Utils->TreeRoot->ChildCount();
95 }
96
97 int ModuleSpanningTree::CountServs()
98 {
99         return Utils->serverlist.size();
100 }
101
102 void ModuleSpanningTree::HandleLinks(const char** parameters, int pcnt, userrec* user)
103 {
104         ShowLinks(Utils->TreeRoot,user,0);
105         user->WriteServ("365 %s * :End of /LINKS list.",user->nick);
106         return;
107 }
108
109 void ModuleSpanningTree::HandleLusers(const char** parameters, int pcnt, userrec* user)
110 {
111         unsigned int n_users = ServerInstance->UserCount();
112
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;
118
119         unsigned int ulined_count = 0;
120         unsigned int ulined_local_count = 0;
121
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)
124          */
125         if ((Utils->HideULines) && (!*user->oper))
126         {
127                 for (server_hash::iterator q = Utils->serverlist.begin(); q != Utils->serverlist.end(); q++)
128                 {
129                         if (ServerInstance->ULine(q->second->GetName().c_str()))
130                         {
131                                 ulined_count++;
132                                 if (q->second->GetParent() == Utils->TreeRoot)
133                                         ulined_local_count++;
134                         }
135                 }
136         }
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);
147         return;
148 }
149
150 std::string ModuleSpanningTree::TimeToStr(time_t secs)
151 {
152         time_t mins_up = secs / 60;
153         time_t hours_up = mins_up / 60;
154         time_t days_up = hours_up / 24;
155         secs = secs % 60;
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");
162 }
163
164 const std::string ModuleSpanningTree::MapOperInfo(TreeServer* Current)
165 {
166         time_t secs_up = ServerInstance->Time() - Current->age;
167         return (" [Up: " + TimeToStr(secs_up) + " Lag: "+ConvToStr(Current->rtt)+"s]");
168 }
169
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)
172 {
173         if (line < 128)
174         {
175                 for (int t = 0; t < depth; t++)
176                 {
177                         matrix[line][t] = ' ';
178                 }
179                 // For Aligning, we need to work out exactly how deep this thing is, and produce
180                 // a 'Spacer' String to compensate.
181                 char spacer[40];
182                 memset(spacer,' ',40);
183                 if ((40 - Current->GetName().length() - depth) > 1) {
184                         spacer[40 - Current->GetName().length() - depth] = '\0';
185                 }
186                 else
187                 {
188                         spacer[5] = '\0';
189                 }
190                 float percent;
191                 char text[128];
192                 /* Neat and tidy default values, as we're dealing with a matrix not a simple string */
193                 memset(text, 0, 128);
194
195                 if (ServerInstance->clientlist->size() == 0) {
196                         // If there are no users, WHO THE HELL DID THE /MAP?!?!?!
197                         percent = 0;
198                 }
199                 else
200                 {
201                         percent = ((float)Current->GetUserCount() / (float)ServerInstance->clientlist->size()) * 100;
202                 }
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();
206                 totservers++;
207                 strlcpy(&matrix[line][depth],text,126);
208                 line++;
209                 for (unsigned int q = 0; q < Current->ChildCount(); q++)
210                 {
211                         if ((Current->GetChild(q)->Hidden) || ((Utils->HideULines) && (ServerInstance->ULine(Current->GetChild(q)->GetName().c_str()))))
212                         {
213                                 if (*user->oper)
214                                 {
215                                         ShowMap(Current->GetChild(q),user,(Utils->FlatLinks && (!*user->oper)) ? depth : depth+2,matrix,totusers,totservers);
216                                 }
217                         }
218                         else
219                         {
220                                 ShowMap(Current->GetChild(q),user,(Utils->FlatLinks && (!*user->oper)) ? depth : depth+2,matrix,totusers,totservers);
221                         }
222                 }
223         }
224 }
225
226 int ModuleSpanningTree::HandleMotd(const char** parameters, int pcnt, userrec* user)
227 {
228         if (pcnt > 0)
229         {
230                 if (match(ServerInstance->Config->ServerName, parameters[0]))
231                         return 0;
232
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]);
238                 if (s)
239                         Utils->DoOneToOne(user->nick, "MOTD", params, s->GetName());
240                 else
241                         user->WriteServ( "402 %s %s :No such server", user->nick, parameters[0]);
242                 return 1;
243         }
244         return 0;
245 }
246
247 int ModuleSpanningTree::HandleAdmin(const char** parameters, int pcnt, userrec* user)
248 {
249         if (pcnt > 0)
250         {
251                 if (match(ServerInstance->Config->ServerName, parameters[0]))
252                         return 1;
253
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]);
259                 if (s)
260                         Utils->DoOneToOne(user->nick, "ADMIN", params, s->GetName());
261                 else
262                         user->WriteServ( "402 %s %s :No such server", user->nick, parameters[0]);
263                 return 1;
264         }
265         return 0;
266 }
267
268 int ModuleSpanningTree::HandleModules(const char** parameters, int pcnt, userrec* user)
269 {
270         if (match(ServerInstance->Config->ServerName, parameters[0]))
271                 return 1;
272
273         std::deque<std::string> params;
274         params.push_back(parameters[0]);
275         TreeServer* s = Utils->FindServerMask(parameters[0]);
276         if (s)
277                 Utils->DoOneToOne(user->nick, "MODULES", params, s->GetName());
278         else
279                 user->WriteServ( "402 %s %s :No such server", user->nick, parameters[0]);
280         return 1;
281 }
282
283 int ModuleSpanningTree::HandleStats(const char** parameters, int pcnt, userrec* user)
284 {
285         if (pcnt > 1)
286         {
287                 if (match(ServerInstance->Config->ServerName, parameters[1]))
288                         return 0;
289
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 */
295
296                 TreeServer* s = Utils->FindServerMask(parameters[1]);
297                 if (s)
298                 {
299                         params[1] = s->GetName();
300                         Utils->DoOneToOne(user->nick, "STATS", params, s->GetName());
301                 }
302                 else
303                 {
304                         user->WriteServ( "402 %s %s :No such server", user->nick, parameters[1]);
305                 }
306                 return 1;
307         }
308         return 0;
309 }
310
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)
320 {
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.
324         float totusers = 0;
325         float totservers = 0;
326         char matrix[128][128];
327         for (unsigned int t = 0; t < 128; t++)
328         {
329                 matrix[t][0] = '\0';
330         }
331         line = 0;
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++)
339         {
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
343                 // are related to)
344                 int first_nonspace = 0;
345                 while (matrix[l][first_nonspace] == ' ')
346                 {
347                         first_nonspace++;
348                 }
349                 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] = '`';
355                 int l2 = l - 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] == '`'))
359                 {
360                         matrix[l2][first_nonspace-1] = '|';
361                         l2--;
362                 }
363         }
364         // dump the whole lot to the user. This is the easy bit, honest.
365         for (int t = 0; t < line; t++)
366         {
367                 user->WriteServ("006 %s :%s",user->nick,&matrix[t][0]);
368         }
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);
372         return;
373 }
374
375 int ModuleSpanningTree::HandleSquit(const char** parameters, int pcnt, userrec* user)
376 {
377         TreeServer* s = Utils->FindServerMask(parameters[0]);
378         if (s)
379         {
380                 if (s == Utils->TreeRoot)
381                 {
382                         user->WriteServ("NOTICE %s :*** SQUIT: Foolish mortal, you cannot make a server SQUIT itself! (%s matches local server name)",user->nick,parameters[0]);
383                         return 1;
384                 }
385                 TreeSocket* sock = s->GetSocket();
386                 if (sock)
387                 {
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);
391                         sock->Close();
392                         delete sock;
393                 }
394                 else
395                 {
396                         /* route it */
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]);
401                 }
402         }
403         else
404         {
405                  user->WriteServ("NOTICE %s :*** SQUIT: The server \002%s\002 does not exist on the network.",user->nick,parameters[0]);
406         }
407         return 1;
408 }
409
410 int ModuleSpanningTree::HandleTime(const char** parameters, int pcnt, userrec* user)
411 {
412         if ((IS_LOCAL(user)) && (pcnt))
413         {
414                 TreeServer* found = Utils->FindServerMask(parameters[0]);
415                 if (found)
416                 {
417                         // we dont' override for local server
418                         if (found == Utils->TreeRoot)
419                                 return 0;
420                         
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());
425                 }
426                 else
427                 {
428                         user->WriteServ("402 %s %s :No such server",user->nick,parameters[0]);
429                 }
430         }
431         return 1;
432 }
433
434 int ModuleSpanningTree::HandleRemoteWhois(const char** parameters, int pcnt, userrec* user)
435 {
436         if ((IS_LOCAL(user)) && (pcnt > 1))
437         {
438                 userrec* remote = ServerInstance->FindNick(parameters[1]);
439                 if ((remote) && (remote->GetFd() < 0))
440                 {
441                         std::deque<std::string> params;
442                         params.push_back(parameters[1]);
443                         Utils->DoOneToOne(user->nick,"IDLE",params,remote->server);
444                         return 1;
445                 }
446                 else if (!remote)
447                 {
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]);
450                         return 1;
451                 }
452         }
453         return 0;
454 }
455
456 void ModuleSpanningTree::DoPingChecks(time_t curtime)
457 {
458         for (unsigned int j = 0; j < Utils->TreeRoot->ChildCount(); j++)
459         {
460                 TreeServer* serv = Utils->TreeRoot->GetChild(j);
461                 TreeSocket* sock = serv->GetSocket();
462                 if (sock)
463                 {
464                         if (curtime >= serv->NextPingTime())
465                         {
466                                 if (serv->AnsweredLastPing())
467                                 {
468                                         sock->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" PING "+serv->GetName());
469                                         serv->SetNextPingTime(curtime + 60);
470                                         serv->LastPing = curtime;
471                                         serv->Warned = false;
472                                 }
473                                 else
474                                 {
475                                         /* they didnt answer, boot them */
476                                         sock->SendError("Ping timeout");
477                                         sock->Squit(serv,"Ping timeout");
478                                         ServerInstance->SE->DelFd(sock);
479                                         sock->Close();
480                                         delete sock;
481                                         return;
482                                 }
483                         }
484                         else if ((Utils->PingWarnTime) && (!serv->Warned) && (curtime >= serv->NextPingTime() - (60 - Utils->PingWarnTime)) && (!serv->AnsweredLastPing()))
485                         {
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);
488                                 serv->Warned = true;
489                         }
490                 }
491         }
492
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
495          */
496         for (server_hash::iterator i = Utils->serverlist.begin(); i != Utils->serverlist.end(); i++)
497                 Utils->SetRemoteBursting(i->second, false);
498 }
499
500 void ModuleSpanningTree::ConnectServer(Link* x)
501 {
502         bool ipvalid = true;
503         QueryType start_type = DNS_QUERY_A;
504 #ifdef IPV6
505         start_type = DNS_QUERY_AAAA;
506         if (strchr(x->IPAddr.c_str(),':'))
507         {
508                 in6_addr n;
509                 if (inet_pton(AF_INET6, x->IPAddr.c_str(), &n) < 1)
510                         ipvalid = false;
511         }
512         else
513         {
514                 in_addr n;
515                 if (inet_aton(x->IPAddr.c_str(),&n) < 1)
516                         ipvalid = false;
517         }
518 #else
519                 in_addr n;
520                 if (inet_aton(x->IPAddr.c_str(),&n) < 1)
521                         ipvalid = false;
522 #endif
523
524         /* Do we already have an IP? If so, no need to resolve it. */
525         if (ipvalid)
526         {
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()))
529                         return;
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)
532                 {
533                         /* Handled automatically on success */
534                 }
535                 else
536                 {
537                         ServerInstance->SNO->WriteToSnoMask('l',"CONNECT: Error connecting \002%s\002: %s.",x->Name.c_str(),strerror(errno));
538                         delete newsocket;
539                         Utils->DoFailOver(x);
540                 }
541         }
542         else
543         {
544                 try
545                 {
546                         bool cached;
547                         ServernameResolver* snr = new ServernameResolver((Module*)this, Utils, ServerInstance,x->IPAddr, *x, cached, start_type);
548                         ServerInstance->AddResolver(snr, cached);
549                 }
550                 catch (ModuleException& e)
551                 {
552                         ServerInstance->SNO->WriteToSnoMask('l',"CONNECT: Error connecting \002%s\002: %s.",x->Name.c_str(), e.GetReason());
553                         Utils->DoFailOver(x);
554                 }
555         }
556 }
557
558 void ModuleSpanningTree::AutoConnectServers(time_t curtime)
559 {
560         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
561         {
562                 if ((x->AutoConnect) && (curtime >= x->NextConnectTime))
563                 {
564                         x->NextConnectTime = curtime + x->AutoConnect;
565                         TreeServer* CheckDupe = Utils->FindServer(x->Name.c_str());
566                         if (x->FailOver.length())
567                         {
568                                 TreeServer* CheckFailOver = Utils->FindServer(x->FailOver.c_str());
569                                 if (CheckFailOver)
570                                 {
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.
574                                          */
575                                         continue;
576                                 }
577                         }
578                         if (!CheckDupe)
579                         {
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));
583                         }
584                 }
585         }
586 }
587
588 int ModuleSpanningTree::HandleVersion(const char** parameters, int pcnt, userrec* user)
589 {
590         // we've already checked if pcnt > 0, so this is safe
591         TreeServer* found = Utils->FindServerMask(parameters[0]);
592         if (found)
593         {
594                 std::string Version = found->GetVersion();
595                 user->WriteServ("351 %s :%s",user->nick,Version.c_str());
596                 if (found == Utils->TreeRoot)
597                 {
598                         ServerInstance->Config->Send005(user);
599                 }
600         }
601         else
602         {
603                 user->WriteServ("402 %s %s :No such server",user->nick,parameters[0]);
604         }
605         return 1;
606 }
607         
608 int ModuleSpanningTree::HandleConnect(const char** parameters, int pcnt, userrec* user)
609 {
610         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
611         {
612                 if (ServerInstance->MatchText(x->Name.c_str(),parameters[0]))
613                 {
614                         TreeServer* CheckDupe = Utils->FindServer(x->Name.c_str());
615                         if (!CheckDupe)
616                         {
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));
619                                 return 1;
620                         }
621                         else
622                         {
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());
624                                 return 1;
625                         }
626                 }
627         }
628         user->WriteServ("NOTICE %s :*** CONNECT: No server matching \002%s\002 could be found in the config file.",user->nick,parameters[0]);
629         return 1;
630 }
631
632 void ModuleSpanningTree::BroadcastTimeSync()
633 {
634         if (Utils->MasterTime)
635         {
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);
640         }
641 }
642
643 int ModuleSpanningTree::OnStats(char statschar, userrec* user, string_list &results)
644 {
645         if ((statschar == 'c') || (statschar == 'n'))
646         {
647                 for (unsigned int i = 0; i < Utils->LinkBlocks.size(); i++)
648                 {
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());
652                 }
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);
655                 return 1;
656         }
657
658         if (statschar == 'p')
659         {
660                 /* show all server ports, after showing client ports. -- w00t */
661
662                 for (unsigned int i = 0; i < Utils->Bindings.size(); i++)
663                 {
664                         /*
665                          * XXX - todo:
666                          *  we need to remove duplicate ports from this list, and possibly also
667                          *  show the correct number of servers on the port.
668                          *
669                          * XXX also, this doesn't currently work, as something is not initialising ->port, so it's 0 all the time.
670                          *
671                          */
672                         if (Utils->Bindings[i]->port)
673                         {
674                                 std::string ip = Utils->Bindings[i]->IP;
675                                 if (ip.empty())
676                                         ip = "*";
677                                 results.push_back(ConvToStr(ServerInstance->Config->ServerName) + " 249 "+user->nick+" :" + ip + ":" + ConvToStr(Utils->Bindings[i]->port)+" (X servers) "+
678                                                 ServerInstance->Config->ports[i]->GetDescription());
679                         }
680                 }
681         }
682         return 0;
683 }
684
685 int ModuleSpanningTree::OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated, const std::string &original_line)
686 {
687         /* If the command doesnt appear to be valid, we dont want to mess with it. */
688         if (!validated)
689                 return 0;
690
691         if (command == "CONNECT")
692         {
693                 return this->HandleConnect(parameters,pcnt,user);
694         }
695         else if (command == "STATS")
696         {
697                 return this->HandleStats(parameters,pcnt,user);
698         }
699         else if (command == "MOTD")
700         {
701                 return this->HandleMotd(parameters,pcnt,user);
702         }
703         else if (command == "ADMIN")
704         {
705                 return this->HandleAdmin(parameters,pcnt,user);
706         }
707         else if (command == "SQUIT")
708         {
709                 return this->HandleSquit(parameters,pcnt,user);
710         }
711         else if (command == "MAP")
712         {
713                 this->HandleMap(parameters,pcnt,user);
714                 return 1;
715         }
716         else if ((command == "TIME") && (pcnt > 0))
717         {
718                 return this->HandleTime(parameters,pcnt,user);
719         }
720         else if (command == "LUSERS")
721         {
722                 this->HandleLusers(parameters,pcnt,user);
723                 return 1;
724         }
725         else if (command == "LINKS")
726         {
727                 this->HandleLinks(parameters,pcnt,user);
728                 return 1;
729         }
730         else if (command == "WHOIS")
731         {
732                 if (pcnt > 1)
733                 {
734                         // remote whois
735                         return this->HandleRemoteWhois(parameters,pcnt,user);
736                 }
737         }
738         else if ((command == "VERSION") && (pcnt > 0))
739         {
740                 this->HandleVersion(parameters,pcnt,user);
741                 return 1;
742         }
743         else if ((command == "MODULES") && (pcnt > 0))
744         {
745                 return this->HandleModules(parameters,pcnt,user);
746         }
747         return 0;
748 }
749
750 void ModuleSpanningTree::OnPostCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, CmdResult result, const std::string &original_line)
751 {
752         if ((result == CMD_SUCCESS) && (ServerInstance->IsValidModuleCommand(command, pcnt, user)))
753         {
754                 // this bit of code cleverly routes all module commands
755                 // to all remote severs *automatically* so that modules
756                 // can just handle commands locally, without having
757                 // to have any special provision in place for remote
758                 // commands and linking protocols.
759                 std::deque<std::string> params;
760                 params.clear();
761                 for (int j = 0; j < pcnt; j++)
762                 {
763                         if (strchr(parameters[j],' '))
764                         {
765                                 params.push_back(":" + std::string(parameters[j]));
766                         }
767                         else
768                         {
769                                 params.push_back(std::string(parameters[j]));
770                         }
771                 }
772                 Utils->DoOneToMany(user->nick,command,params);
773         }
774 }
775
776 void ModuleSpanningTree::OnGetServerDescription(const std::string &servername,std::string &description)
777 {
778         TreeServer* s = Utils->FindServer(servername);
779         if (s)
780         {
781                 description = s->GetDesc();
782         }
783 }
784
785 void ModuleSpanningTree::OnUserInvite(userrec* source,userrec* dest,chanrec* channel)
786 {
787         if (IS_LOCAL(source))
788         {
789                 std::deque<std::string> params;
790                 params.push_back(dest->nick);
791                 params.push_back(channel->name);
792                 Utils->DoOneToMany(source->nick,"INVITE",params);
793         }
794 }
795
796 void ModuleSpanningTree::OnPostLocalTopicChange(userrec* user, chanrec* chan, const std::string &topic)
797 {
798         std::deque<std::string> params;
799         params.push_back(chan->name);
800         params.push_back(":"+topic);
801         Utils->DoOneToMany(user->nick,"TOPIC",params);
802 }
803
804 void ModuleSpanningTree::OnWallops(userrec* user, const std::string &text)
805 {
806         if (IS_LOCAL(user))
807         {
808                 std::deque<std::string> params;
809                 params.push_back(":"+text);
810                 Utils->DoOneToMany(user->nick,"WALLOPS",params);
811         }
812 }
813
814 void ModuleSpanningTree::OnUserNotice(userrec* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
815 {
816         if (target_type == TYPE_USER)
817         {
818                 userrec* d = (userrec*)dest;
819                 if ((d->GetFd() < 0) && (IS_LOCAL(user)))
820                 {
821                         std::deque<std::string> params;
822                         params.clear();
823                         params.push_back(d->nick);
824                         params.push_back(":"+text);
825                         Utils->DoOneToOne(user->nick,"NOTICE",params,d->server);
826                 }
827         }
828         else if (target_type == TYPE_CHANNEL)
829         {
830                 if (IS_LOCAL(user))
831                 {
832                         chanrec *c = (chanrec*)dest;
833                         if (c)
834                         {
835                                 std::string cname = c->name;
836                                 if (status)
837                                         cname = status + cname;
838                                 TreeServerList list;
839                                 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
840                                 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
841                                 {
842                                         TreeSocket* Sock = i->second->GetSocket();
843                                         if (Sock)
844                                                 Sock->WriteLine(":"+std::string(user->nick)+" NOTICE "+cname+" :"+text);
845                                 }
846                         }
847                 }
848         }
849         else if (target_type == TYPE_SERVER)
850         {
851                 if (IS_LOCAL(user))
852                 {
853                         char* target = (char*)dest;
854                         std::deque<std::string> par;
855                         par.push_back(target);
856                         par.push_back(":"+text);
857                         Utils->DoOneToMany(user->nick,"NOTICE",par);
858                 }
859         }
860 }
861
862 void ModuleSpanningTree::OnUserMessage(userrec* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
863 {
864         if (target_type == TYPE_USER)
865         {
866                 // route private messages which are targetted at clients only to the server
867                 // which needs to receive them
868                 userrec* d = (userrec*)dest;
869                 if ((d->GetFd() < 0) && (IS_LOCAL(user)))
870                 {
871                         std::deque<std::string> params;
872                         params.clear();
873                         params.push_back(d->nick);
874                         params.push_back(":"+text);
875                         Utils->DoOneToOne(user->nick,"PRIVMSG",params,d->server);
876                 }
877         }
878         else if (target_type == TYPE_CHANNEL)
879         {
880                 if (IS_LOCAL(user))
881                 {
882                         chanrec *c = (chanrec*)dest;
883                         if (c)
884                         {
885                                 std::string cname = c->name;
886                                 if (status)
887                                         cname = status + cname;
888                                 TreeServerList list;
889                                 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
890                                 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
891                                 {
892                                         TreeSocket* Sock = i->second->GetSocket();
893                                         if (Sock)
894                                                 Sock->WriteLine(":"+std::string(user->nick)+" PRIVMSG "+cname+" :"+text);
895                                 }
896                         }
897                 }
898         }
899         else if (target_type == TYPE_SERVER)
900         {
901                 if (IS_LOCAL(user))
902                 {
903                         char* target = (char*)dest;
904                         std::deque<std::string> par;
905                         par.push_back(target);
906                         par.push_back(":"+text);
907                         Utils->DoOneToMany(user->nick,"PRIVMSG",par);
908                 }
909         }
910 }
911
912 void ModuleSpanningTree::OnBackgroundTimer(time_t curtime)
913 {
914         AutoConnectServers(curtime);
915         DoPingChecks(curtime);
916 }
917
918 void ModuleSpanningTree::OnUserJoin(userrec* user, chanrec* channel, bool &silent)
919 {
920         // Only do this for local users
921         if (IS_LOCAL(user))
922         {
923                 if (channel->GetUserCounter() == 1)
924                 {
925                         std::deque<std::string> params;
926                         // set up their permissions and the channel TS with FJOIN.
927                         // All users are FJOINed now, because a module may specify
928                         // new joining permissions for the user.
929                         params.push_back(channel->name);
930                         params.push_back(ConvToStr(channel->age));
931                         params.push_back(std::string(channel->GetAllPrefixChars(user))+","+std::string(user->nick));
932                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"FJOIN",params);
933                         /* First user in, sync the modes for the channel */
934                         params.pop_back();
935                         params.push_back(channel->ChanModes(true));
936                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"FMODE",params);
937                 }
938                 else
939                 {
940                         std::deque<std::string> params;
941                         params.push_back(channel->name);
942                         params.push_back(ConvToStr(channel->age));
943                         Utils->DoOneToMany(user->nick,"JOIN",params);
944                 }
945         }
946 }
947
948 void ModuleSpanningTree::OnChangeHost(userrec* user, const std::string &newhost)
949 {
950         // only occurs for local clients
951         if (user->registered != REG_ALL)
952                 return;
953         std::deque<std::string> params;
954         params.push_back(newhost);
955         Utils->DoOneToMany(user->nick,"FHOST",params);
956 }
957
958 void ModuleSpanningTree::OnChangeName(userrec* user, const std::string &gecos)
959 {
960         // only occurs for local clients
961         if (user->registered != REG_ALL)
962                 return;
963         std::deque<std::string> params;
964         params.push_back(gecos);
965         Utils->DoOneToMany(user->nick,"FNAME",params);
966 }
967
968 void ModuleSpanningTree::OnUserPart(userrec* user, chanrec* channel, const std::string &partmessage, bool &silent)
969 {
970         if (IS_LOCAL(user))
971         {
972                 std::deque<std::string> params;
973                 params.push_back(channel->name);
974                 if (partmessage != "")
975                         params.push_back(":"+partmessage);
976                 Utils->DoOneToMany(user->nick,"PART",params);
977         }
978 }
979
980 void ModuleSpanningTree::OnUserConnect(userrec* user)
981 {
982         char agestr[MAXBUF];
983         if (IS_LOCAL(user))
984         {
985                 std::deque<std::string> params;
986                 snprintf(agestr,MAXBUF,"%lu",(unsigned long)user->age);
987                 params.push_back(agestr);
988                 params.push_back(user->nick);
989                 params.push_back(user->host);
990                 params.push_back(user->dhost);
991                 params.push_back(user->ident);
992                 params.push_back("+"+std::string(user->FormatModes()));
993                 params.push_back(user->GetIPString());
994                 params.push_back(":"+std::string(user->fullname));
995                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"NICK",params);
996                 // User is Local, change needs to be reflected!
997                 TreeServer* SourceServer = Utils->FindServer(user->server);
998                 if (SourceServer)
999                 {
1000                         SourceServer->AddUserCount();
1001                 }
1002         }
1003 }
1004
1005 void ModuleSpanningTree::OnUserQuit(userrec* user, const std::string &reason, const std::string &oper_message)
1006 {
1007         if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
1008         {
1009                 std::deque<std::string> params;
1010
1011                 if (oper_message != reason)
1012                 {
1013                         params.push_back(":"+oper_message);
1014                         Utils->DoOneToMany(user->nick,"OPERQUIT",params);
1015                 }
1016                 params.clear();
1017                 params.push_back(":"+reason);
1018                 Utils->DoOneToMany(user->nick,"QUIT",params);
1019         }
1020         // Regardless, We need to modify the user Counts..
1021         TreeServer* SourceServer = Utils->FindServer(user->server);
1022         if (SourceServer)
1023         {
1024                 SourceServer->DelUserCount();
1025         }
1026 }
1027
1028 void ModuleSpanningTree::OnUserPostNick(userrec* user, const std::string &oldnick)
1029 {
1030         if (IS_LOCAL(user))
1031         {
1032                 std::deque<std::string> params;
1033                 params.push_back(user->nick);
1034                 Utils->DoOneToMany(oldnick,"NICK",params);
1035         }
1036 }
1037
1038 void ModuleSpanningTree::OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason, bool &silent)
1039 {
1040         if ((source) && (IS_LOCAL(source)))
1041         {
1042                 std::deque<std::string> params;
1043                 params.push_back(chan->name);
1044                 params.push_back(user->nick);
1045                 params.push_back(":"+reason);
1046                 Utils->DoOneToMany(source->nick,"KICK",params);
1047         }
1048         else if (!source)
1049         {
1050                 std::deque<std::string> params;
1051                 params.push_back(chan->name);
1052                 params.push_back(user->nick);
1053                 params.push_back(":"+reason);
1054                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"KICK",params);
1055         }
1056 }
1057
1058 void ModuleSpanningTree::OnRemoteKill(userrec* source, userrec* dest, const std::string &reason)
1059 {
1060         std::deque<std::string> params;
1061         params.push_back(dest->nick);
1062         params.push_back(":"+reason);
1063         Utils->DoOneToMany(source->nick,"KILL",params);
1064 }
1065
1066 void ModuleSpanningTree::OnRehash(userrec* user, const std::string &parameter)
1067 {
1068         if (parameter != "")
1069         {
1070                 std::deque<std::string> params;
1071                 params.push_back(parameter);
1072                 Utils->DoOneToMany(user ? user->nick : ServerInstance->Config->ServerName, "REHASH", params);
1073                 // check for self
1074                 if (ServerInstance->MatchText(ServerInstance->Config->ServerName,parameter))
1075                 {
1076                         ServerInstance->WriteOpers("*** Remote rehash initiated locally by \002%s\002", user ? user->nick : ServerInstance->Config->ServerName);
1077                         ServerInstance->RehashServer();
1078                 }
1079         }
1080         Utils->ReadConfiguration(false);
1081         InitializeDisabledCommands(ServerInstance->Config->DisabledCommands, ServerInstance);
1082 }
1083
1084 // note: the protocol does not allow direct umode +o except
1085 // via NICK with 8 params. sending OPERTYPE infers +o modechange
1086 // locally.
1087 void ModuleSpanningTree::OnOper(userrec* user, const std::string &opertype)
1088 {
1089         if (IS_LOCAL(user))
1090         {
1091                 std::deque<std::string> params;
1092                 params.push_back(opertype);
1093                 Utils->DoOneToMany(user->nick,"OPERTYPE",params);
1094         }
1095 }
1096
1097 void ModuleSpanningTree::OnLine(userrec* source, const std::string &host, bool adding, char linetype, long duration, const std::string &reason)
1098 {
1099         if (!source)
1100         {
1101                 /* Server-set lines */
1102                 char data[MAXBUF];
1103                 snprintf(data,MAXBUF,"%c %s %s %lu %lu :%s", linetype, host.c_str(), ServerInstance->Config->ServerName, (unsigned long)ServerInstance->Time(false),
1104                                 (unsigned long)duration, reason.c_str());
1105                 std::deque<std::string> params;
1106                 params.push_back(data);
1107                 Utils->DoOneToMany(ServerInstance->Config->ServerName, "ADDLINE", params);
1108         }
1109         else
1110         {
1111                 if (IS_LOCAL(source))
1112                 {
1113                         char type[8];
1114                         snprintf(type,8,"%cLINE",linetype);
1115                         std::string stype = type;
1116                         if (adding)
1117                         {
1118                                 char sduration[MAXBUF];
1119                                 snprintf(sduration,MAXBUF,"%ld",duration);
1120                                 std::deque<std::string> params;
1121                                 params.push_back(host);
1122                                 params.push_back(sduration);
1123                                 params.push_back(":"+reason);
1124                                 Utils->DoOneToMany(source->nick,stype,params);
1125                         }
1126                         else
1127                         {
1128                                 std::deque<std::string> params;
1129                                 params.push_back(host);
1130                                 Utils->DoOneToMany(source->nick,stype,params);
1131                         }
1132                 }
1133         }
1134 }
1135
1136 void ModuleSpanningTree::OnAddGLine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
1137 {
1138         OnLine(source,hostmask,true,'G',duration,reason);
1139 }
1140         
1141 void ModuleSpanningTree::OnAddZLine(long duration, userrec* source, const std::string &reason, const std::string &ipmask)
1142 {
1143         OnLine(source,ipmask,true,'Z',duration,reason);
1144 }
1145
1146 void ModuleSpanningTree::OnAddQLine(long duration, userrec* source, const std::string &reason, const std::string &nickmask)
1147 {
1148         OnLine(source,nickmask,true,'Q',duration,reason);
1149 }
1150
1151 void ModuleSpanningTree::OnAddELine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
1152 {
1153         OnLine(source,hostmask,true,'E',duration,reason);
1154 }
1155
1156 void ModuleSpanningTree::OnDelGLine(userrec* source, const std::string &hostmask)
1157 {
1158         OnLine(source,hostmask,false,'G',0,"");
1159 }
1160
1161 void ModuleSpanningTree::OnDelZLine(userrec* source, const std::string &ipmask)
1162 {
1163         OnLine(source,ipmask,false,'Z',0,"");
1164 }
1165
1166 void ModuleSpanningTree::OnDelQLine(userrec* source, const std::string &nickmask)
1167 {
1168         OnLine(source,nickmask,false,'Q',0,"");
1169 }
1170
1171 void ModuleSpanningTree::OnDelELine(userrec* source, const std::string &hostmask)
1172 {
1173         OnLine(source,hostmask,false,'E',0,"");
1174 }
1175
1176 void ModuleSpanningTree::OnMode(userrec* user, void* dest, int target_type, const std::string &text)
1177 {
1178         if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
1179         {
1180                 std::deque<std::string> params;
1181                 std::string command;
1182
1183                 if (target_type == TYPE_USER)
1184                 {
1185                         userrec* u = (userrec*)dest;
1186                         params.push_back(u->nick);
1187                         params.push_back(text);
1188                         command = "MODE";
1189                 }
1190                 else
1191                 {
1192                         chanrec* c = (chanrec*)dest;
1193                         params.push_back(c->name);
1194                         params.push_back(ConvToStr(c->age));
1195                         params.push_back(text);
1196                         command = "FMODE";
1197                 }
1198                 Utils->DoOneToMany(user->nick, command, params);
1199         }
1200 }
1201
1202 void ModuleSpanningTree::OnSetAway(userrec* user)
1203 {
1204         if (IS_LOCAL(user))
1205         {
1206                 std::deque<std::string> params;
1207                 params.push_back(":"+std::string(user->awaymsg));
1208                 Utils->DoOneToMany(user->nick,"AWAY",params);
1209         }
1210 }
1211
1212 void ModuleSpanningTree::OnCancelAway(userrec* user)
1213 {
1214         if (IS_LOCAL(user))
1215         {
1216                 std::deque<std::string> params;
1217                 params.clear();
1218                 Utils->DoOneToMany(user->nick,"AWAY",params);
1219         }
1220 }
1221
1222 void ModuleSpanningTree::ProtoSendMode(void* opaque, int target_type, void* target, const std::string &modeline)
1223 {
1224         TreeSocket* s = (TreeSocket*)opaque;
1225         if (target)
1226         {
1227                 if (target_type == TYPE_USER)
1228                 {
1229                         userrec* u = (userrec*)target;
1230                         s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" FMODE "+u->nick+" "+ConvToStr(u->age)+" "+modeline);
1231                 }
1232                 else
1233                 {
1234                         chanrec* c = (chanrec*)target;
1235                         s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" FMODE "+c->name+" "+ConvToStr(c->age)+" "+modeline);
1236                 }
1237         }
1238 }
1239
1240 void ModuleSpanningTree::ProtoSendMetaData(void* opaque, int target_type, void* target, const std::string &extname, const std::string &extdata)
1241 {
1242         TreeSocket* s = (TreeSocket*)opaque;
1243         if (target)
1244         {
1245                 if (target_type == TYPE_USER)
1246                 {
1247                         userrec* u = (userrec*)target;
1248                         s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA "+u->nick+" "+extname+" :"+extdata);
1249                 }
1250                 else if (target_type == TYPE_CHANNEL)
1251                 {
1252                         chanrec* c = (chanrec*)target;
1253                         s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA "+c->name+" "+extname+" :"+extdata);
1254                 }
1255         }
1256         if (target_type == TYPE_OTHER)
1257         {
1258                 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA * "+extname+" :"+extdata);
1259         }
1260 }
1261
1262 void ModuleSpanningTree::OnEvent(Event* event)
1263 {
1264         std::deque<std::string>* params = (std::deque<std::string>*)event->GetData();
1265         if (event->GetEventID() == "send_metadata")
1266         {
1267                 if (params->size() < 3)
1268                         return;
1269                 (*params)[2] = ":" + (*params)[2];
1270                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"METADATA",*params);
1271         }
1272         else if (event->GetEventID() == "send_topic")
1273         {
1274                 if (params->size() < 2)
1275                         return;
1276                 (*params)[1] = ":" + (*params)[1];
1277                 params->insert(params->begin() + 1,ServerInstance->Config->ServerName);
1278                 params->insert(params->begin() + 1,ConvToStr(ServerInstance->Time(true)));
1279                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"FTOPIC",*params);
1280         }
1281         else if (event->GetEventID() == "send_mode")
1282         {
1283                 if (params->size() < 2)
1284                         return;
1285                 // Insert the TS value of the object, either userrec or chanrec
1286                 time_t ourTS = 0;
1287                 userrec* a = ServerInstance->FindNick((*params)[0]);
1288                 if (a)
1289                 {
1290                         ourTS = a->age;
1291                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"MODE",*params);
1292                         return;
1293                 }
1294                 else
1295                 {
1296                         chanrec* a = ServerInstance->FindChan((*params)[0]);
1297                         if (a)
1298                         {
1299                                 ourTS = a->age;
1300                                 params->insert(params->begin() + 1,ConvToStr(ourTS));
1301                                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"FMODE",*params);
1302                         }
1303                 }
1304         }
1305         else if (event->GetEventID() == "send_mode_explicit")
1306         {
1307                 if (params->size() < 2)
1308                         return;
1309                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"MODE",*params);
1310         }
1311         else if (event->GetEventID() == "send_opers")
1312         {
1313                 if (params->size() < 1)
1314                         return;
1315                 (*params)[0] = ":" + (*params)[0];
1316                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"OPERNOTICE",*params);
1317         }
1318         else if (event->GetEventID() == "send_modeset")
1319         {
1320                 if (params->size() < 2)
1321                         return;
1322                 (*params)[1] = ":" + (*params)[1];
1323                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"MODENOTICE",*params);
1324         }
1325         else if (event->GetEventID() == "send_snoset")
1326         {
1327                 if (params->size() < 2)
1328                         return;
1329                 (*params)[1] = ":" + (*params)[1];
1330                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"SNONOTICE",*params);
1331         }
1332         else if (event->GetEventID() == "send_push")
1333         {
1334                 if (params->size() < 2)
1335                         return;
1336                         
1337                 userrec *a = ServerInstance->FindNick((*params)[0]);
1338                         
1339                 if (!a)
1340                         return;
1341                         
1342                 (*params)[1] = ":" + (*params)[1];
1343                 Utils->DoOneToOne(ServerInstance->Config->ServerName, "PUSH", *params, a->server);
1344         }
1345 }
1346
1347 ModuleSpanningTree::~ModuleSpanningTree()
1348 {
1349         /* This will also free the listeners */
1350         delete Utils;
1351         if (SyncTimer)
1352                 ServerInstance->Timers->DelTimer(SyncTimer);
1353
1354         ServerInstance->Timers->DelTimer(RefreshTimer);
1355
1356         ServerInstance->DoneWithInterface("InspSocketHook");
1357 }
1358
1359 Version ModuleSpanningTree::GetVersion()
1360 {
1361         return Version(1,1,0,2,VF_VENDOR,API_VERSION);
1362 }
1363
1364 void ModuleSpanningTree::Implements(char* List)
1365 {
1366         List[I_OnPreCommand] = List[I_OnGetServerDescription] = List[I_OnUserInvite] = List[I_OnPostLocalTopicChange] = 1;
1367         List[I_OnWallops] = List[I_OnUserNotice] = List[I_OnUserMessage] = List[I_OnBackgroundTimer] = 1;
1368         List[I_OnUserJoin] = List[I_OnChangeHost] = List[I_OnChangeName] = List[I_OnUserPart] = List[I_OnUserConnect] = 1;
1369         List[I_OnUserQuit] = List[I_OnUserPostNick] = List[I_OnUserKick] = List[I_OnRemoteKill] = List[I_OnRehash] = 1;
1370         List[I_OnOper] = List[I_OnAddGLine] = List[I_OnAddZLine] = List[I_OnAddQLine] = List[I_OnAddELine] = 1;
1371         List[I_OnDelGLine] = List[I_OnDelZLine] = List[I_OnDelQLine] = List[I_OnDelELine] = List[I_ProtoSendMode] = List[I_OnMode] = 1;
1372         List[I_OnStats] = List[I_ProtoSendMetaData] = List[I_OnEvent] = List[I_OnSetAway] = List[I_OnCancelAway] = List[I_OnPostCommand] = 1;
1373 }
1374
1375 /* It is IMPORTANT that m_spanningtree is the last module in the chain
1376  * so that any activity it sees is FINAL, e.g. we arent going to send out
1377  * a NICK message before m_cloaking has finished putting the +x on the user,
1378  * etc etc.
1379  * Therefore, we return PRIORITY_LAST to make sure we end up at the END of
1380  * the module call queue.
1381  */
1382 Priority ModuleSpanningTree::Prioritize()
1383 {
1384         return PRIORITY_LAST;
1385 }
1386
1387 class ModuleSpanningTreeFactory : public ModuleFactory
1388 {
1389  public:
1390         ModuleSpanningTreeFactory()
1391         {
1392         }
1393         
1394         ~ModuleSpanningTreeFactory()
1395         {
1396         }
1397         
1398         virtual Module * CreateModule(InspIRCd* Me)
1399         {
1400                 return new ModuleSpanningTree(Me);
1401         }
1402         
1403 };
1404
1405
1406 extern "C" DllExport void * init_module( void )
1407 {
1408         return new ModuleSpanningTreeFactory;
1409 }