Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

118 lines
2.6 KiB

#include "stdafx.h"
#include "thread.h"
// general purpose worker thread
// The thread exists in an alertable Wait state and does most of its work
// in APCs.
CEventThread::CEventThread()
{
InitializeCriticalSection(&m_cs);
m_uRef=0;
m_dwFlags = 0;
m_hThread = 0;
m_hSignalEvent = CreateEvent(NULL,FALSE,FALSE,NULL);
m_hAckEvent = CreateEvent(NULL,FALSE,FALSE,NULL);
}
CEventThread::~CEventThread()
{
DeleteCriticalSection(&m_cs);
if (m_hSignalEvent)
CloseHandle(m_hSignalEvent);
if (m_hAckEvent)
CloseHandle(m_hAckEvent);
}
BOOL
CEventThread::Start()
{
BOOL fRet;
EnterCriticalSection(&m_cs);
if (++m_uRef == 1)
{
m_hThread = CreateThread(
NULL,0,
EventThreadProc,
this, 0, &m_idThread);
}
fRet = !!m_hThread;
LeaveCriticalSection(&m_cs);
return fRet;
}
BOOL
CEventThread::Stop()
{
BOOL fRet;
EnterCriticalSection(&m_cs);
if (--m_uRef == 0)
{
m_dwFlags |= CTHREADF_STOP;
SetEvent(m_hSignalEvent);
WaitForSingleObject(m_hThread, INFINITE);
CloseHandle(m_hThread);
m_hThread = NULL;
m_dwFlags &= ~CTHREADF_STOP;
}
LeaveCriticalSection(&m_cs);
return TRUE;
}
DWORD
CEventThread::ThreadMethod()
{
DWORD dwWait;
while (1) {
dwWait = WaitForSingleObjectEx(m_hSignalEvent, INFINITE, TRUE);
if (dwWait == WAIT_OBJECT_0)
{
if (m_dwFlags & CTHREADF_STOP)
{
break; // exit thread
}
}
else if (dwWait == WAIT_IO_COMPLETION)
{
}
else
if (dwWait == WAIT_FAILED)
{
break; // exit thread
}
}
return dwWait;
}
// Executes the CallProc in the worker thread in a synchronous fashion
// (similar to SendMessage). When this method returns the CallProc has been called
// and the return value is returned.
BOOL CEventThread::CallNow(THREADCALLBACK CallProc, PVOID pParam, DWORD dwParam)
{
BOOL fRet = FALSE;
EnterCriticalSection(&m_cs);
m_Callback.CallProc = CallProc;
m_Callback.pParam = pParam;
m_Callback.dwParam = dwParam;
ResetEvent(m_hAckEvent);
if (QueueUserAPC(HandleCallNowAPC, m_hThread, (DWORD)this))
{
DWORD dwWait;
dwWait = WaitForSingleObject(m_hAckEvent, INFINITE);
if (dwWait == WAIT_OBJECT_0)
fRet = (BOOL) m_Callback.dwParam; //get the cached return value
}
LeaveCriticalSection(&m_cs);
return fRet;
}
void APIENTRY CEventThread::HandleCallNowAPC(DWORD dwArg)
{
CEventThread *pThis = (CEventThread *)dwArg;
if (pThis->m_Callback.CallProc)
{
pThis->m_Callback.dwParam = pThis->m_Callback.CallProc(pThis->m_Callback.pParam, pThis->m_Callback.dwParam);
SetEvent(pThis->m_hAckEvent);
}
}