summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--include/commands/cmd_who.h2
-rw-r--r--src/cmd_who.cpp125
-rw-r--r--src/command_parse.cpp10
3 files changed, 79 insertions, 58 deletions
diff --git a/include/commands/cmd_who.h b/include/commands/cmd_who.h
index c88233b1d..d76e92aec 100644
--- a/include/commands/cmd_who.h
+++ b/include/commands/cmd_who.h
@@ -31,7 +31,7 @@
class cmd_who : public command_t
{
public:
- cmd_who () : command_t("WHO",0,1) { syntax = "<server>|<nickname>|<channel> [O|o]"; }
+ cmd_who () : command_t("WHO",0,1) { syntax = "<server>|<nickname>|<channel>|<realname>|<host>|0 [ohur]"; }
void Handle(const char** parameters, int pcnt, userrec *user);
};
diff --git a/src/cmd_who.cpp b/src/cmd_who.cpp
index 01c7bfff5..aab1688ba 100644
--- a/src/cmd_who.cpp
+++ b/src/cmd_who.cpp
@@ -20,6 +20,7 @@
#include "modules.h"
#include "commands.h"
#include "helperfuncs.h"
+#include "wildcard.h"
#include "commands/cmd_who.h"
extern ServerConfig* Config;
@@ -44,6 +45,20 @@ static char *getlastchanname(userrec *u)
return "*";
}
+bool whomatch(userrec* user, const char* matchtext, bool opt_realname, bool opt_showrealhost)
+{
+ bool realhost = false;
+ bool realname = false;
+
+ 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)));
+}
+
void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
{
/*
@@ -53,12 +68,24 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
* Currently, we support WHO #chan, WHO nick, WHO 0, WHO *, and the addition of a 'o' flag, as per RFC.
*/
+ /* WHO options */
bool opt_viewopersonly = false;
+ bool opt_showrealhost = false;
+ bool opt_unlimit = false;
+ bool opt_realname = false;
+
chanrec *ch = NULL;
std::vector<std::string> whoresults;
std::string initial = "352 " + std::string(user->nick) + " ";
- if (pcnt == 2)
+ 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];
@@ -69,7 +96,18 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
{
case 'o':
opt_viewopersonly = true;
- break;
+ 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;
}
*iter++;
@@ -78,7 +116,7 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
/* who on a channel? */
- ch = FindChan(parameters[0]);
+ ch = FindChan(matchtext);
if (ch)
{
@@ -91,11 +129,10 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
if (opt_viewopersonly && !*(i->second)->oper)
continue;
-
/* XXX - code duplication; this could be more efficient -- w00t */
std::string wholine = initial;
- wholine = wholine + getlastchanname(i->second) + " " + i->second->ident + " " + i->second->dhost + " " +
+ wholine = wholine + getlastchanname(i->second) + " " + i->second->ident + " " + (opt_showrealhost ? i->second->host : i->second->dhost) + " " +
i->second->server + " " + i->second->nick + " ";
/* away? */
@@ -120,51 +157,20 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
}
else
{
- /* uhggle. who on .. something else. */
- userrec *u = Find(parameters[0]);
+ /* Match against wildcard of nick, server or host */
- if (u)
+ if (opt_viewopersonly)
{
- /* who on a single user */
- std::string wholine = initial;
-
- wholine = wholine + getlastchanname(u) + " " + u->ident + " " + u->dhost + " " +
- u->server + " " + u->nick + " ";
-
- /* away? */
- if (*u->awaymsg)
- {
- wholine.append("G");
- }
- else
+ /* Showing only opers */
+ for (std::vector<userrec*>::iterator i = all_opers.begin(); i != all_opers.end(); i++)
{
- wholine.append("H");
- }
+ userrec* oper = *i;
- /* oper? */
- if (*u->oper)
- {
- wholine.append("*");
- }
-
- wholine = wholine + cmode(u, ch) + " :0 " + u->fullname;
- whoresults.push_back(wholine);
- }
-
- if (*parameters[0] == '*' || *parameters[0] == '0')
- {
- if (!opt_viewopersonly && !*user->oper)
- return; /* No way, jose */
-
- if (opt_viewopersonly)
- {
- for (std::vector<userrec*>::iterator i = all_opers.begin(); i != all_opers.end(); i++)
+ if (whomatch(oper, matchtext, opt_realname, opt_showrealhost))
{
- userrec* oper = (userrec*)*i;
-
std::string wholine = initial;
-
- wholine = wholine + getlastchanname(oper) + " " + oper->ident + " " + oper->dhost + " " +
+
+ wholine = wholine + getlastchanname(oper) + " " + oper->ident + " " + (opt_showrealhost ? oper->host : oper->dhost) + " " +
oper->server + " " + oper->nick + " ";
/* away? */
@@ -176,26 +182,29 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
{
wholine.append("H");
}
-
+
/* oper? */
if (*oper->oper)
{
wholine.append("*");
}
-
+
wholine = wholine + cmode(oper, ch) + " :0 " + oper->fullname;
whoresults.push_back(wholine);
}
}
- else
+ }
+ else
+ {
+ for (user_hash::iterator i = clientlist.begin(); i != clientlist.end(); i++)
{
- for (user_hash::iterator i = clientlist.begin(); i != clientlist.end(); i++)
+ if (whomatch(i->second, matchtext, opt_realname, opt_showrealhost))
{
std::string wholine = initial;
-
- wholine = wholine + getlastchanname(i->second) + " " + i->second->ident + " " + i->second->dhost + " " +
- i->second->server + " " + i->second->nick + " ";
-
+
+ wholine = wholine + getlastchanname(i->second) + " " + i->second->ident + " " + (opt_showrealhost ? i->second->host : i->second->dhost) + " " +
+ i->second->server + " " + i->second->nick + " ";
+
/* away? */
if (*(i->second)->awaymsg)
{
@@ -218,4 +227,16 @@ void cmd_who::Handle (const char** parameters, int pcnt, userrec *user)
}
}
}
+ /* Send the results out */
+ if ((whoresults.size() < (size_t)Config->MaxWhoResults) && (!opt_unlimit))
+ {
+ for (std::vector<std::string>::const_iterator n = whoresults.begin(); n != whoresults.end(); n++)
+ WriteServ_NoFormat(user->fd,n->c_str());
+ WriteServ(user->fd,"315 %s %s :End of /WHO list.",user->nick, parameters[0]);
+ }
+ else
+ {
+ /* BZZT! Too many results. */
+ WriteServ(user->fd,"315 %s %s :Too many results",user->nick, parameters[0]);
+ }
}
diff --git a/src/command_parse.cpp b/src/command_parse.cpp
index 10fe6e1ee..386f09321 100644
--- a/src/command_parse.cpp
+++ b/src/command_parse.cpp
@@ -236,11 +236,11 @@ void CommandParser::ProcessCommand(userrec *user, std::string &cmd)
WriteServ(user->fd,"481 %s :Permission Denied- You do not have the required operator privilages",user->nick);
return;
}
- }
- if ((cm->second->flags_needed) && (!user->HasPermission(command)))
- {
- WriteServ(user->fd,"481 %s :Permission Denied- Oper type %s does not have access to command %s",user->nick,user->oper,command.c_str());
- return;
+ if (!user->HasPermission(command))
+ {
+ WriteServ(user->fd,"481 %s :Permission Denied- Oper type %s does not have access to command %s",user->nick,user->oper,command.c_str());
+ return;
+ }
}
if ((user->registered == REG_ALL) && (!*user->oper) && (cm->second->IsDisabled()))
{