X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_dnsbl.cpp;h=5e02dd0a5dadc5cea64d99375285f24a5a8bb5e6;hb=47332d6e9b990498dd35457090dd8983d8aae8d3;hp=d7d013691f3f100c91cbd4f621d618e6809b2ab7;hpb=6c67546af64cec887bb716f666016eb93fde5d2b;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/modules/m_dnsbl.cpp b/src/modules/m_dnsbl.cpp
index d7d013691..5e02dd0a5 100644
--- a/src/modules/m_dnsbl.cpp
+++ b/src/modules/m_dnsbl.cpp
@@ -240,7 +240,7 @@ class ModuleDNSBL : public Module
 		ServerInstance->Modules->AddService(nameExt);
 		ServerInstance->Modules->AddService(countExt);
 		Implementation eventlist[] = { I_OnRehash, I_OnSetUserIP, I_OnStats, I_OnSetConnectClass, I_OnCheckReady };
-		ServerInstance->Modules->Attach(eventlist, this, 5);
+		ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
 	}
 
 	virtual ~ModuleDNSBL()
@@ -352,16 +352,21 @@ class ModuleDNSBL : public Module
 
 	void OnSetUserIP(LocalUser* user)
 	{
-		if (user->exempt)
+		if ((user->exempt) || (user->client_sa.sa.sa_family != AF_INET))
 			return;
 
+		if (user->MyClass)
+		{
+			if (!user->MyClass->config->getBool("usednsbl", true))
+				return;
+		}
+		else
+			ServerInstance->Logs->Log("m_dnsbl", DEBUG, "User has no connect class in OnSetUserIP");
+
 		unsigned char a, b, c, d;
 		char reversedipbuf[128];
 		std::string reversedip;
 
-		if (user->client_sa.sa.sa_family != AF_INET)
-			return;
-
 		d = (unsigned char) (user->client_sa.in4.sin_addr.s_addr >> 24) & 0xFF;
 		c = (unsigned char) (user->client_sa.in4.sin_addr.s_addr >> 16) & 0xFF;
 		b = (unsigned char) (user->client_sa.in4.sin_addr.s_addr >> 8) & 0xFF;
@@ -370,6 +375,8 @@ class ModuleDNSBL : public Module
 		snprintf(reversedipbuf, 128, "%d.%d.%d.%d", d, c, b, a);
 		reversedip = std::string(reversedipbuf);
 
+		countExt.set(user, DNSBLConfEntries.size());
+
 		// For each DNSBL, we will run through this lookup
 		unsigned int i = 0;
 		while (i < DNSBLConfEntries.size())
@@ -381,9 +388,10 @@ class ModuleDNSBL : public Module
 			bool cached;
 			DNSBLResolver *r = new DNSBLResolver(this, nameExt, countExt, hostname, user, DNSBLConfEntries[i], cached);
 			ServerInstance->AddResolver(r, cached);
+			if (user->quitting)
+				break;
 			i++;
 		}
-		countExt.set(user, i);
 	}
 
 	ModResult OnSetConnectClass(LocalUser* user, ConnectClass* myclass)