src/modules/m_censor.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #define _CRT_SECURE_NO_DEPRECATE
  15 #define _SCL_SECURE_NO_DEPRECATE
  16
  17 #include "inspircd.h"
  18 #include <iostream>
  19
  20 typedef std::map<irc::string,irc::string> censor_t;
  21
  22 /* $ModDesc: Provides user and channel +G mode */
  23
  24 /** Handles usermode +G
  25  */
  26 class CensorUser : public SimpleUserModeHandler
  27 {
  28  public:
  29         CensorUser(Module* Creator) : SimpleUserModeHandler(Creator, 'G') { }
  30 };
  31
  32 /** Handles channel mode +G
  33  */
  34 class CensorChannel : public SimpleChannelModeHandler
  35 {
  36  public:
  37         CensorChannel(Module* Creator) : SimpleChannelModeHandler(Creator, 'G') { }
  38 };
  39
  40 class ModuleCensor : public Module
  41 {
  42         censor_t censors;
  43         CensorUser cu;
  44         CensorChannel cc;
  45
  46  public:
  47         ModuleCensor()
  48                 : cu(this), cc(this)
  49         {
  50                 /* Read the configuration file on startup.
  51                  */
  52                 OnRehash(NULL);
  53                 if (!ServerInstance->Modes->AddMode(&cu) || !ServerInstance->Modes->AddMode(&cc))
  54                         throw ModuleException("Could not add new modes!");
  55                 Implementation eventlist[] = { I_OnRehash, I_OnUserPreMessage, I_OnUserPreNotice, I_OnRunTestSuite };
  56                 ServerInstance->Modules->Attach(eventlist, this, 4);
  57         }
  58
  59
  60         virtual ~ModuleCensor()
  61         {
  62         }
  63
  64         // format of a config entry is <badword text="shit" replace="poo">
  65         virtual ModResult OnUserPreMessage(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
  66         {
  67                 if (!IS_LOCAL(user))
  68                         return MOD_RES_PASSTHRU;
  69
  70                 bool active = false;
  71
  72                 if (target_type == TYPE_USER)
  73                         active = ((User*)dest)->IsModeSet('G');
  74                 else if (target_type == TYPE_CHANNEL)
  75                 {
  76                         active = ((Channel*)dest)->IsModeSet('G');
  77                         Channel* c = (Channel*)dest;
  78                         if (CHANOPS_EXEMPT('G') && c->GetPrefixValue(user) == OP_VALUE)
  79                         {
  80                                 return MOD_RES_PASSTHRU;
  81                         }
  82                 }
  83
  84                 if (!active)
  85                         return MOD_RES_PASSTHRU;
  86
  87                 irc::string text2 = text.c_str();
  88                 for (censor_t::iterator index = censors.begin(); index != censors.end(); index++)
  89                 {
  90                         if (text2.find(index->first) != irc::string::npos)
  91                         {
  92                                 if (index->second.empty())
  93                                 {
  94                                         user->WriteNumeric(ERR_WORDFILTERED, "%s %s %s :Your message contained a censored word, and was blocked", user->nick.c_str(), ((Channel*)dest)->name.c_str(), index->first.c_str());
  95                                         return MOD_RES_DENY;
  96                                 }
  97
  98                                 SearchAndReplace(text2, index->first, index->second);
  99                         }
 100                 }
 101                 text = text2.c_str();
 102                 return MOD_RES_PASSTHRU;
 103         }
 104
 105         virtual ModResult OnUserPreNotice(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
 106         {
 107                 return OnUserPreMessage(user,dest,target_type,text,status,exempt_list);
 108         }
 109
 110         virtual void OnRehash(User* user)
 111         {
 112                 /*
 113                  * reload our config file on rehash - we must destroy and re-allocate the classes
 114                  * to call the constructor again and re-read our data.
 115                  */
 116                 ConfigReader* MyConf = new ConfigReader;
 117                 censors.clear();
 118
 119                 for (int index = 0; index < MyConf->Enumerate("badword"); index++)
 120                 {
 121                         irc::string pattern = (MyConf->ReadValue("badword","text",index)).c_str();
 122                         irc::string replace = (MyConf->ReadValue("badword","replace",index)).c_str();
 123                         censors[pattern] = replace;
 124                 }
 125
 126                 delete MyConf;
 127         }
 128
 129         virtual Version GetVersion()
 130         {
 131                 return Version("Provides user and channel +G mode",VF_COMMON|VF_VENDOR,API_VERSION);
 132         }
 133
 134 };
 135
 136 MODULE_INIT(ModuleCensor)