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