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