]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_callerid.cpp
Update all wiki links to point to the new wiki. This was done automatically with...
[user/henk/code/inspircd.git] / src / modules / m_callerid.cpp
index 43d546f92a9c738f32ff8b959fe18b3663f89123..3592767682b49a481e4d2de80980deb77954d2fd 100644 (file)
@@ -1,9 +1,20 @@
-#include "inspircd.h"
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
+ *
+ *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
+ *
+ * This program is free but copyrighted software; see
+ *            the file COPYING for details.
+ *
+ * ---------------------------------------------------
+ */
 
+#include "inspircd.h"
 #include <set>
+#include <sstream>
+#include <algorithm>
 
 /* $ModDesc: Implementation of callerid (umode +g & /accept, ala hybrid etc) */
 
@@ -12,28 +23,64 @@ class callerid_data : public classbase
  public:
        time_t lastnotify;
        std::set<User*> accepting;
+
+       callerid_data() : lastnotify(0) { }
+       callerid_data(const std::string& str, InspIRCd* ServerInstance)
+       {
+               irc::commasepstream s(str);
+               std::string tok;
+               if (s.GetToken(tok))
+               {
+                       lastnotify = ConvToInt(tok);
+               }
+               while (s.GetToken(tok))
+               {
+                       if (tok.empty())
+                       {
+                               continue;
+                       }
+                       User* u = ServerInstance->FindUUID(tok);
+                       if (!u)
+                       {
+                               u = ServerInstance->FindNick(tok);
+                       }
+                       if (!u)
+                       {
+                               continue;
+                       }
+                       accepting.insert(u);
+               }
+       }
+
+       std::string ToString(bool displayable) const
+       {
+               std::ostringstream oss;
+               oss << lastnotify;
+               for (std::set<User*>::const_iterator i = accepting.begin(); i != accepting.end(); ++i)
+               {
+                       // Encode UIDs.
+                       oss << "," << (displayable ? (*i)->nick : (*i)->uuid);
+               }
+               oss << std::ends;
+               return oss.str();
+       }
 };
 
 callerid_data* GetData(User* who, bool extend = true)
 {
        callerid_data* dat;
        if (who->GetExt("callerid_data", dat))
-       {
                return dat;
-       }
        else
        {
                if (extend)
                {
                        dat = new callerid_data;
-                       dat->lastnotify = 0; // Can't init in struct.
                        who->Extend("callerid_data", dat);
                        return dat;
                }
                else
-               {
                        return NULL;
-               }
        }
 }
 
@@ -41,7 +88,10 @@ void RemoveData(User* who)
 {
        callerid_data* dat;
        who->GetExt("callerid_data", dat);
-       if (!dat) return;
+
+       if (!dat)
+               return;
+
        who->Shrink("callerid_data");
        delete dat;
 }
@@ -51,18 +101,21 @@ void RemoveFromAllAccepts(InspIRCd* ServerInstance, User* who)
        for (user_hash::iterator i = ServerInstance->Users->clientlist->begin(); i != ServerInstance->Users->clientlist->end(); ++i)
        {
                callerid_data* dat = GetData(i->second, false);
-               if (!dat) continue;
-               std::set<User*>& accepting = dat->accepting;
-               std::set<User*>::iterator iter = accepting.find(who);
-               if (iter == accepting.end()) continue;
-               accepting.erase(iter);
+
+               if (!dat)
+                       continue;
+
+               std::set<User*>::iterator iter = dat->accepting.find(who);
+
+               if (iter == dat->accepting.end())
+                       continue;
+
+               dat->accepting.erase(iter);
        }
 }
 
 class User_g : public SimpleUserModeHandler
 {
-private:
-
 public:
        User_g(InspIRCd* Instance) : SimpleUserModeHandler(Instance, 'g') { }
 };
@@ -76,61 +129,86 @@ public:
        {
                source = "m_callerid.so";
                syntax = "{[+|-]<nicks>}|*}";
+               TRANSLATE2(TR_CUSTOM, TR_END);
        }
 
