Added missing mutexes for DefragmentPassEnd().

adam-sawicki-a · adam-sawicki-a · commit ea759b65d440 · 2022-03-04T10:34:26.000+01:00
Code by @medranSolus
diff --git a/include/vk_mem_alloc.h b/include/vk_mem_alloc.h
@@ -13252,6 +13252,8 @@ VkResult VmaDefragmentationContext_T::DefragmentPassEnd(VmaDefragmentationPassMo
                     if (state.operation != StateExtensive::Operation::Cleanup)
                     {
                         VmaBlockVector* vector = m_pBlockVectors[block.data];
+                        VmaMutexLockWrite lock(vector->GetMutex(), vector->GetAllocator()->m_UseMutex);
+
                         for (size_t i = 0, count = vector->GetBlockCount() - m_ImmovableBlockCount; i < count; ++i)
                         {
                             if (vector->GetBlock(i) == block.block)
@@ -13281,6 +13283,8 @@ VkResult VmaDefragmentationContext_T::DefragmentPassEnd(VmaDefragmentationPassMo
             for (const FragmentedBlock& block : immovableBlocks)
             {
                 VmaBlockVector* vector = m_pBlockVectors[block.data];
+                VmaMutexLockWrite lock(vector->GetMutex(), vector->GetAllocator()->m_UseMutex);
+
                 for (size_t i = m_ImmovableBlockCount; i < vector->GetBlockCount(); ++i)
                 {
                     if (vector->GetBlock(i) == block.block)

Original file line number	Diff line number	Diff line change
`@@ -13252,6 +13252,8 @@ VkResult VmaDefragmentationContext_T::DefragmentPassEnd(VmaDefragmentationPassMo`
`13252`	`13252`	`if (state.operation != StateExtensive::Operation::Cleanup)`
`13253`	`13253`	`{`
`13254`	`13254`	`VmaBlockVector* vector = m_pBlockVectors[block.data];`
	`13255`	`+ VmaMutexLockWrite lock(vector->GetMutex(), vector->GetAllocator()->m_UseMutex);`
	`13256`	`+`
`13255`	`13257`	`for (size_t i = 0, count = vector->GetBlockCount() - m_ImmovableBlockCount; i < count; ++i)`
`13256`	`13258`	`{`
`13257`	`13259`	`if (vector->GetBlock(i) == block.block)`
`@@ -13281,6 +13283,8 @@ VkResult VmaDefragmentationContext_T::DefragmentPassEnd(VmaDefragmentationPassMo`
`13281`	`13283`	`for (const FragmentedBlock& block : immovableBlocks)`
`13282`	`13284`	`{`
`13283`	`13285`	`VmaBlockVector* vector = m_pBlockVectors[block.data];`
	`13286`	`+ VmaMutexLockWrite lock(vector->GetMutex(), vector->GetAllocator()->m_UseMutex);`
	`13287`	`+`
`13284`	`13288`	`for (size_t i = m_ImmovableBlockCount; i < vector->GetBlockCount(); ++i)`
`13285`	`13289`	`{`
`13286`	`13290`	`if (vector->GetBlock(i) == block.block)`