src/modes/cmode_k.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
   6  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   7  *   Copyright (C) 2006 Craig Edwards <craigedwards@brainbox.cc>
   8  *
   9  * This file is part of InspIRCd.  InspIRCd is free software: you can
  10  * redistribute it and/or modify it under the terms of the GNU General Public
  11  * License as published by the Free Software Foundation, version 2.
  12  *
  13  * This program is distributed in the hope that it will be useful, but WITHOUT
  14  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  15  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  16  * details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20  */
  21
  22
  23 #include "inspircd.h"
  24 #include "mode.h"
  25 #include "channels.h"
  26 #include "users.h"
  27 #include "builtinmodes.h"
  28
  29 ModeChannelKey::ModeChannelKey() : ModeHandler(NULL, "key", 'k', PARAM_ALWAYS, MODETYPE_CHANNEL)
  30 {
  31 }
  32
  33 ModeAction ModeChannelKey::OnModeChange(User* source, User*, Channel* channel, std::string &parameter, bool adding)
  34 {
  35         bool exists = channel->IsModeSet('k');
  36         if (IS_LOCAL(source))
  37         {
  38                 if (exists == adding)
  39                         return MODEACTION_DENY;
  40                 if (exists && (parameter != channel->GetModeParameter('k')))
  41                 {
  42                         /* Key is currently set and the correct key wasnt given */
  43                         return MODEACTION_DENY;
  44                 }
  45         } else {
  46                 if (exists && adding && parameter == channel->GetModeParameter('k'))
  47                 {
  48                         /* no-op, don't show */
  49                         return MODEACTION_DENY;
  50                 }
  51         }
  52
  53         /* invalid keys */
  54         if (!parameter.length())
  55                 return MODEACTION_DENY;
  56
  57         if (parameter.rfind(' ') != std::string::npos)
  58                 return MODEACTION_DENY;
  59
  60         if (adding)
  61         {
  62                 std::string ckey;
  63                 ckey.assign(parameter, 0, 32);
  64                 parameter = ckey;
  65                 channel->SetModeParam('k', parameter);
  66         }
  67         else
  68         {
  69                 channel->SetModeParam('k', "");
  70         }
  71         return MODEACTION_ALLOW;
  72 }