-       /* Will take any number of nicks, which can be seperated by spaces, commas, or a mix.
+       virtual void EncodeParameter(std::string& parameter, int index)
+       {
+               if (index != 0)
+                       return;
+               std::string out = "";
+               irc::commasepstream nicks(parameter);
+               std::string tok;
+               while (nicks.GetToken(tok))
+               {
+                       if (tok == "*")
+                       {
+                               continue; // Drop list requests, since remote servers ignore them anyway.
+                       }
+                       if (!out.empty())
+                               out.append(",");
+                       bool dash = false;
+                       if (tok[0] == '-')
+                       {
+                               dash = true;
+                               tok.erase(0, 1); // Remove the dash.
+                       }
+                       User* u = ServerInstance->FindNick(tok);
+                       if (u)
+                       {
+                               if (dash)
+                                       out.append("-");
+                               out.append(u->uuid);
+                       }
+                       else
+                       {
+                               if (dash)
+                                       out.append("-");
+                               out.append(tok);
+                       }
+               }
+               parameter = out;
+       }
+
+       /** Will take any number of nicks (up to MaxTargets), which can be seperated by commas.
         * - in front of any nick removes, and an * lists. This effectively means you can do:
-        * /accept nick1,nick2,nick3 *
+        * /accept nick1,nick2,nick3,*
         * to add 3 nicks and then show your list
         */
-
        CmdResult Handle(const std::vector<std::string> &parameters, User* user)
        {
-               if (parameters.size() < 1)
-               {
-                       /* Command stuff should've dealt with this already */
-                       return CMD_FAILURE;
-               }
+               if (ServerInstance->Parser->LoopCall(user, this, parameters, 0))
+                       return CMD_SUCCESS;
                /* Even if callerid mode is not set, we let them manage their ACCEPT list so that if they go +g they can
                 * have a list already setup. */
-               bool atleastonechange = false;
-               for (int i = 0; i < (int)parameters.size(); ++i)
+
+               std::string tok = parameters[0];
+
+               if (tok == "*")
                {
-                       const char* arg = parameters[i].c_str();
-                       irc::commasepstream css(arg);
-                       std::string tok;
-                       while (css.GetToken(tok))
+                       if (IS_LOCAL(user))
+                               ListAccept(user);
+                       return CMD_LOCALONLY;
+               }
+               else if (tok[0] == '-')
+               {
+                       User* whotoremove = ServerInstance->FindNick(tok.substr(1));
+                       if (whotoremove)
+                               return (RemoveAccept(user, whotoremove, false) ? CMD_SUCCESS : CMD_FAILURE);
+                       else
+                               return CMD_FAILURE;
+               }
+               else
+               {
+                       User* whotoadd = ServerInstance->FindNick(tok[0] == '+' ? tok.substr(1) : tok);
+                       if (whotoadd)
+                               return (AddAccept(user, whotoadd, false) ? CMD_SUCCESS : CMD_FAILURE);
+                       else
                        {
-                               if (tok.length() < 1)
-                                       continue;
-                               if (tok == "*")
-                               {
-                                       if (IS_LOCAL(user)) continue;
-                                       ListAccept(user);
-                               }
-                               else if (tok[0] == '-')
-                               {
-                                       User* whotoremove = ServerInstance->FindNick(tok.substr(1));
-                                       if (whotoremove)
-                                       {
-                                               atleastonechange = RemoveAccept(user, whotoremove, false) || atleastonechange;
-                                       }
-                               }
-                               else
-                               {
-                                       User* whotoadd = ServerInstance->FindNick(tok[0] == '+' ? tok.substr(1) : tok);
-                                       if (whotoadd)
-                                       {
-                                               atleastonechange = AddAccept(user, whotoadd, false) || atleastonechange;
-                                       }
-                                       else
-                                       {
-                                               user->WriteNumeric(401, "%s %s :No such nick/channel", user->nick, tok.c_str());
-                                       }
-                               }
+                               user->WriteNumeric(401, "%s %s :No such nick/channel", user->nick.c_str(), tok.c_str());
+                               return CMD_FAILURE;
                        }
                }
-               return atleastonechange ? CMD_FAILURE : CMD_SUCCESS;
        }
 
        void ListAccept(User* user)
