]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/users.cpp
6a31f0a93f02027a2f99e5dcf79bc9b091c1c974
[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                 if ((a->GetMaxLocal()) && (ServerInstance->Users->LocalCloneCount(this) > 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()) && (ServerInstance->Users->GlobalCloneCount(this) > 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->statsConnects++;
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(), 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(), 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->statsCollisions++;
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->WriteTo(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->statsSent += 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
911 /* write text to an destination user from a source user (e.g. user privmsg) */
912
913 void User::WriteTo(User *dest, const char *data, ...)
914 {
915         std::string textbuffer;
916         VAFORMAT(textbuffer, data, data);
917         this->WriteTo(dest, textbuffer);
918 }
919
920 void User::WriteTo(User *dest, const std::string &data)
921 {
922         dest->WriteFrom(this, data);
923 }
924
925 void User::WriteCommon(const char* text, ...)
926 {
927         if (this->registered != REG_ALL || quitting)
928                 return;
929
930         std::string textbuffer;
931         VAFORMAT(textbuffer, text, text);
932         textbuffer = ":" + this->GetFullHost() + " " + textbuffer;
933         this->WriteCommonRaw(textbuffer, true);
934 }
935
936 void User::WriteCommonExcept(const char* text, ...)
937 {
938         if (this->registered != REG_ALL || quitting)
939                 return;
940
941         std::string textbuffer;
942         VAFORMAT(textbuffer, text, text);
943         textbuffer = ":" + this->GetFullHost() + " " + textbuffer;
944         this->WriteCommonRaw(textbuffer, false);
945 }
946
947 void User::WriteCommonRaw(const std::string &line, bool include_self)
948 {
949         if (this->registered != REG_ALL || quitting)
950                 return;
951
952         LocalUser::already_sent_id++;
953
954         IncludeChanList include_c(chans.begin(), chans.end());
955         std::map<User*,bool> exceptions;
956
957         exceptions[this] = include_self;
958
959         FOREACH_MOD(OnBuildNeighborList, (this, include_c, exceptions));
960
961         for (std::map<User*,bool>::iterator i = exceptions.begin(); i != exceptions.end(); ++i)
962         {
963                 LocalUser* u = IS_LOCAL(i->first);
964                 if (u && !u->quitting)
965                 {
966                         u->already_sent = LocalUser::already_sent_id;
967                         if (i->second)
968                                 u->Write(line);
969                 }
970         }
971         for (IncludeChanList::const_iterator v = include_c.begin(); v != include_c.end(); ++v)
972         {
973                 Channel* c = (*v)->chan;
974                 const UserMembList* ulist = c->GetUsers();
975                 for (UserMembList::const_iterator i = ulist->begin(); i != ulist->end(); i++)
976                 {
977                         LocalUser* u = IS_LOCAL(i->first);
978                         if (u && u->already_sent != LocalUser::already_sent_id)
979                         {
980                                 u->already_sent = LocalUser::already_sent_id;
981                                 u->Write(line);
982                         }
983                 }
984         }
985 }
986
987 void User::WriteCommonQuit(const std::string &normal_text, const std::string &oper_text)
988 {
989         if (this->registered != REG_ALL)
990                 return;
991
992         already_sent_t uniq_id = ++LocalUser::already_sent_id;
993
994         const std::string normalMessage = ":" + this->GetFullHost() + " QUIT :" + normal_text;
995         const std::string operMessage = ":" + this->GetFullHost() + " QUIT :" + oper_text;
996
997         IncludeChanList include_c(chans.begin(), chans.end());
998         std::map<User*,bool> exceptions;
999
1000         FOREACH_MOD(OnBuildNeighborList, (this, include_c, exceptions));
1001
1002         for (std::map<User*,bool>::iterator i = exceptions.begin(); i != exceptions.end(); ++i)
1003         {
1004                 LocalUser* u = IS_LOCAL(i->first);
1005                 if (u && !u->quitting)
1006                 {
1007                         u->already_sent = uniq_id;
1008                         if (i->second)
1009                                 u->Write(u->IsOper() ? operMessage : normalMessage);
1010                 }
1011         }
1012         for (IncludeChanList::const_iterator v = include_c.begin(); v != include_c.end(); ++v)
1013         {
1014                 const UserMembList* ulist = (*v)->chan->GetUsers();
1015                 for (UserMembList::const_iterator i = ulist->begin(); i != ulist->end(); i++)
1016                 {
1017                         LocalUser* u = IS_LOCAL(i->first);
1018                         if (u && (u->already_sent != uniq_id))
1019                         {
1020                                 u->already_sent = uniq_id;
1021                                 u->Write(u->IsOper() ? operMessage : normalMessage);
1022                         }
1023                 }
1024         }
1025 }
1026
1027 void LocalUser::SendText(const std::string& line)
1028 {
1029         Write(line);
1030 }
1031
1032 void RemoteUser::SendText(const std::string& line)
1033 {
1034         ServerInstance->PI->PushToClient(this, line);
1035 }
1036
1037 void FakeUser::SendText(const std::string& line)
1038 {
1039 }
1040
1041 void User::SendText(const char *text, ...)
1042 {
1043         std::string line;
1044         VAFORMAT(line, text, text);
1045         SendText(line);
1046 }
1047
1048 void User::SendText(const std::string& linePrefix, std::stringstream& textStream)
1049 {
1050         std::string line;
1051         std::string word;
1052         while (textStream >> word)
1053         {
1054                 size_t lineLength = linePrefix.length() + line.length() + word.length() + 3; // "\s\n\r"
1055                 if (lineLength > ServerInstance->Config->Limits.MaxLine)
1056                 {
1057                         SendText(linePrefix + line);
1058                         line.clear();
1059                 }
1060                 line += " " + word;
1061         }
1062         SendText(linePrefix + line);
1063 }
1064
1065 /* return 0 or 1 depending if users u and u2 share one or more common channels
1066  * (used by QUIT, NICK etc which arent channel specific notices)
1067  *
1068  * The old algorithm in 1.0 for this was relatively inefficient, iterating over
1069  * the first users channels then the second users channels within the outer loop,
1070  * therefore it was a maximum of x*y iterations (upon returning 0 and checking
1071  * all possible iterations). However this new function instead checks against the
1072  * channel's userlist in the inner loop which is a std::map<User*,User*>
1073  * and saves us time as we already know what pointer value we are after.
1074  * Don't quote me on the maths as i am not a mathematician or computer scientist,
1075  * but i believe this algorithm is now x+(log y) maximum iterations instead.
1076  */
1077 bool User::SharesChannelWith(User *other)
1078 {
1079         /* Outer loop */
1080         for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1081         {
1082                 /* Eliminate the inner loop (which used to be ~equal in size to the outer loop)
1083                  * by replacing it with a map::find which *should* be more efficient
1084                  */
1085                 if ((*i)->chan->HasUser(other))
1086                         return true;
1087         }
1088         return false;
1089 }
1090
1091 bool User::ChangeName(const std::string& gecos)
1092 {
1093         if (!this->fullname.compare(gecos))
1094                 return true;
1095
1096         if (IS_LOCAL(this))
1097         {
1098                 ModResult MOD_RESULT;
1099                 FIRST_MOD_RESULT(OnChangeLocalUserGECOS, MOD_RESULT, (IS_LOCAL(this),gecos));
1100                 if (MOD_RESULT == MOD_RES_DENY)
1101                         return false;
1102                 FOREACH_MOD(OnChangeName, (this,gecos));
1103         }
1104         this->fullname.assign(gecos, 0, ServerInstance->Config->Limits.MaxGecos);
1105
1106         return true;
1107 }
1108
1109 bool User::ChangeDisplayedHost(const std::string& shost)
1110 {
1111         if (dhost == shost)
1112                 return true;
1113
1114         if (IS_LOCAL(this))
1115         {
1116                 ModResult MOD_RESULT;
1117                 FIRST_MOD_RESULT(OnChangeLocalUserHost, MOD_RESULT, (IS_LOCAL(this),shost));
1118                 if (MOD_RESULT == MOD_RES_DENY)
1119                         return false;
1120         }
1121
1122         FOREACH_MOD(OnChangeHost, (this,shost));
1123
1124         this->dhost.assign(shost, 0, ServerInstance->Config->Limits.MaxHost);
1125         this->InvalidateCache();
1126
1127         if (IS_LOCAL(this))
1128                 this->WriteNumeric(RPL_YOURDISPLAYEDHOST, "%s :is now your displayed host", this->dhost.c_str());
1129
1130         return true;
1131 }
1132
1133 bool User::ChangeIdent(const std::string& newident)
1134 {
1135         if (this->ident == newident)
1136                 return true;
1137
1138         FOREACH_MOD(OnChangeIdent, (this,newident));
1139
1140         this->ident.assign(newident, 0, ServerInstance->Config->Limits.IdentMax);
1141         this->InvalidateCache();
1142
1143         return true;
1144 }
1145
1146 void User::SendAll(const char* command, const char* text, ...)
1147 {
1148         std::string textbuffer;
1149         VAFORMAT(textbuffer, text, text);
1150         const std::string message = ":" + this->GetFullHost() + " " + command + " $* :" + textbuffer;
1151
1152         for (LocalUserList::const_iterator i = ServerInstance->Users->local_users.begin(); i != ServerInstance->Users->local_users.end(); i++)
1153         {
1154                 if ((*i)->registered == REG_ALL)
1155                         (*i)->Write(message);
1156         }
1157 }
1158
1159 /*
1160  * Sets a user's connection class.
1161  * If the class name is provided, it will be used. Otherwise, the class will be guessed using host/ip/ident/etc.
1162  * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
1163  * then their ip will be taken as 'priority' anyway, so for example,
1164  * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
1165  */
1166 void LocalUser::SetClass(const std::string &explicit_name)
1167 {
1168         ConnectClass *found = NULL;
1169
1170         ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Setting connect class for UID %s", this->uuid.c_str());
1171
1172         if (!explicit_name.empty())
1173         {
1174                 for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
1175                 {
1176                         ConnectClass* c = *i;
1177
1178                         if (explicit_name == c->name)
1179                         {
1180                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Explicitly set to %s", explicit_name.c_str());
1181                                 found = c;
1182                         }
1183                 }
1184         }
1185         else
1186         {
1187                 for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
1188                 {
1189                         ConnectClass* c = *i;
1190                         ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Checking %s", c->GetName().c_str());
1191
1192                         ModResult MOD_RESULT;
1193                         FIRST_MOD_RESULT(OnSetConnectClass, MOD_RESULT, (this,c));
1194                         if (MOD_RESULT == MOD_RES_DENY)
1195                                 continue;
1196                         if (MOD_RESULT == MOD_RES_ALLOW)
1197                         {
1198                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Class forced by module to %s", c->GetName().c_str());
1199                                 found = c;
1200                                 break;
1201                         }
1202
1203                         if (c->type == CC_NAMED)
1204                                 continue;
1205
1206                         bool regdone = (registered != REG_NONE);
1207                         if (c->config->getBool("registered", regdone) != regdone)
1208                                 continue;
1209
1210                         /* check if host matches.. */
1211                         if (!InspIRCd::MatchCIDR(this->GetIPString(), c->GetHost(), NULL) &&
1212                             !InspIRCd::MatchCIDR(this->host, c->GetHost(), NULL))
1213                         {
1214                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "No host match (for %s)", c->GetHost().c_str());
1215                                 continue;
1216                         }
1217
1218                         /*
1219                          * deny change if change will take class over the limit check it HERE, not after we found a matching class,
1220                          * because we should attempt to find another class if this one doesn't match us. -- w00t
1221                          */
1222                         if (c->limit && (c->GetReferenceCount() >= c->limit))
1223                         {
1224                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "OOPS: Connect class limit (%lu) hit, denying", c->limit);
1225                                 continue;
1226                         }
1227
1228                         /* if it requires a port ... */
1229                         int port = c->config->getInt("port");
1230                         if (port)
1231                         {
1232                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Requires port (%d)", port);
1233
1234                                 /* and our port doesn't match, fail. */
1235                                 if (this->GetServerPort() != port)
1236                                         continue;
1237                         }
1238
1239                         if (regdone && !c->config->getString("password").empty())
1240                         {
1241                                 if (!ServerInstance->PassCompare(this, c->config->getString("password"), password, c->config->getString("hash")))
1242                                 {
1243                                         ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Bad password, skipping");
1244                                         continue;
1245                                 }
1246                         }
1247
1248                         /* we stop at the first class that meets ALL critera. */
1249                         found = c;
1250                         break;
1251                 }
1252         }
1253
1254         /*
1255          * Okay, assuming we found a class that matches.. switch us into that class, keeping refcounts up to date.
1256          */
1257         if (found)
1258         {
1259                 MyClass = found;
1260         }
1261 }
1262
1263 void User::PurgeEmptyChannels()
1264 {
1265         // firstly decrement the count on each channel
1266         for (UCListIter i = this->chans.begin(); i != this->chans.end(); )
1267         {
1268                 Channel* c = (*i)->chan;
1269                 ++i;
1270                 c->DelUser(this);
1271         }
1272
1273         this->UnOper();
1274 }
1275
1276 const std::string& FakeUser::GetFullHost()
1277 {
1278         if (!ServerInstance->Config->HideWhoisServer.empty())
1279                 return ServerInstance->Config->HideWhoisServer;
1280         return server->GetName();
1281 }
1282
1283 const std::string& FakeUser::GetFullRealHost()
1284 {
1285         if (!ServerInstance->Config->HideWhoisServer.empty())
1286                 return ServerInstance->Config->HideWhoisServer;
1287         return server->GetName();
1288 }
1289
1290 ConnectClass::ConnectClass(ConfigTag* tag, char t, const std::string& mask)
1291         : config(tag), type(t), fakelag(true), name("unnamed"), registration_timeout(0), host(mask),
1292         pingtime(0), softsendqmax(0), hardsendqmax(0), recvqmax(0),
1293         penaltythreshold(0), commandrate(0), maxlocal(0), maxglobal(0), maxconnwarn(true), maxchans(0),
1294         limit(0), resolvehostnames(true)
1295 {
1296 }
1297
1298 ConnectClass::ConnectClass(ConfigTag* tag, char t, const std::string& mask, const ConnectClass& parent)
1299         : config(tag), type(t), fakelag(parent.fakelag), name("unnamed"),
1300         registration_timeout(parent.registration_timeout), host(mask), pingtime(parent.pingtime),
1301         softsendqmax(parent.softsendqmax), hardsendqmax(parent.hardsendqmax), recvqmax(parent.recvqmax),
1302         penaltythreshold(parent.penaltythreshold), commandrate(parent.commandrate),
1303         maxlocal(parent.maxlocal), maxglobal(parent.maxglobal), maxconnwarn(parent.maxconnwarn), maxchans(parent.maxchans),
1304         limit(parent.limit), resolvehostnames(parent.resolvehostnames)
1305 {
1306 }
1307
1308 void ConnectClass::Update(const ConnectClass* src)
1309 {
1310         config = src->config;
1311         type = src->type;
1312         fakelag = src->fakelag;
1313         name = src->name;
1314         registration_timeout = src->registration_timeout;
1315         host = src->host;
1316         pingtime = src->pingtime;
1317         softsendqmax = src->softsendqmax;
1318         hardsendqmax = src->hardsendqmax;
1319         recvqmax = src->recvqmax;
1320         penaltythreshold = src->penaltythreshold;
1321         commandrate = src->commandrate;
1322         maxlocal = src->maxlocal;
1323         maxglobal = src->maxglobal;
1324         maxconnwarn = src->maxconnwarn;
1325         maxchans = src->maxchans;
1326         limit = src->limit;
1327         resolvehostnames = src->resolvehostnames;
1328 }