src/users.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
   6  * See: http://www.inspircd.org/wiki/index.php/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "configreader.h"
  16 #include "channels.h"
  17 #include "users.h"
  18 #include <stdarg.h>
  19 #include "socketengine.h"
  20 #include "wildcard.h"
  21 #include "xline.h"
  22 #include "commands/cmd_whowas.h"
  23
  24 static unsigned long already_sent[MAX_DESCRIPTORS] = {0};
  25
  26 /* XXX: Used for speeding up WriteCommon operations */
  27 unsigned long uniq_id = 0;
  28
  29 bool InitTypes(ServerConfig* conf, const char* tag)
  30 {
  31         if (conf->opertypes.size())
  32         {
  33                 for (opertype_t::iterator n = conf->opertypes.begin(); n != conf->opertypes.end(); n++)
  34                 {
  35                         if (n->second)
  36                                 delete[] n->second;
  37                 }
  38         }
  39
  40         conf->opertypes.clear();
  41         return true;
  42 }
  43
  44 bool InitClasses(ServerConfig* conf, const char* tag)
  45 {
  46         if (conf->operclass.size())
  47         {
  48                 for (operclass_t::iterator n = conf->operclass.begin(); n != conf->operclass.end(); n++)
  49                 {
  50                         if (n->second)
  51                                 delete[] n->second;
  52                 }
  53         }
  54
  55         conf->operclass.clear();
  56         return true;
  57 }
  58
  59 bool DoType(ServerConfig* conf, const char* tag, char** entries, ValueList &values, int* types)
  60 {
  61         const char* TypeName = values[0].GetString();
  62         const char* Classes = values[1].GetString();
  63
  64         conf->opertypes[TypeName] = strnewdup(Classes);
  65         return true;
  66 }
  67
  68 bool DoClass(ServerConfig* conf, const char* tag, char** entries, ValueList &values, int* types)
  69 {
  70         const char* ClassName = values[0].GetString();
  71         const char* CommandList = values[1].GetString();
  72
  73         conf->operclass[ClassName] = strnewdup(CommandList);
  74         return true;
  75 }
  76
  77 bool DoneClassesAndTypes(ServerConfig* conf, const char* tag)
  78 {
  79         return true;
  80 }
  81
  82 std::string userrec::ProcessNoticeMasks(const char *sm)
  83 {
  84         bool adding = true, oldadding = false;
  85         const char *c = sm;
  86         std::string output;
  87
  88         while (c && *c)
  89         {
  90                 switch (*c)
  91                 {
  92                         case '+':
  93                                 adding = true;
  94                         break;
  95                         case '-':
  96                                 adding = false;
  97                         break;
  98                         case '*':
  99                                 for (unsigned char d = 'A'; d <= 'z'; d++)
 100                                 {
 101                                         if (ServerInstance->SNO->IsEnabled(d))
 102                                         {
 103                                                 if ((!IsNoticeMaskSet(d) && adding) || (IsNoticeMaskSet(d) && !adding))
 104                                                 {
 105                                                         if ((oldadding != adding) || (!output.length()))
 106                                                                 output += (adding ? '+' : '-');
 107
 108                                                         this->SetNoticeMask(d, adding);
 109
 110                                                         output += d;
 111                                                 }
 112                                         }
 113                                         oldadding = adding;
 114                                 }
 115                         break;
 116                         default:
 117                                 if ((*c >= 'A') && (*c <= 'z') && (ServerInstance->SNO->IsEnabled(*c)))
 118                                 {
 119                                         if ((!IsNoticeMaskSet(*c) && adding) || (IsNoticeMaskSet(*c) && !adding))
 120                                         {
 121                                                 if ((oldadding != adding) || (!output.length()))
 122                                                         output += (adding ? '+' : '-');
 123
 124                                                 this->SetNoticeMask(*c, adding);
 125
 126                                                 output += *c;
 127                                         }
 128                                 }
 129                                 oldadding = adding;
 130                         break;
 131                 }
 132
 133                 *c++;
 134         }
 135
 136         return output;
 137 }
 138
 139 void userrec::StartDNSLookup()
 140 {
 141         try
 142         {
 143                 bool cached;
 144                 const char* ip = this->GetIPString();
 145
 146                 /* Special case for 4in6 (Have i mentioned i HATE 4in6?) */
 147                 if (!strncmp(ip, "0::ffff:", 8))
 148                         res_reverse = new UserResolver(this->ServerInstance, this, ip + 8, DNS_QUERY_PTR4, cached);
 149                 else
 150                         res_reverse = new UserResolver(this->ServerInstance, this, ip, this->GetProtocolFamily() == AF_INET ? DNS_QUERY_PTR4 : DNS_QUERY_PTR6, cached);
 151
 152                 this->ServerInstance->AddResolver(res_reverse, cached);
 153         }
 154         catch (CoreException& e)
 155         {
 156                 ServerInstance->Log(DEBUG,"Error in resolver: %s",e.GetReason());
 157         }
 158 }
 159
 160 UserResolver::UserResolver(InspIRCd* Instance, userrec* user, std::string to_resolve, QueryType qt, bool &cache) :
 161         Resolver(Instance, to_resolve, qt, cache), bound_user(user)
 162 {
 163         this->fwd = (qt == DNS_QUERY_A || qt == DNS_QUERY_AAAA);
 164         this->bound_fd = user->GetFd();
 165 }
 166
 167 void UserResolver::OnLookupComplete(const std::string &result, unsigned int ttl, bool cached)
 168 {
 169         if ((!this->fwd) && (ServerInstance->SE->GetRef(this->bound_fd) == this->bound_user))
 170         {
 171                 this->bound_user->stored_host = result;
 172                 try
 173                 {
 174                         /* Check we didnt time out */
 175                         if (this->bound_user->registered != REG_ALL)
 176                         {
 177                                 bool cached;
 178 #ifdef IPV6
 179                                 if (this->bound_user->GetProtocolFamily() == AF_INET6)
 180                                 {
 181                                         /* IPV6 forward lookup (with possibility of 4in6) */
 182                                         const char* ip = this->bound_user->GetIPString();
 183                                         bound_user->res_forward = new UserResolver(this->ServerInstance, this->bound_user, result, (!strncmp(ip, "0::ffff:", 8) ? DNS_QUERY_A : DNS_QUERY_AAAA), cached);
 184                                 }
 185                                 else
 186                                         /* IPV4 lookup (mixed protocol mode) */
 187 #endif
 188                                 /* IPV4 lookup (ipv4 only mode) */
 189                                 bound_user->res_forward = new UserResolver(this->ServerInstance, this->bound_user, result, DNS_QUERY_A, cached);
 190                                 this->ServerInstance->AddResolver(bound_user->res_forward, cached);
 191                         }
 192                 }
 193                 catch (CoreException& e)
 194                 {
 195                         ServerInstance->Log(DEBUG,"Error in resolver: %s",e.GetReason());
 196                 }
 197         }
 198         else if ((this->fwd) && (ServerInstance->SE->GetRef(this->bound_fd) == this->bound_user))
 199         {
 200                 /* Both lookups completed */
 201                 std::string result2("0::ffff:");
 202                 result2.append(result);
 203                 if (this->bound_user->GetIPString() == result || this->bound_user->GetIPString() == result2)
 204                 {
 205                         std::string hostname = this->bound_user->stored_host;
 206                         if (hostname.length() < 65)
 207                         {
 208                                 /* Check we didnt time out */
 209                                 if ((this->bound_user->registered != REG_ALL) && (!this->bound_user->dns_done))
 210                                 {
 211                                         /* Hostnames starting with : are not a good thing (tm) */
 212                                         if (*(hostname.c_str()) == ':')
 213                                                 hostname.insert(0, "0");
 214
 215                                         this->bound_user->WriteServ("NOTICE Auth :*** Found your hostname (%s)%s", hostname.c_str(), (cached ? " -- cached" : ""));
 216                                         this->bound_user->dns_done = true;
 217                                         strlcpy(this->bound_user->dhost, hostname.c_str(),64);
 218                                         strlcpy(this->bound_user->host, hostname.c_str(),64);
 219                                         /* Invalidate cache */
 220                                         this->bound_user->InvalidateCache();
 221                                 }
 222                         }
 223                         else
 224                         {
 225                                 if (!this->bound_user->dns_done)
 226                                 {
 227                                         this->bound_user->WriteServ("NOTICE Auth :*** Your hostname is longer than the maximum of 64 characters, using your IP address (%s) instead.", this->bound_user->GetIPString());
 228                                         this->bound_user->dns_done = true;
 229                                 }
 230                         }
 231                 }
 232                 else
 233                 {
 234                         if (!this->bound_user->dns_done)
 235                         {
 236                                 this->bound_user->WriteServ("NOTICE Auth :*** Your hostname does not match up with your IP address. Sorry, using your IP address (%s) instead.", this->bound_user->GetIPString());
 237                                 this->bound_user->dns_done = true;
 238                         }
 239                 }
 240         }
 241 }
 242
 243 void UserResolver::OnError(ResolverError e, const std::string &errormessage)
 244 {
 245         if (ServerInstance->SE->GetRef(this->bound_fd) == this->bound_user)
 246         {
 247                 /* Since dns timeout is implemented outside of the resolver, this was a race condition that could result in this message being sent *after*
 248                  * the user was fully connected. This check fixes that issue  - Special */
 249                 if (!this->bound_user->dns_done)
 250                 {
 251                         /* Error message here */
 252                         this->bound_user->WriteServ("NOTICE Auth :*** Could not resolve your hostname: %s; using your IP address (%s) instead.", errormessage.c_str(), this->bound_user->GetIPString());
 253                         this->bound_user->dns_done = true;
 254                 }
 255         }
 256 }
 257
 258
 259 bool userrec::IsNoticeMaskSet(unsigned char sm)
 260 {
 261         return (snomasks[sm-65]);
 262 }
 263
 264 void userrec::SetNoticeMask(unsigned char sm, bool value)
 265 {
 266         snomasks[sm-65] = value;
 267 }
 268
 269 const char* userrec::FormatNoticeMasks()
 270 {
 271         static char data[MAXBUF];
 272         int offset = 0;
 273
 274         for (int n = 0; n < 64; n++)
 275         {
 276                 if (snomasks[n])
 277                         data[offset++] = n+65;
 278         }
 279
 280         data[offset] = 0;
 281         return data;
 282 }
 283
 284
 285
 286 bool userrec::IsModeSet(unsigned char m)
 287 {
 288         return (modes[m-65]);
 289 }
 290
 291 void userrec::SetMode(unsigned char m, bool value)
 292 {
 293         modes[m-65] = value;
 294 }
 295
 296 const char* userrec::FormatModes()
 297 {
 298         static char data[MAXBUF];
 299         int offset = 0;
 300         for (int n = 0; n < 64; n++)
 301         {
 302                 if (modes[n])
 303                         data[offset++] = n+65;
 304         }
 305         data[offset] = 0;
 306         return data;
 307 }
 308
 309 void userrec::DecrementModes()
 310 {
 311         for (int n = 0; n < 64; n++)
 312         {
 313                 if (modes[n])
 314                 {
 315                         ModeHandler* mh = ServerInstance->Modes->FindMode(n+65, MODETYPE_USER);
 316                         if (mh)
 317                                 mh->ChangeCount(-1);
 318                 }
 319         }
 320 }
 321
 322 userrec::userrec(InspIRCd* Instance) : ServerInstance(Instance)
 323 {
 324         // the PROPER way to do it, AVOID bzero at *ALL* costs
 325         *password = *nick = *ident = *host = *dhost = *fullname = *awaymsg = *oper = 0;
 326         server = (char*)Instance->FindServerNamePtr(Instance->Config->ServerName);
 327         reset_due = ServerInstance->Time();
 328         age = ServerInstance->Time(true);
 329         lines_in = lastping = signon = idle_lastmsg = nping = registered = 0;
 330         ChannelCount = timeout = flood = bytes_in = bytes_out = cmds_in = cmds_out = 0;
 331         muted = exempt = haspassed = dns_done = false;
 332         fd = -1;
 333         recvq.clear();
 334         sendq.clear();
 335         WriteError.clear();
 336         res_forward = res_reverse = NULL;
 337         Visibility = NULL;
 338         ip = NULL;
 339         chans.clear();
 340         invites.clear();
 341         memset(modes,0,sizeof(modes));
 342         memset(snomasks,0,sizeof(snomasks));
 343         /* Invalidate cache */
 344         operquit = cached_fullhost = cached_hostip = cached_makehost = cached_fullrealhost = NULL;
 345 }
 346
 347 void userrec::RemoveCloneCounts()
 348 {
 349         clonemap::iterator x = ServerInstance->local_clones.find(this->GetIPString());
 350         if (x != ServerInstance->local_clones.end())
 351         {
 352                 x->second--;
 353                 if (!x->second)
 354                 {
 355                         ServerInstance->local_clones.erase(x);
 356                 }
 357         }
 358
 359         clonemap::iterator y = ServerInstance->global_clones.find(this->GetIPString());
 360         if (y != ServerInstance->global_clones.end())
 361         {
 362                 y->second--;
 363                 if (!y->second)
 364                 {
 365                         ServerInstance->global_clones.erase(y);
 366                 }
 367         }
 368 }
 369
 370 userrec::~userrec()
 371 {
 372         this->InvalidateCache();
 373         this->DecrementModes();
 374         if (operquit)
 375                 free(operquit);
 376         if (ip)
 377         {
 378                 this->RemoveCloneCounts();
 379
 380                 if (this->GetProtocolFamily() == AF_INET)
 381                 {
 382                         delete (sockaddr_in*)ip;
 383                 }
 384 #ifdef SUPPORT_IP6LINKS
 385                 else
 386                 {
 387                         delete (sockaddr_in6*)ip;
 388                 }
 389 #endif
 390         }
 391 }
 392
 393 char* userrec::MakeHost()
 394 {
 395         if (this->cached_makehost)
 396                 return this->cached_makehost;
 397
 398         char nhost[MAXBUF];
 399         /* This is much faster than snprintf */
 400         char* t = nhost;
 401         for(char* n = ident; *n; n++)
 402                 *t++ = *n;
 403         *t++ = '@';
 404         for(char* n = host; *n; n++)
 405                 *t++ = *n;
 406         *t = 0;
 407
 408         this->cached_makehost = strdup(nhost);
 409
 410         return this->cached_makehost;
 411 }
 412
 413 char* userrec::MakeHostIP()
 414 {
 415         if (this->cached_hostip)
 416                 return this->cached_hostip;
 417
 418         char ihost[MAXBUF];
 419         /* This is much faster than snprintf */
 420         char* t = ihost;
 421         for(char* n = ident; *n; n++)
 422                 *t++ = *n;
 423         *t++ = '@';
 424         for(const char* n = this->GetIPString(); *n; n++)
 425                 *t++ = *n;
 426         *t = 0;
 427
 428         this->cached_hostip = strdup(ihost);
 429
 430         return this->cached_hostip;
 431 }
 432
 433 void userrec::CloseSocket()
 434 {
 435         shutdown(this->fd,2);
 436         close(this->fd);
 437 }
 438
 439 char* userrec::GetFullHost()
 440 {
 441         if (this->cached_fullhost)
 442                 return this->cached_fullhost;
 443
 444         char result[MAXBUF];
 445         char* t = result;
 446         for(char* n = nick; *n; n++)
 447                 *t++ = *n;
 448         *t++ = '!';
 449         for(char* n = ident; *n; n++)
 450                 *t++ = *n;
 451         *t++ = '@';
 452         for(char* n = dhost; *n; n++)
 453                 *t++ = *n;
 454         *t = 0;
 455
 456         this->cached_fullhost = strdup(result);
 457
 458         return this->cached_fullhost;
 459 }
 460
 461 char* userrec::MakeWildHost()
 462 {
 463         static char nresult[MAXBUF];
 464         char* t = nresult;
 465         *t++ = '*';     *t++ = '!';
 466         *t++ = '*';     *t++ = '@';
 467         for(char* n = dhost; *n; n++)
 468                 *t++ = *n;
 469         *t = 0;
 470         return nresult;
 471 }
 472
 473 int userrec::ReadData(void* buffer, size_t size)
 474 {
 475         if (IS_LOCAL(this))
 476         {
 477 #ifndef WIN32
 478                 return read(this->fd, buffer, size);
 479 #else
 480                 return recv(this->fd, (char*)buffer, size, 0);
 481 #endif
 482         }
 483         else
 484                 return 0;
 485 }
 486
 487
 488 char* userrec::GetFullRealHost()
 489 {
 490         if (this->cached_fullrealhost)
 491                 return this->cached_fullrealhost;
 492
 493         char fresult[MAXBUF];
 494         char* t = fresult;
 495         for(char* n = nick; *n; n++)
 496                 *t++ = *n;
 497         *t++ = '!';
 498         for(char* n = ident; *n; n++)
 499                 *t++ = *n;
 500         *t++ = '@';
 501         for(char* n = host; *n; n++)
 502                 *t++ = *n;
 503         *t = 0;
 504
 505         this->cached_fullrealhost = strdup(fresult);
 506
 507         return this->cached_fullrealhost;
 508 }
 509
 510 bool userrec::IsInvited(const irc::string &channel)
 511 {
 512         for (InvitedList::iterator i = invites.begin(); i != invites.end(); i++)
 513         {
 514                 if (channel == *i)
 515                 {
 516                         return true;
 517                 }
 518         }
 519         return false;
 520 }
 521
 522 InvitedList* userrec::GetInviteList()
 523 {
 524         return &invites;
 525 }
 526
 527 void userrec::InviteTo(const irc::string &channel)
 528 {
 529         invites.push_back(channel);
 530 }
 531
 532 void userrec::RemoveInvite(const irc::string &channel)
 533 {
 534         for (InvitedList::iterator i = invites.begin(); i != invites.end(); i++)
 535         {
 536                 if (channel == *i)
 537                 {
 538                         invites.erase(i);
 539                         return;
 540                 }
 541         }
 542 }
 543
 544 bool userrec::HasPermission(const std::string &command)
 545 {
 546         char* mycmd;
 547         char* savept;
 548         char* savept2;
 549
 550         /*
 551          * users on remote servers can completely bypass all permissions based checks.
 552          * This prevents desyncs when one server has different type/class tags to another.
 553          * That having been said, this does open things up to the possibility of source changes
 554          * allowing remote kills, etc - but if they have access to the src, they most likely have
 555          * access to the conf - so it's an end to a means either way.
 556          */
 557         if (!IS_LOCAL(this))
 558                 return true;
 559
 560         // are they even an oper at all?
 561         if (IS_OPER(this))
 562         {
 563                 opertype_t::iterator iter_opertype = ServerInstance->Config->opertypes.find(this->oper);
 564                 if (iter_opertype != ServerInstance->Config->opertypes.end())
 565                 {
 566                         char* Classes = strdup(iter_opertype->second);
 567                         char* myclass = strtok_r(Classes," ",&savept);
 568                         while (myclass)
 569                         {
 570                                 operclass_t::iterator iter_operclass = ServerInstance->Config->operclass.find(myclass);
 571                                 if (iter_operclass != ServerInstance->Config->operclass.end())
 572                                 {
 573                                         char* CommandList = strdup(iter_operclass->second);
 574                                         mycmd = strtok_r(CommandList," ",&savept2);
 575                                         while (mycmd)
 576                                         {
 577                                                 if ((!strcasecmp(mycmd,command.c_str())) || (*mycmd == '*'))
 578                                                 {
 579                                                         free(Classes);
 580                                                         free(CommandList);
 581                                                         return true;
 582                                                 }
 583                                                 mycmd = strtok_r(NULL," ",&savept2);
 584                                         }
 585                                         free(CommandList);
 586                                 }
 587                                 myclass = strtok_r(NULL," ",&savept);
 588                         }
 589                         free(Classes);
 590                 }
 591         }
 592         return false;
 593 }
 594
 595 /** NOTE: We cannot pass a const reference to this method.
 596  * The string is changed by the workings of the method,
 597  * so that if we pass const ref, we end up copying it to
 598  * something we can change anyway. Makes sense to just let
 599  * the compiler do that copy for us.
 600  */
 601 bool userrec::AddBuffer(std::string a)
 602 {
 603         try
 604         {
 605                 std::string::size_type i = a.rfind('\r');
 606
 607                 while (i != std::string::npos)
 608                 {
 609                         a.erase(i, 1);
 610                         i = a.rfind('\r');
 611                 }
 612
 613                 if (a.length())
 614                         recvq.append(a);
 615
 616                 if (recvq.length() > (unsigned)this->recvqmax)
 617                 {
 618                         this->SetWriteError("RecvQ exceeded");
 619                         ServerInstance->WriteOpers("*** User %s RecvQ of %d exceeds connect class maximum of %d",this->nick,recvq.length(),this->recvqmax);
 620                         return false;
 621                 }
 622
 623                 return true;
 624         }
 625
 626         catch (...)
 627         {
 628                 ServerInstance->Log(DEBUG,"Exception in userrec::AddBuffer()");
 629                 return false;
 630         }
 631 }
 632
 633 bool userrec::BufferIsReady()
 634 {
 635         return (recvq.find('\n') != std::string::npos);
 636 }
 637
 638 void userrec::ClearBuffer()
 639 {
 640         recvq.clear();
 641 }
 642
 643 std::string userrec::GetBuffer()
 644 {
 645         try
 646         {
 647                 if (!recvq.length())
 648                         return "";
 649
 650                 /* Strip any leading \r or \n off the string.
 651                  * Usually there are only one or two of these,
 652                  * so its is computationally cheap to do.
 653                  */
 654                 std::string::iterator t = recvq.begin();
 655                 while (t != recvq.end() && (*t == '\r' || *t == '\n'))
 656                 {
 657                         recvq.erase(t);
 658                         t = recvq.begin();
 659                 }
 660
 661                 for (std::string::iterator x = recvq.begin(); x != recvq.end(); x++)
 662                 {
 663                         /* Find the first complete line, return it as the
 664                          * result, and leave the recvq as whats left
 665                          */
 666                         if (*x == '\n')
 667                         {
 668                                 std::string ret = std::string(recvq.begin(), x);
 669                                 recvq.erase(recvq.begin(), x + 1);
 670                                 return ret;
 671                         }
 672                 }
 673                 return "";
 674         }
 675
 676         catch (...)
 677         {
 678                 ServerInstance->Log(DEBUG,"Exception in userrec::GetBuffer()");
 679                 return "";
 680         }
 681 }
 682
 683 void userrec::AddWriteBuf(const std::string &data)
 684 {
 685         if (*this->GetWriteError())
 686                 return;
 687
 688         if (sendq.length() + data.length() > (unsigned)this->sendqmax)
 689         {
 690                 /*
 691                  * Fix by brain - Set the error text BEFORE calling writeopers, because
 692                  * if we dont it'll recursively  call here over and over again trying
 693                  * to repeatedly add the text to the sendq!
 694                  */
 695                 this->SetWriteError("SendQ exceeded");
 696                 ServerInstance->WriteOpers("*** User %s SendQ of %d exceeds connect class maximum of %d",this->nick,sendq.length() + data.length(),this->sendqmax);
 697                 return;
 698         }
 699
 700         try
 701         {
 702                 if (data.length() > MAXBUF - 2) /* MAXBUF has a value of 514, to account for line terminators */
 703                         sendq.append(data.substr(0,MAXBUF - 4)).append("\r\n"); /* MAXBUF-4 = 510 */
 704                 else
 705                         sendq.append(data);
 706         }
 707         catch (...)
 708         {
 709                 this->SetWriteError("SendQ exceeded");
 710                 ServerInstance->WriteOpers("*** User %s SendQ got an exception",this->nick);
 711         }
 712 }
 713
 714 // send AS MUCH OF THE USERS SENDQ as we are able to (might not be all of it)
 715 void userrec::FlushWriteBuf()
 716 {
 717         try
 718         {
 719                 if ((this->fd == FD_MAGIC_NUMBER) || (*this->GetWriteError()))
 720                 {
 721                         sendq.clear();
 722                 }
 723                 if ((sendq.length()) && (this->fd != FD_MAGIC_NUMBER))
 724                 {
 725                         int old_sendq_length = sendq.length();
 726 #ifndef WIN32
 727                 int n_sent = write(this->fd, this->sendq.data(), this->sendq.length());
 728 #else
 729                 int n_sent = send(this->fd, (const char*)this->sendq.data(), this->sendq.length(), 0);
 730 #endif
 731                         if (n_sent == -1)
 732                         {
 733                                 if (errno == EAGAIN)
 734                                 {
 735                                         /* The socket buffer is full. This isnt fatal,
 736                                          * try again later.
 737                                          */
 738                                         this->ServerInstance->SE->WantWrite(this);
 739                                 }
 740                                 else
 741                                 {
 742                                         /* Fatal error, set write error and bail
 743                                          */
 744                                         this->SetWriteError(errno ? strerror(errno) : "EOF from client");
 745                                         return;
 746                                 }
 747                         }
 748                         else
 749                         {
 750                                 /* advance the queue */
 751                                 if (n_sent)
 752                                         this->sendq = this->sendq.substr(n_sent);
 753                                 /* update the user's stats counters */
 754                                 this->bytes_out += n_sent;
 755                                 this->cmds_out++;
 756                                 if (n_sent != old_sendq_length)
 757                                         this->ServerInstance->SE->WantWrite(this);
 758                         }
 759                 }
 760         }
 761
 762         catch (...)
 763         {
 764                 ServerInstance->Log(DEBUG,"Exception in userrec::FlushWriteBuf()");
 765         }
 766
 767         if (this->sendq.empty())
 768         {
 769                 FOREACH_MOD(I_OnBufferFlushed,OnBufferFlushed(this));
 770         }
 771 }
 772
 773 void userrec::SetWriteError(const std::string &error)
 774 {
 775         try
 776         {
 777                 // don't try to set the error twice, its already set take the first string.
 778                 if (this->WriteError.empty())
 779                         this->WriteError = error;
 780         }
 781
 782         catch (...)
 783         {
 784                 ServerInstance->Log(DEBUG,"Exception in userrec::SetWriteError()");
 785         }
 786 }
 787
 788 const char* userrec::GetWriteError()
 789 {
 790         return this->WriteError.c_str();
 791 }
 792
 793 void userrec::Oper(const std::string &opertype)
 794 {
 795         try
 796         {
 797                 this->modes[UM_OPERATOR] = 1;
 798                 this->WriteServ("MODE %s :+o", this->nick);
 799                 FOREACH_MOD(I_OnOper, OnOper(this, opertype));
 800                 ServerInstance->Log(DEFAULT,"OPER: %s!%s@%s opered as type: %s", this->nick, this->ident, this->host, opertype.c_str());
 801                 strlcpy(this->oper, opertype.c_str(), NICKMAX - 1);
 802                 ServerInstance->all_opers.push_back(this);
 803                 FOREACH_MOD(I_OnPostOper,OnPostOper(this, opertype));
 804         }
 805
 806         catch (...)
 807         {
 808                 ServerInstance->Log(DEBUG,"Exception in userrec::Oper()");
 809         }
 810 }
 811
 812 void userrec::UnOper()
 813 {
 814         try
 815         {
 816                 if (IS_OPER(this))
 817                 {
 818                         // unset their oper type (what IS_OPER checks), and remove +o
 819                         *this->oper = 0;
 820                         this->modes[UM_OPERATOR] = 0;
 821
 822                         // remove them from the opers list.
 823                         for (std::vector<userrec*>::iterator a = ServerInstance->all_opers.begin(); a < ServerInstance->all_opers.end(); a++)
 824                         {
 825                                 if (*a == this)
 826                                 {
 827                                         ServerInstance->all_opers.erase(a);
 828                                         return;
 829                                 }
 830                         }
 831                 }
 832         }
 833
 834         catch (...)
 835         {
 836                 ServerInstance->Log(DEBUG,"Exception in userrec::UnOper()");
 837         }
 838 }
 839
 840 void userrec::QuitUser(InspIRCd* Instance, userrec *user, const std::string &quitreason, const char* operreason)
 841 {
 842         user->muted = true;
 843         Instance->GlobalCulls.AddItem(user, quitreason.c_str(), operreason);
 844 }
 845
 846 /* adds or updates an entry in the whowas list */
 847 void userrec::AddToWhoWas()
 848 {
 849         command_t* whowas_command = ServerInstance->Parser->GetHandler("WHOWAS");
 850         if (whowas_command)
 851         {
 852                 std::deque<classbase*> params;
 853                 params.push_back(this);
 854                 whowas_command->HandleInternal(WHOWAS_ADD, params);
 855         }
 856 }
 857
 858 /* add a client connection to the sockets list */
 859 void userrec::AddClient(InspIRCd* Instance, int socket, int port, bool iscached, int socketfamily, sockaddr* ip)
 860 {
 861         std::string tempnick = ConvToStr(socket) + "-unknown";
 862         user_hash::iterator iter = Instance->clientlist->find(tempnick);
 863         char ipaddr[MAXBUF];
 864 #ifdef IPV6
 865         if (socketfamily == AF_INET6)
 866                 inet_ntop(AF_INET6, &((const sockaddr_in6*)ip)->sin6_addr, ipaddr, sizeof(ipaddr));
 867         else
 868 #endif
 869         inet_ntop(AF_INET, &((const sockaddr_in*)ip)->sin_addr, ipaddr, sizeof(ipaddr));
 870         userrec* New;
 871         int j = 0;
 872
 873         Instance->unregistered_count++;
 874
 875         /*
 876          * fix by brain.
 877          * as these nicknames are 'RFC impossible', we can be sure nobody is going to be
 878          * using one as a registered connection. As they are per fd, we can also safely assume
 879          * that we wont have collisions. Therefore, if the nick exists in the list, its only
 880          * used by a dead socket, erase the iterator so that the new client may reclaim it.
 881          * this was probably the cause of 'server ignores me when i hammer it with reconnects'
 882          * issue in earlier alphas/betas
 883          */
 884         if (iter != Instance->clientlist->end())
 885         {
 886                 userrec* goner = iter->second;
 887                 DELETE(goner);
 888                 Instance->clientlist->erase(iter);
 889         }
 890
 891         New = new userrec(Instance);
 892         (*(Instance->clientlist))[tempnick] = New;
 893         New->fd = socket;
 894         strlcpy(New->nick,tempnick.c_str(),NICKMAX-1);
 895
 896         New->server = Instance->FindServerNamePtr(Instance->Config->ServerName);
 897         /* We don't need range checking here, we KNOW 'unknown\0' will fit into the ident field. */
 898         strcpy(New->ident, "unknown");
 899
 900         New->registered = REG_NONE;
 901         New->signon = Instance->Time() + Instance->Config->dns_timeout;
 902         New->lastping = 1;
 903
 904         New->SetSockAddr(socketfamily, ipaddr, port);
 905
 906         /* Smarter than your average bear^H^H^H^Hset of strlcpys. */
 907         for (const char* temp = New->GetIPString(); *temp && j < 64; temp++, j++)
 908                 New->dhost[j] = New->host[j] = *temp;
 909         New->dhost[j] = New->host[j] = 0;
 910
 911         Instance->AddLocalClone(New);
 912         Instance->AddGlobalClone(New);
 913
 914         /*
 915          * First class check. We do this again in FullConnect after DNS is done, and NICK/USER is recieved.
 916          * See my note down there for why this is required. DO NOT REMOVE. :) -- w00t
 917          */
 918         ConnectClass* i = New->GetClass();
 919
 920         if (!i)
 921         {
 922                 userrec::QuitUser(Instance, New, "Access denied by configuration");
 923                 return;
 924         }
 925
 926         New->CheckClass();
 927
 928         New->pingmax = i->GetPingTime();
 929         New->nping = Instance->Time() + i->GetPingTime() + Instance->Config->dns_timeout;
 930         New->timeout = Instance->Time() + i->GetRegTimeout();
 931         New->flood = i->GetFlood();
 932         New->threshold = i->GetThreshold();
 933         New->sendqmax = i->GetSendqMax();
 934         New->recvqmax = i->GetRecvqMax();
 935
 936         Instance->local_users.push_back(New);
 937
 938         if ((Instance->local_users.size() > Instance->Config->SoftLimit) || (Instance->local_users.size() >= MAXCLIENTS))
 939         {
 940                 Instance->WriteOpers("*** Warning: softlimit value has been reached: %d clients", Instance->Config->SoftLimit);
 941                 userrec::QuitUser(Instance, New,"No more connections allowed");
 942                 return;
 943         }
 944
 945         /*
 946          * XXX -
 947          * this is done as a safety check to keep the file descriptors within range of fd_ref_table.
 948          * its a pretty big but for the moment valid assumption:
 949          * file descriptors are handed out starting at 0, and are recycled as theyre freed.
 950          * therefore if there is ever an fd over 65535, 65536 clients must be connected to the
 951          * irc server at once (or the irc server otherwise initiating this many connections, files etc)
 952          * which for the time being is a physical impossibility (even the largest networks dont have more
 953          * than about 10,000 users on ONE server!)
 954          */
 955 #ifndef WINDOWS
 956         if ((unsigned int)socket >= MAX_DESCRIPTORS)
 957         {
 958                 userrec::QuitUser(Instance, New, "Server is full");
 959                 return;
 960         }
 961 #endif
 962
 963         New->exempt = (Instance->XLines->matches_exception(New) != NULL);
 964         if (!New->exempt)
 965         {
 966                 ZLine* r = Instance->XLines->matches_zline(ipaddr);
 967                 if (r)
 968                 {
 969                         char reason[MAXBUF];
 970                         if (*Instance->Config->MoronBanner)
 971                                 New->WriteServ("NOTICE %s :*** %s", New->nick, Instance->Config->MoronBanner);
 972                         snprintf(reason,MAXBUF,"Z-Lined: %s",r->reason);
 973                         userrec::QuitUser(Instance, New, reason);
 974                         return;
 975                 }
 976         }
 977
 978         if (socket > -1)
 979         {
 980                 if (!Instance->SE->AddFd(New))
 981                 {
 982                         userrec::QuitUser(Instance, New, "Internal error handling connection");
 983                         return;
 984                 }
 985         }
 986
 987         /* NOTE: even if dns lookups are *off*, we still need to display this.
 988          * BOPM and other stuff requires it.
 989          */
 990         New->WriteServ("NOTICE Auth :*** Looking up your hostname...");
 991 }
 992
 993 unsigned long userrec::GlobalCloneCount()
 994 {
 995         clonemap::iterator x = ServerInstance->global_clones.find(this->GetIPString());
 996         if (x != ServerInstance->global_clones.end())
 997                 return x->second;
 998         else
 999                 return 0;
1000 }
1001
1002 unsigned long userrec::LocalCloneCount()
1003 {
1004         clonemap::iterator x = ServerInstance->local_clones.find(this->GetIPString());
1005         if (x != ServerInstance->local_clones.end())
1006                 return x->second;
1007         else
1008                 return 0;
1009 }
1010
1011 /*
1012  * Check class restrictions
1013  */
1014 void userrec::CheckClass()
1015 {
1016         ConnectClass* a = this->GetClass();
1017
1018         if ((!a) || (a->GetType() == CC_DENY))
1019         {
1020                 userrec::QuitUser(ServerInstance, this, "Unauthorised connection");
1021                 return;
1022         }
1023         else if ((a->GetMaxLocal()) && (this->LocalCloneCount() > a->GetMaxLocal()))
1024         {
1025                 userrec::QuitUser(ServerInstance, this, "No more connections allowed from your host via this connect class (local)");
1026                 ServerInstance->WriteOpers("*** WARNING: maximum LOCAL connections (%ld) exceeded for IP %s", a->GetMaxLocal(), this->GetIPString());
1027                 return;
1028         }
1029         else if ((a->GetMaxGlobal()) && (this->GlobalCloneCount() > a->GetMaxGlobal()))
1030         {
1031                 userrec::QuitUser(ServerInstance, this, "No more connections allowed from your host via this connect class (global)");
1032                 ServerInstance->WriteOpers("*** WARNING: maximum GLOBAL connections (%ld) exceeded for IP %s", a->GetMaxGlobal(), this->GetIPString());
1033                 return;
1034         }
1035 }
1036
1037 void userrec::FullConnect()
1038 {
1039         ServerInstance->Log(DEBUG,"FullConnect");
1040         ServerInstance->stats->statsConnects++;
1041         this->idle_lastmsg = ServerInstance->Time();
1042
1043         /*
1044          * You may be thinking "wtf, we checked this in userrec::AddClient!" - and yes, we did, BUT.
1045          * At the time AddClient is called, we don't have a resolved host, by here we probably do - which
1046          * may put the user into a totally seperate class with different restrictions! so we *must* check again.
1047          * Don't remove this! -- w00t
1048          */
1049         this->CheckClass();
1050
1051         /* Check the password, if one is required by the user's connect class.
1052          * This CANNOT be in CheckClass(), because that is called prior to PASS as well!
1053          */
1054         if ((!this->GetClass()->GetPass().empty()) && (!this->haspassed))
1055         {
1056                 userrec::QuitUser(ServerInstance, this, "Invalid password");
1057                 return;
1058         }
1059
1060         if (!this->exempt)
1061         {
1062                 GLine* r = ServerInstance->XLines->matches_gline(this);
1063
1064                 if (r)
1065                 {
1066                         this->muted = true;
1067                         char reason[MAXBUF];
1068                         if (*ServerInstance->Config->MoronBanner)
1069                                 this->WriteServ("NOTICE %s :*** %s", this->nick, ServerInstance->Config->MoronBanner);
1070                         snprintf(reason,MAXBUF,"G-Lined: %s",r->reason);
1071                         userrec::QuitUser(ServerInstance, this, reason);
1072                         return;
1073                 }
1074
1075                 KLine* n = ServerInstance->XLines->matches_kline(this);
1076
1077                 if (n)
1078                 {
1079                         this->muted = true;
1080                         char reason[MAXBUF];
1081                         if (*ServerInstance->Config->MoronBanner)
1082                                 this->WriteServ("NOTICE %s :*** %s", this, ServerInstance->Config->MoronBanner);
1083                         snprintf(reason,MAXBUF,"K-Lined: %s",n->reason);
1084                         userrec::QuitUser(ServerInstance, this, reason);
1085                         return;
1086                 }
1087         }
1088
1089         this->WriteServ("NOTICE Auth :Welcome to \002%s\002!",ServerInstance->Config->Network);
1090         this->WriteServ("001 %s :Welcome to the %s IRC Network %s!%s@%s",this->nick, ServerInstance->Config->Network, this->nick, this->ident, this->host);
1091         this->WriteServ("002 %s :Your host is %s, running version %s",this->nick,ServerInstance->Config->ServerName,VERSION);
1092         this->WriteServ("003 %s :This server was created %s %s", this->nick, __TIME__, __DATE__);
1093         this->WriteServ("004 %s %s %s %s %s %s", this->nick, ServerInstance->Config->ServerName, VERSION, ServerInstance->Modes->UserModeList().c_str(), ServerInstance->Modes->ChannelModeList().c_str(), ServerInstance->Modes->ParaModeList().c_str());
1094
1095         ServerInstance->Config->Send005(this);
1096
1097         this->ShowMOTD();
1098
1099         /* Now registered */
1100         if (ServerInstance->unregistered_count)
1101                 ServerInstance->unregistered_count--;
1102
1103         /* Trigger LUSERS output, give modules a chance too */
1104         int MOD_RESULT = 0;
1105         FOREACH_RESULT(I_OnPreCommand, OnPreCommand("LUSERS", NULL, 0, this, true, "LUSERS"));
1106         if (!MOD_RESULT)
1107                 ServerInstance->CallCommandHandler("LUSERS", NULL, 0, this);
1108
1109         /*
1110          * fix 3 by brain, move registered = 7 below these so that spurious modes and host
1111          * changes dont go out onto the network and produce 'fake direction'.
1112          */
1113         FOREACH_MOD(I_OnUserConnect,OnUserConnect(this));
1114
1115         this->registered = REG_ALL;
1116
1117         FOREACH_MOD(I_OnPostConnect,OnPostConnect(this));
1118
1119         ServerInstance->SNO->WriteToSnoMask('c',"Client connecting on port %d: %s!%s@%s [%s] [%s]", this->GetPort(), this->nick, this->ident, this->host, this->GetIPString(), this->fullname);
1120
1121         ServerInstance->Log(DEBUG,"Exit FullConnect");
1122 }
1123
1124 /** userrec::UpdateNick()
1125  * re-allocates a nick in the user_hash after they change nicknames,
1126  * returns a pointer to the new user as it may have moved
1127  */
1128 userrec* userrec::UpdateNickHash(const char* New)
1129 {
1130         try
1131         {
1132                 //user_hash::iterator newnick;
1133                 user_hash::iterator oldnick = ServerInstance->clientlist->find(this->nick);
1134
1135                 if (!strcasecmp(this->nick,New))
1136                         return oldnick->second;
1137
1138                 if (oldnick == ServerInstance->clientlist->end())
1139                         return NULL; /* doesnt exist */
1140
1141                 userrec* olduser = oldnick->second;
1142                 (*(ServerInstance->clientlist))[New] = olduser;
1143                 ServerInstance->clientlist->erase(oldnick);
1144                 return olduser;
1145         }
1146
1147         catch (...)
1148         {
1149                 ServerInstance->Log(DEBUG,"Exception in userrec::UpdateNickHash()");
1150                 return NULL;
1151         }
1152 }
1153
1154 void userrec::InvalidateCache()
1155 {
1156         /* Invalidate cache */
1157         if (cached_fullhost)
1158                 free(cached_fullhost);
1159         if (cached_hostip)
1160                 free(cached_hostip);
1161         if (cached_makehost)
1162                 free(cached_makehost);
1163         if (cached_fullrealhost)
1164                 free(cached_fullrealhost);
1165         cached_fullhost = cached_hostip = cached_makehost = cached_fullrealhost = NULL;
1166 }
1167
1168 bool userrec::ForceNickChange(const char* newnick)
1169 {
1170         try
1171         {
1172                 int MOD_RESULT = 0;
1173
1174                 this->InvalidateCache();
1175
1176                 FOREACH_RESULT(I_OnUserPreNick,OnUserPreNick(this, newnick));
1177
1178                 if (MOD_RESULT)
1179                 {
1180                         ServerInstance->stats->statsCollisions++;
1181                         return false;
1182                 }
1183
1184                 if (ServerInstance->XLines->matches_qline(newnick))
1185                 {
1186                         ServerInstance->stats->statsCollisions++;
1187                         return false;
1188                 }
1189
1190                 if (this->registered == REG_ALL)
1191                 {
1192                         return (ServerInstance->Parser->CallHandler("NICK", &newnick, 1, this) == CMD_SUCCESS);
1193                 }
1194                 return false;
1195         }
1196
1197         catch (...)
1198         {
1199                 ServerInstance->Log(DEBUG,"Exception in userrec::ForceNickChange()");
1200                 return false;
1201         }
1202 }
1203
1204 void userrec::SetSockAddr(int protocol_family, const char* ip, int port)
1205 {
1206         switch (protocol_family)
1207         {
1208 #ifdef SUPPORT_IP6LINKS
1209                 case AF_INET6:
1210                 {
1211                         sockaddr_in6* sin = new sockaddr_in6;
1212                         sin->sin6_family = AF_INET6;
1213                         sin->sin6_port = port;
1214                         inet_pton(AF_INET6, ip, &sin->sin6_addr);
1215                         this->ip = (sockaddr*)sin;
1216                 }
1217                 break;
1218 #endif
1219                 case AF_INET:
1220                 {
1221                         sockaddr_in* sin = new sockaddr_in;
1222                         sin->sin_family = AF_INET;
1223                         sin->sin_port = port;
1224                         inet_pton(AF_INET, ip, &sin->sin_addr);
1225                         this->ip = (sockaddr*)sin;
1226                 }
1227                 break;
1228                 default:
1229                         ServerInstance->Log(DEBUG,"Ut oh, I dont know protocol %d to be set on '%s'!", protocol_family, this->nick);
1230                 break;
1231         }
1232 }
1233
1234 int userrec::GetPort()
1235 {
1236         if (this->ip == NULL)
1237                 return 0;
1238
1239         switch (this->GetProtocolFamily())
1240         {
1241 #ifdef SUPPORT_IP6LINKS
1242                 case AF_INET6:
1243                 {
1244                         sockaddr_in6* sin = (sockaddr_in6*)this->ip;
1245                         return sin->sin6_port;
1246                 }
1247                 break;
1248 #endif
1249                 case AF_INET:
1250                 {
1251                         sockaddr_in* sin = (sockaddr_in*)this->ip;
1252                         return sin->sin_port;
1253                 }
1254                 break;
1255                 default:
1256                 break;
1257         }
1258         return 0;
1259 }
1260
1261 int userrec::GetProtocolFamily()
1262 {
1263         if (this->ip == NULL)
1264                 return 0;
1265
1266         sockaddr_in* sin = (sockaddr_in*)this->ip;
1267         return sin->sin_family;
1268 }
1269
1270 const char* userrec::GetIPString()
1271 {
1272         static char buf[1024];
1273
1274         if (this->ip == NULL)
1275                 return "";
1276
1277         switch (this->GetProtocolFamily())
1278         {
1279 #ifdef SUPPORT_IP6LINKS
1280                 case AF_INET6:
1281                 {
1282                         static char temp[1024];
1283
1284                         sockaddr_in6* sin = (sockaddr_in6*)this->ip;
1285                         inet_ntop(sin->sin6_family, &sin->sin6_addr, buf, sizeof(buf));
1286                         /* IP addresses starting with a : on irc are a Bad Thing (tm) */
1287                         if (*buf == ':')
1288                         {
1289                                 strlcpy(&temp[1], buf, sizeof(temp) - 1);
1290                                 *temp = '0';
1291                                 return temp;
1292                         }
1293                         return buf;
1294                 }
1295                 break;
1296 #endif
1297                 case AF_INET:
1298                 {
1299                         sockaddr_in* sin = (sockaddr_in*)this->ip;
1300                         inet_ntop(sin->sin_family, &sin->sin_addr, buf, sizeof(buf));
1301                         return buf;
1302                 }
1303                 break;
1304                 default:
1305                 break;
1306         }
1307         return "";
1308 }
1309
1310 const char* userrec::GetIPString(char* buf)
1311 {
1312         if (this->ip == NULL)
1313         {
1314                 *buf = 0;
1315                 return buf;
1316         }
1317
1318         switch (this->GetProtocolFamily())
1319         {
1320 #ifdef SUPPORT_IP6LINKS
1321                 case AF_INET6:
1322                 {
1323                         static char temp[1024];
1324
1325                         sockaddr_in6* sin = (sockaddr_in6*)this->ip;
1326                         inet_ntop(sin->sin6_family, &sin->sin6_addr, buf, sizeof(buf));
1327                         /* IP addresses starting with a : on irc are a Bad Thing (tm) */
1328                         if (*buf == ':')
1329                         {
1330                                 strlcpy(&temp[1], buf, sizeof(temp) - 1);
1331                                 *temp = '0';
1332                                 strlcpy(buf, temp, sizeof(temp));
1333                         }
1334                         return buf;
1335                 }
1336                 break;
1337 #endif
1338                 case AF_INET:
1339                 {
1340                         sockaddr_in* sin = (sockaddr_in*)this->ip;
1341                         inet_ntop(sin->sin_family, &sin->sin_addr, buf, sizeof(buf));
1342                         return buf;
1343                 }
1344                 break;
1345
1346                 default:
1347                 break;
1348         }
1349         return "";
1350 }
1351
1352 /** NOTE: We cannot pass a const reference to this method.
1353  * The string is changed by the workings of the method,
1354  * so that if we pass const ref, we end up copying it to
1355  * something we can change anyway. Makes sense to just let
1356  * the compiler do that copy for us.
1357  */
1358 void userrec::Write(std::string text)
1359 {
1360         if (!ServerInstance->SE->BoundsCheckFd(this))
1361                 return;
1362
1363         try
1364         {
1365                 /* ServerInstance->Log(DEBUG,"C[%d] <- %s", this->GetFd(), text.c_str());
1366                  * WARNING: The above debug line is VERY loud, do NOT
1367                  * enable it till we have a good way of filtering it
1368                  * out of the logs (e.g. 1.2 would be good).
1369                  */
1370                 text.append("\r\n");
1371         }
1372         catch (...)
1373         {
1374                 ServerInstance->Log(DEBUG,"Exception in userrec::Write() std::string::append");
1375                 return;
1376         }
1377
1378         if (ServerInstance->Config->GetIOHook(this->GetPort()))
1379         {
1380                 try
1381                 {
1382                         /* XXX: The lack of buffering here is NOT a bug, modules implementing this interface have to
1383                          * implement their own buffering mechanisms
1384                          */
1385                         ServerInstance->Config->GetIOHook(this->GetPort())->OnRawSocketWrite(this->fd, text.data(), text.length());
1386                 }
1387                 catch (CoreException& modexcept)
1388                 {
1389                         ServerInstance->Log(DEBUG, "%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
1390                 }
1391         }
1392         else
1393         {
1394                 this->AddWriteBuf(text);
1395         }
1396         ServerInstance->stats->statsSent += text.length();
1397         this->ServerInstance->SE->WantWrite(this);
1398 }
1399
1400 /** Write()
1401  */
1402 void userrec::Write(const char *text, ...)
1403 {
1404         va_list argsPtr;
1405         char textbuffer[MAXBUF];
1406
1407         va_start(argsPtr, text);
1408         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1409         va_end(argsPtr);
1410
1411         this->Write(std::string(textbuffer));
1412 }
1413
1414 void userrec::WriteServ(const std::string& text)
1415 {
1416         char textbuffer[MAXBUF];
1417
1418         snprintf(textbuffer,MAXBUF,":%s %s",ServerInstance->Config->ServerName,text.c_str());
1419         this->Write(std::string(textbuffer));
1420 }
1421
1422 /** WriteServ()
1423  *  Same as Write(), except `text' is prefixed with `:server.name '.
1424  */
1425 void userrec::WriteServ(const char* text, ...)
1426 {
1427         va_list argsPtr;
1428         char textbuffer[MAXBUF];
1429
1430         va_start(argsPtr, text);
1431         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1432         va_end(argsPtr);
1433
1434         this->WriteServ(std::string(textbuffer));
1435 }
1436
1437
1438 void userrec::WriteFrom(userrec *user, const std::string &text)
1439 {
1440         char tb[MAXBUF];
1441
1442         snprintf(tb,MAXBUF,":%s %s",user->GetFullHost(),text.c_str());
1443
1444         this->Write(std::string(tb));
1445 }
1446
1447
1448 /* write text from an originating user to originating user */
1449
1450 void userrec::WriteFrom(userrec *user, const char* text, ...)
1451 {
1452         va_list argsPtr;
1453         char textbuffer[MAXBUF];
1454
1455         va_start(argsPtr, text);
1456         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1457         va_end(argsPtr);
1458
1459         this->WriteFrom(user, std::string(textbuffer));
1460 }
1461
1462
1463 /* write text to an destination user from a source user (e.g. user privmsg) */
1464
1465 void userrec::WriteTo(userrec *dest, const char *data, ...)
1466 {
1467         char textbuffer[MAXBUF];
1468         va_list argsPtr;
1469
1470         va_start(argsPtr, data);
1471         vsnprintf(textbuffer, MAXBUF, data, argsPtr);
1472         va_end(argsPtr);
1473
1474         this->WriteTo(dest, std::string(textbuffer));
1475 }
1476
1477 void userrec::WriteTo(userrec *dest, const std::string &data)
1478 {
1479         dest->WriteFrom(this, data);
1480 }
1481
1482
1483 void userrec::WriteCommon(const char* text, ...)
1484 {
1485         char textbuffer[MAXBUF];
1486         va_list argsPtr;
1487
1488         if (this->registered != REG_ALL)
1489                 return;
1490
1491         va_start(argsPtr, text);
1492         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1493         va_end(argsPtr);
1494
1495         this->WriteCommon(std::string(textbuffer));
1496 }
1497
1498 void userrec::WriteCommon(const std::string &text)
1499 {
1500         try
1501         {
1502                 bool sent_to_at_least_one = false;
1503                 char tb[MAXBUF];
1504
1505                 if (this->registered != REG_ALL)
1506                         return;
1507
1508                 uniq_id++;
1509
1510                 /* We dont want to be doing this n times, just once */
1511                 snprintf(tb,MAXBUF,":%s %s",this->GetFullHost(),text.c_str());
1512                 std::string out = tb;
1513
1514                 for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1515                 {
1516                         CUList* ulist = v->first->GetUsers();
1517                         for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1518                         {
1519                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1520                                 {
1521                                         already_sent[i->first->fd] = uniq_id;
1522                                         i->first->Write(out);
1523                                         sent_to_at_least_one = true;
1524                                 }
1525                         }
1526                 }
1527
1528                 /*
1529                  * if the user was not in any channels, no users will receive the text. Make sure the user
1530                  * receives their OWN message for WriteCommon
1531                  */
1532                 if (!sent_to_at_least_one)
1533                 {
1534                         this->Write(std::string(tb));
1535                 }
1536         }
1537
1538         catch (...)
1539         {
1540                 ServerInstance->Log(DEBUG,"Exception in userrec::WriteCommon()");
1541         }
1542 }
1543
1544
1545 /* write a formatted string to all users who share at least one common
1546  * channel, NOT including the source user e.g. for use in QUIT
1547  */
1548
1549 void userrec::WriteCommonExcept(const char* text, ...)
1550 {
1551         char textbuffer[MAXBUF];
1552         va_list argsPtr;
1553
1554         va_start(argsPtr, text);
1555         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1556         va_end(argsPtr);
1557
1558         this->WriteCommonExcept(std::string(textbuffer));
1559 }
1560
1561 void userrec::WriteCommonQuit(const std::string &normal_text, const std::string &oper_text)
1562 {
1563         char tb1[MAXBUF];
1564         char tb2[MAXBUF];
1565
1566         if (this->registered != REG_ALL)
1567                 return;
1568
1569         uniq_id++;
1570         snprintf(tb1,MAXBUF,":%s QUIT :%s",this->GetFullHost(),normal_text.c_str());
1571         snprintf(tb2,MAXBUF,":%s QUIT :%s",this->GetFullHost(),oper_text.c_str());
1572         std::string out1 = tb1;
1573         std::string out2 = tb2;
1574
1575         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1576         {
1577                 CUList *ulist = v->first->GetUsers();
1578                 for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1579                 {
1580                         if (this != i->first)
1581                         {
1582                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1583                                 {
1584                                         already_sent[i->first->fd] = uniq_id;
1585                                         i->first->Write(IS_OPER(i->first) ? out2 : out1);
1586                                 }
1587                         }
1588                 }
1589         }
1590 }
1591
1592 void userrec::WriteCommonExcept(const std::string &text)
1593 {
1594         char tb1[MAXBUF];
1595         std::string out1;
1596
1597         if (this->registered != REG_ALL)
1598                 return;
1599
1600         uniq_id++;
1601         snprintf(tb1,MAXBUF,":%s %s",this->GetFullHost(),text.c_str());
1602         out1 = tb1;
1603
1604         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1605         {
1606                 CUList *ulist = v->first->GetUsers();
1607                 for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1608                 {
1609                         if (this != i->first)
1610                         {
1611                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1612                                 {
1613                                         already_sent[i->first->fd] = uniq_id;
1614                                         i->first->Write(out1);
1615                                 }
1616                         }
1617                 }
1618         }
1619
1620 }
1621
1622 void userrec::WriteWallOps(const std::string &text)
1623 {
1624         if (!IS_OPER(this) && IS_LOCAL(this))
1625                 return;
1626
1627         std::string wallop("WALLOPS :");
1628         wallop.append(text);
1629
1630         for (std::vector<userrec*>::const_iterator i = ServerInstance->local_users.begin(); i != ServerInstance->local_users.end(); i++)
1631         {
1632                 userrec* t = *i;
1633                 if (t->IsModeSet('w'))
1634                         this->WriteTo(t,wallop);
1635         }
1636 }
1637
1638 void userrec::WriteWallOps(const char* text, ...)
1639 {
1640         char textbuffer[MAXBUF];
1641         va_list argsPtr;
1642
1643         va_start(argsPtr, text);
1644         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1645         va_end(argsPtr);
1646
1647         this->WriteWallOps(std::string(textbuffer));
1648 }
1649
1650 /* return 0 or 1 depending if users u and u2 share one or more common channels
1651  * (used by QUIT, NICK etc which arent channel specific notices)
1652  *
1653  * The old algorithm in 1.0 for this was relatively inefficient, iterating over
1654  * the first users channels then the second users channels within the outer loop,
1655  * therefore it was a maximum of x*y iterations (upon returning 0 and checking
1656  * all possible iterations). However this new function instead checks against the
1657  * channel's userlist in the inner loop which is a std::map<userrec*,userrec*>
1658  * and saves us time as we already know what pointer value we are after.
1659  * Don't quote me on the maths as i am not a mathematician or computer scientist,
1660  * but i believe this algorithm is now x+(log y) maximum iterations instead.
1661  */
1662 bool userrec::SharesChannelWith(userrec *other)
1663 {
1664         if ((!other) || (this->registered != REG_ALL) || (other->registered != REG_ALL))
1665                 return false;
1666
1667         /* Outer loop */
1668         for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1669         {
1670                 /* Eliminate the inner loop (which used to be ~equal in size to the outer loop)
1671                  * by replacing it with a map::find which *should* be more efficient
1672                  */
1673                 if (i->first->HasUser(other))
1674                         return true;
1675         }
1676         return false;
1677 }
1678
1679 bool userrec::ChangeName(const char* gecos)
1680 {
1681         if (!strcmp(gecos, this->fullname))
1682                 return true;
1683
1684         if (IS_LOCAL(this))
1685         {
1686                 int MOD_RESULT = 0;
1687                 FOREACH_RESULT(I_OnChangeLocalUserGECOS,OnChangeLocalUserGECOS(this,gecos));
1688                 if (MOD_RESULT)
1689                         return false;
1690                 FOREACH_MOD(I_OnChangeName,OnChangeName(this,gecos));
1691         }
1692         strlcpy(this->fullname,gecos,MAXGECOS+1);
1693
1694         return true;
1695 }
1696
1697 bool userrec::ChangeDisplayedHost(const char* host)
1698 {
1699         if (!strcmp(host, this->dhost))
1700                 return true;
1701
1702         if (IS_LOCAL(this))
1703         {
1704                 int MOD_RESULT = 0;
1705                 FOREACH_RESULT(I_OnChangeLocalUserHost,OnChangeLocalUserHost(this,host));
1706                 if (MOD_RESULT)
1707                         return false;
1708                 FOREACH_MOD(I_OnChangeHost,OnChangeHost(this,host));
1709         }
1710         if (this->ServerInstance->Config->CycleHosts)
1711                 this->WriteCommonExcept("QUIT :Changing hosts");
1712
1713         /* Fix by Om: userrec::dhost is 65 long, this was truncating some long hosts */
1714         strlcpy(this->dhost,host,64);
1715
1716         this->InvalidateCache();
1717
1718         if (this->ServerInstance->Config->CycleHosts)
1719         {
1720                 for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1721                 {
1722                         i->first->WriteAllExceptSender(this, false, 0, "JOIN %s", i->first->name);
1723                         std::string n = this->ServerInstance->Modes->ModeString(this, i->first);
1724                         if (n.length() > 0)
1725                                 i->first->WriteAllExceptSender(this, true, 0, "MODE %s +%s", i->first->name, n.c_str());
1726                 }
1727         }
1728
1729         if (IS_LOCAL(this))
1730                 this->WriteServ("396 %s %s :is now your displayed host",this->nick,this->dhost);
1731
1732         return true;
1733 }
1734
1735 bool userrec::ChangeIdent(const char* newident)
1736 {
1737         if (!strcmp(newident, this->ident))
1738                 return true;
1739
1740         if (this->ServerInstance->Config->CycleHosts)
1741                 this->WriteCommonExcept("%s","QUIT :Changing ident");
1742
1743         strlcpy(this->ident, newident, IDENTMAX+2);
1744
1745         this->InvalidateCache();
1746
1747         if (this->ServerInstance->Config->CycleHosts)
1748         {
1749                 for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1750                 {
1751                         i->first->WriteAllExceptSender(this, false, 0, "JOIN %s", i->first->name);
1752                         std::string n = this->ServerInstance->Modes->ModeString(this, i->first);
1753                         if (n.length() > 0)
1754                                 i->first->WriteAllExceptSender(this, true, 0, "MODE %s +%s", i->first->name, n.c_str());
1755                 }
1756         }
1757
1758         return true;
1759 }
1760
1761 void userrec::SendAll(const char* command, char* text, ...)
1762 {
1763         char textbuffer[MAXBUF];
1764         char formatbuffer[MAXBUF];
1765         va_list argsPtr;
1766
1767         va_start(argsPtr, text);
1768         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1769         va_end(argsPtr);
1770
1771         snprintf(formatbuffer,MAXBUF,":%s %s $* :%s", this->GetFullHost(), command, textbuffer);
1772         std::string fmt = formatbuffer;
1773
1774         for (std::vector<userrec*>::const_iterator i = ServerInstance->local_users.begin(); i != ServerInstance->local_users.end(); i++)
1775         {
1776                 (*i)->Write(fmt);
1777         }
1778 }
1779
1780
1781 std::string userrec::ChannelList(userrec* source)
1782 {
1783         try
1784         {
1785                 std::string list;
1786                 for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1787                 {
1788                         /* If the target is the same as the sender, let them see all their channels.
1789                          * If the channel is NOT private/secret OR the user shares a common channel
1790                          * If the user is an oper, and the <options:operspywhois> option is set.
1791                          */
1792                         if ((source == this) || (IS_OPER(source) && ServerInstance->Config->OperSpyWhois) || (((!i->first->IsModeSet('p')) && (!i->first->IsModeSet('s'))) || (i->first->HasUser(source))))
1793                         {
1794                                 list.append(i->first->GetPrefixChar(this)).append(i->first->name).append(" ");
1795                         }
1796                 }
1797                 return list;
1798         }
1799         catch (...)
1800         {
1801                 ServerInstance->Log(DEBUG,"Exception in userrec::ChannelList()");
1802                 return "";
1803         }
1804 }
1805
1806 void userrec::SplitChanList(userrec* dest, const std::string &cl)
1807 {
1808         std::string line;
1809         std::ostringstream prefix;
1810         std::string::size_type start, pos, length;
1811
1812         try
1813         {
1814                 prefix << this->nick << " " << dest->nick << " :";
1815                 line = prefix.str();
1816                 int namelen = strlen(ServerInstance->Config->ServerName) + 6;
1817
1818                 for (start = 0; (pos = cl.find(' ', start)) != std::string::npos; start = pos+1)
1819                 {
1820                         length = (pos == std::string::npos) ? cl.length() : pos;
1821
1822                         if (line.length() + namelen + length - start > 510)
1823                         {
1824                                 ServerInstance->SendWhoisLine(this, dest, 319, "%s", line.c_str());
1825                                 line = prefix.str();
1826                         }
1827
1828                         if(pos == std::string::npos)
1829                         {
1830                                 line.append(cl.substr(start, length - start));
1831                                 break;
1832                         }
1833                         else
1834                         {
1835                                 line.append(cl.substr(start, length - start + 1));
1836                         }
1837                 }
1838
1839                 if (line.length())
1840                 {
1841                         ServerInstance->SendWhoisLine(this, dest, 319, "%s", line.c_str());
1842                 }
1843         }
1844
1845         catch (...)
1846         {
1847                 ServerInstance->Log(DEBUG,"Exception in userrec::SplitChanList()");
1848         }
1849 }
1850
1851
1852 /* looks up a users password for their connection class (<ALLOW>/<DENY> tags)
1853  * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
1854  * then their ip will be taken as 'priority' anyway, so for example,
1855  * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
1856  */
1857 ConnectClass* userrec::GetClass()
1858 {
1859         for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
1860         {
1861                 if (((match(this->GetIPString(),i->GetHost().c_str(),true)) || (match(this->host,i->GetHost().c_str()))))
1862                 {
1863                         if (i->GetPort())
1864                         {
1865                                 if (this->GetPort() == i->GetPort())
1866                                         return &(*i);
1867                                 else
1868                                         continue;
1869                         }
1870                         else
1871                                 return &(*i);
1872                 }
1873         }
1874         return NULL;
1875 }
1876
1877 void userrec::PurgeEmptyChannels()
1878 {
1879         std::vector<chanrec*> to_delete;
1880
1881         // firstly decrement the count on each channel
1882         for (UCListIter f = this->chans.begin(); f != this->chans.end(); f++)
1883         {
1884                 f->first->RemoveAllPrefixes(this);
1885                 if (f->first->DelUser(this) == 0)
1886                 {
1887                         /* No users left in here, mark it for deletion */
1888                         try
1889                         {
1890                                 to_delete.push_back(f->first);
1891                         }
1892                         catch (...)
1893                         {
1894                                 ServerInstance->Log(DEBUG,"Exception in userrec::PurgeEmptyChannels to_delete.push_back()");
1895                         }
1896                 }
1897         }
1898
1899         for (std::vector<chanrec*>::iterator n = to_delete.begin(); n != to_delete.end(); n++)
1900         {
1901                 chanrec* thischan = *n;
1902                 chan_hash::iterator i2 = ServerInstance->chanlist->find(thischan->name);
1903                 if (i2 != ServerInstance->chanlist->end())
1904                 {
1905                         FOREACH_MOD(I_OnChannelDelete,OnChannelDelete(i2->second));
1906                         DELETE(i2->second);
1907                         ServerInstance->chanlist->erase(i2);
1908                         this->chans.erase(*n);
1909                 }
1910         }
1911
1912         this->UnOper();
1913 }
1914
1915 void userrec::ShowMOTD()
1916 {
1917         if (!ServerInstance->Config->MOTD.size())
1918         {
1919                 this->WriteServ("422 %s :Message of the day file is missing.",this->nick);
1920                 return;
1921         }
1922         this->WriteServ("375 %s :%s message of the day", this->nick, ServerInstance->Config->ServerName);
1923
1924         for (file_cache::iterator i = ServerInstance->Config->MOTD.begin(); i != ServerInstance->Config->MOTD.end(); i++)
1925                 this->WriteServ("372 %s :- %s",this->nick,i->c_str());
1926
1927         this->WriteServ("376 %s :End of message of the day.", this->nick);
1928 }
1929
1930 void userrec::ShowRULES()
1931 {
1932         if (!ServerInstance->Config->RULES.size())
1933         {
1934                 this->WriteServ("434 %s :RULES File is missing",this->nick);
1935                 return;
1936         }
1937
1938         this->WriteServ("308 %s :- %s Server Rules -",this->nick,ServerInstance->Config->ServerName);
1939
1940         for (file_cache::iterator i = ServerInstance->Config->RULES.begin(); i != ServerInstance->Config->RULES.end(); i++)
1941                 this->WriteServ("232 %s :- %s",this->nick,i->c_str());
1942
1943         this->WriteServ("309 %s :End of RULES command.",this->nick);
1944 }
1945
1946 void userrec::HandleEvent(EventType et, int errornum)
1947 {
1948         /* WARNING: May delete this user! */
1949         int thisfd = this->GetFd();
1950
1951         try
1952         {
1953                 switch (et)
1954                 {
1955                         case EVENT_READ:
1956                                 ServerInstance->ProcessUser(this);
1957                         break;
1958                         case EVENT_WRITE:
1959                                 this->FlushWriteBuf();
1960                         break;
1961                         case EVENT_ERROR:
1962                                 /** This should be safe, but dont DARE do anything after it -- Brain */
1963                                 this->SetWriteError(errornum ? strerror(errornum) : "EOF from client");
1964                         break;
1965                 }
1966         }
1967         catch (...)
1968         {
1969                 ServerInstance->Log(DEBUG,"Exception in userrec::HandleEvent intercepted");
1970         }
1971
1972         /* If the user has raised an error whilst being processed, quit them now we're safe to */
1973         if ((ServerInstance->SE->GetRef(thisfd) == this))
1974         {
1975                 if (!WriteError.empty())
1976                 {
1977                         userrec::QuitUser(ServerInstance, this, GetWriteError());
1978                 }
1979         }
1980 }
1981
1982 void userrec::SetOperQuit(const std::string &oquit)
1983 {
1984         if (operquit)
1985                 return;
1986
1987         operquit = strdup(oquit.c_str());
1988 }
1989
1990 const char* userrec::GetOperQuit()
1991 {
1992         return operquit ? operquit : "";
1993 }
1994
1995 VisData::VisData()
1996 {
1997 }
1998
1999 VisData::~VisData()
2000 {
2001 }
2002
2003 bool VisData::VisibleTo(userrec* user)
2004 {
2005         return true;
2006 }
2007