src/mode.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
   6  * See: http://www.inspircd.org/wiki/index.php/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 "users.h"
  16 #include "modules.h"
  17 #include "inspstring.h"
  18 #include "mode.h"
  19
  20 /* +s (secret) */
  21 #include "modes/cmode_s.h"
  22 /* +p (private) */
  23 #include "modes/cmode_p.h"
  24 /* +b (bans) */
  25 #include "modes/cmode_b.h"
  26 /* +m (moderated) */
  27 #include "modes/cmode_m.h"
  28 /* +t (only (half) ops can change topic) */
  29 #include "modes/cmode_t.h"
  30 /* +n (no external messages) */
  31 #include "modes/cmode_n.h"
  32 /* +i (invite only) */
  33 #include "modes/cmode_i.h"
  34 /* +k (keyed channel) */
  35 #include "modes/cmode_k.h"
  36 /* +l (channel user limit) */
  37 #include "modes/cmode_l.h"
  38 /* +o (channel op) */
  39 #include "modes/cmode_o.h"
  40 /* +h (channel halfop) */
  41 #include "modes/cmode_h.h"
  42 /* +v (channel voice) */
  43 #include "modes/cmode_v.h"
  44 /* +s (server notices) */
  45 #include "modes/umode_s.h"
  46 /* +w (see wallops) */
  47 #include "modes/umode_w.h"
  48 /* +i (invisible) */
  49 #include "modes/umode_i.h"
  50 /* +o (operator) */
  51 #include "modes/umode_o.h"
  52 /* +n (notice mask - our implementation of snomasks) */
  53 #include "modes/umode_n.h"
  54
  55 ModeHandler::ModeHandler(InspIRCd* Instance, char modeletter, int parameters_on, int parameters_off, bool listmode, ModeType type, bool operonly, char mprefix)
  56         : ServerInstance(Instance), mode(modeletter), n_params_on(parameters_on), n_params_off(parameters_off), list(listmode), m_type(type), oper(operonly), prefix(mprefix), count(0)
  57 {
  58 }
  59
  60 ModeHandler::~ModeHandler()
  61 {
  62 }
  63
  64 bool ModeHandler::IsListMode()
  65 {
  66         return list;
  67 }
  68
  69 unsigned int ModeHandler::GetPrefixRank()
  70 {
  71         return 0;
  72 }
  73
  74 unsigned int ModeHandler::GetCount()
  75 {
  76         return 0;
  77 }
  78
  79 void ModeHandler::ChangeCount(int modifier)
  80 {
  81         count += modifier;
  82 }
  83
  84 ModeType ModeHandler::GetModeType()
  85 {
  86         return m_type;
  87 }
  88
  89 bool ModeHandler::NeedsOper()
  90 {
  91         return oper;
  92 }
  93
  94 char ModeHandler::GetPrefix()
  95 {
  96         return prefix;
  97 }
  98
  99 int ModeHandler::GetNumParams(bool adding)
 100 {
 101         return adding ? n_params_on : n_params_off;
 102 }
 103
 104 char ModeHandler::GetModeChar()
 105 {
 106         return mode;
 107 }
 108
 109 ModeAction ModeHandler::OnModeChange(userrec* source, userrec* dest, chanrec* channel, std::string &parameter, bool adding)
 110 {
 111         return MODEACTION_DENY;
 112 }
 113
 114 ModePair ModeHandler::ModeSet(userrec* source, userrec* dest, chanrec* channel, const std::string &parameter)
 115 {
 116         if (dest)
 117         {
 118                 return std::make_pair(dest->IsModeSet(this->mode), "");
 119         }
 120         else
 121         {
 122                 return std::make_pair(channel->IsModeSet(this->mode), "");
 123         }
 124 }
 125
 126 void ModeHandler::DisplayList(userrec* user, chanrec* channel)
 127 {
 128 }
 129
 130 bool ModeHandler::CheckTimeStamp(time_t theirs, time_t ours, const std::string &their_param, const std::string &our_param, chanrec* channel)
 131 {
 132         return (ours < theirs);
 133 }
 134
 135 ModeWatcher::ModeWatcher(InspIRCd* Instance, char modeletter, ModeType type) : ServerInstance(Instance), mode(modeletter), m_type(type)
 136 {
 137 }
 138
 139 ModeWatcher::~ModeWatcher()
 140 {
 141 }
 142
 143 char ModeWatcher::GetModeChar()
 144 {
 145         return mode;
 146 }
 147
 148 ModeType ModeWatcher::GetModeType()
 149 {
 150         return m_type;
 151 }
 152
 153 bool ModeWatcher::BeforeMode(userrec* source, userrec* dest, chanrec* channel, std::string &parameter, bool adding, ModeType type)
 154 {
 155         return true;
 156 }
 157
 158 void ModeWatcher::AfterMode(userrec* source, userrec* dest, chanrec* channel, const std::string &parameter, bool adding, ModeType type)
 159 {
 160 }
 161
 162 userrec* ModeParser::SanityChecks(userrec *user,const char *dest,chanrec *chan,int status)
 163 {
 164         userrec *d;
 165         if ((!user) || (!dest) || (!chan) || (!*dest))
 166         {
 167                 return NULL;
 168         }
 169         d = ServerInstance->FindNick(dest);
 170         if (!d)
 171         {
 172                 user->WriteServ("401 %s %s :No such nick/channel",user->nick, dest);
 173                 return NULL;
 174         }
 175         return d;
 176 }
 177
 178 const char* ModeParser::Grant(userrec *d,chanrec *chan,int MASK)
 179 {
 180         if (!chan)
 181                 return "";
 182
 183         UCListIter n = d->chans.find(chan);
 184         if (n != d->chans.end())
 185         {
 186                 if (n->second & MASK)
 187                 {
 188                         return "";
 189                 }
 190                 n->second = n->second | MASK;
 191                 switch (MASK)
 192                 {
 193                         case UCMODE_OP:
 194                                 n->first->AddOppedUser(d);
 195                         break;
 196                         case UCMODE_HOP:
 197                                 n->first->AddHalfoppedUser(d);
 198                         break;
 199                         case UCMODE_VOICE:
 200                                 n->first->AddVoicedUser(d);
 201                         break;
 202                 }
 203                 return d->nick;
 204         }
 205         return "";
 206 }
 207
 208 const char* ModeParser::Revoke(userrec *d,chanrec *chan,int MASK)
 209 {
 210         if (!chan)
 211                 return "";
 212
 213         UCListIter n = d->chans.find(chan);
 214         if (n != d->chans.end())
 215         {
 216                 if ((n->second & MASK) == 0)
 217                 {
 218                         return "";
 219                 }
 220                 n->second ^= MASK;
 221                 switch (MASK)
 222                 {
 223                         case UCMODE_OP:
 224                                 n->first->DelOppedUser(d);
 225                         break;
 226                         case UCMODE_HOP:
 227                                 n->first->DelHalfoppedUser(d);
 228                         break;
 229                         case UCMODE_VOICE:
 230                                 n->first->DelVoicedUser(d);
 231                         break;
 232                 }
 233                 return d->nick;
 234         }
 235         return "";
 236 }
 237
 238 void ModeParser::DisplayCurrentModes(userrec *user, userrec* targetuser, chanrec* targetchannel, const char* text)
 239 {
 240         if (targetchannel)
 241         {
 242                 /* Display channel's current mode string */
 243                 user->WriteServ("324 %s %s +%s",user->nick, targetchannel->name, targetchannel->ChanModes(targetchannel->HasUser(user)));
 244                 user->WriteServ("329 %s %s %lu", user->nick, targetchannel->name, (unsigned long)targetchannel->age);
 245                 return;
 246         }
 247         else if (targetuser)
 248         {
 249                 if ((targetuser == user) || (*user->oper))
 250                 {
 251                         /* Display user's current mode string */
 252                         user->WriteServ("221 %s :+%s",targetuser->nick,targetuser->FormatModes());
 253                         if (*targetuser->oper)
 254                                 user->WriteServ("008 %s +%s :Server notice mask", targetuser->nick, targetuser->FormatNoticeMasks());
 255                         return;
 256                 }
 257                 else
 258                 {
 259                         user->WriteServ("502 %s :Can't change mode for other users", user->nick);
 260                         return;
 261                 }
 262         }
 263
 264         /* No such nick/channel */
 265         user->WriteServ("401 %s %s :No such nick/channel",user->nick, text);
 266         return;
 267 }
 268
 269 void ModeParser::Process(const char** parameters, int pcnt, userrec *user, bool servermode)
 270 {
 271         std::string target = parameters[0];
 272         ModeType type = MODETYPE_USER;
 273         unsigned char mask = 0;
 274         chanrec* targetchannel = ServerInstance->FindChan(parameters[0]);
 275         userrec* targetuser  = ServerInstance->FindNick(parameters[0]);
 276
 277         LastParse = "";
 278
 279         /* Special case for displaying the list for listmodes,
 280          * e.g. MODE #chan b, or MODE #chan +b without a parameter
 281          */
 282         if ((targetchannel) && (pcnt == 2))
 283         {
 284                 const char* mode = parameters[1];
 285                 int nonlistmodes_found = 0;
 286                 bool sent[256];
 287
 288                 mask = MASK_CHANNEL;
 289
 290                 memset(&sent, 0, 256);
 291
 292                 while (mode && *mode)
 293                 {
 294                         unsigned char mletter = *mode;
 295
 296                         if (*mode == '+')
 297                         {
 298                                 mode++;
 299                                 continue;
 300                         }
 301
 302                         /* Ensure the user doesnt request the same mode twice,
 303                          * so they cant flood themselves off out of idiocy.
 304                          */
 305                         if (!sent[mletter])
 306                         {
 307                                 sent[mletter] = true;
 308                         }
 309                         else
 310                         {
 311                                 mode++;
 312                                 continue;
 313                         }
 314
 315                         ModeHandler *mh = this->FindMode(*mode, MODETYPE_CHANNEL);
 316                         bool display = true;
 317
 318                         if ((mh) && (mh->IsListMode()))
 319                         {
 320                                 if (ServerInstance->Config->HideModeLists[mletter] && (targetchannel->GetStatus(user) < STATUS_HOP))
 321                                 {
 322                                         user->WriteServ("482 %s %s :Only half-operators and above may view the +%c list",user->nick, targetchannel->name, *mode++);
 323                                         continue;
 324                                 }
 325
 326                                 /** See below for a description of what craq this is :D
 327                                  */
 328                                 unsigned char handler_id = (*mode - 65) | mask;
 329
 330                                 for(ModeWatchIter watchers = modewatchers[handler_id].begin(); watchers != modewatchers[handler_id].end(); watchers++)
 331                                 {
 332                                         std::string dummyparam;
 333
 334                                         if (!((*watchers)->BeforeMode(user, NULL, targetchannel, dummyparam, true, MODETYPE_CHANNEL)))
 335                                                 display = false;
 336                                 }
 337
 338                                 if (display)
 339                                         mh->DisplayList(user, targetchannel);
 340                         }
 341                         else
 342                                 nonlistmodes_found++;
 343
 344                         mode++;
 345                 }
 346
 347                 /* We didnt have any modes that were non-list, we can return here */
 348                 if (!nonlistmodes_found)
 349                         return;
 350         }
 351
 352         if (pcnt == 1)
 353         {
 354                 this->DisplayCurrentModes(user, targetuser, targetchannel, parameters[0]);
 355         }
 356         else if (pcnt > 1)
 357         {
 358                 if (targetchannel)
 359                 {
 360                         type = MODETYPE_CHANNEL;
 361                         mask = MASK_CHANNEL;
 362
 363                         /* Extra security checks on channel modes
 364                          * (e.g. are they a (half)op?
 365                          */
 366
 367                         if ((IS_LOCAL(user)) && (targetchannel->GetStatus(user) < STATUS_HOP))
 368                         {
 369                                 /* We don't have halfop */
 370                                 int MOD_RESULT = 0;
 371                                 FOREACH_RESULT(I_OnAccessCheck,OnAccessCheck(user, NULL, targetchannel, AC_GENERAL_MODE));
 372                                 if (MOD_RESULT == ACR_DENY)
 373                                         return;
 374
 375                                 if (MOD_RESULT == ACR_DEFAULT)
 376                                 {
 377                                         /* Are we a uline or is it a servermode? */
 378                                         if ((!ServerInstance->ULine(user->server)) && (!servermode))
 379                                         {
 380                                                 /* Not enough permission:
 381                                                  * NOT a uline and NOT a servermode,
 382                                                  * OR, NOT halfop or above.
 383                                                  */
 384                                                 user->WriteServ("482 %s %s :You're not a channel (half)operator",user->nick, targetchannel->name);
 385                                                 return;
 386                                         }
 387                                 }
 388                         }
 389                 }
 390                 else if (targetuser)
 391                 {
 392                         type = MODETYPE_USER;
 393                         mask = MASK_USER;
 394                         if ((user != targetuser) && (!ServerInstance->ULine(user->server)))
 395                         {
 396                                 user->WriteServ("502 %s :Can't change mode for other users", user->nick);
 397                                 return;
 398                         }
 399                 }
 400                 else
 401                 {
 402                         /* No such nick/channel */
 403                         user->WriteServ("401 %s %s :No such nick/channel",user->nick, parameters[0]);
 404                         return;
 405                 }
 406
 407                 std::string mode_sequence = parameters[1];
 408                 std::string parameter = "";
 409                 std::ostringstream parameter_list;
 410                 std::string output_sequence = "";
 411                 bool adding = true, state_change = false;
 412                 unsigned char handler_id = 0;
 413                 int parameter_counter = 2; /* Index of first parameter */
 414                 int parameter_count = 0;
 415                 bool last_successful_state_change = false;
 416
 417                 /* A mode sequence that doesnt start with + or -. Assume +. - Thanks for the suggestion spike (bug#132) */
 418                 if ((*mode_sequence.begin() != '+') && (*mode_sequence.begin() != '-'))
 419                         mode_sequence.insert(0, "+");
 420
 421                 for (std::string::const_iterator letter = mode_sequence.begin(); letter != mode_sequence.end(); letter++)
 422                 {
 423                         unsigned char modechar = *letter;
 424
 425                         switch (modechar)
 426                         {
 427                                 /* NB:
 428                                  * For + and - mode characters, we don't just stick the character into the output sequence.
 429                                  * This is because the user may do something dumb, like: +-+ooo or +oo-+. To prevent this
 430                                  * appearing in the output sequence, we store a flag which says there was a state change,
 431                                  * which is set on any + or -, however, the + or - that we finish on is only appended to
 432                                  * the output stream in the event it is followed by a non "+ or -" character, such as o or v.
 433                                  */
 434                                 case '+':
 435                                         /* The following expression prevents: +o+o nick nick, compressing it to +oo nick nick,
 436                                          * however, will allow the + if it is the first item in the sequence, regardless.
 437                                          */
 438                                         if ((!adding) || (!output_sequence.length()))
 439                                                 state_change = true;
 440                                         adding = true;
 441                                         if (!output_sequence.length())
 442                                                 last_successful_state_change = false;
 443                                         continue;
 444                                 break;
 445                                 case '-':
 446                                         if ((adding) || (!output_sequence.length()))
 447                                                 state_change = true;
 448                                         adding = false;
 449                                         if (!output_sequence.length())
 450                                                 last_successful_state_change = true;
 451                                         continue;
 452                                 break;
 453                                 default:
 454
 455                                         /**
 456                                          * Watch carefully for the sleight of hand trick.
 457                                          * 65 is the ascii value of 'A'. We take this from
 458                                          * the char we're looking at to get a number between
 459                                          * 1 and 127. We then logic-or it to get the hashed
 460                                          * position, dependent on wether its a channel or
 461                                          * a user mode. This is a little stranger, but a lot
 462                                          * faster, than using a map of pairs.
 463                                          */
 464                                         handler_id = (modechar - 65) | mask;
 465
 466                                         if (modehandlers[handler_id])
 467                                         {
 468                                                 bool abort = false;
 469
 470                                                 if (modehandlers[handler_id]->GetModeType() == type)
 471                                                 {
 472                                                         if (modehandlers[handler_id]->GetNumParams(adding))
 473                                                         {
 474                                                                 /* This mode expects a parameter, do we have any parameters left in our list to use? */
 475                                                                 if (parameter_counter < pcnt)
 476                                                                 {
 477                                                                         parameter = parameters[parameter_counter++];
 478
 479                                                                         /* Yerk, invalid! */
 480                                                                         if ((parameter.find(':') == 0) || (parameter.rfind(' ') != std::string::npos))
 481                                                                                 parameter = "";
 482                                                                 }
 483                                                                 else
 484                                                                 {
 485                                                                         /* No parameter, continue to the next mode */
 486                                                                         continue;
 487                                                                 }
 488
 489                                                                 bool had_parameter = !parameter.empty();
 490
 491                                                                 for (ModeWatchIter watchers = modewatchers[handler_id].begin(); watchers != modewatchers[handler_id].end(); watchers++)
 492                                                                 {
 493                                                                         ServerInstance->Log(DEBUG,"Call mode watcher");
 494                                                                         if ((*watchers)->BeforeMode(user, targetuser, targetchannel, parameter, adding, type) == false)
 495                                                                         {
 496                                                                                 abort = true;
 497                                                                                 break;
 498                                                                         }
 499                                                                         /* A module whacked the parameter completely, and there was one. abort. */
 500                                                                         if ((had_parameter) && (parameter.empty()))
 501                                                                         {
 502                                                                                 abort = true;
 503                                                                                 break;
 504                                                                         }
 505                                                                 }
 506
 507                                                                 if (abort)
 508                                                                         continue;
 509                                                         }
 510                                                         else
 511                                                         {
 512                                                                 /* Fix by brain: mode watchers not being called for parameterless modes */
 513                                                                 for (ModeWatchIter watchers = modewatchers[handler_id].begin(); watchers != modewatchers[handler_id].end(); watchers++)
 514                                                                 {
 515                                                                         if ((*watchers)->BeforeMode(user, targetuser, targetchannel, parameter, adding, type) == false)
 516                                                                         {
 517                                                                                 abort = true;
 518                                                                                 break;
 519                                                                         }
 520                                                                 }
 521
 522                                                                 if (abort)
 523                                                                         continue;
 524                                                         }
 525
 526                                                         /* It's an oper only mode, check if theyre an oper. If they arent,
 527                                                          * eat any parameter that  came with the mode, and continue to next
 528                                                          */
 529                                                         if ((IS_LOCAL(user)) && (modehandlers[handler_id]->NeedsOper()) && (!*user->oper))
 530                                                         {
 531                                                                 user->WriteServ("481 %s :Permission Denied - Only IRC operators may %sset %s mode %c", user->nick,
 532                                                                                 adding ? "" : "un", type == MODETYPE_CHANNEL ? "channel" : "user",
 533                                                                                 modehandlers[handler_id]->GetModeChar());
 534                                                                 continue;
 535                                                         }
 536
 537                                                         /* Call the handler for the mode */
 538                                                         ModeAction ma = modehandlers[handler_id]->OnModeChange(user, targetuser, targetchannel, parameter, adding);
 539
 540                                                         if ((modehandlers[handler_id]->GetNumParams(adding)) && (parameter == ""))
 541                                                         {
 542                                                                 /* The handler nuked the parameter and they are supposed to have one.
 543                                                                  * We CANT continue now, even if they actually returned MODEACTION_ALLOW,
 544                                                                  * so we bail to the next mode character.
 545                                                                  */
 546                                                                 continue;
 547                                                         }
 548
 549                                                         if (ma == MODEACTION_ALLOW)
 550                                                         {
 551                                                                 /* We're about to output a valid mode letter - was there previously a pending state-change? */
 552                                                                 if (state_change)
 553                                                                 {
 554                                                                         if (adding != last_successful_state_change)
 555                                                                                 output_sequence.append(adding ? "+" : "-");
 556                                                                         last_successful_state_change = adding;
 557                                                                 }
 558
 559                                                                 /* Add the mode letter */
 560                                                                 output_sequence.push_back(modechar);
 561
 562                                                                 /* Is there a valid parameter for this mode? If so add it to the parameter list */
 563                                                                 if ((modehandlers[handler_id]->GetNumParams(adding)) && (parameter != ""))
 564                                                                 {
 565                                                                         parameter_list << " " << parameter;
 566                                                                         parameter_count++;
 567                                                                         /* Does this mode have a prefix? */
 568                                                                         if (modehandlers[handler_id]->GetPrefix() && targetchannel)
 569                                                                         {
 570                                                                                 userrec* user_to_prefix = ServerInstance->FindNick(parameter);
 571                                                                                 if (user_to_prefix)
 572                                                                                         targetchannel->SetPrefix(user_to_prefix, modehandlers[handler_id]->GetPrefix(),
 573                                                                                                         modehandlers[handler_id]->GetPrefixRank(), adding);
 574                                                                         }
 575                                                                 }
 576
 577                                                                 /* Call all the AfterMode events in the mode watchers for this mode */
 578                                                                 for (ModeWatchIter watchers = modewatchers[handler_id].begin(); watchers != modewatchers[handler_id].end(); watchers++)
 579                                                                         (*watchers)->AfterMode(user, targetuser, targetchannel, parameter, adding, type);
 580
 581                                                                 /* Reset the state change flag */
 582                                                                 state_change = false;
 583
 584                                                                 if ((output_sequence.length() + parameter_list.str().length() > 450) || (output_sequence.length() > 100)
 585                                                                                 || (parameter_count > MAXMODES))
 586                                                                 {
 587                                                                         /* We cant have a mode sequence this long */
 588                                                                         letter = mode_sequence.end() - 1;
 589                                                                         continue;
 590                                                                 }
 591                                                         }
 592                                                 }
 593                                         }
 594                                         else
 595                                         {
 596                                                 /* No mode handler? Unknown mode character then. */
 597                                                 user->WriteServ("472 %s %c :is unknown mode char to me",user->nick, modechar);
 598                                         }
 599                                 break;
 600                         }
 601                 }
 602                 /* Was there at least one valid mode in the sequence? */
 603                 if (output_sequence != "")
 604                 {
 605                         if (servermode)
 606                         {
 607                                 if (type == MODETYPE_CHANNEL)
 608                                 {
 609                                         targetchannel->WriteChannelWithServ(ServerInstance->Config->ServerName, "MODE %s %s%s", targetchannel->name, output_sequence.c_str(), parameter_list.str().c_str());
 610                                         this->LastParse = targetchannel->name;
 611                                 }
 612                                 else
 613                                 {
 614                                         targetuser->WriteServ("MODE %s %s%s",targetuser->nick,output_sequence.c_str(), parameter_list.str().c_str());
 615                                         this->LastParse = targetuser->nick;
 616                                 }
 617                         }
 618                         else
 619                         {
 620                                 if (type == MODETYPE_CHANNEL)
 621                                 {
 622                                         targetchannel->WriteChannel(user,"MODE %s %s%s",targetchannel->name,output_sequence.c_str(),parameter_list.str().c_str());
 623                                         FOREACH_MOD(I_OnMode,OnMode(user, targetchannel, TYPE_CHANNEL, output_sequence + parameter_list.str()));
 624                                         this->LastParse = targetchannel->name;
 625                                 }
 626                                 else
 627                                 {
 628                                         user->WriteTo(targetuser,"MODE %s %s%s",targetuser->nick,output_sequence.c_str(), parameter_list.str().c_str());
 629                                         FOREACH_MOD(I_OnMode,OnMode(user, targetuser, TYPE_USER, output_sequence + parameter_list.str()));
 630                                         this->LastParse = targetuser->nick;
 631                                 }
 632                         }
 633
 634                         LastParse.append(" ");
 635                         LastParse.append(output_sequence);
 636                         LastParse.append(parameter_list.str());
 637                 }
 638         }
 639 }
 640
 641 const std::string& ModeParser::GetLastParse()
 642 {
 643         return LastParse;
 644 }
 645
 646 void ModeParser::CleanMask(std::string &mask)
 647 {
 648         std::string::size_type pos_of_pling = mask.find_first_of('!');
 649         std::string::size_type pos_of_at = mask.find_first_of('@');
 650         std::string::size_type pos_of_dot = mask.find_first_of('.');
 651         std::string::size_type pos_of_colon = mask.find_first_of(':'); /* Because ipv6 addresses are colon delimited */
 652
 653         if ((pos_of_pling == std::string::npos) && (pos_of_at == std::string::npos))
 654         {
 655                 /* Just a nick, or just a host */
 656                 if ((pos_of_dot == std::string::npos) && (pos_of_colon == std::string::npos))
 657                 {
 658                         /* It has no '.' in it, it must be a nick. */
 659                         mask.append("!*@*");
 660                 }
 661                 else
 662                 {
 663                         /* Got a dot in it? Has to be a host */
 664                         mask = "*!*@" + mask;
 665                 }
 666         }
 667         else if ((pos_of_pling == std::string::npos) && (pos_of_at != std::string::npos))
 668         {
 669                 /* Has an @ but no !, its a user@host */
 670                  mask = "*!" + mask;
 671         }
 672         else if ((pos_of_pling != std::string::npos) && (pos_of_at == std::string::npos))
 673         {
 674                 /* Has a ! but no @, it must be a nick!ident */
 675                 mask.append("@*");
 676         }
 677 }
 678
 679 bool ModeParser::AddMode(ModeHandler* mh, unsigned const char modeletter)
 680 {
 681         unsigned char mask = 0;
 682         unsigned char pos = 0;
 683
 684         /* Yes, i know, this might let people declare modes like '_' or '^'.
 685          * If they do that, thats their problem, and if i ever EVER see an
 686          * official InspIRCd developer do that, i'll beat them with a paddle!
 687          */
 688         if ((mh->GetModeChar() < 'A') || (mh->GetModeChar() > 'z'))
 689                 return false;
 690
 691         /* A mode prefix of ',' is not acceptable, it would fuck up server to server.
 692          * A mode prefix of ':' will fuck up both server to server, and client to server.
 693          * A mode prefix of '#' will mess up /whois and /privmsg
 694          */
 695         if ((mh->GetPrefix() == ',') || (mh->GetPrefix() == ':') || (mh->GetPrefix() == '#'))
 696                 return false;
 697
 698         mh->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
 699         pos = (mh->GetModeChar()-65) | mask;
 700
 701         if (modehandlers[pos])
 702                 return false;
 703
 704         modehandlers[pos] = mh;
 705         return true;
 706 }
 707
 708 bool ModeParser::DelMode(ModeHandler* mh)
 709 {
 710         unsigned char mask = 0;
 711         unsigned char pos = 0;
 712
 713         if ((mh->GetModeChar() < 'A') || (mh->GetModeChar() > 'z'))
 714                 return false;
 715
 716         mh->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
 717         pos = (mh->GetModeChar()-65) | mask;
 718
 719         if (!modehandlers[pos])
 720                 return false;
 721
 722         switch (mh->GetModeType())
 723         {
 724                 case MODETYPE_USER:
 725                         for (user_hash::iterator i = ServerInstance->clientlist->begin(); i != ServerInstance->clientlist->end(); i++)
 726                         {
 727                                 mh->RemoveMode(i->second);
 728                         }
 729                 break;
 730                 case MODETYPE_CHANNEL:
 731                         for (chan_hash::iterator i = ServerInstance->chanlist->begin(); i != ServerInstance->chanlist->end(); i++)
 732                         {
 733                                 mh->RemoveMode(i->second);
 734                         }
 735                 break;
 736         }
 737
 738         modehandlers[pos] = NULL;
 739
 740         return true;
 741 }
 742
 743 ModeHandler* ModeParser::FindMode(unsigned const char modeletter, ModeType mt)
 744 {
 745         unsigned char mask = 0;
 746         unsigned char pos = 0;
 747
 748         if ((modeletter < 'A') || (modeletter > 'z'))
 749                 return NULL;
 750
 751         mt == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
 752         pos = (modeletter-65) | mask;
 753
 754         return modehandlers[pos];
 755 }
 756
 757 std::string ModeParser::UserModeList()
 758 {
 759         char modestr[256];
 760         int pointer = 0;
 761
 762         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
 763         {
 764                 unsigned char pos = (mode-65) | MASK_USER;
 765
 766                 if (modehandlers[pos])
 767                         modestr[pointer++] = mode;
 768         }
 769         modestr[pointer++] = 0;
 770         return modestr;
 771 }
 772
 773 std::string ModeParser::ChannelModeList()
 774 {
 775         char modestr[256];
 776         int pointer = 0;
 777
 778         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
 779         {
 780                 unsigned char pos = (mode-65) | MASK_CHANNEL;
 781
 782                 if (modehandlers[pos])
 783                         modestr[pointer++] = mode;
 784         }
 785         modestr[pointer++] = 0;
 786         return modestr;
 787 }
 788
 789 std::string ModeParser::ParaModeList()
 790 {
 791         char modestr[256];
 792         int pointer = 0;
 793
 794         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
 795         {
 796                 unsigned char pos = (mode-65) | MASK_CHANNEL;
 797
 798                 if ((modehandlers[pos]) && (modehandlers[pos]->GetNumParams(true)))
 799                         modestr[pointer++] = mode;
 800         }
 801         modestr[pointer++] = 0;
 802         return modestr;
 803 }
 804
 805 ModeHandler* ModeParser::FindPrefix(unsigned const char pfxletter)
 806 {
 807         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
 808         {
 809                 unsigned char pos = (mode-65) | MASK_CHANNEL;
 810
 811                 if ((modehandlers[pos]) && (modehandlers[pos]->GetPrefix() == pfxletter))
 812                 {
 813                         return modehandlers[pos];
 814                 }
 815         }
 816         return NULL;
 817 }
 818
 819 std::string ModeParser::ModeString(userrec* user, chanrec* channel)
 820 {
 821         std::string types;
 822         std::string pars;
 823
 824         if (!channel || !user)
 825                 return "";
 826
 827         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
 828         {
 829                 unsigned char pos = (mode-65) | MASK_CHANNEL;
 830                 ModeHandler* mh = modehandlers[pos];
 831                 if ((mh) && (mh->GetNumParams(true)) && (mh->GetNumParams(false)))
 832                 {
 833                         ModePair ret;
 834                         ret = mh->ModeSet(NULL, user, channel, user->nick);
 835                         if ((ret.first) && (ret.second == user->nick))
 836                         {
 837                                 pars.append(" ");
 838                                 pars.append(user->nick);
 839                                 types.push_back(mh->GetModeChar());
 840                         }
 841                 }
 842         }
 843
 844         return types+pars;
 845 }
 846
 847 std::string ModeParser::ChanModes()
 848 {
 849         std::string type1;      /* Listmodes EXCEPT those with a prefix */
 850         std::string type2;      /* Modes that take a param when adding or removing */
 851         std::string type3;      /* Modes that only take a param when adding */
 852         std::string type4;      /* Modes that dont take a param */
 853
 854         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
 855         {
 856                 unsigned char pos = (mode-65) | MASK_CHANNEL;
 857                  /* One parameter when adding */
 858                 if (modehandlers[pos])
 859                 {
 860                         if (modehandlers[pos]->GetNumParams(true))
 861                         {
 862                                 if ((modehandlers[pos]->IsListMode()) && (!modehandlers[pos]->GetPrefix()))
 863                                 {
 864                                         type1 += modehandlers[pos]->GetModeChar();
 865                                 }
 866                                 else
 867                                 {
 868                                         /* ... and one parameter when removing */
 869                                         if (modehandlers[pos]->GetNumParams(false))
 870                                         {
 871                                                 /* But not a list mode */
 872                                                 if (!modehandlers[pos]->GetPrefix())
 873                                                 {
 874                                                         type2 += modehandlers[pos]->GetModeChar();
 875                                                 }
 876                                         }
 877                                         else
 878                                         {
 879                                                 /* No parameters when removing */
 880                                                 type3 += modehandlers[pos]->GetModeChar();
 881                                         }
 882                                 }
 883                         }
 884                         else
 885                         {
 886                                 type4 += modehandlers[pos]->GetModeChar();
 887                         }
 888                 }
 889
 890         }
 891
 892         return type1 + "," + type2 + "," + type3 + "," + type4;
 893 }
 894
 895 bool ModeParser::PrefixComparison(const prefixtype one, const prefixtype two)
 896 {
 897         return one.second > two.second;
 898 }
 899
 900 std::string ModeParser::BuildPrefixes()
 901 {
 902         std::string mletters = "";
 903         std::string mprefixes = "";
 904         pfxcontainer pfx;
 905         std::map<char,char> prefix_to_mode;
 906
 907         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
 908         {
 909                 unsigned char pos = (mode-65) | MASK_CHANNEL;
 910
 911                 if ((modehandlers[pos]) && (modehandlers[pos]->GetPrefix()))
 912                 {
 913                         pfx.push_back(std::make_pair<char,unsigned int>(modehandlers[pos]->GetPrefix(), modehandlers[pos]->GetPrefixRank()));
 914                         prefix_to_mode[modehandlers[pos]->GetPrefix()] = modehandlers[pos]->GetModeChar();
 915                 }
 916         }
 917
 918         sort(pfx.begin(), pfx.end(), ModeParser::PrefixComparison);
 919
 920         for (pfxcontainer::iterator n = pfx.begin(); n != pfx.end(); n++)
 921         {
 922                 mletters = mletters + n->first;
 923                 mprefixes = mprefixes + prefix_to_mode.find(n->first)->second;
 924         }
 925
 926         return "(" + mprefixes + ")" + mletters;
 927 }
 928
 929 bool ModeParser::AddModeWatcher(ModeWatcher* mw)
 930 {
 931         unsigned char mask = 0;
 932         unsigned char pos = 0;
 933
 934         if (!mw)
 935                 return false;
 936
 937         if ((mw->GetModeChar() < 'A') || (mw->GetModeChar() > 'z'))
 938                 return false;
 939
 940         mw->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
 941         pos = (mw->GetModeChar()-65) | mask;
 942
 943         modewatchers[pos].push_back(mw);
 944
 945         return true;
 946 }
 947
 948 bool ModeParser::DelModeWatcher(ModeWatcher* mw)
 949 {
 950         unsigned char mask = 0;
 951         unsigned char pos = 0;
 952
 953         if (!mw)
 954                 return false;
 955
 956         if ((mw->GetModeChar() < 'A') || (mw->GetModeChar() > 'z'))
 957                 return false;
 958
 959         mw->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
 960         pos = (mw->GetModeChar()-65) | mask;
 961
 962         ModeWatchIter a = find(modewatchers[pos].begin(),modewatchers[pos].end(),mw);
 963
 964         if (a == modewatchers[pos].end())
 965         {
 966                 return false;
 967         }
 968
 969         modewatchers[pos].erase(a);
 970
 971         return true;
 972 }
 973
 974 /** This default implementation can remove simple user modes
 975  */
 976 void ModeHandler::RemoveMode(userrec* user)
 977 {
 978         char moderemove[MAXBUF];
 979         const char* parameters[] = { user->nick, moderemove };
 980
 981         if (user->IsModeSet(this->GetModeChar()))
 982         {
 983                 sprintf(moderemove,"-%c",this->GetModeChar());
 984                 ServerInstance->Parser->CallHandler("MODE", parameters, 2, user);
 985         }
 986 }
 987
 988 /** This default implementation can remove simple channel modes
 989  * (no parameters)
 990  */
 991 void ModeHandler::RemoveMode(chanrec* channel)
 992 {
 993         char moderemove[MAXBUF];
 994         const char* parameters[] = { channel->name, moderemove };
 995
 996         if (channel->IsModeSet(this->GetModeChar()))
 997         {
 998                 userrec* n = new userrec(ServerInstance);
 999
1000                 sprintf(moderemove,"-%c",this->GetModeChar());
1001                 n->SetFd(FD_MAGIC_NUMBER);
1002
1003                 ServerInstance->SendMode(parameters, 2, n);
1004
1005                 delete n;
1006         }
1007 }
1008
1009 ModeParser::ModeParser(InspIRCd* Instance) : ServerInstance(Instance)
1010 {
1011         struct Initializer
1012         {
1013                 char modechar;
1014                 ModeHandler* handler;
1015         };
1016
1017         Initializer modes[] = {
1018                 { 's', new ModeChannelSecret(Instance) },
1019                 { 'p', new ModeChannelPrivate(Instance) },
1020                 { 'm', new ModeChannelModerated(Instance) },
1021                 { 't', new ModeChannelTopicOps(Instance) },
1022                 { 'n', new ModeChannelNoExternal(Instance) },
1023                 { 'i', new ModeChannelInviteOnly(Instance) },
1024                 { 'k', new ModeChannelKey(Instance) },
1025                 { 'l', new ModeChannelLimit(Instance) },
1026                 { 'b', new ModeChannelBan(Instance) },
1027                 { 'o', new ModeChannelOp(Instance) },
1028                 { 'h', new ModeChannelHalfOp(Instance) },
1029                 { 'v', new ModeChannelVoice(Instance) },
1030                 { 's', new ModeUserServerNotice(Instance) },
1031                 { 'w', new ModeUserWallops(Instance) },
1032                 { 'i', new ModeUserInvisible(Instance) },
1033                 { 'o', new ModeUserOperator(Instance) },
1034                 { 'n', new ModeUserServerNoticeMask(Instance) },
1035                 { 0, NULL }
1036         };
1037
1038         /* Clear mode list */
1039         memset(modehandlers, 0, sizeof(modehandlers));
1040         memset(modewatchers, 0, sizeof(modewatchers));
1041
1042         /* Last parse string */
1043         LastParse = "";
1044
1045         /* Initialise the RFC mode letters */
1046         for (int index = 0; modes[index].modechar; index++)
1047                 this->AddMode(modes[index].handler, modes[index].modechar);
1048 }