1 files changed, 229 insertions, 148 deletions

diff --git a/src/modules/m_ident.cpp b/src/modules/m_ident.cpp
index 247d8eaf8..1fe978f5d 100644
--- a/src/modules/m_ident.cpp
+++ b/src/modules/m_ident.cpp
@@ -6,7 +6,7 @@
  * See: http://www.inspircd.org/wiki/index.php/Credits
  *
  * This program is free but copyrighted software; see
- *            the file COPYING for details.
+ *	    the file COPYING for details.
  *
  * ---------------------------------------------------
  */
@@ -18,117 +18,198 @@
 
 /* $ModDesc: Provides support for RFC1413 ident lookups */
 
-class IdentRequestSocket : public InspSocket
+class IdentRequestSocket : public EventHandler
 {
  private:
-	userrec *user;
-	int original_fd;
+
+	 userrec *user;
+	 InspIRCd* ServerInstance;
+	 bool done;
+	 std::string result;
+
  public:
-	IdentRequestSocket(InspIRCd *Server, userrec *user, int timeout, const std::string &bindip)
-		: InspSocket(Server, user->GetIPString(), 113, false, timeout, bindip), user(user)
-	{
-		original_fd = user->GetFd();
-		Instance->Log(DEBUG, "Ident request against user with fd %d", original_fd);
-	}
 
-	virtual bool OnConnected()
+	IdentRequestSocket(InspIRCd *Server, userrec* u, const std::string &bindip) : user(u), ServerInstance(Server), result(u->ident)
 	{
-		if (Instance->SE->GetRef(original_fd) == user)
+		/* connect here on instantiation, throw on immediate failure */
+
+		socklen_t size = 0;
+
+#ifdef IPV6
+		bool v6 = false;
+		if ((bindip.empty()) || bindip.find(':') != std::string::npos)
+		v6 = true;
+
+		if (v6)
+			SetFd(socket(AF_INET6, SOCK_STREAM, 0));
+		else
+#endif
+			SetFd(socket(AF_INET, SOCK_STREAM, 0));
+
+		if (GetFd() == -1)
+			throw ModuleException("Could not create socket");
+
+		sockaddr* s = new sockaddr[2];
+		sockaddr* addr = new sockaddr[2];
+	
+#ifdef IPV6
+		if (v6)
 		{
-			Instance->Log(DEBUG,"Oh dear, our user has gone AWOL on fd %d", original_fd);
-			return false;
+			in6_addr addy;
+			in6_addr n;
+			if (inet_pton(AF_INET6, user->GetIPString(), &addy) > 0)
+			{
+				((sockaddr_in6*)addr)->sin6_family = AF_INET6;
+				memcpy(&((sockaddr_in6*)addr)->sin6_addr, &addy, sizeof(addy));
+				((sockaddr_in6*)addr)->sin6_port = htons(113);
+				size = sizeof(sockaddr_in6);
+				inet_pton(AF_INET6, IP.c_str(), &n);
+				memcpy(&((sockaddr_in6*)s)->sin6_addr, &n, sizeof(sockaddr_in6));
+				((sockaddr_in6*)s)->sin6_port = 0;
+				((sockaddr_in6*)s)->sin6_family = AF_INET6;
+			}
 		}
+		else
+#endif
+		{
+			in_addr addy;
+			in_addr n;
+			if (inet_aton(user->GetIPString(), &addy) > 0)
+			{
+				((sockaddr_in*)addr)->sin_family = AF_INET;
+				((sockaddr_in*)addr)->sin_addr = addy;
+				((sockaddr_in*)addr)->sin_port = htons(113);
+				size = sizeof(sockaddr_in);
+				inet_aton(bindip.c_str(), &n);
+				((sockaddr_in*)s)->sin_addr = n;
+				((sockaddr_in*)s)->sin_port = 0;
+				((sockaddr_in*)s)->sin_family = AF_INET;
+			}
+		}
+
+		if (ServerInstance->SE->Bind(GetFd(), s, size) < 0)
+		{
+			this->Close();
+			delete[] s;
+			throw ModuleException("failed to bind()");
+		}
+
+		delete[] s;
+		ServerInstance->SE->NonBlocking(GetFd());
+
+		if (ServerInstance->SE->Connect(this, (sockaddr*)addr, size) == -1 && errno != EINPROGRESS)
+		{
+			this->Close();
+			delete[] addr;
+			throw ModuleException("connect() failed");
+		}
+
+		delete[] addr;
+
+		if (!ServerInstance->SE->AddFd(this))
+		{
+			this->Close();
+			throw ModuleException("out of fds");
+		}
+
+		ServerInstance->SE->WantWrite(this);
+	}
+
+	virtual void OnConnected()
+	{
+		ServerInstance->Log(DEBUG,"OnConnected()");
 
 		/* Both sockaddr_in and sockaddr_in6 can be safely casted to sockaddr, especially since the
 		 * only members we use are in a part of the struct that should always be identical (at the
 		 * byte level). */
 
-		Instance->Log(DEBUG, "Sending ident request to %s", user->GetIPString());
-		
 		#ifndef IPV6
 		sockaddr_in laddr, raddr;
 		#else
 		sockaddr_in6 laddr, raddr;
 		#endif
-		
+
 		socklen_t laddrsz = sizeof(laddr);
 		socklen_t raddrsz = sizeof(raddr);
-		
+
 		if ((getsockname(user->GetFd(), (sockaddr*) &laddr, &laddrsz) != 0) || (getpeername(user->GetFd(), (sockaddr*) &raddr, &raddrsz) != 0))
 		{
-			// Error
-			TidyUser();
-			return false;
+			done = true;
+			return;
 		}
-		
+
 		char req[32];
-		
+
 		#ifndef IPV6
-		snprintf(req, sizeof(req), "%d,%d\r\n", ntohs(raddr.sin_port), ntohs(laddr.sin_port));
+		int req_size = snprintf(req, sizeof(req), "%d,%d\r\n", ntohs(raddr.sin_port), ntohs(laddr.sin_port));
 		#else
-		snprintf(req, sizeof(req), "%d,%d\r\n", ntohs(raddr.sin6_port), ntohs(laddr.sin6_port));
+		int req_size = snprintf(req, sizeof(req), "%d,%d\r\n", ntohs(raddr.sin6_port), ntohs(laddr.sin6_port));
 		#endif
 		
-		this->Write(req);
-		
-		return true;
+		if (ServerInstance->SE->Send(this, req, req_size, 0) < req_size)
+			done = true;
 	}
 
-	virtual void OnClose()
+	virtual void HandleEvent(EventType et, int errornum = 0)
 	{
-		/* There used to be a check against the fd table here, to make sure the user hadn't been
-		 * deleted but not yet had their socket closed or something along those lines, dated june
-		 * 2006. Since we added the global cull list and such, I don't *think* that is necessary
-		 * 
-		 * -- YES IT IS!!!! DO NOT REMOVE IT, THIS IS WHAT THE WARNING ABOVE THE OLD CODE SAID :P
-		 */
-		if (Instance->SE->GetRef(original_fd) == user)
+		switch (et)
 		{
-			Instance->Log(DEBUG,"Oh dear, our user has gone AWOL on fd %d", original_fd);
-			return;
+			case EVENT_READ:
+				/* fd readable event, received ident response */
+				ReadResponse();
+			break;
+			case EVENT_WRITE:
+				/* fd writeable event, successfully connected! */
+				OnConnected();
+			break;
+			case EVENT_ERROR:
+				/* fd error event, ohshi- */
+				ServerInstance->Log(DEBUG,"EVENT_ERROR");
+				Close();
+				done = true;
+			break;
 		}
-
-		if (*user->ident == '~' && user->GetExt("ident_socket"))
-			user->WriteServ("NOTICE Auth :*** Could not find your ident, using %s instead", user->ident);
-
-		TidyUser();
-		Instance->next_call = Instance->Time();
 	}
-	
-	virtual void OnError(InspSocketError e)
+
+	void Close()
 	{
-		if (Instance->SE->GetRef(original_fd) == user)
+		if (GetFd() > -1)
 		{
-			Instance->Log(DEBUG,"Oh dear, our user has gone AWOL on fd %d", original_fd);
-			return;
+			ServerInstance->Log(DEBUG,"Close ident socket %d", GetFd());
+			ServerInstance->SE->DelFd(this);
+			ServerInstance->SE->Close(GetFd());
+			ServerInstance->SE->Shutdown(GetFd(), SHUT_WR);
+			this->SetFd(-1);
 		}
+	}
 
-		// Quick check to make sure that this hasn't been sent ;)
-		if (*user->ident == '~' && user->GetExt("ident_socket"))
-			user->WriteServ("NOTICE Auth :*** Could not find your ident, using %s instead", user->ident);
-		
-		TidyUser();
-		Instance->next_call = Instance->Time();
+	bool HasResult()
+	{
+		return done;
 	}
-	
-	virtual bool OnDataReady()
+
+	const char* GetResult()
 	{
-		if (Instance->SE->GetRef(original_fd) == user)
-		{
-			Instance->Log(DEBUG,"Oh dear, our user has gone AWOL on fd %d", original_fd);
-			return false;
-		}
+		return result.c_str();
+	}
+
+	void ReadResponse()
+	{
+		ServerInstance->Log(DEBUG,"ReadResponse()");
 
-		char *ibuf = this->Read();
-		if (!ibuf)
-		{
-			TidyUser();
-			return false;
-		}
-		
 		// We don't really need to buffer for incomplete replies here, since IDENT replies are
 		// extremely short - there is *no* sane reason it'd be in more than one packet
-		
+
+		char ibuf[MAXBUF];
+		int recvresult = ServerInstance->SE->Recv(this, ibuf, MAXBUF-1, 0);
+
+		/* Cant possibly be a valid response shorter than 3 chars */
+		if (recvresult < 3)
+		{
+			done = true;
+			return;
+		}
+
 		irc::sepstream sep(ibuf, ':');
 		std::string token;
 		for (int i = 0; sep.GetToken(token); i++)
@@ -136,52 +217,41 @@ class IdentRequestSocket : public InspSocket
 			// We only really care about the 4th portion
 			if (i < 3)
 				continue;
-			
+
 			char ident[IDENTMAX + 2];
-			
+
 			// Truncate the ident at any characters we don't like, skip leading spaces
 			int k = 0;
 			for (const char *j = token.c_str(); *j && (k < IDENTMAX + 1); j++)
 			{
 				if (*j == ' ')
 					continue;
-				
+
 				// Rules taken from InspIRCd::IsIdent
 				if (((*j >= 'A') && (*j <= '}')) || ((*j >= '0') && (*j <= '9')) || (*j == '-') || (*j == '.'))
 				{
 					ident[k++] = *j;
 					continue;
 				}
-				
+
 				break;
 			}
-			
+
 			ident[k] = '\0';
-			
+
 			// Redundant check with IsIdent, in case that changes and this doesn't (paranoia!)
-			if (*ident && Instance->IsIdent(ident))
+			if (*ident && ServerInstance->IsIdent(ident))
 			{
-				strlcpy(user->ident, ident, IDENTMAX);
-				user->WriteServ("NOTICE Auth :*** Found your ident: %s", user->ident);
-				Instance->next_call = Instance->Time();
+				result = ident;
+				ServerInstance->next_call = ServerInstance->Time();
 			}
-			
+
 			break;
 		}
-		
-		TidyUser();
-		return false;
-	}
 
