2 * InspIRCd -- Internet Relay Chat Daemon
4 * Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
5 * Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
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.
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
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/>.
24 #include <netinet/tcp.h>
27 ListenSocket::ListenSocket(ConfigTag* tag, const irc::sockets::sockaddrs& bind_to)
29 , iohookprov(NULL, std::string())
31 irc::sockets::satoap(bind_to, bind_addr, bind_port);
32 bind_desc = bind_to.str();
34 fd = socket(bind_to.sa.sa_family, SOCK_STREAM, 0);
40 /* This OS supports IPv6 sockets that can also listen for IPv4
41 * connections. If our address is "*" or empty, enable both v4 and v6 to
42 * allow for simpler configuration on dual-stack hosts. Otherwise, if it
43 * is "::" or an IPv6 address, disable support so that an IPv4 bind will
44 * work on the port (by us or another application).
46 if (bind_to.sa.sa_family == AF_INET6)
48 std::string addr = tag->getString("address");
49 /* This must be >= sizeof(DWORD) on Windows */
50 const int enable = (addr.empty() || addr == "*") ? 0 : 1;
51 /* This must be before bind() */
52 setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, reinterpret_cast<const char *>(&enable), sizeof(enable));
53 // errors ignored intentionally
57 SocketEngine::SetReuse(fd);
58 int rv = SocketEngine::Bind(this->fd, bind_to);
60 rv = SocketEngine::Listen(this->fd, ServerInstance->Config->MaxConn);
62 int timeout = tag->getInt("defer", 0);
65 #if defined TCP_DEFER_ACCEPT
66 setsockopt(fd, IPPROTO_TCP, TCP_DEFER_ACCEPT, &timeout, sizeof(timeout));
67 #elif defined SO_ACCEPTFILTER
68 struct accept_filter_arg afa;
69 memset(&afa, 0, sizeof(afa));
70 strcpy(afa.af_name, "dataready");
71 setsockopt(fd, SOL_SOCKET, SO_ACCEPTFILTER, &afa, sizeof(afa));
78 SocketEngine::Shutdown(this, 2);
79 SocketEngine::Close(this->GetFd());
85 SocketEngine::NonBlocking(this->fd);
86 SocketEngine::AddFd(this, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
88 this->ResetIOHookProvider();
92 ListenSocket::~ListenSocket()
94 if (this->GetFd() > -1)
96 ServerInstance->Logs->Log("SOCKET", LOG_DEBUG, "Shut down listener on fd %d", this->fd);
97 SocketEngine::Shutdown(this, 2);
98 if (SocketEngine::Close(this) != 0)
99 ServerInstance->Logs->Log("SOCKET", LOG_DEBUG, "Failed to cancel listener: %s", strerror(errno));
103 void ListenSocket::OnEventHandlerRead()
105 irc::sockets::sockaddrs client;
106 irc::sockets::sockaddrs server;
108 socklen_t length = sizeof(client);
109 int incomingSockfd = SocketEngine::Accept(this, &client.sa, &length);
111 ServerInstance->Logs->Log("SOCKET", LOG_DEBUG, "Accepting connection on socket %s fd %d", bind_desc.c_str(), incomingSockfd);
112 if (incomingSockfd < 0)
114 ServerInstance->stats.Refused++;
118 socklen_t sz = sizeof(server);
119 if (getsockname(incomingSockfd, &server.sa, &sz))
121 ServerInstance->Logs->Log("SOCKET", LOG_DEBUG, "Can't get peername: %s", strerror(errno));
122 irc::sockets::aptosa(bind_addr, bind_port, server);
125 if (client.sa.sa_family == AF_INET6)
128 * This case is the be all and end all patch to catch and nuke 4in6
129 * instead of special-casing shit all over the place and wreaking merry
130 * havoc with crap, instead, we just recreate sockaddr and strip ::ffff: prefix
133 * This is, of course, much improved over the older way of handling this
134 * (pretend it doesn't exist + hack around it -- yes, both were done!)
136 * Big, big thanks to danieldg for his work on this.
139 static const unsigned char prefix4in6[12] = { 0,0,0,0, 0,0,0,0, 0,0,0xFF,0xFF };
140 if (!memcmp(prefix4in6, &client.in6.sin6_addr, 12))
142 // recreate as a sockaddr_in using the IPv4 IP
143 uint16_t sport = client.in6.sin6_port;
144 client.in4.sin_family = AF_INET;
145 client.in4.sin_port = sport;
146 memcpy(&client.in4.sin_addr.s_addr, client.in6.sin6_addr.s6_addr + 12, sizeof(uint32_t));
148 sport = server.in6.sin6_port;
149 server.in4.sin_family = AF_INET;
150 server.in4.sin_port = sport;
151 memcpy(&server.in4.sin_addr.s_addr, server.in6.sin6_addr.s6_addr + 12, sizeof(uint32_t));
155 SocketEngine::NonBlocking(incomingSockfd);
158 FIRST_MOD_RESULT(OnAcceptConnection, res, (incomingSockfd, this, &client, &server));
159 if (res == MOD_RES_PASSTHRU)
161 std::string type = bind_tag->getString("type", "clients");
162 if (type == "clients")
164 ServerInstance->Users->AddUser(incomingSockfd, this, &client, &server);
168 if (res == MOD_RES_ALLOW)
170 ServerInstance->stats.Accept++;
174 ServerInstance->stats.Refused++;
175 ServerInstance->Logs->Log("SOCKET", LOG_DEFAULT, "Refusing connection on %s - %s",
176 bind_desc.c_str(), res == MOD_RES_DENY ? "Connection refused by module" : "Module for this port not found");
177 SocketEngine::Close(incomingSockfd);
181 bool ListenSocket::ResetIOHookProvider()
183 std::string provname = bind_tag->getString("ssl");
184 if (!provname.empty())
185 provname.insert(0, "ssl/");
187 // Set the new provider name, dynref handles the rest
188 iohookprov.SetProvider(provname);
190 // Return true if no provider was set, or one was set and it was also found
191 return (provname.empty() || iohookprov);