]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/cmd_whowas.cpp
Add config <options:disablehmac> to support disabling of HMAC, and tidy up to detect...
[user/henk/code/inspircd.git] / src / cmd_whowas.cpp
index fe0d8c90d68f966acb97106ab7f4df634621d51e..46228f0dc04ac410aedf523bfe6d71e640ae87e3 100644 (file)
@@ -2,41 +2,57 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                       E-mail:
- *                <brain@chatspike.net>
- *                <Craig@chatspike.net>
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
  *
- * Written by Craig Edwards, Craig McLure, and others.
  * This program is free but copyrighted software; see
  *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-#include "inspircd_config.h"
 #include "configreader.h"
 #include "users.h"
-#include "commands.h"
-
 #include "commands/cmd_whowas.h"
 
-void cmd_whowas::Handle (const char** parameters, int pcnt, userrec* user)
+extern "C" command_t* init_command(InspIRCd* Instance)
 {
-       irc::whowas::whowas_users::iterator i = ServerInstance->whowas.find(parameters[0]);
+       return new cmd_whowas(Instance);
+}
 
-       if (i == ServerInstance->whowas.end())
+cmd_whowas::cmd_whowas(InspIRCd* Instance)
+: command_t(Instance, "WHOWAS", 0, 1)
+{
+       syntax = "<nick>{,<nick>}";
+       timer = new MaintainTimer(Instance, 3600);
+       Instance->Timers->AddTimer(timer);
+}
+
+CmdResult cmd_whowas::Handle (const char** parameters, int pcnt, userrec* user)
+{
+       /* if whowas disabled in config */
+       if (ServerInstance->Config->WhoWasGroupSize == 0 || ServerInstance->Config->WhoWasMaxGroups == 0)
+       {
+               user->WriteServ("421 %s %s :This command has been disabled.",user->nick,command.c_str());
+               return CMD_FAILURE;
+       }
+                        
+       whowas_users::iterator i = whowas.find(parameters[0]);
+
+       if (i == whowas.end())
        {
                user->WriteServ("406 %s %s :There was no such nickname",user->nick,parameters[0]);
+               user->WriteServ("369 %s %s :End of WHOWAS",user->nick,parameters[0]);
+               return CMD_FAILURE;
        }
        else
        {
-               irc::whowas::whowas_set* grp = i->second;
+               whowas_set* grp = i->second;
                if (grp->size())
                {
-                       for (irc::whowas::whowas_set::iterator ux = grp->begin(); ux != grp->end(); ux++)
+                       for (whowas_set::iterator ux = grp->begin(); ux != grp->end(); ux++)
                        {
-                               irc::whowas::WhoWasGroup* u = *ux;
+                               WhoWasGroup* u = *ux;
                                time_t rawtime = u->signon;
                                tm *timeinfo;
                                char b[MAXBUF];
@@ -61,8 +77,261 @@ void cmd_whowas::Handle (const char** parameters, int pcnt, userrec* user)
                else
                {
                        user->WriteServ("406 %s %s :There was no such nickname",user->nick,parameters[0]);
+                       user->WriteServ("369 %s %s :End of WHOWAS",user->nick,parameters[0]);
+                       return CMD_FAILURE;
                }
        }
-       
+
        user->WriteServ("369 %s %s :End of WHOWAS",user->nick,parameters[0]);
+       return CMD_SUCCESS;
+}
+
+CmdResult cmd_whowas::HandleInternal(const unsigned int id, const std::deque<classbase*> &parameters)
+{
+       switch (id)
+       {
+               case WHOWAS_ADD:
+                       AddToWhoWas((userrec*)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;
+       }
+       return CMD_SUCCESS;
+}
+
+void cmd_whowas::GetStats(Extensible* ext)
+{
+       int whowas_size = 0;
+       int whowas_bytes = 0;
+       whowas_users_fifo::iterator iter;
+       for (iter = whowas_fifo.begin(); iter != whowas_fifo.end(); iter++)
+       {
+               whowas_set* n = (whowas_set*)whowas.find(iter->second)->second;
+               if (n->size())
+               {
+                       whowas_size += n->size();
+                       whowas_bytes += (sizeof(whowas_set) + ( sizeof(WhoWasGroup) * n->size() ) );
+               }
+       }
+       ext->Extend("stats", std::string("Whowas(MAPSETS) " +ConvToStr(whowas_size)+" ("+ConvToStr(whowas_bytes)+" bytes)").c_str());
+}
+
+void cmd_whowas::AddToWhoWas(userrec* user)
+{
+       /* if whowas disabled */
+       if (ServerInstance->Config->WhoWasGroupSize == 0 || ServerInstance->Config->WhoWasMaxGroups == 0)
+       {
+               return;
+       }
+
+       whowas_users::iterator iter = whowas.find(user->nick);
+
+       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));
+
+               if ((int)(whowas.size()) > ServerInstance->Config->WhoWasMaxGroups)
+               {
+                       whowas_users::iterator iter = whowas.find(whowas_fifo[0].second);
+                       if (iter != whowas.end())
+                       {
+                               whowas_set* n = (whowas_set*)iter->second;
+                               if (n->size())
+                               {
+                                       while (n->begin() != n->end())
+                                       {
+                                               WhoWasGroup *a = *(n->begin());
+                                               DELETE(a);
+                                               n->pop_front();
+                                       }
+                               }
+                               DELETE(n);
+                               whowas.erase(iter);
+                       }
+                       whowas_fifo.pop_front();
+               }
+       }
+       else
+       {
+               whowas_set* group = (whowas_set*)iter->second;
+               WhoWasGroup *a = new WhoWasGroup(user);
+               group->push_back(a);
+
+               if ((int)(group->size()) > ServerInstance->Config->WhoWasGroupSize)
+               {
+                       WhoWasGroup *a = (WhoWasGroup*)*(group->begin());
+                       DELETE(a);
+                       group->pop_front();
+               }
+       }
+}
+
+/* on rehash, refactor maps according to new conf values */
+void cmd_whowas::PruneWhoWas(time_t t)
+{
+       /* config values */
+       int groupsize = ServerInstance->Config->WhoWasGroupSize;
+       int maxgroups = ServerInstance->Config->WhoWasMaxGroups;
+       int maxkeep =   ServerInstance->Config->WhoWasMaxKeep;
+
+       /* first cut the list to new size (maxgroups) and also prune entries that are timed out. */
+       whowas_users::iterator iter;
+       int fifosize;
+       while ((fifosize = (int)whowas_fifo.size()) > 0)
+       {
+               if (fifosize > maxgroups || whowas_fifo[0].first < t - maxkeep)
+               {
+                       iter = whowas.find(whowas_fifo[0].second);
+                       /* hopefully redundant integrity check, but added while debugging r6216 */
+                       if (iter == whowas.end())
+                       {
+                               /* this should never happen, if it does maps are corrupt */
+                               ServerInstance->Log(DEBUG, "BUG: Whowas maps got corrupted! (1)");
+                               return;
+                       }
+                       whowas_set* n = (whowas_set*)iter->second;
+                       if (n->size())
+                       {
+                               while (n->begin() != n->end())
+                               {
+                                       WhoWasGroup *a = *(n->begin());
+                                       DELETE(a);
+                                       n->pop_front();
+                               }
+                       }
+                       DELETE(n);
+                       whowas.erase(iter);
+                       whowas_fifo.pop_front();
+               }
+               else
+                       break;
+       }
+
+       /* Then cut the whowas sets to new size (groupsize) */
+       fifosize = (int)whowas_fifo.size();
+       for (int i = 0; i < fifosize; i++)
+       {
+               iter = whowas.find(whowas_fifo[0].second);
+               /* hopefully redundant integrity check, but added while debugging r6216 */
+               if (iter == whowas.end())
+               {
+                       /* this should never happen, if it does maps are corrupt */
+                       ServerInstance->Log(DEBUG, "BUG: Whowas maps got corrupted! (2)");
+                       return;
+               }
+               whowas_set* n = (whowas_set*)iter->second;
+               if (n->size())
+               {
+                       int nickcount = n->size();
+                       while (n->begin() != n->end() && nickcount > groupsize)
+                       {
+                               WhoWasGroup *a = *(n->begin());
+                               DELETE(a);
+                               n->pop_front();
+                               nickcount--;
+                       }
+               }
+       }
+}
+
+/* call maintain once an hour to remove expired nicks */
+void cmd_whowas::MaintainWhoWas(time_t t)
+{
+       for (whowas_users::iterator iter = whowas.begin(); iter != whowas.end(); iter++)
+       {
+               whowas_set* n = (whowas_set*)iter->second;
+               if (n->size())
+               {
+                       while ((n->begin() != n->end()) && ((*n->begin())->signon < t - ServerInstance->Config->WhoWasMaxKeep))
+                       {
+                               WhoWasGroup *a = *(n->begin());
+                               DELETE(a);
+                               n->erase(n->begin());
+                       }
+               }
+       }
+}
+
+cmd_whowas::~cmd_whowas()
+{
+       if (timer)
+       {
+               ServerInstance->Timers->DelTimer(timer);
+       }
+
+       whowas_users::iterator iter;
+       int fifosize;
+       while ((fifosize = (int)whowas_fifo.size()) > 0)
+       {
+               iter = whowas.find(whowas_fifo[0].second);
+               /* hopefully redundant integrity check, but added while debugging r6216 */
+               if (iter == whowas.end())
+               {
+                       /* this should never happen, if it does maps are corrupt */
+                       ServerInstance->Log(DEBUG, "BUG: Whowas maps got corrupted! (3)");
+                       return;
+               }
+               whowas_set* n = (whowas_set*)iter->second;
+               if (n->size())
+               {
+                       while (n->begin() != n->end())
+                       {
+                               WhoWasGroup *a = *(n->begin());
+                               DELETE(a);
+                               n->pop_front();
+                       }
+               }
+               DELETE(n);
+               whowas.erase(iter);
+               whowas_fifo.pop_front();
+       }
+}
+
+WhoWasGroup::WhoWasGroup(userrec* user) : host(NULL), dhost(NULL), ident(NULL), server(NULL), gecos(NULL), 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 MaintainTimer::Tick(time_t t)
+{
+       command_t* whowas_command = ServerInstance->Parser->GetHandler("WHOWAS");
+       if (whowas_command)
+       {
+               std::deque<classbase*> params;
+               whowas_command->HandleInternal(WHOWAS_MAINTAIN, params);
+       }
 }