X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_nicklock.cpp;h=df8a551dd2bd3b1ca2d29204f63d5b20f8204ba3;hb=3a7023f2c595d14778b3f1f7e53d3914698dd500;hp=731704d6850c9ddeb25e46c83739dee503698a65;hpb=73e65d1d242c161e2aaa13d19bee28be2c04cb64;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/src/modules/m_nicklock.cpp b/src/modules/m_nicklock.cpp
index 731704d68..df8a551dd 100644
--- a/src/modules/m_nicklock.cpp
+++ b/src/modules/m_nicklock.cpp
@@ -26,17 +26,18 @@ using namespace std;
 
 /* $ModDesc: Provides the NICKLOCK command, allows an oper to chage a users nick and lock them to it until they quit */
 
-Server *Srv;
+static Server *Srv;
 
 class cmd_nicklock : public command_t
 {
+	char* dummy;
  public:
-	 cmd_nicklock () : command_t("NICKLOCK", 'o', 2)
-	 {
+	cmd_nicklock () : command_t("NICKLOCK", 'o', 2)
+	{
 		this->source = "m_nicklock.so";
-	 }
+	}
 
-	void Handle(char **parameters, int pcnt, userrec *user)
+	void Handle(const char** parameters, int pcnt, userrec *user)
 	{
 		userrec* source = Srv->FindNick(std::string(parameters[0]));
 		irc::string server;
@@ -44,7 +45,7 @@ class cmd_nicklock : public command_t
 
 		if (source)
 		{
-			if (source->GetExt("nick_locked"))
+			if (source->GetExt("nick_locked", dummy))
 			{
 				WriteServ(user->fd,"946 %s %s :This user's nickname is already locked.",user->nick,source->nick);
 				return;
@@ -71,7 +72,7 @@ class cmd_nickunlock : public command_t
 		this->source = "m_nickunlock.so";
 	}
 
-	void Handle (char **parameters, int pcnt, userrec *user)
+	void Handle (const char** parameters, int pcnt, userrec *user)
 	{
 		userrec* source = Srv->FindNick(std::string(parameters[0]));
 		if (source)
@@ -88,6 +89,7 @@ class ModuleNickLock : public Module
 {
 	cmd_nicklock*	cmd1;
 	cmd_nickunlock*	cmd2;
+	char* n;
  public:
 	ModuleNickLock(Server* Me)
 		: Module::Module(Me)
@@ -110,12 +112,12 @@ class ModuleNickLock : public Module
 
 	void Implements(char* List)
 	{
-		List[I_OnUserPreNick] = List[I_OnUserQuit] = 1;
+		List[I_OnUserPreNick] = List[I_OnUserQuit] = List[I_OnCleanup] = 1;
 	}
 
-	virtual int OnUserPreNick(userrec* user, std::string newnick)
+	virtual int OnUserPreNick(userrec* user, const std::string &newnick)
 	{
-		if (user->GetExt("nick_locked"))
+		if (user->GetExt("nick_locked", n))
 		{
 			WriteServ(user->fd,"447 %s :You cannot change your nickname (your nick is locked)",user->nick);
 			return 1;
@@ -123,11 +125,19 @@ class ModuleNickLock : public Module
 		return 0;
 	}
 
-        virtual void OnUserQuit(userrec* user, std::string reason)
-        {
-                user->Shrink("nick_locked");
-        }
+	virtual void OnUserQuit(userrec* user, const std::string &reason)
+	{
+		user->Shrink("nick_locked");
+	}
 
+	virtual void OnCleanup(int target_type, void* item)
+	{
+		if(target_type == TYPE_USER)
+		{
+			userrec* user = (userrec*)item;
+			user->Shrink("nick_locked");
+		}
+	}
 };
 
 // stuff down here is the module-factory stuff. For basic modules you can ignore this.