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(InspIRCd* Instance, Module* Creator) : ModeHandler(Creator, 'J', PARAM_SETONLY, MODETYPE_CHANNEL),
  35                 ext("norejoinusers", Creator) { }
  36
  37         ModePair ModeSet(User* source, User* dest, Channel* channel, const std::string &parameter)
  38         {
  39                 if (channel->IsModeSet('J'))
  40                         return std::make_pair(true, channel->GetModeParameter('J'));
  41                 else
  42                         return std::make_pair(false, parameter);
  43         }
  44
  45         ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
  46         {
  47                 if (!adding)
  48                 {
  49                         ext.unset(channel);
  50
  51                         if (!channel->IsModeSet('J'))
  52                         {
  53                                 return MODEACTION_DENY;
  54                         }
  55                         else
  56                         {
  57                                 channel->SetModeParam('J', "");
  58                                 return MODEACTION_ALLOW;
  59                         }
  60                 }
  61                 else if (atoi(parameter.c_str()) > 0)
  62                 {
  63                         if (!channel->IsModeSet('J'))
  64                         {
  65                                 parameter = ConvToStr(atoi(parameter.c_str()));
  66                                 channel->SetModeParam('J', parameter);
  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                                         if (parameter != "0")
  82                                         {
  83                                                 channel->SetModeParam('J', parameter);
  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         KickRejoin kr;
 104
 105 public:
 106
 107         ModuleKickNoRejoin(InspIRCd* Me)
 108                 : Module(Me), kr(Me, this)
 109         {
 110                 if (!ServerInstance->Modes->AddMode(&kr))
 111                         throw ModuleException("Could not add new modes!");
 112                 Extensible::Register(&kr.ext);
 113                 Implementation eventlist[] = { I_OnUserPreJoin, I_OnUserKick };
 114                 ServerInstance->Modules->Attach(eventlist, this, 2);
 115         }
 116
 117         ModResult OnUserPreJoin(User* user, Channel* chan, const char* cname, std::string &privs, const std::string &keygiven)
 118         {
 119                 if (chan)
 120                 {
 121                         delaylist* dl = kr.ext.get(chan);
 122                         if (dl)
 123                         {
 124                                 std::vector<User*> itemstoremove;
 125
 126                                 for (delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
 127                                 {
 128                                         if (iter->second > ServerInstance->Time())
 129                                         {
 130                                                 if (iter->first == user)
 131                                                 {
 132                                                         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());
 133                                                         return MOD_RES_DENY;
 134                                                 }
 135                                         }
 136                                         else
 137                                         {
 138                                                 // Expired record, remove.
 139                                                 itemstoremove.push_back(iter->first);
 140                                         }
 141                                 }
 142
 143                                 for (unsigned int i = 0; i < itemstoremove.size(); i++)
 144                                         dl->erase(itemstoremove[i]);
 145
 146                                 if (!dl->size())
 147                                         kr.ext.unset(chan);
 148                         }
 149                 }
 150                 return MOD_RES_PASSTHRU;
 151         }
 152
 153         void OnUserKick(User* source, Membership* memb, const std::string &reason, CUList& excepts)
 154         {
 155                 if (memb->chan->IsModeSet('J') && (source != memb->user))
 156                 {
 157                         delaylist* dl = kr.ext.get(memb->chan);
 158                         if (dl)
 159                         {
 160                                 dl = new delaylist;
 161                                 kr.ext.set(memb->chan, dl);
 162                         }
 163                         (*dl)[memb->user] = ServerInstance->Time() + strtoint(memb->chan->GetModeParameter('J'));
 164                 }
 165         }
 166
 167         ~ModuleKickNoRejoin()
 168         {
 169                 ServerInstance->Modes->DelMode(&kr);
 170         }
 171
 172         Version GetVersion()
 173         {
 174                 return Version("Channel mode J, kick-no-rejoin", VF_COMMON | VF_VENDOR);
 175         }
 176 };
 177
 178
 179 MODULE_INIT(ModuleKickNoRejoin)