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