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