]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_ident.cpp
Fix crash on propogation after routed squit has reached it's destination. Thx HiroP.
[user/henk/code/inspircd.git] / src / modules / m_ident.cpp
index c415741b2ff89af65a9bc93ebf9024122b2c84f1..eb9f3f365c6bde4a3b6e018650d67c560d1ccd77 100644 (file)
@@ -2,39 +2,29 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *               <Craig@chatspike.net>
- *     
- * Written by Craig Edwards, Craig McLure, and others.
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-using namespace std;
-
-#include <stdio.h>
-#include <string>
+#include "inspircd.h"
 #include "users.h"
 #include "channels.h"
 #include "modules.h"
-#include "inspircd.h"
-
-extern userrec* fd_ref_table[MAX_DESCRIPTORS];
 
 /* $ModDesc: Provides support for RFC 1413 ident lookups */
 
 // Version 1.5.0.0 - Updated to use InspSocket, faster and neater.
 
+/** Handles RFC1413 ident connections to users
+ */
 class RFC1413 : public InspSocket
 {
  protected:
-       Server* Srv;             // Server* class used for core communications
-       sockaddr_in sock_us;     // our port number
-       sockaddr_in sock_them;   // their port number
        socklen_t uslen;         // length of our port number
        socklen_t themlen;       // length of their port number
        char ident_request[128]; // buffer used to make up the request string
@@ -43,19 +33,19 @@ class RFC1413 : public InspSocket
        userrec* u;              // user record that the lookup is associated with
        int ufd;
 
-       RFC1413(userrec* user, int maxtime, Server* S) : InspSocket((char*)inet_ntoa(user->ip4), 113, false, maxtime), Srv(S), u(user), ufd(user->fd)
+       RFC1413(InspIRCd* SI, userrec* user, int maxtime, const std::string &bindto) : InspSocket(SI, user->GetIPString(), 113, false, maxtime, bindto), u(user)
        {
-               Srv->Log(DEBUG,"Ident: associated.");
+               ufd = user->GetFd();
        }
 
        virtual void OnTimeout()
        {
                // When we timeout, the connection failed within the allowed timeframe,
                // so we just display a notice, and tidy off the ident_data.
-               if (u && (fd_ref_table[ufd] == u))
+               if (u && (Instance->SE->GetRef(ufd) == u))
                {
                        u->Shrink("ident_data");
-                       Srv->SendServ(u->fd,"NOTICE "+std::string(u->nick)+" :*** Could not find your ident, using "+std::string(u->ident)+" instead.");
+                       Instance->next_call = Instance->Time();
                }
        }
 
@@ -83,11 +73,14 @@ class RFC1413 : public InspSocket
                                                                *j = '\0'; // truncate at invalid chars
                                                        if (*section)
                                                        {
-                                                               if (u && (fd_ref_table[ufd] == u))
+                                                               if (u && (Instance->SE->GetRef(ufd) == u))
                                                                {
-                                                                       strlcpy(u->ident,section,IDENTMAX);
-                                                                       Srv->Log(DEBUG,"IDENT SET: "+std::string(u->ident));
-                                                                       Srv->SendServ(u->fd,"NOTICE "+std::string(u->nick)+" :*** Found your ident: "+std::string(u->ident));
+                                                                       if (this->Instance->IsIdent(section))
+                                                                       {
+                                                                               u->Extend("IDENT", new std::string(std::string(section) + "," + std::string(u->ident)));
+                                                                               strlcpy(u->ident,section,IDENTMAX);
+                                                                               u->WriteServ("NOTICE "+std::string(u->nick)+" :*** Found your ident: "+std::string(u->ident));
+                                                                       }
                                                                }
                                                        }
                                                        return false;
@@ -121,42 +114,66 @@ class RFC1413 : public InspSocket
                // descriptor that they were when the lookup began.
                //
                // Fixes issue reported by webs, 7 Jun 2006
-               if (u && (fd_ref_table[ufd] == u))
+               if (u && (Instance->SE->GetRef(ufd) == u))
                {
+                       Instance->next_call = Instance->Time();
                        u->Shrink("ident_data");
                }
        }
 
        virtual void OnError(InspSocketError e)
        {
-               if (u && (fd_ref_table[ufd] == u))
+               if (u && (Instance->SE->GetRef(ufd) == u))
                {
+                       if (*u->ident == '~')
+                               u->WriteServ("NOTICE "+std::string(u->nick)+" :*** Could not find your ident, using "+std::string(u->ident)+" instead.");
+
+                       Instance->next_call = Instance->Time();
                        u->Shrink("ident_data");
                }
        }
 
        virtual bool OnConnected()
        {
-               if (u && (fd_ref_table[ufd] == u))
+               if (u && (Instance->SE->GetRef(ufd) == u))
                {
-                       uslen = sizeof(sock_us);
-                       themlen = sizeof(sock_them);
-                       if ((getsockname(this->u->fd,(sockaddr*)&sock_us,&uslen) || getpeername(this->u->fd, (sockaddr*)&sock_them, &themlen)))
+                       sockaddr* sock_us = new sockaddr[2];
+                       sockaddr* sock_them = new sockaddr[2];
+                       bool success = false;
+                       uslen = sizeof(sockaddr_in);
+                       themlen = sizeof(sockaddr_in);
+#ifdef IPV6
+                       if (this->u->GetProtocolFamily() == AF_INET6)
                        {
-                               Srv->Log(DEBUG,"Ident: failed to get socket names, bailing");
+                               themlen = sizeof(sockaddr_in6);
+                               uslen = sizeof(sockaddr_in6);
+                       }
+#endif
+                       success = ((getsockname(this->u->GetFd(),sock_us,&uslen) || getpeername(this->u->GetFd(), sock_them, &themlen)));
+                       if (success)
+                       {
+                               delete[] sock_us;
+                               delete[] sock_them;
                                return false;
                        }
                        else
                        {
                                // send the request in the following format: theirsocket,oursocket
-                               snprintf(ident_request,127,"%d,%d\r\n",ntohs(sock_them.sin_port),ntohs(sock_us.sin_port));
+#ifdef IPV6
+                               if (this->u->GetProtocolFamily() == AF_INET6)
+                                       snprintf(ident_request,127,"%d,%d\r\n",ntohs(((sockaddr_in6*)sock_them)->sin6_port),ntohs(((sockaddr_in6*)sock_us)->sin6_port));
+                               else
+#endif
+                               snprintf(ident_request,127,"%d,%d\r\n",ntohs(((sockaddr_in*)sock_them)->sin_port),ntohs(((sockaddr_in*)sock_us)->sin_port));
                                this->Write(ident_request);
-                               Srv->Log(DEBUG,"Sent ident request, waiting for reply");
+                               delete[] sock_us;
+                               delete[] sock_them;
                                return true;
                        }
                }
                else
                {
+                       Instance->next_call = Instance->Time();
                        return true;
                }
        }
