Merge GH #716 Add Barrier type, correct Threaded_Fork

1 files changed, 39 insertions, 0 deletions

diff --git a/src/lib/utils/barrier.cpp b/src/lib/utils/barrier.cpp
new file mode 100644
index 000000000..81c578b72
--- /dev/null
+++ b/src/lib/utils/barrier.cpp
@@ -0,0 +1,39 @@
+/*
+* Barrier
+* (C) 2016 Joel Low
+*
+* Botan is released under the Simplified BSD License (see license.txt)
+*/
+
+#include <botan/internal/barrier.h>
+
+#if defined(BOTAN_TARGET_OS_HAS_THREADS)
+
+namespace Botan {
+
+void Barrier::wait(unsigned delta)
+    {
+    lock_guard_type<mutex_type> lock(m_mutex);
+    m_value += delta;
+    }
+
+void Barrier::sync()
+    {
+    std::unique_lock<mutex_type> lock(m_mutex);
+    --m_value;
+    if(m_value > 0)
+        {
+        unsigned current_syncs = m_syncs;
+        m_cond.wait(lock, [this, &current_syncs] { return m_syncs != current_syncs; });
+        }
+    else
+        {
+        m_value = 0;
+        ++m_syncs;
+        m_cond.notify_all();
+        }
+    }
+
+}
+
+#endif