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