src/modules/m_kicknorejoin.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15
  16 /* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
  17
  18 static inline int strtoint(const std::string &str)
  19 {
  20         std::istringstream ss(str);
  21         int result;
  22         ss >> result;
  23         return result;
  24 }
  25
  26 typedef std::map<User*, time_t> delaylist;
  27
  28 /** Handles channel mode +J
  29  */
  30 class KickRejoin : public ModeHandler
  31 {
  32  public:
  33         SimpleExtItem<delaylist> ext;
  34         KickRejoin(Module* Creator) : ModeHandler(Creator, "kicknorejoin", 'J', PARAM_SETONLY, MODETYPE_CHANNEL),
  35                 ext("norejoinusers", Creator) { }
  36
  37         ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
  38         {
  39                 if (!adding)
  40                 {
  41                         ext.unset(channel);
  42
  43                         if (!channel->IsModeSet('J'))
  44                         {
  45                                 return MODEACTION_DENY;
  46                         }
  47                         else
  48                         {
  49                                 channel->SetModeParam('J', "");
  50                                 return MODEACTION_ALLOW;
  51                         }
  52                 }
  53                 else if (atoi(parameter.c_str()) > 0)
  54                 {
  55                         if (!channel->IsModeSet('J'))
  56                         {
  57                                 parameter = ConvToStr(atoi(parameter.c_str()));
  58                                 channel->SetModeParam('J', parameter);
  59                                 return MODEACTION_ALLOW;
  60                         }
  61                         else
  62                         {
  63                                 std::string cur_param = channel->GetModeParameter('J');
  64                                 if (cur_param == parameter)
  65                                 {
  66                                         // mode params match, don't change mode
  67                                         return MODEACTION_DENY;
  68                                 }
  69                                 else
  70                                 {
  71                                         // new mode param, replace old with new
  72                                         parameter = ConvToStr(atoi(parameter.c_str()));
  73                                         if (parameter != "0")
  74                                         {
  75                                                 channel->SetModeParam('J', parameter);
  76                                                 return MODEACTION_ALLOW;
  77                                         }
  78                                         else
  79                                         {
  80                                                 /* Fix to jamie's fix, dont allow +J 0 on the new value! */
  81                                                 return MODEACTION_DENY;
  82                                         }
  83                                 }
  84                         }
  85                 }
  86                 else
  87                 {
  88                         return MODEACTION_DENY;
  89                 }
  90         }
  91 };
  92
  93 class ModuleKickNoRejoin : public Module
  94 {
  95         KickRejoin kr;
  96
  97 public:
  98
  99         ModuleKickNoRejoin()
 100                 : kr(this)
 101         {
 102                 if (!ServerInstance->Modes->AddMode(&kr))
 103                         throw ModuleException("Could not add new modes!");
 104                 ServerInstance->Extensions.Register(&kr.ext);
 105                 Implementation eventlist[] = { I_OnUserPreJoin, I_OnUserKick };
 106                 ServerInstance->Modules->Attach(eventlist, this, 2);
 107         }
 108
 109         ModResult OnUserPreJoin(User* user, Channel* chan, const char* cname, std::string &privs, const std::string &keygiven)
 110         {
 111                 if (chan)
 112                 {
 113                         delaylist* dl = kr.ext.get(chan);
 114                         if (dl)
 115                         {
 116                                 std::vector<User*> itemstoremove;
 117
 118                                 for (delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
 119                                 {
 120                                         if (iter->second > ServerInstance->Time())
 121                                         {
 122                                                 if (iter->first == user)
 123                                                 {
 124                                                         user->WriteNumeric(ERR_DELAYREJOIN, "%s %s :You must wait %s seconds after being kicked to rejoin (+J)", user->nick.c_str(), chan->name.c_str(), chan->GetModeParameter('J').c_str());
 125                                                         return MOD_RES_DENY;
 126                                                 }
 127                                         }
 128                                         else
 129                                         {
 130                                                 // Expired record, remove.
 131                                                 itemstoremove.push_back(iter->first);
 132                                         }
 133                                 }
 134
 135                                 for (unsigned int i = 0; i < itemstoremove.size(); i++)
 136                                         dl->erase(itemstoremove[i]);
 137
 138                                 if (!dl->size())
 139                                         kr.ext.unset(chan);
 140                         }
 141                 }
 142                 return MOD_RES_PASSTHRU;
 143         }
 144
 145         void OnUserKick(User* source, Membership* memb, const std::string &reason, CUList& excepts)
 146         {
 147                 if (memb->chan->IsModeSet('J') && (source != memb->user))
 148                 {
 149                         delaylist* dl = kr.ext.get(memb->chan);
 150                         if (dl)
 151                         {
 152                                 dl = new delaylist;
 153                                 kr.ext.set(memb->chan, dl);
 154                         }
 155                         (*dl)[memb->user] = ServerInstance->Time() + strtoint(memb->chan->GetModeParameter('J'));
 156                 }
 157         }
 158
 159         ~ModuleKickNoRejoin()
 160         {
 161         }
 162
 163         Version GetVersion()
 164         {
 165                 return Version("Channel mode J, kick-no-rejoin", VF_COMMON | VF_VENDOR);
 166         }
 167 };
 168
 169
 170 MODULE_INIT(ModuleKickNoRejoin)