]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/users.cpp
Merge pull request #1222 from SaberUK/master+warnings
[user/henk/code/inspircd.git] / src / users.cpp
1 /*
2  * InspIRCd -- Internet Relay Chat Daemon
3  *
4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
5  *   Copyright (C) 2006-2009 Robin Burchell <robin+git@viroteck.net>
6  *   Copyright (C) 2006-2007, 2009 Dennis Friis <peavey@inspircd.org>
7  *   Copyright (C) 2008 John Brooks <john.brooks@dereferenced.net>
8  *   Copyright (C) 2008 Thomas Stagner <aquanight@inspircd.org>
9  *   Copyright (C) 2008 Oliver Lupton <oliverlupton@gmail.com>
10  *   Copyright (C) 2003-2008 Craig Edwards <craigedwards@brainbox.cc>
11  *
12  * This file is part of InspIRCd.  InspIRCd is free software: you can
13  * redistribute it and/or modify it under the terms of the GNU General Public
14  * License as published by the Free Software Foundation, version 2.
15  *
16  * This program is distributed in the hope that it will be useful, but WITHOUT
17  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
18  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
19  * details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
23  */
24
25
26 #include "inspircd.h"
27 #include "xline.h"
28
29 bool User::IsNoticeMaskSet(unsigned char sm)
30 {
31         if (!isalpha(sm))
32                 return false;
33         return (snomasks[sm-65]);
34 }
35
36 bool User::IsModeSet(unsigned char m)
37 {
38         ModeHandler* mh = ServerInstance->Modes->FindMode(m, MODETYPE_USER);
39         return (mh && modes[mh->GetId()]);
40 }
41
42 const char* User::FormatModes(bool showparameters)
43 {
44         static std::string data;
45         std::string params;
46         data.clear();
47
48         for (unsigned char n = 0; n < 64; n++)
49         {
50                 ModeHandler* mh = ServerInstance->Modes->FindMode(n + 65, MODETYPE_USER);
51                 if (mh && IsModeSet(mh))
52                 {
53                         data.push_back(n + 65);
54                         if (showparameters && mh->NeedsParam(true))
55                         {
56                                 std::string p = mh->GetUserParameter(this);
57                                 if (p.length())
58                                         params.append(" ").append(p);
59                         }
60                 }
61         }
62         data += params;
63         return data.c_str();
64 }
65
66 User::User(const std::string& uid, Server* srv, int type)
67         : age(ServerInstance->Time())
68         , signon(0)
69         , uuid(uid)
70         , server(srv)
71         , registered(REG_NONE)
72         , quitting(false)
73         , usertype(type)
74 {
75         client_sa.sa.sa_family = AF_UNSPEC;
76
77         ServerInstance->Logs->Log("USERS", LOG_DEBUG, "New UUID for user: %s", uuid.c_str());
78
79         // Do not insert FakeUsers into the uuidlist so FindUUID() won't return them which is the desired behavior
80         if (type != USERTYPE_SERVER)
81         {
82                 if (!ServerInstance->Users.uuidlist.insert(std::make_pair(uuid, this)).second)
83                         throw CoreException("Duplicate UUID in User constructor: " + uuid);
84         }
85 }
86
87 LocalUser::LocalUser(int myfd, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* servaddr)
88         : User(ServerInstance->UIDGen.GetUID(), ServerInstance->FakeClient->server, USERTYPE_LOCAL)
89         , eh(this)
90         , bytes_in(0)
91         , bytes_out(0)
92         , cmds_in(0)
93         , cmds_out(0)
94         , quitting_sendq(false)
95         , lastping(true)
96         , exempt(false)
97         , nping(0)
98         , idle_lastmsg(0)
99         , CommandFloodPenalty(0)
100         , already_sent(0)
101 {
102         signon = ServerInstance->Time();
103         // The user's default nick is their UUID
104         nick = uuid;
105         ident = "unknown";
106         eh.SetFd(myfd);
107         memcpy(&client_sa, client, sizeof(irc::sockets::sockaddrs));
108         memcpy(&server_sa, servaddr, sizeof(irc::sockets::sockaddrs));
109         dhost = host = GetIPString();
110 }
111
112 User::~User()
113 {
114 }
115
116 const std::string& User::MakeHost()
117 {
118         if (!this->cached_makehost.empty())
119                 return this->cached_makehost;
120
121         // XXX: Is there really a need to cache this?
122         this->cached_makehost = ident + "@" + host;
123         return this->cached_makehost;
124 }
125
126 const std::string& User::MakeHostIP()
127 {
128         if (!this->cached_hostip.empty())
129                 return this->cached_hostip;
130
131         // XXX: Is there really a need to cache this?
132         this->cached_hostip = ident + "@" + this->GetIPString();
133         return this->cached_hostip;
134 }
135
136 const std::string& User::GetFullHost()
137 {
138         if (!this->cached_fullhost.empty())
139                 return this->cached_fullhost;
140
141         // XXX: Is there really a need to cache this?
142         this->cached_fullhost = nick + "!" + ident + "@" + dhost;
143         return this->cached_fullhost;
144 }
145
146 const std::string& User::GetFullRealHost()
147 {
148         if (!this->cached_fullrealhost.empty())
149                 return this->cached_fullrealhost;
150
151         // XXX: Is there really a need to cache this?
152         this->cached_fullrealhost = nick + "!" + ident + "@" + host;
153         return this->cached_fullrealhost;
154 }
155
156 bool User::HasModePermission(const ModeHandler* mh) const
157 {
158         return true;
159 }
160
161 bool LocalUser::HasModePermission(const ModeHandler* mh) const
162 {
163         if (!this->IsOper())
164                 return false;
165
166         const unsigned char mode = mh->GetModeChar();
167         if (mode < 'A' || mode > ('A' + 64)) return false;
168
169         return ((mh->GetModeType() == MODETYPE_USER ? oper->AllowedUserModes : oper->AllowedChanModes))[(mode - 'A')];
170
171 }
172 /*
173  * users on remote servers can completely bypass all permissions based checks.
174  * This prevents desyncs when one server has different type/class tags to another.
175  * That having been said, this does open things up to the possibility of source changes
176  * allowing remote kills, etc - but if they have access to the src, they most likely have
177  * access to the conf - so it's an end to a means either way.
178  */
179 bool User::HasPermission(const std::string&)
180 {
181         return true;
182 }
183
184 bool LocalUser::HasPermission(const std::string &command)
185 {
186         // are they even an oper at all?
187         if (!this->IsOper())
188         {
189                 return false;
190         }
191
192         if (oper->AllowedOperCommands.find(command) != oper->AllowedOperCommands.end())
193                 return true;
194         else if (oper->AllowedOperCommands.find("*") != oper->AllowedOperCommands.end())
195                 return true;
196
197         return false;
198 }
199
200 bool User::HasPrivPermission(const std::string &privstr, bool noisy)
201 {
202         return true;
203 }
204
205 bool LocalUser::HasPrivPermission(const std::string &privstr, bool noisy)
206 {
207         if (!this->IsOper())
208         {
209                 if (noisy)
210                         this->WriteNotice("You are not an oper");
211                 return false;
212         }
213
214         if (oper->AllowedPrivs.find(privstr) != oper->AllowedPrivs.end())
215         {
216                 return true;
217         }
218         else if (oper->AllowedPrivs.find("*") != oper->AllowedPrivs.end())
219         {
220                 return true;
221         }
222
223         if (noisy)
224                 this->WriteNotice("Oper type " + oper->name + " does not have access to priv " + privstr);
225
226         return false;
227 }
228
229 void UserIOHandler::OnDataReady()
230 {
231         if (user->quitting)
232                 return;
233
234         if (recvq.length() > user->MyClass->GetRecvqMax() && !user->HasPrivPermission("users/flood/increased-buffers"))
235         {
236                 ServerInstance->Users->QuitUser(user, "RecvQ exceeded");
237                 ServerInstance->SNO->WriteToSnoMask('a', "User %s RecvQ of %lu exceeds connect class maximum of %lu",
238                         user->nick.c_str(), (unsigned long)recvq.length(), user->MyClass->GetRecvqMax());
239                 return;
240         }
241         unsigned long sendqmax = ULONG_MAX;
242         if (!user->HasPrivPermission("users/flood/increased-buffers"))
243                 sendqmax = user->MyClass->GetSendqSoftMax();
244         unsigned long penaltymax = ULONG_MAX;
245         if (!user->HasPrivPermission("users/flood/no-fakelag"))
246                 penaltymax = user->MyClass->GetPenaltyThreshold() * 1000;
247
248         while (user->CommandFloodPenalty < penaltymax && getSendQSize() < sendqmax)
249         {
250                 std::string line;
251                 line.reserve(ServerInstance->Config->Limits.MaxLine);
252                 std::string::size_type qpos = 0;
253                 while (qpos < recvq.length())
254                 {
255                         char c = recvq[qpos++];
256                         switch (c)
257                         {
258                         case '\0':
259                                 c = ' ';
260                                 break;
261                         case '\r':
262                                 continue;
263                         case '\n':
264                                 goto eol_found;
265                         }
266                         if (line.length() < ServerInstance->Config->Limits.MaxLine - 2)
267                                 line.push_back(c);
268                 }
269                 // if we got here, the recvq ran out before we found a newline
270                 return;
271 eol_found:
272                 // just found a newline. Terminate the string, and pull it out of recvq
273                 recvq.erase(0, qpos);
274
275                 // TODO should this be moved to when it was inserted in recvq?
276                 ServerInstance->stats.Recv += qpos;
277                 user->bytes_in += qpos;
278                 user->cmds_in++;
279
280                 ServerInstance->Parser.ProcessBuffer(line, user);
281                 if (user->quitting)
282                         return;
283         }
284         if (user->CommandFloodPenalty >= penaltymax && !user->MyClass->fakelag)
285                 ServerInstance->Users->QuitUser(user, "Excess Flood");
286 }
287
288 void UserIOHandler::AddWriteBuf(const std::string &data)
289 {
290         if (user->quitting_sendq)
291                 return;
292         if (!user->quitting && getSendQSize() + data.length() > user->MyClass->GetSendqHardMax() &&
293                 !user->HasPrivPermission("users/flood/increased-buffers"))
294         {
295                 user->quitting_sendq = true;
296                 ServerInstance->GlobalCulls.AddSQItem(user);
297                 return;
298         }
299
300         // We still want to append data to the sendq of a quitting user,
301         // e.g. their ERROR message that says 'closing link'
302
303         WriteData(data);
304 }
305
306 void UserIOHandler::OnError(BufferedSocketError)
307 {
308         ServerInstance->Users->QuitUser(user, getError());
309 }
310
311 CullResult User::cull()
312 {
313         if (!quitting)
314                 ServerInstance->Users->QuitUser(this, "Culled without QuitUser");
315
316         if (client_sa.sa.sa_family != AF_UNSPEC)
317                 ServerInstance->Users->RemoveCloneCounts(this);
318
319         return Extensible::cull();
320 }
321
322 CullResult LocalUser::cull()
323 {
324         eh.cull();
325         return User::cull();
326 }
327
328 CullResult FakeUser::cull()
329 {
330         // Fake users don't quit, they just get culled.
331         quitting = true;
332         // Fake users are not inserted into UserManager::clientlist or uuidlist, so we don't need to modify those here
333         return User::cull();
334 }
335
336 void User::Oper(OperInfo* info)
337 {
338         ModeHandler* opermh = ServerInstance->Modes->FindMode('o', MODETYPE_USER);
339         if (this->IsModeSet(opermh))
340                 this->UnOper();
341
342         this->SetMode(opermh, true);
343         this->oper = info;
344         this->WriteCommand("MODE", "+o");
345         FOREACH_MOD(OnOper, (this, info->name));
346
347         std::string opername;
348         if (info->oper_block)
349                 opername = info->oper_block->getString("name");
350
351         if (IS_LOCAL(this))
352         {
353                 LocalUser* l = IS_LOCAL(this);
354                 std::string vhost = oper->getConfig("vhost");
355                 if (!vhost.empty())
356                         l->ChangeDisplayedHost(vhost);
357                 std::string opClass = oper->getConfig("class");
358                 if (!opClass.empty())
359                         l->SetClass(opClass);
360         }
361
362         ServerInstance->SNO->WriteToSnoMask('o',"%s (%s@%s) is now an IRC operator of type %s (using oper '%s')",
363                 nick.c_str(), ident.c_str(), host.c_str(), oper->name.c_str(), opername.c_str());
364         this->WriteNumeric(RPL_YOUAREOPER, InspIRCd::Format("You are now %s %s", strchr("aeiouAEIOU", oper->name[0]) ? "an" : "a", oper->name.c_str()));
365
366         ServerInstance->Logs->Log("OPER", LOG_DEFAULT, "%s opered as type: %s", GetFullRealHost().c_str(), oper->name.c_str());
367         ServerInstance->Users->all_opers.push_back(this);
368
369         // Expand permissions from config for faster lookup
370         if (IS_LOCAL(this))
371                 oper->init();
372
373         FOREACH_MOD(OnPostOper, (this, oper->name, opername));
374 }
375
376 void OperInfo::init()
377 {
378         AllowedOperCommands.clear();
379         AllowedPrivs.clear();
380         AllowedUserModes.reset();
381         AllowedChanModes.reset();
382         AllowedUserModes['o' - 'A'] = true; // Call me paranoid if you want.
383
384         for(std::vector<reference<ConfigTag> >::iterator iter = class_blocks.begin(); iter != class_blocks.end(); ++iter)
385         {
386                 ConfigTag* tag = *iter;
387                 std::string mycmd, mypriv;
388                 /* Process commands */
389                 irc::spacesepstream CommandList(tag->getString("commands"));
390                 while (CommandList.GetToken(mycmd))
391                 {
392                         AllowedOperCommands.insert(mycmd);
393                 }
394
395                 irc::spacesepstream PrivList(tag->getString("privs"));
396                 while (PrivList.GetToken(mypriv))
397                 {
398                         AllowedPrivs.insert(mypriv);
399                 }
400
401                 std::string modes = tag->getString("usermodes");
402                 for (std::string::const_iterator c = modes.begin(); c != modes.end(); ++c)
403                 {
404                         if (*c == '*')
405                         {
406                                 this->AllowedUserModes.set();
407                         }
408                         else if (*c >= 'A' && *c <= 'z')
409                         {
410                                 this->AllowedUserModes[*c - 'A'] = true;
411                         }
412                 }
413
414                 modes = tag->getString("chanmodes");
415                 for (std::string::const_iterator c = modes.begin(); c != modes.end(); ++c)
416                 {
417                         if (*c == '*')
418                         {
419                                 this->AllowedChanModes.set();
420                         }
421                         else if (*c >= 'A' && *c <= 'z')
422                         {
423                                 this->AllowedChanModes[*c - 'A'] = true;
424                         }
425                 }
426         }
427 }
428
429 void User::UnOper()
430 {
431         if (!this->IsOper())
432                 return;
433
434         /*
435          * unset their oper type (what IS_OPER checks).
436          * note, order is important - this must come before modes as -o attempts
437          * to call UnOper. -- w00t
438          */
439         oper = NULL;
440
441
442         /* Remove all oper only modes from the user when the deoper - Bug #466*/
443         Modes::ChangeList changelist;
444         const ModeParser::ModeHandlerMap& usermodes = ServerInstance->Modes->GetModes(MODETYPE_USER);
445         for (ModeParser::ModeHandlerMap::const_iterator i = usermodes.begin(); i != usermodes.end(); ++i)
446         {
447                 ModeHandler* mh = i->second;
448                 if (mh->NeedsOper())
449                         changelist.push_remove(mh);
450         }
451
452         ServerInstance->Modes->Process(this, NULL, this, changelist);
453
454         // Remove the user from the oper list
455         stdalgo::vector::swaperase(ServerInstance->Users->all_opers, this);
456
457         ModeHandler* opermh = ServerInstance->Modes->FindMode('o', MODETYPE_USER);
458         this->SetMode(opermh, false);
459 }
460
461 /*
462  * Check class restrictions
463  */
464 void LocalUser::CheckClass(bool clone_count)
465 {
466         ConnectClass* a = this->MyClass;
467
468         if (!a)
469         {
470                 ServerInstance->Users->QuitUser(this, "Access denied by configuration");
471                 return;
472         }
473         else if (a->type == CC_DENY)
474         {
475                 ServerInstance->Users->QuitUser(this, a->config->getString("reason", "Unauthorised connection"));
476                 return;
477         }
478         else if (clone_count)
479         {
480                 const UserManager::CloneCounts& clonecounts = ServerInstance->Users->GetCloneCounts(this);
481                 if ((a->GetMaxLocal()) && (clonecounts.local > a->GetMaxLocal()))
482                 {
483                         ServerInstance->Users->QuitUser(this, "No more connections allowed from your host via this connect class (local)");
484                         if (a->maxconnwarn)
485                                 ServerInstance->SNO->WriteToSnoMask('a', "WARNING: maximum LOCAL connections (%ld) exceeded for IP %s", a->GetMaxLocal(), this->GetIPString().c_str());
486                         return;
487                 }
488                 else if ((a->GetMaxGlobal()) && (clonecounts.global > a->GetMaxGlobal()))
489                 {
490                         ServerInstance->Users->QuitUser(this, "No more connections allowed from your host via this connect class (global)");
491                         if (a->maxconnwarn)
492                                 ServerInstance->SNO->WriteToSnoMask('a', "WARNING: maximum GLOBAL connections (%ld) exceeded for IP %s", a->GetMaxGlobal(), this->GetIPString().c_str());
493                         return;
494                 }
495         }
496
497         this->nping = ServerInstance->Time() + a->GetPingTime() + ServerInstance->Config->dns_timeout;
498 }
499
500 bool LocalUser::CheckLines(bool doZline)
501 {
502         const char* check[] = { "G" , "K", (doZline) ? "Z" : NULL, NULL };
503
504         if (!this->exempt)
505         {
506                 for (int n = 0; check[n]; ++n)
507                 {
508                         XLine *r = ServerInstance->XLines->MatchesLine(check[n], this);
509
510                         if (r)
511                         {
512                                 r->Apply(this);
513                                 return true;
514                         }
515                 }
516         }
517
518         return false;
519 }
520
521 void LocalUser::FullConnect()
522 {
523         ServerInstance->stats.Connects++;
524         this->idle_lastmsg = ServerInstance->Time();
525
526         /*
527          * You may be thinking "wtf, we checked this in User::AddClient!" - and yes, we did, BUT.
528          * At the time AddClient is called, we don't have a resolved host, by here we probably do - which
529          * may put the user into a totally seperate class with different restrictions! so we *must* check again.
530          * Don't remove this! -- w00t
531          */
532         MyClass = NULL;
533         SetClass();
534         CheckClass();
535         CheckLines();
536
537         if (quitting)
538                 return;
539
540         this->WriteNumeric(RPL_WELCOME, InspIRCd::Format("Welcome to the %s IRC Network %s", ServerInstance->Config->Network.c_str(), GetFullRealHost().c_str()));
541         this->WriteNumeric(RPL_YOURHOSTIS, InspIRCd::Format("Your host is %s, running version %s", ServerInstance->Config->ServerName.c_str(), INSPIRCD_BRANCH));
542         this->WriteNumeric(RPL_SERVERCREATED, InspIRCd::Format("This server was created %s %s", __TIME__, __DATE__));
543
544         const std::string& modelist = ServerInstance->Modes->GetModeListFor004Numeric();
545         this->WriteNumeric(RPL_SERVERVERSION, ServerInstance->Config->ServerName, INSPIRCD_BRANCH, modelist);
546
547         ServerInstance->ISupport.SendTo(this);
548
549         /* Now registered */
550         if (ServerInstance->Users->unregistered_count)
551                 ServerInstance->Users->unregistered_count--;
552
553         /* Trigger MOTD and LUSERS output, give modules a chance too */
554         ModResult MOD_RESULT;
555         std::string command("LUSERS");
556         std::vector<std::string> parameters;
557         FIRST_MOD_RESULT(OnPreCommand, MOD_RESULT, (command, parameters, this, true, command));
558         if (!MOD_RESULT)
559                 ServerInstance->Parser.CallHandler(command, parameters, this);
560
561         MOD_RESULT = MOD_RES_PASSTHRU;
562         command = "MOTD";
563         FIRST_MOD_RESULT(OnPreCommand, MOD_RESULT, (command, parameters, this, true, command));
564         if (!MOD_RESULT)
565                 ServerInstance->Parser.CallHandler(command, parameters, this);
566
567         if (ServerInstance->Config->RawLog)
568                 WriteServ("PRIVMSG %s :*** Raw I/O logging is enabled on this server. All messages, passwords, and commands are being recorded.", nick.c_str());
569
570         /*
571          * We don't set REG_ALL until triggering OnUserConnect, so some module events don't spew out stuff
572          * for a user that doesn't exist yet.
573          */
574         FOREACH_MOD(OnUserConnect, (this));
575
576         this->registered = REG_ALL;
577
578         FOREACH_MOD(OnPostConnect, (this));
579
580         ServerInstance->SNO->WriteToSnoMask('c',"Client connecting on port %d (class %s): %s (%s) [%s]",
581                 this->GetServerPort(), this->MyClass->name.c_str(), GetFullRealHost().c_str(), this->GetIPString().c_str(), this->fullname.c_str());
582         ServerInstance->Logs->Log("BANCACHE", LOG_DEBUG, "BanCache: Adding NEGATIVE hit for " + this->GetIPString());
583         ServerInstance->BanCache.AddHit(this->GetIPString(), "", "");
584         // reset the flood penalty (which could have been raised due to things like auto +x)
585         CommandFloodPenalty = 0;
586 }
587
588 void User::InvalidateCache()
589 {
590         /* Invalidate cache */
591         cached_fullhost.clear();
592         cached_hostip.clear();
593         cached_makehost.clear();
594         cached_fullrealhost.clear();
595 }
596
597 bool User::ChangeNick(const std::string& newnick, time_t newts)
598 {
599         if (quitting)
600         {
601                 ServerInstance->Logs->Log("USERS", LOG_DEFAULT, "ERROR: Attempted to change nick of a quitting user: " + this->nick);
602                 return false;
603         }
604
605         User* const InUse = ServerInstance->FindNickOnly(newnick);
606         if (InUse == this)
607         {
608                 // case change, don't need to check campers
609                 // and, if it's identical including case, we can leave right now
610                 // We also don't update the nick TS if it's a case change, either
611                 if (newnick == nick)
612                         return true;
613         }
614         else
615         {
616                 /*
617                  * Uh oh.. if the nickname is in use, and it's not in use by the person using it (doh) --
618                  * then we have a potential collide. Check whether someone else is camping on the nick
619                  * (i.e. connect -> send NICK, don't send USER.) If they are camping, force-change the
620                  * camper to their UID, and allow the incoming nick change.
621                  *
622                  * If the guy using the nick is already using it, tell the incoming nick change to gtfo,
623                  * because the nick is already (rightfully) in use. -- w00t
624                  */
625                 if (InUse)
626                 {
627                         if (InUse->registered != REG_ALL)
628                         {
629                                 /* force the camper to their UUID, and ask them to re-send a NICK. */
630                                 InUse->WriteFrom(InUse, "NICK %s", InUse->uuid.c_str());
631                                 InUse->WriteNumeric(ERR_NICKNAMEINUSE, InUse->nick, "Nickname overruled.");
632
633                                 InUse->registered &= ~REG_NICK;
634                                 InUse->ChangeNick(InUse->uuid);
635                         }
636                         else
637                         {
638                                 /* No camping, tell the incoming user  to stop trying to change nick ;p */
639                                 this->WriteNumeric(ERR_NICKNAMEINUSE, newnick, "Nickname is already in use.");
640                                 return false;
641                         }
642                 }
643
644                 age = newts ? newts : ServerInstance->Time();
645         }
646
647         if (this->registered == REG_ALL)
648                 this->WriteCommon("NICK %s",newnick.c_str());
649         std::string oldnick = nick;
650         nick = newnick;
651
652         InvalidateCache();
653         ServerInstance->Users->clientlist.erase(oldnick);
654         ServerInstance->Users->clientlist[newnick] = this;
655
656         if (registered == REG_ALL)
657                 FOREACH_MOD(OnUserPostNick, (this,oldnick));
658
659         return true;
660 }
661
662 int LocalUser::GetServerPort()
663 {
664         switch (this->server_sa.sa.sa_family)
665         {
666                 case AF_INET6:
667                         return htons(this->server_sa.in6.sin6_port);
668                 case AF_INET:
669                         return htons(this->server_sa.in4.sin_port);
670         }
671         return 0;
672 }
673
674 const std::string& User::GetIPString()
675 {
676         int port;
677         if (cachedip.empty())
678         {
679                 irc::sockets::satoap(client_sa, cachedip, port);
680                 /* IP addresses starting with a : on irc are a Bad Thing (tm) */
681                 if (cachedip[0] == ':')
682                         cachedip.insert(cachedip.begin(),1,'0');
683         }
684
685         return cachedip;
686 }
687
688 irc::sockets::cidr_mask User::GetCIDRMask()
689 {
690         int range = 0;
691         switch (client_sa.sa.sa_family)
692         {
693                 case AF_INET6:
694                         range = ServerInstance->Config->c_ipv6_range;
695                         break;
696                 case AF_INET:
697                         range = ServerInstance->Config->c_ipv4_range;
698                         break;
699         }
700         return irc::sockets::cidr_mask(client_sa, range);
701 }
702
703 bool User::SetClientIP(const char* sip, bool recheck_eline)
704 {
705         cachedip.clear();
706         cached_hostip.clear();
707         return irc::sockets::aptosa(sip, 0, client_sa);
708 }
709
710 void User::SetClientIP(const irc::sockets::sockaddrs& sa, bool recheck_eline)
711 {
712         cachedip.clear();
713         cached_hostip.clear();
714         memcpy(&client_sa, &sa, sizeof(irc::sockets::sockaddrs));
715 }
716
717 bool LocalUser::SetClientIP(const char* sip, bool recheck_eline)
718 {
719         irc::sockets::sockaddrs sa;
720         if (!irc::sockets::aptosa(sip, 0, sa))
721                 // Invalid
722                 return false;
723
724         LocalUser::SetClientIP(sa, recheck_eline);
725         return true;
726 }
727
728 void LocalUser::SetClientIP(const irc::sockets::sockaddrs& sa, bool recheck_eline)
729 {
730         if (sa != client_sa)
731         {
732                 User::SetClientIP(sa);
733                 if (recheck_eline)
734                         this->exempt = (ServerInstance->XLines->MatchesLine("E", this) != NULL);
735
736                 FOREACH_MOD(OnSetUserIP, (this));
737         }
738 }
739
740 static std::string wide_newline("\r\n");
741
742 void User::Write(const std::string& text)
743 {
744 }
745
746 void User::Write(const char *text, ...)
747 {
748 }
749
750 void LocalUser::Write(const std::string& text)
751 {
752         if (!SocketEngine::BoundsCheckFd(&eh))
753                 return;
754
755         if (text.length() > ServerInstance->Config->Limits.MaxLine - 2)
756         {
757                 // this should happen rarely or never. Crop the string at 512 and try again.
758                 std::string try_again(text, 0, ServerInstance->Config->Limits.MaxLine - 2);
759                 Write(try_again);
760                 return;
761         }
762
763         ServerInstance->Logs->Log("USEROUTPUT", LOG_RAWIO, "C[%s] O %s", uuid.c_str(), text.c_str());
764
765         eh.AddWriteBuf(text);
766         eh.AddWriteBuf(wide_newline);
767
768         ServerInstance->stats.Sent += text.length() + 2;
769         this->bytes_out += text.length() + 2;
770         this->cmds_out++;
771 }
772
773 /** Write()
774  */
775 void LocalUser::Write(const char *text, ...)
776 {
777         std::string textbuffer;
778         VAFORMAT(textbuffer, text, text);
779         this->Write(textbuffer);
780 }
781
782 void User::WriteServ(const std::string& text)
783 {
784         this->Write(":%s %s",ServerInstance->Config->ServerName.c_str(),text.c_str());
785 }
786
787 /** WriteServ()
788  *  Same as Write(), except `text' is prefixed with `:server.name '.
789  */
790 void User::WriteServ(const char* text, ...)
791 {
792         std::string textbuffer;
793         VAFORMAT(textbuffer, text, text);
794         this->WriteServ(textbuffer);
795 }
796
797 void User::WriteCommand(const char* command, const std::string& text)
798 {
799         this->WriteServ(command + (this->registered & REG_NICK ? " " + this->nick : " *") + " " + text);
800 }
801
802 namespace
803 {
804         std::string BuildNumeric(const std::string& source, User* targetuser, unsigned int num, const std::vector<std::string>& params)
805         {
806                 const char* const target = (targetuser->registered & REG_NICK ? targetuser->nick.c_str() : "*");
807                 std::string raw = InspIRCd::Format(":%s %03u %s", source.c_str(), num, target);
808                 if (!params.empty())
809                 {
810                         for (std::vector<std::string>::const_iterator i = params.begin(); i != params.end()-1; ++i)
811                                 raw.append(1, ' ').append(*i);
812                         raw.append(" :").append(params.back());
813                 }
814                 return raw;
815         }
816 }
817
818 void User::WriteNumeric(const Numeric::Numeric& numeric)
819 {
820         ModResult MOD_RESULT;
821
822         FIRST_MOD_RESULT(OnNumeric, MOD_RESULT, (this, numeric));
823
824         if (MOD_RESULT == MOD_RES_DENY)
825                 return;
826
827         const std::string& servername = (numeric.GetServer() ? numeric.GetServer()->GetName() : ServerInstance->Config->ServerName);
828         this->Write(BuildNumeric(servername, this, numeric.GetNumeric(), numeric.GetParams()));
829 }
830
831 void User::WriteFrom(User *user, const std::string &text)
832 {
833         const std::string message = ":" + user->GetFullHost() + " " + text;
834         this->Write(message);
835 }
836
837
838 /* write text from an originating user to originating user */
839
840 void User::WriteFrom(User *user, const char* text, ...)
841 {
842         std::string textbuffer;
843         VAFORMAT(textbuffer, text, text);
844         this->WriteFrom(user, textbuffer);
845 }
846
847 void User::WriteRemoteNotice(const std::string& text)
848 {
849         ServerInstance->PI->SendUserNotice(this, text);
850 }
851
852 void LocalUser::WriteRemoteNotice(const std::string& text)
853 {
854         WriteNotice(text);
855 }
856
857 namespace
858 {
859         class WriteCommonRawHandler : public User::ForEachNeighborHandler
860         {
861                 const std::string& msg;
862
863                 void Execute(LocalUser* user) CXX11_OVERRIDE
864                 {
865                         user->Write(msg);
866                 }
867
868          public:
869                 WriteCommonRawHandler(const std::string& message)
870                         : msg(message)
871                 {
872                 }
873         };
874 }
875
876 void User::WriteCommon(const char* text, ...)
877 {
878         std::string textbuffer;
879         VAFORMAT(textbuffer, text, text);
880         textbuffer = ":" + this->GetFullHost() + " " + textbuffer;
881         this->WriteCommonRaw(textbuffer, true);
882 }
883
884 void User::WriteCommonRaw(const std::string &line, bool include_self)
885 {
886         WriteCommonRawHandler handler(line);
887         ForEachNeighbor(handler, include_self);
888 }
889
890 void User::ForEachNeighbor(ForEachNeighborHandler& handler, bool include_self)
891 {
892         // The basic logic for visiting the neighbors of a user is to iterate the channel list of the user
893         // and visit all users on those channels. Because two users may share more than one common channel,
894         // we must skip users that we have already visited.
895         // To do this, we make use of a global counter and an integral 'already_sent' field in LocalUser.
896         // The global counter is incremented every time we do something for each neighbor of a user. Then,
897         // before visiting a member we examine user->already_sent. If it's equal to the current counter, we
898         // skip the member. Otherwise, we set it to the current counter and visit the member.
899
900         // Ask modules to build a list of exceptions.
901         // Mods may also exclude entire channels by erasing them from include_chans.
902         IncludeChanList include_chans(chans.begin(), chans.end());
903         std::map<User*, bool> exceptions;
904         exceptions[this] = include_self;
905         FOREACH_MOD(OnBuildNeighborList, (this, include_chans, exceptions));
906
907         // Get next id, guaranteed to differ from the already_sent field of all users
908         const already_sent_t newid = ServerInstance->Users.NextAlreadySentId();
909
910         // Handle exceptions first
911         for (std::map<User*, bool>::const_iterator i = exceptions.begin(); i != exceptions.end(); ++i)
912         {
913                 LocalUser* curr = IS_LOCAL(i->first);
914                 if (curr)
915                 {
916                         // Mark as visited to ensure we won't visit again if there is a common channel
917                         curr->already_sent = newid;
918                         // Always treat quitting users as excluded
919                         if ((i->second) && (!curr->quitting))
920                                 handler.Execute(curr);
921                 }
922         }
923
924         // Now consider the real neighbors
925         for (IncludeChanList::const_iterator i = include_chans.begin(); i != include_chans.end(); ++i)
926         {
927                 Channel* chan = (*i)->chan;
928                 const Channel::MemberMap& userlist = chan->GetUsers();
929                 for (Channel::MemberMap::const_iterator j = userlist.begin(); j != userlist.end(); ++j)
930                 {
931                         LocalUser* curr = IS_LOCAL(j->first);
932                         // User not yet visited?
933                         if ((curr) && (curr->already_sent != newid))
934                         {
935                                 // Mark as visited and execute function
936                                 curr->already_sent = newid;
937                                 handler.Execute(curr);
938                         }
939                 }
940         }
941 }
942
943 void User::WriteRemoteNumeric(const Numeric::Numeric& numeric)
944 {
945         WriteNumeric(numeric);
946 }
947
948 /* return 0 or 1 depending if users u and u2 share one or more common channels
949  * (used by QUIT, NICK etc which arent channel specific notices)
950  *
951  * The old algorithm in 1.0 for this was relatively inefficient, iterating over
952  * the first users channels then the second users channels within the outer loop,
953  * therefore it was a maximum of x*y iterations (upon returning 0 and checking
954  * all possible iterations). However this new function instead checks against the
955  * channel's userlist in the inner loop which is a std::map<User*,User*>
956  * and saves us time as we already know what pointer value we are after.
957  * Don't quote me on the maths as i am not a mathematician or computer scientist,
958  * but i believe this algorithm is now x+(log y) maximum iterations instead.
959  */
960 bool User::SharesChannelWith(User *other)
961 {
962         /* Outer loop */
963         for (User::ChanList::iterator i = this->chans.begin(); i != this->chans.end(); ++i)
964         {
965                 /* Eliminate the inner loop (which used to be ~equal in size to the outer loop)
966                  * by replacing it with a map::find which *should* be more efficient
967                  */
968                 if ((*i)->chan->HasUser(other))
969                         return true;
970         }
971         return false;
972 }
973
974 bool User::ChangeName(const std::string& gecos)
975 {
976         if (!this->fullname.compare(gecos))
977                 return true;
978
979         if (IS_LOCAL(this))
980         {
981                 ModResult MOD_RESULT;
982                 FIRST_MOD_RESULT(OnChangeLocalUserGECOS, MOD_RESULT, (IS_LOCAL(this),gecos));
983                 if (MOD_RESULT == MOD_RES_DENY)
984                         return false;
985                 FOREACH_MOD(OnChangeName, (this,gecos));
986         }
987         this->fullname.assign(gecos, 0, ServerInstance->Config->Limits.MaxGecos);
988
989         return true;
990 }
991
992 bool User::ChangeDisplayedHost(const std::string& shost)
993 {
994         if (dhost == shost)
995                 return true;
996
997         if (IS_LOCAL(this))
998         {
999                 ModResult MOD_RESULT;
1000                 FIRST_MOD_RESULT(OnChangeLocalUserHost, MOD_RESULT, (IS_LOCAL(this),shost));
1001                 if (MOD_RESULT == MOD_RES_DENY)
1002                         return false;
1003         }
1004
1005         FOREACH_MOD(OnChangeHost, (this,shost));
1006
1007         this->dhost.assign(shost, 0, ServerInstance->Config->Limits.MaxHost);
1008         this->InvalidateCache();
1009
1010         if (IS_LOCAL(this))
1011                 this->WriteNumeric(RPL_YOURDISPLAYEDHOST, this->dhost, "is now your displayed host");
1012
1013         return true;
1014 }
1015
1016 bool User::ChangeIdent(const std::string& newident)
1017 {
1018         if (this->ident == newident)
1019                 return true;
1020
1021         FOREACH_MOD(OnChangeIdent, (this,newident));
1022
1023         this->ident.assign(newident, 0, ServerInstance->Config->Limits.IdentMax);
1024         this->InvalidateCache();
1025
1026         return true;
1027 }
1028
1029 /*
1030  * Sets a user's connection class.
1031  * If the class name is provided, it will be used. Otherwise, the class will be guessed using host/ip/ident/etc.
1032  * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
1033  * then their ip will be taken as 'priority' anyway, so for example,
1034  * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
1035  */
1036 void LocalUser::SetClass(const std::string &explicit_name)
1037 {
1038         ConnectClass *found = NULL;
1039
1040         ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Setting connect class for UID %s", this->uuid.c_str());
1041
1042         if (!explicit_name.empty())
1043         {
1044                 for (ServerConfig::ClassVector::const_iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); ++i)
1045                 {
1046                         ConnectClass* c = *i;
1047
1048                         if (explicit_name == c->name)
1049                         {
1050                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Explicitly set to %s", explicit_name.c_str());
1051                                 found = c;
1052                         }
1053                 }
1054         }
1055         else
1056         {
1057                 for (ServerConfig::ClassVector::const_iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); ++i)
1058                 {
1059                         ConnectClass* c = *i;
1060                         ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Checking %s", c->GetName().c_str());
1061
1062                         ModResult MOD_RESULT;
1063                         FIRST_MOD_RESULT(OnSetConnectClass, MOD_RESULT, (this,c));
1064                         if (MOD_RESULT == MOD_RES_DENY)
1065                                 continue;
1066                         if (MOD_RESULT == MOD_RES_ALLOW)
1067                         {
1068                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Class forced by module to %s", c->GetName().c_str());
1069                                 found = c;
1070                                 break;
1071                         }
1072
1073                         if (c->type == CC_NAMED)
1074                                 continue;
1075
1076                         bool regdone = (registered != REG_NONE);
1077                         if (c->config->getBool("registered", regdone) != regdone)
1078                                 continue;
1079
1080                         /* check if host matches.. */
1081                         if (!InspIRCd::MatchCIDR(this->GetIPString(), c->GetHost(), NULL) &&
1082                             !InspIRCd::MatchCIDR(this->host, c->GetHost(), NULL))
1083                         {
1084                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "No host match (for %s)", c->GetHost().c_str());
1085                                 continue;
1086                         }
1087
1088                         /*
1089                          * deny change if change will take class over the limit check it HERE, not after we found a matching class,
1090                          * because we should attempt to find another class if this one doesn't match us. -- w00t
1091                          */
1092                         if (c->limit && (c->GetReferenceCount() >= c->limit))
1093                         {
1094                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "OOPS: Connect class limit (%lu) hit, denying", c->limit);
1095                                 continue;
1096                         }
1097
1098                         /* if it requires a port ... */
1099                         int port = c->config->getInt("port");
1100                         if (port)
1101                         {
1102                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Requires port (%d)", port);
1103
1104                                 /* and our port doesn't match, fail. */
1105                                 if (this->GetServerPort() != port)
1106                                         continue;
1107                         }
1108
1109                         if (regdone && !c->config->getString("password").empty())
1110                         {
1111                                 if (!ServerInstance->PassCompare(this, c->config->getString("password"), password, c->config->getString("hash")))
1112                                 {
1113                                         ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Bad password, skipping");
1114                                         continue;
1115                                 }
1116                         }
1117
1118                         /* we stop at the first class that meets ALL critera. */
1119                         found = c;
1120                         break;
1121                 }
1122         }
1123
1124         /*
1125          * Okay, assuming we found a class that matches.. switch us into that class, keeping refcounts up to date.
1126          */
1127         if (found)
1128         {
1129                 MyClass = found;
1130         }
1131 }
1132
1133 void User::PurgeEmptyChannels()
1134 {
1135         // firstly decrement the count on each channel
1136         for (User::ChanList::iterator i = this->chans.begin(); i != this->chans.end(); )
1137         {
1138                 Channel* c = (*i)->chan;
1139                 ++i;
1140                 c->DelUser(this);
1141         }
1142
1143         this->UnOper();
1144 }
1145
1146 const std::string& FakeUser::GetFullHost()
1147 {
1148         if (!ServerInstance->Config->HideWhoisServer.empty())
1149                 return ServerInstance->Config->HideWhoisServer;
1150         return server->GetName();
1151 }
1152
1153 const std::string& FakeUser::GetFullRealHost()
1154 {
1155         if (!ServerInstance->Config->HideWhoisServer.empty())
1156                 return ServerInstance->Config->HideWhoisServer;
1157         return server->GetName();
1158 }
1159
1160 ConnectClass::ConnectClass(ConfigTag* tag, char t, const std::string& mask)
1161         : config(tag), type(t), fakelag(true), name("unnamed"), registration_timeout(0), host(mask),
1162         pingtime(0), softsendqmax(0), hardsendqmax(0), recvqmax(0),
1163         penaltythreshold(0), commandrate(0), maxlocal(0), maxglobal(0), maxconnwarn(true), maxchans(ServerInstance->Config->MaxChans),
1164         limit(0), resolvehostnames(true)
1165 {
1166 }
1167
1168 ConnectClass::ConnectClass(ConfigTag* tag, char t, const std::string& mask, const ConnectClass& parent)
1169         : config(tag), type(t), fakelag(parent.fakelag), name("unnamed"),
1170         registration_timeout(parent.registration_timeout), host(mask), pingtime(parent.pingtime),
1171         softsendqmax(parent.softsendqmax), hardsendqmax(parent.hardsendqmax), recvqmax(parent.recvqmax),
1172         penaltythreshold(parent.penaltythreshold), commandrate(parent.commandrate),
1173         maxlocal(parent.maxlocal), maxglobal(parent.maxglobal), maxconnwarn(parent.maxconnwarn), maxchans(parent.maxchans),
1174         limit(parent.limit), resolvehostnames(parent.resolvehostnames)
1175 {
1176 }
1177
1178 void ConnectClass::Update(const ConnectClass* src)
1179 {
1180         config = src->config;
1181         type = src->type;
1182         fakelag = src->fakelag;
1183         name = src->name;
1184         registration_timeout = src->registration_timeout;
1185         host = src->host;
1186         pingtime = src->pingtime;
1187         softsendqmax = src->softsendqmax;
1188         hardsendqmax = src->hardsendqmax;
1189         recvqmax = src->recvqmax;
1190         penaltythreshold = src->penaltythreshold;
1191         commandrate = src->commandrate;
1192         maxlocal = src->maxlocal;
1193         maxglobal = src->maxglobal;
1194         maxconnwarn = src->maxconnwarn;
1195         maxchans = src->maxchans;
1196         limit = src->limit;
1197         resolvehostnames = src->resolvehostnames;
1198 }