@@ -139,27 +217,30 @@ public:
                if (dat)
                {
                        for (std::set<User*>::iterator i = dat->accepting.begin(); i != dat->accepting.end(); ++i)
-                       {
-                               user->WriteNumeric(281, "%s %s", user->nick, (*i)->nick);
-                       }
+                               user->WriteNumeric(281, "%s %s", user->nick.c_str(), (*i)->nick.c_str());
                }
-               user->WriteNumeric(282, "%s :End of ACCEPT list", user->nick);
+               user->WriteNumeric(282, "%s :End of ACCEPT list", user->nick.c_str());
        }
 
        bool AddAccept(User* user, User* whotoadd, bool quiet)
        {
                callerid_data* dat = GetData(user, true);
-               std::set<User*>& accepting = dat->accepting;
-               if (accepting.size() >= maxaccepts)
+               if (dat->accepting.size() >= maxaccepts)
                {
-                       if (!quiet) user->WriteNumeric(456, "%s :Accept list is full (limit is %d)", user->nick, maxaccepts);
+                       if (!quiet)
+                               user->WriteNumeric(456, "%s :Accept list is full (limit is %d)", user->nick.c_str(), maxaccepts);
+
                        return false;
                }
-               if (!accepting.insert(whotoadd).second)
+               if (!dat->accepting.insert(whotoadd).second)
                {
-                       if (!quiet) user->WriteNumeric(457, "%s %s :is already on your accept list", user->nick, whotoadd->nick);
+                       if (!quiet)
+                               user->WriteNumeric(457, "%s %s :is already on your accept list", user->nick.c_str(), whotoadd->nick.c_str());
+
                        return false;
                }
+
+               user->WriteServ("NOTICE %s :%s is now on your accept list", user->nick.c_str(), whotoadd->nick.c_str());
                return true;
        }
 
@@ -168,17 +249,22 @@ public:
                callerid_data* dat = GetData(user, false);
                if (!dat)
                {
-                       if (!quiet) user->WriteNumeric(458, "%s %s :is not on your accept list", user->nick, whotoremove->nick);
+                       if (!quiet)
+                               user->WriteNumeric(458, "%s %s :is not on your accept list", user->nick.c_str(), whotoremove->nick.c_str());
+
                        return false;
                }
-               std::set<User*>& accepting = dat->accepting;
-               std::set<User*>::iterator i = accepting.find(whotoremove);
-               if (i == accepting.end())
+               std::set<User*>::iterator i = dat->accepting.find(whotoremove);
+               if (i == dat->accepting.end())
                {
-                       if (!quiet) user->WriteNumeric(458, "%s %s :is not on your accept list", user->nick, whotoremove->nick);
+                       if (!quiet)
+                               user->WriteNumeric(458, "%s %s :is not on your accept list", user->nick.c_str(), whotoremove->nick.c_str());
+
                        return false;
                }
-               accepting.erase(i);
+
+               user->WriteServ("NOTICE %s :%s is no longer on your accept list", user->nick.c_str(), whotoremove->nick.c_str());
+               dat->accepting.erase(i);
                return true;
        }
 };
@@ -201,147 +287,143 @@ public:
                OnRehash(NULL, "");
                mycommand = new CommandAccept(ServerInstance, maxaccepts);
                myumode = new User_g(ServerInstance);
+
+               if (!ServerInstance->Modes->AddMode(myumode))
+               {
+                       delete mycommand;
+                       delete myumode;
+                       throw ModuleException("Could not add usermode +g");
+               }
                try
                {
                        ServerInstance->AddCommand(mycommand);
                }
                catch (const ModuleException& e)
-               {
-                       delete mycommand;
-                       throw ModuleException("Could not add command!");
-               }
-               if (!ServerInstance->Modes->AddMode(myumode))
                {
                        delete mycommand;
                        delete myumode;
-                       throw ModuleException("Could not add usermode +g");
+                       throw ModuleException("Could not add command!");
                }
+
                Implementation eventlist[] = { I_OnRehash, I_OnUserPreNick, I_OnUserQuit, I_On005Numeric, I_OnUserPreNotice, I_OnUserPreMessage, I_OnCleanup };
                ServerInstance->Modules->Attach(eventlist, this, 7);
        }
 
