]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/m_spanningtree/main.cpp
0fddd6e0ebbd64b39071b1b0255954a88f5664c7
[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                         std::string ip = Utils->Bindings[i]->IP;
665                         if (ip.empty())
666                                 ip = "*";
667
668                         std::string transport("plaintext");
669                         if (Utils->Bindings[i]->GetHook())
670                                 transport = InspSocketNameRequest(this, Utils->Bindings[i]->GetHook()).Send();
671
672                         results.push_back(ConvToStr(ServerInstance->Config->ServerName) + " 249 "+user->nick+" :" + ip + ":" + ConvToStr(Utils->Bindings[i]->port)+
673                                 " (server, " + transport + ")");
674                 }
675         }
676         return 0;
677 }
678
679 int ModuleSpanningTree::OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated, const std::string &original_line)
680 {
681         /* If the command doesnt appear to be valid, we dont want to mess with it. */
682         if (!validated)
683                 return 0;
684
685         if (command == "CONNECT")
686         {
687                 return this->HandleConnect(parameters,pcnt,user);
688         }
689         else if (command == "STATS")
690         {
691                 return this->HandleStats(parameters,pcnt,user);
692         }
693         else if (command == "MOTD")
694         {
695                 return this->HandleMotd(parameters,pcnt,user);
696         }
697         else if (command == "ADMIN")
698         {
699                 return this->HandleAdmin(parameters,pcnt,user);
700         }
701         else if (command == "SQUIT")
702         {
703                 return this->HandleSquit(parameters,pcnt,user);
704         }
705         else if (command == "MAP")
706         {
707                 this->HandleMap(parameters,pcnt,user);
708                 return 1;
709         }
710         else if ((command == "TIME") && (pcnt > 0))
711         {
712                 return this->HandleTime(parameters,pcnt,user);
713         }
714         else if (command == "LUSERS")
715         {
716                 this->HandleLusers(parameters,pcnt,user);
717                 return 1;
718         }
719         else if (command == "LINKS")
720         {
721                 this->HandleLinks(parameters,pcnt,user);
722                 return 1;
723         }
724         else if (command == "WHOIS")
725         {
726                 if (pcnt > 1)
727                 {
728                         // remote whois
729                         return this->HandleRemoteWhois(parameters,pcnt,user);
730                 }
731         }
732         else if ((command == "VERSION") && (pcnt > 0))
733         {
734                 this->HandleVersion(parameters,pcnt,user);
735                 return 1;
736         }
737         else if ((command == "MODULES") && (pcnt > 0))
738         {
739                 return this->HandleModules(parameters,pcnt,user);
740         }
741         return 0;
742 }
743
744 void ModuleSpanningTree::OnPostCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, CmdResult result, const std::string &original_line)
745 {
746         if ((result == CMD_SUCCESS) && (ServerInstance->IsValidModuleCommand(command, pcnt, user)))
747         {
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;
754                 params.clear();
755                 for (int j = 0; j < pcnt; j++)
756                 {
757                         if (strchr(parameters[j],' '))
758                         {
759                                 params.push_back(":" + std::string(parameters[j]));
760                         }
761                         else
762                         {
763                                 params.push_back(std::string(parameters[j]));
764                         }
765                 }
766                 Utils->DoOneToMany(user->nick,command,params);
767         }
768 }
769
770 void ModuleSpanningTree::OnGetServerDescription(const std::string &servername,std::string &description)
771 {
772         TreeServer* s = Utils->FindServer(servername);
773         if (s)
774         {
775                 description = s->GetDesc();
776         }
777 }
778
779 void ModuleSpanningTree::OnUserInvite(userrec* source,userrec* dest,chanrec* channel)
780 {
781         if (IS_LOCAL(source))
782         {
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);
787         }
788 }
789
790 void ModuleSpanningTree::OnPostLocalTopicChange(userrec* user, chanrec* chan, const std::string &topic)
791 {
792         std::deque<std::string> params;
793         params.push_back(chan->name);
794         params.push_back(":"+topic);
795         Utils->DoOneToMany(user->nick,"TOPIC",params);
796 }
797
798 void ModuleSpanningTree::OnWallops(userrec* user, const std::string &text)
799 {
800         if (IS_LOCAL(user))
801         {
802                 std::deque<std::string> params;
803                 params.push_back(":"+text);
804                 Utils->DoOneToMany(user->nick,"WALLOPS",params);
805         }
806 }
807
808 void ModuleSpanningTree::OnUserNotice(userrec* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
809 {
810         if (target_type == TYPE_USER)
811         {
812                 userrec* d = (userrec*)dest;
813                 if ((d->GetFd() < 0) && (IS_LOCAL(user)))
814                 {
815                         std::deque<std::string> params;
816                         params.clear();
817                         params.push_back(d->nick);
818                         params.push_back(":"+text);
819                         Utils->DoOneToOne(user->nick,"NOTICE",params,d->server);
820                 }
821         }
822         else if (target_type == TYPE_CHANNEL)
823         {
824                 if (IS_LOCAL(user))
825                 {
826                         chanrec *c = (chanrec*)dest;
827                         if (c)
828                         {
829                                 std::string cname = c->name;
830                                 if (status)
831                                         cname = status + cname;
832                                 TreeServerList list;
833                                 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
834                                 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
835                                 {
836                                         TreeSocket* Sock = i->second->GetSocket();
837                                         if (Sock)
838                                                 Sock->WriteLine(":"+std::string(user->nick)+" NOTICE "+cname+" :"+text);
839                                 }
840                         }
841                 }
842         }
843         else if (target_type == TYPE_SERVER)
844         {
845                 if (IS_LOCAL(user))
846                 {
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);
852                 }
853         }
854 }
855
856 void ModuleSpanningTree::OnUserMessage(userrec* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
857 {
858         if (target_type == TYPE_USER)
859         {
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)))
864                 {
865                         std::deque<std::string> params;
866                         params.clear();
867                         params.push_back(d->nick);
868                         params.push_back(":"+text);
869                         Utils->DoOneToOne(user->nick,"PRIVMSG",params,d->server);
870                 }
871         }
872         else if (target_type == TYPE_CHANNEL)
873         {
874                 if (IS_LOCAL(user))
875                 {
876                         chanrec *c = (chanrec*)dest;
877                         if (c)
878                         {
879                                 std::string cname = c->name;
880                                 if (status)
881                                         cname = status + cname;
882                                 TreeServerList list;
883                                 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
884                                 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
885                                 {
886                                         TreeSocket* Sock = i->second->GetSocket();
887                                         if (Sock)
888                                                 Sock->WriteLine(":"+std::string(user->nick)+" PRIVMSG "+cname+" :"+text);
889                                 }
890                         }
891                 }
892         }
893         else if (target_type == TYPE_SERVER)
894         {
895                 if (IS_LOCAL(user))
896                 {
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);
902                 }
903         }
904 }
905
906 void ModuleSpanningTree::OnBackgroundTimer(time_t curtime)
907 {
908         AutoConnectServers(curtime);
909         DoPingChecks(curtime);
910 }
911
912 void ModuleSpanningTree::OnUserJoin(userrec* user, chanrec* channel, bool &silent)
913 {
914         // Only do this for local users
915         if (IS_LOCAL(user))
916         {
917                 if (channel->GetUserCounter() == 1)
918                 {
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 */
928                         params.pop_back();
929                         params.push_back(channel->ChanModes(true));
930                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"FMODE",params);
931                 }
932                 else
933                 {
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);
938                 }
939         }
940 }
941
942 void ModuleSpanningTree::OnChangeHost(userrec* user, const std::string &newhost)
943 {
944         // only occurs for local clients
945         if (user->registered != REG_ALL)
946                 return;
947         std::deque<std::string> params;
948         params.push_back(newhost);
949         Utils->DoOneToMany(user->nick,"FHOST",params);
950 }
951
952 void ModuleSpanningTree::OnChangeName(userrec* user, const std::string &gecos)
953 {
954         // only occurs for local clients
955         if (user->registered != REG_ALL)
956                 return;
957         std::deque<std::string> params;
958         params.push_back(gecos);
959         Utils->DoOneToMany(user->nick,"FNAME",params);
960 }
961
962 void ModuleSpanningTree::OnUserPart(userrec* user, chanrec* channel, const std::string &partmessage, bool &silent)
963 {
964         if (IS_LOCAL(user))
965         {
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);
971         }
972 }
973
974 void ModuleSpanningTree::OnUserConnect(userrec* user)
975 {
976         char agestr[MAXBUF];
977         if (IS_LOCAL(user))
978         {
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);
992                 if (SourceServer)
993                 {
994                         SourceServer->AddUserCount();
995                 }
996         }
997 }
998
999 void ModuleSpanningTree::OnUserQuit(userrec* user, const std::string &reason, const std::string &oper_message)
1000 {
1001         if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
1002         {
1003                 std::deque<std::string> params;
1004
1005                 if (oper_message != reason)
1006                 {
1007                         params.push_back(":"+oper_message);
1008                         Utils->DoOneToMany(user->nick,"OPERQUIT",params);
1009                 }
1010                 params.clear();
1011                 params.push_back(":"+reason);
1012                 Utils->DoOneToMany(user->nick,"QUIT",params);
1013         }
1014         // Regardless, We need to modify the user Counts..
1015         TreeServer* SourceServer = Utils->FindServer(user->server);
1016         if (SourceServer)
1017         {
1018                 SourceServer->DelUserCount();
1019         }
1020 }
1021
1022 void ModuleSpanningTree::OnUserPostNick(userrec* user, const std::string &oldnick)
1023 {
1024         if (IS_LOCAL(user))
1025         {
1026                 std::deque<std::string> params;
1027                 params.push_back(user->nick);
1028                 Utils->DoOneToMany(oldnick,"NICK",params);
1029         }
1030 }
1031
1032 void ModuleSpanningTree::OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason, bool &silent)
1033 {
1034         if ((source) && (IS_LOCAL(source)))
1035         {
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);
1041         }
1042         else if (!source)
1043         {
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);
1049         }
1050 }
1051
1052 void ModuleSpanningTree::OnRemoteKill(userrec* source, userrec* dest, const std::string &reason)
1053 {
1054         std::deque<std::string> params;
1055         params.push_back(dest->nick);
1056         params.push_back(":"+reason);
1057         Utils->DoOneToMany(source->nick,"KILL",params);
1058 }
1059
1060 void ModuleSpanningTree::OnRehash(userrec* user, const std::string &parameter)
1061 {
1062         if (parameter != "")
1063         {
1064                 std::deque<std::string> params;
1065                 params.push_back(parameter);
1066                 Utils->DoOneToMany(user ? user->nick : ServerInstance->Config->ServerName, "REHASH", params);
1067                 // check for self
1068                 if (ServerInstance->MatchText(ServerInstance->Config->ServerName,parameter))
1069                 {
1070                         ServerInstance->WriteOpers("*** Remote rehash initiated locally by \002%s\002", user ? user->nick : ServerInstance->Config->ServerName);
1071                         ServerInstance->RehashServer();
1072                 }
1073         }
1074         Utils->ReadConfiguration(false);
1075         InitializeDisabledCommands(ServerInstance->Config->DisabledCommands, ServerInstance);
1076 }
1077
1078 // note: the protocol does not allow direct umode +o except
1079 // via NICK with 8 params. sending OPERTYPE infers +o modechange
1080 // locally.
1081 void ModuleSpanningTree::OnOper(userrec* user, const std::string &opertype)
1082 {
1083         if (IS_LOCAL(user))
1084         {
1085                 std::deque<std::string> params;
1086                 params.push_back(opertype);
1087                 Utils->DoOneToMany(user->nick,"OPERTYPE",params);
1088         }
1089 }
1090
1091 void ModuleSpanningTree::OnLine(userrec* source, const std::string &host, bool adding, char linetype, long duration, const std::string &reason)
1092 {
1093         if (!source)
1094         {
1095                 /* Server-set lines */
1096                 char data[MAXBUF];
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);
1102         }
1103         else
1104         {
1105                 if (IS_LOCAL(source))
1106                 {
1107                         char type[8];
1108                         snprintf(type,8,"%cLINE",linetype);
1109                         std::string stype = type;
1110                         if (adding)
1111                         {
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);
1119                         }
1120                         else
1121                         {
1122                                 std::deque<std::string> params;
1123                                 params.push_back(host);
1124                                 Utils->DoOneToMany(source->nick,stype,params);
1125                         }
1126                 }
1127         }
1128 }
1129
1130 void ModuleSpanningTree::OnAddGLine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
1131 {
1132         OnLine(source,hostmask,true,'G',duration,reason);
1133 }
1134         
1135 void ModuleSpanningTree::OnAddZLine(long duration, userrec* source, const std::string &reason, const std::string &ipmask)
1136 {
1137         OnLine(source,ipmask,true,'Z',duration,reason);
1138 }
1139
1140 void ModuleSpanningTree::OnAddQLine(long duration, userrec* source, const std::string &reason, const std::string &nickmask)
1141 {
1142         OnLine(source,nickmask,true,'Q',duration,reason);
1143 }
1144
1145 void ModuleSpanningTree::OnAddELine(long duration, userrec* source, const std::string &reason, const std::string &hostmask)
1146 {
1147         OnLine(source,hostmask,true,'E',duration,reason);
1148 }
1149
1150 void ModuleSpanningTree::OnDelGLine(userrec* source, const std::string &hostmask)
1151 {
1152         OnLine(source,hostmask,false,'G',0,"");
1153 }
1154
1155 void ModuleSpanningTree::OnDelZLine(userrec* source, const std::string &ipmask)
1156 {
1157         OnLine(source,ipmask,false,'Z',0,"");
1158 }
1159
1160 void ModuleSpanningTree::OnDelQLine(userrec* source, const std::string &nickmask)
1161 {
1162         OnLine(source,nickmask,false,'Q',0,"");
1163 }
1164
1165 void ModuleSpanningTree::OnDelELine(userrec* source, const std::string &hostmask)
1166 {
1167         OnLine(source,hostmask,false,'E',0,"");
1168 }
1169
1170 void ModuleSpanningTree::OnMode(userrec* user, void* dest, int target_type, const std::string &text)
1171 {
1172         if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
1173         {
1174                 std::deque<std::string> params;
1175                 std::string command;
1176
1177                 if (target_type == TYPE_USER)
1178                 {
1179                         userrec* u = (userrec*)dest;
1180                         params.push_back(u->nick);
1181                         params.push_back(text);
1182                         command = "MODE";
1183                 }
1184                 else
1185                 {
1186                         chanrec* c = (chanrec*)dest;
1187                         params.push_back(c->name);
1188                         params.push_back(ConvToStr(c->age));
1189                         params.push_back(text);
1190                         command = "FMODE";
1191                 }
1192                 Utils->DoOneToMany(user->nick, command, params);
1193         }
1194 }
1195
1196 void ModuleSpanningTree::OnSetAway(userrec* user)
1197 {
1198         if (IS_LOCAL(user))
1199         {
1200                 std::deque<std::string> params;
1201                 params.push_back(":"+std::string(user->awaymsg));
1202                 Utils->DoOneToMany(user->nick,"AWAY",params);
1203         }
1204 }
1205
1206 void ModuleSpanningTree::OnCancelAway(userrec* user)
1207 {
1208         if (IS_LOCAL(user))
1209         {
1210                 std::deque<std::string> params;
1211                 params.clear();
1212                 Utils->DoOneToMany(user->nick,"AWAY",params);
1213         }
1214 }
1215
1216 void ModuleSpanningTree::ProtoSendMode(void* opaque, int target_type, void* target, const std::string &modeline)
1217 {
1218         TreeSocket* s = (TreeSocket*)opaque;
1219         if (target)
1220         {
1221                 if (target_type == TYPE_USER)
1222                 {
1223                         userrec* u = (userrec*)target;
1224                         s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" FMODE "+u->nick+" "+ConvToStr(u->age)+" "+modeline);
1225                 }
1226                 else
1227                 {
1228                         chanrec* c = (chanrec*)target;
1229                         s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" FMODE "+c->name+" "+ConvToStr(c->age)+" "+modeline);
1230                 }
1231         }
1232 }
1233
1234 void ModuleSpanningTree::ProtoSendMetaData(void* opaque, int target_type, void* target, const std::string &extname, const std::string &extdata)
1235 {
1236         TreeSocket* s = (TreeSocket*)opaque;
1237         if (target)
1238         {
1239                 if (target_type == TYPE_USER)
1240                 {
1241                         userrec* u = (userrec*)target;
1242                         s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA "+u->nick+" "+extname+" :"+extdata);
1243                 }
1244                 else if (target_type == TYPE_CHANNEL)
1245                 {
1246                         chanrec* c = (chanrec*)target;
1247                         s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA "+c->name+" "+extname+" :"+extdata);
1248                 }
1249         }
1250         if (target_type == TYPE_OTHER)
1251         {
1252                 s->WriteLine(std::string(":")+ServerInstance->Config->ServerName+" METADATA * "+extname+" :"+extdata);
1253         }
1254 }
1255
1256 void ModuleSpanningTree::OnEvent(Event* event)
1257 {
1258         std::deque<std::string>* params = (std::deque<std::string>*)event->GetData();
1259         if (event->GetEventID() == "send_metadata")
1260         {
1261                 if (params->size() < 3)
1262                         return;
1263                 (*params)[2] = ":" + (*params)[2];
1264                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"METADATA",*params);
1265         }
1266         else if (event->GetEventID() == "send_topic")
1267         {
1268                 if (params->size() < 2)
1269                         return;
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);
1274         }
1275         else if (event->GetEventID() == "send_mode")
1276         {
1277                 if (params->size() < 2)
1278                         return;
1279                 // Insert the TS value of the object, either userrec or chanrec
1280                 time_t ourTS = 0;
1281                 userrec* a = ServerInstance->FindNick((*params)[0]);
1282                 if (a)
1283                 {
1284                         ourTS = a->age;
1285                         Utils->DoOneToMany(ServerInstance->Config->ServerName,"MODE",*params);
1286                         return;
1287                 }
1288                 else
1289                 {
1290                         chanrec* a = ServerInstance->FindChan((*params)[0]);
1291                         if (a)
1292                         {
1293                                 ourTS = a->age;
1294                                 params->insert(params->begin() + 1,ConvToStr(ourTS));
1295                                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"FMODE",*params);
1296                         }
1297                 }
1298         }
1299         else if (event->GetEventID() == "send_mode_explicit")
1300         {
1301                 if (params->size() < 2)
1302                         return;
1303                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"MODE",*params);
1304         }
1305         else if (event->GetEventID() == "send_opers")
1306         {
1307                 if (params->size() < 1)
1308                         return;
1309                 (*params)[0] = ":" + (*params)[0];
1310                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"OPERNOTICE",*params);
1311         }
1312         else if (event->GetEventID() == "send_modeset")
1313         {
1314                 if (params->size() < 2)
1315                         return;
1316                 (*params)[1] = ":" + (*params)[1];
1317                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"MODENOTICE",*params);
1318         }
1319         else if (event->GetEventID() == "send_snoset")
1320         {
1321                 if (params->size() < 2)
1322                         return;
1323                 (*params)[1] = ":" + (*params)[1];
1324                 Utils->DoOneToMany(ServerInstance->Config->ServerName,"SNONOTICE",*params);
1325         }
1326         else if (event->GetEventID() == "send_push")
1327         {
1328                 if (params->size() < 2)
1329                         return;
1330                         
1331                 userrec *a = ServerInstance->FindNick((*params)[0]);
1332                         
1333                 if (!a)
1334                         return;
1335                         
1336                 (*params)[1] = ":" + (*params)[1];
1337                 Utils->DoOneToOne(ServerInstance->Config->ServerName, "PUSH", *params, a->server);
1338         }
1339 }
1340
1341 ModuleSpanningTree::~ModuleSpanningTree()
1342 {
1343         /* This will also free the listeners */
1344         delete Utils;
1345         if (SyncTimer)
1346                 ServerInstance->Timers->DelTimer(SyncTimer);
1347
1348         ServerInstance->Timers->DelTimer(RefreshTimer);
1349
1350         ServerInstance->DoneWithInterface("InspSocketHook");
1351 }
1352
1353 Version ModuleSpanningTree::GetVersion()
1354 {
1355         return Version(1,1,0,2,VF_VENDOR,API_VERSION);
1356 }
1357
1358 void ModuleSpanningTree::Implements(char* List)
1359 {
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;
1367 }
1368
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,
1372  * etc etc.
1373  * Therefore, we return PRIORITY_LAST to make sure we end up at the END of
1374  * the module call queue.
1375  */
1376 Priority ModuleSpanningTree::Prioritize()
1377 {
1378         return PRIORITY_LAST;
1379 }
1380
1381 class ModuleSpanningTreeFactory : public ModuleFactory
1382 {
1383  public:
1384         ModuleSpanningTreeFactory()
1385         {
1386         }
1387         
1388         ~ModuleSpanningTreeFactory()
1389         {
1390         }
1391         
1392         virtual Module * CreateModule(InspIRCd* Me)
1393         {
1394                 return new ModuleSpanningTree(Me);
1395         }
1396         
1397 };
1398
1399
1400 extern "C" DllExport void * init_module( void )
1401 {
1402         return new ModuleSpanningTreeFactory;
1403 }