]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/mode.cpp
Get rid of ModePair
[user/henk/code/inspircd.git] / src / mode.cpp
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
6  * See: http://wiki.inspircd.org/Credits
7  *
8  * This program is free but copyrighted software; see
9  *            the file COPYING for details.
10  *
11  * ---------------------------------------------------
12  */
13
14 #include "inspircd.h"
15 #include "inspstring.h"
16
17 /* +s (secret) */
18 #include "modes/cmode_s.h"
19 /* +p (private) */
20 #include "modes/cmode_p.h"
21 /* +b (bans) */
22 #include "modes/cmode_b.h"
23 /* +m (moderated) */
24 #include "modes/cmode_m.h"
25 /* +t (only (half) ops can change topic) */
26 #include "modes/cmode_t.h"
27 /* +n (no external messages) */
28 #include "modes/cmode_n.h"
29 /* +i (invite only) */
30 #include "modes/cmode_i.h"
31 /* +k (keyed channel) */
32 #include "modes/cmode_k.h"
33 /* +l (channel user limit) */
34 #include "modes/cmode_l.h"
35 /* +o (channel op) */
36 #include "modes/cmode_o.h"
37 /* +h (channel halfop) */
38 #include "modes/cmode_h.h"
39 /* +v (channel voice) */
40 #include "modes/cmode_v.h"
41 /* +w (see wallops) */
42 #include "modes/umode_w.h"
43 /* +i (invisible) */
44 #include "modes/umode_i.h"
45 /* +o (operator) */
46 #include "modes/umode_o.h"
47 /* +s (server notice masks) */
48 #include "modes/umode_s.h"
49
50 ModeHandler::ModeHandler(Module* Creator, const std::string& Name, char modeletter, ParamSpec Params, ModeType type)
51         : m_paramtype(TR_TEXT), parameters_taken(Params), mode(modeletter), prefix(0), oper(false),
52         list(false), m_type(type), count(0), levelrequired(HALFOP_VALUE), creator(Creator), name(Name)
53 {
54 }
55
56 CullResult ModeHandler::cull()
57 {
58         if (ServerInstance->Modes)
59                 ServerInstance->Modes->DelMode(this);
60         return classbase::cull();
61 }
62
63 ModeHandler::~ModeHandler()
64 {
65         if (ServerInstance && ServerInstance->Modes && ServerInstance->Modes->FindMode(mode, m_type) == this)
66                 ServerInstance->Logs->Log("MODE", ERROR, "ERROR: Destructor for mode %c called while still registered", mode);
67 }
68
69 bool ModeHandler::IsListMode()
70 {
71         return list;
72 }
73
74 unsigned int ModeHandler::GetPrefixRank()
75 {
76         return 0;
77 }
78
79 unsigned int ModeHandler::GetCount()
80 {
81         return 0;
82 }
83
84 void ModeHandler::ChangeCount(int modifier)
85 {
86         count += modifier;
87         ServerInstance->Logs->Log("MODE", DEBUG,"Change count for mode %c is now %d", mode, count);
88 }
89
90 int ModeHandler::GetNumParams(bool adding)
91 {
92         switch (parameters_taken)
93         {
94                 case PARAM_ALWAYS:
95                         return 1;
96                 case PARAM_SETONLY:
97                         return adding ? 1 : 0;
98                 case PARAM_NONE:
99                         break;
100         }
101         return 0;
102 }
103
104 char ModeHandler::GetModeChar()
105 {
106         return mode;
107 }
108
109 std::string ModeHandler::GetUserParameter(User* user)
110 {
111         return "";
112 }
113
114 ModResult ModeHandler::AccessCheck(User*, Channel*, std::string &, bool)
115 {
116         return MOD_RES_PASSTHRU;
117 }
118
119 ModeAction ModeHandler::OnModeChange(User*, User*, Channel*, std::string&, bool)
120 {
121         return MODEACTION_DENY;
122 }
123
124 void ModeHandler::DisplayList(User*, Channel*)
125 {
126 }
127
128 void ModeHandler::DisplayEmptyList(User*, Channel*)
129 {
130 }
131
132 void ModeHandler::OnParameterMissing(User* user, User* dest, Channel* channel)
133 {
134 }
135
136 bool ModeHandler::ResolveModeConflict(std::string& theirs, const std::string& ours, Channel*)
137 {
138         return (theirs < ours);
139 }
140
141 ModeAction SimpleUserModeHandler::OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
142 {
143         if (adding)
144         {
145                 if (!dest->IsModeSet(this->GetModeChar()))
146                 {
147                         dest->SetMode(this->GetModeChar(),true);
148                         return MODEACTION_ALLOW;
149                 }
150         }
151         else
152         {
153                 if (dest->IsModeSet(this->GetModeChar()))
154                 {
155                         dest->SetMode(this->GetModeChar(),false);
156                         return MODEACTION_ALLOW;
157                 }
158         }
159
160         return MODEACTION_DENY;
161 }
162
163
164 ModeAction SimpleChannelModeHandler::OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
165 {
166         if (adding)
167         {
168                 if (!channel->IsModeSet(this->GetModeChar()))
169                 {
170                         channel->SetMode(this->GetModeChar(),true);
171                         return MODEACTION_ALLOW;
172                 }
173         }
174         else
175         {
176                 if (channel->IsModeSet(this->GetModeChar()))
177                 {
178                         channel->SetMode(this->GetModeChar(),false);
179                         return MODEACTION_ALLOW;
180                 }
181         }
182
183         return MODEACTION_DENY;
184 }
185
186 ModeWatcher::ModeWatcher(Module* Creator, char modeletter, ModeType type)
187         : mode(modeletter), m_type(type), creator(Creator)
188 {
189 }
190
191 ModeWatcher::~ModeWatcher()
192 {
193 }
194
195 char ModeWatcher::GetModeChar()
196 {
197         return mode;
198 }
199
200 ModeType ModeWatcher::GetModeType()
201 {
202         return m_type;
203 }
204
205 bool ModeWatcher::BeforeMode(User*, User*, Channel*, std::string&, bool, ModeType)
206 {
207         return true;
208 }
209
210 void ModeWatcher::AfterMode(User*, User*, Channel*, const std::string&, bool, ModeType)
211 {
212 }
213
214 User* ModeParser::SanityChecks(User *user, const char *dest, Channel *chan, int)
215 {
216         User *d;
217         if ((!user) || (!dest) || (!chan) || (!*dest))
218         {
219                 return NULL;
220         }
221         d = ServerInstance->FindNick(dest);
222         if (!d)
223         {
224                 user->WriteNumeric(ERR_NOSUCHNICK, "%s %s :No such nick/channel",user->nick.c_str(), dest);
225                 return NULL;
226         }
227         return d;
228 }
229
230 void ModeParser::DisplayCurrentModes(User *user, User* targetuser, Channel* targetchannel, const char* text)
231 {
232         if (targetchannel)
233         {
234                 /* Display channel's current mode string */
235                 user->WriteNumeric(RPL_CHANNELMODEIS, "%s %s +%s",user->nick.c_str(), targetchannel->name.c_str(), targetchannel->ChanModes(targetchannel->HasUser(user)));
236                 user->WriteNumeric(RPL_CHANNELCREATED, "%s %s %lu", user->nick.c_str(), targetchannel->name.c_str(), (unsigned long)targetchannel->age);
237                 return;
238         }
239         else
240         {
241                 if (targetuser == user || user->HasPrivPermission("users/auspex"))
242                 {
243                         /* Display user's current mode string */
244                         user->WriteNumeric(RPL_UMODEIS, "%s :+%s",targetuser->nick.c_str(),targetuser->FormatModes());
245                         if (IS_OPER(targetuser))
246                                 user->WriteNumeric(RPL_SNOMASKIS, "%s +%s :Server notice mask", targetuser->nick.c_str(), targetuser->FormatNoticeMasks());
247                         return;
248                 }
249                 else
250                 {
251                         user->WriteNumeric(ERR_USERSDONTMATCH, "%s :Can't view modes for other users", user->nick.c_str());
252                         return;
253                 }
254         }
255 }
256
257 ModeAction ModeParser::TryMode(User* user, User* targetuser, Channel* chan, bool adding, const unsigned char modechar,
258                 std::string &parameter, bool SkipACL)
259 {
260         ModeType type = chan ? MODETYPE_CHANNEL : MODETYPE_USER;
261         unsigned char mask = chan ? MASK_CHANNEL : MASK_USER;
262
263         ModeHandler *mh = FindMode(modechar, type);
264         int pcnt = mh->GetNumParams(adding);
265
266         ModResult MOD_RESULT;
267         FIRST_MOD_RESULT(OnRawMode, MOD_RESULT, (user, chan, modechar, parameter, adding, pcnt));
268
269         if (IS_LOCAL(user) && (MOD_RESULT == MOD_RES_DENY))
270                 return MODEACTION_DENY;
271
272         if (chan && !SkipACL && (MOD_RESULT != MOD_RES_ALLOW))
273         {
274                 MOD_RESULT = mh->AccessCheck(user, chan, parameter, adding);
275
276                 if (MOD_RESULT == MOD_RES_DENY)
277                         return MODEACTION_DENY;
278                 if (MOD_RESULT == MOD_RES_PASSTHRU)
279                 {
280                         unsigned int neededrank = mh->GetLevelRequired();
281                         /* Compare our rank on the channel against the rank of the required prefix,
282                          * allow if >= ours. Because mIRC and xchat throw a tizz if the modes shown
283                          * in NAMES(X) are not in rank order, we know the most powerful mode is listed
284                          * first, so we don't need to iterate, we just look up the first instead.
285                          */
286                         unsigned int ourrank = chan->GetPrefixValue(user);
287                         if (ourrank < neededrank)
288                         {
289                                 /* Bog off */
290                                 // TODO replace with a real search for the proper prefix
291                                 char needed = neededrank > HALFOP_VALUE ? '@' : '%';
292                                 user->WriteNumeric(ERR_CHANOPRIVSNEEDED, "%s %s :You must have channel privilege %c or above to %sset channel mode %c",
293                                                 user->nick.c_str(), chan->name.c_str(), needed, adding ? "" : "un", modechar);
294                                 return MODEACTION_DENY;
295                         }
296                 }
297         }
298
299         unsigned char handler_id = (modechar - 'A') | mask;
300
301         for (ModeWatchIter watchers = modewatchers[handler_id].begin(); watchers != modewatchers[handler_id].end(); watchers++)
302         {
303                 if ((*watchers)->BeforeMode(user, targetuser, chan, parameter, adding, type) == false)
304                         return MODEACTION_DENY;
305                 /* A module whacked the parameter completely, and there was one. abort. */
306                 if (pcnt && parameter.empty())
307                         return MODEACTION_DENY;
308         }
309
310         if (IS_LOCAL(user) && !IS_OPER(user))
311         {
312                 char* disabled = (type == MODETYPE_CHANNEL) ? ServerInstance->Config->DisabledCModes : ServerInstance->Config->DisabledUModes;
313                 if (disabled[modechar - 'A'])
314                 {
315                         user->WriteNumeric(ERR_NOPRIVILEGES, "%s :Permission Denied - %s mode %c has been locked by the administrator",
316                                 user->nick.c_str(), type == MODETYPE_CHANNEL ? "channel" : "user", modechar);
317                         return MODEACTION_DENY;
318                 }
319         }
320
321         if (adding && IS_LOCAL(user) && mh->NeedsOper() && !user->HasModePermission(modechar, type))
322         {
323                 /* It's an oper only mode, and they don't have access to it. */
324                 if (IS_OPER(user))
325                 {
326                         user->WriteNumeric(ERR_NOPRIVILEGES, "%s :Permission Denied - Oper type %s does not have access to set %s mode %c",
327                                         user->nick.c_str(), user->oper->NameStr(), type == MODETYPE_CHANNEL ? "channel" : "user", modechar);
328                 }
329                 else
330                 {
331                         user->WriteNumeric(ERR_NOPRIVILEGES, "%s :Permission Denied - Only operators may set %s mode %c",
332                                         user->nick.c_str(), type == MODETYPE_CHANNEL ? "channel" : "user", modechar);
333                 }
334                 return MODEACTION_DENY;
335         }
336
337         if (mh->GetTranslateType() == TR_NICK && !ServerInstance->FindNick(parameter))
338         {
339                 user->WriteNumeric(ERR_NOSUCHNICK, "%s %s :No such nick/channel", user->nick.c_str(), parameter.c_str());
340                 return MODEACTION_DENY;
341         }
342
343         /* Call the handler for the mode */
344         ModeAction ma = mh->OnModeChange(user, targetuser, chan, parameter, adding);
345
346         if (pcnt && parameter.empty())
347                 return MODEACTION_DENY;
348
349         if (ma != MODEACTION_ALLOW)
350                 return ma;
351
352         mh->ChangeCount(adding ? 1 : -1);
353
354         if (mh->GetPrefixRank() && chan)
355         {
356                 User* user_to_prefix = ServerInstance->FindNick(parameter);
357                 if (user_to_prefix)
358                         chan->SetPrefix(user_to_prefix, modechar, adding);
359         }
360
361         for (ModeWatchIter watchers = modewatchers[handler_id].begin(); watchers != modewatchers[handler_id].end(); watchers++)
362                 (*watchers)->AfterMode(user, targetuser, chan, parameter, adding, type);
363
364         return MODEACTION_ALLOW;
365 }
366
367 void ModeParser::Process(const std::vector<std::string>& parameters, User *user, bool merge)
368 {
369         std::string target = parameters[0];
370         Channel* targetchannel = ServerInstance->FindChan(target);
371         User* targetuser  = ServerInstance->FindNick(target);
372         ModeType type = targetchannel ? MODETYPE_CHANNEL : MODETYPE_USER;
373
374         LastParse.clear();
375         LastParseParams.clear();
376         LastParseTranslate.clear();
377
378         if (!targetchannel && !targetuser)
379         {
380                 user->WriteNumeric(ERR_NOSUCHNICK, "%s %s :No such nick/channel",user->nick.c_str(),target.c_str());
381                 return;
382         }
383         if (parameters.size() == 1)
384         {
385                 this->DisplayCurrentModes(user, targetuser, targetchannel, target.c_str());
386                 return;
387         }
388
389         ModResult MOD_RESULT;
390         FIRST_MOD_RESULT(OnPreMode, MOD_RESULT, (user, targetuser, targetchannel, parameters));
391
392         bool SkipAccessChecks = false;
393
394         if (!IS_LOCAL(user) || ServerInstance->ULine(user->server) || MOD_RESULT == MOD_RES_ALLOW)
395                 SkipAccessChecks = true;
396         else if (MOD_RESULT == MOD_RES_DENY)
397                 return;
398
399         if (targetuser && !SkipAccessChecks && user != targetuser)
400         {
401                 user->WriteNumeric(ERR_USERSDONTMATCH, "%s :Can't change mode for other users", user->nick.c_str());
402                 return;
403         }
404
405         std::string mode_sequence = parameters[1];
406
407         std::string output_mode;
408         std::ostringstream output_parameters;
409         LastParseParams.push_back(output_mode);
410         LastParseTranslate.push_back(TR_TEXT);
411
412         bool adding = true;
413         char output_pm = '\0'; // current output state, '+' or '-'
414         unsigned int param_at = 2;
415
416         for (std::string::const_iterator letter = mode_sequence.begin(); letter != mode_sequence.end(); letter++)
417         {
418                 unsigned char modechar = *letter;
419                 if (modechar == '+' || modechar == '-')
420                 {
421                         adding = (modechar == '+');
422                         continue;
423                 }
424
425                 ModeHandler *mh = this->FindMode(modechar, type);
426                 if (!mh)
427                 {
428                         /* No mode handler? Unknown mode character then. */
429                         user->WriteServ("%d %s %c :is unknown mode char to me", type == MODETYPE_CHANNEL ? 472 : 501, user->nick.c_str(), modechar);
430                         continue;
431                 }
432
433                 std::string parameter = "";
434                 int pcnt = mh->GetNumParams(adding);
435                 if (pcnt && param_at == parameters.size())
436                 {
437                         /* No parameter, continue to the next mode */
438                         mh->OnParameterMissing(user, targetuser, targetchannel);
439                         continue;
440                 }
441                 else if (pcnt)
442                 {
443                         parameter = parameters[param_at++];
444                         /* Make sure the user isn't trying to slip in an invalid parameter */
445                         if ((parameter.find(':') == 0) || (parameter.rfind(' ') != std::string::npos))
446                                 continue;
447                         if (merge && targetchannel && targetchannel->IsModeSet(modechar) && !mh->IsListMode())
448                         {
449                                 std::string ours = targetchannel->GetModeParameter(modechar);
450                                 if (!mh->ResolveModeConflict(parameter, ours, targetchannel))
451                                         /* we won the mode merge, don't apply this mode */
452                                         continue;
453                         }
454                 }
455
456                 ModeAction ma = TryMode(user, targetuser, targetchannel, adding, modechar, parameter, SkipAccessChecks);
457
458                 if (ma != MODEACTION_ALLOW)
459                         continue;
460
461                 char needed_pm = adding ? '+' : '-';
462                 if (needed_pm != output_pm)
463                 {
464                         output_pm = needed_pm;
465                         output_mode.append(1, output_pm);
466                 }
467                 output_mode.append(1, modechar);
468
469                 if (pcnt)
470                 {
471                         TranslateType tt = mh->GetTranslateType();
472                         if (tt == TR_NICK)
473                         {
474                                 User* u = ServerInstance->FindNick(parameter);
475                                 if (u)
476                                         parameter = u->nick;
477                         }
478                         output_parameters << " " << parameter;
479                         LastParseParams.push_back(parameter);
480                         LastParseTranslate.push_back(tt);
481                 }
482
483                 if ( (output_mode.length() + output_parameters.str().length() > 450)
484                                 || (output_mode.length() > 100)
485                                 || (LastParseParams.size() > ServerInstance->Config->Limits.MaxModes))
486                 {
487                         /* mode sequence is getting too long */
488                         break;
489                 }
490         }
491
492         LastParseParams[0] = output_mode;
493
494         if (!output_mode.empty())
495         {
496                 LastParse = targetchannel ? targetchannel->name : targetuser->nick;
497                 LastParse.append(" ");
498                 LastParse.append(output_mode);
499                 LastParse.append(output_parameters.str());
500
501                 if (targetchannel)
502                 {
503                         targetchannel->WriteChannel(user, "MODE %s", LastParse.c_str());
504                         FOREACH_MOD(I_OnMode,OnMode(user, targetchannel, TYPE_CHANNEL, LastParseParams, LastParseTranslate));
505                 }
506                 else
507                 {
508                         targetuser->WriteFrom(user, "MODE %s", LastParse.c_str());
509                         FOREACH_MOD(I_OnMode,OnMode(user, targetuser, TYPE_USER, LastParseParams, LastParseTranslate));
510                 }
511         }
512         else if (targetchannel && parameters.size() == 2)
513         {
514                 /* Special case for displaying the list for listmodes,
515                  * e.g. MODE #chan b, or MODE #chan +b without a parameter
516                  */
517                 this->DisplayListModes(user, targetchannel, mode_sequence);
518         }
519 }
520
521 void ModeParser::DisplayListModes(User* user, Channel* chan, std::string &mode_sequence)
522 {
523         seq++;
524
525         for (std::string::const_iterator letter = mode_sequence.begin(); letter != mode_sequence.end(); letter++)
526         {
527                 unsigned char mletter = *letter;
528                 if (mletter == '+')
529                         continue;
530
531                 /* Ensure the user doesnt request the same mode twice,
532                  * so they cant flood themselves off out of idiocy.
533                  */
534                 if (sent[mletter] == seq)
535                         continue;
536
537                 sent[mletter] = seq;
538
539                 ModeHandler *mh = this->FindMode(mletter, MODETYPE_CHANNEL);
540
541                 if (!mh || !mh->IsListMode())
542                         return;
543
544                 ModResult MOD_RESULT;
545                 FIRST_MOD_RESULT(OnRawMode, MOD_RESULT, (user, chan, mletter, "", true, 0));
546                 if (MOD_RESULT == MOD_RES_DENY)
547                         continue;
548
549                 bool display = true;
550                 if (!user->HasPrivPermission("channels/auspex") && ServerInstance->Config->HideModeLists[mletter] && (chan->GetPrefixValue(user) < HALFOP_VALUE))
551                 {
552                         user->WriteNumeric(ERR_CHANOPRIVSNEEDED, "%s %s :Only half-operators and above may view the +%c list",
553                                 user->nick.c_str(), chan->name.c_str(), mletter);
554                         display = false;
555                 }
556
557                 /** See below for a description of what craq this is :D
558                  */
559                 unsigned char handler_id = (mletter - 'A') | MASK_CHANNEL;
560
561                 for(ModeWatchIter watchers = modewatchers[handler_id].begin(); watchers != modewatchers[handler_id].end(); watchers++)
562                 {
563                         std::string dummyparam;
564
565                         if (!((*watchers)->BeforeMode(user, NULL, chan, dummyparam, true, MODETYPE_CHANNEL)))
566                                 display = false;
567                 }
568                 if (display)
569                         mh->DisplayList(user, chan);
570                 else
571                         mh->DisplayEmptyList(user, chan);
572         }
573 }
574
575 const std::string& ModeParser::GetLastParse()
576 {
577         return LastParse;
578 }
579
580 void ModeParser::CleanMask(std::string &mask)
581 {
582         std::string::size_type pos_of_pling = mask.find_first_of('!');
583         std::string::size_type pos_of_at = mask.find_first_of('@');
584         std::string::size_type pos_of_dot = mask.find_first_of('.');
585         std::string::size_type pos_of_colons = mask.find("::"); /* Because ipv6 addresses are colon delimited -- double so it treats extban as nick */
586
587         if (mask.length() >= 2 && mask[1] == ':')
588                 return; // if it's an extban, don't even try guess how it needs to be formed.
589
590         if ((pos_of_pling == std::string::npos) && (pos_of_at == std::string::npos))
591         {
592                 /* Just a nick, or just a host - or clearly ipv6 (starting with :) */
593                 if ((pos_of_dot == std::string::npos) && (pos_of_colons == std::string::npos) && mask[0] != ':')
594                 {
595                         /* It has no '.' in it, it must be a nick. */
596                         mask.append("!*@*");
597                 }
598                 else
599                 {
600                         /* Got a dot in it? Has to be a host */
601                         mask = "*!*@" + mask;
602                 }
603         }
604         else if ((pos_of_pling == std::string::npos) && (pos_of_at != std::string::npos))
605         {
606                 /* Has an @ but no !, its a user@host */
607                  mask = "*!" + mask;
608         }
609         else if ((pos_of_pling != std::string::npos) && (pos_of_at == std::string::npos))
610         {
611                 /* Has a ! but no @, it must be a nick!ident */
612                 mask.append("@*");
613         }
614 }
615
616 bool ModeParser::AddMode(ModeHandler* mh)
617 {
618         unsigned char mask = 0;
619         unsigned char pos = 0;
620
621         /* Yes, i know, this might let people declare modes like '_' or '^'.
622          * If they do that, thats their problem, and if i ever EVER see an
623          * official InspIRCd developer do that, i'll beat them with a paddle!
624          */
625         if ((mh->GetModeChar() < 'A') || (mh->GetModeChar() > 'z') || (mh->GetPrefix() > 126))
626                 return false;
627
628         /* A mode prefix of ',' is not acceptable, it would fuck up server to server.
629          * A mode prefix of ':' will fuck up both server to server, and client to server.
630          * A mode prefix of '#' will mess up /whois and /privmsg
631          */
632         if ((mh->GetPrefix() == ',') || (mh->GetPrefix() == ':') || (mh->GetPrefix() == '#'))
633                 return false;
634
635         mh->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
636         pos = (mh->GetModeChar()-65) | mask;
637
638         if (modehandlers[pos])
639                 return false;
640
641         modehandlers[pos] = mh;
642         return true;
643 }
644
645 bool ModeParser::DelMode(ModeHandler* mh)
646 {
647         unsigned char mask = 0;
648         unsigned char pos = 0;
649
650         if ((mh->GetModeChar() < 'A') || (mh->GetModeChar() > 'z'))
651                 return false;
652
653         mh->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
654         pos = (mh->GetModeChar()-65) | mask;
655
656         if (modehandlers[pos] != mh)
657                 return false;
658
659         /* Note: We can't stack here, as we have modes potentially being removed across many different channels.
660          * To stack here we have to make the algorithm slower. Discuss.
661          */
662         switch (mh->GetModeType())
663         {
664                 case MODETYPE_USER:
665                         for (user_hash::iterator i = ServerInstance->Users->clientlist->begin(); i != ServerInstance->Users->clientlist->end(); i++)
666                         {
667                                 mh->RemoveMode(i->second);
668                         }
669                 break;
670                 case MODETYPE_CHANNEL:
671                         for (chan_hash::iterator i = ServerInstance->chanlist->begin(); i != ServerInstance->chanlist->end(); i++)
672                         {
673                                 mh->RemoveMode(i->second);
674                         }
675                 break;
676         }
677
678         modehandlers[pos] = NULL;
679
680         return true;
681 }
682
683 ModeHandler* ModeParser::FindMode(unsigned const char modeletter, ModeType mt)
684 {
685         unsigned char mask = 0;
686         unsigned char pos = 0;
687
688         if ((modeletter < 'A') || (modeletter > 'z'))
689                 return NULL;
690
691         mt == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
692         pos = (modeletter-65) | mask;
693
694         return modehandlers[pos];
695 }
696
697 std::string ModeParser::UserModeList()
698 {
699         char modestr[256];
700         int pointer = 0;
701
702         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
703         {
704                 unsigned char pos = (mode-65) | MASK_USER;
705
706                 if (modehandlers[pos])
707                         modestr[pointer++] = mode;
708         }
709         modestr[pointer++] = 0;
710         return modestr;
711 }
712
713 std::string ModeParser::ChannelModeList()
714 {
715         char modestr[256];
716         int pointer = 0;
717
718         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
719         {
720                 unsigned char pos = (mode-65) | MASK_CHANNEL;
721
722                 if (modehandlers[pos])
723                         modestr[pointer++] = mode;
724         }
725         modestr[pointer++] = 0;
726         return modestr;
727 }
728
729 std::string ModeParser::ParaModeList()
730 {
731         char modestr[256];
732         int pointer = 0;
733
734         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
735         {
736                 unsigned char pos = (mode-65) | MASK_CHANNEL;
737
738                 if ((modehandlers[pos]) && (modehandlers[pos]->GetNumParams(true)))
739                         modestr[pointer++] = mode;
740         }
741         modestr[pointer++] = 0;
742         return modestr;
743 }
744
745 ModeHandler* ModeParser::FindPrefix(unsigned const char pfxletter)
746 {
747         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
748         {
749                 unsigned char pos = (mode-65) | MASK_CHANNEL;
750
751                 if ((modehandlers[pos]) && (modehandlers[pos]->GetPrefix() == pfxletter))
752                 {
753                         return modehandlers[pos];
754                 }
755         }
756         return NULL;
757 }
758
759 std::string ModeParser::GiveModeList(ModeMasks m)
760 {
761         std::string type1;      /* Listmodes EXCEPT those with a prefix */
762         std::string type2;      /* Modes that take a param when adding or removing */
763         std::string type3;      /* Modes that only take a param when adding */
764         std::string type4;      /* Modes that dont take a param */
765
766         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
767         {
768                 unsigned char pos = (mode-65) | m;
769                  /* One parameter when adding */
770                 if (modehandlers[pos])
771                 {
772                         if (modehandlers[pos]->GetNumParams(true))
773                         {
774                                 if ((modehandlers[pos]->IsListMode()) && (!modehandlers[pos]->GetPrefix()))
775                                 {
776                                         type1 += modehandlers[pos]->GetModeChar();
777                                 }
778                                 else
779                                 {
780                                         /* ... and one parameter when removing */
781                                         if (modehandlers[pos]->GetNumParams(false))
782                                         {
783                                                 /* But not a list mode */
784                                                 if (!modehandlers[pos]->GetPrefix())
785                                                 {
786                                                         type2 += modehandlers[pos]->GetModeChar();
787                                                 }
788                                         }
789                                         else
790                                         {
791                                                 /* No parameters when removing */
792                                                 type3 += modehandlers[pos]->GetModeChar();
793                                         }
794                                 }
795                         }
796                         else
797                         {
798                                 type4 += modehandlers[pos]->GetModeChar();
799                         }
800                 }
801         }
802
803         return type1 + "," + type2 + "," + type3 + "," + type4;
804 }
805
806 std::string ModeParser::BuildPrefixes()
807 {
808         std::string mletters;
809         std::string mprefixes;
810         std::map<int,std::pair<char,char> > prefixes;
811
812         for (unsigned char mode = 'A'; mode <= 'z'; mode++)
813         {
814                 unsigned char pos = (mode-65) | MASK_CHANNEL;
815
816                 if ((modehandlers[pos]) && (modehandlers[pos]->GetPrefix()))
817                 {
818                         prefixes[modehandlers[pos]->GetPrefixRank()] = std::make_pair(
819                                 modehandlers[pos]->GetPrefix(), modehandlers[pos]->GetModeChar());
820                 }
821         }
822
823         for(std::map<int,std::pair<char,char> >::reverse_iterator n = prefixes.rbegin(); n != prefixes.rend(); n++)
824         {
825                 mletters = mletters + n->second.first;
826                 mprefixes = mprefixes + n->second.second;
827         }
828
829         return "(" + mprefixes + ")" + mletters;
830 }
831
832 bool ModeParser::AddModeWatcher(ModeWatcher* mw)
833 {
834         unsigned char mask = 0;
835         unsigned char pos = 0;
836
837         if (!mw)
838                 return false;
839
840         if ((mw->GetModeChar() < 'A') || (mw->GetModeChar() > 'z'))
841                 return false;
842
843         mw->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
844         pos = (mw->GetModeChar()-65) | mask;
845
846         modewatchers[pos].push_back(mw);
847
848         return true;
849 }
850
851 bool ModeParser::DelModeWatcher(ModeWatcher* mw)
852 {
853         unsigned char mask = 0;
854         unsigned char pos = 0;
855
856         if (!mw)
857                 return false;
858
859         if ((mw->GetModeChar() < 'A') || (mw->GetModeChar() > 'z'))
860                 return false;
861
862         mw->GetModeType() == MODETYPE_USER ? mask = MASK_USER : mask = MASK_CHANNEL;
863         pos = (mw->GetModeChar()-65) | mask;
864
865         ModeWatchIter a = find(modewatchers[pos].begin(),modewatchers[pos].end(),mw);
866
867         if (a == modewatchers[pos].end())
868         {
869                 return false;
870         }
871
872         modewatchers[pos].erase(a);
873
874         return true;
875 }
876
877 /** This default implementation can remove simple user modes
878  */
879 void ModeHandler::RemoveMode(User* user, irc::modestacker* stack)
880 {
881         char moderemove[MAXBUF];
882         std::vector<std::string> parameters;
883
884         if (user->IsModeSet(this->GetModeChar()))
885         {
886                 if (stack)
887                 {
888                         stack->Push(this->GetModeChar());
889                 }
890                 else
891                 {
892                         sprintf(moderemove,"-%c",this->GetModeChar());
893                         parameters.push_back(user->nick);
894                         parameters.push_back(moderemove);
895                         ServerInstance->Modes->Process(parameters, ServerInstance->FakeClient);
896                 }
897         }
898 }
899
900 /** This default implementation can remove simple channel modes
901  * (no parameters)
902  */
903 void ModeHandler::RemoveMode(Channel* channel, irc::modestacker* stack)
904 {
905         char moderemove[MAXBUF];
906         std::vector<std::string> parameters;
907
908         if (channel->IsModeSet(this->GetModeChar()))
909         {
910                 if (stack)
911                 {
912                         stack->Push(this->GetModeChar());
913                 }
914                 else
915                 {
916                         sprintf(moderemove,"-%c",this->GetModeChar());
917                         parameters.push_back(channel->name);
918                         parameters.push_back(moderemove);
919                         ServerInstance->SendMode(parameters, ServerInstance->FakeClient);
920                 }
921         }
922 }
923
924 struct builtin_modes
925 {
926         ModeChannelSecret s;
927         ModeChannelPrivate p;
928         ModeChannelModerated m;
929         ModeChannelTopicOps t;
930
931         ModeChannelNoExternal n;
932         ModeChannelInviteOnly i;
933         ModeChannelKey k;
934         ModeChannelLimit l;
935
936         ModeChannelBan b;
937         ModeChannelOp o;
938         // halfop is added by configreader
939         ModeChannelVoice v;
940
941         ModeUserWallops uw;
942         ModeUserInvisible ui;
943         ModeUserOperator uo;
944         ModeUserServerNoticeMask us;
945
946         void init(ModeParser* modes)
947         {
948                 modes->AddMode(&s);
949                 modes->AddMode(&p);
950                 modes->AddMode(&m);
951                 modes->AddMode(&t);
952                 modes->AddMode(&n);
953                 modes->AddMode(&i);
954                 modes->AddMode(&k);
955                 modes->AddMode(&l);
956                 modes->AddMode(&b);
957                 modes->AddMode(&o);
958                 modes->AddMode(&v);
959                 modes->AddMode(&uw);
960                 modes->AddMode(&ui);
961                 modes->AddMode(&uo);
962                 modes->AddMode(&us);
963         }
964 };
965
966 static builtin_modes static_modes;
967
968 ModeParser::ModeParser()
969 {
970         /* Clear mode handler list */
971         memset(modehandlers, 0, sizeof(modehandlers));
972
973         /* Last parse string */
974         LastParse.clear();
975
976         seq = 0;
977         memset(&sent, 0, sizeof(sent));
978
979         static_modes.init(this);
980 }
981
982 ModeParser::~ModeParser()
983 {
984         ModeHandler* mh = FindMode('h', MODETYPE_CHANNEL);
985         if (mh)
986         {
987                 mh->cull();
988                 delete mh;
989         }
990 }