]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/m_spanningtree/main.cpp
0deaeb3ba247c464615927886b09976c495e8936
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / main.cpp
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  InspIRCd: (C) 2002-2008 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 "commands/cmd_whois.h"
18 #include "commands/cmd_stats.h"
19 #include "socket.h"
20 #include "wildcard.h"
21 #include "xline.h"
22 #include "transport.h"
23
24 #include "m_spanningtree/timesynctimer.h"
25 #include "m_spanningtree/resolvers.h"
26 #include "m_spanningtree/main.h"
27 #include "m_spanningtree/utils.h"
28 #include "m_spanningtree/treeserver.h"
29 #include "m_spanningtree/link.h"
30 #include "m_spanningtree/treesocket.h"
31 #include "m_spanningtree/rconnect.h"
32 #include "m_spanningtree/rsquit.h"
33
34 /* $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 m_spanningtree/rsquit.h */
35
36 ModuleSpanningTree::ModuleSpanningTree(InspIRCd* Me)
37         : Module(Me), max_local(0), max_global(0)
38 {
39         ServerInstance->Modules->UseInterface("BufferedSocketHook");
40         Utils = new SpanningTreeUtilities(Me, this);
41         command_rconnect = new cmd_rconnect(ServerInstance, this, Utils);
42         ServerInstance->AddCommand(command_rconnect);
43         command_rsquit = new cmd_rsquit(ServerInstance, this, Utils);
44         ServerInstance->AddCommand(command_rsquit);
45         if (Utils->EnableTimeSync)
46         {
47                 SyncTimer = new TimeSyncTimer(ServerInstance, this);
48                 ServerInstance->Timers->AddTimer(SyncTimer);
49         }
50         else
51                 SyncTimer = NULL;
52
53         RefreshTimer = new CacheRefreshTimer(ServerInstance, Utils);
54         ServerInstance->Timers->AddTimer(RefreshTimer);
55
56         Implementation eventlist[] =
57         {
58                 I_OnPreCommand, I_OnGetServerDescription, I_OnUserInvite, I_OnPostLocalTopicChange,
59                 I_OnWallops, I_OnUserNotice, I_OnUserMessage, I_OnBackgroundTimer,
60                 I_OnUserJoin, I_OnChangeHost, I_OnChangeName, I_OnUserPart, I_OnPostConnect,
61                 I_OnUserQuit, I_OnUserPostNick, I_OnUserKick, I_OnRemoteKill, I_OnRehash,
62                 I_OnOper, I_OnAddLine, I_OnDelLine, I_ProtoSendMode, I_OnMode,
63                 I_OnStats, I_ProtoSendMetaData, I_OnEvent, I_OnSetAway, I_OnCancelAway, I_OnPostCommand
64         };
65         ServerInstance->Modules->Attach(eventlist, this, 29);
66
67         for (std::vector<User*>::const_iterator i = ServerInstance->Users->local_users.begin(); i != ServerInstance->Users->local_users.end(); i++)
68         {
69                 this->OnPostConnect((*i));
70         }
71 }
72
73 void ModuleSpanningTree::ShowLinks(TreeServer* Current, User* user, int hops)
74 {
75         std::string Parent = Utils->TreeRoot->GetName();
76         if (Current->GetParent())
77         {
78                 Parent = Current->GetParent()->GetName();
79         }
80         for (unsigned int q = 0; q < Current->ChildCount(); q++)
81         {
82                 if ((Current->GetChild(q)->Hidden) || ((Utils->HideULines) && (ServerInstance->ULine(Current->GetChild(q)->GetName().c_str()))))
83                 {
84                         if (*user->oper)
85                         {
86                                  ShowLinks(Current->GetChild(q),user,hops+1);
87                         }
88                 }
89                 else
90                 {
91                         ShowLinks(Current->GetChild(q),user,hops+1);
92                 }
93         }
94         /* Don't display the line if its a uline, hide ulines is on, and the user isnt an oper */
95         if ((Utils->HideULines) && (ServerInstance->ULine(Current->GetName().c_str())) && (!IS_OPER(user)))
96                 return;
97         /* Or if the server is hidden and they're not an oper */
98         else if ((Current->Hidden) && (!IS_OPER(user)))
99                 return;
100
101         user->WriteServ("364 %s %s %s :%d %s",  user->nick,Current->GetName().c_str(),
102                         (Utils->FlatLinks && (!IS_OPER(user))) ? ServerInstance->Config->ServerName : Parent.c_str(),
103                         (Utils->FlatLinks && (!IS_OPER(user))) ? 0 : hops,
104                         Current->GetDesc().c_str());
105 }
106
107 int ModuleSpanningTree::CountLocalServs()
108 {
109         return Utils->TreeRoot->ChildCount();
110 }
111
112 int ModuleSpanningTree::CountServs()
113 {
114         return Utils->serverlist.size();
115 }
116
117 void ModuleSpanningTree::HandleLinks(const char** parameters, int pcnt, User* user)
118 {
119         ShowLinks(Utils->TreeRoot,user,0);
120         user->WriteServ("365 %s * :End of /LINKS list.",user->nick);
121         return;
122 }
123
124 void ModuleSpanningTree::HandleLusers(const char** parameters, int pcnt, User* user)
125 {
126         unsigned int n_users = ServerInstance->Users->UserCount();
127
128         /* Only update these when someone wants to see them, more efficient */
129         if ((unsigned int)ServerInstance->Users->LocalUserCount() > max_local)
130                 max_local = ServerInstance->Users->LocalUserCount();
131         if (n_users > max_global)
132                 max_global = n_users;
133
134         unsigned int ulined_count = 0;
135         unsigned int ulined_local_count = 0;
136
137         /* If ulined are hidden and we're not an oper, count the number of ulined servers hidden,
138          * locally and globally (locally means directly connected to us)
139          */
140         if ((Utils->HideULines) && (!*user->oper))
141         {
142                 for (server_hash::iterator q = Utils->serverlist.begin(); q != Utils->serverlist.end(); q++)
143                 {
144                         if (ServerInstance->ULine(q->second->GetName().c_str()))
145                         {
146                                 ulined_count++;
147                                 if (q->second->GetParent() == Utils->TreeRoot)
148                                         ulined_local_count++;
149                         }
150                 }
151         }
152         user->WriteServ("251 %s :There are %d users and %d invisible on %d servers",user->nick,
153                         n_users-ServerInstance->Users->ModeCount('i'),
154                         ServerInstance->Users->ModeCount('i'),
155                         ulined_count ? this->CountServs() - ulined_count : this->CountServs());
156
157         if (ServerInstance->Users->OperCount())
158                 user->WriteServ("252 %s %d :operator(s) online",user->nick,ServerInstance->Users->OperCount());
159
160         if (ServerInstance->Users->UnregisteredUserCount())
161                 user->WriteServ("253 %s %d :unknown connections",user->nick,ServerInstance->Users->UnregisteredUserCount());
162         
163         if (ServerInstance->ChannelCount())
164                 user->WriteServ("254 %s %d :channels formed",user->nick,ServerInstance->ChannelCount());
165         
166         user->WriteServ("255 %s :I have %d clients and %d servers",user->nick,ServerInstance->Users->LocalUserCount(),ulined_local_count ? this->CountLocalServs() - ulined_local_count : this->CountLocalServs());
167         user->WriteServ("265 %s :Current Local Users: %d  Max: %d",user->nick,ServerInstance->Users->LocalUserCount(),max_local);
168         user->WriteServ("266 %s :Current Global Users: %d  Max: %d",user->nick,n_users,max_global);
169         return;
170 }
171
172 std::string ModuleSpanningTree::TimeToStr(time_t secs)
173 {
174         time_t mins_up = secs / 60;
175         time_t hours_up = mins_up / 60;
176         time_t days_up = hours_up / 24;
177         secs = secs % 60;
178         mins_up = mins_up % 60;
179         hours_up = hours_up % 24;
180         return ((days_up ? (ConvToStr(days_up) + "d") : std::string(""))
181                         + (hours_up ? (ConvToStr(hours_up) + "h") : std::string(""))
182                         + (mins_up ? (ConvToStr(mins_up) + "m") : std::string(""))
183                         + ConvToStr(secs) + "s");
184 }
185
186 void ModuleSpanningTree::DoPingChecks(time_t curtime)
187 {
188         for (unsigned int j = 0; j < Utils->TreeRoot->ChildCount(); j++)
189         {
190                 TreeServer* serv = Utils->TreeRoot->GetChild(j);
191                 TreeSocket* sock = serv->GetSocket();
192                 if (sock)
193                 {
194                         if (curtime >= serv->NextPingTime())
195                         {
196                                 if (serv->AnsweredLastPing())
197                                 {
198                                         sock->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" PING "+serv->GetID());
199                                         serv->SetNextPingTime(curtime + Utils->PingFreq);
200                                         timeval t;
201                                         gettimeofday(&t, NULL);
202                                         long ts = (t.tv_sec * 1000) + (t.tv_usec / 1000);
203                                         serv->LastPingMsec = ts;
204                                         serv->Warned = false;
205                                 }
206                                 else
207                                 {
208                                         /* they didnt answer, boot them */
209                                         sock->SendError("Ping timeout");
210                                         sock->Squit(serv,"Ping timeout");
211                                         ServerInstance->SE->DelFd(sock);
212                                         sock->Close();
213                                         return;
214                                 }
215                         }
216                         else if ((Utils->PingWarnTime) && (!serv->Warned) && (curtime >= serv->NextPingTime() - (Utils->PingFreq - Utils->PingWarnTime)) && (!serv->AnsweredLastPing()))
217                         {
218                                 /* The server hasnt responded, send a warning to opers */
219                                 ServerInstance->SNO->WriteToSnoMask('l',"Server \002%s\002 has not responded to PING for %d seconds, high latency.", serv->GetName().c_str(), Utils->PingWarnTime);
220                                 serv->Warned = true;
221                         }
222                 }
223         }
224
225         /*
226          * Cancel remote burst mode on any servers which still have it enabled due to latency/lack of data.
227          * This prevents lost REMOTECONNECT notices
228          * XXX this should probably not do this until server has been bursting for, say, 60 seconds or something
229          */
230         for (server_hash::iterator i = Utils->serverlist.begin(); i != Utils->serverlist.end(); i++)
231         {
232                 if (i->second->bursting)
233                 {
234                         ServerInstance->SNO->WriteToSnoMask('l',"Server \002%s\002 has not finished burst, forcing end of burst.", i->second->GetName().c_str());
235                         i->second->FinishBurst();
236                 }
237         }
238 }
239
240 void ModuleSpanningTree::ConnectServer(Link* x)
241 {
242         bool ipvalid = true;
243         QueryType start_type = DNS_QUERY_A;
244 #ifdef IPV6
245         start_type = DNS_QUERY_AAAA;
246         if (strchr(x->IPAddr.c_str(),':'))
247         {
248                 in6_addr n;
249                 if (inet_pton(AF_INET6, x->IPAddr.c_str(), &n) < 1)
250                         ipvalid = false;
251         }
252         else
253 #endif
254         {
255                 in_addr n;
256                 if (inet_aton(x->IPAddr.c_str(),&n) < 1)
257                         ipvalid = false;
258         }
259
260         /* Do we already have an IP? If so, no need to resolve it. */
261         if (ipvalid)
262         {
263                 /* Gave a hook, but it wasnt one we know */
264                 if ((!x->Hook.empty()) && (Utils->hooks.find(x->Hook.c_str()) == Utils->hooks.end()))
265                         return;
266                 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()]);
267                 if (newsocket->GetFd() > -1)
268                 {
269                         /* Handled automatically on success */
270                 }
271                 else
272                 {
273                         RemoteMessage(NULL, "CONNECT: Error connecting \002%s\002: %s.",x->Name.c_str(),strerror(errno));
274                         if (ServerInstance->SocketCull.find(newsocket) == ServerInstance->SocketCull.end())
275                                 ServerInstance->SocketCull[newsocket] = newsocket;
276                         Utils->DoFailOver(x);
277                 }
278         }
279         else
280         {
281                 try
282                 {
283                         bool cached;
284                         ServernameResolver* snr = new ServernameResolver((Module*)this, Utils, ServerInstance,x->IPAddr, *x, cached, start_type);
285                         ServerInstance->AddResolver(snr, cached);
286                 }
287                 catch (ModuleException& e)
288                 {
289                         RemoteMessage(NULL, "CONNECT: Error connecting \002%s\002: %s.",x->Name.c_str(), e.GetReason());
290                         Utils->DoFailOver(x);
291                 }
292         }
293 }
294
295 void ModuleSpanningTree::AutoConnectServers(time_t curtime)
296 {
297         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
298         {
299                 if ((x->AutoConnect) && (curtime >= x->NextConnectTime))
300                 {
301                         x->NextConnectTime = curtime + x->AutoConnect;
302                         TreeServer* CheckDupe = Utils->FindServer(x->Name.c_str());
303                         if (x->FailOver.length())
304                         {
305                                 TreeServer* CheckFailOver = Utils->FindServer(x->FailOver.c_str());
306                                 if (CheckFailOver)
307                                 {
308                                         /* The failover for this server is currently a member of the network.
309                                          * The failover probably succeeded, where the main link did not.
310                                          * Don't try the main link until the failover is gone again.
311                                          */
312                                         continue;
313                                 }
314                         }
315                         if (!CheckDupe)
316                         {
317                                 // an autoconnected server is not connected. Check if its time to connect it
318                                 ServerInstance->SNO->WriteToSnoMask('l',"AUTOCONNECT: Auto-connecting server \002%s\002 (%lu seconds until next attempt)",x->Name.c_str(),x->AutoConnect);
319                                 this->ConnectServer(&(*x));
320                         }
321                 }
322         }
323 }
324
325 int ModuleSpanningTree::HandleVersion(const char** parameters, int pcnt, User* user)
326 {
327         // we've already checked if pcnt > 0, so this is safe
328         TreeServer* found = Utils->FindServerMask(parameters[0]);
329         if (found)
330         {
331                 std::string Version = found->GetVersion();
332                 user->WriteServ("351 %s :%s",user->nick,Version.c_str());
333                 if (found == Utils->TreeRoot)
334                 {
335                         ServerInstance->Config->Send005(user);
336                 }
337         }
338         else
339         {
340                 user->WriteServ("402 %s %s :No such server",user->nick,parameters[0]);
341         }
342         return 1;
343 }
344
345 /* This method will attempt to get a link message out to as many people as is required.
346  * If a user is provided, and that user is local, then the user is sent the message using
347  * WriteServ (they are the local initiator of that message). If the user is remote, they are
348  * sent that message remotely via PUSH.
349  * If the user is NULL, then the notice is sent locally via WriteToSnoMask with snomask 'l',
350  * and remotely via SNONOTICE with mask 'l'.
351  */
352 void ModuleSpanningTree::RemoteMessage(User* user, const char* format, ...)
353 {
354         /* This could cause an infinite loop, because DoOneToMany() will, on error,
355          * call TreeSocket::OnError(), which in turn will call this function to
356          * notify everyone of the error. So, drop any messages that are generated
357          * during the sending of another message. -Special */
358         static bool SendingRemoteMessage = false;
359         if (SendingRemoteMessage)
360                 return;
361         SendingRemoteMessage = true;
362
363         std::deque<std::string> params;
364         char text[MAXBUF];
365         va_list argsPtr;
366
367         va_start(argsPtr, format);
368         vsnprintf(text, MAXBUF, format, argsPtr);
369         va_end(argsPtr);
370
371         if (!user)
372         {
373                 /* No user, target it generically at everyone */
374                 ServerInstance->SNO->WriteToSnoMask('l', "%s", text);
375                 params.push_back("l");
376                 params.push_back(std::string(":") + text);
377                 Utils->DoOneToMany(ServerInstance->Config->GetSID(), "SNONOTICE", params);
378         }
379         else
380         {
381                 if (IS_LOCAL(user))
382                         user->WriteServ("NOTICE %s :%s", user->nick, text);
383                 else
384                 {
385                         params.push_back(user->uuid);
386                         params.push_back(std::string("::") + ServerInstance->Config->ServerName + " NOTICE " + user->nick + " :*** From " +
387                                         ServerInstance->Config->ServerName+ ": " + text);
388                         Utils->DoOneToMany(ServerInstance->Config->GetSID(), "PUSH", params);
389                 }
390         }
391         
392         SendingRemoteMessage = false;
393 }
394         
395 int ModuleSpanningTree::HandleConnect(const char** parameters, int pcnt, User* user)
396 {
397         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
398         {
399                 if (ServerInstance->MatchText(x->Name.c_str(),parameters[0]))
400                 {
401                         TreeServer* CheckDupe = Utils->FindServer(x->Name.c_str());
402                         if (!CheckDupe)
403                         {
404                                 RemoteMessage(user, "*** CONNECT: Connecting to server: \002%s\002 (%s:%d)",x->Name.c_str(),(x->HiddenFromStats ? "<hidden>" : x->IPAddr.c_str()),x->Port);
405                                 ConnectServer(&(*x));
406                                 return 1;
407                         }
408                         else
409                         {
410                                 RemoteMessage(user, "*** CONNECT: Server \002%s\002 already exists on the network and is connected via \002%s\002",x->Name.c_str(),CheckDupe->GetParent()->GetName().c_str());
411                                 return 1;
412                         }
413                 }
414         }
415         RemoteMessage(user, "*** CONNECT: No server matching \002%s\002 could be found in the config file.",parameters[0]);
416         return 1;
417 }
418
419 void ModuleSpanningTree::BroadcastTimeSync()
420 {
421         if (Utils->MasterTime)
422         {
423                 std::deque<std::string> params;
424                 params.push_back(ConvToStr(ServerInstance->Time(false)));
425                 params.push_back("FORCE");
426                 Utils->DoOneToMany(ServerInstance->Config->GetSID(), "TIMESET", params);
427         }
428 }
429
430 void ModuleSpanningTree::OnGetServerDescription(const std::string &servername,std::string &description)
431 {
432         TreeServer* s = Utils->FindServer(servername);
433         if (s)
434         {
435                 description = s->GetDesc();
436         }
437 }
438
439 void ModuleSpanningTree::OnUserInvite(User* source,User* dest,Channel* channel)
440 {
441         if (IS_LOCAL(source))
442         {
443                 std::deque<std::string> params;
444                 params.push_back(dest->uuid);
445                 params.push_back(channel->name);
446                 Utils->DoOneToMany(source->uuid,"INVITE",params);
447         }
448 }
449
450 void ModuleSpanningTree::OnPostLocalTopicChange(User* user, Channel* chan, const std::string &topic)
451 {
452         std::deque<std::string> params;
453         params.push_back(chan->name);
454         params.push_back(":"+topic);
455         Utils->DoOneToMany(user->uuid,"TOPIC",params);
456 }
457
458 void ModuleSpanningTree::OnWallops(User* user, const std::string &text)
459 {
460         if (IS_LOCAL(user))
461         {
462                 std::deque<std::string> params;
463                 params.push_back(":"+text);
464                 Utils->DoOneToMany(user->uuid,"WALLOPS",params);
465         }
466 }
467
468 void ModuleSpanningTree::OnUserNotice(User* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
469 {
470         if (target_type == TYPE_USER)
471         {
472                 User* d = (User*)dest;
473                 if ((d->GetFd() < 0) && (IS_LOCAL(user)))
474                 {
475                         std::deque<std::string> params;
476                         params.clear();
477                         params.push_back(d->uuid);
478                         params.push_back(":"+text);
479                         Utils->DoOneToOne(user->uuid,"NOTICE",params,d->server);
480                 }
481         }
482         else if (target_type == TYPE_CHANNEL)
483         {
484                 if (IS_LOCAL(user))
485                 {
486                         Channel *c = (Channel*)dest;
487                         if (c)
488                         {
489                                 std::string cname = c->name;
490                                 if (status)
491                                         cname = status + cname;
492                                 TreeServerList list;
493                                 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
494                                 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
495                                 {
496                                         TreeSocket* Sock = i->second->GetSocket();
497                                         if (Sock)
498                                                 Sock->WriteLine(":"+std::string(user->uuid)+" NOTICE "+cname+" :"+text);
499                                 }
500                         }
501                 }
502         }
503         else if (target_type == TYPE_SERVER)
504         {
505                 if (IS_LOCAL(user))
506                 {
507                         char* target = (char*)dest;
508                         std::deque<std::string> par;
509                         par.push_back(target);
510                         par.push_back(":"+text);
511                         Utils->DoOneToMany(user->uuid,"NOTICE",par);
512                 }
513         }
514 }
515
516 void ModuleSpanningTree::OnUserMessage(User* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
517 {
518         if (target_type == TYPE_USER)
519         {
520                 // route private messages which are targetted at clients only to the server
521                 // which needs to receive them
522                 User* d = (User*)dest;
523                 if ((d->GetFd() < 0) && (IS_LOCAL(user)))
524                 {
525                         std::deque<std::string> params;
526                         params.clear();
527                         params.push_back(d->uuid);
528                         params.push_back(":"+text);
529                         Utils->DoOneToOne(user->uuid,"PRIVMSG",params,d->server);
530                 }
531         }
532         else if (target_type == TYPE_CHANNEL)
533         {
534                 if (IS_LOCAL(user))
535                 {
536                         Channel *c = (Channel*)dest;
537                         if (c)
538                         {
539                                 std::string cname = c->name;
540                                 if (status)
541                                         cname = status + cname;
542                                 TreeServerList list;
543                                 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
544                                 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
545                                 {
546                                         TreeSocket* Sock = i->second->GetSocket();
547                                         if (Sock)
548                                                 Sock->WriteLine(":"+std::string(user->uuid)+" PRIVMSG "+cname+" :"+text);
549                                 }
550                         }
551                 }
552         }
553         else if (target_type == TYPE_SERVER)
554         {
555                 if (IS_LOCAL(user))
556                 {
557                         char* target = (char*)dest;
558                         std::deque<std::string> par;
559                         par.push_back(target);
560                         par.push_back(":"+text);
561                         Utils->DoOneToMany(user->uuid,"PRIVMSG",par);
562                 }
563         }
564 }
565
566 void ModuleSpanningTree::OnBackgroundTimer(time_t curtime)
567 {
568         AutoConnectServers(curtime);
569         DoPingChecks(curtime);
570 }
571
572 void ModuleSpanningTree::OnUserJoin(User* user, Channel* channel, bool sync, bool &silent)
573 {
574         // Only do this for local users
575         if (IS_LOCAL(user))
576         {
577                 if (channel->GetUserCounter() == 1)
578                 {
579                         std::deque<std::string> params;
580                         // set up their permissions and the channel TS with FJOIN.
581                         // All users are FJOINed now, because a module may specify
582                         // new joining permissions for the user.
583                         params.push_back(channel->name);
584                         params.push_back(ConvToStr(channel->age));
585                         params.push_back(std::string(channel->GetAllPrefixChars(user))+","+std::string(user->uuid));
586                         Utils->DoOneToMany(ServerInstance->Config->GetSID(),"FJOIN",params);
587                         /* First user in, sync the modes for the channel */
588                         params.pop_back();
589                         params.push_back(std::string("+") + channel->ChanModes(true));
590                         Utils->DoOneToMany(ServerInstance->Config->GetSID(),"FMODE",params);
591                 }
592                 else
593                 {
594                         std::deque<std::string> params;
595                         params.push_back(channel->name);
596                         params.push_back(ConvToStr(channel->age));
597                         Utils->DoOneToMany(user->uuid,"JOIN",params);
598                 }
599         }
600 }
601
602 void ModuleSpanningTree::OnChangeHost(User* user, const std::string &newhost)
603 {
604         // only occurs for local clients
605         if (user->registered != REG_ALL)
606                 return;
607         std::deque<std::string> params;
608         params.push_back(newhost);
609         Utils->DoOneToMany(user->uuid,"FHOST",params);
610 }
611
612 void ModuleSpanningTree::OnChangeName(User* user, const std::string &gecos)
613 {
614         // only occurs for local clients
615         if (user->registered != REG_ALL)
616                 return;
617         std::deque<std::string> params;
618         params.push_back(gecos);
619         Utils->DoOneToMany(user->uuid,"FNAME",params);
620 }
621
622 void ModuleSpanningTree::OnUserPart(User* user, Channel* channel, const std::string &partmessage, bool &silent)
623 {
624         if (IS_LOCAL(user))
625         {
626                 std::deque<std::string> params;
627                 params.push_back(channel->name);
628                 if (!partmessage.empty())
629                         params.push_back(":"+partmessage);
630                 Utils->DoOneToMany(user->uuid,"PART",params);
631         }
632 }
633
634 void ModuleSpanningTree::OnPostConnect(User* user)
635 {
636         if (IS_LOCAL(user))
637         {
638                 std::deque<std::string> params;
639                 params.push_back(user->uuid);
640                 params.push_back(ConvToStr(user->age));
641                 params.push_back(user->nick);
642                 params.push_back(user->host);
643                 params.push_back(user->dhost);
644                 params.push_back(user->ident);
645                 params.push_back("+"+std::string(user->FormatModes()));
646                 params.push_back(user->GetIPString());
647                 params.push_back(ConvToStr(user->signon));
648                 params.push_back(":"+std::string(user->fullname));
649                 Utils->DoOneToMany(ServerInstance->Config->GetSID(), "UID", params);
650         }
651
652         TreeServer* SourceServer = Utils->FindServer(user->server);
653         if (SourceServer)
654         {
655                 SourceServer->SetUserCount(1); // increment by 1
656         }
657 }
658
659 void ModuleSpanningTree::OnUserQuit(User* user, const std::string &reason, const std::string &oper_message)
660 {
661         if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
662         {
663                 std::deque<std::string> params;
664
665                 if (oper_message != reason)
666                 {
667                         params.push_back(":"+oper_message);
668                         Utils->DoOneToMany(user->uuid,"OPERQUIT",params);
669                 }
670                 params.clear();
671                 params.push_back(":"+reason);
672                 Utils->DoOneToMany(user->uuid,"QUIT",params);
673         }
674
675         // Regardless, We need to modify the user Counts..
676         TreeServer* SourceServer = Utils->FindServer(user->server);
677         if (SourceServer)
678         {
679                 SourceServer->SetUserCount(-1); // decrement by 1
680         }
681 }
682
683 void ModuleSpanningTree::OnUserPostNick(User* user, const std::string &oldnick)
684 {
685         if (IS_LOCAL(user))
686         {
687                 std::deque<std::string> params;
688                 params.push_back(user->nick);
689
690                 /** IMPORTANT: We don't update the TS if the oldnick is just a case change of the newnick!
691                  */
692                 if (irc::string(user->nick) != assign(oldnick))
693                         user->age = ServerInstance->Time(true);
694
695                 params.push_back(ConvToStr(user->age));
696                 Utils->DoOneToMany(user->uuid,"NICK",params);
697         }
698 }
699
700 void ModuleSpanningTree::OnUserKick(User* source, User* user, Channel* chan, const std::string &reason, bool &silent)
701 {
702         if ((source) && (IS_LOCAL(source)))
703         {
704                 std::deque<std::string> params;
705                 params.push_back(chan->name);
706                 params.push_back(user->uuid);
707                 params.push_back(":"+reason);
708                 Utils->DoOneToMany(source->uuid,"KICK",params);
709         }
710         else if (!source)
711         {
712                 std::deque<std::string> params;
713                 params.push_back(chan->name);
714                 params.push_back(user->uuid);
715                 params.push_back(":"+reason);
716                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"KICK",params);
717         }
718 }
719
720 void ModuleSpanningTree::OnRemoteKill(User* source, User* dest, const std::string &reason, const std::string &operreason)
721 {
722         std::deque<std::string> params;
723         params.push_back(":"+reason);
724         Utils->DoOneToMany(dest->uuid,"OPERQUIT",params);
725         params.clear();
726         params.push_back(dest->uuid);
727         params.push_back(":"+reason);
728         dest->SetOperQuit(operreason);
729         Utils->DoOneToMany(source->uuid,"KILL",params);
730 }
731
732 void ModuleSpanningTree::OnRehash(User* user, const std::string &parameter)
733 {
734         if (!parameter.empty())
735         {
736                 std::deque<std::string> params;
737                 params.push_back(parameter);
738                 Utils->DoOneToMany(user ? user->nick : ServerInstance->Config->GetSID(), "REHASH", params);
739                 // check for self
740                 if (ServerInstance->MatchText(ServerInstance->Config->ServerName,parameter))
741                 {
742                         ServerInstance->SNO->WriteToSnoMask('A', "Remote rehash initiated locally by \002%s\002", user ? user->nick : ServerInstance->Config->ServerName);
743                         ServerInstance->RehashServer();
744                 }
745         }
746         Utils->ReadConfiguration(true);
747         InitializeDisabledCommands(ServerInstance->Config->DisabledCommands, ServerInstance);
748 }
749
750 // note: the protocol does not allow direct umode +o except
751 // via NICK with 8 params. sending OPERTYPE infers +o modechange
752 // locally.
753 void ModuleSpanningTree::OnOper(User* user, const std::string &opertype)
754 {
755         if (IS_LOCAL(user))
756         {
757                 std::deque<std::string> params;
758                 params.push_back(opertype);
759                 Utils->DoOneToMany(user->uuid,"OPERTYPE",params);
760         }
761 }
762
763 void ModuleSpanningTree::OnAddLine(XLine* line, User* user)
764 {
765         if (line->type == "K")
766                 return;
767
768         char data[MAXBUF];
769         snprintf(data,MAXBUF,"%s %s %s %lu %lu :%s", line->type.c_str(), line->Displayable(), 
770         ServerInstance->Config->ServerName, (unsigned long)line->set_time, (unsigned long)line->duration, line->reason);
771         std::deque<std::string> params;
772         params.push_back(data);
773
774         if (!user)
775         {
776                 /* Server-set lines */
777                 Utils->DoOneToMany(ServerInstance->Config->GetSID(), "ADDLINE", params);
778         }
779         else if (IS_LOCAL(user))
780         {
781                 /* User-set lines */
782                 Utils->DoOneToMany(user->uuid, "ADDLINE", params);
783         }
784 }
785
786 void ModuleSpanningTree::OnDelLine(XLine* line, User* user)
787 {
788         if (line->type == "K")
789                 return;
790
791         char data[MAXBUF];
792         snprintf(data,MAXBUF,"%s %s", line->type.c_str(), line->Displayable());
793         std::deque<std::string> params;
794         params.push_back(data);
795
796         if (!user)
797         {
798                 /* Server-unset lines */
799                 Utils->DoOneToMany(ServerInstance->Config->GetSID(), "DELLINE", params);
800         }
801         else if (IS_LOCAL(user))
802         {
803                 /* User-unset lines */
804                 Utils->DoOneToMany(user->uuid, "DELLINE", params);
805         }
806 }
807
808 void ModuleSpanningTree::OnMode(User* user, void* dest, int target_type, const std::string &text)
809 {
810         if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
811         {
812                 std::deque<std::string> params;
813                 std::string command;
814                 std::string output_text;
815
816                 ServerInstance->Parser->TranslateUIDs(TR_SPACENICKLIST, text, output_text);
817
818                 if (target_type == TYPE_USER)
819                 {
820                         User* u = (User*)dest;
821                         params.push_back(u->uuid);
822                         params.push_back(output_text);
823                         command = "MODE";
824                 }
825                 else
826                 {
827                         Channel* c = (Channel*)dest;
828                         params.push_back(c->name);
829                         params.push_back(ConvToStr(c->age));
830                         params.push_back(output_text);
831                         command = "FMODE";
832                 }
833
834                 Utils->DoOneToMany(user->uuid, command, params);
835         }
836 }
837
838 void ModuleSpanningTree::OnSetAway(User* user)
839 {
840         if (IS_LOCAL(user))
841         {
842                 std::deque<std::string> params;
843                 params.push_back(":"+std::string(user->awaymsg));
844                 Utils->DoOneToMany(user->uuid,"AWAY",params);
845         }
846 }
847
848 void ModuleSpanningTree::OnCancelAway(User* user)
849 {
850         if (IS_LOCAL(user))
851         {
852                 std::deque<std::string> params;
853                 params.clear();
854                 Utils->DoOneToMany(user->uuid,"AWAY",params);
855         }
856 }
857
858 void ModuleSpanningTree::ProtoSendMode(void* opaque, int target_type, void* target, const std::string &modeline)
859 {
860         TreeSocket* s = (TreeSocket*)opaque;
861         std::string output_text;
862
863         ServerInstance->Parser->TranslateUIDs(TR_SPACENICKLIST, modeline, output_text);
864
865         if (target)
866         {
867                 if (target_type == TYPE_USER)
868                 {
869                         User* u = (User*)target;
870                         s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" FMODE "+u->uuid+" "+ConvToStr(u->age)+" "+output_text);
871                 }
872                 else
873                 {
874                         Channel* c = (Channel*)target;
875                         s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" FMODE "+c->name+" "+ConvToStr(c->age)+" "+output_text);
876                 }
877         }
878 }
879
880 void ModuleSpanningTree::ProtoSendMetaData(void* opaque, int target_type, void* target, const std::string &extname, const std::string &extdata)
881 {
882         TreeSocket* s = (TreeSocket*)opaque;
883         if (target)
884         {
885                 if (target_type == TYPE_USER)
886                 {
887                         User* u = (User*)target;
888                         s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" METADATA "+u->uuid+" "+extname+" :"+extdata);
889                 }
890                 else if (target_type == TYPE_CHANNEL)
891                 {
892                         Channel* c = (Channel*)target;
893                         s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" METADATA "+c->name+" "+extname+" :"+extdata);
894                 }
895         }
896         if (target_type == TYPE_OTHER)
897         {
898                 s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" METADATA * "+extname+" :"+extdata);
899         }
900 }
901
902 void ModuleSpanningTree::OnEvent(Event* event)
903 {
904         std::deque<std::string>* params = (std::deque<std::string>*)event->GetData();
905         if (event->GetEventID() == "send_metadata")
906         {
907                 if (params->size() < 3)
908                         return;
909                 (*params)[2] = ":" + (*params)[2];
910                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"METADATA",*params);
911         }
912         else if (event->GetEventID() == "send_topic")
913         {
914                 if (params->size() < 2)
915                         return;
916                 (*params)[1] = ":" + (*params)[1];
917                 params->insert(params->begin() + 1,ServerInstance->Config->ServerName);
918                 params->insert(params->begin() + 1,ConvToStr(ServerInstance->Time(true)));
919                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"FTOPIC",*params);
920         }
921         else if (event->GetEventID() == "send_mode")
922         {
923                 if (params->size() < 2)
924                         return;
925                 // Insert the TS value of the object, either User or Channel
926                 time_t ourTS = 0;
927                 std::string output_text;
928
929                 /* Warning: in-place translation is only safe for type TR_NICK */
930                 for (size_t n = 0; n < params->size(); n++)
931                         ServerInstance->Parser->TranslateUIDs(TR_NICK, (*params)[n], (*params)[n]);
932
933                 User* a = ServerInstance->FindNick((*params)[0]);
934                 if (a)
935                 {
936                         ourTS = a->age;
937                         Utils->DoOneToMany(ServerInstance->Config->GetSID(),"MODE",*params);
938                         return;
939                 }
940                 else
941                 {
942                         Channel* a = ServerInstance->FindChan((*params)[0]);
943                         if (a)
944                         {
945                                 ourTS = a->age;
946                                 params->insert(params->begin() + 1,ConvToStr(ourTS));
947                                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"FMODE",*params);
948                         }
949                 }
950         }
951         else if (event->GetEventID() == "send_mode_explicit")
952         {
953                 if (params->size() < 2)
954                         return;
955                 std::string output_text;
956
957                 /* Warning: in-place translation is only safe for type TR_NICK */
958                 for (size_t n = 0; n < params->size(); n++)
959                         ServerInstance->Parser->TranslateUIDs(TR_NICK, (*params)[n], (*params)[n]);
960
961                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"MODE",*params);
962         }
963         else if (event->GetEventID() == "send_opers")
964         {
965                 if (params->size() < 1)
966                         return;
967                 (*params)[0] = ":" + (*params)[0];
968                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"OPERNOTICE",*params);
969         }
970         else if (event->GetEventID() == "send_modeset")
971         {
972                 if (params->size() < 2)
973                         return;
974                 (*params)[1] = ":" + (*params)[1];
975                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"MODENOTICE",*params);
976         }
977         else if (event->GetEventID() == "send_snoset")
978         {
979                 if (params->size() < 2)
980                         return;
981                 (*params)[1] = ":" + (*params)[1];
982                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"SNONOTICE",*params);
983         }
984         else if (event->GetEventID() == "send_push")
985         {
986                 if (params->size() < 2)
987                         return;
988                         
989                 User *a = ServerInstance->FindNick((*params)[0]);
990                         
991                 if (!a)
992                         return;
993
994                 (*params)[0] = a->uuid;
995                 (*params)[1] = ":" + (*params)[1];
996                 Utils->DoOneToOne(ServerInstance->Config->GetSID(), "PUSH", *params, a->server);
997         }
998 }
999
1000 ModuleSpanningTree::~ModuleSpanningTree()
1001 {
1002         /* This will also free the listeners */
1003         delete Utils;
1004         if (SyncTimer)
1005                 ServerInstance->Timers->DelTimer(SyncTimer);
1006
1007         ServerInstance->Timers->DelTimer(RefreshTimer);
1008
1009         ServerInstance->Modules->DoneWithInterface("BufferedSocketHook");
1010 }
1011
1012 Version ModuleSpanningTree::GetVersion()
1013 {
1014         return Version(1,1,0,2,VF_VENDOR,API_VERSION);
1015 }
1016
1017 /* It is IMPORTANT that m_spanningtree is the last module in the chain
1018  * so that any activity it sees is FINAL, e.g. we arent going to send out
1019  * a NICK message before m_cloaking has finished putting the +x on the user,
1020  * etc etc.
1021  * Therefore, we return PRIORITY_LAST to make sure we end up at the END of
1022  * the module call queue.
1023  */
1024 void ModuleSpanningTree::Prioritize()
1025 {
1026         ServerInstance->Modules->SetPriority(this, PRIO_LAST);
1027 }
1028
1029 MODULE_INIT(ModuleSpanningTree)