-/* +------------------------------------+
- * | Inspire Internet Relay Chat Daemon |
- * +------------------------------------+
- *
- * InspIRCd: (C) 2002-2007 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
- *
- * This program is free but copyrighted software; see
- * the file COPYING for details.
- *
- * ---------------------------------------------------
- */
-
-#include "inspircd.h"
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
-#include "wildcard.h"
-
-/** Holds a users m_safelist state
- */
-class ListData : public classbase
-{
- public:
- long list_start;
- long list_position;
- bool list_ended;
- const std::string glob;
- int minusers;
- int maxusers;
-
- ListData() : list_start(0), list_position(0), list_ended(false) {};
- ListData(long pos, time_t t, const std::string &pattern, int mi, int ma) : list_start(t), list_position(pos), list_ended(false), glob(pattern), minusers(mi), maxusers(ma) {};
-};
-
-/* $ModDesc: A module overriding /list, and making it safe - stop those sendq problems. */
-
-class ModuleSafeList : public Module
-{
- time_t ThrottleSecs;
- size_t ServerNameSize;
- int global_listing;
- int LimitList;
- public:
- ModuleSafeList(InspIRCd* Me) : Module(Me)
- {
- OnRehash(NULL, "");
- }
-
- virtual ~ModuleSafeList()
- {
- }
-
- virtual void OnRehash(userrec* user, const std::string ¶meter)
- {
- ConfigReader MyConf(ServerInstance);
- ThrottleSecs = MyConf.ReadInteger("safelist", "throttle", "60", 0, true);
- LimitList = MyConf.ReadInteger("safelist", "maxlisters", "50", 0, true);
- ServerNameSize = strlen(ServerInstance->Config->ServerName) + 4;
- global_listing = 0;
- }
-
- virtual Version GetVersion()
- {
- return Version(1,1,0,0,VF_VENDOR,API_VERSION);
- }
-
- void Implements(char* List)
- {
- List[I_OnBufferFlushed] = List[I_OnPreCommand] = List[I_OnCleanup] = List[I_OnUserQuit] = List[I_On005Numeric] = List[I_OnRehash] = 1;
- }
-
- /*
- * OnPreCommand()
- * Intercept the LIST command.
- */
- virtual int OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated, const std::string &original_line)
- {
- /* If the command doesnt appear to be valid, we dont want to mess with it. */
- if (!validated)
- return 0;
-
- if (command == "LIST")
- {
- return this->HandleList(parameters, pcnt, user);
- }
- return 0;
- }
-
- /*
- * HandleList()
- * Handle (override) the LIST command.
- */
- int HandleList(const char** parameters, int pcnt, userrec* user)
- {
- int minusers = 0, maxusers = 0;
-
- if (global_listing >= LimitList)
- {
- user->WriteServ("NOTICE %s :*** Server load is currently too heavy. Please try again later.", user->nick);
- user->WriteServ("321 %s Channel :Users Name",user->nick);
- user->WriteServ("323 %s :End of channel list.",user->nick);
- return 1;
- }
-
- /* First, let's check if the user is currently /list'ing */
- ListData *ld;
- user->GetExt("safelist_cache", ld);
-
- if (ld)
- {
- /* user is already /list'ing, we don't want to do shit. */
- return 1;
- }
-
- /* Work around mIRC suckyness. YOU SUCK, KHALED! */
- if (pcnt == 1)
- {
- if (*parameters[0] == '<')
- {
- maxusers = atoi(parameters[0]+1);
- ServerInstance->Log(DEBUG,"Max users: %d", maxusers);
- pcnt = 0;
- }
- else if (*parameters[0] == '>')
- {
- minusers = atoi(parameters[0]+1);
- ServerInstance->Log(DEBUG,"Min users: %d", minusers);
- pcnt = 0;
- }
- }
-
- time_t* last_list_time;
- user->GetExt("safelist_last", last_list_time);
- if (last_list_time)
- {
- if (ServerInstance->Time() < (*last_list_time)+ThrottleSecs)
- {
- user->WriteServ("NOTICE %s :*** Woah there, slow down a little, you can't /LIST so often!",user->nick);
- user->WriteServ("321 %s Channel :Users Name",user->nick);
- user->WriteServ("323 %s :End of channel list.",user->nick);
- return 1;
- }
-
- DELETE(last_list_time);
- user->Shrink("safelist_last");
- }
-
-
- /*
- * start at channel 0! ;)
- */
- ld = new ListData(0,ServerInstance->Time(), pcnt ? parameters[0] : "*", minusers, maxusers);
- user->Extend("safelist_cache", ld);
-
- time_t* llt = new time_t;
- *llt = ServerInstance->Time();
- user->Extend("safelist_last", llt);
-
- user->WriteServ("321 %s Channel :Users Name",user->nick);
-
- global_listing++;
-
- return 1;
- }
-
- virtual void OnBufferFlushed(userrec* user)
- {
- char buffer[MAXBUF];
- ListData* ld;
- if (user->GetExt("safelist_cache", ld))
- {
- chanrec* chan = NULL;
- long amount_sent = 0;
- do
- {
- chan = ServerInstance->GetChannelIndex(ld->list_position);
- bool has_user = (chan && chan->HasUser(user));
- long users = chan ? chan->GetUserCounter() : 0;
-
- bool too_few = (ld->minusers && (users <= ld->minusers));
- bool too_many = (ld->maxusers && (users >= ld->maxusers));
-
- if (chan && (too_many || too_few))
- {
- ld->list_position++;
- continue;
- }
-
- if ((chan) && (chan->modes[CM_PRIVATE]))
- {
- bool display = (match(chan->name, ld->glob.c_str()) || (*chan->topic && match(chan->topic, ld->glob.c_str())));
- if ((users) && (display))
- {
- int counter = snprintf(buffer, MAXBUF, "322 %s *", user->nick);
- amount_sent += counter + ServerNameSize;
- user->WriteServ(std::string(buffer));
- }
- }
- else if ((chan) && (((!(chan->modes[CM_PRIVATE])) && (!(chan->modes[CM_SECRET]))) || (has_user)))
- {
- bool display = (match(chan->name, ld->glob.c_str()) || (*chan->topic && match(chan->topic, ld->glob.c_str())));
- if ((users) && (display))
- {
- int counter = snprintf(buffer, MAXBUF, "322 %s %s %ld :[+%s] %s",user->nick, chan->name, users, chan->ChanModes(has_user), chan->topic);
- amount_sent += counter + ServerNameSize;
- user->WriteServ(std::string(buffer));
- }
- }
- else
- {
- if (!chan)
- {
- if (!ld->list_ended)
- {
- ld->list_ended = true;
- user->WriteServ("323 %s :End of channel list.",user->nick);
- }
- }
- }
- ld->list_position++;
- }
- while ((chan != NULL) && (amount_sent < (user->sendqmax / 4)));
- if (ld->list_ended)
- {
- user->Shrink("safelist_cache");
- DELETE(ld);
- global_listing--;
- }
- }
- }
-
- virtual void OnCleanup(int target_type, void* item)
- {
- if(target_type == TYPE_USER)
- {
- userrec* u = (userrec*)item;
- ListData* ld;
- u->GetExt("safelist_cache", ld);
- if (ld)
- {
- u->Shrink("safelist_cache");
- DELETE(ld);
- global_listing--;
- }
- time_t* last_list_time;
- u->GetExt("safelist_last", last_list_time);
- if (last_list_time)
- {
- DELETE(last_list_time);
- u->Shrink("safelist_last");
- }
- }
- }
-
- virtual void On005Numeric(std::string &output)
- {
- output.append(" SAFELIST");
- }
-
- virtual void OnUserQuit(userrec* user, const std::string &message, const std::string &oper_message)
- {
- this->OnCleanup(TYPE_USER,user);
- }
-
-};
-
-MODULE_INIT(ModuleSafeList)
+/* +------------------------------------+\r * | Inspire Internet Relay Chat Daemon |\r * +------------------------------------+\r *\r * InspIRCd: (C) 2002-2007 InspIRCd Development Team\r * See: http://www.inspircd.org/wiki/index.php/Credits\r *\r * This program is free but copyrighted software; see\r * the file COPYING for details.\r *\r * ---------------------------------------------------\r */\r\r#include "inspircd.h" \r#include "users.h"\r#include "channels.h"\r#include "modules.h"\r#include "wildcard.h"\r\r/** Holds a users m_safelist state\r */\rclass ListData : public classbase\r{\r public:\r long list_start;\r long list_position;\r bool list_ended;\r const std::string glob;\r int minusers;\r int maxusers;\r\r ListData() : list_start(0), list_position(0), list_ended(false) {};\r ListData(long pos, time_t t, const std::string &pattern, int mi, int ma) : list_start(t), list_position(pos), list_ended(false), glob(pattern), minusers(mi), maxusers(ma) {};\r};\r\r/* $ModDesc: A module overriding /list, and making it safe - stop those sendq problems. */\r\rclass ModuleSafeList : public Module\r{\r time_t ThrottleSecs;\r size_t ServerNameSize;\r int global_listing;\r int LimitList;\r public:\r ModuleSafeList(InspIRCd* Me) : Module(Me)\r {\r OnRehash(NULL, "");\r }\r \r virtual ~ModuleSafeList()\r {\r }\r\r virtual void OnRehash(userrec* user, const std::string ¶meter)\r {\r ConfigReader MyConf(ServerInstance);\r ThrottleSecs = MyConf.ReadInteger("safelist", "throttle", "60", 0, true);\r LimitList = MyConf.ReadInteger("safelist", "maxlisters", "50", 0, true);\r ServerNameSize = strlen(ServerInstance->Config->ServerName) + 4;\r global_listing = 0;\r }\r \r virtual Version GetVersion()\r {\r return Version(1,1,0,0,VF_VENDOR,API_VERSION);\r }\r \r void Implements(char* List)\r {\r List[I_OnBufferFlushed] = List[I_OnPreCommand] = List[I_OnCleanup] = List[I_OnUserQuit] = List[I_On005Numeric] = List[I_OnRehash] = 1;\r }\r\r /*\r * OnPreCommand()\r * Intercept the LIST command.\r */ \r virtual int OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated, const std::string &original_line)\r {\r /* If the command doesnt appear to be valid, we dont want to mess with it. */\r if (!validated)\r return 0;\r \r if (command == "LIST")\r {\r return this->HandleList(parameters, pcnt, user);\r }\r return 0;\r }\r \r /*\r * HandleList()\r * Handle (override) the LIST command.\r */\r int HandleList(const char** parameters, int pcnt, userrec* user)\r {\r int minusers = 0, maxusers = 0;\r\r if (global_listing >= LimitList)\r {\r user->WriteServ("NOTICE %s :*** Server load is currently too heavy. Please try again later.", user->nick);\r user->WriteServ("321 %s Channel :Users Name",user->nick);\r user->WriteServ("323 %s :End of channel list.",user->nick);\r return 1;\r }\r\r /* First, let's check if the user is currently /list'ing */\r ListData *ld;\r user->GetExt("safelist_cache", ld);\r \r if (ld)\r {\r /* user is already /list'ing, we don't want to do shit. */\r return 1;\r }\r\r /* Work around mIRC suckyness. YOU SUCK, KHALED! */\r if (pcnt == 1)\r {\r if (*parameters[0] == '<')\r {\r maxusers = atoi(parameters[0]+1);\r ServerInstance->Log(DEBUG,"Max users: %d", maxusers);\r pcnt = 0;\r }\r else if (*parameters[0] == '>')\r {\r minusers = atoi(parameters[0]+1);\r ServerInstance->Log(DEBUG,"Min users: %d", minusers);\r pcnt = 0;\r }\r }\r\r time_t* last_list_time;\r user->GetExt("safelist_last", last_list_time);\r if (last_list_time)\r {\r if (ServerInstance->Time() < (*last_list_time)+ThrottleSecs)\r {\r user->WriteServ("NOTICE %s :*** Woah there, slow down a little, you can't /LIST so often!",user->nick);\r user->WriteServ("321 %s Channel :Users Name",user->nick);\r user->WriteServ("323 %s :End of channel list.",user->nick);\r return 1;\r }\r\r DELETE(last_list_time);\r user->Shrink("safelist_last");\r }\r\r \r /*\r * start at channel 0! ;)\r */\r ld = new ListData(0,ServerInstance->Time(), pcnt ? parameters[0] : "*", minusers, maxusers);\r user->Extend("safelist_cache", ld);\r\r time_t* llt = new time_t;\r *llt = ServerInstance->Time();\r user->Extend("safelist_last", llt);\r\r user->WriteServ("321 %s Channel :Users Name",user->nick);\r\r global_listing++;\r\r return 1;\r }\r\r virtual void OnBufferFlushed(userrec* user)\r {\r char buffer[MAXBUF];\r ListData* ld;\r if (user->GetExt("safelist_cache", ld))\r {\r chanrec* chan = NULL;\r long amount_sent = 0;\r do\r {\r chan = ServerInstance->GetChannelIndex(ld->list_position);\r bool has_user = (chan && chan->HasUser(user));\r long users = chan ? chan->GetUserCounter() : 0;\r\r bool too_few = (ld->minusers && (users <= ld->minusers));\r bool too_many = (ld->maxusers && (users >= ld->maxusers));\r\r if (chan && (too_many || too_few))\r {\r ld->list_position++;\r continue;\r }\r\r if ((chan) && (chan->modes[CM_PRIVATE]))\r {\r bool display = (match(chan->name, ld->glob.c_str()) || (*chan->topic && match(chan->topic, ld->glob.c_str())));\r if ((users) && (display))\r {\r int counter = snprintf(buffer, MAXBUF, "322 %s *", user->nick);\r amount_sent += counter + ServerNameSize;\r user->WriteServ(std::string(buffer));\r }\r }\r else if ((chan) && (((!(chan->modes[CM_PRIVATE])) && (!(chan->modes[CM_SECRET]))) || (has_user)))\r {\r bool display = (match(chan->name, ld->glob.c_str()) || (*chan->topic && match(chan->topic, ld->glob.c_str())));\r if ((users) && (display))\r {\r int counter = snprintf(buffer, MAXBUF, "322 %s %s %ld :[+%s] %s",user->nick, chan->name, users, chan->ChanModes(has_user), chan->topic);\r amount_sent += counter + ServerNameSize;\r user->WriteServ(std::string(buffer));\r }\r }\r else\r {\r if (!chan)\r {\r if (!ld->list_ended)\r {\r ld->list_ended = true;\r user->WriteServ("323 %s :End of channel list.",user->nick);\r }\r }\r }\r ld->list_position++;\r }\r while ((chan != NULL) && (amount_sent < (user->sendqmax / 4)));\r if (ld->list_ended)\r {\r user->Shrink("safelist_cache");\r DELETE(ld);\r global_listing--;\r }\r }\r }\r\r virtual void OnCleanup(int target_type, void* item)\r {\r if(target_type == TYPE_USER)\r {\r userrec* u = (userrec*)item;\r ListData* ld;\r u->GetExt("safelist_cache", ld);\r if (ld)\r {\r u->Shrink("safelist_cache");\r DELETE(ld);\r global_listing--;\r }\r time_t* last_list_time;\r u->GetExt("safelist_last", last_list_time);\r if (last_list_time)\r {\r DELETE(last_list_time);\r u->Shrink("safelist_last");\r }\r }\r }\r\r virtual void On005Numeric(std::string &output)\r {\r output.append(" SAFELIST");\r }\r\r virtual void OnUserQuit(userrec* user, const std::string &message, const std::string &oper_message)\r {\r this->OnCleanup(TYPE_USER,user);\r }\r\r};\r\rMODULE_INIT(ModuleSafeList)\r
\ No newline at end of file