include/command_parse.h

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
   6  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   7  *   Copyright (C) 2005-2007 Craig Edwards <craigedwards@brainbox.cc>
   8  *
   9  * This file is part of InspIRCd.  InspIRCd is free software: you can
  10  * redistribute it and/or modify it under the terms of the GNU General Public
  11  * License as published by the Free Software Foundation, version 2.
  12  *
  13  * This program is distributed in the hope that it will be useful, but WITHOUT
  14  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  15  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  16  * details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20  */
  21
  22
  23 #pragma once
  24
  25 /** This class handles command management and parsing.
  26  * It allows you to add and remove commands from the map,
  27  * call command handlers by name, and chop up comma seperated
  28  * parameters into multiple calls.
  29  */
  30 class CoreExport CommandParser
  31 {
  32  private:
  33         /** Process a command from a user.
  34          * @param user The user to parse the command for
  35          * @param cmd The command string to process
  36          */
  37         void ProcessCommand(LocalUser* user, std::string& cmd);
  38
  39  public:
  40         /** Command list, a hash_map of command names to Command*
  41          */
  42         Commandtable cmdlist;
  43
  44         /** Default constructor.
  45          */
  46         CommandParser();
  47
  48         /** Calls the handler for a given command.
  49          * @param commandname The command to find. This should be in uppercase.
  50          * @param parameters Parameter list
  51          * @param user The user to call the handler on behalf of
  52          * @param cmd If non-NULL and the command was executed it is set to the command handler,
  53          * otherwise it isn't written to.
  54          * @return This method will return CMD_SUCCESS if the command handler was found and called,
  55          * and the command completeld successfully. It will return CMD_FAILURE if the command handler was found
  56          * and called, but the command did not complete successfully, and it will return CMD_INVALID if the
  57          * command simply did not exist at all or the wrong number of parameters were given, or the user
  58          * was not privilaged enough to execute the command.
  59          */
  60         CmdResult CallHandler(const std::string& commandname, const std::vector<std::string>& parameters, User* user, Command** cmd = NULL);
  61
  62         /** Get the handler function for a command.
  63          * @param commandname The command required. Always use uppercase for this parameter.
  64          * @return a pointer to the command handler, or NULL
  65          */
  66         Command* GetHandler(const std::string &commandname);
  67
  68         /** LoopCall is used to call a command handler repeatedly based on the contents of a comma seperated list.
  69          * There are two ways to call this method, either with one potential list or with two potential lists.
  70          * We need to handle two potential lists for JOIN, because a JOIN may contain two lists of items at once:
  71          * the channel names and their keys as follows:
  72          *
  73          * JOIN \#chan1,\#chan2,\#chan3 key1,,key3
  74          *
  75          * Therefore, we need to deal with both lists concurrently. If there are two lists then the method reads
  76          * them both together until the first runs out of tokens.
  77          * With one list it is much simpler, and is used in NAMES, WHOIS, PRIVMSG etc.
  78          *
  79          * If there is only one list and there are duplicates in it, then the command handler is only called for
  80          * unique items. Entries are compared using "irc comparision" (see irc::string).
  81          * If the usemax parameter is true (the default) the function only parses until it reaches
  82          * ServerInstance->Config->MaxTargets number of targets, to stop abuse via spam.
  83          *
  84          * The OnPostCommand hook is executed for each item after it has been processed by the handler, with the
  85          * original line parameter being empty (to indicate that the command in that form was created by this function).
  86          * This only applies if the user executing the command is local.
  87          *
  88          * If there are two lists and the second list runs out of tokens before the first list then parameters[extra]
  89          * will be an EMPTY string when Handle() is called for the remaining tokens in the first list, even if it is
  90          * in the middle of parameters[]! Moreover, empty tokens in the second list are allowed, and those will also
  91          * result in the appropiate entry being empty in parameters[].
  92          * This is different than what command handlers usually expect; the command parser only allows an empty param
  93          * as the last item in the vector.
  94          *
  95          * @param user The user who sent the command
  96          * @param handler The command handler to call for each parameter in the list
  97          * @param parameters Parameter list as a vector of strings
  98          * @param splithere The first parameter index to split as a comma seperated list
  99          * @param extra The second parameter index to split as a comma seperated list, or -1 (the default) if there is only one list
 100          * @param usemax True to limit the command to MaxTargets targets (default), or false to process all tokens
 101          * @return This function returns true when it identified a list in the given parameter and finished calling the
 102          * command handler for each entry on the list. When this occurs, the caller should return without doing anything,
 103          * otherwise it should continue into its main section of code.
 104          */
 105         static bool LoopCall(User* user, Command* handler, const std::vector<std::string>& parameters, unsigned int splithere, int extra = -1, bool usemax = true);
 106
 107         /** Take a raw input buffer from a recvq, and process it on behalf of a user.
 108          * @param buffer The buffer line to process
 109          * @param user The user to whom this line belongs
 110          */
 111         void ProcessBuffer(std::string &buffer,LocalUser *user);
 112
 113         /** Add a new command to the commands hash
 114          * @param f The new Command to add to the list
 115          * @return True if the command was added
 116          */
 117         bool AddCommand(Command *f);
 118
 119         /** Removes a command.
 120          */
 121         void RemoveCommand(Command* x);
 122
 123         /** Translate a single item based on the TranslationType given.
 124          * @param to The translation type to use for the process
 125          * @param item The input string
 126          * @param dest The output string. The translation result will be appended to this string
 127          * @param custom_translator Used to translate the parameter if the translation type is TR_CUSTOM, if NULL, TR_CUSTOM will act like TR_TEXT
 128          * @param paramnumber The index of the parameter we are translating.
 129          */
 130         static void TranslateSingleParam(TranslateType to, const std::string& item, std::string& dest, CommandBase* custom_translator = NULL, unsigned int paramnumber = 0);
 131
 132         /** Translate nicknames in a list of strings into UIDs, based on the TranslateTypes given.
 133          * @param to The translation types to use for the process. If this list is too short, TR_TEXT is assumed for the rest.
 134          * @param source The strings to translate
 135          * @param prefix_final True if the final source argument should have a colon prepended (if it could contain a space)
 136          * @param custom_translator Used to translate the parameter if the translation type is TR_CUSTOM, if NULL, TR_CUSTOM will act like TR_TEXT
 137          * @return dest The output string
 138          */
 139         static std::string TranslateUIDs(const std::vector<TranslateType>& to, const std::vector<std::string>& source, bool prefix_final = false, CommandBase* custom_translator = NULL);
 140 };
 141
 142 /** A lookup table of values for multiplier characters used by
 143  * InspIRCd::Duration(). In this lookup table, the indexes for
 144  * the ascii values 'm' and 'M' have the value '60', the indexes
 145  * for the ascii values 'D' and 'd' have a value of '86400', etc.
 146  */
 147 const int duration_multi[] =
 148 {
 149         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 150         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 151         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 152         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 153         1, 1, 1, 1, 1, 1, 1, 1, 86400, 1, 1, 1, 3600,
 154         1, 1, 1, 1, 60, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 155         604800, 1, 31557600, 1, 1, 1, 1, 1, 1, 1, 1,
 156         1, 1, 86400, 1, 1, 1, 3600, 1, 1, 1, 1, 60,
 157         1, 1, 1, 1, 1, 1, 1, 1, 1, 604800, 1, 31557600,
 158         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 159         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 160         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 161         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 162         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 163         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 164         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 165         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 166         1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1
 167 };