src/users.cpp

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