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