-       ~ModuleCallerID()
+       virtual ~ModuleCallerID()
        {
                ServerInstance->Modes->DelMode(myumode);
                delete myumode;
        }
 
-       Version GetVersion()
+       virtual Version GetVersion()
        {
-               return Version(1, 0, 0, 0, VF_COMMON | VF_VENDOR, API_VERSION);
+               return Version("$Id$", VF_COMMON | VF_VENDOR, API_VERSION);
        }
 
-       void On005Numeric(std::string& output)
+       virtual void On005Numeric(std::string& output)
        {
                output += " CALLERID=g";
        }
 
        int PreText(User* user, User* dest, std::string& text, bool notice)
        {
-               if (!dest->IsModeSet('g')) return 0;
-               if (operoverride && IS_OPER(user)) return 0;
+               if (!dest->IsModeSet('g'))
+                       return 0;
+
+               if (operoverride && IS_OPER(user))
+                       return 0;
+
                callerid_data* dat = GetData(dest, true);
-               std::set<User*>& accepting = dat->accepting;
-               time_t& lastnotify = dat->lastnotify;
-               std::set<User*>::iterator i = accepting.find(dest);
-               if (i == accepting.end())
+               std::set<User*>::iterator i = dat->accepting.find(user);
+
+               if (i == dat->accepting.end())
                {
-                       time_t now = time(NULL);
+                       time_t now = ServerInstance->Time();
                        /* +g and *not* accepted */
-                       user->WriteNumeric(716, "%s %s :is in +g mode (server-side ignore).", user->nick, dest->nick);
-                       if (now > (lastnotify + (time_t)notify_cooldown))
+                       if (IS_LOCAL(user))
+                               user->WriteNumeric(716, "%s %s :is in +g mode (server-side ignore).", user->nick.c_str(), dest->nick.c_str());
+                       else
+                               ServerInstance->PI->PushToClient(user, std::string("::") + ServerInstance->Config->ServerName + " 716 " + user->nick + dest->nick + " :is in +g mode (server-side ignore).");
+                       if (now > (dat->lastnotify + (time_t)notify_cooldown))
                        {
-                               user->WriteNumeric(717, "%s %s :has been informed that you messaged them.", user->nick, dest->nick);
-                               dest->WriteNumeric(718, "%s %s %s@%s :is messaging you, and you have umode +g", dest->nick, user->nick, user->ident, user->dhost);
-                               lastnotify = now;
+                               user->WriteNumeric(717, "%s %s :has been informed that you messaged them.", user->nick.c_str(), dest->nick.c_str());
+                               dest->WriteNumeric(718, "%s %s %s@%s :is messaging you, and you have umode +g. Use /ACCEPT +%s to allow.", dest->nick.c_str(), user->nick.c_str(), user->ident.c_str(), user->dhost.c_str(), user->nick.c_str());
+                               dat->lastnotify = now;
                        }
                        return 1;
                }
                return 0;
        }
 
-       int OnUserPreMessage(User* user, void* dest, int target_type, std::string& text, char status, CUList &exempt_list)
+       virtual int OnUserPreMessage(User* user, void* dest, int target_type, std::string& text, char status, CUList &exempt_list)
        {
                if (IS_LOCAL(user) && target_type == TYPE_USER)
                        return PreText(user, (User*)dest, text, true);
+
                return 0;
        }
 
-       int OnUserPreNotice(User* user, void* dest, int target_type, std::string& text, char status, CUList &exempt_list)
+       virtual int OnUserPreNotice(User* user, void* dest, int target_type, std::string& text, char status, CUList &exempt_list)
        {
                if (IS_LOCAL(user) && target_type == TYPE_USER)
                        return PreText(user, (User*)dest, text, true);
+
                return 0;
        }
 
-       void OnCleanup(int type, void* item)
+       virtual void OnCleanup(int type, void* item)
        {
-               if (type != TYPE_USER) return;
+               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);
        }
 
