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