src/inspsocket.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2008 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 /* $Core: libIRCDinspsocket */
  15
  16 #include "socket.h"
  17 #include "inspstring.h"
  18 #include "socketengine.h"
  19 #include "inspircd.h"
  20
  21 using irc::sockets::OpenTCPSocket;
  22
  23 bool BufferedSocket::Readable()
  24 {
  25         return ((this->state != I_CONNECTING) && (this->WaitingForWriteEvent == false));
  26 }
  27
  28 BufferedSocket::BufferedSocket(InspIRCd* SI)
  29 {
  30         this->Timeout = NULL;
  31         this->state = I_DISCONNECTED;
  32         this->fd = -1;
  33         this->WaitingForWriteEvent = false;
  34         this->Instance = SI;
  35         this->IsIOHooked = false;
  36 }
  37
  38 BufferedSocket::BufferedSocket(InspIRCd* SI, int newfd, const char* ip)
  39 {
  40         this->Timeout = NULL;
  41         this->fd = newfd;
  42         this->state = I_CONNECTED;
  43         strlcpy(this->IP,ip,MAXBUF);
  44         this->WaitingForWriteEvent = false;
  45         this->Instance = SI;
  46         this->IsIOHooked = false;
  47         if (this->fd > -1)
  48                 this->Instance->SE->AddFd(this);
  49 }
  50
  51 BufferedSocket::BufferedSocket(InspIRCd* SI, const std::string &ipaddr, int aport, bool listening, unsigned long maxtime, const std::string &connectbindip)
  52 {
  53         this->cbindip = connectbindip;
  54         this->fd = -1;
  55         this->Instance = SI;
  56         strlcpy(host,ipaddr.c_str(),MAXBUF);
  57         this->WaitingForWriteEvent = false;
  58         this->IsIOHooked = false;
  59         this->Timeout = NULL;
  60         if (listening)
  61         {
  62                 if ((this->fd = OpenTCPSocket(host)) == ERROR)
  63                 {
  64                         this->fd = -1;
  65                         this->state = I_ERROR;
  66                         this->OnError(I_ERR_SOCKET);
  67                         return;
  68                 }
  69                 else
  70                 {
  71                         if (!SI->BindSocket(this->fd,aport,(char*)ipaddr.c_str()))
  72                         {
  73                                 this->Close();
  74                                 this->fd = -1;
  75                                 this->state = I_ERROR;
  76                                 this->OnError(I_ERR_BIND);
  77                                 this->ClosePending = true;
  78                                 return;
  79                         }
  80                         else
  81                         {
  82                                 this->state = I_LISTENING;
  83                                 this->port = aport;
  84                                 if (this->fd > -1)
  85                                 {
  86                                         if (!this->Instance->SE->AddFd(this))
  87                                         {
  88                                                 this->Close();
  89                                                 this->state = I_ERROR;
  90                                                 this->OnError(I_ERR_NOMOREFDS);
  91                                         }
  92                                 }
  93                                 return;
  94                         }
  95                 }
  96         }
  97         else
  98         {
  99                 strlcpy(this->host,ipaddr.c_str(),MAXBUF);
 100                 this->port = aport;
 101
 102                 bool ipvalid = true;
 103 #ifdef IPV6
 104                 if (strchr(host,':'))
 105                 {
 106                         in6_addr n;
 107                         if (inet_pton(AF_INET6, host, &n) < 1)
 108                                 ipvalid = false;
 109                 }
 110                 else
 111 #endif
 112                 {
 113                         in_addr n;
 114                         if (inet_aton(host,&n) < 1)
 115                                 ipvalid = false;
 116                 }
 117                 if (!ipvalid)
 118                 {
 119                         this->Instance->Logs->Log("SOCKET", DEBUG,"BUG: Hostname passed to BufferedSocket, rather than an IP address!");
 120                         this->OnError(I_ERR_CONNECT);
 121                         this->Close();
 122                         this->fd = -1;
 123                         this->state = I_ERROR;
 124                         return;
 125                 }
 126                 else
 127                 {
 128                         strlcpy(this->IP,host,MAXBUF);
 129                         timeout_val = maxtime;
 130                         if (!this->DoConnect())
 131                         {
 132                                 this->OnError(I_ERR_CONNECT);
 133                                 this->Close();
 134                                 this->fd = -1;
 135                                 this->state = I_ERROR;
 136                                 return;
 137                         }
 138                 }
 139         }
 140 }
 141
 142 void BufferedSocket::WantWrite()
 143 {
 144         this->Instance->SE->WantWrite(this);
 145         this->WaitingForWriteEvent = true;
 146 }
 147
 148 void BufferedSocket::SetQueues(int nfd)
 149 {
 150         // attempt to increase socket sendq and recvq as high as its possible
 151         int sendbuf = 32768;
 152         int recvbuf = 32768;
 153         if(setsockopt(nfd,SOL_SOCKET,SO_SNDBUF,(const char *)&sendbuf,sizeof(sendbuf)) || setsockopt(nfd,SOL_SOCKET,SO_RCVBUF,(const char *)&recvbuf,sizeof(sendbuf)))
 154         {
 155                 //this->Instance->Log(DEFAULT, "Could not increase SO_SNDBUF/SO_RCVBUF for socket %u", GetFd());
 156                 ; // do nothing. I'm a little sick of people trying to interpret this message as a result of why their incorrect setups don't work.
 157         }
 158 }
 159
 160 /* Most irc servers require you to specify the ip you want to bind to.
 161  * If you dont specify an IP, they rather dumbly bind to the first IP
 162  * of the box (e.g. INADDR_ANY). In InspIRCd, we scan thought the IP
 163  * addresses we've bound server ports to, and we try and bind our outbound
 164  * connections to the first usable non-loopback and non-any IP we find.
 165  * This is easier to configure when you have a lot of links and a lot
 166  * of servers to configure.
 167  */
 168 bool BufferedSocket::BindAddr(const std::string &ip)
 169 {
 170         ConfigReader Conf(this->Instance);
 171         socklen_t size = sizeof(sockaddr_in);
 172 #ifdef IPV6
 173         bool v6 = false;
 174         /* Are we looking for a binding to fit an ipv6 host? */
 175         if ((ip.empty()) || (ip.find(':') != std::string::npos))
 176                 v6 = true;
 177 #endif
 178         int j = 0;
 179         while (j < Conf.Enumerate("bind") || (!ip.empty()))
 180         {
 181                 std::string sIP = ip.empty() ? Conf.ReadValue("bind","address",j) : ip;
 182                 if (!ip.empty() || Conf.ReadValue("bind","type",j) == "servers")
 183                 {
 184                         if (!ip.empty() || ((sIP != "*") && (sIP != "127.0.0.1") && (!sIP.empty()) && (sIP != "::1")))
 185                         {
 186                                 /* The [2] is required because we may write a sockaddr_in6 here, and sockaddr_in6 is larger than sockaddr, where sockaddr_in4 is not. */
 187                                 sockaddr* s = new sockaddr[2];
 188 #ifdef IPV6
 189                                 if (v6)
 190                                 {
 191                                         in6_addr n;
 192                                         if (inet_pton(AF_INET6, sIP.c_str(), &n) > 0)
 193                                         {
 194                                                 memcpy(&((sockaddr_in6*)s)->sin6_addr, &n, sizeof(sockaddr_in6));
 195                                                 ((sockaddr_in6*)s)->sin6_port = 0;
 196                                                 ((sockaddr_in6*)s)->sin6_family = AF_INET6;
 197                                                 size = sizeof(sockaddr_in6);
 198                                         }
 199                                         else
 200                                         {
 201                                                 delete[] s;
 202                                                 j++;
 203                                                 continue;
 204                                         }
 205                                 }
 206                                 else
 207 #endif
 208                                 {
 209                                         in_addr n;
 210                                         if (inet_aton(sIP.c_str(), &n) > 0)
 211                                         {
 212                                                 ((sockaddr_in*)s)->sin_addr = n;
 213                                                 ((sockaddr_in*)s)->sin_port = 0;
 214                                                 ((sockaddr_in*)s)->sin_family = AF_INET;
 215                                         }
 216                                         else
 217                                         {
 218                                                 delete[] s;
 219                                                 j++;
 220                                                 continue;
 221                                         }
 222                                 }
 223
 224                                 if (Instance->SE->Bind(this->fd, s, size) < 0)
 225                                 {
 226                                         this->state = I_ERROR;
 227                                         this->OnError(I_ERR_BIND);
 228                                         this->fd = -1;
 229                                         delete[] s;
 230                                         return false;
 231                                 }
 232
 233                                 delete[] s;
 234                                 return true;
 235                         }
 236                 }
 237                 j++;
 238         }
 239         Instance->Logs->Log("SOCKET", DEBUG,"nothing in the config to bind()!");
 240         return true;
 241 }
 242
 243 bool BufferedSocket::DoConnect()
 244 {
 245         /* The [2] is required because we may write a sockaddr_in6 here, and sockaddr_in6 is larger than sockaddr, where sockaddr_in4 is not. */
 246         sockaddr* addr = new sockaddr[2];
 247         socklen_t size = sizeof(sockaddr_in);
 248 #ifdef IPV6
 249         bool v6 = false;
 250         if ((!*this->host) || strchr(this->host, ':'))
 251                 v6 = true;
 252
 253         if (v6)
 254         {
 255                 this->fd = socket(AF_INET6, SOCK_STREAM, 0);
 256                 if ((this->fd > -1) && ((strstr(this->IP,"::ffff:") != (char*)&this->IP) && (strstr(this->IP,"::FFFF:") != (char*)&this->IP)))
 257                 {
 258                         if (!this->BindAddr(this->cbindip))
 259                         {
 260                                 delete[] addr;
 261                                 return false;
 262                         }
 263                 }
 264         }
 265         else
 266 #endif
 267         {
 268                 this->fd = socket(AF_INET, SOCK_STREAM, 0);
 269                 if (this->fd > -1)
 270                 {
 271                         if (!this->BindAddr(this->cbindip))
 272                         {
 273                                 delete[] addr;
 274                                 return false;
 275                         }
 276                 }
 277         }
 278
 279         if (this->fd == -1)
 280         {
 281                 this->state = I_ERROR;
 282                 this->OnError(I_ERR_SOCKET);
 283                 delete[] addr;
 284                 return false;
 285         }
 286
 287 #ifdef IPV6
 288         if (v6)
 289         {
 290                 in6_addr addy;
 291                 if (inet_pton(AF_INET6, this->host, &addy) > 0)
 292                 {
 293                         ((sockaddr_in6*)addr)->sin6_family = AF_INET6;
 294                         memcpy(&((sockaddr_in6*)addr)->sin6_addr, &addy, sizeof(addy));
 295                         ((sockaddr_in6*)addr)->sin6_port = htons(this->port);
 296                         size = sizeof(sockaddr_in6);
 297                 }
 298         }
 299         else
 300 #endif
 301         {
 302                 in_addr addy;
 303                 if (inet_aton(this->host, &addy) > 0)
 304                 {
 305                         ((sockaddr_in*)addr)->sin_family = AF_INET;
 306                         ((sockaddr_in*)addr)->sin_addr = addy;
 307                         ((sockaddr_in*)addr)->sin_port = htons(this->port);
 308                 }
 309         }
 310
 311         Instance->SE->NonBlocking(this->fd);
 312
 313 #ifdef WIN32
 314         /* UGH for the LOVE OF ZOMBIE JESUS SOMEONE FIX THIS!!!!!!!!!!! */
 315         Instance->SE->Blocking(this->fd);
 316 #endif
 317
 318         if (Instance->SE->Connect(this, (sockaddr*)addr, size) == -1)
 319         {
 320                 if (errno != EINPROGRESS)
 321                 {
 322                         this->OnError(I_ERR_CONNECT);
 323                         this->Close();
 324                         this->state = I_ERROR;
 325                         return false;
 326                 }
 327
 328                 this->Timeout = new SocketTimeout(this->GetFd(), this->Instance, this, timeout_val, this->Instance->Time());
 329                 this->Instance->Timers->AddTimer(this->Timeout);
 330         }
 331 #ifdef WIN32
 332         /* CRAQ SMOKING STUFF TO BE FIXED */
 333         Instance->SE->NonBlocking(this->fd);
 334 #endif
 335         this->state = I_CONNECTING;
 336         if (this->fd > -1)
 337         {
 338                 if (!this->Instance->SE->AddFd(this))
 339                 {
 340                         this->OnError(I_ERR_NOMOREFDS);
 341                         this->Close();
 342                         this->state = I_ERROR;
 343                         return false;
 344                 }
 345                 this->SetQueues(this->fd);
 346         }
 347
 348         Instance->Logs->Log("SOCKET", DEBUG,"BufferedSocket::DoConnect success");
 349         return true;
 350 }
 351
 352
 353 void BufferedSocket::Close()
 354 {
 355         /* Save this, so we dont lose it,
 356          * otherise on failure, error messages
 357          * might be inaccurate.
 358          */
 359         int save = errno;
 360         if (this->fd > -1)
 361         {
 362                 if (this->IsIOHooked && Instance->Config->GetIOHook(this))
 363                 {
 364                         try
 365                         {
 366                                 Instance->Config->GetIOHook(this)->OnRawSocketClose(this->fd);
 367                         }
 368                         catch (CoreException& modexcept)
 369                         {
 370                                 Instance->Logs->Log("SOCKET", DEFAULT,"%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
 371                         }
 372                 }
 373                 Instance->SE->Shutdown(this, 2);
 374                 if (Instance->SE->Close(this) != -1)
 375                         this->OnClose();
 376
 377                 if (Instance->SocketCull.find(this) == Instance->SocketCull.end())
 378                         Instance->SocketCull[this] = this;
 379         }
 380         errno = save;
 381 }
 382
 383 std::string BufferedSocket::GetIP()
 384 {
 385         return this->IP;
 386 }
 387
 388 const char* BufferedSocket::Read()
 389 {
 390         if (!Instance->SE->BoundsCheckFd(this))
 391                 return NULL;
 392
 393         int n = 0;
 394
 395         if (this->IsIOHooked)
 396         {
 397                 int result2 = 0;
 398                 int MOD_RESULT = 0;
 399                 try
 400                 {
 401                         MOD_RESULT = Instance->Config->GetIOHook(this)->OnRawSocketRead(this->fd,this->ibuf,sizeof(this->ibuf),result2);
 402                 }
 403                 catch (CoreException& modexcept)
 404                 {
 405                         Instance->Logs->Log("SOCKET", DEFAULT,"%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
 406                 }
 407                 if (MOD_RESULT < 0)
 408                 {
 409                         n = -1;
 410                         errno = EAGAIN;
 411                 }
 412                 else
 413                 {
 414                         n = result2;
 415                 }
 416         }
 417         else
 418         {
 419                 n = recv(this->fd,this->ibuf,sizeof(this->ibuf),0);
 420         }
 421
 422         if ((n > 0) && (n <= (int)sizeof(this->ibuf)))
 423         {
 424                 ibuf[n] = 0;
 425                 return ibuf;
 426         }
 427         else
 428         {
 429                 int err = errno;
 430                 if (err == EAGAIN)
 431                         return "";
 432                 else
 433                         return NULL;
 434         }
 435 }
 436
 437 /*
 438  * This function formerly tried to flush write buffer each call.
 439  * While admirable in attempting to get the data out to wherever
 440  * it is going, on a full socket, it's just going to syscall write() and
 441  * EAGAIN constantly, instead of waiting in the SE to know if it can write
 442  * which will chew a bit of CPU.
 443  *
 444  * So, now this function returns void (take note) and just adds to the sendq.
 445  *
 446  * It'll get written at a determinate point when the socketengine tells us it can write.
 447  *              -- w00t (april 1, 2008)
 448  */
 449 void BufferedSocket::Write(const std::string &data)
 450 {
 451         /* Append the data to the back of the queue ready for writing */
 452         outbuffer.push_back(data);
 453
 454         /* Mark ourselves as wanting write */
 455         this->Instance->SE->WantWrite(this);
 456 }
 457
 458 bool BufferedSocket::FlushWriteBuffer()
 459 {
 460         errno = 0;
 461         if ((this->fd > -1) && (this->state == I_CONNECTED))
 462         {
 463                 if (this->IsIOHooked)
 464                 {
 465                         while (outbuffer.size() && (errno != EAGAIN))
 466                         {
 467                                 try
 468                                 {
 469                                         /* XXX: The lack of buffering here is NOT a bug, modules implementing this interface have to
 470                                          * implement their own buffering mechanisms
 471                                          */
 472                                         Instance->Config->GetIOHook(this)->OnRawSocketWrite(this->fd, outbuffer[0].c_str(), outbuffer[0].length());
 473                                         outbuffer.pop_front();
 474                                 }
 475                                 catch (CoreException& modexcept)
 476                                 {
 477                                         Instance->Logs->Log("SOCKET", DEBUG,"%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
 478                                         return true;
 479                                 }
 480                         }
 481                 }
 482                 else
 483                 {
 484                         /* If we have multiple lines, try to send them all,
 485                          * not just the first one -- Brain
 486                          */
 487                         while (outbuffer.size() && (errno != EAGAIN))
 488                         {
 489                                 /* Send a line */
 490                                 int result = Instance->SE->Send(this, outbuffer[0].c_str(), outbuffer[0].length(), 0);
 491
 492                                 if (result > 0)
 493                                 {
 494                                         if ((unsigned int)result >= outbuffer[0].length())
 495                                         {
 496                                                 /* The whole block was written (usually a line)
 497                                                  * Pop the block off the front of the queue,
 498                                                  * dont set errno, because we are clear of errors
 499                                                  * and want to try and write the next block too.
 500                                                  */
 501                                                 outbuffer.pop_front();
 502                                         }
 503                                         else
 504                                         {
 505                                                 std::string temp = outbuffer[0].substr(result);
 506                                                 outbuffer[0] = temp;
 507                                                 /* We didnt get the whole line out. arses.
 508                                                  * Try again next time, i guess. Set errno,
 509                                                  * because we shouldnt be writing any more now,
 510                                                  * until the socketengine says its safe to do so.
 511                                                  */
 512                                                 errno = EAGAIN;
 513                                         }
 514                                 }
 515                                 else if (result == 0)
 516                                 {
 517                                         this->Instance->SE->DelFd(this);
 518                                         this->Close();
 519                                         return true;
 520                                 }
 521                                 else if ((result == -1) && (errno != EAGAIN))
 522                                 {
 523                                         this->OnError(I_ERR_WRITE);
 524                                         this->state = I_ERROR;
 525                                         this->Instance->SE->DelFd(this);
 526                                         this->Close();
 527                                         return true;
 528                                 }
 529                         }
 530                 }
 531         }
 532
 533         if ((errno == EAGAIN) && (fd > -1))
 534         {
 535                 this->Instance->SE->WantWrite(this);
 536         }
 537
 538         return (fd < 0);
 539 }
 540
 541 void SocketTimeout::Tick(time_t)
 542 {
 543         ServerInstance->Logs->Log("SOCKET", DEBUG,"SocketTimeout::Tick");
 544
 545         if (ServerInstance->SE->GetRef(this->sfd) != this->sock)
 546                 return;
 547
 548         if (this->sock->state == I_CONNECTING)
 549         {
 550                 // for non-listening sockets, the timeout can occur
 551                 // which causes termination of the connection after
 552                 // the given number of seconds without a successful
 553                 // connection.
 554                 this->sock->OnTimeout();
 555                 this->sock->OnError(I_ERR_TIMEOUT);
 556                 this->sock->timeout = true;
 557
 558                 /* NOTE: We must set this AFTER DelFd, as we added
 559                  * this socket whilst writeable. This means that we
 560                  * must DELETE the socket whilst writeable too!
 561                  */
 562                 this->sock->state = I_ERROR;
 563
 564                 if (ServerInstance->SocketCull.find(this->sock) == ServerInstance->SocketCull.end())
 565                         ServerInstance->SocketCull[this->sock] = this->sock;
 566         }
 567
 568         this->sock->Timeout = NULL;
 569 }
 570
 571 bool BufferedSocket::Poll()
 572 {
 573         int incoming = -1;
 574
 575 #ifndef WINDOWS
 576         if (!Instance->SE->BoundsCheckFd(this))
 577                 return false;
 578 #endif
 579
 580         if (Instance->SE->GetRef(this->fd) != this)
 581                 return false;
 582
 583         switch (this->state)
 584         {
 585                 case I_CONNECTING:
 586                         /* Our socket was in write-state, so delete it and re-add it
 587                          * in read-state.
 588                          */
 589 #ifndef WINDOWS
 590                         if (this->fd > -1)
 591                         {
 592                                 this->Instance->SE->DelFd(this);
 593                                 if (!this->Instance->SE->AddFd(this))
 594                                         return false;
 595                         }
 596 #endif
 597                         this->SetState(I_CONNECTED);
 598
 599                         if (Instance->Config->GetIOHook(this))
 600                         {
 601                                 Instance->Logs->Log("SOCKET",DEBUG,"Hook for raw connect");
 602                                 try
 603                                 {
 604                                         Instance->Config->GetIOHook(this)->OnRawSocketConnect(this->fd);
 605                                 }
 606                                 catch (CoreException& modexcept)
 607                                 {
 608                                         Instance->Logs->Log("SOCKET",DEBUG,"%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
 609                                 }
 610                         }
 611                         return this->OnConnected();
 612                 break;
 613                 case I_LISTENING:
 614                 {
 615                         /* The [2] is required because we may write a sockaddr_in6 here, and sockaddr_in6 is larger than sockaddr, where sockaddr_in4 is not. */
 616                         sockaddr* client = new sockaddr[2];
 617                         length = sizeof (sockaddr_in);
 618                         std::string recvip;
 619 #ifdef IPV6
 620                         if ((!*this->host) || strchr(this->host, ':'))
 621                                 length = sizeof(sockaddr_in6);
 622 #endif
 623                         incoming = Instance->SE->Accept(this, client, &length);
 624 #ifdef IPV6
 625                         if ((!*this->host) || strchr(this->host, ':'))
 626                         {
 627                                 char buf[1024];
 628                                 recvip = inet_ntop(AF_INET6, &((sockaddr_in6*)client)->sin6_addr, buf, sizeof(buf));
 629                         }
 630                         else
 631 #endif
 632                         {
 633                                 // FIX: we were doing this for IPv6 connections as well, which was fucking recvip..
 634                                 // Add brackets to make this a bit clearer. -- w00t (Jan 15, 2008)
 635                                 recvip = inet_ntoa(((sockaddr_in*)client)->sin_addr);
 636                         }
 637
 638                         Instance->SE->NonBlocking(incoming);
 639
 640                         this->OnIncomingConnection(incoming, (char*)recvip.c_str());
 641
 642                         if (this->IsIOHooked)
 643                         {
 644                                 try
 645                                 {
 646                                         Instance->Config->GetIOHook(this)->OnRawSocketAccept(incoming, recvip.c_str(), this->port);
 647                                 }
 648                                 catch (CoreException& modexcept)
 649                                 {
 650                                         Instance->Logs->Log("SOCKET",DEBUG,"%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
 651                                 }
 652                         }
 653
 654                         this->SetQueues(incoming);
 655
 656                         delete[] client;
 657                         return true;
 658                 }
 659                 break;
 660                 case I_CONNECTED:
 661                         /* Process the read event */
 662                         return this->OnDataReady();
 663                 break;
 664                 default:
 665                 break;
 666         }
 667         return true;
 668 }
 669
 670 void BufferedSocket::SetState(BufferedSocketState s)
 671 {
 672         this->state = s;
 673 }
 674
 675 BufferedSocketState BufferedSocket::GetState()
 676 {
 677         return this->state;
 678 }
 679
 680 int BufferedSocket::GetFd()
 681 {
 682         return this->fd;
 683 }
 684
 685 bool BufferedSocket::OnConnected() { return true; }
 686 void BufferedSocket::OnError(BufferedSocketError) { return; }
 687 int BufferedSocket::OnDisconnect() { return 0; }
 688 int BufferedSocket::OnIncomingConnection(int, char*) { return 0; }
 689 bool BufferedSocket::OnDataReady() { return true; }
 690 bool BufferedSocket::OnWriteReady() { return true; }
 691 void BufferedSocket::OnTimeout() { return; }
 692 void BufferedSocket::OnClose() { return; }
 693
 694 BufferedSocket::~BufferedSocket()
 695 {
 696         this->Close();
 697         if (Timeout)
 698         {
 699                 Instance->Timers->DelTimer(Timeout);
 700                 Timeout = NULL;
 701         }
 702 }
 703
 704 void BufferedSocket::HandleEvent(EventType et, int errornum)
 705 {
 706         switch (et)
 707         {
 708                 case EVENT_ERROR:
 709                         switch (errornum)
 710                         {
 711                                 case ETIMEDOUT:
 712                                         this->OnError(I_ERR_TIMEOUT);
 713                                 break;
 714                                 case ECONNREFUSED:
 715                                 case 0:
 716                                         this->OnError(this->state == I_CONNECTING ? I_ERR_CONNECT : I_ERR_WRITE);
 717                                 break;
 718                                 case EADDRINUSE:
 719                                         this->OnError(I_ERR_BIND);
 720                                 break;
 721                                 case EPIPE:
 722                                 case EIO:
 723                                         this->OnError(I_ERR_WRITE);
 724                                 break;
 725                         }
 726                         if (this->Instance->SocketCull.find(this) == this->Instance->SocketCull.end())
 727                                 this->Instance->SocketCull[this] = this;
 728                         return;
 729                 break;
 730                 case EVENT_READ:
 731                         if (!this->Poll())
 732                         {
 733                                 if (this->Instance->SocketCull.find(this) == this->Instance->SocketCull.end())
 734                                         this->Instance->SocketCull[this] = this;
 735                                 return;
 736                         }
 737                 break;
 738                 case EVENT_WRITE:
 739                         if (this->WaitingForWriteEvent)
 740                         {
 741                                 this->WaitingForWriteEvent = false;
 742                                 if (!this->OnWriteReady())
 743                                 {
 744                                         if (this->Instance->SocketCull.find(this) == this->Instance->SocketCull.end())
 745                                                 this->Instance->SocketCull[this] = this;
 746                                         return;
 747                                 }
 748                         }
 749                         if (this->state == I_CONNECTING)
 750                         {
 751                                 /* This might look wrong as if we should be actually calling
 752                                  * with EVENT_WRITE, but trust me it is correct. There are some
 753                                  * writeability-state things in the read code, because of how
 754                                  * BufferedSocket used to work regarding write buffering in previous
 755                                  * versions of InspIRCd. - Brain
 756                                  */
 757                                 this->HandleEvent(EVENT_READ);
 758                                 return;
 759                         }
 760                         else
 761                         {
 762                                 if (this->FlushWriteBuffer())
 763                                 {
 764                                         if (this->Instance->SocketCull.find(this) == this->Instance->SocketCull.end())
 765                                                 this->Instance->SocketCull[this] = this;
 766                                         return;
 767                                 }
 768                         }
 769                 break;
 770         }
 771 }
 772