X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Finspsocket.cpp;h=eb0ee86c82f524d8a5c48311f5847d055ed00ccf;hb=82243d9beb827fca5708efe9e047ff2fec4bfe8c;hp=814517bf16ad9c4fe4c858112c90dd06ca00bb79;hpb=104cf9cbee8ae0c6994060b565fe410e8f140ed7;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/inspsocket.cpp b/src/inspsocket.cpp
index 814517bf1..eb0ee86c8 100644
--- a/src/inspsocket.cpp
+++ b/src/inspsocket.cpp
@@ -14,12 +14,6 @@
  * ---------------------------------------------------
  */
 
-#include <string>
-#include <sstream>
-#include <iostream>
-#include <fstream>
-#include <stdexcept>
-#include "inspircd_config.h"
 #include "socket.h"
 #include "configreader.h"
 #include "inspstring.h"
@@ -117,7 +111,7 @@ InspSocket::InspSocket(InspIRCd* SI, const std::string &ipaddr, int aport, bool
 		{
 			this->Instance->Log(DEBUG,"No need to resolve %s",this->host);
 			strlcpy(this->IP,host,MAXBUF);
-			timeout_end = time(NULL) + maxtime;
+			timeout_val = maxtime;
 			this->DoConnect();
 		}
 	}
@@ -254,6 +248,9 @@ bool InspSocket::DoConnect()
 			this->state = I_ERROR;
 			return false;
 		}
+
+		this->Timeout = new SocketTimeout(this->GetFd(), this->Instance, this, timeout_val, this->Instance->Time());
+		this->Instance->Timers->AddTimer(this->Timeout);
 	}
 	this->state = I_CONNECTING;
 	if (this->fd > -1)
@@ -359,6 +356,8 @@ bool InspSocket::FlushWriteBuffer()
 				this->Instance->Log(DEBUG,"Write error on socket: %s",strerror(errno));
 				this->OnError(I_ERR_WRITE);
 				this->state = I_ERROR;
+				this->Instance->SE->DelFd(this);
+				this->Close();
 				return true;
 			}
 		}
@@ -366,34 +365,31 @@ bool InspSocket::FlushWriteBuffer()
 	return (fd < 0);
 }
 
-bool InspSocket::Timeout(time_t current)
+void SocketTimeout::Tick(time_t now)
 {
-	if (this->Instance->SE->GetRef(this->fd) != this)
+	if (ServerInstance->SE->GetRef(this->sfd) != this->sock)
 	{
-		this->Instance->Log(DEBUG,"No FD or socket ref");
-		return false;
+		ServerInstance->Log(DEBUG,"No FD or socket ref");
+		return;
 	}
 
-	if (this->ClosePending)
+	if (this->sock->state == I_CONNECTING)
 	{
-		this->Instance->Log(DEBUG,"Close is pending");
-		return true;
-	}
-
-	if ((this->state == I_CONNECTING) && (current > timeout_end))
-	{
-		this->Instance->Log(DEBUG,"Timed out, current=%lu timeout_end=%lu");
+		ServerInstance->Log(DEBUG,"Timed out, current=%lu",now);
 		// for non-listening sockets, the timeout can occur
 		// which causes termination of the connection after
 		// the given number of seconds without a successful
 		// connection.
-		this->OnTimeout();
-		this->OnError(I_ERR_TIMEOUT);
-		timeout = true;
-		this->state = I_ERROR;
-		return true;
+		this->sock->OnTimeout();
+		this->sock->OnError(I_ERR_TIMEOUT);
+		this->sock->timeout = true;
+		this->sock->state = I_ERROR;
+		ServerInstance->SE->DelFd(this->sock);
+		this->sock->Close();
+		delete this->sock;
+		return;
 	}
-	return this->FlushWriteBuffer();
+	this->sock->FlushWriteBuffer();
 }
 
 bool InspSocket::Poll()