src/modules/m_kicknorejoin.cpp

   1 #include <time.h>
   2 #include <map>
   3 #include <vector>
   4 #include <sstream>
   5 #include "users.h"
   6 #include "channels.h"
   7 #include "modules.h"
   8 #include "helperfuncs.h"
   9 #include "inspircd.h"
  10
  11 /* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
  12
  13 inline int strtoint(const std::string &str)
  14 {
  15         std::istringstream ss(str);
  16         int result;
  17         ss >> result;
  18         return result;
  19 }
  20
  21 typedef std::map<userrec*, time_t> delaylist;
  22
  23 class KickRejoin : public ModeHandler
  24 {
  25  public:
  26         KickRejoin() : ModeHandler('J', 1, 0, false, MODETYPE_CHANNEL, false) { }
  27
  28         ModePair ModeSet(userrec* source, userrec* dest, chanrec* channel, const std::string &parameter)
  29         {
  30                 if (channel->IsModeSet('J'))
  31                         return std::make_pair(true, channel->GetModeParameter('J'));
  32                 else
  33                         return std::make_pair(false, parameter);
  34         }
  35
  36         bool CheckTimeStamp(time_t theirs, time_t ours, const std::string &their_param, const std::string &our_param, chanrec* channel)
  37         {
  38                 /* When TS is equal, the alphabetically later one wins */
  39                 return (their_param < our_param);
  40         }
  41
  42         ModeAction OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string &parameter, bool adding)
  43         {
  44                 if (!adding)
  45                 {
  46                         // Taking the mode off, we need to clean up.
  47                         delaylist* dl;
  48
  49                         if (channel->GetExt("norejoinusers", dl))
  50                         {
  51                                 DELETE(dl);
  52                                 channel->Shrink("norejoinusers");
  53                         }
  54                 }
  55                 if ((!adding) || (atoi(parameter.c_str()) > 0))
  56                 {
  57                         parameter = ConvToStr(atoi(parameter.c_str()));
  58                         channel->SetModeParam('J', parameter.c_str(), adding);
  59                         channel->SetMode('J', adding);
  60                         return MODEACTION_ALLOW;
  61                 }
  62                 else
  63                 {
  64                         return MODEACTION_DENY;
  65                 }
  66         }
  67 };
  68
  69 class ModuleKickNoRejoin : public Module
  70 {
  71         Server *Srv;
  72         KickRejoin* kr;
  73
  74 public:
  75
  76         ModuleKickNoRejoin(Server* Me)
  77                 : Module::Module(Me)
  78         {
  79                 Srv = Me;
  80                 kr = new KickRejoin();
  81                 Srv->AddMode(kr, 'J');
  82         }
  83
  84         virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname)
  85         {
  86                 if (chan)
  87                 {
  88                         delaylist* dl;
  89                         if (chan->GetExt("norejoinusers", dl))
  90                         {
  91                                 log(DEBUG, "m_kicknorejoin.so: got delay list, iterating over it");
  92                                 std::vector<userrec*> itemstoremove;
  93
  94                                 for (delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
  95                                 {
  96                                         log(DEBUG, "m_kicknorejoin.so:\t[%s] => %d", iter->first->nick, iter->second);
  97                                         if (iter->second > time(NULL))
  98                                         {
  99                                                 log(DEBUG, "m_kicknorejoin.so: still inside time slot");
 100                                                 if (iter->first == user)
 101                                                 {
 102                                                         log(DEBUG, "m_kicknorejoin.so: and we have the right user");
 103                                                         WriteServ(user->fd, "495 %s %s :You cannot rejoin this channel yet after being kicked (+J)", user->nick, chan->name);
 104                                                         return 1;
 105                                                 }
 106                                         }
 107                                         else
 108                                         {
 109                                                 // Expired record, remove.
 110                                                 log(DEBUG, "m_kicknorejoin.so: record expired");
 111                                                 itemstoremove.push_back(iter->first);
 112                                         }
 113                                 }
 114
 115                                 for (unsigned int i = 0; i < itemstoremove.size(); i++)
 116                                         dl->erase(itemstoremove[i]);
 117
 118                                 if (!dl->size())
 119                                 {
 120                                         // Now it's empty..
 121                                                 DELETE(dl);
 122                                                 chan->Shrink("norejoinusers");
 123                                 }
 124                         }
 125                 }
 126                 return 0;
 127         }
 128
 129         virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason)
 130         {
 131                 if (chan->IsModeSet('J') && (source != user))
 132                 {
 133                         delaylist* dl;
 134                         if (!chan->GetExt("norejoinusers", dl))
 135                         {
 136                                 dl = new delaylist;
 137                                 chan->Extend("norejoinusers", dl);
 138                         }
 139
 140                         log(DEBUG, "m_kicknorejoin.so: setting record for %s, %d second delay", user->nick, strtoint(chan->GetModeParameter('J')));
 141                         (*dl)[user] = time(NULL) + strtoint(chan->GetModeParameter('J'));
 142                 }
 143         }
 144
 145         virtual void OnChannelDelete(chanrec* chan)
 146         {
 147                 delaylist* dl;
 148
 149                 if (chan->GetExt("norejoinusers", dl))
 150                 {
 151                         DELETE(dl);
 152                         chan->Shrink("norejoinusers");
 153                 }
 154         }
 155
 156         virtual void OnCleanup(int target_type, void* item)
 157         {
 158                 if(target_type == TYPE_CHANNEL)
 159                         OnChannelDelete((chanrec*)item);
 160         }
 161
 162         virtual void Implements(char* List)
 163         {
 164                 List[I_OnCleanup] = List[I_On005Numeric] = List[I_OnChannelDelete] = List[I_OnUserPreJoin] = List[I_OnUserKick] = 1;
 165         }
 166
 167         virtual void On005Numeric(std::string &output)
 168         {
 169                 InsertMode(output, "J", 3);
 170         }
 171
 172         virtual ~ModuleKickNoRejoin()
 173         {
 174                 DELETE(kr);
 175         }
 176
 177         virtual Version GetVersion()
 178         {
 179                 return Version(1, 0, 0, 0, VF_STATIC | VF_VENDOR);
 180         }
 181 };
 182
 183
 184 class ModuleKickNoRejoinFactory : public ModuleFactory
 185 {
 186  public:
 187         ModuleKickNoRejoinFactory()
 188         {
 189         }
 190
 191         ~ModuleKickNoRejoinFactory()
 192         {
 193         }
 194
 195         virtual Module * CreateModule(Server* Me)
 196         {
 197                 return new ModuleKickNoRejoin(Me);
 198         }
 199
 200 };
 201
 202
 203 extern "C" void * init_module( void )
 204 {
 205         return new ModuleKickNoRejoinFactory;
 206 }
 207