/* * InspIRCd -- Internet Relay Chat Daemon * * Copyright (C) 2009-2010 Daniel De Graaf * Copyright (C) 2008 Craig Edwards * * This file is part of InspIRCd. InspIRCd is free software: you can * redistribute it and/or modify it under the terms of the GNU General Public * License as published by the Free Software Foundation, version 2. * * This program is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more * details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ #include "inspircd.h" #include "xline.h" #include #ifdef WINDOWS # pragma comment(lib, "GeoIP.lib") #endif /* $ModDesc: Provides a way to restrict users by country using GeoIP lookup */ /* $LinkerFlags: -lGeoIP */ class ModuleGeoIP : public Module { LocalStringExt ext; GeoIP* gi; public: ModuleGeoIP() : ext("geoip_cc", this), gi(NULL) { } void init() { gi = GeoIP_new(GEOIP_STANDARD); if (gi == NULL) throw ModuleException("Unable to initialize geoip, are you missing GeoIP.dat?"); ServerInstance->Modules->AddService(ext); Implementation eventlist[] = { I_OnSetConnectClass }; ServerInstance->Modules->Attach(eventlist, this, 1); } ~ModuleGeoIP() { GeoIP_delete(gi); } Version GetVersion() { return Version("Provides a way to assign users to connect classes by country using GeoIP lookup", VF_VENDOR); } ModResult OnSetConnectClass(LocalUser* user, ConnectClass* myclass) { std::string* cc = ext.get(user); if (!cc) { const char* c = GeoIP_country_code_by_addr(gi, user->GetIPString()); if (!c) c = "UNK"; cc = new std::string(c); ext.set(user, cc); } std::string geoip = myclass->config->getString("geoip"); if (geoip.empty()) return MOD_RES_PASSTHRU; irc::commasepstream list(geoip); std::string country; while (list.GetToken(country)) if (country == *cc) return MOD_RES_PASSTHRU; return MOD_RES_DENY; } }; MODULE_INIT(ModuleGeoIP)