From b867007d201b1a3b130186c54e41481c0374a7f6 Mon Sep 17 00:00:00 2001
From: Sadie Powell
Date: Sun, 16 Jan 2022 01:15:50 +0000
Subject: Allow setting extension data on connect classes.

---
 src/modules.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/modules.cpp')

diff --git a/src/modules.cpp b/src/modules.cpp
index 28a231268..b6738524b 100644
--- a/src/modules.cpp
+++ b/src/modules.cpp
@@ -429,6 +429,12 @@ void ModuleManager::DoSafeUnload(Module* mod)
 		user->UnhookExtensions(items);
 	}
 
+	for (const auto& klass : ServerInstance->Config->Classes)
+	{
+		mod->OnCleanup(ExtensionType::CONNECT_CLASS, klass.get());
+		klass->UnhookExtensions(items);
+	}
+
 	for (DataProviderMap::iterator i = DataProviders.begin(); i != DataProviders.end(); )
 	{
 		DataProviderMap::iterator curr = i++;
-- 
cgit v1.3.1-10-gc9f91