Socket engine tweaks to fix a glitch, and improvements to new m_ident

[user/henk/code/inspircd.git] / src / modules / m_restrictbanned.cpp

diff --git a/src/modules/m_restrictbanned.cpp b/src/modules/m_restrictbanned.cpp
index 9315385a1aeb416bf0d2ad63a0ec5967b54404ee..a5a5ab1aee4cab2408d027f0713102d06c3de225 100644 (file)
--- a/src/modules/m_restrictbanned.cpp
+++ b/src/modules/m_restrictbanned.cpp
@@ -12,9 +12,6 @@
  */
 
 #include "inspircd.h"
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
 
 /* $ModDesc: Restricts banned users in a channel. May not speak, etc. */
 
@@ -40,7 +37,7 @@ class ModuleRestrictBanned : public Module
                List[I_OnLocalTopicChange] = List[I_OnUserPreNick] = List[I_OnUserPreNotice] = List[I_OnUserPreMessage] = 1;
        }
 
-       int CheckRestricted(userrec *user, chanrec *channel, const std::string &action)
+       int CheckRestricted(User *user, Channel *channel, const std::string &action)
        {
                /* aren't local? we don't care. */
                if (!IS_LOCAL(user))
@@ -56,12 +53,16 @@ class ModuleRestrictBanned : public Module
                return 0;
        }
 
-       virtual int OnUserPreNick(userrec *user, const std::string &newnick)
+       virtual int OnUserPreNick(User *user, const std::string &newnick)
        {
                /* if they aren't local, we don't care */
                if (!IS_LOCAL(user))
                        return 0;
 
+               /* Allow changes to UID */
+               if (isdigit(newnick[0]))
+                       return 0;
+
                /* bit of a special case. */
                for (UCListIter i = user->chans.begin(); i != user->chans.end(); i++)
                {
@@ -72,21 +73,21 @@ class ModuleRestrictBanned : public Module
                return 0;
        }
 
-       virtual int OnLocalTopicChange(userrec *user, chanrec *channel, const std::string &topic)
+       virtual int OnLocalTopicChange(User *user, Channel *channel, const std::string &topic)
        {
                return CheckRestricted(user, channel, "change the topic");
        }
        
-       virtual int OnUserPreMessage(userrec* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
+       virtual int OnUserPreMessage(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
        {
                return OnUserPreNotice(user,dest,target_type,text,status,exempt_list);
        }
 
-       virtual int OnUserPreNotice(userrec* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
+       virtual int OnUserPreNotice(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
        {
                if (target_type == TYPE_CHANNEL)
                {
-                       chanrec *channel = (chanrec *)dest;
+                       Channel *channel = (Channel *)dest;
 
                        return CheckRestricted(user, channel, "message the channel");
                }