From 259cf9f027adb41925e075799c7e46ccbcaec881 Mon Sep 17 00:00:00 2001
From: madmaxoft <github@xoft.cz>
Date: Wed, 14 Aug 2013 22:27:49 +0200
Subject: cEvent can now wait for the event with a specified timeout.

---
 source/OSSupport/Event.cpp | 104 +++++++++++++++++++++++++++++++++------------
 source/OSSupport/Event.h   |  10 +++++
 2 files changed, 88 insertions(+), 26 deletions(-)

(limited to 'source/OSSupport')

diff --git a/source/OSSupport/Event.cpp b/source/OSSupport/Event.cpp
index 13b5c1d3f..f3c9053b1 100644
--- a/source/OSSupport/Event.cpp
+++ b/source/OSSupport/Event.cpp
@@ -15,7 +15,7 @@
 cEvent::cEvent(void)
 {
 #ifdef _WIN32
-	m_Event = CreateEvent( 0, FALSE, FALSE, 0 );
+	m_Event = CreateEvent(NULL, FALSE, FALSE, NULL);
 	if (m_Event == NULL)
 	{
 		LOGERROR("cEvent: cannot create event, GLE = %d. Aborting server.", GetLastError());
@@ -78,19 +78,19 @@ cEvent::~cEvent()
 
 void cEvent::Wait(void)
 {
-#ifdef _WIN32
-	DWORD res = WaitForSingleObject(m_Event, INFINITE);
-	if (res != WAIT_OBJECT_0)
-	{
-		LOGWARN("cEvent: waiting for the event failed: %d, GLE = %d. Continuing, but server may be unstable.", res, GetLastError());
-	}
-#else
-	int res = sem_wait(m_Event);
-	if (res != 0 )
-	{
-		LOGWARN("cEvent: waiting for the event failed: %i, errno = %i. Continuing, but server may be unstable.", res, errno);
-	}
-#endif
+	#ifdef _WIN32
+		DWORD res = WaitForSingleObject(m_Event, INFINITE);
+		if (res != WAIT_OBJECT_0)
+		{
+			LOGWARN("cEvent: waiting for the event failed: %d, GLE = %d. Continuing, but server may be unstable.", res, GetLastError());
+		}
+	#else
+		int res = sem_wait(m_Event);
+		if (res != 0 )
+		{
+			LOGWARN("cEvent: waiting for the event failed: %i, errno = %i. Continuing, but server may be unstable.", res, errno);
+		}
+	#endif
 }
 
 
@@ -99,18 +99,70 @@ void cEvent::Wait(void)
 
 void cEvent::Set(void)
 {
-#ifdef _WIN32
-	if (!SetEvent(m_Event))
-	{
-		LOGWARN("cEvent: Could not set cEvent: GLE = %d", GetLastError());
-	}
-#else
-	int res = sem_post(m_Event);
-	if (res != 0)
-	{
-		LOGWARN("cEvent: Could not set cEvent: %i, errno = %d", res, errno);
-	}
-#endif
+	#ifdef _WIN32
+		if (!SetEvent(m_Event))
+		{
+			LOGWARN("cEvent: Could not set cEvent: GLE = %d", GetLastError());
+		}
+	#else
+		int res = sem_post(m_Event);
+		if (res != 0)
+		{
+			LOGWARN("cEvent: Could not set cEvent: %i, errno = %d", res, errno);
+		}
+	#endif
+}
+
+
+
+
+
+cEvent::eWaitResult cEvent::Wait(int a_TimeoutMilliSec)
+{
+	#ifdef _WIN32
+		DWORD res = WaitForSingleObject(m_Event, (DWORD)a_TimeoutMilliSec);
+		switch (res)
+		{
+			case WAIT_OBJECT_0:
+			{
+				// The semaphore was signalled
+				return wrSignalled;
+			}
+			case WAIT_TIMEOUT:
+			{
+				// The timeout was hit
+				return wrTimeout;
+			}
+			default:
+			{
+				LOGWARNING("cEvent: timed-waiting for the event failed: %d, GLE = %d. Continuing, but server may be unstable.", res, GetLastError());
+				return wrError;
+			}
+		}
+	#else
+		timespec timeout;
+		timeout.tv_sec = a_TimeoutMilliSec / 1000;
+		timeout.tv_nsec = (a_TimeoutMilliSec % 1000) * 1000000;
+		int res = sem_timedwait(m_Event, &timeout);
+		switch (res)
+		{
+			case 0:
+			{
+				// The semaphore was signalled
+				return wrSignalled;
+			}
+			case ETIMEDOUT:
+			{
+				// The timeout was hit
+				return wrTimeout;
+			}
+			default:
+			{
+				LOGWARNING("cEvent: timed-waiting for the event failed: %i, errno = %i. Continuing, but server may be unstable.", res, errno);
+				return wrError;
+			}
+		}
+	#endif
 }
 
 
diff --git a/source/OSSupport/Event.h b/source/OSSupport/Event.h
index 71f418c0c..803d73b7e 100644
--- a/source/OSSupport/Event.h
+++ b/source/OSSupport/Event.h
@@ -19,12 +19,22 @@
 class cEvent
 {
 public:
+	enum eWaitResult
+	{
+		wrSignalled,
+		wrTimeout,
+		wrError,
+	} ;
+	
 	cEvent(void);
 	~cEvent();
 
 	void Wait(void);
 	void Set (void);
 	
+	/// Waits for the semaphore with a timeout
+	eWaitResult Wait(int a_TimeoutMilliSec);
+	
 private:
 
 	#ifdef _WIN32
-- 
cgit v1.2.3