@@ -166,23 +183,24 @@ class ModuleIdent : public Module
 {
 
        ConfigReader* Conf;
-       Server* Srv;
        int IdentTimeout;
+       std::string PortBind;
 
  public:
        void ReadSettings()
        {
-               Conf = new ConfigReader;
-               IdentTimeout = Conf->ReadInteger("ident","timeout",0,true);
+               Conf = new ConfigReader(ServerInstance);
+               IdentTimeout = Conf->ReadInteger("ident", "timeout", 0, true);
+               PortBind = Conf->ReadValue("ident", "bind", 0);
                if (!IdentTimeout)
                        IdentTimeout = 1;
                DELETE(Conf);
        }
 
-       ModuleIdent(Server* Me)
-               : Module::Module(Me)
+       ModuleIdent(InspIRCd* Me)
+               : Module(Me)
        {
-               Srv = Me;
+
                ReadSettings();
        }
 
@@ -191,12 +209,23 @@ class ModuleIdent : public Module
                List[I_OnCleanup] = List[I_OnRehash] = List[I_OnUserRegister] = List[I_OnCheckReady] = List[I_OnUserDisconnect] = 1;
        }
 
-       virtual void OnRehash(const std::string &parameter)
+       void OnSyncUserMetaData(userrec* user, Module* proto,void* opaque, const std::string &extname, bool displayable)
+       {
+               if ((displayable) && (extname == "IDENT"))
+               {
+                       std::string* ident;
+                       if (GetExt("IDENT", ident))
+                               proto->ProtoSendMetaData(opaque, TYPE_USER, user, extname, *ident);
+               }
+       }
+
+
+       virtual void OnRehash(userrec* user, const std::string &parameter)
        {
                ReadSettings();
        }
 
