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