X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_shun.cpp;h=c013776ed217029838b9018601908a1eb2d565bc;hb=a23a840749db9646884907ae28588b1962dc2449;hp=b3b9b2010bbfe19c81d75407195e7793dc0b3880;hpb=480407f4ff12a014efdfa0eaa01172787409850c;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/modules/m_shun.cpp b/src/modules/m_shun.cpp index b3b9b2010..c013776ed 100644 --- a/src/modules/m_shun.cpp +++ b/src/modules/m_shun.cpp @@ -1,148 +1,139 @@ -#include -#include -#include -#include #include "inspircd.h" -#include "modules.h" -#include "hashcomp.h" -#include "configreader.h" +#include "xline.h" -/* $ModDesc: Gives the /shun command, which stops a user executing all commands except PING and PONG. */ -/* $ModAuthor: Om */ -/* $ModAuthorMail: om@inspircd.org */ +/* $ModDesc: Provides the /shun command, which stops a user executing all commands except PING and PONG. */ -class Shun +class Shun : public XLine { public: - std::string banmask; - std::string set_by; - time_t set_on; - long length; - std::string reason; + std::string matchtext; - Shun() { } + Shun(InspIRCd* Instance, time_t s_time, long d, const char* src, const char* re, const char *shunmask) : XLine(Instance, s_time, d, src, re, "SHUN") + { + this->matchtext = shunmask; + } - Shun(std::string bm, std::string sb, time_t so, long ln, std::string rs) : banmask(bm), set_by(sb), set_on(so), length(ln), reason(rs) { } - - bool operator<(const Shun &ban2) const + ~Shun() { - return ((this->set_on + this->length) < (ban2.set_on + ban2.length)); } -}; -typedef std::vector shunlist; + bool Matches(User *u) + { + if (InspIRCd::Match(u->GetFullHost(), matchtext) || InspIRCd::Match(u->GetFullRealHost(), matchtext)) + return true; -class ModuleShunBase -{ - public: - /* shuns is declared here, as our type is right above. Don't try move it. */ - shunlist shuns; - - InspIRCd* Srv; - - std::string EncodeShun(const Shun &shun) + return false; + } + + bool Matches(const std::string &s) + { + if (matchtext == s) + return true; + return false; + } + + void Apply(User *u) { - std::ostringstream stream; - stream << shun.banmask << " " << shun.set_by << " " << shun.set_on << " " << shun.length << " " << shun.reason; - return stream.str(); + if (!u->GetExt("shunned")) + u->Extend("shunned"); } - Shun DecodeShun(const std::string &data) + + void DisplayExpiry() { - Shun res; - std::istringstream stream(data); - stream >> res.banmask; - stream >> res.set_by; - stream >> res.set_on; - stream >> res.length; - res.reason = stream.str(); - - return res; + ServerInstance->SNO->WriteToSnoMask('x',"Expiring timed shun %s (set by %s %ld seconds ago)", this->matchtext.c_str(), this->source, this->duration); } - void ExpireBans() + const char* Displayable() { - while(shuns.size() && shuns.begin()->length && ((shuns.begin()->set_on + shuns.begin()->length) <= Srv->Time())) - { - shunlist::iterator iter = shuns.begin(); - - Srv->SNO->WriteToSnoMask('X', "*** %ld second shun on '%s' (%s) set by %s %ld seconds ago expired", iter->length, iter->banmask.c_str(), iter->reason.c_str(), iter->set_by.c_str(), Srv->Time() - iter->set_on); - shuns.erase(iter); - } + return matchtext.c_str(); } }; +/** An XLineFactory specialized to generate shun pointers + */ +class ShunFactory : public XLineFactory +{ + public: + ShunFactory(InspIRCd* Instance) : XLineFactory(Instance, "SHUN") { } + + /** Generate a shun + */ + XLine* Generate(time_t set_time, long duration, const char* source, const char* reason, const char* xline_specific_mask) + { + return new Shun(ServerInstance, set_time, duration, source, reason, xline_specific_mask); + } +}; + +//typedef std::vector shunlist; + class cmd_shun : public Command { private: InspIRCd *Srv; - ModuleShunBase* base; public: - cmd_shun(InspIRCd* Me, ModuleShunBase* b) - : Command(Me, "SHUN", "o", 1), Srv(Me), base(b) + cmd_shun(InspIRCd* Me) : Command(Me, "SHUN", "o", 1, 3), Srv(Me) { this->source = "m_shun.so"; + this->syntax = " [] :"; } - CmdResult Handle(const char* const*parameters, int pcnt, User *user) + CmdResult Handle(const std::vector& parameters, User *user) { /* syntax: SHUN nick!user@host time :reason goes here */ /* 'time' is a human-readable timestring, like 2d3h2s. */ - - base->ExpireBans(); - if(pcnt == 1) + if (parameters.size() == 1) { - /* form: SHUN mask removes a SHUN */ - for(shunlist::iterator iter = base->shuns.begin(); iter != base->shuns.end(); iter++) + if (ServerInstance->XLines->DelLine(parameters[0].c_str(), "SHUN", user)) { - if(parameters[0] == iter->banmask) - { - Srv->SNO->WriteToSnoMask('X', "*** %s removed shun '%s', set %ld seconds ago with reason '%s'", user->nick, iter->banmask.c_str(), Srv->Time() - iter->set_on, iter->reason.c_str()); - base->shuns.erase(iter); - return CMD_SUCCESS; - } + ServerInstance->SNO->WriteToSnoMask('x',"%s Removed shun on %s.",user->nick.c_str(),parameters[0].c_str()); + } + else + { + // XXX todo implement stats + user->WriteServ("NOTICE %s :*** Shun %s not found in list, try /stats s.",user->nick.c_str(),parameters[0].c_str()); } - user->WriteServ("NOTICE %s :*** The mask %s is not currently shunned, try /stats s", user->nick, parameters[0]); - return CMD_FAILURE; + return CMD_SUCCESS; } - else if (pcnt >= 2) + else if (parameters.size() >= 2) { - /* full form to add a shun */ - if(Srv->IsValidMask(parameters[0])) + // Adding - XXX todo make this respect tag perhaps.. + long duration = ServerInstance->Duration(parameters[1]); + Shun *r = NULL; + + try + { + r = new Shun(ServerInstance, ServerInstance->Time(), duration, user->nick.c_str(), parameters[2].c_str(), parameters[0].c_str()); + } + catch (...) + { + ; // Do nothing. If we get here, the regex was fucked up, and they already got told it fucked up. + } + + if (r) { - // parameters[0] = Foamy!*@* - // parameters[1] = 1h3m2s - // parameters[2] = Tortoise abuser - for(shunlist::iterator iter = base->shuns.begin(); iter != base->shuns.end(); iter++) + if (ServerInstance->XLines->AddLine(r, user)) { - if (parameters[0] == iter->banmask) + if (!duration) + { + ServerInstance->SNO->WriteToSnoMask('x',"%s added permanent shun for %s: %s", user->nick.c_str(), parameters[0].c_str(), parameters[2].c_str()); + } + else { - user->WriteServ("NOTICE %s :*** Shun on %s already exists", user->nick, parameters[0]); - return CMD_FAILURE; + time_t c_requires_crap = duration + ServerInstance->Time(); + ServerInstance->SNO->WriteToSnoMask('x', "%s added timed shun for %s, expires on %s: %s", user->nick.c_str(), parameters[0].c_str(), ServerInstance->TimeString(c_requires_crap).c_str(), parameters[2].c_str()); } + + ServerInstance->XLines->ApplyLines(); } - - long length = Srv->Duration(parameters[1]); - - std::string reason = (pcnt > 2) ? parameters[2] : "No reason supplied"; - - base->shuns.push_back(Shun(parameters[0], user->nick, Srv->Time(), length, reason)); - - std::sort(base->shuns.begin(), base->shuns.end()); - - if(length > 0) - Srv->SNO->WriteToSnoMask('X', "*** %s added %ld second shun on '%s' (%s)", user->nick, length, parameters[0], reason.c_str()); else - Srv->SNO->WriteToSnoMask('X', "*** %s added permanent shun on '%s' (%s)", user->nick, parameters[0], reason.c_str()); - - return CMD_SUCCESS; - } - else - { - user->WriteServ("NOTICE %s :*** Invalid ban mask (%s)", user->nick, parameters[0]); + { + delete r; + user->WriteServ("NOTICE %s :*** Shun for %s already exists", user->nick.c_str(), parameters[0].c_str()); + } } } @@ -150,79 +141,116 @@ class cmd_shun : public Command } }; -class ModuleShun : public Module, public ModuleShunBase +class ModuleShun : public Module { cmd_shun* mycommand; + ShunFactory *f; + std::map ShunEnabledCommands; + bool NotifyOfShun; public: - ModuleShun(InspIRCd* Me) - : Module::Module(Me) + ModuleShun(InspIRCd* Me) : Module(Me) { - this->Srv = Me; - mycommand = new cmd_shun(Srv, this); - Srv->AddCommand(mycommand); + f = new ShunFactory(ServerInstance); + ServerInstance->XLines->RegisterFactory(f); + + mycommand = new cmd_shun(ServerInstance); + ServerInstance->AddCommand(mycommand); + + Implementation eventlist[] = { I_OnStats, I_OnPreCommand, I_OnUserConnect, I_OnRehash }; + ServerInstance->Modules->Attach(eventlist, this, 4); + OnRehash(NULL, ""); } - void Implements(char* List) + virtual ~ModuleShun() { - List[I_OnPreCommand] = List[I_OnSyncOtherMetaData] = List[I_OnDecodeMetaData] = List[I_OnStats] = 1; + ServerInstance->XLines->DelAll("SHUN"); + ServerInstance->XLines->UnregisterFactory(f); } - + virtual int OnStats(char symbol, User* user, string_list& out) { - ExpireBans(); - - if(symbol == 's') - { - for(shunlist::iterator iter = shuns.begin(); iter != shuns.end(); iter++) - { - std::ostringstream format; - format << Srv->Config->ServerName << " 223 " << user->nick << " :" << iter->banmask << " " << iter->set_on << " " << iter->length << " " << iter->set_by << " " << iter->reason; - out.push_back(format.str()); - } - } - - return 0; + if (symbol != 'S') + return 0; + + ServerInstance->XLines->InvokeStats("SHUN", 223, user, out); + return 1; } - virtual int OnPreCommand(const std::string &command, const char* const*parameters, int pcnt, User* user, bool validated, const std::string &original_line) + virtual void OnRehash(User* user, const std::string ¶meter) { - if((command != "PONG") && (command != "PING")) + ConfigReader MyConf(ServerInstance); + std::string cmds = MyConf.ReadValue("shun", "enabledcommands", 0); + + if (cmds.empty()) + cmds = "PING PONG QUIT"; + + ShunEnabledCommands.clear(); + NotifyOfShun = true; + + std::stringstream dcmds(cmds); + std::string thiscmd; + + while (dcmds >> thiscmd) { - ExpireBans(); - - for(shunlist::iterator iter = shuns.begin(); iter != shuns.end(); iter++) - if(Srv->MatchText(user->GetFullHost(), iter->banmask) || Srv->MatchText(user->GetFullRealHost(), iter->banmask)) - return 1; + ShunEnabledCommands[thiscmd] = true; } - - return 0; + + NotifyOfShun = MyConf.ReadFlag("shun", "notifyuser", "yes", 0); } - virtual void OnSyncOtherMetaData(Module* proto, void* opaque, bool displayable) + virtual void OnUserConnect(User* user) { - for(shunlist::iterator iter = shuns.begin(); iter != shuns.end(); iter++) + if (!IS_LOCAL(user)) + return; + + // Apply lines on user connect + XLine *rl = ServerInstance->XLines->MatchesLine("SHUN", user); + + if (rl) { - proto->ProtoSendMetaData(opaque, TYPE_OTHER, NULL, "shun", EncodeShun(*iter)); + // Bang. :P + rl->Apply(user); } } - virtual void OnDecodeMetaData(int target_type, void* target, const std::string &extname, const std::string &extdata) + virtual int OnPreCommand(std::string &command, std::vector& parameters, User* user, bool validated, const std::string &original_line) { - if((target_type == TYPE_OTHER) && (extname == "shun")) + if (validated || !user->GetExt("shunned")) + return 0; + + if (!ServerInstance->XLines->MatchesLine("SHUN", user)) { - shuns.push_back(DecodeShun(extdata)); - std::sort(shuns.begin(), shuns.end()); + /* The shun previously set on this user has expired or been removed */ + user->Shrink("shunned"); + return 0; } - } - virtual ~ModuleShun() - { + std::map::iterator i = ShunEnabledCommands.find(command); + + if (i == ShunEnabledCommands.end()) + { + user->WriteServ("NOTICE %s :*** Command %s not processed, as you have been blocked from issuing commands (SHUN)", user->nick.c_str(), command.c_str()); + return 1; + } + + if (command == "QUIT") + { + /* Allow QUIT but dont show any quit message */ + parameters.clear(); + } + else if (command == "PART") + { + /* same for PART */ + parameters.clear(); + } + + return 1; } virtual Version GetVersion() { - return Version(1,0,0,0,0,API_VERSION); + return Version("$Id$",VF_VENDOR|VF_COMMON,API_VERSION); } };