src/modules/m_namesx.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2019 linuxdaemon <linuxdaemon.irc@gmail.com>
   5  *   Copyright (C) 2013, 2018-2021 Sadie Powell <sadie@witchery.services>
   6  *   Copyright (C) 2012-2016 Attila Molnar <attilamolnar@hush.com>
   7  *   Copyright (C) 2012, 2019 Robby <robby@chatbelgie.be>
   8  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   9  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
  10  *   Copyright (C) 2006, 2008 Craig Edwards <brain@inspircd.org>
  11  *
  12  * This file is part of InspIRCd.  InspIRCd is free software: you can
  13  * redistribute it and/or modify it under the terms of the GNU General Public
  14  * License as published by the Free Software Foundation, version 2.
  15  *
  16  * This program is distributed in the hope that it will be useful, but WITHOUT
  17  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  18  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  19  * details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  23  */
  24
  25
  26 #include "inspircd.h"
  27 #include "modules/cap.h"
  28 #include "modules/names.h"
  29 #include "modules/who.h"
  30 #include "modules/whois.h"
  31
  32 class ModuleNamesX
  33         : public Module
  34         , public Names::EventListener
  35         , public Who::EventListener
  36         , public Whois::LineEventListener
  37 {
  38  private:
  39         Cap::Capability cap;
  40
  41  public:
  42         ModuleNamesX()
  43                 : Names::EventListener(this)
  44                 , Who::EventListener(this)
  45                 , Whois::LineEventListener(this)
  46                 , cap(this, "multi-prefix")
  47         {
  48         }
  49
  50         Version GetVersion() CXX11_OVERRIDE
  51         {
  52                 return Version("Provides the IRCv3 multi-prefix client capability.", VF_VENDOR);
  53         }
  54
  55         void On005Numeric(std::map<std::string, std::string>& tokens) CXX11_OVERRIDE
  56         {
  57                 // The legacy PROTOCTL system is a wrapper around the cap.
  58                 dynamic_reference_nocheck<Cap::Manager> capmanager(this, "capmanager");
  59                 if (capmanager)
  60                         tokens["NAMESX"];
  61         }
  62
  63         ModResult OnPreCommand(std::string& command, CommandBase::Params& parameters, LocalUser* user, bool validated) CXX11_OVERRIDE
  64         {
  65                 /* We don't actually create a proper command handler class for PROTOCTL,
  66                  * because other modules might want to have PROTOCTL hooks too.
  67                  * Therefore, we just hook its as an unvalidated command therefore we
  68                  * can capture it even if it doesnt exist! :-)
  69                  */
  70                 if (command == "PROTOCTL")
  71                 {
  72                         if (!parameters.empty() && irc::equals(parameters[0], "NAMESX"))
  73                         {
  74                                 cap.set(user, true);
  75                                 return MOD_RES_DENY;
  76                         }
  77                 }
  78                 return MOD_RES_PASSTHRU;
  79         }
  80
  81         ModResult OnNamesListItem(LocalUser* issuer, Membership* memb, std::string& prefixes, std::string& nick) CXX11_OVERRIDE
  82         {
  83                 if (cap.get(issuer))
  84                         prefixes = memb->GetAllPrefixChars();
  85
  86                 return MOD_RES_PASSTHRU;
  87         }
  88
  89         ModResult OnWhoLine(const Who::Request& request, LocalUser* source, User* user, Membership* memb, Numeric::Numeric& numeric) CXX11_OVERRIDE
  90         {
  91                 if ((!memb) || (!cap.get(source)))
  92                         return MOD_RES_PASSTHRU;
  93
  94                 // Don't do anything if the user has only one prefix
  95                 std::string prefixes = memb->GetAllPrefixChars();
  96                 if (prefixes.length() <= 1)
  97                         return MOD_RES_PASSTHRU;
  98
  99                 size_t flag_index;
 100                 if (!request.GetFieldIndex('f', flag_index))
 101                         return MOD_RES_PASSTHRU;
 102
 103                 // #chan ident localhost insp22.test nick H@ :0 Attila
 104                 if (numeric.GetParams().size() <= flag_index)
 105                         return MOD_RES_PASSTHRU;
 106
 107                 numeric.GetParams()[flag_index].append(prefixes, 1, std::string::npos);
 108                 return MOD_RES_PASSTHRU;
 109         }
 110
 111         ModResult OnWhoisLine(Whois::Context& whois, Numeric::Numeric& numeric) CXX11_OVERRIDE
 112         {
 113                 if (numeric.GetNumeric() != RPL_WHOISCHANNELS || !cap.get(whois.GetSource()))
 114                         return MOD_RES_PASSTHRU;
 115
 116                 // :testnet.inspircd.org 319 test Sadie :#test ~#inspircd
 117                 if (numeric.GetParams().size() < 2 || numeric.GetParams().back().empty())
 118                         return MOD_RES_PASSTHRU;
 119
 120                 std::stringstream newchannels;
 121                 irc::spacesepstream channelstream(numeric.GetParams().back());
 122                 for (std::string channel; channelstream.GetToken(channel); )
 123                 {
 124                         size_t hashpos = channel.find('#');
 125                         if (!hashpos || hashpos == std::string::npos)
 126                         {
 127                                 // The entry is malformed or the user has no privs.
 128                                 newchannels << channel << ' ';
 129                                 continue;
 130                         }
 131
 132                         Channel* chan = ServerInstance->FindChan(channel.substr(hashpos));
 133                         if (!chan)
 134                         {
 135                                 // Should never happen.
 136                                 newchannels << channel << ' ';
 137                                 continue;
 138                         }
 139
 140                         Membership* memb = chan->GetUser(whois.GetTarget());
 141                         if (!memb)
 142                         {
 143                                 // Should never happen.
 144                                 newchannels << channel << ' ';
 145                                 continue;
 146                         }
 147
 148                         newchannels << memb->GetAllPrefixChars() << chan->name << ' ';
 149                 }
 150
 151                 numeric.GetParams().back() = newchannels.str();
 152                 return MOD_RES_PASSTHRU;
 153         }
 154 };
 155
 156 MODULE_INIT(ModuleNamesX)