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