src/listensocket.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
   6  *
   7  * This file is part of InspIRCd.  InspIRCd is free software: you can
   8  * redistribute it and/or modify it under the terms of the GNU General Public
   9  * License as published by the Free Software Foundation, version 2.
  10  *
  11  * This program is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  13  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  14  * details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20
  21 #include "inspircd.h"
  22 #include "socket.h"
  23 #include "socketengine.h"
  24
  25 ListenSocket::ListenSocket(ConfigTag* tag, const irc::sockets::sockaddrs& bind_to)
  26         : bind_tag(tag)
  27 {
  28         irc::sockets::satoap(bind_to, bind_addr, bind_port);
  29         bind_desc = irc::sockets::satouser(bind_to);
  30
  31         fd = socket(bind_to.sa.sa_family, SOCK_STREAM, 0);
  32
  33         if (this->fd == -1)
  34                 return;
  35
  36 #ifdef IPV6_V6ONLY
  37         /* This OS supports IPv6 sockets that can also listen for IPv4
  38          * connections. If our address is "*" or empty, enable both v4 and v6 to
  39          * allow for simpler configuration on dual-stack hosts. Otherwise, if it
  40          * is "::" or an IPv6 address, disable support so that an IPv4 bind will
  41          * work on the port (by us or another application).
  42          */
  43         if (bind_to.sa.sa_family == AF_INET6)
  44         {
  45                 std::string addr = tag->getString("address");
  46                 /* This must be >= sizeof(DWORD) on Windows */
  47                 const int enable = (addr.empty() || addr == "*") ? 0 : 1;
  48                 /* This must be before bind() */
  49                 setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, reinterpret_cast<const char *>(&enable), sizeof(enable));
  50                 // errors ignored intentionally
  51         }
  52 #endif
  53
  54         ServerInstance->SE->SetReuse(fd);
  55         int rv = ServerInstance->SE->Bind(this->fd, bind_to);
  56         if (rv >= 0)
  57                 rv = ServerInstance->SE->Listen(this->fd, ServerInstance->Config->MaxConn);
  58
  59         if (rv < 0)
  60         {
  61                 int errstore = errno;
  62                 ServerInstance->SE->Shutdown(this, 2);
  63                 ServerInstance->SE->Close(this);
  64                 this->fd = -1;
  65                 errno = errstore;
  66         }
  67         else
  68         {
  69                 ServerInstance->SE->NonBlocking(this->fd);
  70                 ServerInstance->SE->AddFd(this, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
  71         }
  72 }
  73
  74 ListenSocket::~ListenSocket()
  75 {
  76         if (this->GetFd() > -1)
  77         {
  78                 ServerInstance->SE->DelFd(this);
  79                 ServerInstance->Logs->Log("SOCKET", DEBUG,"Shut down listener on fd %d", this->fd);
  80                 ServerInstance->SE->Shutdown(this, 2);
  81                 if (ServerInstance->SE->Close(this) != 0)
  82                         ServerInstance->Logs->Log("SOCKET", DEBUG,"Failed to cancel listener: %s", strerror(errno));
  83                 this->fd = -1;
  84         }
  85 }
  86
  87 /* Just seperated into another func for tidiness really.. */
  88 void ListenSocket::AcceptInternal()
  89 {
  90         irc::sockets::sockaddrs client;
  91         irc::sockets::sockaddrs server;
  92
  93         socklen_t length = sizeof(client);
  94         int incomingSockfd = ServerInstance->SE->Accept(this, &client.sa, &length);
  95
  96         ServerInstance->Logs->Log("SOCKET",DEBUG,"HandleEvent for Listensocket %s nfd=%d", bind_desc.c_str(), incomingSockfd);
  97         if (incomingSockfd < 0)
  98         {
  99                 ServerInstance->stats->statsRefused++;
 100                 return;
 101         }
 102
 103         socklen_t sz = sizeof(server);
 104         if (getsockname(incomingSockfd, &server.sa, &sz))
 105         {
 106                 ServerInstance->Logs->Log("SOCKET", DEBUG, "Can't get peername: %s", strerror(errno));
 107                 irc::sockets::aptosa(bind_addr, bind_port, server);
 108         }
 109
 110         /*
 111          * XXX -
 112          * this is done as a safety check to keep the file descriptors within range of fd_ref_table.
 113          * its a pretty big but for the moment valid assumption:
 114          * file descriptors are handed out starting at 0, and are recycled as theyre freed.
 115          * therefore if there is ever an fd over 65535, 65536 clients must be connected to the
 116          * irc server at once (or the irc server otherwise initiating this many connections, files etc)
 117          * which for the time being is a physical impossibility (even the largest networks dont have more
 118          * than about 10,000 users on ONE server!)
 119          */
 120         if (incomingSockfd >= ServerInstance->SE->GetMaxFds())
 121         {
 122                 ServerInstance->Logs->Log("SOCKET", DEBUG, "Server is full");
 123                 ServerInstance->SE->Shutdown(incomingSockfd, 2);
 124                 ServerInstance->SE->Close(incomingSockfd);
 125                 ServerInstance->stats->statsRefused++;
 126                 return;
 127         }
 128
 129         if (client.sa.sa_family == AF_INET6)
 130         {
 131                 /*
 132                  * This case is the be all and end all patch to catch and nuke 4in6
 133                  * instead of special-casing shit all over the place and wreaking merry
 134                  * havoc with crap, instead, we just recreate sockaddr and strip ::ffff: prefix
 135                  * if it's a 4in6 IP.
 136                  *
 137                  * This is, of course, much improved over the older way of handling this
 138                  * (pretend it doesn't exist + hack around it -- yes, both were done!)
 139                  *
 140                  * Big, big thanks to danieldg for his work on this.
 141                  * -- w00t
 142                  */
 143                 static const unsigned char prefix4in6[12] = { 0,0,0,0, 0,0,0,0, 0,0,0xFF,0xFF };
 144                 if (!memcmp(prefix4in6, &client.in6.sin6_addr, 12))
 145                 {
 146                         // recreate as a sockaddr_in using the IPv4 IP
 147                         uint16_t sport = client.in6.sin6_port;
 148                         client.in4.sin_family = AF_INET;
 149                         client.in4.sin_port = sport;
 150                         memcpy(&client.in4.sin_addr.s_addr, client.in6.sin6_addr.s6_addr + 12, sizeof(uint32_t));
 151
 152                         sport = server.in6.sin6_port;
 153                         server.in4.sin_family = AF_INET;
 154                         server.in4.sin_port = sport;
 155                         memcpy(&server.in4.sin_addr.s_addr, server.in6.sin6_addr.s6_addr + 12, sizeof(uint32_t));
 156                 }
 157         }
 158
 159         ServerInstance->SE->NonBlocking(incomingSockfd);
 160
 161         ModResult res;
 162         FIRST_MOD_RESULT(OnAcceptConnection, res, (incomingSockfd, this, &client, &server));
 163         if (res == MOD_RES_PASSTHRU)
 164         {
 165                 std::string type = bind_tag->getString("type", "clients");
 166                 if (type == "clients")
 167                 {
 168                         ServerInstance->Users->AddUser(incomingSockfd, this, &client, &server);
 169                         res = MOD_RES_ALLOW;
 170                 }
 171         }
 172         if (res == MOD_RES_ALLOW)
 173         {
 174                 ServerInstance->stats->statsAccept++;
 175         }
 176         else
 177         {
 178                 ServerInstance->stats->statsRefused++;
 179                 ServerInstance->Logs->Log("SOCKET",DEFAULT,"Refusing connection on %s - %s",
 180                         bind_desc.c_str(), res == MOD_RES_DENY ? "Connection refused by module" : "Module for this port not found");
 181                 ServerInstance->SE->Close(incomingSockfd);
 182         }
 183 }
 184
 185 void ListenSocket::HandleEvent(EventType e, int err)
 186 {
 187         switch (e)
 188         {
 189                 case EVENT_ERROR:
 190                         ServerInstance->Logs->Log("SOCKET",DEFAULT,"ListenSocket::HandleEvent() received a socket engine error event! well shit! '%s'", strerror(err));
 191                         break;
 192                 case EVENT_WRITE:
 193                         ServerInstance->Logs->Log("SOCKET",DEBUG,"*** BUG *** ListenSocket::HandleEvent() got a WRITE event!!!");
 194                         break;
 195                 case EVENT_READ:
 196                         this->AcceptInternal();
 197                         break;
 198         }
 199 }