LosaDev source snapshot 2019

extra/LS_Service/LS_network/cIocpQueue.cpp

@@ -0,0 +1,100 @@
+// cIocpQueue.cpp: implementation of the cIocpQueue class.
+//
+//////////////////////////////////////////////////////////////////////
+
+#include "StdAfx.h"
+#include "cIocpQueue.h"
+
+
+
+
+//////////////////////////////////////////////////////////////////////
+// Construction/Destruction
+//////////////////////////////////////////////////////////////////////
+
+cIocpQueue::cIocpQueue() : m_count(0), m_completionPort(NULL)
+{
+}
+
+cIocpQueue::~cIocpQueue()
+{
+	Cleanup();
+}
+
+//////////////////////////////////////////////////////////////////////
+// Startup/Cleanup
+//////////////////////////////////////////////////////////////////////
+BOOL cIocpQueue::Startup( int32 timeout )
+{
+	if( !m_completionPort )
+	{
+		m_completionPort = CreateIoCompletionPort(	INVALID_HANDLE_VALUE, 
+													NULL, 
+													0, 
+													0 );
+		if( !m_completionPort ) return FALSE;
+
+		m_timeout = timeout;
+		return TRUE;
+	}
+	
+	return FALSE;
+}
+
+void	cIocpQueue::Cleanup()
+{
+	if( m_completionPort )
+	{
+		CloseHandle( m_completionPort );
+		m_completionPort = NULL;
+	}	
+}
+
+
+//////////////////////////////////////////////////////////////////////
+// Enqueue/Dequeue
+//////////////////////////////////////////////////////////////////////
+BOOL cIocpQueue::Enqueue( const DWORD completionKey, const DWORD bytes )
+{
+	if( m_completionPort )
+	{
+		if(PostQueuedCompletionStatus( m_completionPort,
+										bytes,
+										completionKey,
+										NULL ))
+		{
+			InterlockedIncrement( &m_count );
+			return TRUE;
+		}
+	}
+	return FALSE;
+}
+
+LPDWORD	cIocpQueue::Dequeue()
+{
+	DWORD bytes = 0;
+ 	return Dequeue( bytes );
+}
+
+LPDWORD	cIocpQueue::Dequeue( DWORD &bytes )
+{
+	LPDWORD completionKey = NULL;
+	LPOVERLAPPED overlapped	= NULL;
+
+	if( m_completionPort )
+	{
+		if(GetQueuedCompletionStatus(	m_completionPort,
+										&bytes,
+										reinterpret_cast<LPDWORD>(&completionKey),
+										&overlapped,
+										m_timeout ))
+		{
+			InterlockedDecrement( &m_count );
+		}
+	}
+
+ 	return completionKey;
+}
+
+
+