]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/cmd_whowas.cpp
class command_t -> class Command. Whey :D
[user/henk/code/inspircd.git] / src / cmd_whowas.cpp
index aab457243fdc5a653a2523abf2093a38519e689c..f982d48737ad740fb63faa5039ac6ded28630c3a 100644 (file)
@@ -1 +1,349 @@
-/*       +------------------------------------+\r *       | Inspire Internet Relay Chat Daemon |\r *       +------------------------------------+\r *\r *  InspIRCd: (C) 2002-2007 InspIRCd Development Team\r * See: http://www.inspircd.org/wiki/index.php/Credits\r *\r * This program is free but copyrighted software; see\r *            the file COPYING for details.\r *\r * ---------------------------------------------------\r */\r\r#include "inspircd.h"\r#include "configreader.h"\r#include "users.h"\r#include "commands/cmd_whowas.h"\r\rWhoWasMaintainTimer * timer;\r\rextern "C" DllExport command_t* init_command(InspIRCd* Instance)\r{\r     return new cmd_whowas(Instance);\r}\r\rcmd_whowas::cmd_whowas(InspIRCd* Instance)\r: command_t(Instance, "WHOWAS", 0, 1)\r{\r syntax = "<nick>{,<nick>}";\r    timer = new WhoWasMaintainTimer(Instance, 3600);\r       Instance->Timers->AddTimer(timer);\r}\r\rCmdResult cmd_whowas::Handle (const char** parameters, int pcnt, userrec* user)\r{\r        /* if whowas disabled in config */\r     if (ServerInstance->Config->WhoWasGroupSize == 0 || ServerInstance->Config->WhoWasMaxGroups == 0)\r      {\r              user->WriteServ("421 %s %s :This command has been disabled.",user->nick,command.c_str());\r              return CMD_FAILURE;\r    }\r\r     whowas_users::iterator i = whowas.find(parameters[0]);\r\r        if (i == whowas.end())\r {\r              user->WriteServ("406 %s %s :There was no such nickname",user->nick,parameters[0]);\r             user->WriteServ("369 %s %s :End of WHOWAS",user->nick,parameters[0]);\r          return CMD_FAILURE;\r    }\r      else\r   {\r              whowas_set* grp = i->second;\r           if (grp->size())\r               {\r                      for (whowas_set::iterator ux = grp->begin(); ux != grp->end(); ux++)\r                   {\r                              WhoWasGroup* u = *ux;\r                          time_t rawtime = u->signon;\r                            tm *timeinfo;\r                          char b[MAXBUF];\r        \r                               timeinfo = localtime(&rawtime);\r                                \r                               /* XXX - 'b' could be only 25 chars long and then strlcpy() would terminate it for us too? */\r                          strlcpy(b,asctime(timeinfo),MAXBUF);\r                           b[24] = 0;\r\r                            user->WriteServ("314 %s %s %s %s * :%s",user->nick,parameters[0],u->ident,u->dhost,u->gecos);\r                          \r                               if (IS_OPER(user))\r                                     user->WriteServ("379 %s %s :was connecting from *@%s", user->nick, parameters[0], u->host);\r                            \r                               if (*ServerInstance->Config->HideWhoisServer && !IS_OPER(user))\r                                        user->WriteServ("312 %s %s %s :%s",user->nick,parameters[0], ServerInstance->Config->HideWhoisServer, b);\r                              else\r                                   user->WriteServ("312 %s %s %s :%s",user->nick,parameters[0], u->server, b);\r                    }\r              }\r              else\r           {\r                      user->WriteServ("406 %s %s :There was no such nickname",user->nick,parameters[0]);\r                     user->WriteServ("369 %s %s :End of WHOWAS",user->nick,parameters[0]);\r                  return CMD_FAILURE;\r            }\r      }\r\r     user->WriteServ("369 %s %s :End of WHOWAS",user->nick,parameters[0]);\r  return CMD_SUCCESS;\r}\r\rCmdResult cmd_whowas::HandleInternal(const unsigned int id, const std::deque<classbase*> &parameters)\r{\r switch (id)\r    {\r              case WHOWAS_ADD:\r                       AddToWhoWas((userrec*)parameters[0]);\r          break;\r\r                case WHOWAS_STATS:\r                     GetStats((Extensible*)parameters[0]);\r          break;\r\r                case WHOWAS_PRUNE:\r                     PruneWhoWas(ServerInstance->Time());\r           break;\r\r                case WHOWAS_MAINTAIN:\r                  MaintainWhoWas(ServerInstance->Time());\r                break;\r\r                default:\r               break;\r }\r      return CMD_SUCCESS;\r}\r\rvoid cmd_whowas::GetStats(Extensible* ext)\r{\r    int whowas_size = 0;\r   int whowas_bytes = 0;\r  whowas_users_fifo::iterator iter;\r      for (iter = whowas_fifo.begin(); iter != whowas_fifo.end(); iter++)\r    {\r              whowas_set* n = (whowas_set*)whowas.find(iter->second)->second;\r                if (n->size())\r         {\r                      whowas_size += n->size();\r                      whowas_bytes += (sizeof(whowas_set) + ( sizeof(WhoWasGroup) * n->size() ) );\r           }\r      }\r      stats.assign("Whowas(MAPSETS) " +ConvToStr(whowas_size)+" ("+ConvToStr(whowas_bytes)+" bytes)");\r       ext->Extend("stats", stats.c_str());\r}\r\rvoid cmd_whowas::AddToWhoWas(userrec* user)\r{\r  /* if whowas disabled */\r       if (ServerInstance->Config->WhoWasGroupSize == 0 || ServerInstance->Config->WhoWasMaxGroups == 0)\r      {\r              return;\r        }\r\r     whowas_users::iterator iter = whowas.find(user->nick);\r\r        if (iter == whowas.end())\r      {\r              whowas_set* n = new whowas_set;\r                WhoWasGroup *a = new WhoWasGroup(user);\r                n->push_back(a);\r               whowas[user->nick] = n;\r                whowas_fifo.push_back(std::make_pair(ServerInstance->Time(),user->nick));\r\r             if ((int)(whowas.size()) > ServerInstance->Config->WhoWasMaxGroups)\r            {\r                      whowas_users::iterator iter = whowas.find(whowas_fifo[0].second);\r                      if (iter != whowas.end())\r                      {\r                              whowas_set* n = (whowas_set*)iter->second;\r                             if (n->size())\r                         {\r                                      while (n->begin() != n->end())\r                                 {\r                                              WhoWasGroup *a = *(n->begin());\r                                                DELETE(a);\r                                             n->pop_front();\r                                        }\r                              }\r                              DELETE(n);\r                             whowas.erase(iter);\r                    }\r                      whowas_fifo.pop_front();\r               }\r      }\r      else\r   {\r              whowas_set* group = (whowas_set*)iter->second;\r         WhoWasGroup *a = new WhoWasGroup(user);\r                group->push_back(a);\r\r          if ((int)(group->size()) > ServerInstance->Config->WhoWasGroupSize)\r            {\r                      WhoWasGroup *a = (WhoWasGroup*)*(group->begin());\r                      DELETE(a);\r                     group->pop_front();\r            }\r      }\r}\r\r/* on rehash, refactor maps according to new conf values */\rvoid cmd_whowas::PruneWhoWas(time_t t)\r{\r      /* config values */\r    int groupsize = ServerInstance->Config->WhoWasGroupSize;\r       int maxgroups = ServerInstance->Config->WhoWasMaxGroups;\r       int maxkeep =   ServerInstance->Config->WhoWasMaxKeep;\r\r        /* first cut the list to new size (maxgroups) and also prune entries that are timed out. */\r    whowas_users::iterator iter;\r   int fifosize;\r  while ((fifosize = (int)whowas_fifo.size()) > 0)\r       {\r              if (fifosize > maxgroups || whowas_fifo[0].first < t - maxkeep)\r                {\r                      iter = whowas.find(whowas_fifo[0].second);\r                     /* hopefully redundant integrity check, but added while debugging r6216 */\r                     if (iter == whowas.end())\r                      {\r                              /* this should never happen, if it does maps are corrupt */\r                            ServerInstance->Log(DEFAULT, "BUG: Whowas maps got corrupted! (1)");\r                           return;\r                        }\r                      whowas_set* n = (whowas_set*)iter->second;\r                     if (n->size())\r                 {\r                              while (n->begin() != n->end())\r                         {\r                                      WhoWasGroup *a = *(n->begin());\r                                        DELETE(a);\r                                     n->pop_front();\r                                }\r                      }\r                      DELETE(n);\r                     whowas.erase(iter);\r                    whowas_fifo.pop_front();\r               }\r              else\r                   break;\r }\r\r     /* Then cut the whowas sets to new size (groupsize) */\r fifosize = (int)whowas_fifo.size();\r    for (int i = 0; i < fifosize; i++)\r     {\r              iter = whowas.find(whowas_fifo[0].second);\r             /* hopefully redundant integrity check, but added while debugging r6216 */\r             if (iter == whowas.end())\r              {\r                      /* this should never happen, if it does maps are corrupt */\r                    ServerInstance->Log(DEFAULT, "BUG: Whowas maps got corrupted! (2)");\r                   return;\r                }\r              whowas_set* n = (whowas_set*)iter->second;\r             if (n->size())\r         {\r                      int nickcount = n->size();\r                     while (n->begin() != n->end() && nickcount > groupsize)\r                        {\r                              WhoWasGroup *a = *(n->begin());\r                                DELETE(a);\r                             n->pop_front();\r                                nickcount--;\r                   }\r              }\r      }\r}\r\r/* call maintain once an hour to remove expired nicks */\rvoid cmd_whowas::MaintainWhoWas(time_t t)\r{\r      for (whowas_users::iterator iter = whowas.begin(); iter != whowas.end(); iter++)\r       {\r              whowas_set* n = (whowas_set*)iter->second;\r             if (n->size())\r         {\r                      while ((n->begin() != n->end()) && ((*n->begin())->signon < t - ServerInstance->Config->WhoWasMaxKeep))\r                        {\r                              WhoWasGroup *a = *(n->begin());\r                                DELETE(a);\r                             n->erase(n->begin());\r                  }\r              }\r      }\r}\r\rcmd_whowas::~cmd_whowas()\r{\r       if (timer)\r     {\r              ServerInstance->Timers->DelTimer(timer);\r       }\r\r     whowas_users::iterator iter;\r   int fifosize;\r  while ((fifosize = (int)whowas_fifo.size()) > 0)\r       {\r              iter = whowas.find(whowas_fifo[0].second);\r             /* hopefully redundant integrity check, but added while debugging r6216 */\r             if (iter == whowas.end())\r              {\r                      /* this should never happen, if it does maps are corrupt */\r                    ServerInstance->Log(DEFAULT, "BUG: Whowas maps got corrupted! (3)");\r                   return;\r                }\r              whowas_set* n = (whowas_set*)iter->second;\r             if (n->size())\r         {\r                      while (n->begin() != n->end())\r                 {\r                              WhoWasGroup *a = *(n->begin());\r                                DELETE(a);\r                             n->pop_front();\r                        }\r              }\r              DELETE(n);\r             whowas.erase(iter);\r            whowas_fifo.pop_front();\r       }\r}\r\rWhoWasGroup::WhoWasGroup(userrec* user) : host(NULL), dhost(NULL), ident(NULL), server(NULL), gecos(NULL), signon(user->signon)\r{\r this->host = strdup(user->host);\r       this->dhost = strdup(user->dhost);\r     this->ident = strdup(user->ident);\r     this->server = user->server;\r   this->gecos = strdup(user->fullname);\r}\r\rWhoWasGroup::~WhoWasGroup()\r{\r if (host)\r              free(host);\r    if (dhost)\r             free(dhost);\r   if (ident)\r             free(ident);\r   if (gecos)\r             free(gecos);\r}\r\r/* every hour, run this function which removes all entries older than Config->WhoWasMaxKeep */\rvoid WhoWasMaintainTimer::Tick(time_t t)\r{\r      command_t* whowas_command = ServerInstance->Parser->GetHandler("WHOWAS");\r      if (whowas_command)\r    {\r              std::deque<classbase*> params;\r         whowas_command->HandleInternal(WHOWAS_MAINTAIN, params);\r       }\r}\r
\ No newline at end of file
+/*       +------------------------------------+
+ *       | Inspire Internet Relay Chat Daemon |
+ *       +------------------------------------+
+ *
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
+ *
+ * This program is free but copyrighted software; see
+ *            the file COPYING for details.
+ *
+ * ---------------------------------------------------
+ */
+
+#include "inspircd.h"
+#include "commands/cmd_whowas.h"
+
+WhoWasMaintainTimer * timer;
+
+extern "C" DllExport Command* init_command(InspIRCd* Instance)
+{
+       return new cmd_whowas(Instance);
+}
+
+cmd_whowas::cmd_whowas(InspIRCd* Instance)
+: Command(Instance, "WHOWAS", 0, 1)
+{
+       syntax = "<nick>{,<nick>}";
+       timer = new WhoWasMaintainTimer(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
+       {
+               whowas_set* grp = i->second;
+               if (grp->size())
+               {
+                       for (whowas_set::iterator ux = grp->begin(); ux != grp->end(); ux++)
+                       {
+                               WhoWasGroup* u = *ux;
+                               time_t rawtime = u->signon;
+                               tm *timeinfo;
+                               char b[MAXBUF];
+       
+                               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);
+                               b[24] = 0;
+
+                               user->WriteServ("314 %s %s %s %s * :%s",user->nick,parameters[0],u->ident,u->dhost,u->gecos);
+                               
+                               if (IS_OPER(user))
+                                       user->WriteServ("379 %s %s :was connecting from *@%s", user->nick, parameters[0], u->host);
+                               
+                               if (*ServerInstance->Config->HideWhoisServer && !IS_OPER(user))
+                                       user->WriteServ("312 %s %s %s :%s",user->nick,parameters[0], ServerInstance->Config->HideWhoisServer, b);
+                               else
+                                       user->WriteServ("312 %s %s %s :%s",user->nick,parameters[0], u->server, b);
+                       }
+               }
+               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() ) );
+               }
+       }
+       stats.assign("Whowas(MAPSETS) " +ConvToStr(whowas_size)+" ("+ConvToStr(whowas_bytes)+" bytes)");
+       ext->Extend("stats", stats.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(DEFAULT, "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(DEFAULT, "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(DEFAULT, "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 WhoWasMaintainTimer::Tick(time_t t)
+{
+       Command* whowas_command = ServerInstance->Parser->GetHandler("WHOWAS");
+       if (whowas_command)
+       {
+               std::deque<classbase*> params;
+               whowas_command->HandleInternal(WHOWAS_MAINTAIN, params);
+       }
+}