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