]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/commands/cmd_whowas.cpp
Make cmd_whowas act like a module, remove special handling
[user/henk/code/inspircd.git] / src / commands / cmd_whowas.cpp
index 8bc188eee6e7c483d38c20ec05aa94eb04599add..dfe513dff78c8d34ef36f87d51bad701542ee9c7 100644 (file)
@@ -1,31 +1,32 @@
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
  *
- *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
+ *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2008 Thomas Stagner <aquanight@inspircd.org>
+ *   Copyright (C) 2008 Craig Edwards <craigedwards@brainbox.cc>
+ *   Copyright (C) 2007-2008 Robin Burchell <robin+git@viroteck.net>
  *
- * This program is free but copyrighted software; see
- *            the file COPYING for details.
+ * 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/>.
  */
 
+
 #include "inspircd.h"
 #include "commands/cmd_whowas.h"
 
-WhoWasMaintainTimer * timer;
-
-extern "C" DllExport Command* init_command(InspIRCd* Instance)
-{
-       return new CommandWhowas(Instance);
-}
-
-CommandWhowas::CommandWhowas(InspIRCd* Instance) : Command(Instance, "WHOWAS", 0, 1, false, 2)
+CommandWhowas::CommandWhowas( Module* parent) : Command(parent, "WHOWAS", 1)
 {
        syntax = "<nick>{,<nick>}";
-       timer = new WhoWasMaintainTimer(Instance, 3600);
-       Instance->Timers->AddTimer(timer);
+       Penalty = 2;
 }
 
 CmdResult CommandWhowas::Handle (const std::vector<std::string>& parameters, User* user)
@@ -33,7 +34,7 @@ CmdResult CommandWhowas::Handle (const std::vector<std::string>& parameters, Use
        /* if whowas disabled in config */
        if (ServerInstance->Config->WhoWasGroupSize == 0 || ServerInstance->Config->WhoWasMaxGroups == 0)
        {
-               user->WriteNumeric(421, "%s %s :This command has been disabled.",user->nick,command.c_str());
+               user->WriteNumeric(421, "%s %s :This command has been disabled.",user->nick.c_str(),name.c_str());
                return CMD_FAILURE;
        }
 
@@ -41,8 +42,8 @@ CmdResult CommandWhowas::Handle (const std::vector<std::string>& parameters, Use
 
        if (i == whowas.end())
        {
-               user->WriteNumeric(406, "%s %s :There was no such nickname",user->nick,parameters[0].c_str());
-               user->WriteNumeric(369, "%s %s :End of WHOWAS",user->nick,parameters[0].c_str());
+               user->WriteNumeric(406, "%s %s :There was no such nickname",user->nick.c_str(),parameters[0].c_str());
+               user->WriteNumeric(369, "%s %s :End of WHOWAS",user->nick.c_str(),parameters[0].c_str());
                return CMD_FAILURE;
        }
        else
@@ -55,64 +56,39 @@ CmdResult CommandWhowas::Handle (const std::vector<std::string>& parameters, Use
                                WhoWasGroup* u = *ux;
                                time_t rawtime = u->signon;
                                tm *timeinfo;
-                               char b[MAXBUF];
-       
+                               char b[25];
+
                                timeinfo = localtime(&rawtime);
-                               
-                               /* XXX - 'b' could be only 25 chars long and then strlcpy() would terminate it for us too? */
-                               strlcpy(b,asctime(timeinfo),MAXBUF);
+
+                               strncpy(b,asctime(timeinfo),24);
                                b[24] = 0;
 
-                               user->WriteNumeric(314, "%s %s %s %s * :%s",user->nick,parameters[0].c_str(),u->ident,u->dhost,u->gecos);
-                               
-                               if (IS_OPER(user))
-                                       user->WriteNumeric(379, "%s %s :was connecting from *@%s", user->nick, parameters[0].c_str(), u->host);
-                               
-                               if (*ServerInstance->Config->HideWhoisServer && !IS_OPER(user))
-                                       user->WriteNumeric(312, "%s %s %s :%s",user->nick,parameters[0].c_str(), ServerInstance->Config->HideWhoisServer, b);
+                               user->WriteNumeric(314, "%s %s %s %s * :%s",user->nick.c_str(),parameters[0].c_str(),
+                                       u->ident.c_str(),u->dhost.c_str(),u->gecos.c_str());
+
+                               if (user->HasPrivPermission("users/auspex"))
+                                       user->WriteNumeric(379, "%s %s :was connecting from *@%s",
+                                               user->nick.c_str(), parameters[0].c_str(), u->host.c_str());
+
+                               if (!ServerInstance->Config->HideWhoisServer.empty() && !user->HasPrivPermission("servers/auspex"))
+                                       user->WriteNumeric(312, "%s %s %s :%s",user->nick.c_str(),parameters[0].c_str(), ServerInstance->Config->HideWhoisServer.c_str(), b);
                                else
-                                       user->WriteNumeric(312, "%s %s %s :%s",user->nick,parameters[0].c_str(), u->server, b);
+                                       user->WriteNumeric(312, "%s %s %s :%s",user->nick.c_str(),parameters[0].c_str(), u->server.c_str(), b);
                        }
                }
                else
                {
-                       user->WriteNumeric(406, "%s %s :There was no such nickname",user->nick,parameters[0].c_str());
-                       user->WriteNumeric(369, "%s %s :End of WHOWAS",user->nick,parameters[0].c_str());
+                       user->WriteNumeric(406, "%s %s :There was no such nickname",user->nick.c_str(),parameters[0].c_str());
+                       user->WriteNumeric(369, "%s %s :End of WHOWAS",user->nick.c_str(),parameters[0].c_str());
                        return CMD_FAILURE;
                }
        }
 
