src/mode.cpp

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