-       virtual void OnUserRegister(userrec* user)
+       virtual int OnUserRegister(userrec* user)
        {
                /*
                 * when the new user connects, before they authenticate with USER/NICK/PASS, we do
@@ -204,18 +233,24 @@ class ModuleIdent : public Module
                 * is derived from InspSocket, and inserting it into the socket engine using the
                 * Server::AddSocket() call.
                 */
-               Srv->SendServ(user->fd,"NOTICE "+std::string(user->nick)+" :*** Looking up your ident...");
-               RFC1413* ident = new RFC1413(user, IdentTimeout, Srv);
-               if (ident->GetState() != I_ERROR)
+               char newident[MAXBUF];
+               strcpy(newident,"~");
+               strlcat(newident,user->ident,IDENTMAX);
+               strlcpy(user->ident,newident,IDENTMAX);
+               
+
+               user->WriteServ("NOTICE "+std::string(user->nick)+" :*** Looking up your ident...");
+               RFC1413* ident = new RFC1413(ServerInstance, user, IdentTimeout, PortBind);
+               if ((ident->GetState() == I_CONNECTING) || (ident->GetState() == I_CONNECTED))
                {
                        user->Extend("ident_data", (char*)ident);
-                       Srv->AddSocket(ident);
                }
                else
                {
-                       Srv->SendServ(user->fd,"NOTICE "+std::string(user->nick)+" :*** Could not find your ident, using "+std::string(user->ident)+" instead.");
-                       DELETE(ident);
+                       user->WriteServ("NOTICE "+std::string(user->nick)+" :*** Could not find your ident, using "+std::string(user->ident)+" instead.");
+                       ServerInstance->next_call = ServerInstance->Time();
                }
+               return 0;
        }
 
        virtual bool OnCheckReady(userrec* user)
@@ -225,8 +260,8 @@ class ModuleIdent : public Module
                 * either due to timeout or due to closing, so, we just hold them until they dont
                 * have an ident field any more.
                 */
-               RFC1413* ident = (RFC1413*)user->GetExt("ident_data");
-               return (!ident);
+               RFC1413* ident;
+               return (!user->GetExt("ident_data", ident));
        }
 
        virtual void OnCleanup(int target_type, void* item)
@@ -234,15 +269,21 @@ class ModuleIdent : public Module
                if (target_type == TYPE_USER)
                {
                        userrec* user = (userrec*)item;
-                       RFC1413* ident = (RFC1413*)user->GetExt("ident_data");
-                       if (ident)
+                       RFC1413* ident;
+                       std::string* identstr;
+                       if (user->GetExt("ident_data", ident))
                        {
                                // FIX: If the user record is deleted, the socket wont be removed
                                // immediately so there is chance of the socket trying to write to
                                // a user which has now vanished! To prevent this, set ident::u
                                // to NULL and check it so that we dont write users who have gone away.
                                ident->u = NULL;
-                               Srv->RemoveSocket(ident);
+                               ServerInstance->SE->DelFd(ident);
+                               //delete ident;
+                       }
+                       if (user->GetExt("IDENT", identstr))
+                       {
+                               delete identstr;
                        }
                }
        }
@@ -257,23 +298,29 @@ class ModuleIdent : public Module
                 * who have quit, as class RFC1459 is only loosely bound to userrec* via a pair of pointers
                 * and this would leave at least one of the invalid ;)
                 */
-               RFC1413* ident = (RFC1413*)user->GetExt("ident_data");
-               if (ident)
+               RFC1413* ident;
+               std::string* identstr;
+               if (user->GetExt("ident_data", ident))
                {
                        ident->u = NULL;
-                       Srv->RemoveSocket(ident);
+                       ServerInstance->SE->DelFd(ident);
+               }
+               if (user->GetExt("IDENT", identstr))
+               {
+                       delete identstr;
                }
        }
-       
+
        virtual ~ModuleIdent()
        {
+               ServerInstance->next_call = ServerInstance->Time();
        }
-       
+
        virtual Version GetVersion()
        {
-               return Version(1,5,0,0,VF_VENDOR);
+               return Version(1,1,0,0,VF_VENDOR,API_VERSION);
        }
-       
+
 };
 
 class ModuleIdentFactory : public ModuleFactory
@@ -282,20 +329,20 @@ class ModuleIdentFactory : public ModuleFactory
        ModuleIdentFactory()
        {
        }
-       
+
        ~ModuleIdentFactory()
        {
        }
-       
-       virtual Module * CreateModule(Server* Me)
+
+       virtual Module * CreateModule(InspIRCd* Me)
        {
                return new ModuleIdent(Me);
        }
-       
+
 };
 
 
-extern "C" void * init_module( void )
+extern "C" DllExport void * init_module( void )
 {
        return new ModuleIdentFactory;
 }