]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/users.cpp
4f9b6c945680327f928293e06073e9fd123fac93
[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         Modes::ChangeList changelist;
453         for (unsigned char letter = 'A'; letter <= 'z'; letter++)
454         {
455                 ModeHandler* mh = ServerInstance->Modes->FindMode(letter, MODETYPE_USER);
456                 if (mh && mh->NeedsOper())
457                         changelist.push_remove(mh);
458         }
459
460         ServerInstance->Modes->Process(this, NULL, this, changelist);
461
462         // Remove the user from the oper list
463         stdalgo::vector::swaperase(ServerInstance->Users->all_opers, this);
464
465         ModeHandler* opermh = ServerInstance->Modes->FindMode('o', MODETYPE_USER);
466         this->SetMode(opermh, false);
467 }
468
469 /*
470  * Check class restrictions
471  */
472 void LocalUser::CheckClass(bool clone_count)
473 {
474         ConnectClass* a = this->MyClass;
475
476         if (!a)
477         {
478                 ServerInstance->Users->QuitUser(this, "Access denied by configuration");
479                 return;
480         }
481         else if (a->type == CC_DENY)
482         {
483                 ServerInstance->Users->QuitUser(this, a->config->getString("reason", "Unauthorised connection"));
484                 return;
485         }
486         else if (clone_count)
487         {
488                 const UserManager::CloneCounts& clonecounts = ServerInstance->Users->GetCloneCounts(this);
489                 if ((a->GetMaxLocal()) && (clonecounts.local > a->GetMaxLocal()))
490                 {
491                         ServerInstance->Users->QuitUser(this, "No more connections allowed from your host via this connect class (local)");
492                         if (a->maxconnwarn)
493                                 ServerInstance->SNO->WriteToSnoMask('a', "WARNING: maximum LOCAL connections (%ld) exceeded for IP %s", a->GetMaxLocal(), this->GetIPString().c_str());
494                         return;
495                 }
496                 else if ((a->GetMaxGlobal()) && (clonecounts.global > a->GetMaxGlobal()))
497                 {
498                         ServerInstance->Users->QuitUser(this, "No more connections allowed from your host via this connect class (global)");
499                         if (a->maxconnwarn)
500                                 ServerInstance->SNO->WriteToSnoMask('a', "WARNING: maximum GLOBAL connections (%ld) exceeded for IP %s", a->GetMaxGlobal(), this->GetIPString().c_str());
501                         return;
502                 }
503         }
504
505         this->nping = ServerInstance->Time() + a->GetPingTime() + ServerInstance->Config->dns_timeout;
506 }
507
508 bool LocalUser::CheckLines(bool doZline)
509 {
510         const char* check[] = { "G" , "K", (doZline) ? "Z" : NULL, NULL };
511
512         if (!this->exempt)
513         {
514                 for (int n = 0; check[n]; ++n)
515                 {
516                         XLine *r = ServerInstance->XLines->MatchesLine(check[n], this);
517
518                         if (r)
519                         {
520                                 r->Apply(this);
521                                 return true;
522                         }
523                 }
524         }
525
526         return false;
527 }
528
529 void LocalUser::FullConnect()
530 {
531         ServerInstance->stats.Connects++;
532         this->idle_lastmsg = ServerInstance->Time();
533
534         /*
535          * You may be thinking "wtf, we checked this in User::AddClient!" - and yes, we did, BUT.
536          * At the time AddClient is called, we don't have a resolved host, by here we probably do - which
537          * may put the user into a totally seperate class with different restrictions! so we *must* check again.
538          * Don't remove this! -- w00t
539          */
540         MyClass = NULL;
541         SetClass();
542         CheckClass();
543         CheckLines();
544
545         if (quitting)
546                 return;
547
548         this->WriteNumeric(RPL_WELCOME, ":Welcome to the %s IRC Network %s", ServerInstance->Config->Network.c_str(), GetFullRealHost().c_str());
549         this->WriteNumeric(RPL_YOURHOSTIS, ":Your host is %s, running version %s", ServerInstance->Config->ServerName.c_str(), INSPIRCD_BRANCH);
550         this->WriteNumeric(RPL_SERVERCREATED, ":This server was created %s %s", __TIME__, __DATE__);
551
552         const std::string& modelist = ServerInstance->Modes->GetModeListFor004Numeric();
553         this->WriteNumeric(RPL_SERVERVERSION, "%s %s %s", ServerInstance->Config->ServerName.c_str(), INSPIRCD_BRANCH, modelist.c_str());
554
555         ServerInstance->ISupport.SendTo(this);
556
557         /* Now registered */
558         if (ServerInstance->Users->unregistered_count)
559                 ServerInstance->Users->unregistered_count--;
560
561         /* Trigger MOTD and LUSERS output, give modules a chance too */
562         ModResult MOD_RESULT;
563         std::string command("LUSERS");
564         std::vector<std::string> parameters;
565         FIRST_MOD_RESULT(OnPreCommand, MOD_RESULT, (command, parameters, this, true, command));
566         if (!MOD_RESULT)
567                 ServerInstance->Parser.CallHandler(command, parameters, this);
568
569         MOD_RESULT = MOD_RES_PASSTHRU;
570         command = "MOTD";
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         if (ServerInstance->Config->RawLog)
576                 WriteServ("PRIVMSG %s :*** Raw I/O logging is enabled on this server. All messages, passwords, and commands are being recorded.", nick.c_str());
577
578         /*
579          * We don't set REG_ALL until triggering OnUserConnect, so some module events don't spew out stuff
580          * for a user that doesn't exist yet.
581          */
582         FOREACH_MOD(OnUserConnect, (this));
583
584         this->registered = REG_ALL;
585
586         FOREACH_MOD(OnPostConnect, (this));
587
588         ServerInstance->SNO->WriteToSnoMask('c',"Client connecting on port %d (class %s): %s (%s) [%s]",
589                 this->GetServerPort(), this->MyClass->name.c_str(), GetFullRealHost().c_str(), this->GetIPString().c_str(), this->fullname.c_str());
590         ServerInstance->Logs->Log("BANCACHE", LOG_DEBUG, "BanCache: Adding NEGATIVE hit for " + this->GetIPString());
591         ServerInstance->BanCache.AddHit(this->GetIPString(), "", "");
592         // reset the flood penalty (which could have been raised due to things like auto +x)
593         CommandFloodPenalty = 0;
594 }
595
596 void User::InvalidateCache()
597 {
598         /* Invalidate cache */
599         cached_fullhost.clear();
600         cached_hostip.clear();
601         cached_makehost.clear();
602         cached_fullrealhost.clear();
603 }
604
605 bool User::ChangeNick(const std::string& newnick, time_t newts)
606 {
607         if (quitting)
608         {
609                 ServerInstance->Logs->Log("USERS", LOG_DEFAULT, "ERROR: Attempted to change nick of a quitting user: " + this->nick);
610                 return false;
611         }
612
613         User* const InUse = ServerInstance->FindNickOnly(newnick);
614         if (InUse == this)
615         {
616                 // case change, don't need to check campers
617                 // and, if it's identical including case, we can leave right now
618                 // We also don't update the nick TS if it's a case change, either
619                 if (newnick == nick)
620                         return true;
621         }
622         else
623         {
624                 /*
625                  * Uh oh.. if the nickname is in use, and it's not in use by the person using it (doh) --
626                  * then we have a potential collide. Check whether someone else is camping on the nick
627                  * (i.e. connect -> send NICK, don't send USER.) If they are camping, force-change the
628                  * camper to their UID, and allow the incoming nick change.
629                  *
630                  * If the guy using the nick is already using it, tell the incoming nick change to gtfo,
631                  * because the nick is already (rightfully) in use. -- w00t
632                  */
633                 if (InUse)
634                 {
635                         if (InUse->registered != REG_ALL)
636                         {
637                                 /* force the camper to their UUID, and ask them to re-send a NICK. */
638                                 InUse->WriteFrom(InUse, "NICK %s", InUse->uuid.c_str());
639                                 InUse->WriteNumeric(ERR_NICKNAMEINUSE, "%s :Nickname overruled.", InUse->nick.c_str());
640
641                                 InUse->registered &= ~REG_NICK;
642                                 InUse->ChangeNick(InUse->uuid);
643                         }
644                         else
645                         {
646                                 /* No camping, tell the incoming user  to stop trying to change nick ;p */
647                                 this->WriteNumeric(ERR_NICKNAMEINUSE, "%s :Nickname is already in use.", newnick.c_str());
648                                 return false;
649                         }
650                 }
651
652                 age = newts ? newts : ServerInstance->Time();
653         }
654
655         if (this->registered == REG_ALL)
656                 this->WriteCommon("NICK %s",newnick.c_str());
657         std::string oldnick = nick;
658         nick = newnick;
659
660         InvalidateCache();
661         ServerInstance->Users->clientlist.erase(oldnick);
662         ServerInstance->Users->clientlist[newnick] = this;
663
664         if (registered == REG_ALL)
665                 FOREACH_MOD(OnUserPostNick, (this,oldnick));
666
667         return true;
668 }
669
670 int LocalUser::GetServerPort()
671 {
672         switch (this->server_sa.sa.sa_family)
673         {
674                 case AF_INET6:
675                         return htons(this->server_sa.in6.sin6_port);
676                 case AF_INET:
677                         return htons(this->server_sa.in4.sin_port);
678         }
679         return 0;
680 }
681
682 const std::string& User::GetIPString()
683 {
684         int port;
685         if (cachedip.empty())
686         {
687                 irc::sockets::satoap(client_sa, cachedip, port);
688                 /* IP addresses starting with a : on irc are a Bad Thing (tm) */
689                 if (cachedip[0] == ':')
690                         cachedip.insert(cachedip.begin(),1,'0');
691         }
692
693         return cachedip;
694 }
695
696 irc::sockets::cidr_mask User::GetCIDRMask()
697 {
698         int range = 0;
699         switch (client_sa.sa.sa_family)
700         {
701                 case AF_INET6:
702                         range = ServerInstance->Config->c_ipv6_range;
703                         break;
704                 case AF_INET:
705                         range = ServerInstance->Config->c_ipv4_range;
706                         break;
707         }
708         return irc::sockets::cidr_mask(client_sa, range);
709 }
710
711 bool User::SetClientIP(const char* sip, bool recheck_eline)
712 {
713         cachedip.clear();
714         cached_hostip.clear();
715         return irc::sockets::aptosa(sip, 0, client_sa);
716 }
717
718 void User::SetClientIP(const irc::sockets::sockaddrs& sa, bool recheck_eline)
719 {
720         cachedip.clear();
721         cached_hostip.clear();
722         memcpy(&client_sa, &sa, sizeof(irc::sockets::sockaddrs));
723 }
724
725 bool LocalUser::SetClientIP(const char* sip, bool recheck_eline)
726 {
727         irc::sockets::sockaddrs sa;
728         if (!irc::sockets::aptosa(sip, 0, sa))
729                 // Invalid
730                 return false;
731
732         LocalUser::SetClientIP(sa, recheck_eline);
733         return true;
734 }
735
736 void LocalUser::SetClientIP(const irc::sockets::sockaddrs& sa, bool recheck_eline)
737 {
738         if (sa != client_sa)
739         {
740                 User::SetClientIP(sa);
741                 if (recheck_eline)
742                         this->exempt = (ServerInstance->XLines->MatchesLine("E", this) != NULL);
743
744                 FOREACH_MOD(OnSetUserIP, (this));
745         }
746 }
747
748 static std::string wide_newline("\r\n");
749
750 void User::Write(const std::string& text)
751 {
752 }
753
754 void User::Write(const char *text, ...)
755 {
756 }
757
758 void LocalUser::Write(const std::string& text)
759 {
760         if (!SocketEngine::BoundsCheckFd(&eh))
761                 return;
762
763         if (text.length() > ServerInstance->Config->Limits.MaxLine - 2)
764         {
765                 // this should happen rarely or never. Crop the string at 512 and try again.
766                 std::string try_again = text.substr(0, ServerInstance->Config->Limits.MaxLine - 2);
767                 Write(try_again);
768                 return;
769         }
770
771         ServerInstance->Logs->Log("USEROUTPUT", LOG_RAWIO, "C[%s] O %s", uuid.c_str(), text.c_str());
772
773         eh.AddWriteBuf(text);
774         eh.AddWriteBuf(wide_newline);
775
776         ServerInstance->stats.Sent += text.length() + 2;
777         this->bytes_out += text.length() + 2;
778         this->cmds_out++;
779 }
780
781 /** Write()
782  */
783 void LocalUser::Write(const char *text, ...)
784 {
785         std::string textbuffer;
786         VAFORMAT(textbuffer, text, text);
787         this->Write(textbuffer);
788 }
789
790 void User::WriteServ(const std::string& text)
791 {
792         this->Write(":%s %s",ServerInstance->Config->ServerName.c_str(),text.c_str());
793 }
794
795 /** WriteServ()
796  *  Same as Write(), except `text' is prefixed with `:server.name '.
797  */
798 void User::WriteServ(const char* text, ...)
799 {
800         std::string textbuffer;
801         VAFORMAT(textbuffer, text, text);
802         this->WriteServ(textbuffer);
803 }
804
805 void User::WriteCommand(const char* command, const std::string& text)
806 {
807         this->WriteServ(command + (this->registered & REG_NICK ? " " + this->nick : " *") + " " + text);
808 }
809
810 void User::WriteNumeric(unsigned int numeric, const char* text, ...)
811 {
812         std::string textbuffer;
813         VAFORMAT(textbuffer, text, text);
814         this->WriteNumeric(numeric, textbuffer);
815 }
816
817 void User::WriteNumeric(unsigned int numeric, const std::string &text)
818 {
819         ModResult MOD_RESULT;
820
821         FIRST_MOD_RESULT(OnNumeric, MOD_RESULT, (this, numeric, text));
822
823         if (MOD_RESULT == MOD_RES_DENY)
824                 return;
825
826         const std::string message = InspIRCd::Format(":%s %03u %s %s", ServerInstance->Config->ServerName.c_str(),
827                 numeric, this->registered & REG_NICK ? this->nick.c_str() : "*", text.c_str());
828         this->Write(message);
829 }
830
831 void User::WriteFrom(User *user, const std::string &text)
832 {
833         const std::string message = ":" + user->GetFullHost() + " " + text;
834         this->Write(message);
835 }
836
837
838 /* write text from an originating user to originating user */
839
840 void User::WriteFrom(User *user, const char* text, ...)
841 {
842         std::string textbuffer;
843         VAFORMAT(textbuffer, text, text);
844         this->WriteFrom(user, textbuffer);
845 }
846
847 void User::WriteCommon(const char* text, ...)
848 {
849         if (this->registered != REG_ALL || quitting)
850                 return;
851
852         std::string textbuffer;
853         VAFORMAT(textbuffer, text, text);
854         textbuffer = ":" + this->GetFullHost() + " " + textbuffer;
855         this->WriteCommonRaw(textbuffer, true);
856 }
857
858 void User::WriteCommonRaw(const std::string &line, bool include_self)
859 {
860         if (this->registered != REG_ALL || quitting)
861                 return;
862
863         LocalUser::already_sent_id++;
864
865         IncludeChanList include_c(chans.begin(), chans.end());
866         std::map<User*,bool> exceptions;
867
868         exceptions[this] = include_self;
869
870         FOREACH_MOD(OnBuildNeighborList, (this, include_c, exceptions));
871
872         for (std::map<User*,bool>::iterator i = exceptions.begin(); i != exceptions.end(); ++i)
873         {
874                 LocalUser* u = IS_LOCAL(i->first);
875                 if (u && !u->quitting)
876                 {
877                         u->already_sent = LocalUser::already_sent_id;
878                         if (i->second)
879                                 u->Write(line);
880                 }
881         }
882         for (IncludeChanList::const_iterator v = include_c.begin(); v != include_c.end(); ++v)
883         {
884                 Channel* c = (*v)->chan;
885                 const Channel::MemberMap& ulist = c->GetUsers();
886                 for (Channel::MemberMap::const_iterator i = ulist.begin(); i != ulist.end(); ++i)
887                 {
888                         LocalUser* u = IS_LOCAL(i->first);
889                         if (u && u->already_sent != LocalUser::already_sent_id)
890                         {
891                                 u->already_sent = LocalUser::already_sent_id;
892                                 u->Write(line);
893                         }
894                 }
895         }
896 }
897
898 void User::WriteCommonQuit(const std::string &normal_text, const std::string &oper_text)
899 {
900         if (this->registered != REG_ALL)
901                 return;
902
903         already_sent_t uniq_id = ++LocalUser::already_sent_id;
904
905         const std::string normalMessage = ":" + this->GetFullHost() + " QUIT :" + normal_text;
906         const std::string operMessage = ":" + this->GetFullHost() + " QUIT :" + oper_text;
907
908         IncludeChanList include_c(chans.begin(), chans.end());
909         std::map<User*,bool> exceptions;
910
911         FOREACH_MOD(OnBuildNeighborList, (this, include_c, exceptions));
912
913         for (std::map<User*,bool>::iterator i = exceptions.begin(); i != exceptions.end(); ++i)
914         {
915                 LocalUser* u = IS_LOCAL(i->first);
916                 if (u && !u->quitting)
917                 {
918                         u->already_sent = uniq_id;
919                         if (i->second)
920                                 u->Write(u->IsOper() ? operMessage : normalMessage);
921                 }
922         }
923         for (IncludeChanList::const_iterator v = include_c.begin(); v != include_c.end(); ++v)
924         {
925                 const Channel::MemberMap& ulist = (*v)->chan->GetUsers();
926                 for (Channel::MemberMap::const_iterator i = ulist.begin(); i != ulist.end(); i++)
927                 {
928                         LocalUser* u = IS_LOCAL(i->first);
929                         if (u && (u->already_sent != uniq_id))
930                         {
931                                 u->already_sent = uniq_id;
932                                 u->Write(u->IsOper() ? operMessage : normalMessage);
933                         }
934                 }
935         }
936 }
937
938 void LocalUser::SendText(const std::string& line)
939 {
940         Write(line);
941 }
942
943 void RemoteUser::SendText(const std::string& line)
944 {
945         ServerInstance->PI->PushToClient(this, line);
946 }
947
948 void FakeUser::SendText(const std::string& line)
949 {
950 }
951
952 void User::SendText(const char *text, ...)
953 {
954         std::string line;
955         VAFORMAT(line, text, text);
956         SendText(line);
957 }
958
959 void User::SendText(const std::string& linePrefix, std::stringstream& textStream)
960 {
961         std::string line;
962         std::string word;
963         while (textStream >> word)
964         {
965                 size_t lineLength = linePrefix.length() + line.length() + word.length() + 3; // "\s\n\r"
966                 if (lineLength > ServerInstance->Config->Limits.MaxLine)
967                 {
968                         SendText(linePrefix + line);
969                         line.clear();
970                 }
971                 line += " " + word;
972         }
973         SendText(linePrefix + line);
974 }
975
976 /* return 0 or 1 depending if users u and u2 share one or more common channels
977  * (used by QUIT, NICK etc which arent channel specific notices)
978  *
979  * The old algorithm in 1.0 for this was relatively inefficient, iterating over
980  * the first users channels then the second users channels within the outer loop,
981  * therefore it was a maximum of x*y iterations (upon returning 0 and checking
982  * all possible iterations). However this new function instead checks against the
983  * channel's userlist in the inner loop which is a std::map<User*,User*>
984  * and saves us time as we already know what pointer value we are after.
985  * Don't quote me on the maths as i am not a mathematician or computer scientist,
986  * but i believe this algorithm is now x+(log y) maximum iterations instead.
987  */
988 bool User::SharesChannelWith(User *other)
989 {
990         /* Outer loop */
991         for (User::ChanList::iterator i = this->chans.begin(); i != this->chans.end(); ++i)
992         {
993                 /* Eliminate the inner loop (which used to be ~equal in size to the outer loop)
994                  * by replacing it with a map::find which *should* be more efficient
995                  */
996                 if ((*i)->chan->HasUser(other))
997                         return true;
998         }
999         return false;
1000 }
1001
1002 bool User::ChangeName(const std::string& gecos)
1003 {
1004         if (!this->fullname.compare(gecos))
1005                 return true;
1006
1007         if (IS_LOCAL(this))
1008         {
1009                 ModResult MOD_RESULT;
1010                 FIRST_MOD_RESULT(OnChangeLocalUserGECOS, MOD_RESULT, (IS_LOCAL(this),gecos));
1011                 if (MOD_RESULT == MOD_RES_DENY)
1012                         return false;
1013                 FOREACH_MOD(OnChangeName, (this,gecos));
1014         }
1015         this->fullname.assign(gecos, 0, ServerInstance->Config->Limits.MaxGecos);
1016
1017         return true;
1018 }
1019
1020 bool User::ChangeDisplayedHost(const std::string& shost)
1021 {
1022         if (dhost == shost)
1023                 return true;
1024
1025         if (IS_LOCAL(this))
1026         {
1027                 ModResult MOD_RESULT;
1028                 FIRST_MOD_RESULT(OnChangeLocalUserHost, MOD_RESULT, (IS_LOCAL(this),shost));
1029                 if (MOD_RESULT == MOD_RES_DENY)
1030                         return false;
1031         }
1032
1033         FOREACH_MOD(OnChangeHost, (this,shost));
1034
1035         this->dhost.assign(shost, 0, ServerInstance->Config->Limits.MaxHost);
1036         this->InvalidateCache();
1037
1038         if (IS_LOCAL(this))
1039                 this->WriteNumeric(RPL_YOURDISPLAYEDHOST, "%s :is now your displayed host", this->dhost.c_str());
1040
1041         return true;
1042 }
1043
1044 bool User::ChangeIdent(const std::string& newident)
1045 {
1046         if (this->ident == newident)
1047                 return true;
1048
1049         FOREACH_MOD(OnChangeIdent, (this,newident));
1050
1051         this->ident.assign(newident, 0, ServerInstance->Config->Limits.IdentMax);
1052         this->InvalidateCache();
1053
1054         return true;
1055 }
1056
1057 /*
1058  * Sets a user's connection class.
1059  * If the class name is provided, it will be used. Otherwise, the class will be guessed using host/ip/ident/etc.
1060  * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
1061  * then their ip will be taken as 'priority' anyway, so for example,
1062  * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
1063  */
1064 void LocalUser::SetClass(const std::string &explicit_name)
1065 {
1066         ConnectClass *found = NULL;
1067
1068         ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Setting connect class for UID %s", this->uuid.c_str());
1069
1070         if (!explicit_name.empty())
1071         {
1072                 for (ServerConfig::ClassVector::const_iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); ++i)
1073                 {
1074                         ConnectClass* c = *i;
1075
1076                         if (explicit_name == c->name)
1077                         {
1078                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Explicitly set to %s", explicit_name.c_str());
1079                                 found = c;
1080                         }
1081                 }
1082         }
1083         else
1084         {
1085                 for (ServerConfig::ClassVector::const_iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); ++i)
1086                 {
1087                         ConnectClass* c = *i;
1088                         ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Checking %s", c->GetName().c_str());
1089
1090                         ModResult MOD_RESULT;
1091                         FIRST_MOD_RESULT(OnSetConnectClass, MOD_RESULT, (this,c));
1092                         if (MOD_RESULT == MOD_RES_DENY)
1093                                 continue;
1094                         if (MOD_RESULT == MOD_RES_ALLOW)
1095                         {
1096                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Class forced by module to %s", c->GetName().c_str());
1097                                 found = c;
1098                                 break;
1099                         }
1100
1101                         if (c->type == CC_NAMED)
1102                                 continue;
1103
1104                         bool regdone = (registered != REG_NONE);
1105                         if (c->config->getBool("registered", regdone) != regdone)
1106                                 continue;
1107
1108                         /* check if host matches.. */
1109                         if (!InspIRCd::MatchCIDR(this->GetIPString(), c->GetHost(), NULL) &&
1110                             !InspIRCd::MatchCIDR(this->host, c->GetHost(), NULL))
1111                         {
1112                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "No host match (for %s)", c->GetHost().c_str());
1113                                 continue;
1114                         }
1115
1116                         /*
1117                          * deny change if change will take class over the limit check it HERE, not after we found a matching class,
1118                          * because we should attempt to find another class if this one doesn't match us. -- w00t
1119                          */
1120                         if (c->limit && (c->GetReferenceCount() >= c->limit))
1121                         {
1122                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "OOPS: Connect class limit (%lu) hit, denying", c->limit);
1123                                 continue;
1124                         }
1125
1126                         /* if it requires a port ... */
1127                         int port = c->config->getInt("port");
1128                         if (port)
1129                         {
1130                                 ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Requires port (%d)", port);
1131
1132                                 /* and our port doesn't match, fail. */
1133                                 if (this->GetServerPort() != port)
1134                                         continue;
1135                         }
1136
1137                         if (regdone && !c->config->getString("password").empty())
1138                         {
1139                                 if (!ServerInstance->PassCompare(this, c->config->getString("password"), password, c->config->getString("hash")))
1140                                 {
1141                                         ServerInstance->Logs->Log("CONNECTCLASS", LOG_DEBUG, "Bad password, skipping");
1142                                         continue;
1143                                 }
1144                         }
1145
1146                         /* we stop at the first class that meets ALL critera. */
1147                         found = c;
1148                         break;
1149                 }
1150         }
1151
1152         /*
1153          * Okay, assuming we found a class that matches.. switch us into that class, keeping refcounts up to date.
1154          */
1155         if (found)
1156         {
1157                 MyClass = found;
1158         }
1159 }
1160
1161 void User::PurgeEmptyChannels()
1162 {
1163         // firstly decrement the count on each channel
1164         for (User::ChanList::iterator i = this->chans.begin(); i != this->chans.end(); )
1165         {
1166                 Channel* c = (*i)->chan;
1167                 ++i;
1168                 c->DelUser(this);
1169         }
1170
1171         this->UnOper();
1172 }
1173
1174 const std::string& FakeUser::GetFullHost()
1175 {
1176         if (!ServerInstance->Config->HideWhoisServer.empty())
1177                 return ServerInstance->Config->HideWhoisServer;
1178         return server->GetName();
1179 }
1180
1181 const std::string& FakeUser::GetFullRealHost()
1182 {
1183         if (!ServerInstance->Config->HideWhoisServer.empty())
1184                 return ServerInstance->Config->HideWhoisServer;
1185         return server->GetName();
1186 }
1187
1188 ConnectClass::ConnectClass(ConfigTag* tag, char t, const std::string& mask)
1189         : config(tag), type(t), fakelag(true), name("unnamed"), registration_timeout(0), host(mask),
1190         pingtime(0), softsendqmax(0), hardsendqmax(0), recvqmax(0),
1191         penaltythreshold(0), commandrate(0), maxlocal(0), maxglobal(0), maxconnwarn(true), maxchans(ServerInstance->Config->MaxChans),
1192         limit(0), resolvehostnames(true)
1193 {
1194 }
1195
1196 ConnectClass::ConnectClass(ConfigTag* tag, char t, const std::string& mask, const ConnectClass& parent)
1197         : config(tag), type(t), fakelag(parent.fakelag), name("unnamed"),
1198         registration_timeout(parent.registration_timeout), host(mask), pingtime(parent.pingtime),
1199         softsendqmax(parent.softsendqmax), hardsendqmax(parent.hardsendqmax), recvqmax(parent.recvqmax),
1200         penaltythreshold(parent.penaltythreshold), commandrate(parent.commandrate),
1201         maxlocal(parent.maxlocal), maxglobal(parent.maxglobal), maxconnwarn(parent.maxconnwarn), maxchans(parent.maxchans),
1202         limit(parent.limit), resolvehostnames(parent.resolvehostnames)
1203 {
1204 }
1205
1206 void ConnectClass::Update(const ConnectClass* src)
1207 {
1208         config = src->config;
1209         type = src->type;
1210         fakelag = src->fakelag;
1211         name = src->name;
1212         registration_timeout = src->registration_timeout;
1213         host = src->host;
1214         pingtime = src->pingtime;
1215         softsendqmax = src->softsendqmax;
1216         hardsendqmax = src->hardsendqmax;
1217         recvqmax = src->recvqmax;
1218         penaltythreshold = src->penaltythreshold;
1219         commandrate = src->commandrate;
1220         maxlocal = src->maxlocal;
1221         maxglobal = src->maxglobal;
1222         maxconnwarn = src->maxconnwarn;
1223         maxchans = src->maxchans;
1224         limit = src->limit;
1225         resolvehostnames = src->resolvehostnames;
1226 }