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