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