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