src/modules/m_censor.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2010 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, "u_censor", 'G') { }
  30 };
  31
  32 /** Handles channel mode +G
  33  */
  34 class CensorChannel : public SimpleChannelModeHandler
  35 {
  36  public:
  37         CensorChannel(Module* Creator) : SimpleChannelModeHandler(Creator, "censor", 'G') { }
  38 };
  39
  40 class ModuleCensor : public Module
  41 {
  42         censor_t censors;
  43         CensorUser cu;
  44         CensorChannel cc;
  45
  46  public:
  47         ModuleCensor() : cu(this), cc(this) { }
  48
  49         void init()
  50         {
  51                 /* Read the configuration file on startup.
  52                  */
  53                 OnRehash(NULL);
  54                 ServerInstance->Modules->AddService(cu);
  55                 ServerInstance->Modules->AddService(cc);
  56                 Implementation eventlist[] = { I_OnRehash, I_OnUserPreMessage, I_OnUserPreNotice };
  57                 ServerInstance->Modules->Attach(eventlist, this, 3);
  58         }
  59
  60
  61         virtual ~ModuleCensor()
  62         {
  63         }
  64
  65         // format of a config entry is <badword text="shit" replace="poo">
  66         virtual ModResult OnUserPreMessage(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
  67         {
  68                 if (!IS_LOCAL(user))
  69                         return MOD_RES_PASSTHRU;
  70
  71                 bool active = false;
  72
  73                 if (target_type == TYPE_USER)
  74                         active = ((User*)dest)->IsModeSet('G');
  75                 else if (target_type == TYPE_CHANNEL)
  76                 {
  77                         active = ((Channel*)dest)->IsModeSet('G');
  78                         Channel* c = (Channel*)dest;
  79                         ModResult res;
  80                         FIRST_MOD_RESULT(OnChannelRestrictionApply, res, (user,c,"censor"));
  81
  82                         if (res == MOD_RES_ALLOW)
  83                                 return MOD_RES_PASSTHRU;
  84                 }
  85
  86                 if (!active)
  87                         return MOD_RES_PASSTHRU;
  88
  89                 irc::string text2 = text.c_str();
  90                 for (censor_t::iterator index = censors.begin(); index != censors.end(); index++)
  91                 {
  92                         if (text2.find(index->first) != irc::string::npos)
  93                         {
  94                                 if (index->second.empty())
  95                                 {
  96                                         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());
  97                                         return MOD_RES_DENY;
  98                                 }
  99
 100                                 SearchAndReplace(text2, index->first, index->second);
 101                         }
 102                 }
 103                 text = text2.c_str();
 104                 return MOD_RES_PASSTHRU;
 105         }
 106
 107         virtual ModResult OnUserPreNotice(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
 108         {
 109                 return OnUserPreMessage(user,dest,target_type,text,status,exempt_list);
 110         }
 111
 112         virtual void OnRehash(User* user)
 113         {
 114                 /*
 115                  * reload our config file on rehash - we must destroy and re-allocate the classes
 116                  * to call the constructor again and re-read our data.
 117                  */
 118                 ConfigReader MyConf;
 119                 censors.clear();
 120
 121                 for (int index = 0; index < MyConf.Enumerate("badword"); index++)
 122                 {
 123                         irc::string pattern = (MyConf.ReadValue("badword","text",index)).c_str();
 124                         irc::string replace = (MyConf.ReadValue("badword","replace",index)).c_str();
 125                         censors[pattern] = replace;
 126                 }
 127         }
 128
 129         virtual Version GetVersion()
 130         {
 131                 return Version("Provides user and channel +G mode",VF_VENDOR);
 132         }
 133
 134 };
 135
 136 MODULE_INIT(ModuleCensor)