src/users.cpp

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