[core] Handle spurious wake up in shared mutex locking functions

Haivision · Jul 30, 2024 · 58d575f · 58d575f
1 parent deeb9c9
commit 58d575f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/srtcore/sync.cpp b/srtcore/sync.cpp
@@ -386,12 +386,12 @@ srt::sync::SharedMutex::~SharedMutex()
 void srt::sync::SharedMutex::lock()
 {
     UniqueLock l1(m_Mutex);
-    if (m_bWriterLocked)
+    while (m_bWriterLocked)
         m_LockWriteCond.wait(l1);
 
     m_bWriterLocked = true;
 
-    if (m_iCountRead)
+    while (m_iCountRead)
         m_LockReadCond.wait(l1);
 }
 
@@ -418,7 +418,7 @@ void srt::sync::SharedMutex::unlock()
 void srt::sync::SharedMutex::lock_shared()
 {
     UniqueLock lk(m_Mutex);
-    if (m_bWriterLocked)
+    while (m_bWriterLocked)
         m_LockWriteCond.wait(lk);
 
     m_iCountRead++;