]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/m_spanningtree/treesocket2.cpp
e27c1d879739abba2f8e42e824edeb611ea15df0
[user/henk/code/inspircd.git] / src / modules / m_spanningtree / treesocket2.cpp
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
6  * See: http://www.inspircd.org/wiki/index.php/Credits
7  *
8  * This program is free but copyrighted software; see
9  *            the file COPYING for details.
10  *
11  * ---------------------------------------------------
12  */
13
14 #include "inspircd.h"
15 #include "commands/cmd_whois.h"
16 #include "commands/cmd_stats.h"
17 #include "socket.h"
18 #include "xline.h"
19 #include "transport.h"
20 #include "socketengine.h"
21
22 #include "m_spanningtree/main.h"
23 #include "m_spanningtree/utils.h"
24 #include "m_spanningtree/treeserver.h"
25 #include "m_spanningtree/link.h"
26 #include "m_spanningtree/treesocket.h"
27 #include "m_spanningtree/resolvers.h"
28 #include "m_spanningtree/handshaketimer.h"
29
30 /* $ModDep: m_spanningtree/resolvers.h m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/link.h m_spanningtree/treesocket.h m_spanningtree/handshaketimer.h */
31
32 void TreeSocket::WriteLine(std::string line)
33 {
34         ServerInstance->Logs->Log("m_spanningtree",DEBUG, "S[%d] O %s", this->GetFd(), line.c_str());
35         line.append("\r\n");
36         this->Write(line);
37 }
38
39
40 /* Handle ERROR command */
41 bool TreeSocket::Error(std::deque<std::string> &params)
42 {
43         if (params.size() < 1)
44                 return false;
45         this->ServerInstance->SNO->WriteToSnoMask('l',"ERROR from %s: %s",(!InboundServerName.empty() ? InboundServerName.c_str() : myhost.c_str()),params[0].c_str());
46         /* we will return false to cause the socket to close. */
47         return false;
48 }
49
50 void TreeSocket::Split(const std::string &line, std::deque<std::string> &n)
51 {
52         n.clear();
53         irc::tokenstream tokens(line);
54         std::string param;
55         while (tokens.GetToken(param))
56         {
57                 n.push_back(param);
58         }
59         return;
60 }
61
62 bool TreeSocket::ProcessLine(std::string &line)
63 {
64         std::deque<std::string> params;
65         irc::string command;
66         std::string prefix;
67
68         line = line.substr(0, line.find_first_of("\r\n"));
69
70         if (line.empty())
71                 return true;
72
73         ServerInstance->Logs->Log("m_spanningtree",DEBUG, "S[%d] I %s", this->GetFd(), line.c_str());
74
75         this->Split(line.c_str(),params);
76
77         if (params.empty())
78                 return true;
79
80         if ((params[0][0] == ':') && (params.size() > 1))
81         {
82                 prefix = params[0].substr(1);
83                 params.pop_front();
84
85                 if (prefix.empty())
86                 {
87                         this->SendError("BUG (?) Empty prefix recieved.");
88                         return false;
89                 }
90         }
91
92         command = params[0].c_str();
93         params.pop_front();
94
95         switch (this->LinkState)
96         {
97                 TreeServer* Node;
98
99                 case WAIT_AUTH_1:
100                         /*
101                          * State WAIT_AUTH_1:
102                          *  Waiting for SERVER command from remote server. Server initiating
103                          *  the connection sends the first SERVER command, listening server
104                          *  replies with theirs if its happy, then if the initiator is happy,
105                          *  it starts to send its net sync, which starts the merge, otherwise
106                          *  it sends an ERROR.
107                          */
108                         if (command == "PASS")
109                         {
110                                 /*
111                                  * Ignore this silently. Some services packages insist on sending PASS, even
112                                  * when it is not required (i.e. by us). We have to ignore this here, otherwise
113                                  * as it's an unknown command (effectively), it will cause the connection to be
114                                  * closed, which probably isn't what people want. -- w00t
115                                  */
116                         }
117                         else if (command == "SERVER")
118                         {
119                                 return this->Inbound_Server(params);
120                         }
121                         else if (command == "ERROR")
122                         {
123                                 return this->Error(params);
124                         }
125                         else if (command == "USER")
126                         {
127                                 this->SendError("Client connections to this port are prohibited.");
128                                 return false;
129                         }
130                         else if (command == "CAPAB")
131                         {
132                                 return this->Capab(params);
133                         }
134                         else
135                         {
136                                 // XXX ...wtf.
137                                 irc::string error = "Invalid command in negotiation phase: " + command;
138                                 this->SendError(assign(error));
139                                 return false;
140                         }
141                 break;
142                 case WAIT_AUTH_2:
143                         /*
144                          * State WAIT_AUTH_2:
145                          *  We have sent SERVER to the other side of the connection. Now we're waiting for them to start BURST.
146                          *  The other option at this stage of things, of course, is for them to close our connection thanks
147                          *  to invalid credentials.. -- w
148                          */
149                         if (command == "SERVER")
150                         {
151                                 /*
152                                  * Connection is either attempting to re-auth itself (stupid) or sending netburst without sending BURST.
153                                  * Both of these aren't allowable, so block them here. -- w
154                                  */
155                                 this->SendError("You may not re-authenticate or commence netburst without sending BURST.");
156                                 return true;
157                         }
158                         else if (command == "BURST")
159                         {
160                                 if (params.size())
161                                 {
162                                         time_t them = atoi(params[0].c_str());
163                                         time_t delta = them - ServerInstance->Time();
164                                         if ((delta < -600) || (delta > 600))
165                                         {
166                                                 ServerInstance->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((long)delta));
167                                                 SendError("Your clocks are out by "+ConvToStr(abs((long)delta))+" seconds (this is more than five minutes). Link aborted, PLEASE SYNC YOUR CLOCKS!");
168                                                 return false;
169                                         }
170                                         else if ((delta < -30) || (delta > 30))
171                                         {
172                                                 ServerInstance->SNO->WriteToSnoMask('l',"\2WARNING\2: Your clocks are out by %d seconds. Please consider synching your clocks.", abs((long)delta));
173                                         }
174                                 }
175                                 this->LinkState = CONNECTED;
176
177                                 Utils->timeoutlist.erase(this);
178
179                                 Link* lnk = Utils->FindLink(InboundServerName);
180
181                                 Node = new TreeServer(this->Utils, this->ServerInstance, InboundServerName, InboundDescription, InboundSID, Utils->TreeRoot, this, lnk ? lnk->Hidden : false);
182
183                                 if (Node->DuplicateID())
184                                 {
185                                         this->SendError("Server ID "+InboundSID+" already exists on the network!");
186                                         this->ServerInstance->SNO->WriteToSnoMask('l',"Server \2"+InboundServerName+"\2 being introduced from \2" + prefix + "\2 denied, server ID already exists on the network. Closing link.");
187                                         return false;
188                                 }
189
190                                 Utils->TreeRoot->AddChild(Node);
191                                 params.clear();
192                                 params.push_back(InboundServerName);
193                                 params.push_back("*");
194                                 params.push_back("1");
195                                 params.push_back(InboundSID);
196                                 params.push_back(":"+InboundDescription);
197                                 Utils->DoOneToAllButSender(ServerInstance->Config->GetSID(),"SERVER",params,InboundServerName);
198                                 Node->bursting = true;
199                                 this->DoBurst(Node);
200                         }
201                         else if (command == "ERROR")
202                         {
203                                 return this->Error(params);
204                         }
205                         else if (command == "CAPAB")
206                         {
207                                 return this->Capab(params);
208                         }
209
210                 break;
211                 case CONNECTING:
212                         /*
213                          * State CONNECTING:
214                          *  We're connecting (OUTGOING) to another server. They are in state WAIT_AUTH_1 until they verify
215                          *  our credentials, when they proceed into WAIT_AUTH_2 and send SERVER to us. We then send BURST
216                          *  + our netburst, which will put them into CONNECTED state. -- w
217                          */
218                         if (command == "SERVER")
219                         {
220                                 // Our credentials have been accepted, send netburst. (this puts US into the CONNECTED state)
221                                 return this->Outbound_Reply_Server(params);
222                         }
223                         else if (command == "ERROR")
224                         {
225                                 return this->Error(params);
226                         }
227                         else if (command == "CAPAB")
228                         {
229                                 return this->Capab(params);
230                         }
231                 break;
232                 case CONNECTED:
233                         /*
234                         * State CONNECTED:
235                          *  Credentials have been exchanged, we've gotten their 'BURST' (or sent ours).
236                          *  Anything from here on should be accepted a little more reasonably.
237                          */
238                         if (!prefix.empty())
239                         {
240                                 /*
241                                  * Check for fake direction here, and drop any instances that are found.
242                                  * What is fake direction? Imagine the following server setup:
243                                  *    0AA <-> 0AB <-> 0AC
244                                  * Fake direction would be 0AC sending a message to 0AB claiming to be from
245                                  * 0AA, or something similar. Basically, a message taking a path that *cannot*
246                                  * be correct.
247                                  *
248                                  * When would this be seen?
249                                  * Well, hopefully never. It could be caused by race conditions, bugs, or
250                                  * "miscreant" servers, though, so let's check anyway. -- w
251                                  */
252                                 std::string direction = prefix;
253
254                                 User *t = this->ServerInstance->FindUUID(prefix);
255                                 if (t)
256                                 {
257                                         direction = t->server;
258                                 }
259
260                                 TreeServer* route_back_again = Utils->BestRouteTo(direction);
261                                 if ((!route_back_again) || (route_back_again->GetSocket() != this))
262                                 {
263                                         if (route_back_again)
264                                                 ServerInstance->Logs->Log("m_spanningtree",DEBUG,"Protocol violation: Fake direction in command '%s' from connection '%s'",line.c_str(),this->GetName().c_str());
265                                         return true;
266                                 }
267                                 /* Fix by brain:
268                                  * When there is activity on the socket, reset the ping counter so
269                                  * that we're not wasting bandwidth pinging an active server.
270                                  */
271                                 route_back_again->SetNextPingTime(ServerInstance->Time() + Utils->PingFreq);
272                                 route_back_again->SetPingFlag();
273                         }
274                         else
275                         {
276                                 /*
277                                  * Empty prefix from a server to server link:
278                                  *  This is somewhat bad/naughty, so let's set the prefix
279                                  *  to be the link that we got it from, so we don't break anything. -- w
280                                  */
281                                 TreeServer* n = Utils->FindServer(GetName());
282                                 if (n)
283                                         prefix = n->GetID();
284                                 else
285                                         prefix = GetName();
286                         }
287
288                         /*
289                          * First up, check for any malformed commands (e.g. MODE without a timestamp)
290                          * and rewrite commands where necessary (SVSMODE -> MODE for services). -- w
291                          */
292                         if (command == "SVSMODE") // This isn't in an "else if" so we still force FMODE for changes on channels.
293                                 command = "MODE";
294
295                         /*
296                          * Now, check for (and parse) commands as appropriate. -- w
297                          */
298
299                         /* Find the server that this command originated from, used in the handlers below */
300                         TreeServer *ServerSource = Utils->FindServer(prefix);
301
302                         /* Find the link we just got this from so we don't bounce it back incorrectly */
303                         std::string sourceserv = this->myhost;
304                         if (!this->InboundServerName.empty())
305                         {
306                                 sourceserv = this->InboundServerName;
307                         }
308
309                         /*
310                          * XXX one of these days, this needs to be moved into class Commands.
311                          */
312                         if (command == "UID")
313                         {
314                                 return this->ParseUID(prefix, params);
315                         }
316                         else if (command == "FJOIN")
317                         {
318                                 return this->ForceJoin(prefix,params);
319                         }
320                         else if ((command == "NOTICE" || command == "PRIVMSG") && (Utils->IsServer(prefix)))
321                         {
322                                 return this->ServerMessage(assign(command), prefix, params, sourceserv);
323                         }
324                         else if (command == "STATS")
325                         {
326                                 return this->Stats(prefix, params);
327                         }
328                         else if (command == "MOTD")
329                         {
330                                 return this->Motd(prefix, params);
331                         }
332                         else if (command == "KILL" && ServerSource)
333                         {
334                                 // Kill from a server
335                                 return this->RemoteKill(prefix,params);
336                         }
337                         else if (command == "MODULES")
338                         {
339                                 return this->Modules(prefix, params);
340                         }
341                         else if (command == "ADMIN")
342                         {
343                                 return this->Admin(prefix, params);
344                         }
345                         else if (command == "MAP")
346                         {
347                                 User* user = ServerInstance->FindNick(prefix);
348                                 if (user)
349                                 {
350                                         std::vector<std::string> p(params.begin(), params.end());
351                                         return Utils->Creator->HandleMap(p, user);
352                                 }
353                         }
354                         else if (command == "SERVER")
355                         {
356                                 return this->RemoteServer(prefix,params);
357                         }
358                         else if (command == "ERROR")
359                         {
360                                 return this->Error(params);
361                         }
362                         else if (command == "OPERTYPE")
363                         {
364                                 return this->OperType(prefix,params);
365                         }
366                         else if (command == "FMODE")
367                         {
368                                 return this->ForceMode(prefix,params);
369                         }
370                         else if (command == "FTOPIC")
371                         {
372                                 return this->ForceTopic(prefix,params);
373                         }
374                         else if (command == "METADATA")
375                         {
376                                 return this->MetaData(prefix,params);
377                         }
378                         else if (command == "PING")
379                         {
380                                 return this->LocalPing(prefix,params);
381                         }
382                         else if (command == "PONG")
383                         {
384                                 TreeServer *s = Utils->FindServer(prefix);
385                                 if (s && s->bursting)
386                                 {
387                                         ServerInstance->SNO->WriteToSnoMask('l',"Server \002%s\002 has not finished burst, forcing end of burst (send ENDBURST!)", prefix.c_str());
388                                         s->FinishBurst();
389                                 }
390                                 return this->LocalPong(prefix,params);
391                         }
392                         else if (command == "VERSION")
393                         {
394                                 return this->ServerVersion(prefix,params);
395                         }
396                         else if (command == "FHOST")
397                         {
398                                 return this->ChangeHost(prefix,params);
399                         }
400                         else if (command == "FNAME")
401                         {
402                                 return this->ChangeName(prefix,params);
403                         }
404                         else if (command == "ADDLINE")
405                         {
406                                 return this->AddLine(prefix,params);
407                         }
408                         else if (command == "DELLINE")
409                         {
410                                 return this->DelLine(prefix,params);
411                         }
412                         else if (command == "SVSNICK")
413                         {
414                                 return this->ForceNick(prefix,params);
415                         }
416                         else if (command == "OPERQUIT")
417                         {
418                                 return this->OperQuit(prefix,params);
419                         }
420                         else if (command == "IDLE")
421                         {
422                                 return this->Whois(prefix,params);
423                         }
424                         else if (command == "PUSH")
425                         {
426                                 return this->Push(prefix,params);
427                         }
428                         else if (command == "TIME")
429                         {
430                                 return this->Time(prefix,params);
431                         }
432                         else if ((command == "KICK") && (Utils->IsServer(prefix)))
433                         {
434                                 if (params.size() == 3)
435                                 {
436                                         TreeServer* pf = Utils->FindServer(prefix);
437                                         User* user = this->ServerInstance->FindNick(params[1]);
438                                         Channel* chan = this->ServerInstance->FindChan(params[0]);
439                                         if (pf && user && chan)
440                                         {
441                                                 if (!chan->ServerKickUser(user, params[2].c_str(), false, pf->GetName().c_str()))
442                                                         /* Yikes, the channels gone! */
443                                                         delete chan;
444                                         }
445                                 }
446
447                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
448                         }
449                         else if (command == "SVSJOIN")
450                         {
451                                 return this->ServiceJoin(prefix,params);
452                         }
453                         else if (command == "SVSPART")
454                         {
455                                 return this->ServicePart(prefix,params);
456                         }
457                         else if (command == "SQUIT")
458                         {
459                                 if (params.size() == 2)
460                                 {
461                                         this->Squit(Utils->FindServer(params[0]),params[1]);
462                                 }
463                                 return true;
464                         }
465                         else if (command == "MODENOTICE")
466                         {
467                                 if (params.size() >= 2)
468                                 {
469                                         if (ServerSource)
470                                                 ServerInstance->Users->WriteMode(params[0].c_str(), WM_AND, "*** From %s: %s", (ServerSource ? ServerSource->GetName().c_str() : prefix.c_str()), params[1].c_str());
471                                 }
472                                 return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
473                         }
474                         else if (command == "SNONOTICE")
475                         {
476                                 if (params.size() >= 2)
477                                 {
478                                         ServerInstance->SNO->WriteToSnoMask(*(params[0].c_str()), "From " + (ServerSource ? ServerSource->GetName().c_str() : prefix) + ": "+ params[1]);
479                                         return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
480                                 }
481
482                         }
483                         else if (command == "BURST")
484                         {
485                                 // Set prefix server as bursting
486                                 if (!ServerSource)
487                                 {
488                                         this->ServerInstance->SNO->WriteToSnoMask('l', "WTF: Got BURST from a nonexistant server(?): %s", (ServerSource ? ServerSource->GetName().c_str() : prefix.c_str()));
489                                         return false;
490                                 }
491
492                                 ServerSource->bursting = true;
493                                 return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
494                         }
495                         else if (command == "ENDBURST")
496                         {
497                                 if (!ServerSource)
498                                 {
499                                         this->ServerInstance->SNO->WriteToSnoMask('l', "WTF: Got ENDBURST from a nonexistant server(?): %s", (ServerSource ? ServerSource->GetName().c_str() : prefix.c_str()));
500                                         return false;
501                                 }
502
503                                 ServerSource->FinishBurst();
504                                 return Utils->DoOneToAllButSenderRaw(line, sourceserv, prefix, command, params);
505                         }
506                         else if (command == "ENCAP")
507                         {
508                                 return this->Encap(prefix, params);
509                         }
510                         else if (command == "MODE" && !this->ServerInstance->FindUUID(prefix)) // XXX we should check for no such serv?
511                         {
512                                 // Server-prefix MODE.
513                                 std::vector<std::string> modelist(params.begin(), params.end());
514
515                                 /* We don't support this for channel mode changes any more! */
516                                 if (params.size() >= 1)
517                                 {
518                                         if (ServerInstance->FindChan(params[0]))
519                                         {
520                                                 this->SendError("Protocol violation by '"+(ServerSource ? ServerSource->GetName().c_str() : prefix)+"'! MODE for channel mode changes is not supported by the InspIRCd 1.2 protocol. You must use FMODE to preserve channel timestamps.");
521                                                 return false;
522                                         }
523                                 }
524
525                                 // Insert into the parser
526                                 this->ServerInstance->SendMode(modelist, this->ServerInstance->FakeClient);
527
528                                 // Pass out to the network
529                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
530                         }
531                         else
532                         {
533                                 /*
534                                  * Not a special s2s command. Emulate the user doing it.
535                                  * This saves us having a huge ugly command parser again.
536                                  */
537                                 User* who = this->ServerInstance->FindUUID(prefix);
538
539                                 if (!who)
540                                 {
541                                         /* this looks ugly because command is an irc::string
542                                          * It is important that we dont close the link here, unknown prefix can occur
543                                          * due to various race conditions such as the KILL message for a user somehow
544                                          * crossing the users QUIT further upstream from the server. Thanks jilles!
545                                          */
546                                         ServerInstance->Logs->Log("m_spanningtree", DEBUG, "Command " + std::string(command.c_str()) + " from unknown prefix " + prefix + "! Dropping entire command.");
547                                         return true;
548                                 }
549
550                                 if (command == "NICK")
551                                 {
552                                         if (params.size() != 2)
553                                         {
554                                                 SendError("Protocol violation: NICK message without TS - :"+std::string(who->uuid)+" NICK "+params[0]);
555                                                 return false;
556                                         }
557                                         /* Update timestamp on user when they change nicks */
558                                         who->age = atoi(params[1].c_str());
559
560                                         /*
561                                          * On nick messages, check that the nick doesnt already exist here.
562                                          * If it does, perform collision logic.
563                                          */
564                                         User* x = this->ServerInstance->FindNickOnly(params[0]);
565                                         if ((x) && (x != who))
566                                         {
567                                                 int collideret = 0;
568                                                 /* x is local, who is remote */
569                                                 collideret = this->DoCollision(x, who->age, who->ident, who->GetIPString(), who->uuid);
570                                                 if (collideret != 1)
571                                                 {
572                                                         /*
573                                                          * Remote client lost, or both lost, parsing this nickchange would be
574                                                          * pointless, as the incoming client's server will soon recieve SVSNICK to
575                                                          * change its nick to its UID. :) -- w00t
576                                                          */
577                                                         return true;
578                                                 }
579                                         }
580                                 }
581
582                                 // its a user
583                                 std::vector<std::string> strparams(params.begin(), params.end());
584
585                                 switch (this->ServerInstance->CallCommandHandler(command.c_str(), strparams, who))
586                                 {
587                                         case CMD_INVALID:
588                                                 /*
589                                                  * XXX: command is irc::string, hence ugliness
590                                                  */
591                                                 this->SendError("Unrecognised or malformed command '" + std::string(command.c_str()) + "' -- possibly loaded mismatched modules");
592                                                 return false;
593                                                 break;
594                                         case CMD_FAILURE:
595                                                 /*
596                                                  * CMD_LOCALONLY is aliased to CMD_FAILURE, so this won't go out onto the network.
597                                                  */
598                                                 return true;
599                                                 break;
600                                         default:
601                                                 /* CMD_SUCCESS falls through here */
602                                                 break;
603                                 }
604
605                                 return Utils->DoOneToAllButSenderRaw(line,sourceserv,prefix,command,params);
606
607                         }
608                         return true;
609                         break; // end of state CONNECTED (phew).
610         }
611         return true;
612 }
613
614 std::string TreeSocket::GetName()
615 {
616         std::string sourceserv = this->myhost;
617         if (!this->InboundServerName.empty())
618         {
619                 sourceserv = this->InboundServerName;
620         }
621         return sourceserv;
622 }
623
624 void TreeSocket::OnTimeout()
625 {
626         if (this->LinkState == CONNECTING)
627         {
628                 Utils->Creator->RemoteMessage(NULL, "CONNECT: Connection to \002%s\002 timed out.", myhost.c_str());
629                 Link* MyLink = Utils->FindLink(myhost);
630                 if (MyLink)
631                         Utils->DoFailOver(MyLink);
632         }
633 }
634
635 void TreeSocket::OnClose()
636 {
637         // Test fix for big fuckup
638         if (this->LinkState != CONNECTED)
639                 return;
640
641         // Connection closed.
642         // If the connection is fully up (state CONNECTED)
643         // then propogate a netsplit to all peers.
644         std::string quitserver = this->myhost;
645         if (!this->InboundServerName.empty())
646         {
647                 quitserver = this->InboundServerName;
648         }
649         TreeServer* s = Utils->FindServer(quitserver);
650         if (s)
651         {
652                 Squit(s,"Remote host closed the connection");
653         }
654
655         if (!quitserver.empty())
656         {
657                 Utils->Creator->RemoteMessage(NULL,"Connection to '\2%s\2' failed.",quitserver.c_str());
658                 time_t server_uptime = ServerInstance->Time() - this->age;
659                 if (server_uptime)
660                         Utils->Creator->RemoteMessage(NULL,"Connection to '\2%s\2' was established for %s", quitserver.c_str(), Utils->Creator->TimeToStr(server_uptime).c_str());
661         }
662 }
663
664