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