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