Allow m_sqlauth to set connect class

1 files changed, 10 insertions, 1 deletions

diff --git a/src/users.cpp b/src/users.cpp
index b6247ca98..f40286bc4 100644
--- a/src/users.cpp
+++ b/src/users.cpp
@@ -801,7 +801,16 @@ void LocalUser::FullConnect()
 	 * may put the user into a totally seperate class with different restrictions! so we *must* check again.
 	 * Don't remove this! -- w00t
 	 */
-	SetClass();
+	if (ServerInstance->ForcedClass.get(this))
+	{
+		std::string* cls = ServerInstance->ForcedClass.get(this);
+		SetClass(*cls);
+		ServerInstance->ForcedClass.unset(this);
+	}
+	else
+	{
+		SetClass();
+	}
 	CheckClass();
 	CheckLines();