-	void TidyUser()
-	{
-		user->Shrink("ident_socket");
-		int* delfd;
-		if (user->GetExt("ident_socket_fd", delfd))
-		{
-			delete delfd;
-			user->Shrink("ident_socket_fd");
-		}
+		Close();
+		done = true;
+		return;
 	}
 };
 
@@ -196,10 +266,6 @@ class ModuleIdent : public Module
 		OnRehash(NULL, "");
 	}
 	
-	virtual ~ModuleIdent()
-	{
-	}
-	
 	virtual Version GetVersion()
 	{
 		return Version(1, 1, 1, 0, VF_VENDOR, API_VERSION);
@@ -221,15 +287,15 @@ class ModuleIdent : public Module
 	
 	virtual int OnUserRegister(userrec *user)
 	{
-		/* userrec::ident is currently the username field from USER; with m_ident loaded, that
+		/* User::ident is currently the username field from USER; with m_ident loaded, that
 		 * should be preceded by a ~. The field is actually IDENTMAX+2 characters wide. */
 		memmove(user->ident + 1, user->ident, IDENTMAX);
 		user->ident[0] = '~';
 		// Ensure that it is null terminated
 		user->ident[IDENTMAX + 1] = '\0';
-		
+
 		user->WriteServ("NOTICE Auth :*** Looking up your ident...");
-		
+
 		// Get the IP that the user is connected to, and bind to that for the outgoing connection
 		#ifndef IPV6
 		sockaddr_in laddr;
@@ -237,80 +303,95 @@ class ModuleIdent : public Module
 		sockaddr_in6 laddr;
 		#endif
 		socklen_t laddrsz = sizeof(laddr);
-		
+
 		if (getsockname(user->GetFd(), (sockaddr*) &laddr, &laddrsz) != 0)
 		{
 			user->WriteServ("NOTICE Auth :*** Could not find your ident, using %s instead.", user->ident);
 			return 0;
 		}
-		
+
 		#ifndef IPV6
 		const char *ip = inet_ntoa(laddr.sin_addr);
 		#else
 		char ip[INET6_ADDRSTRLEN + 1];
 		inet_ntop(laddr.sin6_family, &laddr.sin6_addr, ip, INET6_ADDRSTRLEN);
 		#endif
-		
-		IdentRequestSocket *isock = new IdentRequestSocket(ServerInstance, user, RequestTimeout, ip);
-		if (isock->GetFd() > -1)
+
+		IdentRequestSocket *isock = NULL;
+		try
 		{
-			user->Extend("ident_socket_fd", new int(isock->GetFd()));
-			user->Extend("ident_socket", isock);
+			isock = new IdentRequestSocket(ServerInstance, user, ip);
 		}
-		else
-			if (ServerInstance->SocketCull.find(isock) == ServerInstance->SocketCull.end())
-				ServerInstance->SocketCull[isock] = isock;
+		catch (ModuleException &e)
+		{
+			ServerInstance->Log(DEBUG,"Ident exception: %s", e.GetReason());
+			return 0;
+		}
+
+		user->Extend("ident_socket", isock);
 		return 0;
 	}
-	
+
 	virtual bool OnCheckReady(userrec *user)
 	{
-		return (!user->GetExt("ident_socket"));
+		ServerInstance->Log(DEBUG,"OnCheckReady %s", user->nick);
+
+		/* Does user have an ident socket attached at all? */
+		IdentRequestSocket *isock = NULL;
+		if (!user->GetExt("ident_socket", isock))
+		{
+			ServerInstance->Log(DEBUG, "No ident socket :(");
+			return true;
+		}
+
+		ServerInstance->Log(DEBUG, "Has ident_socket");
+
+		if (isock->age + RequestTimeout > ServerInstance->Time() && !isock->HasResult())
+		{
+			/* Ident timeout */
+			user->WriteServ("NOTICE Auth :*** Ident request timed out.");
+			ServerInstance->Log(DEBUG, "Timeout");
+			OnUserDisconnect(user);
+			return true;
+		}
+
+		/* Got a result yet? */
+		if (!isock->HasResult())
+		{
+			ServerInstance->Log(DEBUG, "No result yet");
+			return false;
+		}
+
+		ServerInstance->Log(DEBUG, "Yay, result!");
+
+		/* wooo, got a result! */
+		if (*(isock->GetResult()) != '~')
+			user->WriteServ("NOTICE Auth :*** Found your ident, '%s'", isock->GetResult());
+		else
+			user->WriteServ("NOTICE Auth :*** Could not find your ident, using %s instead.", isock->GetResult());
+
+		strlcpy(user->ident, isock->GetResult(), IDENTMAX+1);
+		OnUserDisconnect(user);
+		return true;
 	}
-	
+
 	virtual void OnCleanup(int target_type, void *item)
 	{
+		/* Module unloading, tidy up users */
 		if (target_type == TYPE_USER)
-		{
-			IdentRequestSocket *isock;
-			userrec *user = (userrec*)item;
-			if (user->GetExt("ident_socket", isock))
-			{
-				int *fd;
-				if (user->GetExt("ident_socket_fd", fd))
-				{
-					if (ServerInstance->SE->GetRef(*fd) == isock)
-						isock->Close();
-
-					/* Check again, isock->Close() can confuse us */
-					if (user->GetExt("ident_socket_fd", fd))
-					{
-						user->Shrink("ident_socket_fd");
-						delete fd;
-					}
-				}
-			}
-		}
+			OnUserDisconnect((userrec*)item);
 	}
 
 	virtual void OnUserDisconnect(userrec *user)
 	{
-		IdentRequestSocket *isock;
+		/* User disconnect (generic socket detatch event) */
+		IdentRequestSocket *isock = NULL;
 		if (user->GetExt("ident_socket", isock))
 		{
-			int *fd;
-			if (user->GetExt("ident_socket_fd", fd))
-			{
-				if (ServerInstance->SE->GetRef(*fd) == isock)
-					isock->Close();
-
-				/* Check again, isock->Close() can confuse us */
-				if (user->GetExt("ident_socket_fd", fd))
-				{
-					user->Shrink("ident_socket_fd");
-					delete fd;
-				}
-			}
+			isock->Close();
+			delete isock;
+			user->Shrink("ident_socket");
+			ServerInstance->Log(DEBUG, "Removed ident socket from %s", user->nick);
 		}
 	}
 };