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