-#include <time.h>
-#include <map>
-#include <vector>
-#include <sstream>
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-#include "helperfuncs.h"
-
-/* $ModDesc: Provides channel mode +J (delay rejoin after kick) */
-
-inline int strtoint(const std::string &str)
-{
- std::istringstream ss(str);
- int result;
- ss >> result;
- return result;
-}
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
+ *
+ * Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ * Copyright (C) 2008 Pippijn van Steenhoven <pip88nl@gmail.com>
+ * Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ * Copyright (C) 2006-2007 Robin Burchell <robin+git@viroteck.net>
+ * Copyright (C) 2006 John Brooks <john.brooks@dereferenced.net>
+ * Copyright (C) 2006 Craig Edwards <craigedwards@brainbox.cc>
+ * Copyright (C) 2006 Oliver Lupton <oliverlupton@gmail.com>
+ *
+ * This file is part of InspIRCd. InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
-typedef std::map<userrec*, time_t> delaylist;
-class ModuleKickNoRejoin : public Module
+#include "inspircd.h"
+
+class KickRejoinData
{
- Server *Srv;
-
-public:
-
- ModuleKickNoRejoin(Server* Me)
- : Module::Module(Me)
+ struct KickedUser
{
- Srv = Me;
-
- Srv->AddExtendedMode('J', MT_CHANNEL, false, 1, 0);
- }
-
- virtual int OnExtendedMode(userrec* user, void* target, char modechar, int type, bool mode_on, string_list ¶ms)
- {
- if ((modechar == 'J') && (type == MT_CHANNEL))
+ std::string uuid;
+ time_t expire;
+
+ KickedUser(User* user, unsigned int Delay)
+ : uuid(user->uuid)
+ , expire(ServerInstance->Time() + Delay)
{
- if(!mode_on)
- {
- // Taking the mode off, we need to clean up.
- chanrec* c = (chanrec*)target;
-
- delaylist* dl = (delaylist*)c->GetExt("norejoinusers");
-
- if(dl)
- {
- delete dl;
- c->Shrink("norejoinusers");
- }
- }
- return 1;
}
- return 0;
- }
+ };
+
+ typedef std::vector<KickedUser> KickedList;
+
+ mutable KickedList kicked;
+
+ public:
+ const unsigned int delay;
+
+ KickRejoinData(unsigned int Delay) : delay(Delay) { }
- virtual int OnUserPreJoin(userrec* user, chanrec* chan, const char* cname)
+ bool canjoin(LocalUser* user) const
{
- if(chan)
+ for (KickedList::iterator i = kicked.begin(); i != kicked.end(); )
{
- delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
- log(DEBUG, "m_kicknorejoin.so: tried to grab delay list");
-
- if(dl)
+ KickedUser& rec = *i;
+ if (rec.expire > ServerInstance->Time())
{
- log(DEBUG, "m_kicknorejoin.so: got delay list, iterating over it");
- std::vector<userrec*> itemstoremove;
-
- for(delaylist::iterator iter = dl->begin(); iter != dl->end(); iter++)
- {
- log(DEBUG, "m_kicknorejoin.so:\t[%s] => %d", iter->first->nick, iter->second);
- if(iter->second > time(NULL))
- {
- log(DEBUG, "m_kicknorejoin.so: still inside time slot");
- if(iter->first == user)
- {
- log(DEBUG, "m_kicknorejoin.so: and we have the right user");
- WriteServ(user->fd, "495 %s %s :You cannot rejoin this channel yet after being kicked (+J)", user->nick, chan->name);
- return 1;
- }
- }
- else
- {
- // Expired record, remove.
- log(DEBUG, "m_kicknorejoin.so: record expired");
- itemstoremove.push_back(iter->first);
- }
- }
-
- for(unsigned int i = 0; i < itemstoremove.size(); i++)
- dl->erase(itemstoremove[i]);
-
- if(!dl->size())
- {
- // Now it's empty..
- delete dl;
- chan->Shrink("norejoinusers");
- }
+ if (rec.uuid == user->uuid)
+ return false;
+ ++i;
}
- }
- return 0;
- }
-
- virtual void OnUserKick(userrec* source, userrec* user, chanrec* chan, std::string reason)
- {
- if(chan->IsCustomModeSet('J') && (source != user))
- {
- delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
-
- if(!dl)
+ else
{
- dl = new delaylist;
- chan->Extend("norejoinusers", (char*)dl);
+ // Expired record, remove.
+ stdalgo::vector::swaperase(kicked, i);
+ if (kicked.empty())
+ break;
}
-
- log(DEBUG, "m_kicknorejoin.so: setting record for %s, %d second delay", user->nick, strtoint(chan->GetModeParameter('J')));
- (*dl)[user] = time(NULL) + strtoint(chan->GetModeParameter('J'));
}
- }
-
- virtual void OnChannelDelete(chanrec* chan)
- {
- delaylist* dl = (delaylist*)chan->GetExt("norejoinusers");
-
- if(dl)
- {
- delete dl;
- chan->Shrink("norejoinusers");
- }
- }
-
- virtual void OnCleanup(int target_type, void* item)
- {
- if(target_type == TYPE_CHANNEL)
- OnChannelDelete((chanrec*)item);
+ return true;
}
- virtual void Implements(char* List)
+ void add(User* user)
{
- List[I_OnCleanup] = List[I_On005Numeric] = List[I_OnExtendedMode] = List[I_OnChannelDelete] = List[I_OnUserPreJoin] = List[I_OnUserKick] = 1;
+ // One user can be in the list multiple times if the user gets kicked, force joins
+ // (skipping OnUserPreJoin) and gets kicked again, but that's okay because canjoin()
+ // works correctly in this case as well
+ kicked.push_back(KickedUser(user, delay));
}
+};
- virtual void On005Numeric(std::string &output)
+/** Handles channel mode +J
+ */
+class KickRejoin : public ParamMode<KickRejoin, SimpleExtItem<KickRejoinData> >
+{
+ static const unsigned int max = 60;
+ public:
+ KickRejoin(Module* Creator)
+ : ParamMode<KickRejoin, SimpleExtItem<KickRejoinData> >(Creator, "kicknorejoin", 'J')
{
- InsertMode(output, "J", 3);
}
- virtual ~ModuleKickNoRejoin()
+ ModeAction OnSet(User* source, Channel* channel, std::string& parameter)
{
+ int v = ConvToInt(parameter);
+ if (v <= 0)
+ return MODEACTION_DENY;
+
+ if ((IS_LOCAL(source) && ((unsigned int)v > max)))
+ v = max;
+
+ ext.set(channel, new KickRejoinData(v));
+ return MODEACTION_ALLOW;
}
-
- virtual Version GetVersion()
+
+ void SerializeParam(Channel* chan, const KickRejoinData* krd, std::string& out)
{
- return Version(1, 0, 0, 0, VF_STATIC | VF_VENDOR);
+ out.append(ConvToStr(krd->delay));
}
};
-
-class ModuleKickNoRejoinFactory : public ModuleFactory
+class ModuleKickNoRejoin : public Module
{
- public:
- ModuleKickNoRejoinFactory()
+ KickRejoin kr;
+
+public:
+ ModuleKickNoRejoin()
+ : kr(this)
{
}
-
- ~ModuleKickNoRejoinFactory()
+
+ ModResult OnUserPreJoin(LocalUser* user, Channel* chan, const std::string& cname, std::string& privs, const std::string& keygiven) CXX11_OVERRIDE
{
+ if (chan)
+ {
+ const KickRejoinData* data = kr.ext.get(chan);
+ if ((data) && (!data->canjoin(user)))
+ {
+ user->WriteNumeric(ERR_DELAYREJOIN, "%s :You must wait %u seconds after being kicked to rejoin (+J)", chan->name.c_str(), data->delay);
+ return MOD_RES_DENY;
+ }
+ }
+ return MOD_RES_PASSTHRU;
}
-
- virtual Module * CreateModule(Server* Me)
+
+ void OnUserKick(User* source, Membership* memb, const std::string &reason, CUList& excepts) CXX11_OVERRIDE
{
- return new ModuleKickNoRejoin(Me);
- }
-
-};
+ if ((!IS_LOCAL(memb->user)) || (source == memb->user))
+ return;
+ KickRejoinData* data = kr.ext.get(memb->chan);
+ if (data)
+ {
+ data->add(memb->user);
+ }
+ }
-extern "C" void * init_module( void )
-{
- return new ModuleKickNoRejoinFactory;
-}
+ Version GetVersion() CXX11_OVERRIDE
+ {
+ return Version("Channel mode to delay rejoin after kick", VF_VENDOR);
+ }
+};
+MODULE_INIT(ModuleKickNoRejoin)