-       user->WriteNumeric(369, "%s %s :End of WHOWAS",user->nick,parameters[0].c_str());
-       return CMD_SUCCESS;
-}
-
-CmdResult CommandWhowas::HandleInternal(const unsigned int id, const std::deque<classbase*> &parameters)
-{
-       switch (id)
-       {
-               case WHOWAS_ADD:
-                       AddToWhoWas((User*)parameters[0]);
-               break;
-
-               case WHOWAS_STATS:
-                       GetStats((Extensible*)parameters[0]);
-               break;
-
-               case WHOWAS_PRUNE:
-                       PruneWhoWas(ServerInstance->Time());
-               break;
-
-               case WHOWAS_MAINTAIN:
-                       MaintainWhoWas(ServerInstance->Time());
-               break;
-
-               default:
-               break;
-       }
+       user->WriteNumeric(369, "%s %s :End of WHOWAS",user->nick.c_str(),parameters[0].c_str());
        return CMD_SUCCESS;
 }
 
-void CommandWhowas::GetStats(Extensible* ext)
+std::string CommandWhowas::GetStats()
 {
        int whowas_size = 0;
        int whowas_bytes = 0;
@@ -126,8 +102,7 @@ void CommandWhowas::GetStats(Extensible* ext)
                        whowas_bytes += (sizeof(whowas_set) + ( sizeof(WhoWasGroup) * n->size() ) );
                }
        }
-       stats.assign("Whowas(MAPSETS) " +ConvToStr(whowas_size)+" ("+ConvToStr(whowas_bytes)+" bytes)");
-       ext->Extend("stats", stats.c_str());
+       return "Whowas entries: " +ConvToStr(whowas_size)+" ("+ConvToStr(whowas_bytes)+" bytes)";
 }
 
 void CommandWhowas::AddToWhoWas(User* user)
@@ -138,15 +113,15 @@ void CommandWhowas::AddToWhoWas(User* user)
                return;
        }
 
-       whowas_users::iterator iter = whowas.find(user->nick);
+       whowas_users::iterator iter = whowas.find(irc::string(user->nick.c_str()));
 
        if (iter == whowas.end())
        {
                whowas_set* n = new whowas_set;
                WhoWasGroup *a = new WhoWasGroup(user);
                n->push_back(a);
-               whowas[user->nick] = n;
-               whowas_fifo.push_back(std::make_pair(ServerInstance->Time(),user->nick));
+               whowas[user->nick.c_str()] = n;
+               whowas_fifo.push_back(std::make_pair(ServerInstance->Time(),user->nick.c_str()));
 
                if ((int)(whowas.size()) > ServerInstance->Config->WhoWasMaxGroups)
                {
@@ -278,11 +253,6 @@ void CommandWhowas::MaintainWhoWas(time_t t)
 
 CommandWhowas::~CommandWhowas()
 {
-       if (timer)
-       {
-               ServerInstance->Timers->DelTimer(timer);
-       }
-
        whowas_users::iterator iter;
        int fifosize;
        while ((fifosize = (int)whowas_fifo.size()) > 0)
@@ -315,34 +285,58 @@ CommandWhowas::~CommandWhowas()
        }
 }
 
-WhoWasGroup::WhoWasGroup(User* user) : host(NULL), dhost(NULL), ident(NULL), server(NULL), gecos(NULL), signon(user->signon)
+WhoWasGroup::WhoWasGroup(User* user) : host(user->host), dhost(user->dhost), ident(user->ident),
+       server(user->server), gecos(user->fullname), signon(user->signon)
 {
-       this->host = strdup(user->host);
-       this->dhost = strdup(user->dhost);
-       this->ident = strdup(user->ident);
-       this->server = user->server;
-       this->gecos = strdup(user->fullname);
 }
 
 WhoWasGroup::~WhoWasGroup()
 {
-       if (host)
-               free(host);
-       if (dhost)
-               free(dhost);
-       if (ident)
-               free(ident);
-       if (gecos)
-               free(gecos);
 }
 
-/* every hour, run this function which removes all entries older than Config->WhoWasMaxKeep */
-void WhoWasMaintainTimer::Tick(time_t)
+class ModuleWhoWas : public Module
 {
-       Command* whowas_command = ServerInstance->Parser->GetHandler("WHOWAS");
-       if (whowas_command)
+       CommandWhowas cmd;
+ public:
+       ModuleWhoWas() : cmd(this)
        {
-               std::deque<classbase*> params;
-               whowas_command->HandleInternal(WHOWAS_MAINTAIN, params);
        }
-}
+
+       void init()
+       {
+               ServerInstance->Modules->AddService(cmd);
+               Implementation eventlist[] = { I_OnGarbageCollect, I_OnUserQuit, I_OnStats, I_OnRehash };
+               ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
+       }
+
+       void OnGarbageCollect()
+       {
+               /* Removes all entries older than WhoWasMaxKeep */
+               cmd.MaintainWhoWas(ServerInstance->Time());
+       }
+
+       void OnUserQuit(User* user, const std::string& message, const std::string& oper_message)
+       {
+               cmd.AddToWhoWas(user);
+       }
+
+       ModResult OnStats(char symbol, User* user, string_list &results)
+       {
+               if (symbol == 'z')
+                       results.push_back(ServerInstance->Config->ServerName+" 249 "+user->nick+" :"+cmd.GetStats());
+
+               return MOD_RES_PASSTHRU;
+       }
+
+       void OnRehash(User* user)
+       {
+               cmd.PruneWhoWas(ServerInstance->Time());
+       }
+
+       Version GetVersion()
+       {
+               return Version("WHOWAS", VF_VENDOR);
+       }
+};
+
+MODULE_INIT(ModuleWhoWas)