]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/m_spanningtree/treesocket2.cpp
Next part of Development/Hooking (see wiki)
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / treesocket2.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 #include "inspircd.h"
15 #include "commands/cmd_whois.h"
16 #include "commands/cmd_stats.h"
17 #include "socket.h"
18 #include "wildcard.h"
19 #include "xline.h"
20 #include "transport.h"
21 #include "socketengine.h"
22
23 #include "m_spanningtree/main.h"
24 #include "m_spanningtree/utils.h"
25 #include "m_spanningtree/treeserver.h"
26 #include "m_spanningtree/link.h"
27 #include "m_spanningtree/treesocket.h"
28 #include "m_spanningtree/resolvers.h"
29 #include "m_spanningtree/handshaketimer.h"
30
31 /* $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 */
32
33 static std::map<std::string, std::string> warned;       /* Server names that have had protocol violation warnings displayed for them */
34
35 int TreeSocket::WriteLine(std::string line)
36 {
37         Instance->Log(DEBUG, "S[%d] O %s", this->GetFd(), line.c_str());
38         line.append("\r\n");
39         return this->Write(line);
40 }
41
42
43 /* Handle ERROR command */
44 bool TreeSocket::Error(std::deque<std::string> &params)
45 {
46         if (params.size() < 1)
47                 return false;
48         this->Instance->SNO->WriteToSnoMask('l',"ERROR from %s: %s",(!InboundServerName.empty() ? InboundServerName.c_str() : myhost.c_str()),params[0].c_str());
49         /* we will return false to cause the socket to close. */
50         return false;
51 }
52
53 bool TreeSocket::Modules(const std::string &prefix, std::deque<std::string> &params)
54 {
55         if (params.empty())
56                 return true;
57
58         if (!this->Instance->MatchText(this->Instance->Config->ServerName, params[0]))
59         {
60                 /* Pass it on, not for us */
61                 Utils->DoOneToOne(prefix, "MODULES", params, params[0]);
62                 return true;
63         }
64
65         char strbuf[MAXBUF];
66         std::deque<std::string> par;
67         par.push_back(prefix);
68         par.push_back("");
69
70         User* source = this->Instance->FindNick(prefix);
71         if (!source)
72                 return true;
73
74         for (unsigned int i = 0; i < Instance->Config->module_names.size(); i++)
75         {
76                 Version V = Instance->Modules->modules[i]->GetVersion();
77                 char modulename[MAXBUF];
78                 char flagstate[MAXBUF];
79                 *flagstate = 0;
80                 if (V.Flags & VF_STATIC)
81                         strlcat(flagstate,", static",MAXBUF);
82                 if (V.Flags & VF_VENDOR)
83                         strlcat(flagstate,", vendor",MAXBUF);
84                 if (V.Flags & VF_COMMON)
85                         strlcat(flagstate,", common",MAXBUF);
86                 if (V.Flags & VF_SERVICEPROVIDER)
87                         strlcat(flagstate,", service provider",MAXBUF);
88                 if (!flagstate[0])
89                         strcpy(flagstate,"  <no flags>");
90                 strlcpy(modulename,Instance->Config->module_names[i].c_str(),256);
91                 if (*source->oper)
92                 {
93                         snprintf(strbuf, MAXBUF, "::%s 900 %s :0x%08lx %d.%d.%d.%d %s (%s)",Instance->Config->ServerName,source->nick,(long unsigned int)Instance->Modules->modules[i],V.Major,V.Minor,V.Revision,V.Build,ServerConfig::CleanFilename(modulename),flagstate+2);
94                 }
95                 else
96                 {
97                         snprintf(strbuf, MAXBUF, "::%s 900 %s :%s",Instance->Config->ServerName,source->nick,ServerConfig::CleanFilename(modulename));
98                 }
99                 par[1] = strbuf;
100                 Utils->DoOneToOne(Instance->Config->GetSID(), "PUSH", par, source->server);
101         }
102         snprintf(strbuf, MAXBUF, "::%s 901 %s :End of MODULES list", Instance->Config->ServerName, source->nick);
103         par[1] = strbuf;
104         Utils->DoOneToOne(Instance->Config->GetSID(), "PUSH", par, source->server);
105         return true;
106 }
107
108 /** remote MOTD. leet, huh? */
109 bool TreeSocket::Motd(const std::string &prefix, std::deque<std::string> &params)
110 {
111         if (params.size() > 0)
112         {
113                 if (this->Instance->MatchText(this->Instance->Config->ServerName, params[0]))
114                 {
115                         /* It's for our server */
116                         string_list results;
117                         User* source = this->Instance->FindNick(prefix);
118
119                         if (source)
120                         {
121                                 std::deque<std::string> par;
122                                 par.push_back(prefix);
123                                 par.push_back("");
124
125                                 if (!Instance->Config->MOTD.size())
126                                 {
127                                         par[1] = std::string("::")+Instance->Config->ServerName+" 422 "+source->nick+" :Message of the day file is missing.";
128                                         Utils->DoOneToOne(this->Instance->Config->GetSID(), "PUSH",par, source->server);
129                                         return true;
130                                 }
131
132                                 par[1] = std::string("::")+Instance->Config->ServerName+" 375 "+source->nick+" :"+Instance->Config->ServerName+" message of the day";
133                                 Utils->DoOneToOne(this->Instance->Config->GetSID(), "PUSH",par, source->server);
134
135                                 for (unsigned int i = 0; i < Instance->Config->MOTD.size(); i++)
136                                 {
137                                         par[1] = std::string("::")+Instance->Config->ServerName+" 372 "+source->nick+" :- "+Instance->Config->MOTD[i];
138                                         Utils->DoOneToOne(this->Instance->Config->GetSID(), "PUSH",par, source->server);
139                                 }
140
141                                 par[1] = std::string("::")+Instance->Config->ServerName+" 376 "+source->nick+" :End of message of the day.";
142                                 Utils->DoOneToOne(this->Instance->Config->GetSID(), "PUSH",par, source->server);
143                         }
144                 }
145                 else
146                 {
147                         /* Pass it on */
148                         User* source = this->Instance->FindNick(prefix);
149                         if (source)
150                                 Utils->DoOneToOne(prefix, "MOTD", params, params[0]);
151                 }
152         }
153         return true;
154 }
155
156 /** remote ADMIN. leet, huh? */
157 bool TreeSocket::Admin(const std::string &prefix, std::deque<std::string> &params)
158 {
159         if (params.size() > 0)
160         {
161                 if (this->Instance->MatchText(this->Instance->Config->ServerName, params[0]))
162                 {
163                         /* It's for our server */
164                         string_list results;
165                         User* source = this->Instance->FindNick(prefix);
166                         if (source)
167                         {
168                                 std::deque<std::string> par;
169                                 par.push_back(prefix);
170                                 par.push_back("");
171                                 par[1] = std::string("::")+Instance->Config->ServerName+" 256 "+source->nick+" :Administrative info for "+Instance->Config->ServerName;
172                                 Utils->DoOneToOne(this->Instance->Config->GetSID(), "PUSH",par, source->server);
173                                 par[1] = std::string("::")+Instance->Config->ServerName+" 257 "+source->nick+" :Name     - "+Instance->Config->AdminName;
174                                 Utils->DoOneToOne(this->Instance->Config->GetSID(), "PUSH",par, source->server);
175                                 par[1] = std::string("::")+Instance->Config->ServerName+" 258 "+source->nick+" :Nickname - "+Instance->Config->AdminNick;
176                                 Utils->DoOneToOne(this->Instance->Config->GetSID(), "PUSH",par, source->server);
177                                 par[1] = std::string("::")+Instance->Config->ServerName+" 258 "+source->nick+" :E-Mail   - "+Instance->Config->AdminEmail;
178                                 Utils->DoOneToOne(this->Instance->Config->GetSID(), "PUSH",par, source->server);
179                         }
180                 }
181                 else
182                 {
183                         /* Pass it on */
184                         User* source = this->Instance->FindNick(prefix);
185                         if (source)
186                                 Utils->DoOneToOne(prefix, "ADMIN", params, params[0]);
187                 }
188         }
189         return true;
190 }
191
192 bool TreeSocket::Stats(const std::string &prefix, std::deque<std::string> &params)
193 {
194         /* Get the reply to a STATS query if it matches this servername,
195          * and send it back as a load of PUSH queries
196          */
197         if (params.size() > 1)
198         {
199                 if (this->Instance->MatchText(this->Instance->Config->ServerName, params[1]))
200                 {
201                         /* It's for our server */
202                         string_list results;
203                         User* source = this->Instance->FindNick(prefix);
204                         if (source)
205                         {
206                                 std::deque<std::string> par;
207                                 par.push_back(prefix);
208                                 par.push_back("");
209                                 DoStats(this->Instance, *(params[0].c_str()), source, results);
210                                 for (size_t i = 0; i < results.size(); i++)
211                                 {
212                                         par[1] = "::" + results[i];
213                                         Utils->DoOneToOne(this->Instance->Config->GetSID(), "PUSH",par, source->server);
214                                 }
215                         }
216                 }
217                 else
218                 {
219                         /* Pass it on */
220                         User* source = this->Instance->FindNick(prefix);
221                         if (source)
222                                 Utils->DoOneToOne(source->uuid, "STATS", params, params[1]);
223                 }
224         }
225         return true;
226 }
227
228
229 /** Because the core won't let users or even SERVERS set +o,
230  * we use the OPERTYPE command to do this.
231  */
232 bool TreeSocket::OperType(const std::string &prefix, std::deque<std::string> &params)
233 {
234         if (params.size() != 1)
235                 return true;
236         std::string opertype = params[0];
237         User* u = this->Instance->FindNick(prefix);
238         if (u)
239         {
240                 if (!u->IsModeSet('o'))
241                         this->Instance->all_opers.push_back(u);
242                 u->modes[UM_OPERATOR] = 1;
243                 strlcpy(u->oper,opertype.c_str(),NICKMAX-1);
244                 Utils->DoOneToAllButSender(u->nick,"OPERTYPE",params,u->server);
245
246                 TreeServer* remoteserver = Utils->FindServer(u->server);
247                 bool dosend = true;
248
249                 if (this->Utils->quiet_bursts)
250                 {
251                         /*
252                          * If quiet bursts are enabled, and server is bursting or silent uline (i.e. services),
253                          * then do nothing. -- w00t
254                          */
255                         if (
256                                 this->bursting ||
257                                 Utils->FindRemoteBurstServer(remoteserver) ||
258                                 this->Instance->SilentULine(this->Instance->FindServerNamePtr(u->server))
259                            )
260                         {
261                                 dosend = false;
262                         }
263                 }
264
265                 if (dosend)
266                         this->Instance->SNO->WriteToSnoMask('o',"From %s: User %s (%s@%s) is now an IRC operator of type %s",u->server, u->nick,u->ident,u->host,irc::Spacify(opertype.c_str()));
267         }
268         return true;
269 }
270
271 /** Because Andy insists that services-compatible servers must
272  * implement SVSNICK and SVSJOIN, that's exactly what we do :p
273  */
274 bool TreeSocket::ForceNick(const std::string &prefix, std::deque<std::string> &params)
275 {
276         if (params.size() < 3)
277                 return true;
278
279         User* u = this->Instance->FindNick(params[0]);
280
281         if (u)
282         {
283                 Utils->DoOneToAllButSender(prefix,"SVSNICK",params,prefix);
284
285                 if (IS_LOCAL(u))
286                 {
287                         std::deque<std::string> par;
288                         par.push_back(params[1]);
289
290                         if (!u->ForceNickChange(params[1].c_str()))
291                         {
292                                 /* buh. UID them */
293                                 if (!u->ForceNickChange(u->uuid))
294                                 {
295                                         User::QuitUser(this->Instance, u, "Nickname collision");
296                                         return true;
297                                 }
298                         }
299
300                         u->age = atoi(params[2].c_str());
301                 }
302         }
303
304         return true;
305 }
306
307 bool TreeSocket::OperQuit(const std::string &prefix, std::deque<std::string> &params)
308 {
309         if (params.size() < 1)
310                 return true;
311
312         User* u = this->Instance->FindNick(prefix);
313
314         if (u)
315         {
316                 u->SetOperQuit(params[0]);
317                 params[0] = ":" + params[0];
318                 Utils->DoOneToAllButSender(prefix,"OPERQUIT",params,prefix);
319         }
320         return true;
321 }
322
323 bool TreeSocket::ServiceJoin(const std::string &prefix, std::deque<std::string> &params)
324 {
325         if (params.size() < 2)
326                 return true;
327
328         if (!this->Instance->IsChannel(params[1].c_str()))
329                 return true;
330
331         User* u = this->Instance->FindNick(params[0]);
332
333         if (u)
334         {
335                 /* only join if it's local, otherwise just pass it on! */
336                 if (IS_LOCAL(u))
337                         Channel::JoinUser(this->Instance, u, params[1].c_str(), false, "", Instance->Time());
338                 Utils->DoOneToAllButSender(prefix,"SVSJOIN",params,prefix);
339         }
340         return true;
341 }
342
343 bool TreeSocket::ServicePart(const std::string &prefix, std::deque<std::string> &params)
344 {
345         if (params.size() < 2)
346                 return true;
347
348         if (!this->Instance->IsChannel(params[1].c_str()))
349                 return true;
350
351         User* u = this->Instance->FindNick(params[0]);
352         Channel* c = this->Instance->FindChan(params[1]);
353
354         if (u)
355         {
356                 /* only part if it's local, otherwise just pass it on! */
357                 if (IS_LOCAL(u))
358                         if (!c->PartUser(u, "Services forced part"))
359                                 delete c;
360                 Utils->DoOneToAllButSender(prefix,"SVSPART",params,prefix);
361         }
362
363         return true;
364 }
365
366 bool TreeSocket::RemoteRehash(const std::string &prefix, std::deque<std::string> &params)
367 {
368         if (params.size() < 1)
369                 return false;
370
371         std::string servermask = params[0];
372
373         if (this->Instance->MatchText(this->Instance->Config->ServerName,servermask))
374         {
375                 this->Instance->SNO->WriteToSnoMask('l',"Remote rehash initiated by \002"+prefix+"\002.");
376                 this->Instance->RehashServer();
377                 Utils->ReadConfiguration(false);
378                 InitializeDisabledCommands(Instance->Config->DisabledCommands, Instance);
379         }
380         Utils->DoOneToAllButSender(prefix,"REHASH",params,prefix);
381         return true;
382 }
383
384 bool TreeSocket::RemoteKill(const std::string &prefix, std::deque<std::string> &params)
385 {        
386         if (params.size() != 2)
387                 return true;
388
389         User* who = this->Instance->FindNick(params[0]);
390
391         if (who)
392         {
393                 /* Prepend kill source, if we don't have one */          
394                 if (*(params[1].c_str()) != '[')
395                 {
396                         params[1] = "[" + prefix + "] Killed (" + params[1] +")";
397                 }
398                 std::string reason = params[1];
399                 params[1] = ":" + params[1];
400                 Utils->DoOneToAllButSender(prefix,"KILL",params,prefix);
401                 // NOTE: This is safe with kill hiding on, as RemoteKill is only reached if we have a server prefix.
402                 // in short this is not executed for USERS.
403                 who->Write(":%s KILL %s :%s (%s)", prefix.c_str(), who->nick, prefix.c_str(), reason.c_str());
404                 User::QuitUser(this->Instance,who,reason);
405         }
406         return true;
407 }
408
409 bool TreeSocket::LocalPong(const std::string &prefix, std::deque<std::string> &params)
410 {
411         if (params.size() < 1)
412                 return true;
413
414         if (params.size() == 1)
415         {
416                 TreeServer* ServerSource = Utils->FindServer(prefix);
417                 if (ServerSource)
418                 {
419                         ServerSource->SetPingFlag();
420                         timeval t;
421                         gettimeofday(&t, NULL);
422                         long ts = (t.tv_sec * 1000) + (t.tv_usec / 1000);
423                         ServerSource->rtt = ts - ServerSource->LastPingMsec;
424                 }
425         }
426         else
427         {
428                 std::string forwardto = params[1];
429                 if (forwardto == Instance->Config->GetSID() || forwardto == Instance->Config->ServerName)
430                 {
431                         /*
432                          * this is a PONG for us
433                          * if the prefix is a user, check theyre local, and if they are,
434                          * dump the PONG reply back to their fd. If its a server, do nowt.
435                          * Services might want to send these s->s, but we dont need to yet.
436                          */
437                         User* u = this->Instance->FindNick(prefix);
438                         if (u)
439                         {
440                                 u->WriteServ("PONG %s %s",params[0].c_str(),params[1].c_str());
441                         }
442                 }
443                 else
444                 {
445                         // not for us, pass it on :)
446                         Utils->DoOneToOne(prefix,"PONG",params,forwardto);
447                 }
448         }
449
450         return true;
451 }
452
453 bool TreeSocket::MetaData(const std::string &prefix, std::deque<std::string> &params)
454 {
455         if (params.size() < 2)
456                 return true;
457         else if (params.size() < 3)
458                 params.push_back("");
459         TreeServer* ServerSource = Utils->FindServer(prefix);
460         if (ServerSource)
461         {
462                 Utils->SetRemoteBursting(ServerSource, false);
463
464                 if (params[0] == "*")
465                 {
466                         FOREACH_MOD_I(this->Instance,I_OnDecodeMetaData,OnDecodeMetaData(TYPE_OTHER,NULL,params[1],params[2]));
467                 }
468                 else if (*(params[0].c_str()) == '#')
469                 {
470                         Channel* c = this->Instance->FindChan(params[0]);
471                         if (c)
472                         {
473                                 FOREACH_MOD_I(this->Instance,I_OnDecodeMetaData,OnDecodeMetaData(TYPE_CHANNEL,c,params[1],params[2]));
474                         }
475                 }
476                 else if (*(params[0].c_str()) != '#')
477                 {
478                         User* u = this->Instance->FindNick(params[0]);
479                         if (u)
480                         {
481                                 FOREACH_MOD_I(this->Instance,I_OnDecodeMetaData,OnDecodeMetaData(TYPE_USER,u,params[1],params[2]));
482                         }
483                 }
484         }
485
486         params[2] = ":" + params[2];
487         Utils->DoOneToAllButSender(prefix,"METADATA",params,prefix);
488         return true;
489 }
490
491 bool TreeSocket::ServerVersion(const std::string &prefix, std::deque<std::string> &params)
492 {
493         if (params.size() < 1)
494                 return true;
495
496         TreeServer* ServerSource = Utils->FindServer(prefix);
497
498         if (ServerSource)
499         {
500                 ServerSource->SetVersion(params[0]);
501         }
502         params[0] = ":" + params[0];
503         Utils->DoOneToAllButSender(prefix,"VERSION",params,prefix);
504         return true;
505 }
506
507 bool TreeSocket::ChangeHost(const std::string &prefix, std::deque<std::string> &params)
508 {
509         if (params.size() < 1)
510                 return true;
511         User* u = this->Instance->FindNick(prefix);
512
513         if (u)
514         {
515                 u->ChangeDisplayedHost(params[0].c_str());
516                 Utils->DoOneToAllButSender(prefix,"FHOST",params,u->server);
517         }
518         return true;
519 }
520
521 bool TreeSocket::AddLine(const std::string &prefix, std::deque<std::string> &params)
522 {
523         if (params.size() < 6)
524                 return true;
525
526         XLineFactory* xlf = Instance->XLines->GetFactory(params[0]);
527
528         if (!xlf)
529                 return false;
530
531         XLine* xl = xlf->Generate(Instance->Time(), atoi(params[4].c_str()), params[2].c_str(), params[5].c_str(), params[1].c_str());
532         xl->SetCreateTime(atoi(params[3].c_str()));
533         if (Instance->XLines->AddLine(xl,NULL))
534         {
535                 if (xl->duration)
536                 {
537                         this->Instance->SNO->WriteToSnoMask('x',"%s added %s%s on %s to expire on %s (%s).",prefix.c_str(),params[0].c_str(),params[0].length() == 1 ? "LINE" : "",
538                                         params[1].c_str(),Instance->TimeString(xl->expiry).c_str(),params[5].c_str());
539                 }
540                 else
541                 {
542                         this->Instance->SNO->WriteToSnoMask('x',"%s added permenant %s%s on %s (%s).",prefix.c_str(),params[0].c_str(),params[0].length() == 1 ? "LINE" : "",
543                                         params[1].c_str(),params[5].c_str());
544                 }
545                 params[5] = ":" + params[5];
546
547                 User* u = Instance->FindNick(prefix);
548                 Utils->DoOneToAllButSender(prefix, "ADDLINE", params, u ? u->server : prefix);
549         }
550         else
551                 delete xl;
552
553         if (!this->bursting)
554         {
555                 Instance->XLines->ApplyLines();
556         }
557
558         return true;
559 }
560
561 bool TreeSocket::DelLine(const std::string &prefix, std::deque<std::string> &params)
562 {
563         if (params.size() < 2)
564                 return true;
565
566         User* user = Instance->FindNick(prefix);
567
568         /* NOTE: No check needed on 'user', this function safely handles NULL */
569         if (Instance->XLines->DelLine(params[0].c_str(), params[1], user))
570         {
571                 this->Instance->SNO->WriteToSnoMask('x',"%s removed %s%s on %s.", prefix.c_str(),
572                                 params[0].c_str(), params[0].length() == 1 ? "LINE" : "", params[1].c_str());
573                 Utils->DoOneToAllButSender(prefix,"DELLINE", params, prefix);
574         }
575         return true;
576 }
577
578 bool TreeSocket::ChangeName(const std::string &prefix, std::deque<std::string> &params)
579 {
580         if (params.size() < 1)
581                 return true;
582         User* u = this->Instance->FindNick(prefix);
583         if (u)
584         {
585                 u->ChangeName(params[0].c_str());
586                 params[0] = ":" + params[0];
587                 Utils->DoOneToAllButSender(prefix,"FNAME",params,u->server);
588         }
589         return true;
590 }
591
592 bool TreeSocket::Whois(const std::string &prefix, std::deque<std::string> &params)
593 {
594         if (params.size() < 1)
595                 return true;
596         User* u = this->Instance->FindNick(prefix);
597         if (u)
598         {
599                 // an incoming request
600                 if (params.size() == 1)
601                 {
602                         User* x = this->Instance->FindNick(params[0]);
603                         if ((x) && (IS_LOCAL(x)))
604                         {
605                                 User* x = this->Instance->FindNick(params[0]);
606                                 char signon[MAXBUF];
607                                 char idle[MAXBUF];
608                                 snprintf(signon, MAXBUF, "%lu", (unsigned long)x->signon);
609                                 snprintf(idle, MAXBUF, "%lu", (unsigned long)abs((x->idle_lastmsg) - Instance->Time(true)));
610                                 std::deque<std::string> par;
611                                 par.push_back(prefix);
612                                 par.push_back(signon);
613                                 par.push_back(idle);
614                                 // ours, we're done, pass it BACK
615                                 Utils->DoOneToOne(params[0], "IDLE", par, u->server);
616                         }
617                         else
618                         {
619                                 // not ours pass it on
620                                 if (x)
621                                         Utils->DoOneToOne(prefix, "IDLE", params, x->server);
622                         }
623                 }
624                 else if (params.size() == 3)
625                 {
626                         std::string who_did_the_whois = params[0];
627                         User* who_to_send_to = this->Instance->FindNick(who_did_the_whois);
628                         if ((who_to_send_to) && (IS_LOCAL(who_to_send_to)))
629                         {
630                                 // an incoming reply to a whois we sent out
631                                 std::string nick_whoised = prefix;
632                                 unsigned long signon = atoi(params[1].c_str());
633                                 unsigned long idle = atoi(params[2].c_str());
634                                 if ((who_to_send_to) && (IS_LOCAL(who_to_send_to)))
635                                 {
636                                         do_whois(this->Instance, who_to_send_to, u, signon, idle, nick_whoised.c_str());
637                                 }
638                         }
639                         else
640                         {
641                                 // not ours, pass it on
642                                 if (who_to_send_to)
643                                         Utils->DoOneToOne(prefix, "IDLE", params, who_to_send_to->server);
644                         }
645                 }
646         }
647         return true;
648 }
649
650 bool TreeSocket::Push(const std::string &prefix, std::deque<std::string> &params)
651 {
652         if (params.size() < 2)
653                 return true;
654         User* u = this->Instance->FindNick(params[0]);
655         if (!u)
656                 return true;
657         if (IS_LOCAL(u))
658         {
659                 u->Write(params[1]);
660         }
661         else
662         {
663                 // continue the raw onwards
664                 params[1] = ":" + params[1];
665                 Utils->DoOneToOne(prefix,"PUSH",params,u->server);
666         }
667         return true;
668 }
669
670 bool TreeSocket::HandleSetTime(const std::string &prefix, std::deque<std::string> &params)
671 {
672         if (!params.size() || !Utils->EnableTimeSync)
673                 return true;
674
675         bool force = false;
676
677         if ((params.size() == 2) && (params[1] == "FORCE"))
678                 force = true;
679
680         time_t them = atoi(params[0].c_str());
681         time_t us = Instance->Time(false);
682
683         time_t diff = them - us;
684
685         Utils->DoOneToAllButSender(prefix, "TIMESET", params, prefix);
686
687         if (force || (them != us))
688         {
689                 time_t old = Instance->SetTimeDelta(diff);
690                 Instance->Log(DEBUG, "TS (diff %d) from %s applied (old delta was %d)", diff, prefix.c_str(), old);
691         }
692
693         return true;
694 }
695
696 bool TreeSocket::Time(const std::string &prefix, std::deque<std::string> &params)
697 {
698         // :source.server TIME remote.server sendernick
699         // :remote.server TIME source.server sendernick TS
700         if (params.size() == 2)
701         {
702                 // someone querying our time?
703                 if (this->Instance->Config->ServerName == params[0] || this->Instance->Config->GetSID() == params[0])
704                 {
705                         User* u = this->Instance->FindNick(params[1]);
706                         if (u)
707                         {
708                                 params.push_back(ConvToStr(Instance->Time(false)));
709                                 params[0] = prefix;
710                                 Utils->DoOneToOne(this->Instance->Config->GetSID(),"TIME",params,params[0]);
711                         }
712                 }
713                 else
714                 {
715                         // not us, pass it on
716                         User* u = this->Instance->FindNick(params[1]);
717                         if (u)
718                                 Utils->DoOneToOne(prefix,"TIME",params,params[0]);
719                 }
720         }
721         else if (params.size() == 3)
722         {
723                 // a response to a previous TIME
724                 User* u = this->Instance->FindNick(params[1]);
725                 if ((u) && (IS_LOCAL(u)))
726                 {
727                         time_t rawtime = atol(params[2].c_str());
728                         struct tm * timeinfo;
729                         timeinfo = localtime(&rawtime);
730                         char tms[26];
731                         snprintf(tms,26,"%s",asctime(timeinfo));
732                         tms[24] = 0;
733                         u->WriteServ("391 %s %s :%s",u->nick,prefix.c_str(),tms);
734                 }
735                 else
736                 {
737                         if (u)
738                                 Utils->DoOneToOne(prefix,"TIME",params,u->server);
739                 }
740         }
741         return true;
742 }
743
744 bool TreeSocket::LocalPing(const std::string &prefix, std::deque<std::string> &params)
745 {
746         if (params.size() < 1)
747                 return true;
748         if (params.size() == 1)
749         {
750                 std::string stufftobounce = params[0];
751                 this->WriteLine(std::string(":")+this->Instance->Config->GetSID()+" PONG "+stufftobounce);
752                 return true;
753         }
754         else
755         {
756                 std::string forwardto = params[1];
757                 if (forwardto == this->Instance->Config->ServerName || forwardto == this->Instance->Config->GetSID())
758                 {
759                         // this is a ping for us, send back PONG to the requesting server
760                         params[1] = params[0];
761                         params[0] = forwardto;
762                         Utils->DoOneToOne(forwardto,"PONG",params,params[1]);
763                 }
764                 else
765                 {
766                         // not for us, pass it on :)
767                         Utils->DoOneToOne(prefix,"PING",params,forwardto);
768                 }
769                 return true;
770         }
771 }
772
773 /** TODO: This creates a total mess of output and needs to really use irc::modestacker.
774  */
775 bool TreeSocket::RemoveStatus(const std::string &prefix, std::deque<std::string> &params)
776 {
777         if (params.size() < 1)
778                 return true;
779         Channel* c = Instance->FindChan(params[0]);
780         if (c)
781         {
782                 for (char modeletter = 'A'; modeletter <= 'z'; modeletter++)
783                 {
784                         ModeHandler* mh = Instance->Modes->FindMode(modeletter, MODETYPE_CHANNEL);
785                         if (mh)
786                                 mh->RemoveMode(c);
787                 }
788         }
789         return true;
790 }
791
792 bool TreeSocket::RemoteServer(const std::string &prefix, std::deque<std::string> &params)
793 {
794         if (params.size() < 5)
795         {
796                 SendError("Protocol error - Missing SID");
797                 return false;
798         }
799
800         std::string servername = params[0];
801         std::string password = params[1];
802         // hopcount is not used for a remote server, we calculate this ourselves
803         std::string sid = params[3];
804         std::string description = params[4];
805         TreeServer* ParentOfThis = Utils->FindServer(prefix);
806         if (!ParentOfThis)
807         {
808                 this->SendError("Protocol error - Introduced remote server from unknown server "+prefix);
809                 return false;
810         }
811         if (!Utils->IsSID(sid))
812         {
813                 this->SendError("Invalid format server ID: "+sid+"!");
814                 return false;
815         }
816         TreeServer* CheckDupe = Utils->FindServer(servername);
817         if (CheckDupe)
818         {
819                 this->SendError("Server "+servername+" already exists!");
820                 this->Instance->SNO->WriteToSnoMask('l',"Server \2"+servername+"\2 being introduced from \2" + prefix + "\2 denied, already exists. Closing link with " + prefix);
821                 return false;
822         }
823
824         Link* lnk = Utils->FindLink(servername);
825
826         TreeServer *Node = new TreeServer(this->Utils, this->Instance, servername, description, sid, ParentOfThis,NULL, lnk ? lnk->Hidden : false);
827
828         if (Node->DuplicateID())
829         {
830                 this->SendError("Server ID "+sid+" already exists on the network!");
831                 this->Instance->SNO->WriteToSnoMask('l',"Server \2"+servername+"\2 being introduced from \2" + prefix + "\2 denied, server ID already exists on the network. Closing link with " + prefix);
832                 return false;
833         }
834
835         ParentOfThis->AddChild(Node);
836         params[4] = ":" + params[4];
837         Utils->SetRemoteBursting(Node, true);
838         Utils->DoOneToAllButSender(prefix,"SERVER",params,prefix);
839         this->Instance->SNO->WriteToSnoMask('l',"Server \002"+prefix+"\002 introduced server \002"+servername+"\002 ("+description+")");
840         return true;
841 }
842
843 bool TreeSocket::ComparePass(const std::string &ours, const std::string &theirs)
844 {
845         if ((!strncmp(ours.c_str(), "HMAC-SHA256:", 12)) || (!strncmp(theirs.c_str(), "HMAC-SHA256:", 12)))
846         {
847                 /* One or both of us specified hmac sha256, but we don't have sha256 module loaded!
848                  * We can't allow this password as valid.
849                  */
850                 if (!Instance->Modules->Find("m_sha256.so") || !Utils->ChallengeResponse)
851                                 return false;
852                 else
853                         /* Straight string compare of hashes */
854                         return ours == theirs;
855         }
856         else
857                 /* Straight string compare of plaintext */
858                 return ours == theirs;
859 }
860
861 bool TreeSocket::Outbound_Reply_Server(std::deque<std::string> &params)
862 {
863         if (params.size() < 5)
864         {
865                 SendError("Protocol error - Missing SID");
866                 return false;
867         }
868
869         irc::string servername = params[0].c_str();
870         std::string sname = params[0];
871         std::string password = params[1];
872         std::string sid = params[3];
873         std::string description = params[4];
874         int hops = atoi(params[2].c_str());
875
876         this->InboundServerName = sname;
877         this->InboundDescription = description;
878         this->InboundSID = sid;
879
880         if (!sentcapab)
881                 this->SendCapabilities();
882
883         if (hops)
884         {
885                 this->SendError("Server too far away for authentication");
886                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, server is too far away for authentication");
887                 return false;
888         }
889
890         if (!Utils->IsSID(sid))
891         {
892                 this->SendError("Invalid format server ID: "+sid+"!");
893                 return false;
894         }
895
896         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
897         {
898                 if ((x->Name == servername) && ((ComparePass(this->MakePass(x->RecvPass,this->GetOurChallenge()),password)) || (x->RecvPass == password && (this->GetTheirChallenge().empty()))))
899                 {
900                         TreeServer* CheckDupe = Utils->FindServer(sname);
901                         if (CheckDupe)
902                         {
903                                 this->SendError("Server "+sname+" already exists on server "+CheckDupe->GetParent()->GetName()+"!");
904                                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName());
905                                 return false;
906                         }
907                         // Begin the sync here. this kickstarts the
908                         // other side, waiting in WAIT_AUTH_2 state,
909                         // into starting their burst, as it shows
910                         // that we're happy.
911                         this->LinkState = CONNECTED;
912                         // we should add the details of this server now
913                         // to the servers tree, as a child of the root
914                         // node.
915
916                         TreeServer *Node = new TreeServer(this->Utils, this->Instance, sname, description, sid, Utils->TreeRoot, this, x->Hidden);
917
918                         if (Node->DuplicateID())
919                         {
920                                 this->SendError("Server ID "+sid+" already exists on the network!");
921                                 this->Instance->SNO->WriteToSnoMask('l',"Server \2"+assign(servername)+"\2 being introduced denied, server ID already exists on the network. Closing link.");
922                                 return false;
923                         }
924
925                         Utils->TreeRoot->AddChild(Node);
926                         params[4] = ":" + params[4];
927                         Utils->DoOneToAllButSender(Instance->Config->GetSID(),"SERVER",params,sname);
928                         this->bursting = true;
929                         this->DoBurst(Node);
930                         return true;
931                 }
932         }
933         this->SendError("Invalid credentials");
934         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, invalid link credentials");
935         return false;
936 }
937
938 bool TreeSocket::Inbound_Server(std::deque<std::string> &params)
939 {
940         if (params.size() < 5)
941         {
942                 SendError("Protocol error - Missing SID");
943                 return false;
944         }
945
946         irc::string servername = params[0].c_str();
947         std::string sname = params[0];
948         std::string password = params[1];
949         std::string sid = params[3];
950         std::string description = params[4];
951         int hops = atoi(params[2].c_str());
952
953         this->InboundServerName = sname;
954         this->InboundDescription = description;
955         this->InboundSID = sid;
956
957         if (!sentcapab)
958                 this->SendCapabilities();
959
960         if (hops)
961         {
962                 this->SendError("Server too far away for authentication");
963                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, server is too far away for authentication");
964                 return false;
965         }
966
967         if (!Utils->IsSID(sid))
968         {
969                 this->SendError("Invalid format server ID: "+sid+"!");
970                 return false;
971         }
972
973         for (std::vector<Link>::iterator x = Utils->LinkBlocks.begin(); x < Utils->LinkBlocks.end(); x++)
974         {
975                 if ((x->Name == servername) && ((ComparePass(this->MakePass(x->RecvPass,this->GetOurChallenge()),password) || x->RecvPass == password && (this->GetTheirChallenge().empty()))))
976                 {
977                         /* First check for instances of the server that are waiting between the inbound and outbound SERVER command */
978                         TreeSocket* CheckDupeSocket = Utils->FindBurstingServer(sname);
979                         if (CheckDupeSocket)
980                         {
981                                 /* If we find one, we abort the link to prevent a race condition */
982                                 this->SendError("Negotiation collision");
983                                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists in a negotiating state.");
984                                 CheckDupeSocket->SendError("Negotiation collision");
985                                 Instance->SE->DelFd(CheckDupeSocket);
986                                 CheckDupeSocket->Close();
987                                 return false;
988                         }
989                         /* Check for fully initialized instances of the server by id */
990                         Instance->Log(DEBUG,"Looking for dupe SID %s", sid.c_str());
991                         TreeServer* CheckDupeSID = Utils->FindServerID(sid);
992                         if (CheckDupeSID)
993                         {
994                                 this->SendError("Server ID "+CheckDupeSID->GetID()+" already exists on server "+CheckDupeSID->GetName()+"!");
995                                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, server ID '"+CheckDupeSID->GetID()+
996                                                 "' already exists on server "+CheckDupeSID->GetName());
997                                 return false;
998                         }
999                         /* Now check for fully initialized instances of the server by name */
1000                         TreeServer* CheckDupe = Utils->FindServer(sname);
1001                         if (CheckDupe)
1002                         {
1003                                 this->SendError("Server "+sname+" already exists on server "+CheckDupe->GetParent()->GetName()+"!");
1004                                 this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, already exists on server "+CheckDupe->GetParent()->GetName());
1005                                 return false;
1006                         }
1007                         this->Instance->SNO->WriteToSnoMask('l',"Verified incoming server connection from \002"+sname+"\002["+(x->HiddenFromStats ? "<hidden>" : this->GetIP())+"] ("+description+")");
1008                         if (this->Hook)
1009                         {
1010                                 std::string name = BufferedSocketNameRequest((Module*)Utils->Creator, this->Hook).Send();
1011                                 this->Instance->SNO->WriteToSnoMask('l',"Connection from \2"+sname+"\2["+(x->HiddenFromStats ? "<hidden>" : this->GetIP())+"] using transport \2"+name+"\2");
1012                         }
1013
1014                         Utils->AddBurstingServer(sname,this);
1015
1016                         // this is good. Send our details: Our server name and description and hopcount of 0,
1017                         // along with the sendpass from this block.
1018                         this->WriteLine(std::string("SERVER ")+this->Instance->Config->ServerName+" "+this->MakePass(x->SendPass, this->GetTheirChallenge())+" 0 "+Instance->Config->GetSID()+" :"+this->Instance->Config->ServerDesc);
1019                         // move to the next state, we are now waiting for THEM.
1020                         this->LinkState = WAIT_AUTH_2;
1021                         return true;
1022                 }
1023         }
1024         this->SendError("Invalid credentials");
1025         this->Instance->SNO->WriteToSnoMask('l',"Server connection from \2"+sname+"\2 denied, invalid link credentials");
1026         return false;
1027 }
1028
1029 void TreeSocket::Split(const std::string &line, std::deque<std::string> &n)
1030 {
1031         n.clear();
1032         irc::tokenstream tokens(line);
1033         std::string param;
1034         while (tokens.GetToken(param))
1035         {
1036                 n.push_back(param);
1037         }
1038         return;
1039 }
1040
1041 bool TreeSocket::ProcessLine(std::string &line)
1042 {
1043         std::deque<std::string> params;
1044         irc::string command;
1045         std::string prefix;
1046
1047         line = line.substr(0, line.find_first_of("\r\n"));
1048
1049         if (line.empty())
1050                 return true;
1051
1052         Instance->Log(DEBUG, "S[%d] I %s", this->GetFd(), line.c_str());
1053
1054         this->Split(line.c_str(),params);
1055         
1056         if (params.empty())
1057                 return true;
1058         
1059         if ((params[0][0] == ':') && (params.size() > 1))
1060         {
1061                 prefix = params[0].substr(1);
1062                 params.pop_front();
1063         }
1064         command = params[0].c_str();
1065         params.pop_front();
1066         switch (this->LinkState)
1067         {
1068                 TreeServer* Node;
1069
1070                 case WAIT_AUTH_1:
1071                         // Waiting for SERVER command from remote server. Server initiating
1072                         // the connection sends the first SERVER command, listening server
1073                         // replies with theirs if its happy, then if the initiator is happy,
1074                         // it starts to send its net sync, which starts the merge, otherwise
1075                         // it sends an ERROR.
1076                         if (command == "PASS")
1077                         {
1078                                 /* Silently ignored */
1079                         }
1080                         else if (command == "SERVER")
1081                         {
1082                                 return this->Inbound_Server(params);
1083                         }
1084                         else if (command == "ERROR")
1085                         {
1086                                 return this->Error(params);
1087                         }
1088                         else if (command == "USER")
1089                         {
1090                                 this->SendError("Client connections to this port are prohibited.");
1091                                 return false;
1092                         }
1093                         else if (command == "CAPAB")
1094                         {
1095                                 return this->Capab(params);
1096                         }
1097                         else
1098                         {
1099                                 irc::string error = "Invalid command in negotiation phase: " + command;
1100                                 this->SendError(assign(error));
1101                                 return false;
1102                         }
1103                 break;
1104                 case WAIT_AUTH_2:
1105                         // Waiting for start of other side's netmerge to say they liked our
1106                         // password.
1107                         if (command == "SERVER")
1108                         {
1109                                 // cant do this, they sent it to us in the WAIT_AUTH_1 state!
1110                                 // silently ignore.
1111                                 return true;
1112                         }
1113                         else if (command == "BURST")
1114                         {
1115                                 if (params.size() && Utils->EnableTimeSync)
1116                                 {
1117                                         bool we_have_delta = (Instance->Time(false) != Instance->Time(true));
1118                                         time_t them = atoi(params[0].c_str());
1119                                         time_t delta = them - Instance->Time(false);
1120                                         if ((delta < -600) || (delta > 600))
1121                                         {
1122                                                 Instance->SNO->WriteToSnoMask('l',"\2ERROR\2: Your clocks are out by %d seconds (this is more than five minutes). Link aborted, \2PLEASE SYNC YOUR CLOCKS!\2",abs(delta));
1123                                                 SendError("Your clocks are out by "+ConvToStr(abs(delta))+" seconds (this is more than five minutes). Link aborted, PLEASE SYNC YOUR CLOCKS!");
1124                                                 return false;
1125                                         }
1126                                         else if ((delta < -30) || (delta > 30))
1127                                         {
1128                                                 Instance->SNO->WriteToSnoMask('l',"\2WARNING\2: Your clocks are out by %d seconds. Please consider synching your clocks.", abs(delta));
1129                                         }
1130
1131                                         if (!Utils->MasterTime && !we_have_delta)
1132                                         {
1133                                                 this->Instance->SetTimeDelta(delta);
1134                                                 // Send this new timestamp to any other servers
1135                                                 Utils->DoOneToMany(Instance->Config->GetSID(), "TIMESET", params);
1136                                         }
1137                                 }
1138                                 this->LinkState = CONNECTED;
1139                                 Link* lnk = Utils->FindLink(InboundServerName);
1140
1141                                 Node = new TreeServer(this->Utils, this->Instance, InboundServerName, InboundDescription, InboundSID, Utils->TreeRoot, this, lnk ? lnk->Hidden : false);
1142
1143                                 if (Node->DuplicateID())
1144                                 {
1145                                         this->SendError("Server ID "+InboundSID+" already exists on the network!");
1146                                         this->Instance->SNO->WriteToSnoMask('l',"Server \2"+InboundServerName+"\2 being introduced from \2" + prefix + "\2 denied, server ID already exists on the network. Closing link.");
1147                                         return false;
1148                                 }
1149                                 Utils->DelBurstingServer(this);
1150                                 Utils->TreeRoot->AddChild(Node);
1151                                 params.clear();
1152                                 params.push_back(InboundServerName);
1153                                 params.push_back("*");
1154                                 params.push_back("1");
1155                                 params.push_back(InboundSID);
1156                                 params.push_back(":"+InboundDescription);
1157                                 Utils->DoOneToAllButSender(Instance->Config->GetSID(),"SERVER",params,InboundServerName);
1158                                 this->bursting = true;
1159                                 this->DoBurst(Node);
1160                         }
1161                         else if (command == "ERROR")
1162                         {
1163                                 return this->Error(params);
1164                         }
1165                         else if (command == "CAPAB")
1166                         {
1167                                 return this->Capab(params);
1168                         }
1169
1170                 break;
1171                 case LISTENER:
1172                         this->SendError("Internal error -- listening socket accepted its own descriptor!!!");
1173                         return false;
1174                 break;
1175                 case CONNECTING:
1176                         if (command == "SERVER")
1177                         {
1178                                 // another server we connected to, which was in WAIT_AUTH_1 state,
1179                                 // has just sent us their credentials. If we get this far, theyre
1180                                 // happy with OUR credentials, and they are now in WAIT_AUTH_2 state.
1181                                 // if we're happy with this, we should send our netburst which
1182                                 // kickstarts the merge.
1183                                 return this->Outbound_Reply_Server(params);
1184                         }
1185                         else if (command == "ERROR")
1186                         {
1187                                 return this->Error(params);
1188                         }
1189                         else if (command == "CAPAB")
1190                         {
1191                                 return this->Capab(params);
1192                         }
1193                 break;
1194                 case CONNECTED:
1195                         // This is the 'authenticated' state, when all passwords
1196                         // have been exchanged and anything past this point is taken
1197                         // as gospel.
1198
1199                         if (!prefix.empty())
1200                         {
1201                                 std::string direction = prefix;
1202
1203                                 User *t = this->Instance->FindUUID(prefix);
1204                                 if (t)
1205                                 {
1206                                         direction = t->server;
1207                                 }
1208
1209                                 TreeServer* route_back_again = Utils->BestRouteTo(direction);
1210                                 if ((!route_back_again) || (route_back_again->GetSocket() != this))
1211                                 {
1212                                         if (route_back_again)
1213                                                 Instance->Log(DEBUG,"Protocol violation: Fake direction in command '%s' from connection '%s'",line.c_str(),this->GetName().c_str());
1214                                         return true;
1215                                 }
1216                                 /* Fix by brain:
1217                                  * When there is activity on the socket, reset the ping counter so
1218                                  * that we're not wasting bandwidth pinging an active server.
1219                                  */
1220                                 route_back_again->SetNextPingTime(time(NULL) + Utils->PingFreq);
1221                                 route_back_again->SetPingFlag();
1222                         }
1223                         else
1224                         {
1225                                 TreeServer* n = Utils->FindServer(GetName());
1226                                 if (n)
1227                                         prefix = n->GetID();
1228                                 else
1229                                         prefix = GetName();
1230                         }
1231
1232                         if ((command == "MODE") && (params.size() >= 2))
1233                         {
1234                                 Channel* channel = Instance->FindChan(params[0]);
1235                                 if (channel)
1236                                 {
1237                                         User* x = Instance->FindNick(prefix);
1238                                         if (x)
1239                                         {
1240                                                 if (warned.find(x->server) == warned.end())
1241                                                 {
1242                                                         Instance->Log(DEFAULT,"WARNING: I revceived modes '%s' from another server '%s'. This is not compliant with InspIRCd. Please check that server for bugs.", params[1].c_str(), x->server);
1243                                                         Instance->SNO->WriteToSnoMask('d', "WARNING: The server %s is sending nonstandard modes: '%s MODE %s' where FMODE should be used, and may cause desyncs.", x->server, x->nick, params[1].c_str());
1244                                                         warned[x->server] = x->nick;
1245                                                 }
1246                                         }
1247                                 }
1248                         }
1249
1250                         if (command == "SVSMODE")
1251                         {
1252                                 /* Services expects us to implement
1253                                  * SVSMODE. In inspircd its the same as
1254                                  * MODE anyway.
1255                                  */
1256                                 command = "MODE";
1257                         }
1258                         std::string target;
1259                         /* Yes, know, this is a mess. Its reasonably fast though as we're
1260                          * working with std::string here.
1261                          */
1262                         if (command == "UID")
1263                         {
1264                                 return this->ParseUID(prefix, params);
1265                         }
1266                         else if (command == "FJOIN")
1267                         {
1268                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1269                                 if (ServerSource)
1270                                         Utils->SetRemoteBursting(ServerSource, false);
1271                                 return this->ForceJoin(prefix,params);
1272                         }
1273                         else if (command == "STATS")
1274                         {
1275                                 return this->Stats(prefix, params);
1276                         }
1277                         else if (command == "MOTD")
1278                         {
1279                                 return this->Motd(prefix, params);
1280                         }
1281                         else if (command == "KILL" && Utils->IsServer(prefix))
1282                         {
1283                                 return this->RemoteKill(prefix,params);
1284                         }
1285                         else if (command == "MODULES")
1286                         {
1287                                 return this->Modules(prefix, params);
1288                         }
1289                         else if (command == "ADMIN")
1290                         {
1291                                 return this->Admin(prefix, params);
1292                         }
1293                         else if (command == "SERVER")
1294                         {
1295                                 return this->RemoteServer(prefix,params);
1296                         }
1297                         else if (command == "ERROR")
1298                         {
1299                                 return this->Error(params);
1300                         }
1301                         else if (command == "OPERTYPE")
1302                         {
1303                                 return this->OperType(prefix,params);
1304                         }
1305                         else if (command == "FMODE")
1306                         {
1307                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1308                                 if (ServerSource)
1309                                         Utils->SetRemoteBursting(ServerSource, false);
1310                                 return this->ForceMode(prefix,params);
1311                         }
1312                         else if (command == "FTOPIC")
1313                         {
1314                                 return this->ForceTopic(prefix,params);
1315                         }
1316                         else if (command == "REHASH")
1317                         {
1318                                 return this->RemoteRehash(prefix,params);
1319                         }
1320                         else if (command == "METADATA")
1321                         {
1322                                 return this->MetaData(prefix,params);
1323                         }
1324                         else if (command == "REMSTATUS")
1325                         {
1326                                 return this->RemoveStatus(prefix,params);
1327                         }
1328                         else if (command == "PING")
1329                         {
1330                                 /*
1331                                  * We just got a ping from a server that's bursting.
1332                                  * This can't be right, so set them to not bursting, and
1333                                  * apply their lines.
1334                                  */
1335                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1336                                 if (ServerSource)
1337                                         Utils->SetRemoteBursting(ServerSource, false);
1338
1339                                 if (this->bursting)
1340                                 {
1341                                         this->bursting = false;
1342                                         Instance->XLines->ApplyLines();
1343                                 }
1344
1345                                 return this->LocalPing(prefix,params);
1346                         }
1347                         else if (command == "PONG")
1348                         {
1349                                 /*
1350                                  * We just got a pong from a server that's bursting.
1351                                  * This can't be right, so set them to not bursting, and
1352                                  * apply their lines.
1353                                  */
1354                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1355                                 if (ServerSource)
1356                                         Utils->SetRemoteBursting(ServerSource, false);
1357
1358                                 if (this->bursting)
1359                                 {
1360                                         this->bursting = false;
1361                                         Instance->XLines->ApplyLines();
1362                                 }
1363
1364                                 return this->LocalPong(prefix,params);
1365                         }
1366                         else if (command == "VERSION")
1367                         {
1368                                 return this->ServerVersion(prefix,params);
1369                         }
1370                         else if (command == "FHOST")
1371                         {
1372                                 return this->ChangeHost(prefix,params);
1373                         }
1374                         else if (command == "FNAME")
1375                         {
1376                                 return this->ChangeName(prefix,params);
1377                         }
1378                         else if (command == "ADDLINE")
1379                         {
1380                                 TreeServer* ServerSource = Utils->FindServer(prefix);
1381                                 if (ServerSource)
1382                                         Utils->SetRemoteBursting(ServerSource, false);
1383                                 return this->AddLine(prefix,params);
1384                         }
1385                         else if (command == "DELLINE")
1386                         {
1387                                 return this->DelLine(prefix,params);
1388                         }
1389                         else if (command == "SVSNICK")
1390                         {
1391                                 return this->ForceNick(prefix,params);
1392                         }
1393                         else if (command == "OPERQUIT")
1394                         {
1395                                 return this->OperQuit(prefix,params);
1396                         }
1397                         else if (command == "IDLE")
1398                         {
1399                                 return this->Whois(prefix,params);
1400                         }
1401                         else if (command == "PUSH")
1402                         {
1403                                 return this->Push(prefix,params);
1404                         }
1405                         else if (command == "TIMESET")
1406                         {
1407                                 return this->HandleSetTime(prefix, params);
1408                         }
1409                         else if (command == "TIME")
1410                         {
1411                                 return this->Time(prefix,params);
1412                         }
1413                         else if ((command == "KICK") && (Utils->IsServer(prefix)))
1414                         {
1415                                 std::string sourceserv = this->myhost;
1416                                 if (params.size() == 3)
1417                                 {
1418                                         User* user = this->Instance->FindNick(params[1]);
1419                                         Channel* chan = this->Instance->FindChan(params[0]);
1420                                         if (user && chan)
1421                                         {
1422                                                 if (!chan->ServerKickUser(user, params[2].c_str(), false))
1423                                                         /* Yikes, the channels gone! */
1424                                                         delete chan;
1425                                         }
1426                                 }
1427                                 if (!this->InboundServerName.empty())
1428                                 {
1429                                         sourceserv = this->InboundServerName;
1430                                 }
1431                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
1432                         }
1433                         else if (command == "SVSJOIN")
1434                         {
1435                                 return this->ServiceJoin(prefix,params);
1436                         }
1437                         else if (command == "SVSPART")
1438                         {
1439                                 return this->ServicePart(prefix,params);
1440                         }
1441                         else if (command == "SQUIT")
1442                         {
1443                                 if (params.size() == 2)
1444                                 {
1445                                         this->Squit(Utils->FindServer(params[0]),params[1]);
1446                                 }
1447                                 return true;
1448                         }
1449                         else if (command == "OPERNOTICE")
1450                         {
1451                                 std::string sourceserv = this->myhost;
1452                                 if (!this->InboundServerName.empty())
1453                                         sourceserv = this->InboundServerName;
1454                                 if (params.size() >= 1)
1455                                         Instance->WriteOpers("*** From " + prefix + ": " + params[0]);
1456                                 return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
1457                         }
1458                         else if (command == "MODENOTICE")
1459                         {
1460                                 std::string sourceserv = this->myhost;
1461                                 if (!this->InboundServerName.empty())
1462                                         sourceserv = this->InboundServerName;
1463                                 if (params.size() >= 2)
1464                                 {
1465                                         Instance->WriteMode(params[0].c_str(), WM_AND, "*** From %s: %s", prefix.c_str(), params[1].c_str());
1466                                 }
1467                                 return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
1468                         }
1469                         else if (command == "SNONOTICE")
1470                         {
1471                                 std::string sourceserv = this->myhost;
1472                                 if (!this->InboundServerName.empty())
1473                                         sourceserv = this->InboundServerName;
1474                                 if (params.size() >= 2)
1475                                 {
1476                                         Instance->SNO->WriteToSnoMask(*(params[0].c_str()), "From " + prefix + ": "+ params[1]);
1477                                 }
1478                                 return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
1479                         }
1480                         else if (command == "ENDBURST")
1481                         {
1482                                 this->bursting = false;
1483                                 Instance->XLines->ApplyLines();
1484                                 std::string sourceserv = this->myhost;
1485                                 if (!this->InboundServerName.empty())
1486                                         sourceserv = this->InboundServerName;
1487                                 this->Instance->SNO->WriteToSnoMask('l',"Received end of netburst from \2%s\2",sourceserv.c_str());
1488
1489                                 Event rmode((char*)sourceserv.c_str(), (Module*)Utils->Creator, "new_server");
1490                                 rmode.Send(Instance);
1491
1492                                 return true;
1493                         }
1494                         else
1495                         {
1496                                 /*
1497                                  * Not a special s2s command. Emulate the user doing it.
1498                                  * This saves us having a huge ugly command parser again.
1499                                  */
1500                                 User *who = this->Instance->FindUUID(prefix);
1501
1502                                 std::string sourceserv = this->myhost;
1503                                 if (!this->InboundServerName.empty())
1504                                 {
1505                                         sourceserv = this->InboundServerName;
1506                                 }
1507                                 if ((!who) && (command == "MODE"))
1508                                 {
1509                                         if (Utils->IsServer(prefix))
1510                                         {
1511                                                 const char* modelist[MAXPARAMETERS];
1512                                                 for (size_t i = 0; i < params.size(); i++)
1513                                                         modelist[i] = params[i].c_str();
1514                                                 this->Instance->SendMode(modelist, params.size(), this->Instance->FakeClient);
1515                                                 /* Hot potato! pass it on! */
1516                                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
1517                                         }
1518                                 }
1519                                 if (who)
1520                                 {
1521                                         if (command == "NICK")
1522                                         {
1523                                                 if (params.size() != 2)
1524                                                 {
1525                                                         SendError("Protocol violation: NICK message without TS - :"+std::string(who->uuid)+" NICK "+params[0]);
1526                                                         return false;
1527                                                 }
1528                                                 /* Update timestamp on user when they change nicks */
1529                                                 who->age = atoi(params[1].c_str());
1530
1531                                                 /* On nick messages, check that the nick doesnt
1532                                                  * already exist here. If it does, kill their copy,
1533                                                  * and our copy.
1534                                                  */
1535                                                 User* x = this->Instance->FindNickOnly(params[0]);
1536                                                 if ((x) && (x != who))
1537                                                 {
1538                                                         int collideret = 0;
1539                                                         /* x is local, who is remote */
1540                                                         collideret = this->DoCollision(x, who->age, who->ident, who->GetIPString(), who->uuid);
1541                                                         if (collideret != 1)
1542                                                         {
1543                                                                 /*
1544                                                                  * Remote client lost, or both lost,
1545                                                                  * parsing this nickchange would be
1546                                                                  * pointless, as the incoming client's
1547                                                                  * server will soon recieve SVSNICK to
1548                                                                  * change its nick to its UID. :)
1549                                                                  *   -- w00t
1550                                                                  */
1551                                                                 return true;
1552                                                         }
1553 /*
1554 Old nickname collision logic..
1555                                                         std::deque<std::string> p;
1556                                                         p.push_back(params[0]);
1557                                                         p.push_back(":Nickname collision ("+prefix+" -> "+params[0]+")");
1558                                                         Utils->DoOneToMany(this->Instance->Config->ServerName,"KILL",p);
1559                                                         p.clear();
1560                                                         p.push_back(prefix);
1561                                                         p.push_back(":Nickname collision");
1562                                                         Utils->DoOneToMany(this->Instance->Config->ServerName,"KILL",p);
1563                                                         User::QuitUser(this->Instance,x,"Nickname collision ("+prefix+" -> "+params[0]+")");
1564                                                         User* y = this->Instance->FindNick(prefix);
1565                                                         if (y)
1566                                                         {
1567                                                                 User::QuitUser(this->Instance,y,"Nickname collision");
1568                                                         }
1569                                                         return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
1570 */
1571                                                 }
1572                                         }
1573                                         // its a user
1574                                         target = who->server;
1575                                         const char* strparams[127];
1576                                         for (unsigned int q = 0; q < params.size(); q++)
1577                                         {
1578                                                 strparams[q] = params[q].c_str();
1579                                         }
1580                                         switch (this->Instance->CallCommandHandler(command.c_str(), strparams, params.size(), who))
1581                                         {
1582                                                 case CMD_INVALID:
1583                                                         this->SendError("Unrecognised or malformed command '"+std::string(command.c_str())+"' -- possibly loaded mismatched modules");
1584                                                         return false;
1585                                                 break;
1586                                                 case CMD_FAILURE:
1587                                                         return true;
1588                                                 break;
1589                                                 default:
1590                                                         /* CMD_SUCCESS and CMD_USER_DELETED fall through here */
1591                                                 break;
1592                                         }
1593                                 }
1594                                 else
1595                                 {
1596                                         // its not a user. Its either a server, or somethings screwed up.
1597                                         if (Utils->IsServer(prefix))
1598                                                 target = this->Instance->Config->GetSID();
1599                                         else
1600                                                 return true;
1601                                 }
1602                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
1603
1604                         }
1605                         return true;
1606                 break;
1607         }
1608         return true;
1609 }
1610
1611 std::string TreeSocket::GetName()
1612 {
1613         std::string sourceserv = this->myhost;
1614         if (!this->InboundServerName.empty())
1615         {
1616                 sourceserv = this->InboundServerName;
1617         }
1618         return sourceserv;
1619 }
1620
1621 void TreeSocket::OnTimeout()
1622 {
1623         if (this->LinkState == CONNECTING)
1624         {
1625                 Utils->Creator->RemoteMessage(NULL, "CONNECT: Connection to \002%s\002 timed out.", myhost.c_str());
1626                 Link* MyLink = Utils->FindLink(myhost);
1627                 if (MyLink)
1628                         Utils->DoFailOver(MyLink);
1629         }
1630 }
1631
1632 void TreeSocket::OnClose()
1633 {
1634         if (this->LinkState == LISTENER)
1635                 return;
1636
1637         // Connection closed.
1638         // If the connection is fully up (state CONNECTED)
1639         // then propogate a netsplit to all peers.
1640         std::string quitserver = this->myhost;
1641         if (!this->InboundServerName.empty())
1642         {
1643                 quitserver = this->InboundServerName;
1644         }
1645         TreeServer* s = Utils->FindServer(quitserver);
1646         if (s)
1647         {
1648                 Squit(s,"Remote host closed the connection");
1649         }
1650
1651         if (!quitserver.empty())
1652         {
1653                 Utils->Creator->RemoteMessage(NULL,"Connection to '\2%s\2' failed.",quitserver.c_str());
1654                 time_t server_uptime = Instance->Time() - this->age;    
1655                 if (server_uptime)
1656                         Utils->Creator->RemoteMessage(NULL,"Connection to '\2%s\2' was established for %s", quitserver.c_str(), Utils->Creator->TimeToStr(server_uptime).c_str());
1657         }
1658 }
1659
1660 int TreeSocket::OnIncomingConnection(int newsock, char* ip)
1661 {
1662         /* To prevent anyone from attempting to flood opers/DDoS by connecting to the server port,
1663          * or discovering if this port is the server port, we don't allow connections from any
1664          * IPs for which we don't have a link block.
1665          */
1666         bool found = false;
1667
1668         found = (std::find(Utils->ValidIPs.begin(), Utils->ValidIPs.end(), ip) != Utils->ValidIPs.end());
1669         if (!found)
1670         {
1671                 for (vector<std::string>::iterator i = Utils->ValidIPs.begin(); i != Utils->ValidIPs.end(); i++)
1672                         if (irc::sockets::MatchCIDR(ip, (*i).c_str()))
1673                                 found = true;
1674
1675                 if (!found)
1676                 {
1677                         Utils->Creator->RemoteMessage(NULL,"Server connection from %s denied (no link blocks with that IP address)", ip);
1678                         Instance->SE->Close(newsock);
1679                         return false;
1680                 }
1681         }
1682
1683         TreeSocket* s = new TreeSocket(this->Utils, this->Instance, newsock, ip, this->Hook);
1684         s = s; /* Whinge whinge whinge, thats all GCC ever does. */
1685         return true;
1686 }