Include explicit routing information in Command, will replace CMD_LOCALONLY return...

[user/henk/code/inspircd.git] / src / modules / m_cloaking.cpp

diff --git a/src/modules/m_cloaking.cpp b/src/modules/m_cloaking.cpp
index 59dd008af9609bf0c4f5bcec69f194dab0e654f5..a236300f18404bf623e7ad73a12b738862dd4530 100644 (file)
--- a/src/modules/m_cloaking.cpp
+++ b/src/modules/m_cloaking.cpp
@@ -250,16 +250,14 @@ class CloakUser : public ModeHandler
 class ModuleCloaking : public Module
 {
  private:
-
        CloakUser* cu;
-       Module* HashModule;
 
  public:
        ModuleCloaking(InspIRCd* Me)
                : Module(Me)
        {
                /* Attempt to locate the md5 service provider, bail if we can't find it */
-               HashModule = ServerInstance->Modules->Find("m_md5.so");
+               Module* HashModule = ServerInstance->Modules->Find("m_md5.so");
                if (!HashModule)
                        throw ModuleException("Can't find m_md5.so. Please load m_md5.so before m_cloaking.so.");
 
@@ -410,35 +408,21 @@ class ModuleCloaking : public Module
                                 * Their ISP shouldnt go to town on subdomains, or they shouldnt have a kiddie
                                 * vhost.
                                 */
-#ifdef IPV6
-                               in6_addr testaddr;
-                               in_addr testaddr2;
-                               if ((dest->GetProtocolFamily() == AF_INET6) && (inet_pton(AF_INET6,dest->host.c_str(),&testaddr) < 1) && (hostcloak.length() <= 64))
-                                       /* Invalid ipv6 address, and ipv6 user (resolved host) */
-                                       b = hostcloak;
-                               else if ((dest->GetProtocolFamily() == AF_INET) && (inet_aton(dest->host.c_str(),&testaddr2) < 1) && (hostcloak.length() <= 64))
-                                       /* Invalid ipv4 address, and ipv4 user (resolved host) */
-                                       b = hostcloak;
-                               else
-                                       /* Valid ipv6 or ipv4 address (not resolved) ipv4 or ipv6 user */
-                                       b = ((!strchr(dest->host.c_str(),':')) ? cu->Cloak4(dest->host.c_str()) : cu->Cloak6(dest->host.c_str()));
-#else
-                               in_addr testaddr;
-                               if ((inet_aton(dest->host.c_str(),&testaddr) < 1) && (hostcloak.length() <= 64))
-                                       /* Invalid ipv4 address, and ipv4 user (resolved host) */
+                               std::string testaddr;
+                               int testport;
+                               if (!irc::sockets::satoap(&dest->client_sa, testaddr, testport) && (hostcloak.length() <= 64))
+                                       /* not a valid address, must have been a host, so cloak as a host */
                                        b = hostcloak;
+                               else if (dest->client_sa.sa.sa_family == AF_INET6)
+                                       b = cu->Cloak6(dest->GetIPString());
                                else
-                                       /* Valid ipv4 address (not resolved) ipv4 user */
-                                       b = cu->Cloak4(dest->host.c_str());
-#endif
+                                       b = cu->Cloak4(dest->GetIPString());
                        }
                        else
                        {
-#ifdef IPV6
-                               if (dest->GetProtocolFamily() == AF_INET6)
+                               if (dest->client_sa.sa.sa_family == AF_INET6)
                                        b = cu->Cloak6(dest->GetIPString());
-#endif
-                               if (dest->GetProtocolFamily() == AF_INET)
+                               else
                                        b = cu->Cloak4(dest->GetIPString());
                        }