]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/m_spanningtree/main.cpp
c38d54d4d0e5c1a36f806b12518beb36d1d1e785
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / main.cpp
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
6  * See: http://wiki.inspircd.org/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 "socket.h"
18 #include "xline.h"
19
20 #include "cachetimer.h"
21 #include "resolvers.h"
22 #include "main.h"
23 #include "utils.h"
24 #include "treeserver.h"
25 #include "link.h"
26 #include "treesocket.h"
27 #include "commands.h"
28 #include "protocolinterface.h"
29
30 ModuleSpanningTree::ModuleSpanningTree()
31 {
32         Utils = new SpanningTreeUtilities(this);
33         commands = new SpanningTreeCommands(this);
34         RefreshTimer = NULL;
35 }
36
37 SpanningTreeCommands::SpanningTreeCommands(ModuleSpanningTree* module)
38         : rconnect(module, module->Utils), rsquit(module, module->Utils),
39         svsjoin(module), svspart(module), svsnick(module), metadata(module),
40         uid(module), opertype(module), fjoin(module), fmode(module), ftopic(module),
41         fhost(module), fident(module), fname(module)
42 {
43 }
44
45 void ModuleSpanningTree::init()
46 {
47         ServerInstance->Modules->AddService(commands->rconnect);
48         ServerInstance->Modules->AddService(commands->rsquit);
49         ServerInstance->Modules->AddService(commands->svsjoin);
50         ServerInstance->Modules->AddService(commands->svspart);
51         ServerInstance->Modules->AddService(commands->svsnick);
52         ServerInstance->Modules->AddService(commands->metadata);
53         ServerInstance->Modules->AddService(commands->uid);
54         ServerInstance->Modules->AddService(commands->opertype);
55         ServerInstance->Modules->AddService(commands->fjoin);
56         ServerInstance->Modules->AddService(commands->fmode);
57         ServerInstance->Modules->AddService(commands->ftopic);
58         ServerInstance->Modules->AddService(commands->fhost);
59         ServerInstance->Modules->AddService(commands->fident);
60         ServerInstance->Modules->AddService(commands->fname);
61         RefreshTimer = new CacheRefreshTimer(Utils);
62         ServerInstance->Timers->AddTimer(RefreshTimer);
63
64         Implementation eventlist[] =
65         {
66                 I_OnPreCommand, I_OnGetServerDescription, I_OnUserInvite, I_OnPostTopicChange,
67                 I_OnWallops, I_OnUserNotice, I_OnUserMessage, I_OnBackgroundTimer, I_OnUserJoin,
68                 I_OnChangeHost, I_OnChangeName, I_OnChangeIdent, I_OnUserPart, I_OnUnloadModule,
69                 I_OnUserQuit, I_OnUserPostNick, I_OnUserKick, I_OnRemoteKill, I_OnRehash, I_OnPreRehash,
70                 I_OnOper, I_OnAddLine, I_OnDelLine, I_OnMode, I_OnLoadModule, I_OnStats,
71                 I_OnSetAway, I_OnPostCommand, I_OnUserConnect, I_OnAcceptConnection
72         };
73         ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
74
75         delete ServerInstance->PI;
76         ServerInstance->PI = new SpanningTreeProtocolInterface(this, Utils);
77         loopCall = false;
78
79         // update our local user count
80         Utils->TreeRoot->SetUserCount(ServerInstance->Users->local_users.size());
81 }
82
83 void ModuleSpanningTree::ShowLinks(TreeServer* Current, User* user, int hops)
84 {
85         std::string Parent = Utils->TreeRoot->GetName();
86         if (Current->GetParent())
87         {
88                 Parent = Current->GetParent()->GetName();
89         }
90         for (unsigned int q = 0; q < Current->ChildCount(); q++)
91         {
92                 if ((Current->GetChild(q)->Hidden) || ((Utils->HideULines) && (ServerInstance->ULine(Current->GetChild(q)->GetName().c_str()))))
93                 {
94                         if (IS_OPER(user))
95                         {
96                                  ShowLinks(Current->GetChild(q),user,hops+1);
97                         }
98                 }
99                 else
100                 {
101                         ShowLinks(Current->GetChild(q),user,hops+1);
102                 }
103         }
104         /* Don't display the line if its a uline, hide ulines is on, and the user isnt an oper */
105         if ((Utils->HideULines) && (ServerInstance->ULine(Current->GetName().c_str())) && (!IS_OPER(user)))
106                 return;
107         /* Or if the server is hidden and they're not an oper */
108         else if ((Current->Hidden) && (!IS_OPER(user)))
109                 return;
110
111         user->WriteNumeric(364, "%s %s %s :%d %s",      user->nick.c_str(),Current->GetName().c_str(),
112                         (Utils->FlatLinks && (!IS_OPER(user))) ? ServerInstance->Config->ServerName.c_str() : Parent.c_str(),
113                         (Utils->FlatLinks && (!IS_OPER(user))) ? 0 : hops,
114                         Current->GetDesc().c_str());
115 }
116
117 int ModuleSpanningTree::CountServs()
118 {
119         return Utils->serverlist.size();
120 }
121
122 void ModuleSpanningTree::HandleLinks(const std::vector<std::string>& parameters, User* user)
123 {
124         ShowLinks(Utils->TreeRoot,user,0);
125         user->WriteNumeric(365, "%s * :End of /LINKS list.",user->nick.c_str());
126         return;
127 }
128
129 std::string ModuleSpanningTree::TimeToStr(time_t secs)
130 {
131         time_t mins_up = secs / 60;
132         time_t hours_up = mins_up / 60;
133         time_t days_up = hours_up / 24;
134         secs = secs % 60;
135         mins_up = mins_up % 60;
136         hours_up = hours_up % 24;
137         return ((days_up ? (ConvToStr(days_up) + "d") : std::string(""))
138                         + (hours_up ? (ConvToStr(hours_up) + "h") : std::string(""))
139                         + (mins_up ? (ConvToStr(mins_up) + "m") : std::string(""))
140                         + ConvToStr(secs) + "s");
141 }
142
143 void ModuleSpanningTree::DoPingChecks(time_t curtime)
144 {
145         /*
146          * Cancel remote burst mode on any servers which still have it enabled due to latency/lack of data.
147          * This prevents lost REMOTECONNECT notices
148          */
149         long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
150
151 restart:
152         for (server_hash::iterator i = Utils->serverlist.begin(); i != Utils->serverlist.end(); i++)
153         {
154                 TreeServer *s = i->second;
155                 
156                 if (s->GetSocket() && s->GetSocket()->GetLinkState() == DYING)
157                 {
158                         s->GetSocket()->Close();
159                         goto restart;
160                 }
161
162                 // Fix for bug #792, do not ping servers that are not connected yet!
163                 // Remote servers have Socket == NULL and local connected servers have
164                 // Socket->LinkState == CONNECTED
165                 if (s->GetSocket() && s->GetSocket()->GetLinkState() != CONNECTED)
166                         continue;
167
168                 // Now do PING checks on all servers
169                 TreeServer *mts = Utils->BestRouteTo(s->GetID());
170
171                 if (mts)
172                 {
173                         // Only ping if this server needs one
174                         if (curtime >= s->NextPingTime())
175                         {
176                                 // And if they answered the last
177                                 if (s->AnsweredLastPing())
178                                 {
179                                         // They did, send a ping to them
180                                         s->SetNextPingTime(curtime + Utils->PingFreq);
181                                         TreeSocket *tsock = mts->GetSocket();
182
183                                         // ... if we can find a proper route to them
184                                         if (tsock)
185                                         {
186                                                 tsock->WriteLine(std::string(":") + ServerInstance->Config->GetSID() + " PING " +
187                                                                 ServerInstance->Config->GetSID() + " " + s->GetID());
188                                                 s->LastPingMsec = ts;
189                                         }
190                                 }
191                                 else
192                                 {
193                                         // They didn't answer the last ping, if they are locally connected, get rid of them.
194                                         TreeSocket *sock = s->GetSocket();
195                                         if (sock)
196                                         {
197                                                 sock->SendError("Ping timeout");
198                                                 sock->Close();
199                                                 goto restart;
200                                         }
201                                 }
202                         }
203
204                         // If warn on ping enabled and not warned and the difference is sufficient and they didn't answer the last ping...
205                         if ((Utils->PingWarnTime) && (!s->Warned) && (curtime >= s->NextPingTime() - (Utils->PingFreq - Utils->PingWarnTime)) && (!s->AnsweredLastPing()))
206                         {
207                                 /* The server hasnt responded, send a warning to opers */
208                                 ServerInstance->SNO->WriteToSnoMask('l',"Server \002%s\002 has not responded to PING for %d seconds, high latency.", s->GetName().c_str(), Utils->PingWarnTime);
209                                 s->Warned = true;
210                         }
211                 }
212         }
213 }
214
215 void ModuleSpanningTree::ConnectServer(Autoconnect* a, bool on_timer)
216 {
217         if (!a)
218                 return;
219         for(unsigned int j=0; j < a->servers.size(); j++)
220         {
221                 if (Utils->FindServer(a->servers[j]))
222                 {
223                         // found something in this block. Should the server fail,
224                         // we want to start at the start of the list, not in the
225                         // middle where we left off
226                         a->position = -1;
227                         return;
228                 }
229         }
230         if (on_timer && a->position >= 0)
231                 return;
232         if (!on_timer && a->position < 0)
233                 return;
234
235         a->position++;
236         while (a->position < (int)a->servers.size())
237         {
238                 Link* x = Utils->FindLink(a->servers[a->position]);
239                 if (x)
240                 {
241                         ServerInstance->SNO->WriteToSnoMask('l', "AUTOCONNECT: Auto-connecting server \002%s\002", x->Name.c_str());
242                         ConnectServer(x, a);
243                         return;
244                 }
245                 a->position++;
246         }
247         // Autoconnect chain has been fully iterated; start at the beginning on the
248         // next AutoConnectServers run
249         a->position = -1;
250 }
251
252 void ModuleSpanningTree::ConnectServer(Link* x, Autoconnect* y)
253 {
254         bool ipvalid = true;
255
256         if (InspIRCd::Match(ServerInstance->Config->ServerName, assign(x->Name)))
257         {
258                 ServerInstance->SNO->WriteToSnoMask('l', "CONNECT: Not connecting to myself.");
259                 return;
260         }
261
262         QueryType start_type = DNS_QUERY_A;
263         start_type = DNS_QUERY_AAAA;
264         if (strchr(x->IPAddr.c_str(),':'))
265         {
266                 in6_addr n;
267                 if (inet_pton(AF_INET6, x->IPAddr.c_str(), &n) < 1)
268                         ipvalid = false;
269         }
270         else
271         {
272                 in_addr n;
273                 if (inet_aton(x->IPAddr.c_str(),&n) < 1)
274                         ipvalid = false;
275         }
276
277         /* Do we already have an IP? If so, no need to resolve it. */
278         if (ipvalid)
279         {
280                 /* Gave a hook, but it wasnt one we know */
281                 TreeSocket* newsocket = new TreeSocket(Utils, x, y, x->IPAddr);
282                 if (newsocket->GetFd() > -1)
283                 {
284                         /* Handled automatically on success */
285                 }
286                 else
287                 {
288                         ServerInstance->SNO->WriteToSnoMask('l', "CONNECT: Error connecting \002%s\002: %s.",
289                                 x->Name.c_str(), newsocket->getError().c_str());
290                         ServerInstance->GlobalCulls.AddItem(newsocket);
291                 }
292         }
293         else
294         {
295                 try
296                 {
297                         bool cached;
298                         ServernameResolver* snr = new ServernameResolver(Utils, x->IPAddr, x, cached, start_type, y);
299                         ServerInstance->AddResolver(snr, cached);
300                 }
301                 catch (ModuleException& e)
302                 {
303                         ServerInstance->SNO->WriteToSnoMask('l', "CONNECT: Error connecting \002%s\002: %s.",x->Name.c_str(), e.GetReason());
304                         ConnectServer(y, false);
305                 }
306         }
307 }
308
309 void ModuleSpanningTree::AutoConnectServers(time_t curtime)
310 {
311         for (std::vector<reference<Autoconnect> >::iterator i = Utils->AutoconnectBlocks.begin(); i < Utils->AutoconnectBlocks.end(); ++i)
312         {
313                 Autoconnect* x = *i;
314                 if (curtime >= x->NextConnectTime)
315                 {
316                         x->NextConnectTime = curtime + x->Period;
317                         ConnectServer(x, true);
318                 }
319         }
320 }
321
322 void ModuleSpanningTree::DoConnectTimeout(time_t curtime)
323 {
324         std::map<TreeSocket*, std::pair<std::string, int> >::iterator i = Utils->timeoutlist.begin();
325         while (i != Utils->timeoutlist.end())
326         {
327                 TreeSocket* s = i->first;
328                 std::pair<std::string, int> p = i->second;
329                 std::map<TreeSocket*, std::pair<std::string, int> >::iterator me = i;
330                 i++;
331                 if (s->GetLinkState() == DYING)
332                 {
333                         Utils->timeoutlist.erase(me);
334                         s->Close();
335                 }
336                 else if (curtime > s->age + p.second)
337                 {
338                         ServerInstance->SNO->WriteToSnoMask('l',"CONNECT: Error connecting \002%s\002 (timeout of %d seconds)",p.first.c_str(),p.second);
339                         Utils->timeoutlist.erase(me);
340                         s->Close();
341                 }
342         }
343 }
344
345 ModResult ModuleSpanningTree::HandleVersion(const std::vector<std::string>& parameters, User* user)
346 {
347         // we've already checked if pcnt > 0, so this is safe
348         TreeServer* found = Utils->FindServerMask(parameters[0]);
349         if (found)
350         {
351                 std::string Version = found->GetVersion();
352                 user->WriteNumeric(351, "%s :%s",user->nick.c_str(),Version.c_str());
353                 if (found == Utils->TreeRoot)
354                 {
355                         ServerInstance->Config->Send005(user);
356                 }
357         }
358         else
359         {
360                 user->WriteNumeric(402, "%s %s :No such server",user->nick.c_str(),parameters[0].c_str());
361         }
362         return MOD_RES_DENY;
363 }
364
365 /* This method will attempt to get a message to a remote user.
366  */
367 void ModuleSpanningTree::RemoteMessage(User* user, const char* format, ...)
368 {
369         char text[MAXBUF];
370         va_list argsPtr;
371
372         va_start(argsPtr, format);
373         vsnprintf(text, MAXBUF, format, argsPtr);
374         va_end(argsPtr);
375
376         if (IS_LOCAL(user))
377                 user->WriteServ("NOTICE %s :%s", user->nick.c_str(), text);
378         else
379                 ServerInstance->PI->SendUserNotice(user, text);
380 }
381
382 ModResult ModuleSpanningTree::HandleConnect(const std::vector<std::string>& parameters, User* user)
383 {
384         for (std::vector<reference<Link> >::iterator i = Utils->LinkBlocks.begin(); i < Utils->LinkBlocks.end(); i++)
385         {
386                 Link* x = *i;
387                 if (InspIRCd::Match(x->Name.c_str(),parameters[0]))
388                 {
389                         if (InspIRCd::Match(ServerInstance->Config->ServerName, assign(x->Name)))
390                         {
391                                 RemoteMessage(user, "*** CONNECT: Server \002%s\002 is ME, not connecting.",x->Name.c_str());
392                                 return MOD_RES_DENY;
393                         }
394
395                         TreeServer* CheckDupe = Utils->FindServer(x->Name.c_str());
396                         if (!CheckDupe)
397                         {
398                                 RemoteMessage(user, "*** CONNECT: Connecting to server: \002%s\002 (%s:%d)",x->Name.c_str(),(x->HiddenFromStats ? "<hidden>" : x->IPAddr.c_str()),x->Port);
399                                 ConnectServer(x);
400                                 return MOD_RES_DENY;
401                         }
402                         else
403                         {
404                                 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());
405                                 return MOD_RES_DENY;
406                         }
407                 }
408         }
409         RemoteMessage(user, "*** CONNECT: No server matching \002%s\002 could be found in the config file.",parameters[0].c_str());
410         return MOD_RES_DENY;
411 }
412
413 void ModuleSpanningTree::OnGetServerDescription(const std::string &servername,std::string &description)
414 {
415         TreeServer* s = Utils->FindServer(servername);
416         if (s)
417         {
418                 description = s->GetDesc();
419         }
420 }
421
422 void ModuleSpanningTree::OnUserInvite(User* source,User* dest,Channel* channel, time_t expiry)
423 {
424         if (IS_LOCAL(source))
425         {
426                 parameterlist params;
427                 params.push_back(dest->uuid);
428                 params.push_back(channel->name);
429                 params.push_back(ConvToStr(expiry));
430                 Utils->DoOneToMany(source->uuid,"INVITE",params);
431         }
432 }
433
434 void ModuleSpanningTree::OnPostTopicChange(User* user, Channel* chan, const std::string &topic)
435 {
436         // Drop remote events on the floor.
437         if (!IS_LOCAL(user))
438                 return;
439
440         parameterlist params;
441         params.push_back(chan->name);
442         params.push_back(":"+topic);
443         Utils->DoOneToMany(user->uuid,"TOPIC",params);
444 }
445
446 void ModuleSpanningTree::OnWallops(User* user, const std::string &text)
447 {
448         if (IS_LOCAL(user))
449         {
450                 parameterlist params;
451                 params.push_back(":"+text);
452                 Utils->DoOneToMany(user->uuid,"WALLOPS",params);
453         }
454 }
455
456 void ModuleSpanningTree::OnUserNotice(User* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
457 {
458         /* Server origin */
459         if (user == NULL)
460                 return;
461
462         if (target_type == TYPE_USER)
463         {
464                 User* d = (User*)dest;
465                 if (!IS_LOCAL(d) && IS_LOCAL(user))
466                 {
467                         parameterlist params;
468                         params.push_back(d->uuid);
469                         params.push_back(":"+text);
470                         Utils->DoOneToOne(user->uuid,"NOTICE",params,d->server);
471                 }
472         }
473         else if (target_type == TYPE_CHANNEL)
474         {
475                 if (IS_LOCAL(user))
476                 {
477                         Channel *c = (Channel*)dest;
478                         if (c)
479                         {
480                                 std::string cname = c->name;
481                                 if (status)
482                                         cname = status + cname;
483                                 TreeServerList list;
484                                 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
485                                 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
486                                 {
487                                         TreeSocket* Sock = i->second->GetSocket();
488                                         if (Sock)
489                                                 Sock->WriteLine(":"+std::string(user->uuid)+" NOTICE "+cname+" :"+text);
490                                 }
491                         }
492                 }
493         }
494         else if (target_type == TYPE_SERVER)
495         {
496                 if (IS_LOCAL(user))
497                 {
498                         char* target = (char*)dest;
499                         parameterlist par;
500                         par.push_back(target);
501                         par.push_back(":"+text);
502                         Utils->DoOneToMany(user->uuid,"NOTICE",par);
503                 }
504         }
505 }
506
507 void ModuleSpanningTree::OnUserMessage(User* user, void* dest, int target_type, const std::string &text, char status, const CUList &exempt_list)
508 {
509         /* Server origin */
510         if (user == NULL)
511                 return;
512
513         if (target_type == TYPE_USER)
514         {
515                 // route private messages which are targetted at clients only to the server
516                 // which needs to receive them
517                 User* d = (User*)dest;
518                 if (!IS_LOCAL(d) && (IS_LOCAL(user)))
519                 {
520                         parameterlist params;
521                         params.push_back(d->uuid);
522                         params.push_back(":"+text);
523                         Utils->DoOneToOne(user->uuid,"PRIVMSG",params,d->server);
524                 }
525         }
526         else if (target_type == TYPE_CHANNEL)
527         {
528                 if (IS_LOCAL(user))
529                 {
530                         Channel *c = (Channel*)dest;
531                         if (c)
532                         {
533                                 std::string cname = c->name;
534                                 if (status)
535                                         cname = status + cname;
536                                 TreeServerList list;
537                                 Utils->GetListOfServersForChannel(c,list,status,exempt_list);
538                                 for (TreeServerList::iterator i = list.begin(); i != list.end(); i++)
539                                 {
540                                         TreeSocket* Sock = i->second->GetSocket();
541                                         if (Sock)
542                                                 Sock->WriteLine(":"+std::string(user->uuid)+" PRIVMSG "+cname+" :"+text);
543                                 }
544                         }
545                 }
546         }
547         else if (target_type == TYPE_SERVER)
548         {
549                 if (IS_LOCAL(user))
550                 {
551                         char* target = (char*)dest;
552                         parameterlist par;
553                         par.push_back(target);
554                         par.push_back(":"+text);
555                         Utils->DoOneToMany(user->uuid,"PRIVMSG",par);
556                 }
557         }
558 }
559
560 void ModuleSpanningTree::OnBackgroundTimer(time_t curtime)
561 {
562         AutoConnectServers(curtime);
563         DoPingChecks(curtime);
564         DoConnectTimeout(curtime);
565 }
566
567 void ModuleSpanningTree::OnUserConnect(LocalUser* user)
568 {
569         if (user->quitting)
570                 return;
571
572         parameterlist params;
573         params.push_back(user->uuid);
574         params.push_back(ConvToStr(user->age));
575         params.push_back(user->nick);
576         params.push_back(user->host);
577         params.push_back(user->dhost);
578         params.push_back(user->ident);
579         params.push_back(user->GetIPString());
580         params.push_back(ConvToStr(user->signon));
581         params.push_back("+"+std::string(user->FormatModes(true)));
582         params.push_back(":"+std::string(user->fullname));
583         Utils->DoOneToMany(ServerInstance->Config->GetSID(), "UID", params);
584
585         if (IS_OPER(user))
586         {
587                 params.clear();
588                 params.push_back(user->oper->name);
589                 Utils->DoOneToMany(user->uuid,"OPERTYPE",params);
590         }
591
592         for(Extensible::ExtensibleStore::const_iterator i = user->GetExtList().begin(); i != user->GetExtList().end(); i++)
593         {
594                 ExtensionItem* item = i->first;
595                 std::string value = item->serialize(FORMAT_NETWORK, user, i->second);
596                 if (!value.empty())
597                         ServerInstance->PI->SendMetaData(user, item->name, value);
598         }
599
600         Utils->TreeRoot->SetUserCount(1); // increment by 1
601 }
602
603 void ModuleSpanningTree::OnUserJoin(Membership* memb, bool sync, bool created, CUList& excepts)
604 {
605         // Only do this for local users
606         if (IS_LOCAL(memb->user))
607         {
608                 parameterlist params;
609                 // set up their permissions and the channel TS with FJOIN.
610                 // All users are FJOINed now, because a module may specify
611                 // new joining permissions for the user.
612                 params.push_back(memb->chan->name);
613                 params.push_back(ConvToStr(memb->chan->age));
614                 params.push_back(std::string("+") + memb->chan->ChanModes(true));
615                 params.push_back(memb->modes+","+std::string(memb->user->uuid));
616                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"FJOIN",params);
617         }
618 }
619
620 void ModuleSpanningTree::OnChangeHost(User* user, const std::string &newhost)
621 {
622         if (user->registered != REG_ALL || !IS_LOCAL(user))
623                 return;
624
625         parameterlist params;
626         params.push_back(newhost);
627         Utils->DoOneToMany(user->uuid,"FHOST",params);
628 }
629
630 void ModuleSpanningTree::OnChangeName(User* user, const std::string &gecos)
631 {
632         if (user->registered != REG_ALL || !IS_LOCAL(user))
633                 return;
634
635         parameterlist params;
636         params.push_back(gecos);
637         Utils->DoOneToMany(user->uuid,"FNAME",params);
638 }
639
640 void ModuleSpanningTree::OnChangeIdent(User* user, const std::string &ident)
641 {
642         // only occurs for local clients
643         if (user->registered != REG_ALL)
644                 return;
645
646         parameterlist params;
647         params.push_back(ident);
648         Utils->DoOneToMany(user->uuid,"FIDENT",params);
649 }
650
651 void ModuleSpanningTree::OnUserPart(Membership* memb, std::string &partmessage, CUList& excepts)
652 {
653         if (IS_LOCAL(memb->user))
654         {
655                 parameterlist params;
656                 params.push_back(memb->chan->name);
657                 if (!partmessage.empty())
658                         params.push_back(":"+partmessage);
659                 Utils->DoOneToMany(memb->user->uuid,"PART",params);
660         }
661 }
662
663 void ModuleSpanningTree::OnUserQuit(User* user, const std::string &reason, const std::string &oper_message)
664 {
665         if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
666         {
667                 parameterlist params;
668
669                 if (oper_message != reason)
670                 {
671                         params.push_back(":"+oper_message);
672                         Utils->DoOneToMany(user->uuid,"OPERQUIT",params);
673                 }
674                 params.clear();
675                 params.push_back(":"+reason);
676                 Utils->DoOneToMany(user->uuid,"QUIT",params);
677         }
678
679         // Regardless, We need to modify the user Counts..
680         TreeServer* SourceServer = Utils->FindServer(user->server);
681         if (SourceServer)
682         {
683                 SourceServer->SetUserCount(-1); // decrement by 1
684         }
685 }
686
687 void ModuleSpanningTree::OnUserPostNick(User* user, const std::string &oldnick)
688 {
689         if (IS_LOCAL(user))
690         {
691                 parameterlist params;
692                 params.push_back(user->nick);
693
694                 /** IMPORTANT: We don't update the TS if the oldnick is just a case change of the newnick!
695                  */
696                 if (irc::string(user->nick.c_str()) != assign(oldnick))
697                         user->age = ServerInstance->Time();
698
699                 params.push_back(ConvToStr(user->age));
700                 Utils->DoOneToMany(user->uuid,"NICK",params);
701         }
702         else if (!loopCall && user->nick == user->uuid)
703         {
704                 parameterlist params;
705                 params.push_back(user->uuid);
706                 params.push_back(ConvToStr(user->age));
707                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"SAVE",params);
708         }
709 }
710
711 void ModuleSpanningTree::OnUserKick(User* source, Membership* memb, const std::string &reason, CUList& excepts)
712 {
713         parameterlist params;
714         params.push_back(memb->chan->name);
715         params.push_back(memb->user->uuid);
716         params.push_back(":"+reason);
717         if (IS_LOCAL(source))
718         {
719                 Utils->DoOneToMany(source->uuid,"KICK",params);
720         }
721         else if (source == ServerInstance->FakeClient)
722         {
723                 Utils->DoOneToMany(ServerInstance->Config->GetSID(),"KICK",params);
724         }
725 }
726
727 void ModuleSpanningTree::OnRemoteKill(User* source, User* dest, const std::string &reason, const std::string &operreason)
728 {
729         if (!IS_LOCAL(source))
730                 return; // Only start routing if we're origin.
731
732         ServerInstance->OperQuit.set(dest, operreason);
733         parameterlist params;
734         params.push_back(":"+operreason);
735         Utils->DoOneToMany(dest->uuid,"OPERQUIT",params);
736         params.clear();
737         params.push_back(dest->uuid);
738         params.push_back(":"+reason);
739         Utils->DoOneToMany(source->uuid,"KILL",params);
740 }
741
742 void ModuleSpanningTree::OnPreRehash(User* user, const std::string &parameter)
743 {
744         ServerInstance->Logs->Log("remoterehash", DEBUG, "called with param %s", parameter.c_str());
745
746         // Send out to other servers
747         if (!parameter.empty() && parameter[0] != '-')
748         {
749                 parameterlist params;
750                 params.push_back(parameter);
751                 Utils->DoOneToAllButSender(user ? user->uuid : ServerInstance->Config->GetSID(), "REHASH", params, user ? user->server : ServerInstance->Config->ServerName);
752         }
753 }
754
755 void ModuleSpanningTree::OnRehash(User* user)
756 {
757         // Re-read config stuff
758         Utils->ReadConfiguration();
759 }
760
761 void ModuleSpanningTree::OnLoadModule(Module* mod)
762 {
763         std::string data;
764         data.push_back('+');
765         data.append(mod->ModuleSourceFile);
766         Version v = mod->GetVersion();
767         if (!v.link_data.empty())
768         {
769                 data.push_back('=');
770                 data.append(v.link_data);
771         }
772         ServerInstance->PI->SendMetaData(NULL, "modules", data);
773 }
774
775 void ModuleSpanningTree::OnUnloadModule(Module* mod)
776 {
777         ServerInstance->PI->SendMetaData(NULL, "modules", "-" + mod->ModuleSourceFile);
778
779         unsigned int items = Utils->TreeRoot->ChildCount();
780         for(unsigned int x = 0; x < items; x++)
781         {
782                 TreeServer* srv = Utils->TreeRoot->GetChild(x);
783                 TreeSocket* sock = srv->GetSocket();
784                 if (sock && sock->GetIOHook() == mod)
785                 {
786                         sock->SendError("SSL module unloaded");
787                         sock->Close();
788                 }
789         }
790 }
791
792 void ModuleSpanningTree::RedoConfig(Module* mod)
793 {
794 }
795
796 // note: the protocol does not allow direct umode +o except
797 // via NICK with 8 params. sending OPERTYPE infers +o modechange
798 // locally.
799 void ModuleSpanningTree::OnOper(User* user, const std::string &opertype)
800 {
801         if (user->registered != REG_ALL || !IS_LOCAL(user))
802                 return;
803         parameterlist params;
804         params.push_back(opertype);
805         Utils->DoOneToMany(user->uuid,"OPERTYPE",params);
806 }
807
808 void ModuleSpanningTree::OnAddLine(User* user, XLine *x)
809 {
810         if (!x->IsBurstable() || loopCall)
811                 return;
812
813         char data[MAXBUF];
814         snprintf(data,MAXBUF,"%s %s %s %lu %lu :%s", x->type.c_str(), x->Displayable(),
815         ServerInstance->Config->ServerName.c_str(), (unsigned long)x->set_time, (unsigned long)x->duration, x->reason.c_str());
816         parameterlist params;
817         params.push_back(data);
818
819         if (!user)
820         {
821                 /* Server-set lines */
822                 Utils->DoOneToMany(ServerInstance->Config->GetSID(), "ADDLINE", params);
823         }
824         else if (IS_LOCAL(user))
825         {
826                 /* User-set lines */
827                 Utils->DoOneToMany(user->uuid, "ADDLINE", params);
828         }
829 }
830
831 void ModuleSpanningTree::OnDelLine(User* user, XLine *x)
832 {
833         if (!x->IsBurstable() || loopCall)
834                 return;
835
836         char data[MAXBUF];
837         snprintf(data,MAXBUF,"%s %s", x->type.c_str(), x->Displayable());
838         parameterlist params;
839         params.push_back(data);
840
841         if (!user)
842         {
843                 /* Server-unset lines */
844                 Utils->DoOneToMany(ServerInstance->Config->GetSID(), "DELLINE", params);
845         }
846         else if (IS_LOCAL(user))
847         {
848                 /* User-unset lines */
849                 Utils->DoOneToMany(user->uuid, "DELLINE", params);
850         }
851 }
852
853 void ModuleSpanningTree::OnMode(User* user, void* dest, int target_type, const parameterlist &text, const std::vector<TranslateType> &translate)
854 {
855         if ((IS_LOCAL(user)) && (user->registered == REG_ALL))
856         {
857                 parameterlist params;
858                 std::string command;
859                 std::string output_text;
860
861                 ServerInstance->Parser->TranslateUIDs(translate, text, output_text);
862
863                 if (target_type == TYPE_USER)
864                 {
865                         User* u = (User*)dest;
866                         params.push_back(u->uuid);
867                         params.push_back(output_text);
868                         command = "MODE";
869                 }
870                 else
871                 {
872                         Channel* c = (Channel*)dest;
873                         params.push_back(c->name);
874                         params.push_back(ConvToStr(c->age));
875                         params.push_back(output_text);
876                         command = "FMODE";
877                 }
878
879                 Utils->DoOneToMany(user->uuid, command, params);
880         }
881 }
882
883 ModResult ModuleSpanningTree::OnSetAway(User* user, const std::string &awaymsg)
884 {
885         if (IS_LOCAL(user))
886         {
887                 if (awaymsg.empty())
888                 {
889                         parameterlist params;
890                         Utils->DoOneToMany(user->uuid,"AWAY",params);
891                 }
892                 else
893                 {
894                         parameterlist params;
895                         params.push_back(ConvToStr(user->awaytime));
896                         params.push_back(":" + awaymsg);
897                         Utils->DoOneToMany(user->uuid,"AWAY",params);
898                 }
899         }
900
901         return MOD_RES_PASSTHRU;
902 }
903
904 void ModuleSpanningTree::ProtoSendMode(void* opaque, TargetTypeFlags target_type, void* target, const parameterlist &modeline, const std::vector<TranslateType> &translate)
905 {
906         TreeSocket* s = (TreeSocket*)opaque;
907         std::string output_text;
908
909         ServerInstance->Parser->TranslateUIDs(translate, modeline, output_text);
910
911         if (target)
912         {
913                 if (target_type == TYPE_USER)
914                 {
915                         User* u = (User*)target;
916                         s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" MODE "+u->uuid+" "+output_text);
917                 }
918                 else if (target_type == TYPE_CHANNEL)
919                 {
920                         Channel* c = (Channel*)target;
921                         s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" FMODE "+c->name+" "+ConvToStr(c->age)+" "+output_text);
922                 }
923         }
924 }
925
926 void ModuleSpanningTree::ProtoSendMetaData(void* opaque, Extensible* target, const std::string &extname, const std::string &extdata)
927 {
928         TreeSocket* s = static_cast<TreeSocket*>(opaque);
929         User* u = dynamic_cast<User*>(target);
930         Channel* c = dynamic_cast<Channel*>(target);
931         if (u)
932                 s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" METADATA "+u->uuid+" "+extname+" :"+extdata);
933         else if (c)
934                 s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" METADATA "+c->name+" "+extname+" :"+extdata);
935         else if (!target)
936                 s->WriteLine(std::string(":")+ServerInstance->Config->GetSID()+" METADATA * "+extname+" :"+extdata);
937 }
938
939 CullResult ModuleSpanningTree::cull()
940 {
941         Utils->cull();
942         ServerInstance->Timers->DelTimer(RefreshTimer);
943         return this->Module::cull();
944 }
945
946 ModuleSpanningTree::~ModuleSpanningTree()
947 {
948         delete ServerInstance->PI;
949         ServerInstance->PI = new ProtocolInterface;
950
951         /* This will also free the listeners */
952         delete Utils;
953
954         delete commands;
955 }
956
957 Version ModuleSpanningTree::GetVersion()
958 {
959         return Version("Allows servers to be linked", VF_VENDOR);
960 }
961
962 /* It is IMPORTANT that m_spanningtree is the last module in the chain
963  * so that any activity it sees is FINAL, e.g. we arent going to send out
964  * a NICK message before m_cloaking has finished putting the +x on the user,
965  * etc etc.
966  * Therefore, we return PRIORITY_LAST to make sure we end up at the END of
967  * the module call queue.
968  */
969 void ModuleSpanningTree::Prioritize()
970 {
971         ServerInstance->Modules->SetPriority(this, PRIORITY_LAST);
972 }
973
974 MODULE_INIT(ModuleSpanningTree)