src/modules/m_cap.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "m_cap.h"
  16
  17 /* $ModDesc: Provides the CAP negotiation mechanism seen in ratbox-derived ircds */
  18
  19 /*
  20 CAP LS
  21 :alfred.staticbox.net CAP * LS :multi-prefix sasl
  22 CAP REQ :multi-prefix
  23 :alfred.staticbox.net CAP * ACK :multi-prefix
  24 CAP CLEAR
  25 :alfred.staticbox.net CAP * ACK :-multi-prefix
  26 CAP REQ :multi-prefix
  27 :alfred.staticbox.net CAP * ACK :multi-prefix
  28 CAP LIST
  29 :alfred.staticbox.net CAP * LIST :multi-prefix
  30 CAP END
  31 */
  32
  33 /** Handle /CAP
  34  */
  35 class CommandCAP : public Command
  36 {
  37  public:
  38         LocalIntExt reghold;
  39         CommandCAP (Module* mod) : Command(mod, "CAP", 1),
  40                 reghold("CAP_REGHOLD", mod)
  41         {
  42                 works_before_reg = true;
  43         }
  44
  45         CmdResult Handle (const std::vector<std::string> &parameters, User *user)
  46         {
  47                 irc::string subcommand = parameters[0].c_str();
  48
  49                 if (subcommand == "REQ")
  50                 {
  51                         CapEvent Data(creator, "cap_req");
  52
  53                         Data.type = subcommand;
  54                         Data.user = user;
  55                         Data.creator = this->creator;
  56
  57                         if (parameters.size() < 2)
  58                                 return CMD_FAILURE;
  59
  60                         // tokenize the input into a nice list of requested caps
  61                         std::string param = parameters[1];
  62                         std::string cap_;
  63                         irc::spacesepstream cap_stream(param);
  64
  65                         while (cap_stream.GetToken(cap_))
  66                         {
  67                                 Data.wanted.push_back(cap_);
  68                         }
  69
  70                         reghold.set(user, 1);
  71                         Data.Send();
  72
  73                         if (Data.ack.size() > 0)
  74                         {
  75                                 std::string AckResult = irc::stringjoiner(" ", Data.ack, 0, Data.ack.size() - 1).GetJoined();
  76                                 user->WriteServ("CAP %s ACK :%s", user->nick.c_str(), AckResult.c_str());
  77                         }
  78
  79                         if (Data.wanted.size() > 0)
  80                         {
  81                                 std::string NakResult = irc::stringjoiner(" ", Data.wanted, 0, Data.wanted.size() - 1).GetJoined();
  82                                 user->WriteServ("CAP %s NAK :%s", user->nick.c_str(), NakResult.c_str());
  83                         }
  84                 }
  85                 else if (subcommand == "END")
  86                 {
  87                         reghold.set(user, 0);
  88                 }
  89                 else if ((subcommand == "LS") || (subcommand == "LIST"))
  90                 {
  91                         CapEvent Data(creator, subcommand == "LS" ? "cap_ls" : "cap_list");
  92
  93                         Data.type = subcommand;
  94                         Data.user = user;
  95                         Data.creator = this->creator;
  96
  97                         reghold.set(user, 1);
  98                         Data.Send();
  99
 100                         std::string Result;
 101                         if (Data.wanted.size() > 0)
 102                                 Result = irc::stringjoiner(" ", Data.wanted, 0, Data.wanted.size() - 1).GetJoined();
 103                         else
 104                                 Result = "";
 105
 106                         user->WriteServ("CAP %s %s :%s", user->nick.c_str(), subcommand.c_str(), Result.c_str());
 107                 }
 108                 else if (subcommand == "CLEAR")
 109                 {
 110                         CapEvent Data(creator, "cap_clear");
 111
 112                         Data.type = subcommand;
 113                         Data.user = user;
 114                         Data.creator = this->creator;
 115
 116                         reghold.set(user, 1);
 117                         Data.Send();
 118
 119                         std::string Result = irc::stringjoiner(" ", Data.ack, 0, Data.ack.size() - 1).GetJoined();
 120                         user->WriteServ("CAP %s ACK :%s", user->nick.c_str(), Result.c_str());
 121                 }
 122                 else
 123                 {
 124                         user->WriteNumeric(ERR_INVALIDCAPSUBCOMMAND, "%s %s :Invalid CAP subcommand", user->nick.c_str(), subcommand.c_str());
 125                 }
 126
 127                 return CMD_FAILURE;
 128         }
 129 };
 130
 131 class ModuleCAP : public Module
 132 {
 133         CommandCAP cmd;
 134  public:
 135         ModuleCAP()
 136                 : cmd(this)
 137         {
 138                 ServerInstance->AddCommand(&cmd);
 139                 ServerInstance->Extensions.Register(&cmd.reghold);
 140
 141                 Implementation eventlist[] = { I_OnCheckReady };
 142                 ServerInstance->Modules->Attach(eventlist, this, 1);
 143         }
 144
 145         ModResult OnCheckReady(LocalUser* user)
 146         {
 147                 /* Users in CAP state get held until CAP END */
 148                 if (cmd.reghold.get(user))
 149                         return MOD_RES_DENY;
 150
 151                 return MOD_RES_PASSTHRU;
 152         }
 153
 154         ~ModuleCAP()
 155         {
 156         }
 157
 158         Version GetVersion()
 159         {
 160                 return Version("Client CAP extension support", VF_VENDOR);
 161         }
 162 };
 163
 164 MODULE_INIT(ModuleCAP)
 165