src/modules/m_services_account.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2012 Shawn Smith <shawn@inspircd.org>
   5  *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
   6  *   Copyright (C) 2006-2008 Robin Burchell <robin+git@viroteck.net>
   7  *   Copyright (C) 2008 Pippijn van Steenhoven <pip88nl@gmail.com>
   8  *   Copyright (C) 2006, 2008 Craig Edwards <craigedwards@brainbox.cc>
   9  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
  10  *
  11  * This file is part of InspIRCd.  InspIRCd is free software: you can
  12  * redistribute it and/or modify it under the terms of the GNU General Public
  13  * License as published by the Free Software Foundation, version 2.
  14  *
  15  * This program is distributed in the hope that it will be useful, but WITHOUT
  16  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  17  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  18  * details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22  */
  23
  24
  25 #include "inspircd.h"
  26 #include "modules/account.h"
  27 #include "modules/exemption.h"
  28
  29 enum
  30 {
  31         // From ircd-hybrid?
  32         ERR_NEEDREGGEDNICK = 477,
  33
  34         // From IRCv3 sasl-3.1.
  35         RPL_LOGGEDIN = 900,
  36         RPL_LOGGEDOUT = 901
  37 };
  38
  39 /** Channel mode +r - mark a channel as identified
  40  */
  41 class Channel_r : public ModeHandler
  42 {
  43  public:
  44         Channel_r(Module* Creator) : ModeHandler(Creator, "c_registered", 'r', PARAM_NONE, MODETYPE_CHANNEL) { }
  45
  46         ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string& parameter, bool adding) CXX11_OVERRIDE
  47         {
  48                 // only a u-lined server may add or remove the +r mode.
  49                 if (!IS_LOCAL(source))
  50                 {
  51                         // Only change the mode if it's not redundant
  52                         if ((adding != channel->IsModeSet(this)))
  53                         {
  54                                 channel->SetMode(this, adding);
  55                                 return MODEACTION_ALLOW;
  56                         }
  57                 }
  58                 else
  59                 {
  60                         source->WriteNumeric(500, "Only a server may modify the +r channel mode");
  61                 }
  62                 return MODEACTION_DENY;
  63         }
  64 };
  65
  66 /** User mode +r - mark a user as identified
  67  */
  68 class User_r : public ModeHandler
  69 {
  70
  71  public:
  72         User_r(Module* Creator) : ModeHandler(Creator, "u_registered", 'r', PARAM_NONE, MODETYPE_USER) { }
  73
  74         ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string& parameter, bool adding) CXX11_OVERRIDE
  75         {
  76                 if (!IS_LOCAL(source))
  77                 {
  78                         if ((adding != dest->IsModeSet(this)))
  79                         {
  80                                 dest->SetMode(this, adding);
  81                                 return MODEACTION_ALLOW;
  82                         }
  83                 }
  84                 else
  85                 {
  86                         source->WriteNumeric(500, "Only a server may modify the +r user mode");
  87                 }
  88                 return MODEACTION_DENY;
  89         }
  90 };
  91
  92 class AccountExtItemImpl : public AccountExtItem
  93 {
  94         Events::ModuleEventProvider eventprov;
  95
  96  public:
  97         AccountExtItemImpl(Module* mod)
  98                 : AccountExtItem("accountname", ExtensionItem::EXT_USER, mod)
  99                 , eventprov(mod, "event/account")
 100         {
 101         }
 102
 103         void unserialize(SerializeFormat format, Extensible* container, const std::string& value) CXX11_OVERRIDE
 104         {
 105                 User* user = static_cast<User*>(container);
 106
 107                 StringExtItem::unserialize(format, container, value);
 108
 109                 // If we are being reloaded then don't send the numeric or run the event
 110                 if (format == FORMAT_INTERNAL)
 111                         return;
 112
 113                 if (IS_LOCAL(user))
 114                 {
 115                         if (value.empty())
 116                         {
 117                                 // Logged out.
 118                                 user->WriteNumeric(RPL_LOGGEDOUT, user->GetFullHost(), "You are now logged out");
 119                         }
 120                         else
 121                         {
 122                                 // Logged in.
 123                                 user->WriteNumeric(RPL_LOGGEDIN, user->GetFullHost(), value, InspIRCd::Format("You are now logged in as %s", value.c_str()));
 124                         }
 125                 }
 126
 127                 FOREACH_MOD_CUSTOM(eventprov, AccountEventListener, OnAccountChange, (user, value));
 128         }
 129 };
 130
 131 class ModuleServicesAccount : public Module, public Whois::EventListener
 132 {
 133         CheckExemption::EventProvider exemptionprov;
 134         SimpleChannelModeHandler m1;
 135         SimpleChannelModeHandler m2;
 136         SimpleUserModeHandler m3;
 137         Channel_r m4;
 138         User_r m5;
 139         AccountExtItemImpl accountname;
 140         bool checking_ban;
 141  public:
 142         ModuleServicesAccount()
 143                 : Whois::EventListener(this)
 144                 , exemptionprov(this)
 145                 , m1(this, "reginvite", 'R')
 146                 , m2(this, "regmoderated", 'M')
 147                 , m3(this, "regdeaf", 'R')
 148                 , m4(this)
 149                 , m5(this)
 150                 , accountname(this)
 151                 , checking_ban(false)
 152         {
 153         }
 154
 155         void On005Numeric(std::map<std::string, std::string>& tokens) CXX11_OVERRIDE
 156         {
 157                 tokens["EXTBAN"].push_back('R');
 158                 tokens["EXTBAN"].push_back('U');
 159         }
 160
 161         /* <- :twisted.oscnet.org 330 w00t2 w00t2 w00t :is logged in as */
 162         void OnWhois(Whois::Context& whois) CXX11_OVERRIDE
 163         {
 164                 std::string* account = accountname.get(whois.GetTarget());
 165
 166                 if (account)
 167                 {
 168                         whois.SendLine(330, *account, "is logged in as");
 169                 }
 170
 171                 if (whois.GetTarget()->IsModeSet(m5))
 172                 {
 173                         /* user is registered */
 174                         whois.SendLine(307, "is a registered nick");
 175                 }
 176         }
 177
 178         void OnUserPostNick(User* user, const std::string &oldnick) CXX11_OVERRIDE
 179         {
 180                 /* On nickchange, if they have +r, remove it */
 181                 if ((user->IsModeSet(m5)) && (ServerInstance->FindNickOnly(oldnick) != user))
 182                         m5.RemoveMode(user);
 183         }
 184
 185         ModResult OnUserPreMessage(User* user, void* dest, int target_type, std::string& text, char status, CUList& exempt_list, MessageType msgtype) CXX11_OVERRIDE
 186         {
 187                 if (!IS_LOCAL(user))
 188                         return MOD_RES_PASSTHRU;
 189
 190                 std::string *account = accountname.get(user);
 191                 bool is_registered = account && !account->empty();
 192
 193                 if (target_type == TYPE_CHANNEL)
 194                 {
 195                         Channel* c = (Channel*)dest;
 196                         ModResult res = CheckExemption::Call(exemptionprov, user, c, "regmoderated");
 197
 198                         if (c->IsModeSet(m2) && !is_registered && res != MOD_RES_ALLOW)
 199                         {
 200                                 // user messaging a +M channel and is not registered
 201                                 user->WriteNumeric(ERR_NEEDREGGEDNICK, c->name, "You need to be identified to a registered account to message this channel");
 202                                 return MOD_RES_DENY;
 203                         }
 204                 }
 205                 else if (target_type == TYPE_USER)
 206                 {
 207                         User* u = (User*)dest;
 208
 209                         if (u->IsModeSet(m3) && !is_registered)
 210                         {
 211                                 // user messaging a +R user and is not registered
 212                                 user->WriteNumeric(ERR_NEEDREGGEDNICK, u->nick, "You need to be identified to a registered account to message this user");
 213                                 return MOD_RES_DENY;
 214                         }
 215                 }
 216                 return MOD_RES_PASSTHRU;
 217         }
 218
 219         ModResult OnCheckBan(User* user, Channel* chan, const std::string& mask) CXX11_OVERRIDE
 220         {
 221                 if (checking_ban)
 222                         return MOD_RES_PASSTHRU;
 223
 224                 if ((mask.length() > 2) && (mask[1] == ':'))
 225                 {
 226                         if (mask[0] == 'R')
 227                         {
 228                                 std::string *account = accountname.get(user);
 229                                 if (account && InspIRCd::Match(*account, mask.substr(2)))
 230                                         return MOD_RES_DENY;
 231                         }
 232                         else if (mask[0] == 'U')
 233                         {
 234                                 std::string *account = accountname.get(user);
 235                                 /* If the user is registered we don't care. */
 236                                 if (account)
 237                                         return MOD_RES_PASSTHRU;
 238
 239                                 /* If we made it this far we know the user isn't registered
 240                                         so just deny if it matches */
 241                                 checking_ban = true;
 242                                 bool result = chan->CheckBan(user, mask.substr(2));
 243                                 checking_ban = false;
 244
 245                                 if (result)
 246                                         return MOD_RES_DENY;
 247                         }
 248                 }
 249
 250                 /* If we made it this far then the ban wasn't an ExtBan
 251                         or the user we were checking for didn't match either ExtBan */
 252                 return MOD_RES_PASSTHRU;
 253         }
 254
 255         ModResult OnUserPreJoin(LocalUser* user, Channel* chan, const std::string& cname, std::string& privs, const std::string& keygiven) CXX11_OVERRIDE
 256         {
 257                 std::string *account = accountname.get(user);
 258                 bool is_registered = account && !account->empty();
 259
 260                 if (chan)
 261                 {
 262                         if (chan->IsModeSet(m1))
 263                         {
 264                                 if (!is_registered)
 265                                 {
 266                                         // joining a +R channel and not identified
 267                                         user->WriteNumeric(ERR_NEEDREGGEDNICK, chan->name, "You need to be identified to a registered account to join this channel");
 268                                         return MOD_RES_DENY;
 269                                 }
 270                         }
 271                 }
 272                 return MOD_RES_PASSTHRU;
 273         }
 274
 275         ModResult OnSetConnectClass(LocalUser* user, ConnectClass* myclass) CXX11_OVERRIDE
 276         {
 277                 if (myclass->config->getBool("requireaccount") && !accountname.get(user))
 278                         return MOD_RES_DENY;
 279                 return MOD_RES_PASSTHRU;
 280         }
 281
 282         Version GetVersion() CXX11_OVERRIDE
 283         {
 284                 return Version("Provides support for ircu-style services accounts, including chmode +R, etc.",VF_OPTCOMMON|VF_VENDOR);
 285         }
 286 };
 287
 288 MODULE_INIT(ModuleServicesAccount)