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