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