src/modules/m_services_account.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2009 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 "account.h"
  16
  17 /* $ModDesc: Povides support for ircu-style services accounts, including chmode +R, etc. */
  18
  19 /** Channel mode +r - mark a channel as identified
  20  */
  21 class Channel_r : public ModeHandler
  22 {
  23
  24  public:
  25         Channel_r(InspIRCd* Instance) : ModeHandler(Instance, 'r', 0, 0, false, MODETYPE_CHANNEL, false) { }
  26
  27         ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding, bool)
  28         {
  29                 // only a u-lined server may add or remove the +r mode.
  30                 if (IS_REMOTE(source) || ServerInstance->ULine(source->nick.c_str()) || ServerInstance->ULine(source->server))
  31                 {
  32                         // Only change the mode if it's not redundant
  33                         if ((adding && !channel->IsModeSet('r')) || (!adding && channel->IsModeSet('r')))
  34                         {
  35                                 channel->SetMode('r',adding);
  36                                 return MODEACTION_ALLOW;
  37                         }
  38
  39                         return MODEACTION_DENY;
  40                 }
  41                 else
  42                 {
  43                         source->WriteNumeric(500, "%s :Only a server may modify the +r channel mode", source->nick.c_str());
  44                         return MODEACTION_DENY;
  45                 }
  46         }
  47 };
  48
  49 /** User mode +r - mark a user as identified
  50  */
  51 class User_r : public ModeHandler
  52 {
  53
  54  public:
  55         User_r(InspIRCd* Instance) : ModeHandler(Instance, 'r', 0, 0, false, MODETYPE_USER, false) { }
  56
  57         ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding, bool servermode)
  58         {
  59                 if (servermode || IS_REMOTE(source) || ServerInstance->ULine(source->nick.c_str()) || ServerInstance->ULine(source->server))
  60                 {
  61                         if ((adding && !dest->IsModeSet('r')) || (!adding && dest->IsModeSet('r')))
  62                         {
  63                                 dest->SetMode('r',adding);
  64                                 return MODEACTION_ALLOW;
  65                         }
  66                         return MODEACTION_DENY;
  67                 }
  68                 else
  69                 {
  70                         source->WriteNumeric(500, "%s :Only a server may modify the +r user mode", source->nick.c_str());
  71                         return MODEACTION_DENY;
  72                 }
  73         }
  74 };
  75
  76 /** Channel mode +R - unidentified users cannot join
  77  */
  78 class AChannel_R : public SimpleChannelModeHandler
  79 {
  80  public:
  81         AChannel_R(InspIRCd* Instance) : SimpleChannelModeHandler(Instance, 'R') { }
  82 };
  83
  84 /** User mode +R - unidentified users cannot message
  85  */
  86 class AUser_R : public SimpleUserModeHandler
  87 {
  88  public:
  89         AUser_R(InspIRCd* Instance) : SimpleUserModeHandler(Instance, 'R') { }
  90 };
  91
  92 /** Channel mode +M - unidentified users cannot message channel
  93  */
  94 class AChannel_M : public SimpleChannelModeHandler
  95 {
  96  public:
  97         AChannel_M(InspIRCd* Instance) : SimpleChannelModeHandler(Instance, 'M') { }
  98 };
  99
 100 class ModuleServicesAccount : public Module
 101 {
 102         AChannel_R* m1;
 103         AChannel_M* m2;
 104         AUser_R* m3;
 105         Channel_r *m4;
 106         User_r *m5;
 107  public:
 108         ModuleServicesAccount(InspIRCd* Me) : Module(Me)
 109         {
 110                 m1 = new AChannel_R(ServerInstance);
 111                 m2 = new AChannel_M(ServerInstance);
 112                 m3 = new AUser_R(ServerInstance);
 113                 m4 = new Channel_r(ServerInstance);
 114                 m5 = new User_r(ServerInstance);
 115
 116                 if (!ServerInstance->Modes->AddMode(m1) || !ServerInstance->Modes->AddMode(m2) || !ServerInstance->Modes->AddMode(m3) || !ServerInstance->Modes->AddMode(m4) || !ServerInstance->Modes->AddMode(m5))
 117                         throw ModuleException("Some other module has claimed our modes!");
 118
 119                 Implementation eventlist[] = { I_OnWhois, I_OnUserPreMessage, I_OnUserPreNotice, I_OnUserPreJoin, I_OnCheckBan,
 120                         I_OnSyncUserMetaData, I_OnUserQuit, I_OnCleanup, I_OnDecodeMetaData, I_On005Numeric, I_OnUserPostNick };
 121
 122                 ServerInstance->Modules->Attach(eventlist, this, 10);
 123         }
 124
 125         virtual void On005Numeric(std::string &t)
 126         {
 127                 ServerInstance->AddExtBanChar('R');
 128                 ServerInstance->AddExtBanChar('M');
 129         }
 130
 131         /* <- :twisted.oscnet.org 330 w00t2 w00t2 w00t :is logged in as */
 132         virtual void OnWhois(User* source, User* dest)
 133         {
 134                 std::string *account;
 135                 dest->GetExt("accountname", account);
 136
 137                 if (account)
 138                 {
 139                         ServerInstance->SendWhoisLine(source, dest, 330, "%s %s %s :is logged in as", source->nick.c_str(), dest->nick.c_str(), account->c_str());
 140                 }
 141
 142                 if (dest->IsModeSet('r'))
 143                 {
 144                         /* user is registered */
 145                         ServerInstance->SendWhoisLine(source, dest, 307, "%s %s :is a registered nick", source->nick.c_str(), dest->nick.c_str());
 146                 }
 147         }
 148
 149         virtual void OnUserPostNick(User* user, const std::string &oldnick)
 150         {
 151                 /* On nickchange, if they have +r, remove it */
 152                 if (user->IsModeSet('r') && assign(user->nick) != oldnick)
 153                 {
 154                         std::vector<std::string> modechange;
 155                         modechange.push_back(user->nick);
 156                         modechange.push_back("-r");
 157                         ServerInstance->SendMode(modechange, user);
 158                 }
 159         }
 160
 161         virtual int OnUserPreMessage(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
 162         {
 163                 if (!IS_LOCAL(user))
 164                         return 0;
 165
 166                 std::string *account;
 167                 bool is_registered = user->GetExt("accountname", account);
 168                 is_registered = is_registered && !account->empty();
 169
 170                 if ((ServerInstance->ULine(user->nick.c_str())) || (ServerInstance->ULine(user->server)))
 171                 {
 172                         // user is ulined, can speak regardless
 173                         return 0;
 174                 }
 175
 176                 if (target_type == TYPE_CHANNEL)
 177                 {
 178                         Channel* c = (Channel*)dest;
 179
 180                         if (c->IsModeSet('M') && !is_registered)
 181                         {
 182                                 // user messaging a +M channel and is not registered
 183                                 user->WriteNumeric(477, ""+std::string(user->nick)+" "+std::string(c->name)+" :You need to be identified to a registered account to message this channel");
 184                                 return 1;
 185                         }
 186
 187                         if (account)
 188                         {
 189                                 if (c->GetExtBanStatus(*account, 'M') < 0)
 190                                 {
 191                                         // may not speak (text is deliberately vague, so they don't know which restriction to evade)
 192                                         user->WriteNumeric(477, ""+std::string(user->nick)+" "+std::string(c->name)+" :You may not speak in this channel");
 193                                         return 1;
 194                                 }
 195                         }
 196                 }
 197                 else if (target_type == TYPE_USER)
 198                 {
 199                         User* u = (User*)dest;
 200
 201                         if (u->IsModeSet('R') && !is_registered)
 202                         {
 203                                 // user messaging a +R user and is not registered
 204                                 user->WriteNumeric(477, ""+ user->nick +" "+ u->nick +" :You need to be identified to a registered account to message this user");
 205                                 return 1;
 206                         }
 207                 }
 208                 return 0;
 209         }
 210
 211         virtual int OnCheckBan(User* user, Channel* chan)
 212         {
 213                 std::string* account;
 214                 if (!user->GetExt("accountname", account))
 215                         return 0;
 216                 return chan->GetExtBanStatus(*account, 'R');
 217         }
 218
 219         virtual int OnUserPreNotice(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
 220         {
 221                 return OnUserPreMessage(user, dest, target_type, text, status, exempt_list);
 222         }
 223
 224         virtual int OnUserPreJoin(User* user, Channel* chan, const char* cname, std::string &privs, const std::string &keygiven)
 225         {
 226                 if (!IS_LOCAL(user))
 227                         return 0;
 228
 229                 std::string *account;
 230                 bool is_registered = user->GetExt("accountname", account);
 231                 is_registered = is_registered && !account->empty();
 232
 233                 if (chan)
 234                 {
 235                         if ((ServerInstance->ULine(user->nick.c_str())) || (ServerInstance->ULine(user->server)))
 236                         {
 237                                 // user is ulined, won't be stopped from joining
 238                                 return 0;
 239                         }
 240
 241                         if (chan->IsModeSet('R'))
 242                         {
 243                                 if (!is_registered)
 244                                 {
 245                                         // joining a +R channel and not identified
 246                                         user->WriteNumeric(477, user->nick + " " + chan->name + " :You need to be identified to a registered account to join this channel");
 247                                         return 1;
 248                                 }
 249                         }
 250                 }
 251                 return 0;
 252         }
 253
 254         // Whenever the linking module wants to send out data, but doesnt know what the data
 255         // represents (e.g. it is metadata, added to a User or Channel by a module) then
 256         // this method is called. We should use the ProtoSendMetaData function after we've
 257         // corrected decided how the data should look, to send the metadata on its way if
 258         // it is ours.
 259         virtual void OnSyncUserMetaData(User* user, Module* proto, void* opaque, const std::string &extname, bool displayable)
 260         {
 261                 // check if the linking module wants to know about OUR metadata
 262                 if (extname == "accountname")
 263                 {
 264                         // check if this user has an swhois field to send
 265                         std::string* account;
 266                         user->GetExt("accountname", account);
 267                         if (account)
 268                         {
 269                                 // remove any accidental leading/trailing spaces
 270                                 trim(*account);
 271
 272                                 // call this function in the linking module, let it format the data how it
 273                                 // sees fit, and send it on its way. We dont need or want to know how.
 274                                 proto->ProtoSendMetaData(opaque,TYPE_USER,user,extname,*account);
 275                         }
 276                 }
 277         }
 278
 279         // when a user quits, tidy up their metadata
 280         virtual void OnUserQuit(User* user, const std::string &message, const std::string &oper_message)
 281         {
 282                 std::string* account;
 283                 user->GetExt("accountname", account);
 284                 if (account)
 285                 {
 286                         user->Shrink("accountname");
 287                         delete account;
 288                 }
 289         }
 290
 291         // if the module is unloaded, tidy up all our dangling metadata
 292         virtual void OnCleanup(int target_type, void* item)
 293         {
 294                 if (target_type == TYPE_USER)
 295                 {
 296                         User* user = (User*)item;
 297                         std::string* account;
 298                         user->GetExt("accountname", account);
 299                         if (account)
 300                         {
 301                                 user->Shrink("accountname");
 302                                 delete account;
 303                         }
 304                 }
 305         }
 306
 307         // Whenever the linking module receives metadata from another server and doesnt know what
 308         // to do with it (of course, hence the 'meta') it calls this method, and it is up to each
 309         // module in turn to figure out if this metadata key belongs to them, and what they want
 310         // to do with it.
 311         // In our case we're only sending a single string around, so we just construct a std::string.
 312         // Some modules will probably get much more complex and format more detailed structs and classes
 313         // in a textual way for sending over the link.
 314         virtual void OnDecodeMetaData(int target_type, void* target, const std::string &extname, const std::string &extdata)
 315         {
 316                 // check if its our metadata key, and its associated with a user
 317                 if ((target_type == TYPE_USER) && (extname == "accountname"))
 318                 {
 319                         User* dest = (User*)target;
 320
 321                         std::string* account;
 322                         if (dest->GetExt("accountname", account)) {
 323                                 // remove old account so that we can set new (or leave unset)
 324                                 dest->Shrink("accountname");
 325                                 delete account;
 326                         }
 327
 328                         if (!extdata.empty())
 329                         {
 330                                 account = new std::string(extdata);
 331                                 // remove any accidental leading/trailing spaces
 332                                 trim(*account);
 333                                 dest->Extend("accountname", account);
 334
 335                                 if (IS_LOCAL(dest))
 336                                         dest->WriteNumeric(900, "%s %s %s :You are now logged in as %s",
 337                                                 dest->nick.c_str(), dest->GetFullHost().c_str(), account->c_str(), account->c_str());
 338
 339                                 AccountData ac;
 340                                 ac.user = dest;
 341                                 ac.account = *account;
 342                                 Event n((char*)&ac, this, "account_login");
 343                                 n.Send(ServerInstance);
 344                         }
 345                 }
 346         }
 347
 348         virtual ~ModuleServicesAccount()
 349         {
 350                 ServerInstance->Modes->DelMode(m1);
 351                 ServerInstance->Modes->DelMode(m2);
 352                 ServerInstance->Modes->DelMode(m3);
 353                 ServerInstance->Modes->DelMode(m4);
 354                 ServerInstance->Modes->DelMode(m5);
 355                 delete m1;
 356                 delete m2;
 357                 delete m3;
 358                 delete m4;
 359                 delete m5;
 360         }
 361
 362         virtual Version GetVersion()
 363         {
 364                 return Version("$Id$",VF_COMMON|VF_VENDOR,API_VERSION);
 365         }
 366 };
 367
 368 MODULE_INIT(ModuleServicesAccount)