X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_cap.cpp;h=6b4387fdd96ccec55db075d0391be07a6478c02f;hb=fc4fc43ec232407b38d7ca182cb92c5cac4287aa;hp=564ed594bd7189a2d9ddd644b2de3ab4cd38920d;hpb=83137d5f9274832f152acc7d19de3cb0897f9630;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/modules/m_cap.cpp b/src/modules/m_cap.cpp index 564ed594b..6b4387fdd 100644 --- a/src/modules/m_cap.cpp +++ b/src/modules/m_cap.cpp @@ -1,112 +1,167 @@ -/* +------------------------------------+ - * | Inspire Internet Relay Chat Daemon | - * +------------------------------------+ +/* + * InspIRCd -- Internet Relay Chat Daemon + * + * Copyright (C) 2009 Daniel De Graaf + * Copyright (C) 2008 Craig Edwards * - * InspIRCd: (C) 2002-2008 InspIRCd Development Team - * See: http://www.inspircd.org/wiki/index.php/Credits + * 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 free but copyrighted software; see - * the file COPYING for details. + * 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" #include "m_cap.h" -/* $ModDesc: Provides a pointless /dalinfo command, demo module */ +/* $ModDesc: Provides the CAP negotiation mechanism seen in ratbox-derived ircds */ /* - * std::string type; - * std::vector parameters; - * User* user; - * Module* creator; - */ +CAP LS +:alfred.staticbox.net CAP * LS :multi-prefix sasl +CAP REQ :multi-prefix +:alfred.staticbox.net CAP * ACK :multi-prefix +CAP CLEAR +:alfred.staticbox.net CAP * ACK :-multi-prefix +CAP REQ :multi-prefix +:alfred.staticbox.net CAP * ACK :multi-prefix +CAP LIST +:alfred.staticbox.net CAP * LIST :multi-prefix +CAP END +*/ /** Handle /CAP */ class CommandCAP : public Command { - Module* Creator; public: - /* Command 'dalinfo', takes no parameters and needs no special modes */ - CommandCAP (InspIRCd* Instance, Module* mod) : Command(Instance,"CAP", 0, 1, true), Creator(mod) + LocalIntExt reghold; + CommandCAP (Module* mod) : Command(mod, "CAP", 1), + reghold("CAP_REGHOLD", mod) { - this->source = "m_cap.so"; + works_before_reg = true; } - CmdResult Handle (const char* const* parameters, int pcnt, User *user) + CmdResult Handle (const std::vector ¶meters, User *user) { - irc::string subcommand = parameters[0]; + irc::string subcommand = parameters[0].c_str(); + if (subcommand == "REQ") { - CapData Data; - Data.type = parameters[1]; - Data.user = user; - Data.creator = this->Creator; - Data.parameter = (pcnt > 1 ? parameters[1] : ""); - - user->Extend("CAP_REGHOLD"); - Event event((char*) &Data, (Module*)this->Creator, "cap_req"); - event.Send(this->ServerInstance); + if (parameters.size() < 2) + return CMD_FAILURE; + + CapEvent Data(creator, user, CapEvent::CAPEVENT_REQ); + + // tokenize the input into a nice list of requested caps + std::string cap_; + irc::spacesepstream cap_stream(parameters[1]); + + while (cap_stream.GetToken(cap_)) + { + // Whilst the handling of extraneous spaces is not currently defined in the CAP specification + // every single other implementation ignores extraneous spaces. Lets copy them for + // compatibility purposes. + trim(cap_); + if (!cap_.empty()) + Data.wanted.push_back(cap_); + } + + reghold.set(user, 1); + Data.Send(); + + if (Data.wanted.empty()) + { + user->WriteServ("CAP %s ACK :%s", user->nick.c_str(), parameters[1].c_str()); + return CMD_SUCCESS; + } + + // HACK: reset all of the caps which were enabled on this user because a cap request is atomic. + for (std::vector >::iterator iter = Data.changed.begin(); iter != Data.changed.end(); ++iter) + iter->first->ext.set(user, iter->second); + + user->WriteServ("CAP %s NAK :%s", user->nick.c_str(), parameters[1].c_str()); } else if (subcommand == "END") { - user->Shrink("CAP_REGHOLD"); + reghold.set(user, 0); + } + else if ((subcommand == "LS") || (subcommand == "LIST")) + { + CapEvent Data(creator, user, subcommand == "LS" ? CapEvent::CAPEVENT_LS : CapEvent::CAPEVENT_LIST); + + reghold.set(user, 1); + Data.Send(); + + std::string Result; + if (Data.wanted.size() > 0) + Result = irc::stringjoiner(" ", Data.wanted, 0, Data.wanted.size() - 1).GetJoined(); + + user->WriteServ("CAP %s %s :%s", user->nick.c_str(), subcommand.c_str(), Result.c_str()); } - else if (subcommand == "LS") + else if (subcommand == "CLEAR") { - CapData Data; - user->Extend("CAP_REGHOLD"); - Data.type = "LS"; - Data.user = user; - Data.creator = this->Creator; - Data.parameter.clear(); + CapEvent Data(creator, user, CapEvent::CAPEVENT_CLEAR); - Event event((char*) &Data, (Module*)this->Creator, "cap_ls"); - event.Send(this->ServerInstance); + reghold.set(user, 1); + Data.Send(); - user->WriteServ("CAP * LS :%s", Data.parameter.c_str()); + std::string Result; + if (!Data.ack.empty()) + Result = irc::stringjoiner(" ", Data.ack, 0, Data.ack.size() - 1).GetJoined(); + user->WriteServ("CAP %s ACK :%s", user->nick.c_str(), Result.c_str()); } else { - user->WriteServ("410 %s %s :Invalid CAP subcommand", user->nick, subcommand.c_str()); + user->WriteNumeric(ERR_INVALIDCAPSUBCOMMAND, "%s %s :Invalid CAP subcommand", user->nick.c_str(), subcommand.c_str()); + return CMD_FAILURE; } - return CMD_FAILURE; + + return CMD_SUCCESS; } }; class ModuleCAP : public Module { - CommandCAP* newcommand; + CommandCAP cmd; public: - ModuleCAP(InspIRCd* Me) - : Module(Me) + ModuleCAP() + : cmd(this) + { + } + + void init() { - // Create a new command - newcommand = new CommandCAP(ServerInstance, this); - ServerInstance->AddCommand(newcommand); + ServerInstance->Modules->AddService(cmd); + ServerInstance->Modules->AddService(cmd.reghold); - Implementation eventlist[] = { I_OnCheckReady, I_OnCleanup, I_OnUserDisconnect, I_OnRequest }; - ServerInstance->Modules->Attach(eventlist, this, 5); + Implementation eventlist[] = { I_OnCheckReady }; + ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation)); } - virtual bool OnCheckReady(User* user) + ModResult OnCheckReady(LocalUser* user) { /* Users in CAP state get held until CAP END */ - if (user->GetExt("CAP_REGHOLD")) - return true; + if (cmd.reghold.get(user)) + return MOD_RES_DENY; - return false; + return MOD_RES_PASSTHRU; } - virtual ~ModuleCAP() + ~ModuleCAP() { } - virtual Version GetVersion() + Version GetVersion() { - return Version(1, 1, 0, 0, VF_VENDOR, API_VERSION); + return Version("Client CAP extension support", VF_VENDOR); } };