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
  10 /* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
  11
  12 inline int strtoint(const std::string &str)
  13 {
  14         std::istringstream ss(str);
  15         int result;
  16         ss >> result;
  17         return result;
  18 }
  19
  20 typedef std::map<userrec*, time_t> delaylist;
  21
  22 class ModuleKickNoRejoin : public Module
  23 {
  24         Server *Srv;
  25
  26 public:
  27
  28         ModuleKickNoRejoin(Server* Me)
  29                 : Module::Module(Me)
  30         {
  31                 Srv = Me;
  32
  33                 Srv->AddExtendedMode('J', MT_CHANNEL, false, 1, 0);
  34         }
  35
  36         virtual int OnExtendedMode(userrec* user, void* target, char modechar, int type, bool mode_on, string_list &params)
  37         {
  38                 if ((modechar == 'J') && (type == MT_CHANNEL))
  39                 {
  40                         if (!mode_on)
  41                         {
  42                                 // Taking the mode off, we need to clean up.
  43                                 chanrec* c = (chanrec*)target;
  44
  45                                 delaylist* dl = (delaylist*)c->GetExt("norejoinusers");
  46
  47                                 if (dl)
  48                                 {
  49                                         delete dl;
  50                                         c->Shrink("norejoinusers");
  51                                 }
  52                         }
  53                         /* Don't allow negative or 0 +J value */
  54                         return ((!mode_on) || (atoi(params[0].c_str()) > 0));
  55                 }
  56                 return 0;
  57         }
  58
  59         virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname)
  60         {
  61                 if (chan)
  62                 {
  63                         delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
  64                         log(DEBUG, "m_kicknorejoin.so: tried to grab delay list");
  65
  66                         if (dl)
  67                         {
  68                                 log(DEBUG, "m_kicknorejoin.so: got delay list, iterating over it");
  69                                 std::vector<userrec*> itemstoremove;
  70
  71                                 for (delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
  72                                 {
  73                                         log(DEBUG, "m_kicknorejoin.so:\t[%s] => %d", iter->first->nick, iter->second);
  74                                         if (iter->second > time(NULL))
  75                                         {
  76                                                 log(DEBUG, "m_kicknorejoin.so: still inside time slot");
  77                                                 if (iter->first == user)
  78                                                 {
  79                                                         log(DEBUG, "m_kicknorejoin.so: and we have the right user");
  80                                                         WriteServ(user->fd, "495 %s %s :You cannot rejoin this channel yet after being kicked (+J)", user->nick, chan->name);
  81                                                         return 1;
  82                                                 }
  83                                         }
  84                                         else
  85                                         {
  86                                                 // Expired record, remove.
  87                                                 log(DEBUG, "m_kicknorejoin.so: record expired");
  88                                                 itemstoremove.push_back(iter->first);
  89                                         }
  90                                 }
  91
  92                                 for (unsigned int i = 0; i < itemstoremove.size(); i++)
  93                                         dl->erase(itemstoremove[i]);
  94
  95                                 if (!dl->size())
  96                                 {
  97                                         // Now it's empty..
  98                                         delete dl;
  99                                         chan->Shrink("norejoinusers");
 100                                 }
 101                         }
 102                 }
 103                 return 0;
 104         }
 105
 106         virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, const std::string &reason)
 107         {
 108                 if (chan->IsModeSet('J') && (source != user))
 109                 {
 110                         delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
 111
 112                         if (!dl)
 113                         {
 114                                 dl = new delaylist;
 115                                 chan->Extend("norejoinusers", (char*)dl);
 116                         }
 117
 118                         log(DEBUG, "m_kicknorejoin.so: setting record for %s, %d second delay", user->nick, strtoint(chan->GetModeParameter('J')));
 119                         (*dl)[user] = time(NULL) + strtoint(chan->GetModeParameter('J'));
 120                 }
 121         }
 122
 123         virtual void OnChannelDelete(chanrec* chan)
 124         {
 125                 delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
 126
 127                 if (dl)
 128                 {
 129                         delete dl;
 130                         chan->Shrink("norejoinusers");
 131                 }
 132         }
 133
 134         virtual void OnCleanup(int target_type, void* item)
 135         {
 136                 if(target_type == TYPE_CHANNEL)
 137                         OnChannelDelete((chanrec*)item);
 138         }
 139
 140         virtual void Implements(char* List)
 141         {
 142                 List[I_OnCleanup] = List[I_On005Numeric] = List[I_OnExtendedMode] = List[I_OnChannelDelete] = List[I_OnUserPreJoin] = List[I_OnUserKick] = 1;
 143         }
 144
 145         virtual void On005Numeric(std::string &output)
 146         {
 147                 InsertMode(output, "J", 3);
 148         }
 149
 150         virtual ~ModuleKickNoRejoin()
 151         {
 152         }
 153
 154         virtual Version GetVersion()
 155         {
 156                 return Version(1, 0, 0, 0, VF_STATIC | VF_VENDOR);
 157         }
 158 };
 159
 160
 161 class ModuleKickNoRejoinFactory : public ModuleFactory
 162 {
 163  public:
 164         ModuleKickNoRejoinFactory()
 165         {
 166         }
 167
 168         ~ModuleKickNoRejoinFactory()
 169         {
 170         }
 171
 172         virtual Module * CreateModule(Server* Me)
 173         {
 174                 return new ModuleKickNoRejoin(Me);
 175         }
 176
 177 };
 178
 179
 180 extern "C" void * init_module( void )
 181 {
 182         return new ModuleKickNoRejoinFactory;
 183 }
 184