src/mode.cpp

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