-       int OnUserPreNick(User* user, const std::string& newnick)
+       virtual void OnSyncUserMetaData(User* user, Module* proto, void* opaque, const std::string& extname, bool displayable)
+       {
+               if (extname == "callerid_data")
+               {
+                       callerid_data* dat = GetData(user, false);
+                       if (dat)
+                       {
+                               std::string str = dat->ToString(displayable);
+                               proto->ProtoSendMetaData(opaque, TYPE_USER, user, extname, str);
+                       }
+               }
+       }
+
+       virtual void OnDecodeMetaData(int target_type, void* target, const std::string& extname, const std::string& extdata)
+       {
+               if (target_type == TYPE_USER && extname == "callerid_data")
+               {
+                       User* u = (User*)target;
+                       callerid_data* dat = new callerid_data(extdata, ServerInstance);
+                       u->Extend("callerid_data", dat);
+               }
+       }
+
+       virtual int OnUserPreNick(User* user, const std::string& newnick)
        {
                if (!tracknick)
                        RemoveFromAllAccepts(ServerInstance, user);
                return 0;
        }
 
-       void OnUserQuit(User* user, const std::string& message, const std::string& oper_message)
+       virtual void OnUserQuit(User* user, const std::string& message, const std::string& oper_message)
        {
                RemoveData(user);
                RemoveFromAllAccepts(ServerInstance, user);
        }
 
-       void OnRehash(User* user, const std::string& parameter)
+       virtual void OnRehash(User* user, const std::string& parameter)
        {
                ConfigReader Conf(ServerInstance);
-               int new_maxaccepts, new_cooldown;
-               bool new_override, new_track;
-               new_maxaccepts = Conf.ReadInteger("callerid", "maxaccepts", "16", 0, true);
-               switch (Conf.GetError())
-               {
-                       case 0: break;
-                       case CONF_VALUE_NOT_FOUND:
-                               new_maxaccepts = 16;
-                               break;
-                       case CONF_NOT_A_NUMBER:
-                               if (user) user->WriteServ("NOTICE %s :Invalid maxaccepts value '%s', not a number", user->nick, Conf.ReadValue("callerid", "maxaccepts", "", 0).c_str());
-                               throw ModuleException("Invalid maxaccepts value, not a number");
-                       case CONF_INT_NEGATIVE:
-                               if (user) user->WriteServ("NOTICE %s :Invalid maxaccepts value '%s', negative", user->nick, Conf.ReadValue("callerid", "maxaccepts", "", 0).c_str());
-                               throw ModuleException("Invalid maxaccepts value, negative");
-                       default:
-                               /* Yikes */
-                               throw ModuleException("Invalid maxaccepts value, unknown config error");
-               }
-               new_override = Conf.ReadFlag("callerid", "operoverride", "0", 0);
-               if (Conf.GetError() == CONF_VALUE_NOT_FOUND) new_override = false;
-               new_track = Conf.ReadFlag("callerid", "tracknick", "0", 0);
-               if (Conf.GetError() == CONF_VALUE_NOT_FOUND) new_track = false;
-               new_cooldown = Conf.ReadInteger("callerid", "cooldown", "60", 0, true);
-               switch (Conf.GetError())
-               {
-                       case 0: break;
-                       case CONF_VALUE_NOT_FOUND:
-                               new_cooldown = 16;
-                               break;
-                       case CONF_NOT_A_NUMBER:
-                               if (user) user->WriteServ("NOTICE %s :Invalid cooldown value '%s', not a number", user->nick, Conf.ReadValue("callerid", "maxaccepts", "", 0).c_str());
-                               throw ModuleException("Invalid cooldown value, not a number");
-                       case CONF_INT_NEGATIVE:
-                               if (user) user->WriteServ("NOTICE %s :Invalid cooldown value '%s', negative", user->nick, Conf.ReadValue("callerid", "maxaccepts", "", 0).c_str());
-                               throw ModuleException("Invalid cooldown value, negative");
-                       default:
-                               /* Yikes */
-                               throw ModuleException("Invalid cooldown value, unknown config error");
-               }
-               maxaccepts = new_maxaccepts;
-               notify_cooldown = new_cooldown;
-               operoverride = new_override;
-               tracknick = new_track;
+               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);
        }
 };