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