src/modules/m_knock.cpp

   1 #include <stdio.h>
   2 #include <string>
   3 #include "users.h"
   4 #include "channels.h"
   5 #include "modules.h"
   6
   7 /* $ModDesc: Provides support for /KNOCK and mode +K */
   8
   9 Server *Srv;
  10
  11 void handle_knock(char **parameters, int pcnt, userrec *user)
  12 {
  13         chanrec* c = Srv->FindChannel(parameters[0]);
  14         std::string line = "";
  15
  16         for (int i = 1; i < pcnt - 1; i++)
  17         {
  18                 line = line + std::string(parameters[i]) + " ";
  19         }
  20         line = line + std::string(parameters[pcnt-1]);
  21
  22         if (c->IsCustomModeSet('K'))
  23         {
  24                 WriteServ(user->fd,"480 %s :Can't KNOCK on %s, +K is set.",user->nick, c->name);
  25                 return;
  26         }
  27         if (c->inviteonly)
  28         {
  29                 WriteChannelWithServ((char*)Srv->GetServerName().c_str(),c,user,"NOTICE %s :User %s is KNOCKing on %s (%s)",c->name,user->nick,c->name,line.c_str());
  30                 WriteServ(user->fd,"NOTICE %s :KNOCKing on %s",user->nick,c->name);
  31                 return;
  32         }
  33         else
  34         {
  35                 WriteServ(user->fd,"480 %s :Can't KNOCK on %s, channel is not invite only so knocking is pointless!",user->nick, c->name);
  36                 return;
  37         }
  38 }
  39
  40
  41 class ModuleKnock : public Module
  42 {
  43  public:
  44         ModuleKnock()
  45         {
  46                 Srv = new Server;
  47
  48                 Srv->AddExtendedMode('K',MT_CHANNEL,false,0,0);
  49                 Srv->AddCommand("KNOCK",handle_knock,0,2);
  50         }
  51
  52         virtual ~ModuleKnock()
  53         {
  54                 delete Srv;
  55         }
  56
  57         virtual Version GetVersion()
  58         {
  59                 return Version(1,0,0,1);
  60         }
  61
  62         virtual int OnExtendedMode(userrec* user, void* target, char modechar, int type, bool mode_on, string_list &params)
  63         {
  64                 // check if this is our mode character...
  65                 if ((modechar == 'K') && (type == MT_CHANNEL))
  66                 {
  67                         return 1;
  68                 }
  69                 else
  70                 {
  71                         return 0;
  72                 }
  73         }
  74
  75 };
  76
  77 // stuff down here is the module-factory stuff. For basic modules you can ignore this.
  78
  79 class ModuleKnockFactory : public ModuleFactory
  80 {
  81  public:
  82         ModuleKnockFactory()
  83         {
  84         }
  85
  86         ~ModuleKnockFactory()
  87         {
  88         }
  89
  90         virtual Module * CreateModule()
  91         {
  92                 return new ModuleKnock;
  93         }
  94
  95 };
  96
  97
  98 extern "C" void * init_module( void )
  99 {
 100         return new ModuleKnockFactory;
 101 }
 102