]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/cmd_who.cpp
/who tidyup - needs QA'ing
[user/henk/code/inspircd.git] / src / cmd_who.cpp
index 607fba68ef94b88661279ccfc15e49f985ff892b..c4c37e52876d260c0ac32c3af203fbdf8b85695e 100644 (file)
  *       | 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.
+ *         the file COPYING for details.
  *
  * ---------------------------------------------------
  */
 
-using namespace std;
-
-#include "inspircd_config.h"
-#include "inspircd.h"
-#include "inspircd_io.h"
-#include <time.h>
-#include <string>
-#ifdef GCC3
-#include <ext/hash_map>
-#else
-#include <hash_map>
-#endif
-#include <map>
-#include <sstream>
-#include <vector>
-#include <deque>
+#include "configreader.h"
 #include "users.h"
-#include "ctables.h"
-#include "globals.h"
 #include "modules.h"
-#include "dynamic.h"
 #include "wildcard.h"
-#include "message.h"
-#include "commands.h"
-#include "mode.h"
-#include "xline.h"
-#include "inspstring.h"
-#include "dnsqueue.h"
-#include "helperfuncs.h"
-#include "hashcomp.h"
-#include "socketengine.h"
-#include "typedefs.h"
-#include "command_parse.h"
-#include "cmd_who.h"
-
-extern ServerConfig* Config;
-extern InspIRCd* ServerInstance;
-extern int MODCOUNT;
-extern std::vector<Module*> modules;
-extern std::vector<ircd_module*> factory;
-extern time_t TIME;
-extern user_hash clientlist;
-extern chan_hash chanlist;
-extern whowas_hash whowas;
-extern std::vector<userrec*> all_opers;
-extern std::vector<userrec*> local_users;
-extern userrec* fd_ref_table[MAX_DESCRIPTORS];
-
-void cmd_who::Handle (char **parameters, int pcnt, userrec *user)
+#include "commands/cmd_who.h"
+
+/* get the last 'visible' chan of a user */
+static char *getlastchanname(userrec *u)
+{
+       UCListIter i = u->chans.begin();
+       if (i != u->chans.end())
+       {
+               if (!i->first->IsModeSet('s'))
+                       return i->first->name;
+       }
+
+       return "*";
+}
+
+bool whomatch(userrec* user, const char* matchtext, bool opt_realname, bool opt_showrealhost, bool opt_mode)
 {
-       chanrec* Ptr = NULL;
-       char tmp[10];
-       
-       /* theres more to do here, but for now just close the socket */
-       if (pcnt == 1)
+       bool realhost = false;
+       bool realname = false;
+       bool positive = true;
+
+       if (user->registered != REG_ALL)
+               return false;
+
+       if (opt_mode)
        {
-               if ((!strcmp(parameters[0],"0")) || (!strcmp(parameters[0],"*")))
+               for (const char* n = matchtext; *n; n++)
                {
-                       if ((user->chans.size()) && (user->chans[0].channel))
+                       if (*n == '+')
                        {
-                               int n_list = 0;
-                               for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++)
-                               {
-                                       Ptr = i->second->chans[0].channel;
-                                       // suggested by phidjit and FCS
-                                       if ((!common_channels(user,i->second)) && (isnick(i->second->nick)))
-                                       {
-                                               // Bug Fix #29
-                                               *tmp = 0;
-                                               if (*i->second->awaymsg) {
-                                                       strlcat(tmp, "G", 9);
-                                               } else {
-                                                       strlcat(tmp, "H", 9);
-                                               }
-                                               if (strchr(i->second->modes,'o')) { strlcat(tmp, "*", 9); }
-                                               WriteServ(user->fd,"352 %s %s %s %s %s %s %s :0 %s",user->nick, Ptr ? Ptr->name : "*", i->second->ident, i->second->dhost, i->second->server, i->second->nick, tmp, i->second->fullname);
-                                               n_list++;
-                                               if (n_list > Config->MaxWhoResults)
-                                               {
-                                                       WriteServ(user->fd,"523 %s WHO :Command aborted: More results than configured limit",user->nick);
-                                                       break;
-                                               }
-                                       }
-                               }
+                               positive = true;
+                               continue;
                        }
-                       if (Ptr)
+                       else if (*n == '-')
                        {
-                               WriteServ(user->fd,"315 %s %s :End of /WHO list.",user->nick , parameters[0]);
+                               positive = false;
+                               continue;
                        }
-                       else
+                       if (user->IsModeSet(*n) != positive)
+                               return false;
+               }
+               return true;
+       }
+
+       if (opt_realname)
+               realname = match(user->fullname, matchtext);
+
+       if (opt_showrealhost)
+               realhost = match(user->host, matchtext);
+
+       return ((realname) || (realhost) || (match(user->dhost, matchtext)) || (match(user->nick, matchtext)) || (match(user->server, matchtext)));
+}
+
+
+
+extern "C" command_t* init_command(InspIRCd* Instance)
+{
+       return new cmd_who(Instance);
+}
+
+bool cmd_who::CanView(chanrec* chan, userrec* user)
+{
+       if (!user || !chan)
+               return false;
+
+       /* Execute items in fastest-to-execute first order */
+       if (*user->oper)
+               return true;
+       else if (!chan->IsModeSet('s') && !chan->IsModeSet('p'))
+               return true;
+       else if (chan->HasUser(user))
+               return true;
+
+       return false;
+}
+
+void cmd_who::SendWhoLine(userrec* user, const std::string &initial, chanrec* ch, userrec* u, std::vector<std::string> &whoresults)
+{
+       std::string lcn = getlastchanname(u);
+       chanrec* chlast = ServerInstance->FindChan(lcn);
+
+       std::string wholine =   initial + (ch ? ch->name : lcn) + " " + u->ident + " " + (opt_showrealhost ? u->host : u->dhost) + " " +
+                               ((*ServerInstance->Config->HideWhoisServer && !*user->oper) ? ServerInstance->Config->HideWhoisServer : u->server) +
+                               " " + u->nick + " ";
+
+       /* away? */
+       if (u->awaymsg)
+       {
+               wholine.append("G");
+       }
+       else
+       {
+               wholine.append("H");
+       }
+
+       /* oper? */
+       if (u->oper)
+       {
+               wholine.append("*");
+       }
+
+       wholine = wholine + (ch ? ch->GetPrefixChar(u) : (chlast ? chlast->GetPrefixChar(u) : "")) + " :0 " + u->fullname;
+       whoresults.push_back(wholine);
+}
+
+CmdResult cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
+{
+       /*
+        * XXX - RFC says:
+        *   The <name> passed to WHO is matched against users' host, server, real
+        *   name and nickname
+        * Currently, we support WHO #chan, WHO nick, WHO 0, WHO *, and the addition of a 'o' flag, as per RFC.
+        */
+
+       /* WHO options */
+       opt_viewopersonly = false;
+       opt_showrealhost = false;
+       opt_unlimit = false;
+       opt_realname = false;
+       opt_mode = false;
+
+       chanrec *ch = NULL;
+       std::vector<std::string> whoresults;
+       std::string initial = "352 " + std::string(user->nick) + " ";
+
+       const char* matchtext = NULL;
+
+       /* Change '0' into '*' so the wildcard matcher can grok it */
+       matchtext = parameters[0];
+       if (!strcmp(matchtext,"0"))
+               matchtext = "*";
+
+       if (pcnt > 1)
+       {
+               /* parse flags */
+               const char *iter = parameters[1];
+
+               while (*iter)
+               {
+                       switch (*iter)
                        {
-                               WriteServ(user->fd,"315 %s %s :End of /WHO list.",user->nick, parameters[0]);
+                               case 'o':
+                                       opt_viewopersonly = true;
+                               break;
+                               case 'h':
+                                       if (*user->oper)
+                                               opt_showrealhost = true;
+                               break;
+                               case 'u':
+                                       if (*user->oper)
+                                               opt_unlimit = true;
+                               break;
+                               case 'r':
+                                       opt_realname = true;
+                               break;
+                               case 'm':
+                                       opt_mode = true;
+                               break;
                        }
-                       return;
+
+                       *iter++;
                }
-               if (parameters[0][0] == '#')
+       }
+
+
+       /* who on a channel? */
+       ch = ServerInstance->FindChan(matchtext);
+
+       if ((ch) && (CanView(ch,user)))
+       {
+               /* who on a channel. */
+               CUList *cu = ch->GetUsers();
+
+               for (CUList::iterator i = cu->begin(); i != cu->end(); i++)
                {
-                       Ptr = FindChan(parameters[0]);
-                       if (Ptr)
+                       /* opers only, please */
+                       if (opt_viewopersonly && !*(i->second)->oper)
+                               continue;
+
+                       SendWhoLine(user, initial, ch, i->second, whoresults);
+               }
+       }
+       else
+       {
+               /* Match against wildcard of nick, server or host */
+
+               if (opt_viewopersonly)
+               {
+                       /* Showing only opers */
+                       for (std::vector<userrec*>::iterator i = ServerInstance->all_opers.begin(); i != ServerInstance->all_opers.end(); i++)
                        {
-                               int n_list = 0;
-                               for (user_hash::const_iterator i = clientlist.begin(); i != clientlist.end(); i++)
+                               userrec* oper = *i;
+
+                               if (whomatch(oper, matchtext, opt_realname, opt_showrealhost, opt_mode))
                                {
-                                       if ((has_channel(i->second,Ptr)) && (isnick(i->second->nick)))
-                                       {
-                                               // Fix Bug #29 - Part 2..
-                                               *tmp = 0;
-                                               if (*i->second->awaymsg) {
-                                                       strlcat(tmp, "G", 9);
-                                               } else {
-                                                       strlcat(tmp, "H", 9);
-                                               }
-                                               if (strchr(i->second->modes,'o')) { strlcat(tmp, "*", 9); }
-                                               strlcat(tmp, cmode(i->second, Ptr),5);
-                                               WriteServ(user->fd,"352 %s %s %s %s %s %s %s :0 %s",user->nick, Ptr->name, i->second->ident, i->second->dhost, i->second->server, i->second->nick, tmp, i->second->fullname);
-                                                n_list++;
-                                                if (n_list > Config->MaxWhoResults)
-                                                {
-                                                        WriteServ(user->fd,"523 %s WHO :Command aborted: More results than configured limit",user->nick);
-                                                        break;
-                                                }
-
-                                       }
+                                       SendWhoLine(user, initial, NULL, oper, whoresults);
                                }
-                               WriteServ(user->fd,"315 %s %s :End of /WHO list.",user->nick, parameters[0]);
-                       }
-                       else
-                       {
-                               WriteServ(user->fd,"401 %s %s :No such nick/channel",user->nick, parameters[0]);
                        }
                }
                else
                {
-                       userrec* u = Find(parameters[0]);
-                       if (u)
+                       for (user_hash::iterator i = ServerInstance->clientlist->begin(); i != ServerInstance->clientlist->end(); i++)
                        {
-                               // Bug Fix #29 -- Part 29..
-                               *tmp = 0;
-                               if (*u->awaymsg) {
-                                       strlcat(tmp, "G" ,9);
-                               } else {
-                                       strlcat(tmp, "H" ,9);
+                               if (whomatch(i->second, matchtext, opt_realname, opt_showrealhost, opt_mode))
+                               {
+                                       SendWhoLine(user, initial, NULL, i->second, whoresults);
                                }
-                               if (strchr(u->modes,'o')) { strlcat(tmp, "*" ,9); }
-                               WriteServ(user->fd,"352 %s %s %s %s %s %s %s :0 %s",user->nick, u->chans.size() ? u->chans[0].channel->name
-                                : "*", u->ident, u->dhost, u->server, u->nick, tmp, u->fullname);
                        }
-                       WriteServ(user->fd,"315 %s %s :End of /WHO list.",user->nick, parameters[0]);
                }
        }
-       if (pcnt == 2)
+       /* Send the results out */
+       if ((whoresults.size() < (size_t)ServerInstance->Config->MaxWhoResults) && (!opt_unlimit))
        {
-                if ((!strcmp(parameters[0],"0")) || (!strcmp(parameters[0],"*")) && (!strcmp(parameters[1],"o")))
-                {
-                       for (std::vector<userrec*>::iterator i = all_opers.begin(); i != all_opers.end(); i++)
-                        {
-                               // If i were a rich man.. I wouldn't need to me making these bugfixes..
-                               // But i'm a poor bastard with nothing better to do.
-                               userrec* oper = *i;
-                               *tmp = 0;
-                               if (*oper->awaymsg) {
-                                       strlcat(tmp, "G" ,9);
-                               } else {
-                                       strlcat(tmp, "H" ,9);
-                               }
-                                WriteServ(user->fd,"352 %s %s %s %s %s %s %s* :0 %s", user->nick, oper->chans.size() ? oper->chans[0].channel->name 
-                               : "*", oper->ident, oper->dhost, oper->server, oper->nick, tmp, oper->fullname);
-                        }
-                        WriteServ(user->fd,"315 %s %s :End of /WHO list.",user->nick, parameters[0]);
-                        return;
-                }
+               for (std::vector<std::string>::const_iterator n = whoresults.begin(); n != whoresults.end(); n++)
+                       user->WriteServ(*n);
+               user->WriteServ("315 %s %s :End of /WHO list.",user->nick, *parameters[0] ? parameters[0] : "*");
+               return CMD_SUCCESS;
+       }
+       else
+       {
+               /* BZZT! Too many results. */
+               user->WriteServ("315 %s %s :Too many results",user->nick, parameters[0]);
+               return CMD_FAILURE;
        }
 }
-
-