src/mode.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2012 Shawn Smith <shawn@inspircd.org>
   5  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   6  *   Copyright (C) 2007, 2009 Dennis Friis <peavey@inspircd.org>
   7  *   Copyright (C) 2006-2008 Robin Burchell <robin+git@viroteck.net>
   8  *   Copyright (C) 2008 Thomas Stagner <aquanight@inspircd.org>
   9  *   Copyright (C) 2004-2008 Craig Edwards <craigedwards@brainbox.cc>
  10  *   Copyright (C) 2006 Oliver Lupton <oliverlupton@gmail.com>
  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 "builtinmodes.h"
  28
  29 ModeHandler::ModeHandler(Module* Creator, const std::string& Name, char modeletter, ParamSpec Params, ModeType type, Class mclass)
  30         : ServiceProvider(Creator, Name, SERVICE_MODE), m_paramtype(TR_TEXT),
  31         parameters_taken(Params), mode(modeletter), oper(false),
  32         list(false), m_type(type), type_id(mclass), levelrequired(HALFOP_VALUE)
  33 {
  34 }
  35
  36 CullResult ModeHandler::cull()
  37 {
  38         if (ServerInstance->Modes)
  39                 ServerInstance->Modes->DelMode(this);
  40         return classbase::cull();
  41 }
  42
  43 ModeHandler::~ModeHandler()
  44 {
  45 }
  46
  47 int ModeHandler::GetNumParams(bool adding)
  48 {
  49         switch (parameters_taken)
  50         {
  51                 case PARAM_ALWAYS:
  52                         return 1;
  53                 case PARAM_SETONLY:
  54                         return adding ? 1 : 0;
  55                 case PARAM_NONE:
  56                         break;
  57         }
  58         return 0;
  59 }
  60
  61 std::string ModeHandler::GetUserParameter(User* user)
  62 {
  63         return "";
  64 }
  65
  66 ModResult ModeHandler::AccessCheck(User*, Channel*, std::string &, bool)
  67 {
  68         return MOD_RES_PASSTHRU;
  69 }
  70
  71 ModeAction ModeHandler::OnModeChange(User*, User*, Channel*, std::string&, bool)
  72 {
  73         return MODEACTION_DENY;
  74 }
  75
  76 void ModeHandler::DisplayList(User*, Channel*)
  77 {
  78 }
  79
  80 void ModeHandler::DisplayEmptyList(User*, Channel*)
  81 {
  82 }
  83
  84 void ModeHandler::OnParameterMissing(User* user, User* dest, Channel* channel)
  85 {
  86 }
  87
  88 bool ModeHandler::ResolveModeConflict(std::string& theirs, const std::string& ours, Channel*)
  89 {
  90         return (theirs < ours);
  91 }
  92
  93 ModeAction SimpleUserModeHandler::OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
  94 {
  95         /* We're either trying to add a mode we already have or
  96                 remove a mode we don't have, deny. */
  97         if (dest->IsModeSet(this) == adding)
  98                 return MODEACTION_DENY;
  99
 100         /* adding will be either true or false, depending on if we
 101                 are adding or removing the mode, since we already checked
 102                 to make sure we aren't adding a mode we have or that we
 103                 aren't removing a mode we don't have, we don't have to do any
 104                 other checks here to see if it's true or false, just add or
 105                 remove the mode */
 106         dest->SetMode(this, adding);
 107
 108         return MODEACTION_ALLOW;
 109 }
 110
 111
 112 ModeAction SimpleChannelModeHandler::OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
 113 {
 114         /* We're either trying to add a mode we already have or
 115                 remove a mode we don't have, deny. */
 116         if (channel->IsModeSet(this) == adding)
 117                 return MODEACTION_DENY;
 118
 119         /* adding will be either true or false, depending on if we
 120                 are adding or removing the mode, since we already checked
 121                 to make sure we aren't adding a mode we have or that we
 122                 aren't removing a mode we don't have, we don't have to do any
 123                 other checks here to see if it's true or false, just add or
 124                 remove the mode */
 125         channel->SetMode(this, adding);
 126
 127         return MODEACTION_ALLOW;
 128 }
 129
 130 ModeAction ParamChannelModeHandler::OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
 131 {
 132         if (adding && !ParamValidate(parameter))
 133                 return MODEACTION_DENY;
 134         std::string now = channel->GetModeParameter(this);
 135         if (parameter == now)
 136                 return MODEACTION_DENY;
 137         return MODEACTION_ALLOW;
 138 }
 139
 140 bool ParamChannelModeHandler::ParamValidate(std::string& parameter)
 141 {
 142         return true;
 143 }
 144
 145 ModeWatcher::ModeWatcher(Module* Creator, const std::string& modename, ModeType type)
 146         : mode(modename), m_type(type), creator(Creator)
 147 {
 148         ServerInstance->Modes->AddModeWatcher(this);
 149 }
 150
 151 ModeWatcher::~ModeWatcher()
 152 {
 153         ServerInstance->Modes->DelModeWatcher(this);
 154 }
 155
 156 ModeType ModeWatcher::GetModeType()
 157 {
 158         return m_type;
 159 }
 160
 161 bool ModeWatcher::BeforeMode(User*, User*, Channel*, std::string&, bool)
 162 {
 163         return true;
 164 }
 165
 166 void ModeWatcher::AfterMode(User*, User*, Channel*, const std::string&, bool)
 167 {
 168 }
 169
 170 void ModeParser::DisplayCurrentModes(User *user, User* targetuser, Channel* targetchannel, const char* text)
 171 {
 172         if (targetchannel)
 173         {
 174                 /* Display channel's current mode string */
 175                 user->WriteNumeric(RPL_CHANNELMODEIS, "%s +%s", targetchannel->name.c_str(), targetchannel->ChanModes(targetchannel->HasUser(user)));
 176                 user->WriteNumeric(RPL_CHANNELCREATED, "%s %lu", targetchannel->name.c_str(), (unsigned long)targetchannel->age);
 177                 return;
 178         }
 179         else
 180         {
 181                 if (targetuser == user || user->HasPrivPermission("users/auspex"))
 182                 {
 183                         /* Display user's current mode string */
 184                         user->WriteNumeric(RPL_UMODEIS, ":+%s", targetuser->FormatModes());
 185                         if ((targetuser->IsOper()))
 186                         {
 187                                 ModeHandler* snomask = FindMode('s', MODETYPE_USER);
 188                                 user->WriteNumeric(RPL_SNOMASKIS, "%s :Server notice mask", snomask->GetUserParameter(user).c_str());
 189                         }
 190                         return;
 191                 }
 192                 else
 193                 {
 194                         user->WriteNumeric(ERR_USERSDONTMATCH, ":Can't view modes for other users");
 195                         return;
 196                 }
 197         }
 198 }
 199
 200 PrefixMode::PrefixMode(Module* Creator, const std::string& Name, char ModeLetter)
 201         : ModeHandler(Creator, Name, ModeLetter, PARAM_ALWAYS, MODETYPE_CHANNEL, MC_PREFIX)
 202         , prefix(0), prefixrank(0)
 203 {
 204         list = true;
 205         m_paramtype = TR_NICK;
 206 }
 207
 208 ModeAction PrefixMode::OnModeChange(User* source, User*, Channel* chan, std::string& parameter, bool adding)
 209 {
 210         User* target = ServerInstance->FindNick(parameter);
 211         if (!target)
 212         {
 213                 source->WriteNumeric(ERR_NOSUCHNICK, "%s :No such nick/channel", parameter.c_str());
 214                 return MODEACTION_DENY;
 215         }
 216
 217         Membership* memb = chan->GetUser(target);
 218         if (!memb)
 219                 return MODEACTION_DENY;
 220
 221         parameter = target->nick;
 222         return (memb->SetPrefix(this, adding) ? MODEACTION_ALLOW : MODEACTION_DENY);
 223 }
 224
 225 ModeAction ModeParser::TryMode(User* user, User* targetuser, Channel* chan, bool adding, const unsigned char modechar,
 226                 std::string &parameter, bool SkipACL)
 227 {
 228         ModeType type = chan ? MODETYPE_CHANNEL : MODETYPE_USER;
 229
 230         ModeHandler *mh = FindMode(modechar, type);
 231         int pcnt = mh->GetNumParams(adding);
 232
 233         // crop mode parameter size to 250 characters
 234         if (parameter.length() > 250 && adding)
 235                 parameter = parameter.substr(0, 250);
 236
 237         ModResult MOD_RESULT;
 238         FIRST_MOD_RESULT(OnRawMode, MOD_RESULT, (user, chan, modechar, parameter, adding, pcnt));
 239
 240         if (IS_LOCAL(user) && (MOD_RESULT == MOD_RES_DENY))
 241                 return MODEACTION_DENY;
 242
 243         if (chan && !SkipACL && (MOD_RESULT != MOD_RES_ALLOW))
 244         {
 245                 MOD_RESULT = mh->AccessCheck(user, chan, parameter, adding);
 246
 247                 if (MOD_RESULT == MOD_RES_DENY)
 248                         return MODEACTION_DENY;
 249                 if (MOD_RESULT == MOD_RES_PASSTHRU)
 250                 {
 251                         unsigned int neededrank = mh->GetLevelRequired();
 252                         /* Compare our rank on the channel against the rank of the required prefix,
 253                          * allow if >= ours. Because mIRC and xchat throw a tizz if the modes shown
 254                          * in NAMES(X) are not in rank order, we know the most powerful mode is listed
 255                          * first, so we don't need to iterate, we just look up the first instead.
 256                          */
 257                         unsigned int ourrank = chan->GetPrefixValue(user);
 258                         if (ourrank < neededrank)
 259                         {
 260                                 PrefixMode* neededmh = NULL;
 261                                 for(char c='A'; c <= 'z'; c++)
 262                                 {
 263                                         PrefixMode* privmh = FindPrefixMode(c);
 264                                         if (privmh && privmh->GetPrefixRank() >= neededrank)
 265                                         {
 266                                                 // this mode is sufficient to allow this action
 267                                                 if (!neededmh || privmh->GetPrefixRank() < neededmh->GetPrefixRank())
 268                                                         neededmh = privmh;
 269                                         }
 270                                 }
 271                                 if (neededmh)
 272                                         user->WriteNumeric(ERR_CHANOPRIVSNEEDED, "%s :You must have channel %s access or above to %sset channel mode %c",
 273                                                 chan->name.c_str(), neededmh->name.c_str(), adding ? "" : "un", modechar);
 274                                 else
 275                                         user->WriteNumeric(ERR_CHANOPRIVSNEEDED, "%s :You cannot %sset channel mode %c",
 276                                                 chan->name.c_str(), adding ? "" : "un", modechar);
 277                                 return MODEACTION_DENY;
 278                         }
 279                 }
 280         }
 281
 282         // Ask mode watchers whether this mode change is OK
 283         std::pair<ModeWatchIter, ModeWatchIter> itpair = modewatchermap.equal_range(mh->name);
 284         for (ModeWatchIter i = itpair.first; i != itpair.second; ++i)
 285         {
 286                 ModeWatcher* mw = i->second;
 287                 if (mw->GetModeType() == type)
 288                 {
 289                         if (!mw->BeforeMode(user, targetuser, chan, parameter, adding))
 290                                 return MODEACTION_DENY;
 291
 292                         // A module whacked the parameter completely, and there was one. Abort.
 293                         if (pcnt && parameter.empty())
 294                                 return MODEACTION_DENY;
 295                 }
 296         }
 297
 298         if (IS_LOCAL(user) && !user->IsOper())
 299         {
 300                 char* disabled = (type == MODETYPE_CHANNEL) ? ServerInstance->Config->DisabledCModes : ServerInstance->Config->DisabledUModes;
 301                 if (disabled[modechar - 'A'])
 302                 {
 303                         user->WriteNumeric(ERR_NOPRIVILEGES, ":Permission Denied - %s mode %c has been locked by the administrator",
 304                                 type == MODETYPE_CHANNEL ? "channel" : "user", modechar);
 305                         return MODEACTION_DENY;
 306                 }
 307         }
 308
 309         if (adding && IS_LOCAL(user) && mh->NeedsOper() && !user->HasModePermission(modechar, type))
 310         {
 311                 /* It's an oper only mode, and they don't have access to it. */
 312                 if (user->IsOper())
 313                 {
 314                         user->WriteNumeric(ERR_NOPRIVILEGES, ":Permission Denied - Oper type %s does not have access to set %s mode %c",
 315                                         user->oper->name.c_str(), type == MODETYPE_CHANNEL ? "channel" : "user", modechar);
 316                 }
 317                 else
 318                 {
 319                         user->WriteNumeric(ERR_NOPRIVILEGES, ":Permission Denied - Only operators may set %s mode %c",
 320                                         type == MODETYPE_CHANNEL ? "channel" : "user", modechar);
 321                 }
 322                 return MODEACTION_DENY;
 323         }
 324
 325         /* Call the handler for the mode */
 326         ModeAction ma = mh->OnModeChange(user, targetuser, chan, parameter, adding);
 327
 328         if (pcnt && parameter.empty())
 329                 return MODEACTION_DENY;
 330
 331         if (ma != MODEACTION_ALLOW)
 332                 return ma;
 333
 334         if ((!mh->IsListMode()) && (mh->GetNumParams(true)) && (chan))
 335                 chan->SetModeParam(mh, (adding ? parameter : ""));
 336
 337         itpair = modewatchermap.equal_range(mh->name);
 338         for (ModeWatchIter i = itpair.first; i != itpair.second; ++i)
 339         {
 340                 ModeWatcher* mw = i->second;
 341                 if (mw->GetModeType() == type)
 342                         mw->AfterMode(user, targetuser, chan, parameter, adding);
 343         }
 344
 345         return MODEACTION_ALLOW;
 346 }
 347
 348 void ModeParser::Process(const std::vector<std::string>& parameters, User* user, ModeProcessFlag flags)
 349 {
 350         std::string target = parameters[0];
 351         Channel* targetchannel = ServerInstance->FindChan(target);
 352         User* targetuser  = ServerInstance->FindNick(target);
 353         ModeType type = targetchannel ? MODETYPE_CHANNEL : MODETYPE_USER;
 354
 355         LastParse.clear();
 356         LastParseParams.clear();
 357         LastParseTranslate.clear();
 358
 359         if ((!targetchannel) && ((!targetuser) || (IS_SERVER(targetuser))))
 360         {
 361                 user->WriteNumeric(ERR_NOSUCHNICK, "%s :No such nick/channel", target.c_str());
 362                 return;
 363         }
 364         if (parameters.size() == 1)
 365         {
 366                 this->DisplayCurrentModes(user, targetuser, targetchannel, target.c_str());
 367                 return;
 368         }
 369
 370         ModResult MOD_RESULT;
 371         FIRST_MOD_RESULT(OnPreMode, MOD_RESULT, (user, targetuser, targetchannel, parameters));
 372
 373         bool SkipAccessChecks = false;
 374
 375         if (!IS_LOCAL(user) || MOD_RESULT == MOD_RES_ALLOW)
 376                 SkipAccessChecks = true;
 377         else if (MOD_RESULT == MOD_RES_DENY)
 378                 return;
 379
 380         if (targetuser && !SkipAccessChecks && user != targetuser)
 381         {
 382                 user->WriteNumeric(ERR_USERSDONTMATCH, ":Can't change mode for other users");
 383                 return;
 384         }
 385
 386         std::string mode_sequence = parameters[1];
 387
 388         std::string output_mode;
 389         std::ostringstream output_parameters;
 390         LastParseParams.push_back(output_mode);
 391         LastParseTranslate.push_back(TR_TEXT);
 392
 393         bool adding = true;
 394         char output_pm = '\0'; // current output state, '+' or '-'
 395         unsigned int param_at = 2;
 396
 397         for (std::string::const_iterator letter = mode_sequence.begin(); letter != mode_sequence.end(); letter++)
 398         {
 399                 unsigned char modechar = *letter;
 400                 if (modechar == '+' || modechar == '-')
 401                 {
 402                         adding = (modechar == '+');
 403                         continue;
 404                 }
 405
 406                 ModeHandler *mh = this->FindMode(modechar, type);
 407                 if (!mh)
 408                 {
 409                         /* No mode handler? Unknown mode character then. */
 410                         user->WriteNumeric(type == MODETYPE_CHANNEL ? ERR_UNKNOWNMODE : ERR_UNKNOWNSNOMASK, "%c :is unknown mode char to me", modechar);
 411                         continue;
 412                 }
 413
 414                 std::string parameter;
 415                 int pcnt = mh->GetNumParams(adding);
 416                 if (pcnt && param_at == parameters.size())
 417                 {
 418                         /* No parameter, continue to the next mode */
 419                         mh->OnParameterMissing(user, targetuser, targetchannel);
 420                         continue;
 421                 }
 422                 else if (pcnt)
 423                 {
 424                         parameter = parameters[param_at++];
 425                         /* Make sure the user isn't trying to slip in an invalid parameter */
 426                         if ((parameter.find(':') == 0) || (parameter.rfind(' ') != std::string::npos))
 427                                 continue;
 428                         if ((flags & MODE_MERGE) && targetchannel && targetchannel->IsModeSet(mh) && !mh->IsListMode())
 429                         {
 430                                 std::string ours = targetchannel->GetModeParameter(mh);
 431                                 if (!mh->ResolveModeConflict(parameter, ours, targetchannel))
 432                                         /* we won the mode merge, don't apply this mode */
 433                                         continue;
 434                         }
 435                 }
 436
 437                 ModeAction ma = TryMode(user, targetuser, targetchannel, adding, modechar, parameter, SkipAccessChecks);
 438
 439                 if (ma != MODEACTION_ALLOW)
 440                         continue;
 441
 442                 char needed_pm = adding ? '+' : '-';
 443                 if (needed_pm != output_pm)
 444                 {
 445                         output_pm = needed_pm;
 446                         output_mode.append(1, output_pm);
 447                 }
 448                 output_mode.append(1, modechar);
 449
 450                 if (pcnt)
 451                 {
 452                         output_parameters << " " << parameter;
 453                         LastParseParams.push_back(parameter);
 454                         LastParseTranslate.push_back(mh->GetTranslateType());
 455                 }
 456
 457                 if ( (output_mode.length() + output_parameters.str().length() > 450)
 458                                 || (output_mode.length() > 100)
 459                                 || (LastParseParams.size() > ServerInstance->Config->Limits.MaxModes))
 460                 {
 461                         /* mode sequence is getting too long */
 462                         break;
 463                 }
 464         }
 465
 466         LastParseParams[0] = output_mode;
 467
 468         if (!output_mode.empty())
 469         {
 470                 LastParse = targetchannel ? targetchannel->name : targetuser->nick;
 471                 LastParse.append(" ");
 472                 LastParse.append(output_mode);
 473                 LastParse.append(output_parameters.str());
 474
 475                 if (!(flags & MODE_LOCALONLY))
 476                         ServerInstance->PI->SendMode(user, targetuser, targetchannel, LastParseParams, LastParseTranslate);
 477
 478                 if (targetchannel)
 479                         targetchannel->WriteChannel(user, "MODE " + LastParse);
 480                 else
 481                         targetuser->WriteFrom(user, "MODE " + LastParse);
 482
 483                 FOREACH_MOD(OnMode, (user, targetuser, targetchannel, LastParseParams, LastParseTranslate));
 484         }
 485         else if (targetchannel && parameters.size() == 2)
 486         {
 487                 /* Special case for displaying the list for listmodes,
 488                  * e.g. MODE #chan b, or MODE #chan +b without a parameter
 489                  */
 490                 this->DisplayListModes(user, targetchannel, mode_sequence);
 491         }
 492 }
 493
 494 void ModeParser::DisplayListModes(User* user, Channel* chan, std::string &mode_sequence)
 495 {
 496         seq++;
 497
 498         for (std::string::const_iterator letter = mode_sequence.begin(); letter != mode_sequence.end(); letter++)
 499         {
 500                 unsigned char mletter = *letter;
 501                 if (mletter == '+')
 502                         continue;
 503
 504                 /* Ensure the user doesnt request the same mode twice,
 505                  * so they cant flood themselves off out of idiocy.
 506                  */
 507                 if (sent[mletter] == seq)
 508                         continue;
 509
 510                 sent[mletter] = seq;
 511
 512                 ModeHandler *mh = this->FindMode(mletter, MODETYPE_CHANNEL);
 513
 514                 if (!mh || !mh->IsListMode())
 515                         return;
 516
 517                 ModResult MOD_RESULT;
 518                 FIRST_MOD_RESULT(OnRawMode, MOD_RESULT, (user, chan, mletter, "", true, 0));
 519                 if (MOD_RESULT == MOD_RES_DENY)
 520                         continue;
 521
 522                 bool display = true;
 523                 if (!user->HasPrivPermission("channels/auspex") && ServerInstance->Config->HideModeLists[mletter] && (chan->GetPrefixValue(user) < HALFOP_VALUE))
 524                 {
 525                         user->WriteNumeric(ERR_CHANOPRIVSNEEDED, "%s :You do not have access to view the +%c list",
 526                                 chan->name.c_str(), mletter);
 527                         display = false;
 528                 }
 529
 530                 // Ask mode watchers whether it's OK to show the list
 531                 std::pair<ModeWatchIter, ModeWatchIter> itpair = modewatchermap.equal_range(mh->name);
 532                 for (ModeWatchIter i = itpair.first; i != itpair.second; ++i)
 533                 {
 534                         ModeWatcher* mw = i->second;
 535                         if (mw->GetModeType() == MODETYPE_CHANNEL)
 536                         {
 537                                 std::string dummyparam;
 538
 539                                 if (!mw->BeforeMode(user, NULL, chan, dummyparam, true))
 540                                 {
 541                                         // A mode watcher doesn't want us to show the list
 542                                         display = false;
 543                                         break;
 544                                 }
 545                         }
 546                 }
 547
 548                 if (display)
 549                         mh->DisplayList(user, chan);
 550                 else
 551                         mh->DisplayEmptyList(user, chan);
 552         }
 553 }
 554
 555 void ModeParser::CleanMask(std::string &mask)
 556 {
 557         std::string::size_type pos_of_pling = mask.find_first_of('!');
 558         std::string::size_type pos_of_at = mask.find_first_of('@');
 559         std::string::size_type pos_of_dot = mask.find_first_of('.');
 560         std::string::size_type pos_of_colons = mask.find("::"); /* Because ipv6 addresses are colon delimited -- double so it treats extban as nick */
 561
 562         if (mask.length() >= 2 && mask[1] == ':')
 563                 return; // if it's an extban, don't even try guess how it needs to be formed.
 564
 565         if ((pos_of_pling == std::string::npos) && (pos_of_at == std::string::npos))
 566         {
 567                 /* Just a nick, or just a host - or clearly ipv6 (starting with :) */
 568                 if ((pos_of_dot == std::string::npos) && (pos_of_colons == std::string::npos) && mask[0] != ':')
 569                 {
 570                         /* It has no '.' in it, it must be a nick. */
 571                         mask.append("!*@*");
 572                 }
 573                 else
 574                 {
 575                         /* Got a dot in it? Has to be a host */
 576                         mask = "*!*@" + mask;
 577                 }
 578         }
 579         else if ((pos_of_pling == std::string::npos) && (pos_of_at != std::string::npos))
 580         {
 581                 /* Has an @ but no !, its a user@host */
 582                  mask = "*!" + mask;
 583         }
 584         else if ((pos_of_pling != std::string::npos) && (pos_of_at == std::string::npos))
 585         {
 586                 /* Has a ! but no @, it must be a nick!ident */
 587                 mask.append("@*");
 588         }
 589 }
 590
 591 bool ModeParser::AddMode(ModeHandler* mh)
 592 {
 593         unsigned char mask = 0;
 594         unsigned char pos = 0;
 595
 596         /* Yes, i know, this might let people declare modes like '_' or '^'.
 597          * If they do that, thats their problem, and if i ever EVER see an
 598          * official InspIRCd developer do that, i'll beat them with a paddle!
 599          */
 600         if ((mh->GetModeChar() < 'A') || (mh->GetModeChar() > 'z'))
 601                 return false;
 602
 603         /* A mode prefix of ',' is not acceptable, it would fuck up server to server.
 604          * A mode prefix of ':' will fuck up both server to server, and client to server.
 605          * A mode prefix of '#' will mess up /whois and /privmsg
 606          */
 607         PrefixMode* pm = mh->IsPrefixMode();
 608         if (pm)
 609         {
 610                 if ((pm->GetPrefix() > 126) || (pm->GetPrefix() == ',') || (pm->GetPrefix() == ':') || (pm->GetPrefix() == '#'))
 611                         return false;
 612
 613                 if (FindPrefix(pm->GetPrefix()))
 614                         return false;
 615         }
 616
 617         mh->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
 618         pos = (mh->GetModeChar()-65) | mask;
 619
 620         if (modehandlers[pos])
 621                 return false;
 622
 623         // Everything is fine, add the mode
 624         modehandlers[pos] = mh;
 625         if (pm)
 626                 mhlist.prefix.push_back(pm);
 627         else if (mh->IsListModeBase())
 628                 mhlist.list.push_back(mh->IsListModeBase());
 629
 630         RecreateModeListFor004Numeric();
 631         return true;
 632 }
 633
 634 bool ModeParser::DelMode(ModeHandler* mh)
 635 {
 636         unsigned char mask = 0;
 637         unsigned char pos = 0;
 638
 639         if ((mh->GetModeChar() < 'A') || (mh->GetModeChar() > 'z'))
 640                 return false;
 641
 642         mh->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
 643         pos = (mh->GetModeChar()-65) | mask;
 644
 645         if (modehandlers[pos] != mh)
 646                 return false;
 647
 648         /* Note: We can't stack here, as we have modes potentially being removed across many different channels.
 649          * To stack here we have to make the algorithm slower. Discuss.
 650          */
 651         switch (mh->GetModeType())
 652         {
 653                 case MODETYPE_USER:
 654                         for (user_hash::iterator i = ServerInstance->Users->clientlist->begin(); i != ServerInstance->Users->clientlist->end(); )
 655                         {
 656                                 User* user = i->second;
 657                                 ++i;
 658                                 mh->RemoveMode(user);
 659                         }
 660                 break;
 661                 case MODETYPE_CHANNEL:
 662                         for (chan_hash::iterator i = ServerInstance->chanlist->begin(); i != ServerInstance->chanlist->end(); )
 663                         {
 664                                 // The channel may not be in the hash after RemoveMode(), see m_permchannels
 665                                 Channel* chan = i->second;
 666                                 ++i;
 667
 668                                 irc::modestacker stack(false);
 669                                 mh->RemoveMode(chan, stack);
 670
 671                                 std::vector<std::string> stackresult;
 672                                 stackresult.push_back(chan->name);
 673                                 while (stack.GetStackedLine(stackresult))
 674                                 {
 675                                         this->Process(stackresult, ServerInstance->FakeClient, MODE_LOCALONLY);
 676                                         stackresult.erase(stackresult.begin() + 1, stackresult.end());
 677                                 }
 678                         }
 679                 break;
 680         }
 681
 682         modehandlers[pos] = NULL;
 683         if (mh->IsPrefixMode())
 684                 mhlist.prefix.erase(std::find(mhlist.prefix.begin(), mhlist.prefix.end(), mh->IsPrefixMode()));
 685         else if (mh->IsListModeBase())
 686                 mhlist.list.erase(std::find(mhlist.list.begin(), mhlist.list.end(), mh->IsListModeBase()));
 687
 688         RecreateModeListFor004Numeric();
 689         return true;
 690 }
 691
 692 ModeHandler* ModeParser::FindMode(unsigned const char modeletter, ModeType mt)
 693 {
 694         unsigned char mask = 0;
 695         unsigned char pos = 0;
 696
 697         if ((modeletter < 'A') || (modeletter > 'z'))
 698                 return NULL;
 699
 700         mt == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
 701         pos = (modeletter-65) | mask;
 702
 703         return modehandlers[pos];
 704 }
 705
 706 PrefixMode* ModeParser::FindPrefixMode(unsigned char modeletter)
 707 {
 708         ModeHandler* mh = FindMode(modeletter, MODETYPE_CHANNEL);
 709         if (!mh)
 710                 return NULL;
 711         return mh->IsPrefixMode();
 712 }
 713
 714 std::string ModeParser::CreateModeList(ModeType mt, bool needparam)
 715 {
 716         std::string modestr;
 717         unsigned char mask = ((mt == MODETYPE_CHANNEL) ? MASK_CHANNEL : MASK_USER);
 718
 719         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
 720         {
 721                 unsigned char pos = (mode-65) | mask;
 722
 723                 if ((modehandlers[pos]) && ((!needparam) || (modehandlers[pos]->GetNumParams(true))))
 724                         modestr.push_back(mode);
 725         }
 726
 727         return modestr;
 728 }
 729
 730 void ModeParser::RecreateModeListFor004Numeric()
 731 {
 732         Cached004ModeList = CreateModeList(MODETYPE_USER) + " " + CreateModeList(MODETYPE_CHANNEL) + " " + CreateModeList(MODETYPE_CHANNEL, true);
 733 }
 734
 735 PrefixMode* ModeParser::FindPrefix(unsigned const char pfxletter)
 736 {
 737         const PrefixModeList& list = GetPrefixModes();
 738         for (PrefixModeList::const_iterator i = list.begin(); i != list.end(); ++i)
 739         {
 740                 PrefixMode* pm = *i;
 741                 if (pm->GetPrefix() == pfxletter)
 742                         return pm;
 743         }
 744         return NULL;
 745 }
 746
 747 std::string ModeParser::GiveModeList(ModeMasks m)
 748 {
 749         std::string type1;      /* Listmodes EXCEPT those with a prefix */
 750         std::string type2;      /* Modes that take a param when adding or removing */
 751         std::string type3;      /* Modes that only take a param when adding */
 752         std::string type4;      /* Modes that dont take a param */
 753
 754         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
 755         {
 756                 unsigned char pos = (mode-65) | m;
 757                  /* One parameter when adding */
 758                 if (modehandlers[pos])
 759                 {
 760                         if (modehandlers[pos]->GetNumParams(true))
 761                         {
 762                                 PrefixMode* pm = modehandlers[pos]->IsPrefixMode();
 763                                 if ((modehandlers[pos]->IsListMode()) && ((!pm) || (pm->GetPrefix() == 0)))
 764                                 {
 765                                         type1 += modehandlers[pos]->GetModeChar();
 766                                 }
 767                                 else
 768                                 {
 769                                         /* ... and one parameter when removing */
 770                                         if (modehandlers[pos]->GetNumParams(false))
 771                                         {
 772                                                 /* But not a list mode */
 773                                                 if (!pm)
 774                                                 {
 775                                                         type2 += modehandlers[pos]->GetModeChar();
 776                                                 }
 777                                         }
 778                                         else
 779                                         {
 780                                                 /* No parameters when removing */
 781                                                 type3 += modehandlers[pos]->GetModeChar();
 782                                         }
 783                                 }
 784                         }
 785                         else
 786                         {
 787                                 type4 += modehandlers[pos]->GetModeChar();
 788                         }
 789                 }
 790         }
 791
 792         return type1 + "," + type2 + "," + type3 + "," + type4;
 793 }
 794
 795 std::string ModeParser::BuildPrefixes(bool lettersAndModes)
 796 {
 797         std::string mletters;
 798         std::string mprefixes;
 799         std::map<int,std::pair<char,char> > prefixes;
 800
 801         const PrefixModeList& list = GetPrefixModes();
 802         for (PrefixModeList::const_iterator i = list.begin(); i != list.end(); ++i)
 803         {
 804                 PrefixMode* pm = *i;
 805                 if (pm->GetPrefix())
 806                         prefixes[pm->GetPrefixRank()] = std::make_pair(pm->GetPrefix(), pm->GetModeChar());
 807         }
 808
 809         for(std::map<int,std::pair<char,char> >::reverse_iterator n = prefixes.rbegin(); n != prefixes.rend(); n++)
 810         {
 811                 mletters = mletters + n->second.first;
 812                 mprefixes = mprefixes + n->second.second;
 813         }
 814
 815         return lettersAndModes ? "(" + mprefixes + ")" + mletters : mletters;
 816 }
 817
 818 void ModeParser::AddModeWatcher(ModeWatcher* mw)
 819 {
 820         modewatchermap.insert(std::make_pair(mw->GetModeName(), mw));
 821 }
 822
 823 bool ModeParser::DelModeWatcher(ModeWatcher* mw)
 824 {
 825         std::pair<ModeWatchIter, ModeWatchIter> itpair = modewatchermap.equal_range(mw->GetModeName());
 826         for (ModeWatchIter i = itpair.first; i != itpair.second; ++i)
 827         {
 828                 if (i->second == mw)
 829                 {
 830                         modewatchermap.erase(i);
 831                         return true;
 832                 }
 833         }
 834
 835         return false;
 836 }
 837
 838 void ModeHandler::RemoveMode(User* user)
 839 {
 840         // Remove the mode if it's set on the user
 841         if (user->IsModeSet(this->GetModeChar()))
 842         {
 843                 std::vector<std::string> parameters;
 844                 parameters.push_back(user->nick);
 845                 parameters.push_back("-");
 846                 parameters[1].push_back(this->GetModeChar());
 847                 ServerInstance->Modes->Process(parameters, ServerInstance->FakeClient, ModeParser::MODE_LOCALONLY);
 848         }
 849 }
 850
 851 void ModeHandler::RemoveMode(Channel* channel, irc::modestacker& stack)
 852 {
 853         if (channel->IsModeSet(this))
 854         {
 855                 if (this->GetNumParams(false))
 856                         // Removing this mode requires a parameter
 857                         stack.Push(this->GetModeChar(), channel->GetModeParameter(this));
 858                 else
 859                         stack.Push(this->GetModeChar());
 860         }
 861 }
 862
 863 void PrefixMode::RemoveMode(Channel* chan, irc::modestacker& stack)
 864 {
 865         const UserMembList* userlist = chan->GetUsers();
 866         for (UserMembCIter i = userlist->begin(); i != userlist->end(); ++i)
 867         {
 868                 if (i->second->hasMode(this->GetModeChar()))
 869                         stack.Push(this->GetModeChar(), i->first->nick);
 870         }
 871 }
 872
 873 struct builtin_modes
 874 {
 875         ModeChannelSecret s;
 876         ModeChannelPrivate p;
 877         ModeChannelModerated m;
 878         ModeChannelTopicOps t;
 879
 880         ModeChannelNoExternal n;
 881         ModeChannelInviteOnly i;
 882         ModeChannelKey k;
 883         ModeChannelLimit l;
 884
 885         ModeChannelBan b;
 886         ModeChannelOp o;
 887         ModeChannelVoice v;
 888
 889         ModeUserWallops uw;
 890         ModeUserInvisible ui;
 891         ModeUserOperator uo;
 892         ModeUserServerNoticeMask us;
 893
 894         void init()
 895         {
 896                 ServiceProvider* modes[] = { &s, &p, &m, &t, &n, &i, &k, &l, &b, &o, &v,
 897                                                                          &uw, &ui, &uo, &us };
 898                 ServerInstance->Modules->AddServices(modes, sizeof(modes)/sizeof(ServiceProvider*));
 899         }
 900 };
 901
 902 static builtin_modes static_modes;
 903
 904 void ModeParser::InitBuiltinModes()
 905 {
 906         static_modes.init();
 907         static_modes.b.DoRehash();
 908 }
 909
 910 ModeParser::ModeParser()
 911 {
 912         /* Clear mode handler list */
 913         memset(modehandlers, 0, sizeof(modehandlers));
 914
 915         seq = 0;
 916         memset(&sent, 0, sizeof(sent));
 917 }
 918
 919 ModeParser::~ModeParser()
 920 {
 921 }