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