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