src/modules/extra/m_geoip.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Craig Edwards <craigedwards@brainbox.cc>
   6  *
   7  * This file is part of InspIRCd.  InspIRCd is free software: you can
   8  * redistribute it and/or modify it under the terms of the GNU General Public
   9  * License as published by the Free Software Foundation, version 2.
  10  *
  11  * This program is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  13  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  14  * details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20
  21 #include "inspircd.h"
  22 #include "xline.h"
  23
  24 #include <GeoIP.h>
  25
  26 #ifdef _WIN32
  27 # pragma comment(lib, "GeoIP.lib")
  28 #endif
  29
  30 /* $LinkerFlags: -lGeoIP */
  31
  32 class ModuleGeoIP : public Module
  33 {
  34         LocalStringExt ext;
  35         GeoIP* gi;
  36
  37         std::string* SetExt(LocalUser* user)
  38         {
  39                 const char* c = GeoIP_country_code_by_addr(gi, user->GetIPString().c_str());
  40                 if (!c)
  41                         c = "UNK";
  42
  43                 std::string* cc = new std::string(c);
  44                 ext.set(user, cc);
  45                 return cc;
  46         }
  47
  48  public:
  49         ModuleGeoIP() : ext("geoip_cc", this), gi(NULL)
  50         {
  51         }
  52
  53         void init() CXX11_OVERRIDE
  54         {
  55                 gi = GeoIP_new(GEOIP_STANDARD);
  56                 if (gi == NULL)
  57                                 throw ModuleException("Unable to initialize geoip, are you missing GeoIP.dat?");
  58
  59                 const UserManager::LocalList& list = ServerInstance->Users.GetLocalUsers();
  60                 for (UserManager::LocalList::const_iterator i = list.begin(); i != list.end(); ++i)
  61                 {
  62                         LocalUser* user = *i;
  63                         if ((user->registered == REG_ALL) && (!ext.get(user)))
  64                         {
  65                                 SetExt(user);
  66                         }
  67                 }
  68         }
  69
  70         ~ModuleGeoIP()
  71         {
  72                 if (gi)
  73                         GeoIP_delete(gi);
  74         }
  75
  76         Version GetVersion() CXX11_OVERRIDE
  77         {
  78                 return Version("Provides a way to assign users to connect classes by country using GeoIP lookup", VF_VENDOR);
  79         }
  80
  81         ModResult OnSetConnectClass(LocalUser* user, ConnectClass* myclass) CXX11_OVERRIDE
  82         {
  83                 std::string* cc = ext.get(user);
  84                 if (!cc)
  85                         cc = SetExt(user);
  86
  87                 std::string geoip = myclass->config->getString("geoip");
  88                 if (geoip.empty())
  89                         return MOD_RES_PASSTHRU;
  90                 irc::commasepstream list(geoip);
  91                 std::string country;
  92                 while (list.GetToken(country))
  93                         if (country == *cc)
  94                                 return MOD_RES_PASSTHRU;
  95                 return MOD_RES_DENY;
  96         }
  97
  98         ModResult OnStats(char symbol, User* user, string_list &out) CXX11_OVERRIDE
  99         {
 100                 if (symbol != 'G')
 101                         return MOD_RES_PASSTHRU;
 102
 103                 unsigned int unknown = 0;
 104                 std::map<std::string, unsigned int> results;
 105
 106                 const UserManager::LocalList& list = ServerInstance->Users.GetLocalUsers();
 107                 for (UserManager::LocalList::const_iterator i = list.begin(); i != list.end(); ++i)
 108                 {
 109                         std::string* cc = ext.get(*i);
 110                         if (cc)
 111                                 results[*cc]++;
 112                         else
 113                                 unknown++;
 114                 }
 115
 116                 std::string p = "801 " + user->nick + " :GeoIPSTATS ";
 117                 for (std::map<std::string, unsigned int>::const_iterator i = results.begin(); i != results.end(); ++i)
 118                 {
 119                         out.push_back(p + i->first + " " + ConvToStr(i->second));
 120                 }
 121
 122                 if (unknown)
 123                         out.push_back(p + "Unknown " + ConvToStr(unknown));
 124
 125                 return MOD_RES_DENY;
 126         }
 127 };
 128
 129 MODULE_INIT(ModuleGeoIP)