X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fcommand_parse.cpp;h=eeb4b3e9fb0ea2b58b05608e7523eb32893b5aac;hb=da29af8cba49d51e53d6e68237ccbf6370b6dd1f;hp=bf2e61ce974c8f4d79a4dde41d43a0a6e57bf23a;hpb=bab14f0dd2345c9d7dcbc47c918563709e1ac094;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/command_parse.cpp b/src/command_parse.cpp index bf2e61ce9..514394a57 100644 --- a/src/command_parse.cpp +++ b/src/command_parse.cpp @@ -1 +1,473 @@ -/* +------------------------------------+ * | 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 "configreader.h" #include #include "users.h" #include "modules.h" #include "wildcard.h" #include "xline.h" #include "socketengine.h" #include "socket.h" #include "command_parse.h" /* Directory Searching for Unix-Only */ #ifndef WIN32 #include #include #endif bool InspIRCd::ULine(const char* server) { if (!server) return false; if (!*server) return true; return (Config->ulines.find(server) != Config->ulines.end()); } bool InspIRCd::SilentULine(const char* server) { std::map::iterator n = Config->ulines.find(server); if (n != Config->ulines.end()) return n->second; else return false; } int InspIRCd::OperPassCompare(const char* data,const char* input, int tagnumber) { int MOD_RESULT = 0; FOREACH_RESULT_I(this,I_OnOperCompare,OnOperCompare(data, input, tagnumber)) if (MOD_RESULT == 1) return 0; if (MOD_RESULT == -1) return 1; return strcmp(data,input); } std::string InspIRCd::TimeString(time_t curtime) { return std::string(ctime(&curtime),24); } /** Refactored by Brain, Jun 2007. Much faster with some clever O(1) array * lookups and pointer maths. */ long InspIRCd::Duration(const std::string &str) { unsigned char multiplier = 0; long total = 0; long times = 1; long subtotal = 0; /* Iterate each item in the string, looking for number or multiplier */ for (std::string::const_reverse_iterator i = str.rbegin(); i != str.rend(); ++i) { /* Found a number, queue it onto the current number */ if ((*i >= '0') && (*i <= '9')) { subtotal = subtotal + ((*i - '0') * times); times = times * 10; } else { /* Found something thats not a number, find out how much * it multiplies the built up number by, multiply the total * and reset the built up number. */ if (subtotal) total += subtotal * duration_multi[multiplier]; /* Next subtotal please */ subtotal = 0; multiplier = *i; times = 1; } } if (multiplier) { total += subtotal * duration_multi[multiplier]; subtotal = 0; } /* Any trailing values built up are treated as raw seconds */ return total + subtotal; } /* LoopCall is used to call a command classes handler repeatedly based on the contents of a comma seperated list. * There are two overriden versions of this method, one of which takes two potential lists and the other takes one. * We need a version which takes two potential lists for JOIN, because a JOIN may contain two lists of items at once, * the channel names and their keys as follows: * JOIN #chan1,#chan2,#chan3 key1,,key3 * Therefore, we need to deal with both lists concurrently. The first instance of this method does that by creating * two instances of irc::commasepstream and reading them both together until the first runs out of tokens. * The second version is much simpler and just has the one stream to read, and is used in NAMES, WHOIS, PRIVMSG etc. * Both will only parse until they reach ServerInstance->Config->MaxTargets number of targets, to stop abuse via spam. */ int CommandParser::LoopCall(userrec* user, command_t* CommandObj, const char** parameters, int pcnt, unsigned int splithere, unsigned int extra) { /* First check if we have more than one item in the list, if we don't we return zero here and the handler * which called us just carries on as it was. */ if (!strchr(parameters[splithere],',')) return 0; /** Some lame ircds will weed out dupes using some shitty O(n^2) algorithm. * By using std::map (thanks for the idea w00t) we can cut this down a ton. * ...VOOODOOOO! */ std::map dupes; /* Create two lists, one for channel names, one for keys */ irc::commasepstream items1(parameters[splithere]); irc::commasepstream items2(parameters[extra]); std::string item("*"); unsigned int max = 0; /* Attempt to iterate these lists and call the command objech * which called us, for every parameter pair until there are * no more left to parse. */ while (((item = items1.GetToken()) != "") && (max++ < ServerInstance->Config->MaxTargets)) { if (dupes.find(item.c_str()) == dupes.end()) { const char* new_parameters[127]; for (int t = 0; (t < pcnt) && (t < 127); t++) new_parameters[t] = parameters[t]; std::string extrastuff = items2.GetToken(); new_parameters[splithere] = item.c_str(); new_parameters[extra] = extrastuff.c_str(); CommandObj->Handle(new_parameters,pcnt,user); dupes[item.c_str()] = true; } } return 1; } int CommandParser::LoopCall(userrec* user, command_t* CommandObj, const char** parameters, int pcnt, unsigned int splithere) { /* First check if we have more than one item in the list, if we don't we return zero here and the handler * which called us just carries on as it was. */ if (!strchr(parameters[splithere],',')) return 0; std::map dupes; /* Only one commasepstream here */ irc::commasepstream items1(parameters[splithere]); std::string item("*"); unsigned int max = 0; /* Parse the commasepstream until there are no tokens remaining. * Each token we parse out, call the command handler that called us * with it */ while (((item = items1.GetToken()) != "") && (max++ < ServerInstance->Config->MaxTargets)) { if (dupes.find(item.c_str()) == dupes.end()) { const char* new_parameters[127]; for (int t = 0; (t < pcnt) && (t < 127); t++) new_parameters[t] = parameters[t]; new_parameters[splithere] = item.c_str(); parameters[splithere] = item.c_str(); /* Execute the command handler over and over. If someone pulls our user * record out from under us (e.g. if we /kill a comma sep list, and we're * in that list ourselves) abort if we're gone. */ CommandObj->Handle(new_parameters,pcnt,user); dupes[item.c_str()] = true; } } /* By returning 1 we tell our caller that nothing is to be done, * as all the previous calls handled the data. This makes the parent * return without doing any processing. */ return 1; } bool CommandParser::IsValidCommand(const std::string &commandname, int pcnt, userrec * user) { command_table::iterator n = cmdlist.find(commandname); if (n != cmdlist.end()) { if ((pcnt>=n->second->min_params) && (n->second->source != "")) { if ((!n->second->flags_needed) || (user->modes[n->second->flags_needed-65])) { if (n->second->flags_needed) { return ((user->HasPermission(commandname)) || (ServerInstance->ULine(user->server))); } return true; } } } return false; } command_t* CommandParser::GetHandler(const std::string &commandname) { command_table::iterator n = cmdlist.find(commandname); if (n != cmdlist.end()) return n->second; return NULL; } // calls a handler function for a command CmdResult CommandParser::CallHandler(const std::string &commandname,const char** parameters, int pcnt, userrec *user) { command_table::iterator n = cmdlist.find(commandname); if (n != cmdlist.end()) { if (pcnt >= n->second->min_params) { if ((!n->second->flags_needed) || (user->modes[n->second->flags_needed-65])) { if (n->second->flags_needed) { if ((user->HasPermission(commandname)) || (!IS_LOCAL(user))) { return n->second->Handle(parameters,pcnt,user); } } else { return n->second->Handle(parameters,pcnt,user); } } } } return CMD_INVALID; } void CommandParser::ProcessCommand(userrec *user, std::string &cmd) { const char *command_p[127]; int items = 0; irc::tokenstream tokens(cmd); std::string command; tokens.GetToken(command); /* A client sent a nick prefix on their command (ick) * rhapsody and some braindead bouncers do this -- * the rfc says they shouldnt but also says the ircd should * discard it if they do. */ if (*command.c_str() == ':') tokens.GetToken(command); while (tokens.GetToken(para[items]) && (items < 127)) { command_p[items] = para[items].c_str(); items++; } std::transform(command.begin(), command.end(), command.begin(), ::toupper); int MOD_RESULT = 0; FOREACH_RESULT(I_OnPreCommand,OnPreCommand(command,command_p,items,user,false,cmd)); if (MOD_RESULT == 1) { return; } command_table::iterator cm = cmdlist.find(command); if (cm != cmdlist.end()) { if (user) { /* activity resets the ping pending timer */ user->nping = ServerInstance->Time() + user->pingmax; if (cm->second->flags_needed) { if (!user->IsModeSet(cm->second->flags_needed)) { user->WriteServ("481 %s :Permission Denied - You do not have the required operator privileges",user->nick); return; } if (!user->HasPermission(command)) { user->WriteServ("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) && (!IS_OPER(user)) && (cm->second->IsDisabled())) { /* command is disabled! */ user->WriteServ("421 %s %s :This command has been disabled.",user->nick,command.c_str()); return; } if (items < cm->second->min_params) { user->WriteServ("461 %s %s :Not enough parameters.", user->nick, command.c_str()); /* If syntax is given, display this as the 461 reply */ if ((ServerInstance->Config->SyntaxHints) && (user->registered == REG_ALL) && (cm->second->syntax.length())) user->WriteServ("304 %s :SYNTAX %s %s", user->nick, cm->second->command.c_str(), cm->second->syntax.c_str()); return; } if ((user->registered == REG_ALL) || (cm->second->WorksBeforeReg())) { /* ikky /stats counters */ cm->second->use_count++; cm->second->total_bytes += cmd.length(); int MOD_RESULT = 0; FOREACH_RESULT(I_OnPreCommand,OnPreCommand(command,command_p,items,user,true,cmd)); if (MOD_RESULT == 1) return; /* * WARNING: nothing may come after the * command handler call, as the handler * may free the user structure! */ CmdResult result = cm->second->Handle(command_p,items,user); FOREACH_MOD(I_OnPostCommand,OnPostCommand(command, command_p, items, user, result,cmd)); return; } else { user->WriteServ("451 %s :You have not registered",command.c_str()); return; } } } else if (user) { ServerInstance->stats->statsUnknown++; user->WriteServ("421 %s %s :Unknown command",user->nick,command.c_str()); } } bool CommandParser::RemoveCommands(const char* source) { command_table::iterator i,safei; for (i = cmdlist.begin(); i != cmdlist.end(); i++) { safei = i; safei++; if (safei != cmdlist.end()) { RemoveCommand(safei, source); } } safei = cmdlist.begin(); if (safei != cmdlist.end()) { RemoveCommand(safei, source); } return true; } void CommandParser::RemoveCommand(command_table::iterator safei, const char* source) { command_t* x = safei->second; if (x->source == std::string(source)) { cmdlist.erase(safei); delete x; } } void CommandParser::ProcessBuffer(std::string &buffer,userrec *user) { std::string::size_type a; if (!user) return; while ((a = buffer.rfind("\n")) != std::string::npos) buffer.erase(a); while ((a = buffer.rfind("\r")) != std::string::npos) buffer.erase(a); if (buffer.length()) { if (!user->muted) { ServerInstance->Log(DEBUG,"C[%d] -> :%s %s",user->GetFd(), user->nick, buffer.c_str()); this->ProcessCommand(user,buffer); } } } bool CommandParser::CreateCommand(command_t *f, void* so_handle) { if (so_handle) { if (RFCCommands.find(f->command) == RFCCommands.end()) RFCCommands[f->command] = so_handle; else { ServerInstance->Log(DEFAULT,"ERK! Somehow, we loaded a cmd_*.so file twice! Only the first instance is being recorded."); return false; } } /* create the command and push it onto the table */ if (cmdlist.find(f->command) == cmdlist.end()) { cmdlist[f->command] = f; return true; } else return false; } CommandParser::CommandParser(InspIRCd* Instance) : ServerInstance(Instance) { para.resize(128); this->SetupCommandTable(); } bool CommandParser::FindSym(void** v, void* h) { *v = dlsym(h, "init_command"); const char* err = dlerror(); if (err && !(*v)) { ServerInstance->Log(SPARSE, "Error loading core command: %s\n", err); return false; } return true; } bool CommandParser::ReloadCommand(const char* cmd) { char filename[MAXBUF]; char commandname[MAXBUF]; int y = 0; for (const char* x = cmd; *x; x++, y++) commandname[y] = toupper(*x); commandname[y] = 0; SharedObjectList::iterator command = RFCCommands.find(commandname); if (command != RFCCommands.end()) { command_t* cmdptr = cmdlist.find(commandname)->second; cmdlist.erase(cmdlist.find(commandname)); for (char* x = commandname; *x; x++) *x = tolower(*x); delete cmdptr; dlclose(command->second); RFCCommands.erase(command); snprintf(filename, MAXBUF, "cmd_%s.so", commandname); this->LoadCommand(filename); return true; } return false; } CmdResult cmd_reload::Handle(const char** parameters, int pcnt, userrec *user) { user->WriteServ("NOTICE %s :*** Reloading command '%s'",user->nick, parameters[0]); if (ServerInstance->Parser->ReloadCommand(parameters[0])) { user->WriteServ("NOTICE %s :*** Successfully reloaded command '%s'", user->nick, parameters[0]); ServerInstance->WriteOpers("*** RELOAD: %s reloaded the '%s' command.", user->nick, parameters[0]); return CMD_SUCCESS; } else { user->WriteServ("NOTICE %s :*** Could not reload command '%s'", user->nick, parameters[0]); return CMD_FAILURE; } } void CommandParser::LoadCommand(const char* name) { char filename[MAXBUF]; void* h; command_t* (*cmd_factory_func)(InspIRCd*); snprintf(filename, MAXBUF, "%s/%s", LIBRARYDIR, name); h = dlopen(filename, RTLD_NOW | RTLD_GLOBAL); if (!h) { ServerInstance->Log(SPARSE, "Error loading core command: %s", dlerror()); return; } if (this->FindSym((void **)&cmd_factory_func, h)) { command_t* newcommand = cmd_factory_func(ServerInstance); this->CreateCommand(newcommand, h); } } void CommandParser::SetupCommandTable() { RFCCommands.clear(); printf("\nLoading core commands"); fflush(stdout); DIR* library = opendir(LIBRARYDIR); if (library) { dirent* entry = NULL; while ((entry = readdir(library))) { if (match(entry->d_name, "cmd_*.so")) { printf("."); fflush(stdout); this->LoadCommand(entry->d_name); } } closedir(library); printf("\n"); } this->CreateCommand(new cmd_reload(ServerInstance)); } \ No newline at end of file +/* + * InspIRCd -- Internet Relay Chat Daemon + * + * Copyright (C) 2009-2010 Daniel De Graaf + * Copyright (C) 2006-2008 Robin Burchell + * Copyright (C) 2008 Thomas Stagner + * Copyright (C) 2005-2008 Craig Edwards + * Copyright (C) 2006-2007 Dennis Friis + * + * This file is part of InspIRCd. InspIRCd is free software: you can + * redistribute it and/or modify it under the terms of the GNU General Public + * License as published by the Free Software Foundation, version 2. + * + * This program is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more + * details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + + +#include "inspircd.h" + +bool InspIRCd::PassCompare(Extensible* ex, const std::string& data, const std::string& input, const std::string& hashtype) +{ + ModResult res; + FIRST_MOD_RESULT(OnPassCompare, res, (ex, data, input, hashtype)); + + /* Module matched */ + if (res == MOD_RES_ALLOW) + return true; + + /* Module explicitly didnt match */ + if (res == MOD_RES_DENY) + return false; + + /* We dont handle any hash types except for plaintext - Thanks tra26 */ + if (!hashtype.empty() && hashtype != "plaintext") + return false; + + return TimingSafeCompare(data, input); +} + +bool CommandParser::LoopCall(User* user, Command* handler, const std::vector& parameters, unsigned int splithere, int extra, bool usemax) +{ + if (splithere >= parameters.size()) + return false; + + /* First check if we have more than one item in the list, if we don't we return false here and the handler + * which called us just carries on as it was. + */ + if (parameters[splithere].find(',') == std::string::npos) + return false; + + /** Some lame ircds will weed out dupes using some shitty O(n^2) algorithm. + * By using std::set (thanks for the idea w00t) we can cut this down a ton. + * ...VOOODOOOO! + * + * Only check for duplicates if there is one list (allow them in JOIN). + */ + insp::flat_set dupes; + bool check_dupes = (extra < 0); + + /* Create two sepstreams, if we have only one list, then initialize the second sepstream with + * an empty string. The second parameter of the constructor of the sepstream tells whether + * or not to allow empty tokens. + * We allow empty keys, so "JOIN #a,#b ,bkey" will be interpreted as "JOIN #a", "JOIN #b bkey" + */ + irc::commasepstream items1(parameters[splithere]); + irc::commasepstream items2(extra >= 0 ? parameters[extra] : "", true); + std::string item; + unsigned int max = 0; + LocalUser* localuser = IS_LOCAL(user); + + /* Attempt to iterate these lists and call the command handler + * for every parameter or parameter pair until there are no more + * left to parse. + */ + while (items1.GetToken(item) && (!usemax || max++ < ServerInstance->Config->MaxTargets)) + { + if ((!check_dupes) || (dupes.insert(item.c_str()).second)) + { + std::vector new_parameters(parameters); + new_parameters[splithere] = item; + + if (extra >= 0) + { + // If we have two lists then get the next item from the second list. + // In case it runs out of elements then 'item' will be an empty string. + items2.GetToken(item); + new_parameters[extra] = item; + } + + CmdResult result = handler->Handle(new_parameters, user); + if (localuser) + { + // Run the OnPostCommand hook with the last parameter (original line) being empty + // to indicate that the command had more targets in its original form. + item.clear(); + FOREACH_MOD(OnPostCommand, (handler, new_parameters, localuser, result, item)); + } + } + } + + return true; +} + +Command* CommandParser::GetHandler(const std::string &commandname) +{ + CommandMap::iterator n = cmdlist.find(commandname); + if (n != cmdlist.end()) + return n->second; + + return NULL; +} + +// calls a handler function for a command + +CmdResult CommandParser::CallHandler(const std::string& commandname, const std::vector& parameters, User* user, Command** cmd) +{ + CommandMap::iterator n = cmdlist.find(commandname); + + if (n != cmdlist.end()) + { + if ((!parameters.empty()) && (parameters.back().empty()) && (!n->second->allow_empty_last_param)) + return CMD_INVALID; + + if (parameters.size() >= n->second->min_params) + { + bool bOkay = false; + + if (IS_LOCAL(user) && n->second->flags_needed) + { + /* if user is local, and flags are needed .. */ + + if (user->IsModeSet(n->second->flags_needed)) + { + /* if user has the flags, and now has the permissions, go ahead */ + if (user->HasPermission(commandname)) + bOkay = true; + } + } + else + { + /* remote or no flags required anyway */ + bOkay = true; + } + + if (bOkay) + { + if (cmd) + *cmd = n->second; + return n->second->Handle(parameters,user); + } + } + } + return CMD_INVALID; +} + +void CommandParser::ProcessCommand(LocalUser *user, std::string &cmd) +{ + std::vector command_p; + irc::tokenstream tokens(cmd); + std::string command, token; + tokens.GetToken(command); + + /* A client sent a nick prefix on their command (ick) + * rhapsody and some braindead bouncers do this -- + * the rfc says they shouldnt but also says the ircd should + * discard it if they do. + */ + if (command[0] == ':') + tokens.GetToken(command); + + while (tokens.GetToken(token)) + command_p.push_back(token); + + std::transform(command.begin(), command.end(), command.begin(), ::toupper); + + /* find the command, check it exists */ + Command* handler = GetHandler(command); + + // Penalty to give if the command fails before the handler is executed + unsigned int failpenalty = 0; + + /* Modify the user's penalty regardless of whether or not the command exists */ + if (!user->HasPrivPermission("users/flood/no-throttle")) + { + // If it *doesn't* exist, give it a slightly heftier penalty than normal to deter flooding us crap + unsigned int penalty = (handler ? handler->Penalty * 1000 : 2000); + user->CommandFloodPenalty += penalty; + + // Increase their penalty later if we fail and the command has 0 penalty by default (i.e. in Command::Penalty) to + // throttle sending ERR_* from the command parser. If the command does have a non-zero penalty then this is not + // needed because we've increased their penalty above. + if (penalty == 0) + failpenalty = 1000; + } + + if (!handler) + { + ModResult MOD_RESULT; + FIRST_MOD_RESULT(OnPreCommand, MOD_RESULT, (command, command_p, user, false, cmd)); + if (MOD_RESULT == MOD_RES_DENY) + return; + + /* + * This double lookup is in case a module (abbreviation) wishes to change a command. + * Sure, the double lookup is a bit painful, but bear in mind this only happens for unknowns anyway. + * + * Thanks dz for making me actually understand why this is necessary! + * -- w00t + */ + handler = GetHandler(command); + if (!handler) + { + if (user->registered == REG_ALL) + user->WriteNumeric(ERR_UNKNOWNCOMMAND, command, "Unknown command"); + ServerInstance->stats.Unknown++; + return; + } + } + + // If we were given more parameters than max_params then append the excess parameter(s) + // to command_p[maxparams-1], i.e. to the last param that is still allowed + if (handler->max_params && command_p.size() > handler->max_params) + { + /* + * command_p input (assuming max_params 1): + * this + * is + * a + * test + */ + + // Iterator to the last parameter that will be kept + const std::vector::iterator lastkeep = command_p.begin() + (handler->max_params - 1); + // Iterator to the first excess parameter + const std::vector::iterator firstexcess = lastkeep + 1; + + // Append all excess parameter(s) to the last parameter, seperated by spaces + for (std::vector::const_iterator i = firstexcess; i != command_p.end(); ++i) + { + lastkeep->push_back(' '); + lastkeep->append(*i); + } + + // Erase the excess parameter(s) + command_p.erase(firstexcess, command_p.end()); + } + + /* + * We call OnPreCommand here seperately if the command exists, so the magic above can + * truncate to max_params if necessary. -- w00t + */ + ModResult MOD_RESULT; + FIRST_MOD_RESULT(OnPreCommand, MOD_RESULT, (command, command_p, user, false, cmd)); + if (MOD_RESULT == MOD_RES_DENY) + return; + + /* activity resets the ping pending timer */ + user->nping = ServerInstance->Time() + user->MyClass->GetPingTime(); + + if (handler->flags_needed) + { + if (!user->IsModeSet(handler->flags_needed)) + { + user->CommandFloodPenalty += failpenalty; + user->WriteNumeric(ERR_NOPRIVILEGES, "Permission Denied - You do not have the required operator privileges"); + return; + } + + if (!user->HasPermission(command)) + { + user->CommandFloodPenalty += failpenalty; + user->WriteNumeric(ERR_NOPRIVILEGES, InspIRCd::Format("Permission Denied - Oper type %s does not have access to command %s", + user->oper->name.c_str(), command.c_str())); + return; + } + } + + if ((user->registered == REG_ALL) && (!user->IsOper()) && (handler->IsDisabled())) + { + /* command is disabled! */ + user->CommandFloodPenalty += failpenalty; + if (ServerInstance->Config->DisabledDontExist) + { + user->WriteNumeric(ERR_UNKNOWNCOMMAND, command, "Unknown command"); + } + else + { + user->WriteNumeric(ERR_UNKNOWNCOMMAND, command, "This command has been disabled."); + } + + ServerInstance->SNO->WriteToSnoMask('a', "%s denied for %s (%s@%s)", + command.c_str(), user->nick.c_str(), user->ident.c_str(), user->host.c_str()); + return; + } + + if ((!command_p.empty()) && (command_p.back().empty()) && (!handler->allow_empty_last_param)) + command_p.pop_back(); + + if (command_p.size() < handler->min_params) + { + user->CommandFloodPenalty += failpenalty; + user->WriteNumeric(ERR_NEEDMOREPARAMS, command, "Not enough parameters."); + if ((ServerInstance->Config->SyntaxHints) && (user->registered == REG_ALL) && (handler->syntax.length())) + user->WriteNumeric(RPL_SYNTAX, InspIRCd::Format("SYNTAX %s %s", handler->name.c_str(), handler->syntax.c_str())); + return; + } + + if ((user->registered != REG_ALL) && (!handler->WorksBeforeReg())) + { + user->CommandFloodPenalty += failpenalty; + user->WriteNumeric(ERR_NOTREGISTERED, command, "You have not registered"); + } + else + { + /* passed all checks.. first, do the (ugly) stats counters. */ + handler->use_count++; + + /* module calls too */ + FIRST_MOD_RESULT(OnPreCommand, MOD_RESULT, (command, command_p, user, true, cmd)); + if (MOD_RESULT == MOD_RES_DENY) + return; + + /* + * WARNING: be careful, the user may be deleted soon + */ + CmdResult result = handler->Handle(command_p, user); + + FOREACH_MOD(OnPostCommand, (handler, command_p, user, result, cmd)); + } +} + +void CommandParser::RemoveCommand(Command* x) +{ + CommandMap::iterator n = cmdlist.find(x->name); + if (n != cmdlist.end() && n->second == x) + cmdlist.erase(n); +} + +CommandBase::CommandBase(Module* mod, const std::string& cmd, unsigned int minpara, unsigned int maxpara) + : ServiceProvider(mod, cmd, SERVICE_COMMAND) + , flags_needed(0) + , min_params(minpara) + , max_params(maxpara) + , use_count(0) + , disabled(false) + , works_before_reg(false) + , allow_empty_last_param(true) + , Penalty(1) +{ +} + +CommandBase::~CommandBase() +{ +} + +void CommandBase::EncodeParameter(std::string& parameter, int index) +{ +} + +RouteDescriptor CommandBase::GetRouting(User* user, const std::vector& parameters) +{ + return ROUTE_LOCALONLY; +} + +Command::Command(Module* mod, const std::string& cmd, unsigned int minpara, unsigned int maxpara) + : CommandBase(mod, cmd, minpara, maxpara) + , force_manual_route(false) +{ +} + +Command::~Command() +{ + ServerInstance->Parser.RemoveCommand(this); +} + +void Command::RegisterService() +{ + if (!ServerInstance->Parser.AddCommand(this)) + throw ModuleException("Command already exists: " + name); +} + +void CommandParser::ProcessBuffer(std::string &buffer,LocalUser *user) +{ + if (buffer.empty()) + return; + + ServerInstance->Logs->Log("USERINPUT", LOG_RAWIO, "C[%s] I :%s %s", + user->uuid.c_str(), user->nick.c_str(), buffer.c_str()); + ProcessCommand(user,buffer); +} + +bool CommandParser::AddCommand(Command *f) +{ + /* create the command and push it onto the table */ + if (cmdlist.find(f->name) == cmdlist.end()) + { + cmdlist[f->name] = f; + return true; + } + return false; +} + +CommandParser::CommandParser() +{ +} + +std::string CommandParser::TranslateUIDs(const std::vector& to, const std::vector& source, bool prefix_final, CommandBase* custom_translator) +{ + std::vector::const_iterator types = to.begin(); + std::string dest; + + for (unsigned int i = 0; i < source.size(); i++) + { + TranslateType t = TR_TEXT; + // They might supply less translation types than parameters, + // in that case pretend that all remaining types are TR_TEXT + if (types != to.end()) + { + t = *types; + types++; + } + + bool last = (i == (source.size() - 1)); + if (prefix_final && last) + dest.push_back(':'); + + TranslateSingleParam(t, source[i], dest, custom_translator, i); + + if (!last) + dest.push_back(' '); + } + + return dest; +} + +void CommandParser::TranslateSingleParam(TranslateType to, const std::string& item, std::string& dest, CommandBase* custom_translator, unsigned int paramnumber) +{ + switch (to) + { + case TR_NICK: + { + /* Translate single nickname */ + User* user = ServerInstance->FindNick(item); + if (user) + dest.append(user->uuid); + else + dest.append(item); + break; + } + case TR_CUSTOM: + { + if (custom_translator) + { + std::string translated = item; + custom_translator->EncodeParameter(translated, paramnumber); + dest.append(translated); + break; + } + // If no custom translator was given, fall through + } + case TR_TEXT: + default: + /* Do nothing */ + dest.append(item); + break; + } +}