diff options
-rw-r--r-- | include/inspsocket.h | 59 | ||||
-rw-r--r-- | src/inspsocket.cpp | 156 |
2 files changed, 44 insertions, 171 deletions
diff --git a/include/inspsocket.h b/include/inspsocket.h index ae18afabd..f0f0e1484 100644 --- a/include/inspsocket.h +++ b/include/inspsocket.h @@ -27,8 +27,6 @@ enum BufferedSocketState I_CONNECTING, /** Socket fully connected */ I_CONNECTED, - /** Socket listening for connections */ - I_LISTENING, /** Socket has an error */ I_ERROR }; @@ -106,44 +104,31 @@ class CoreExport BufferedSocket : public EventHandler { public: - /** - * Bind IP + /** Bind IP */ std::string cbindip; - /** - * Is hooked by a module for low level IO - */ - bool IsIOHooked; - - /** - * Instance we were created by + /** Instance we were created by */ InspIRCd* Instance; - /** - * Timeout class or NULL + /** Timeout class or NULL */ SocketTimeout* Timeout; - /** - * Timeout length + /** Timeout length */ unsigned long timeout_val; - /** - * Socket output buffer (binary safe) + /** Socket output buffer (binary safe) */ std::deque<std::string> outbuffer; - /** - * The hostname connected to + /** The hostname connected to */ char host[MAXBUF]; - /** - * The port connected to, or the port - * this socket is listening on + /** The port connected to */ int port; @@ -221,19 +206,17 @@ class CoreExport BufferedSocket : public EventHandler BufferedSocket(InspIRCd* SI, int newfd, const char* ip); /** - * This constructor is used to create a new - * socket, either listening for connections, or an outbound connection to another host. + * This constructor is used to create a new outbound connection to another host. * Note that if you specify a hostname in the 'ipaddr' parameter, this class will not * connect. You must resolve your hostnames before passing them to BufferedSocket. To do so, * you should use the nonblocking class 'Resolver'. * @param ipaddr The IP to connect to, or bind to - * @param port The port number to connect to, or bind to - * @param listening true to listen on the given host:port pair, or false to connect to them + * @param port The port number to connect to * @param maxtime Number of seconds to wait, if connecting, before the connection times out and an OnTimeout() event is generated * @param connectbindip When creating an outbound connection, the IP to bind the connection to. If not defined, the port is not bound. * @return On exit, GetState() returns I_ERROR if an error occured, and errno can be used to read the socket error. */ - BufferedSocket(InspIRCd* SI, const std::string &ipaddr, int port, bool listening, unsigned long maxtime, const std::string &connectbindip = ""); + BufferedSocket(InspIRCd* SI, const std::string &ipaddr, int port, unsigned long maxtime, const std::string &connectbindip = ""); /** * This method is called when an outbound @@ -298,9 +281,7 @@ class CoreExport BufferedSocket : public EventHandler * Whenever close() is called, OnClose() will be * called first. Please note that this means * OnClose will be called alongside OnError(), - * OnTimeout(), and Close(), and also when - * cancelling a listening socket by calling - * the destructor indirectly. + * OnTimeout(), and Close(). */ virtual void OnClose(); @@ -326,21 +307,6 @@ class CoreExport BufferedSocket : public EventHandler virtual void Write(const std::string &data); /** - * If your socket is a listening socket, when a new - * connection comes in on the socket this method will - * be called. Given the new file descriptor in the - * parameters, and the IP, it is recommended you copy - * them to a new instance of your socket class, - * e.g.: - * - * MySocket* newsocket = new MySocket(newfd,ip); - * - * Once you have done this, you can then associate the - * new socket with the core using Server::AddSocket(). - */ - virtual int OnIncomingConnection(int newfd, char* ip); - - /** * Changes the socket's state. The core uses this * to change socket states, and you should not call * it directly. @@ -392,8 +358,7 @@ class CoreExport BufferedSocket : public EventHandler /** * This method attempts to connect to a hostname. - * This only occurs on a non-listening socket. This - * method is asyncronous. + * This method is asyncronous. */ virtual bool DoConnect(); diff --git a/src/inspsocket.cpp b/src/inspsocket.cpp index 3b28cc3bd..d4f6c9c17 100644 --- a/src/inspsocket.cpp +++ b/src/inspsocket.cpp @@ -18,8 +18,6 @@ #include "socketengine.h" #include "inspircd.h" -using irc::sockets::OpenTCPSocket; - bool BufferedSocket::Readable() { return ((this->state != I_CONNECTING) && (this->WaitingForWriteEvent == false)); @@ -46,7 +44,7 @@ BufferedSocket::BufferedSocket(InspIRCd* SI, int newfd, const char* ip) this->Instance->SE->AddFd(this); } -BufferedSocket::BufferedSocket(InspIRCd* SI, const std::string &ipaddr, int aport, bool listening, unsigned long maxtime, const std::string &connectbindip) +BufferedSocket::BufferedSocket(InspIRCd* SI, const std::string &ipaddr, int aport, unsigned long maxtime, const std::string &connectbindip) { this->cbindip = connectbindip; this->fd = -1; @@ -54,85 +52,46 @@ BufferedSocket::BufferedSocket(InspIRCd* SI, const std::string &ipaddr, int apor strlcpy(host,ipaddr.c_str(),MAXBUF); this->WaitingForWriteEvent = false; this->Timeout = NULL; - if (listening) + + strlcpy(this->host,ipaddr.c_str(),MAXBUF); + this->port = aport; + + bool ipvalid = true; +#ifdef IPV6 + if (strchr(host,':')) { - if ((this->fd = OpenTCPSocket(host)) == ERROR) - { - this->fd = -1; - this->state = I_ERROR; - this->OnError(I_ERR_SOCKET); - return; - } - else - { - if (!SI->BindSocket(this->fd,aport,(char*)ipaddr.c_str())) - { - this->Close(); - this->fd = -1; - this->state = I_ERROR; - this->OnError(I_ERR_BIND); - this->ClosePending = true; - return; - } - else - { - this->state = I_LISTENING; - this->port = aport; - if (this->fd > -1) - { - if (!this->Instance->SE->AddFd(this)) - { - this->Close(); - this->state = I_ERROR; - this->OnError(I_ERR_NOMOREFDS); - } - } - return; - } - } + in6_addr n; + if (inet_pton(AF_INET6, host, &n) < 1) + ipvalid = false; } else - { - strlcpy(this->host,ipaddr.c_str(),MAXBUF); - this->port = aport; - - bool ipvalid = true; -#ifdef IPV6 - if (strchr(host,':')) - { - in6_addr n; - if (inet_pton(AF_INET6, host, &n) < 1) - ipvalid = false; - } - else #endif + { + in_addr n; + if (inet_aton(host,&n) < 1) + ipvalid = false; + } + if (!ipvalid) + { + this->Instance->Logs->Log("SOCKET", DEBUG,"BUG: Hostname passed to BufferedSocket, rather than an IP address!"); + this->OnError(I_ERR_CONNECT); + this->Close(); + this->fd = -1; + this->state = I_ERROR; + return; + } + else + { + strlcpy(this->IP,host,MAXBUF); + timeout_val = maxtime; + if (!this->DoConnect()) { - in_addr n; - if (inet_aton(host,&n) < 1) - ipvalid = false; - } - if (!ipvalid) - { - this->Instance->Logs->Log("SOCKET", DEBUG,"BUG: Hostname passed to BufferedSocket, rather than an IP address!"); this->OnError(I_ERR_CONNECT); this->Close(); this->fd = -1; this->state = I_ERROR; return; } - else - { - strlcpy(this->IP,host,MAXBUF); - timeout_val = maxtime; - if (!this->DoConnect()) - { - this->OnError(I_ERR_CONNECT); - this->Close(); - this->fd = -1; - this->state = I_ERROR; - return; - } - } } } @@ -360,8 +319,7 @@ void BufferedSocket::Close() { try { - if (this->state != I_LISTENING) - this->GetIOHook()->OnRawSocketClose(this->fd); + this->GetIOHook()->OnRawSocketClose(this->fd); } catch (CoreException& modexcept) { @@ -552,7 +510,7 @@ void SocketTimeout::Tick(time_t) if (this->sock->state == I_CONNECTING) { - // for non-listening sockets, the timeout can occur + // for connecting sockets, the timeout can occur // which causes termination of the connection after // the given number of seconds without a successful // connection. @@ -575,8 +533,6 @@ void SocketTimeout::Tick(time_t) bool BufferedSocket::Poll() { - int incoming = -1; - #ifndef WINDOWS if (!Instance->SE->BoundsCheckFd(this)) return false; @@ -615,53 +571,6 @@ bool BufferedSocket::Poll() } return this->OnConnected(); break; - case I_LISTENING: - { - /* The [2] is required because we may write a sockaddr_in6 here, and sockaddr_in6 is larger than sockaddr, where sockaddr_in4 is not. */ - sockaddr* client = new sockaddr[2]; - length = sizeof (sockaddr_in); - std::string recvip; -#ifdef IPV6 - if ((!*this->host) || strchr(this->host, ':')) - length = sizeof(sockaddr_in6); -#endif - incoming = Instance->SE->Accept(this, client, &length); -#ifdef IPV6 - if ((!*this->host) || strchr(this->host, ':')) - { - char buf[1024]; - recvip = inet_ntop(AF_INET6, &((sockaddr_in6*)client)->sin6_addr, buf, sizeof(buf)); - } - else -#endif - { - // FIX: we were doing this for IPv6 connections as well, which was fucking recvip.. - // Add brackets to make this a bit clearer. -- w00t (Jan 15, 2008) - recvip = inet_ntoa(((sockaddr_in*)client)->sin_addr); - } - - Instance->SE->NonBlocking(incoming); - - this->OnIncomingConnection(incoming, (char*)recvip.c_str()); - - if (this->GetIOHook()) - { - try - { - this->GetIOHook()->OnRawSocketAccept(incoming, recvip.c_str(), this->port); - } - catch (CoreException& modexcept) - { - Instance->Logs->Log("SOCKET",DEBUG,"%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason()); - } - } - - this->SetQueues(incoming); - - delete[] client; - return true; - } - break; case I_CONNECTED: /* Process the read event */ return this->OnDataReady(); @@ -690,7 +599,6 @@ int BufferedSocket::GetFd() bool BufferedSocket::OnConnected() { return true; } void BufferedSocket::OnError(BufferedSocketError) { return; } int BufferedSocket::OnDisconnect() { return 0; } -int BufferedSocket::OnIncomingConnection(int, char*) { return 0; } bool BufferedSocket::OnDataReady() { return true; } bool BufferedSocket::OnWriteReady() { return true; } void BufferedSocket::OnTimeout() { return; } |