X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_callerid.cpp;h=45d3f0e5ccddca52eb2d78b0fc89bd7ddb68ab9e;hb=7107ec12d8640d35cfe3d5002db1bc1deb33625d;hp=28c7f80f305d5e3916af50f33f5183318b5f2c39;hpb=0377d937c5d1bf20fa0c29d4a41c7fd89502ab38;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/modules/m_callerid.cpp b/src/modules/m_callerid.cpp index 28c7f80f3..45d3f0e5c 100644 --- a/src/modules/m_callerid.cpp +++ b/src/modules/m_callerid.cpp @@ -32,7 +32,7 @@ class callerid_data : public classbase std::list wholistsme; callerid_data() : lastnotify(0) { } - callerid_data(const std::string& str, InspIRCd* ServerInstance) + callerid_data(const std::string& str) { irc::commasepstream s(str); std::string tok; @@ -56,85 +56,88 @@ class callerid_data : public classbase } } - std::string ToString(Module* proto) const + std::string ToString(SerializeFormat format) const { std::ostringstream oss; oss << lastnotify; for (std::set::const_iterator i = accepting.begin(); i != accepting.end(); ++i) { + User* u = *i; // Encode UIDs. - oss << "," << proto->ProtoTranslate(*i); + oss << "," << (format == FORMAT_USER ? u->nick : u->uuid); } oss << std::ends; return oss.str(); } }; -/** Retrieves the callerid information for a given user record - * @param who The user to retrieve callerid information for - * @param extend true to add callerid information if it doesn't already exist, false to return NULL if it doesn't exist - * @return NULL if extend is false and it doesn't exist, a callerid_data instance otherwise. - */ -callerid_data* GetData(User* who, bool extend = true) +struct CallerIDExtInfo : public ExtensionItem { - callerid_data* dat; - if (who->GetExt("callerid_data", dat)) - return dat; - else + CallerIDExtInfo(Module* parent) + : ExtensionItem("callerid_data", parent) { - if (extend) - { - dat = new callerid_data; - who->Extend("callerid_data", dat); - return dat; - } - else - return NULL; } -} -void RemoveData(User* who) -{ - callerid_data* dat; - who->GetExt("callerid_data", dat); + std::string serialize(SerializeFormat format, const Extensible* container, void* item) + { + callerid_data* dat = static_cast(item); + return dat->ToString(format); + } - if (!dat) - return; + void unserialize(SerializeFormat format, Extensible* container, const std::string& value) + { + callerid_data* dat = new callerid_data(value); + set_raw(container, dat); + } - // We need to walk the list of users on our accept list, and remove ourselves from their wholistsme. - for (std::set::iterator it = dat->accepting.begin(); it != dat->accepting.end(); it++) + callerid_data* get(User* user, bool create) { - callerid_data *targ = GetData(*it, false); + callerid_data* dat = static_cast(get_raw(user)); + if (!dat) + { + dat = new callerid_data; + set_raw(user, dat); + } + return dat; + } - if (!targ) - continue; // shouldn't happen, but oh well. + void free(void* item) + { + callerid_data* dat = static_cast(item); - for (std::list::iterator it2 = targ->wholistsme.begin(); it2 != targ->wholistsme.end(); it2++) + // We need to walk the list of users on our accept list, and remove ourselves from their wholistsme. + for (std::set::iterator it = dat->accepting.begin(); it != dat->accepting.end(); it++) { - if (*it2 == dat) + callerid_data *targ = this->get(*it, false); + + if (!targ) + continue; // shouldn't happen, but oh well. + + for (std::list::iterator it2 = targ->wholistsme.begin(); it2 != targ->wholistsme.end(); it2++) { - targ->wholistsme.erase(it2); - break; + if (*it2 == dat) + { + targ->wholistsme.erase(it2); + break; + } } } } - - who->Shrink("callerid_data"); - delete dat; -} - +}; class User_g : public SimpleUserModeHandler { public: - User_g(InspIRCd* Instance, Module* Creator) : SimpleUserModeHandler(Instance, Creator, 'g') { } + User_g(Module* Creator) : SimpleUserModeHandler(Creator, 'g') { } }; class CommandAccept : public Command { public: + CallerIDExtInfo extInfo; unsigned int maxaccepts; - CommandAccept(InspIRCd* Instance, Module* Creator) : Command(Instance, Creator, "ACCEPT", 0, 1) + CommandAccept(Module* Creator) : Command(Creator, "ACCEPT", 1), + extInfo(Creator) { syntax = "{[+|-]}|*}"; TRANSLATE2(TR_CUSTOM, TR_END); @@ -196,7 +199,7 @@ public: { if (IS_LOCAL(user)) ListAccept(user); - return CMD_LOCALONLY; + return CMD_SUCCESS; } else if (tok[0] == '-') { @@ -221,7 +224,7 @@ public: void ListAccept(User* user) { - callerid_data* dat = GetData(user, false); + callerid_data* dat = extInfo.get(user, false); if (dat) { for (std::set::iterator i = dat->accepting.begin(); i != dat->accepting.end(); ++i) @@ -233,7 +236,7 @@ public: bool AddAccept(User* user, User* whotoadd, bool quiet) { // Add this user to my accept list first, so look me up.. - callerid_data* dat = GetData(user, true); + callerid_data* dat = extInfo.get(user, true); if (dat->accepting.size() >= maxaccepts) { if (!quiet) @@ -250,7 +253,7 @@ public: } // Now, look them up, and add me to their list - callerid_data *targ = GetData(whotoadd, true); + callerid_data *targ = extInfo.get(whotoadd, true); targ->wholistsme.push_back(dat); user->WriteServ("NOTICE %s :%s is now on your accept list", user->nick.c_str(), whotoadd->nick.c_str()); @@ -260,7 +263,7 @@ public: bool RemoveAccept(User* user, User* whotoremove, bool quiet) { // Remove them from my list, so look up my list.. - callerid_data* dat = GetData(user, false); + callerid_data* dat = extInfo.get(user, false); if (!dat) { if (!quiet) @@ -280,7 +283,7 @@ public: dat->accepting.erase(i); // Look up their list to remove me. - callerid_data *dat2 = GetData(whotoremove, false); + callerid_data *dat2 = extInfo.get(whotoremove, false); if (!dat2) { // How the fuck is this possible. @@ -305,7 +308,7 @@ public: class ModuleCallerID : public Module { private: - CommandAccept mycommand; + CommandAccept cmd; User_g myumode; // Configuration variables: @@ -319,7 +322,7 @@ private: void RemoveFromAllAccepts(User* who) { // First, find the list of people who have me on accept - callerid_data *userdata = GetData(who, false); + callerid_data *userdata = cmd.extInfo.get(who, false); if (!userdata) return; @@ -339,17 +342,18 @@ private: } public: - ModuleCallerID(InspIRCd* Me) : Module(Me), mycommand(Me, this), myumode(Me, this) + ModuleCallerID() : cmd(this), myumode(this) { OnRehash(NULL); if (!ServerInstance->Modes->AddMode(&myumode)) throw ModuleException("Could not add usermode +g"); - ServerInstance->AddCommand(&mycommand); + ServerInstance->AddCommand(&cmd); + Extensible::Register(&cmd.extInfo); - Implementation eventlist[] = { I_OnRehash, I_OnUserPreNick, I_OnUserQuit, I_On005Numeric, I_OnUserPreNotice, I_OnUserPreMessage, I_OnCleanup }; - ServerInstance->Modules->Attach(eventlist, this, 7); + Implementation eventlist[] = { I_OnRehash, I_OnUserPreNick, I_OnUserQuit, I_On005Numeric, I_OnUserPreNotice, I_OnUserPreMessage }; + ServerInstance->Modules->Attach(eventlist, this, 6); } virtual ~ModuleCallerID() @@ -359,7 +363,7 @@ public: virtual Version GetVersion() { - return Version("$Id$", VF_COMMON | VF_VENDOR, API_VERSION); + return Version("Implementation of callerid (umode +g & /accept, ala hybrid etc)", VF_COMMON | VF_VENDOR, API_VERSION); } virtual void On005Numeric(std::string& output) @@ -375,7 +379,7 @@ public: if (operoverride && IS_OPER(user)) return MOD_RES_PASSTHRU; - callerid_data* dat = GetData(dest, true); + callerid_data* dat = cmd.extInfo.get(dest, true); std::set::iterator i = dat->accepting.find(user); if (i == dat->accepting.end()) @@ -411,53 +415,22 @@ public: return MOD_RES_PASSTHRU; } - virtual void OnCleanup(int type, void* item) - { - if (type != TYPE_USER) - return; - - User* u = (User*)item; - /* Cleanup only happens on unload (before dtor), so keep this O(n) instead of O(n^2) which deferring to OnUserQuit would do. */ - RemoveData(u); - } - - virtual void OnSyncUser(User* user, Module* proto, void* opaque) - { - callerid_data* dat = GetData(user, false); - if (dat) - { - std::string str = dat->ToString(proto); - proto->ProtoSendMetaData(opaque, user, "callerid_data", str); - } - } - - virtual void OnDecodeMetaData(Extensible* target, const std::string& extname, const std::string& extdata) - { - User* u = dynamic_cast(target); - if (u && extname == "callerid_data") - { - callerid_data* dat = new callerid_data(extdata, ServerInstance); - u->Extend("callerid_data", dat); - } - } - - virtual ModResult OnUserPreNick(User* user, const std::string& newnick) + ModResult OnUserPreNick(User* user, const std::string& newnick) { if (!tracknick) RemoveFromAllAccepts(user); return MOD_RES_PASSTHRU; } - virtual void OnUserQuit(User* user, const std::string& message, const std::string& oper_message) + void OnUserQuit(User* user, const std::string& message, const std::string& oper_message) { RemoveFromAllAccepts(user); - RemoveData(user); } virtual void OnRehash(User* user) { - ConfigReader Conf(ServerInstance); - mycommand.maxaccepts = Conf.ReadInteger("callerid", "maxaccepts", "16", 0, true); + ConfigReader Conf; + cmd.maxaccepts = Conf.ReadInteger("callerid", "maxaccepts", "16", 0, true); operoverride = Conf.ReadFlag("callerid", "operoverride", "0", 0); tracknick = Conf.ReadFlag("callerid", "tracknick", "0", 0); notify_cooldown = Conf.ReadInteger("callerid", "cooldown", "60", 0, true);