Core/Instances: Sprinkle some thread safety on InstanceLockMgr

1 files changed, 12 insertions, 1 deletions

diff --git a/src/server/game/Instances/InstanceLockMgr.cpp b/src/server/game/Instances/InstanceLockMgr.cpp
index b1c645f46e9..1e2eb22cbe6 100644
--- a/src/server/game/Instances/InstanceLockMgr.cpp
+++ b/src/server/game/Instances/InstanceLockMgr.cpp
@@ -215,6 +215,8 @@ InstanceLock* InstanceLockMgr::FindActiveInstanceLock(ObjectGuid const& playerGu
 
 InstanceLock* InstanceLockMgr::FindActiveInstanceLock(ObjectGuid const& playerGuid, MapDb2Entries const& entries, bool ignoreTemporary, bool ignoreExpired) const
 {
+    std::shared_lock<std::shared_mutex> guard(_locksMutex);
+
     InstanceLock* lock = FindInstanceLock(_instanceLocksByPlayer, playerGuid, entries);
 
     // Ignore expired and not extended locks
@@ -227,6 +229,7 @@ InstanceLock* InstanceLockMgr::FindActiveInstanceLock(ObjectGuid const& playerGu
     return FindInstanceLock(_temporaryInstanceLocksByPlayer, playerGuid, entries);
 }
 
+// used in world update thread (THREADUNSAFE packets) - no locking neccessary
 std::vector<InstanceLock const*> InstanceLockMgr::GetInstanceLocksForPlayer(ObjectGuid const& playerGuid) const
 {
     std::vector<InstanceLock const*> locks;
@@ -241,6 +244,7 @@ std::vector<InstanceLock const*> InstanceLockMgr::GetInstanceLocksForPlayer(Obje
     return locks;
 }
 
+// used in world update thread (cross map teleportation) - no locking neccessary
 InstanceLock* InstanceLockMgr::CreateInstanceLockForNewInstance(ObjectGuid const& playerGuid, MapDb2Entries const& entries, uint32 instanceId)
 {
     if (!entries.MapDifficulty->HasResetSchedule())
@@ -272,6 +276,8 @@ InstanceLock* InstanceLockMgr::UpdateInstanceLockForPlayer(CharacterDatabaseTran
     InstanceLock* instanceLock = FindActiveInstanceLock(playerGuid, entries, true, true);
     if (!instanceLock)
     {
+        std::unique_lock<std::shared_mutex> guard(_locksMutex);
+
         // Move lock from temporary storage if it exists there
         // This is to avoid destroying expired locks before any boss is killed in a fresh lock
         // player can still change his mind, exit instance and reactivate old lock
@@ -312,7 +318,12 @@ InstanceLock* InstanceLockMgr::UpdateInstanceLockForPlayer(CharacterDatabaseTran
             instanceLock = new InstanceLock(entries.MapDifficulty->MapID, Difficulty(entries.MapDifficulty->DifficultyID),
                 GetNextResetTime(entries), updateEvent.InstanceId);
 
-        _instanceLocksByPlayer[playerGuid][entries.GetKey()].reset(instanceLock);
+        {
+            std::unique_lock<std::shared_mutex> guard(_locksMutex);
+
+            _instanceLocksByPlayer[playerGuid][entries.GetKey()].reset(instanceLock);
+        }
+
         TC_LOG_DEBUG("instance.locks", "[%u-%s | %u-%s] Created new instance lock for %s in instance %u",
             entries.Map->ID, entries.Map->MapName[sWorld->GetDefaultDbcLocale()],
             uint32(entries.MapDifficulty->DifficultyID), sDifficultyStore.AssertEntry(entries.MapDifficulty->DifficultyID)->Name[sWorld->GetDefaultDbcLocale()],