windows-nt/Source/XPSP1/NT/net/rras/common/rtutils/wt.c
2020-09-26 16:20:57 +08:00

1434 lines
53 KiB
C

#include "wt.h"
#include "wtproto.h"
#include "workerinc.h"
//----------------------------------------------------------------------------------//
// WorkerFunction_ProcessClientNotifyWTEvent //
// the client has locks on wte_CS, so that no one can add or delete //
// if you need to add timers, you have to take lock on wte_CSTimer //
//----------------------------------------------------------------------------------//
VOID
WorkerFunction_ProcessClientNotifyWTEvent (
IN PVOID pContext
)
{
PWAIT_THREAD_ENTRY pwte;
BOOL bLockTimer;
pwte = (PWAIT_THREAD_ENTRY) pContext;
TRACE0(ENTER, "Entering WorkerFunction_ProcessClientNotifyWTEvent:");
switch (pwte->wte_ChangeType) {
case CHANGE_TYPE_ADD :
RegisterClientEventsTimers(pwte);
break;
case CHANGE_TYPE_DELETE :
DeRegisterClientEventsTimers(pwte);
break;
case CHANGE_TYPE_BIND_FUNCTION_ADD :
RegisterClientEventBinding(pwte);
break;
case CHANGE_TYPE_BIND_FUNCTION_DELETE :
DeRegisterClientEventBinding(pwte);
break;
}
SET_EVENT(pwte->wte_WTNotifyClientEvent, "wte_WTNotifyClientEvent",
"WorkerFunction_ProcessClientNotifyWTEvent");
TRACE0(LEAVE, "Leaving WorkerFunction_ProcessClientNotifyWTEvent:");
return;
} //end WorkerFunction_ProcessClientNotifyWTEvent
//----------------------------------------------------------------------------------//
// WorkerFunction_ProcessWorkQueueTimer //
//----------------------------------------------------------------------------------//
VOID
WorkerFunction_ProcessWorkQueueTimer (
IN PVOID pContext
)
{
TRACE0(ENTER, "Entering WorkerFunction_ProcessWorkQueueTimer:");
// Work queue has not been served within specified
// timeout
while (1) {
// Make a local copy of the count
LONG count = ThreadCount;
// Make sure we havn't exceded the limit
if (count>=MAX_WORKER_THREADS)
break;
else {
// Try to increment the value
// use another local variable
// because of MIPS optimizer bug
LONG newCount = count+1;
if (InterlockedCompareExchange (&ThreadCount,
newCount, count)==count) {
HANDLE hThread;
DWORD tid;
// Create new thread if increment succeded
hThread = CreateThread (NULL, 0, WorkerThread, NULL, 0, &tid);
if (hThread!=NULL) {
CloseHandle (hThread);
}
else // Restore the value if thread creation
// failed
InterlockedDecrement (&ThreadCount);
break;
}
// else repeat the loop if ThreadCount was modified
// while we were checking
}
}
TRACE0(LEAVE, "Leaving WorkerFunction_ProcessWorkQueueTimer:");
return;
} //end WorkerFunction_ProcessWorkQueueTimer
//----------------------------------------------------------------------------------//
// WorkerFunction_ProcessAlertableThreadSemaphore //
//----------------------------------------------------------------------------------//
VOID
WorkerFunction_ProcessAlertableThreadSemaphore (
IN PVOID pContext
)
{
WorkItem *workitem;
TRACE0(ENTER, "Entering WorkerFunction_ProcessAlertableThreadSemaphore:");
EnterCriticalSection(&AlertableWorkQueueLock);
ASSERT (!IsListEmpty (&AlertableWorkQueue));
workitem = (WorkItem *) RemoveHeadList (&AlertableWorkQueue) ;
LeaveCriticalSection(&AlertableWorkQueueLock);
(workitem->WI_Function) (workitem->WI_Context);
HeapFree (AlertableWorkerHeap, 0, workitem);
TRACE0(LEAVE, "Leaving WorkerFunction_ProcessAlertableThreadSemaphore:");
return;
}
//++-------------------------------------------------------------------------------*//
// WorkerFunction_ProcessWaitableTimer //
// Process event: waitable timer fired //
// takes lock on wte_CSTimer and releases it in the end //
//----------------------------------------------------------------------------------//
VOID
WorkerFunction_ProcessWaitableTimer(
IN PVOID pContext
)
{
PWAIT_THREAD_ENTRY pwte;
LONGLONG liCurrentTime;
PLIST_ENTRY ple, pHead, pleCurrent;
PWT_TIMER_ENTRY pte;
BOOL bActiveTimers;
TRACE0(ENTER, "Entering WorkerFunction_ProcessWaitableTimer:");
pwte = (PWAIT_THREAD_ENTRY) pContext;
// get lock on server threads timer
ENTER_CRITICAL_SECTION(&pwte->wte_CSTimer, "wte_CSTimer", "WorkerFunction_ProcessWaitableTimer");
NtQuerySystemTime((LARGE_INTEGER*) &liCurrentTime);
// ordered in increasing time, with inactive timers at the end
pHead = &pwte->wteL_ClientTimerEntries;
for (ple=pHead->Flink; ple!=pHead; ) {
pte = CONTAINING_RECORD(ple, WT_TIMER_ENTRY, te_ServerLinks);
if (pte->te_Status == TIMER_INACTIVE) // inactive timers at end of list
break;
if (IS_TIMER_INFINITE(pte->te_Timeout)) { //should have been inactive
ple = ple->Flink;
continue;
}
if (pte->te_Timeout<=liCurrentTime) {
//
// set timer status to inactive and insert at end of timer queue
//
pte->te_Status = TIMER_INACTIVE;
SET_TIMER_INFINITE(pte->te_Timeout);
pleCurrent = ple;
ple = ple->Flink;
RemoveEntryList(pleCurrent);
InsertTailList(pHead, pleCurrent);
// run the function in current thread or dispatch to worker thread
//
if (pte->te_RunInServer) {
(pte->te_Function)(pte->te_Context);
}
else {
QueueWorkItem( pte->te_Function, pte->te_Context, FALSE ); // do not run in alertable thread
}
}
else {
break;
}
}
//
// search for active timers with timeout which is not infinite
//
if (IsListEmpty(pHead))
bActiveTimers = FALSE;
else {
ple = pHead->Flink;
pte = CONTAINING_RECORD(ple, WT_TIMER_ENTRY, te_ServerLinks);
bActiveTimers = (pte->te_Status==TIMER_INACTIVE) ? FALSE : TRUE;
}
//
// if active timers present, then set waitableTimer
//
if (bActiveTimers) {
// set next timeout value for the wait server
pwte->wte_Timeout = pte->te_Timeout;
TRACE2(TIMER, "SetWaitableTimer set to <%lu:%lu> after being fired",
TIMER_HIGH(pte->te_Timeout), TIMER_LOW(pte->te_Timeout));
SetWaitableTimer(pwte->wte_Timer, (LARGE_INTEGER*)&pte->te_Timeout, 0, NULL, NULL, FALSE);
}
// no active timer in queue. do not set the waitable timer
else {
SET_TIMER_INFINITE(pwte->wte_Timeout);
}
#if DBG2
DebugPrintWaitWorkerThreads(DEBUGPRINT_FILTER_EVENTS);
#endif
LEAVE_CRITICAL_SECTION(&pwte->wte_CSTimer, "wte_CSTimer", "WorkerFunction_ProcessWaitableTimer");
TRACE0(LEAVE, "Leaving WorkerFunction_ProcessWaitableTimer:");
return;
} //end WorkerFunction_ProcessWaitableTimer
//---------------------------------------------------------------------------------*//
// CreateServerEventsAndTimer //
// Create the events for a server and the timer(called by server only) //
// assumes lock on server and server timer structure //
// THIS FUNCTION SHOULD NOT HAVE ANY CALLS TO API FUNCTIONS //
//----------------------------------------------------------------------------------//
DWORD
CreateServerEventsAndTimer (
IN PWAIT_THREAD_ENTRY pwte
)
{
PWT_EVENT_ENTRY peeWaitableTimer, peeClientNotifyWTEvent,
peeAlertableThreadSemaphore, peeWorkQueueTimer;
// waitable timer (set to infinity time)
peeWaitableTimer = CreateWaitEvent(
pwte->wte_Timer,
NULL, FALSE, FALSE, NULL,
TRUE, // highPriority
(WORKERFUNCTION) WorkerFunction_ProcessWaitableTimer,
(PVOID)pwte, // context
0, // context size=0, as just a pointer value is being passed
TRUE // run in server context
);
if (!peeWaitableTimer)
return ERROR_NOT_ENOUGH_MEMORY;
peeWaitableTimer->ee_EventId = 1;
RegisterClientEventLocal(peeWaitableTimer, pwte);
// ClientNotifyWTEvent
peeClientNotifyWTEvent = CreateWaitEvent(
pwte->wte_ClientNotifyWTEvent,
NULL, FALSE, FALSE, NULL,
TRUE, // highPriority
(WORKERFUNCTION) WorkerFunction_ProcessClientNotifyWTEvent,
(PVOID)pwte, // context
0, // context size=0, as just a pointer value is being passed
TRUE // run in server context
);
if (!peeClientNotifyWTEvent )
return ERROR_NOT_ENOUGH_MEMORY;
peeClientNotifyWTEvent->ee_EventId = 2;
RegisterClientEventLocal(peeClientNotifyWTEvent, pwte);
// AlertableThreadSemaphore
peeAlertableThreadSemaphore = CreateWaitEvent(
AlertableThreadSemaphore,
NULL, FALSE, FALSE, NULL,
FALSE, // Priority=low
(WORKERFUNCTION) WorkerFunction_ProcessAlertableThreadSemaphore,
NULL, // context
0, // context size=0, as just a pointer value is being passed
TRUE // run in server context
);
peeAlertableThreadSemaphore->ee_EventId = 3;
RegisterClientEventLocal(peeAlertableThreadSemaphore, pwte);
// WorkQueueTimer
peeWorkQueueTimer = CreateWaitEvent(
WorkQueueTimer,
NULL, FALSE, FALSE, NULL,
FALSE, // Priority=low
(WORKERFUNCTION) WorkerFunction_ProcessWorkQueueTimer,
NULL, // context
0, // context size=0, as just a pointer value is being passed
TRUE // run in server context
);
peeWorkQueueTimer->ee_EventId = 4;
RegisterClientEventLocal(peeWorkQueueTimer, pwte);
return NO_ERROR;
} //end CreateServerEventsAndTimer
//---------------------------------------------------------------------------------*//
// InitializeWaitGlobal //
// initialize the global data structure for all wait threads //
//----------------------------------------------------------------------------------//
DWORD
InitializeWaitGlobal(
)
{
BOOL bErr;
DWORD dwErr = NO_ERROR;
ZeroMemory(&WTG, sizeof(WTG));
WTG.g_Initialized = 0x12345678;
//
// initialize tracing and logging
//
WTG.g_TraceId = TraceRegister("WAIT_THREAD");
//todo:set the logging level
WTG.g_LogLevel = WT_LOGGING_ERROR;
WTG.g_LogHandle = RouterLogRegister("WAIT_THREAD");
TRACE0(ENTER, "Entering InitializeWaitGlobal()");
//
// initialize global structure
//
bErr = FALSE;
do { // error breakout loop
//
// create a private heap for Wait-Thread
//
WTG.g_Heap = AlertableWorkerHeap; // created in worker.c
if (WTG.g_Heap==NULL)
WTG.g_Heap = HeapCreate(0, 0, 0);
if (WTG.g_Heap == NULL) {
dwErr = GetLastError();
TRACE1(
ANY, "error %d creating Wait-Thread global heap", dwErr
);
LOGERR0(HEAP_CREATE_FAILED, dwErr);
bErr = FALSE;
break;
}
// initialize list for wait thread entries
//
InitializeListHead(&WTG.gL_WaitThreadEntries);
// initialize critical section
//
try {
InitializeCriticalSection(&WTG.g_CS);
}
except (EXCEPTION_EXECUTE_HANDLER) {
dwErr = GetExceptionCode();
TRACE1(
ANY, "exception %d initializing global critical section",
dwErr
);
LOGERR0(INIT_CRITSEC_FAILED, dwErr);
bErr = TRUE;
break;
}
} while (FALSE);
TRACE1(LEAVE, "leaving InitializeWaitGlobal: %d", dwErr);
if (bErr)
return dwErr;
else
return NO_ERROR;
} //end InitializeWaitGlobal
//---------------------------------------------------------------------------------*//
// DeInitializeWaitGlobal //
//deinitializes the global structure for wait-thread //
//todo: free the server entries and the event/timers etc associated with them //
//----------------------------------------------------------------------------------//
DWORD
DeInitializeWaitGlobal(
)
{
PLIST_ENTRY ple, pHead;
PWAIT_THREAD_ENTRY pwte;
ENTER_CRITICAL_SECTION(&WTG.g_CS, "g_CS", "DeInitializeWaitGlobal");
if (WTG.g_Initialized==0) {
LEAVE_CRITICAL_SECTION(&WTG.g_CS, "g_CS", "DeInitializeWaitGlobal");
TRACE0(LEAVE, "leaving DeInitializeWaitGlobal:Pending");
return ERROR_CAN_NOT_COMPLETE;
}
else
WTG.g_Initialized = 0;
//
// if waitThreadEntries exist, then for each server entry mark each event/binding
// as deleted and return pending
//
if (!IsListEmpty(&WTG.gL_WaitThreadEntries)) {
pHead = &WTG.gL_WaitThreadEntries;
for (ple=pHead->Flink; ple!=pHead; ple=ple->Flink) {
pwte = CONTAINING_RECORD(ple, WAIT_THREAD_ENTRY, wte_Links);
ENTER_CRITICAL_SECTION(&pwte->wte_CS, "deleting wte_CS", "DeInitializeWaitGlobal");
pwte->wte_Status = WT_STATUS_DELETED;
}
//todo should I also mark each event and timer as deleted
TRACE0(LEAVE, "leaving DeInitializeWaitGlobal:Pending");
return ERROR_CAN_NOT_COMPLETE;
}
DeInitializeWaitGlobalComplete();
return NO_ERROR;
} //end DeInitializeWaitGlobal
//----------------------DeInitializeWaitGlobalComplete-----------------------------//
DWORD
DeInitializeWaitGlobalComplete (
)
{
TRACE0(LEAVE, "leaving DeInitializeWaitGlobal");
// for each server entry mark each event/binding as deleted
// and return pending
TraceDeregister(WTG.g_TraceId);
RouterLogDeregister(WTG.g_LogHandle);
// delete critical section
try{
DeleteCriticalSection(&WTG.g_CS);
}
except (EXCEPTION_EXECUTE_HANDLER) {
}
// destroy heap
/*if (WTG.g_Heap != NULL) {
HeapDestroy(WTG.g_Heap);
}*/
return NO_ERROR;
} //end DeInitializeWaitGlobal
//++-------------------------------------------------------------------------------*//
// InsertWaitThreadEntry //
// insert the new wait server thread into a list of increasing ServerIds //
// initializes the serverId //
// assumes: it has lock on WTG.g_cs, and pwte->wte_CS, and pwte->wte_CSTimer //
// NO CALLS TO APIS SHOULD BE MADE HERE //
//----------------------------------------------------------------------------------//
DWORD
InsertWaitThreadEntry (
IN PWAIT_THREAD_ENTRY pwteInsert
)
{
PLIST_ENTRY ple, pHead;
PWAIT_THREAD_ENTRY pwte;
DWORD dwServerId;
//
// find the lowest unallocated SeverId and insert the new server in the ordered list
//
dwServerId = 1;
pHead = &WTG.gL_WaitThreadEntries;
for (ple=pHead->Flink; ple!=pHead; ple=ple->Flink,dwServerId++) {
pwte = CONTAINING_RECORD(ple, WAIT_THREAD_ENTRY, wte_Links);
if (dwServerId==pwte->wte_ServerId) {
continue;
}
else
break;
}
// insert in list
if (dwServerId==1) { // not required. but kept for easy understanding
InsertHeadList(&WTG.gL_WaitThreadEntries, &pwteInsert->wte_Links);
}
else {
InsertTailList(ple, &pwteInsert->wte_Links);
}
// set serverId
pwteInsert->wte_ServerId = dwServerId;
pwteInsert->wte_Status = WT_STATUS_REGISTERED;
return NO_ERROR;
} //end InsertWaitThreadEntry
//---------------------------------------------------------------------------------*//
// RemoveWaitThreadEntry //
// removes a wait server thread from the list of servers //
// assumes: it has lock on WTG.g_cs and wte_CS //
//----------------------------------------------------------------------------------//
DWORD
DeleteWaitThreadEntry (
IN PWAIT_THREAD_ENTRY pwte
)
{
// set status to deleted
pwte->wte_Status = WT_STATUS_DELETED;
// if RefCount == 0, then remove it from the list and free it
if (pwte->wte_RefCount==0) {
// free the wait thread entry
RemoveEntryList(&pwte->wte_Links);
FreeWaitThreadEntry(pwte);
// deinitialize global structure if it is also marked for delete and its
// WaitThreadEntry list is empty
if ((WTG.g_Initialized==WT_STATUS_DELETED)
&&(IsListEmpty(&WTG.gL_WaitThreadEntries)))
{
DeInitializeWaitGlobalComplete();
}
}
return NO_ERROR;
}
//---------------------------------------------------------------------------------*//
// CreateWaitThreadEntry //
// creates a wait thread entry and initializes it //
// no locks required //
//----------------------------------------------------------------------------------//
DWORD
CreateWaitThreadEntry (
IN DWORD dwThreadId,
OUT PWAIT_THREAD_ENTRY *ppwte
)
{
PWAIT_THREAD_ENTRY pwte;
DWORD dwErr = NO_ERROR;
BOOL bErr = TRUE;
TRACE0(ENTER, "Entering CreateWaitThreadEntry");
//
// allocate wait-thread-entry entry
//
*ppwte = pwte = WT_MALLOC(sizeof(WAIT_THREAD_ENTRY));
if (pwte == NULL) {
dwErr = GetLastError();
TRACE2(
ANY, "error %d allocating %d bytes for wait-thread-entry",
dwErr, sizeof(WAIT_THREAD_ENTRY)
);
LOGERR0(HEAP_ALLOC_FAILED, dwErr);
return dwErr;
}
ZeroMemory(pwte, sizeof(WAIT_THREAD_ENTRY));
//
// initialize global structure
//
do { // error breakout loop
// critical section
try {
InitializeCriticalSection(&pwte->wte_CS);
}
except (EXCEPTION_EXECUTE_HANDLER) {
dwErr = GetExceptionCode();
TRACE1(
ANY, "exception %d initializing global critical section",
dwErr
);
LOGERR0(INIT_CRITSEC_FAILED, dwErr);
break;
}
// current clients
//server id is set when it is inserted into global list
pwte->wte_ThreadId = dwThreadId; //ServerId is assigned during insertion of wte
pwte->wte_NumClients = 0;
// list for events/timers
InitializeListHead(&pwte->wteL_ClientEventEntries);
InitializeListHead(&pwte->wteL_ClientTimerEntries);
// create waitable timer
pwte->wte_Timer = CreateWaitableTimer(NULL, FALSE, NULL);
if (pwte->wte_Timer == NULL) {
dwErr = GetLastError();
TRACE1(
ANY, "error creating waitable timer",
dwErr
);
LOGERR0(CREATE_WAITABLE_TIMER_FAILED, dwErr);
break;
}
SET_TIMER_INFINITE(pwte->wte_Timeout); //set timeout to infinity
// critical section for timers
try {
InitializeCriticalSection(&pwte->wte_CSTimer);
}
except (EXCEPTION_EXECUTE_HANDLER) {
dwErr = GetExceptionCode();
TRACE1(
ANY, "exception %d initializing critical section for timer",
dwErr
);
LOGERR0(INIT_CRITSEC_FAILED, dwErr);
break;
}
// array for WaitForMultipleObjects
pwte->wte_LowIndex = 0;
pwte->wte_NumTotalEvents = 0;
pwte->wte_NumActiveEvents = 0;
pwte->wte_NumHighPriorityEvents = 0;
pwte->wte_NumActiveHighPriorityEvents = 0;
//
// adding/deleting events/timers
//
// create event: Client notifies WT: wake up WT to add/delete events/timers
pwte->wte_ClientNotifyWTEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
if (pwte->wte_ClientNotifyWTEvent == NULL) {
dwErr = GetLastError();
TRACE1(START, "error %d creating event Client-notify-WT", dwErr);
LOGERR0(CREATE_EVENT_FAILED, dwErr);
break;
}
// create event: WT notifies Client: the work requested has been done
pwte->wte_WTNotifyClientEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
if (pwte->wte_WTNotifyClientEvent == NULL) {
dwErr = GetLastError();
TRACE1(START, "error:%d creating event WT-notify-Client", dwErr);
LOGERR0(CREATE_EVENT_FAILED, dwErr);
break;
}
// variables used to add/delete events/timers
pwte->wte_ChangeType = CHANGE_TYPE_NONE;
pwte->wte_Status = 0;
pwte->wte_RefCount = 0;
pwte->wtePL_EventsToChange = NULL;
pwte->wtePL_TimersToChange = NULL;
bErr = FALSE;
} while (FALSE);
TRACE1(LEAVE, "Leaving CreateWaitThreadEntry: %d", dwErr);
if (bErr) {
FreeWaitThreadEntry(pwte);
return dwErr;
}
else
return NO_ERROR;
}//end CreateWaitThreadEntry
//---------------------------------------------------------------------------------*//
// FreeWaitThreadEntry //
//----------------------------------------------------------------------------------//
DWORD
FreeWaitThreadEntry (
IN PWAIT_THREAD_ENTRY pwte
)
{
DWORD dwErr = NO_ERROR;
TRACE0(ENTER, "Entering FreeWaitThreadEntry");
DeleteCriticalSection(&pwte->wte_CS);
// delete waitable timer
if (pwte->wte_Timer!=NULL)
CloseHandle(pwte->wte_Timer);
DeleteCriticalSection(&pwte->wte_CSTimer);
// delete event: Client notifies WT
if (pwte->wte_ClientNotifyWTEvent!=NULL)
CloseHandle(&pwte->wte_ClientNotifyWTEvent);
// delete event: WT notifies Client
if (pwte->wte_WTNotifyClientEvent!=NULL)
CloseHandle(pwte->wte_WTNotifyClientEvent);
//
// free wait-thread-entry record
//
WT_FREE(pwte);
// deinitialize global structure if it is also marked for delete and its
// WaitThreadEntry list is empty
if ((WTG.g_Initialized==WT_STATUS_DELETED)
&&(IsListEmpty(&WTG.gL_WaitThreadEntries)))
{
DeInitializeWaitGlobalComplete();
}
TRACE1(LEAVE, "Leaving FreeWaitThreadEntry: %d", dwErr);
return NO_ERROR;
} //end FreeWaitThreadEntry
//----------------------------------------------------------------------------------//
// GetWaitThread //
// returns a wait thread which has the required number of free events //
// assumes lock on g_CS. //
// if NumTotalEvents is low enough, do interlocked increment. //
//----------------------------------------------------------------------------------//
PWAIT_THREAD_ENTRY
GetWaitThread (
IN DWORD dwNumEventsToAdd, //==0, if only timers are being added
IN DWORD dwNumTimersToadd
)
{
PLIST_ENTRY ple, pHead;
PWAIT_THREAD_ENTRY pwte;
BOOL bFound;
DWORD dwIncrRefCount = dwNumEventsToAdd + dwNumTimersToadd;
pHead = &WTG.gL_WaitThreadEntries;
//
// Locate a wait-thread with required events free.
//
bFound = FALSE;
for (ple=pHead->Flink; ple!=pHead; ple=ple->Flink) {
pwte = CONTAINING_RECORD(ple, WAIT_THREAD_ENTRY, wte_Links);
// cannot allocate this server because it is not active
if ((pwte->wte_Status != WT_STATUS_REGISTERED)
&& (pwte->wte_Status != WT_STATUS_ACTIVE)
)
{
ETRACE0("could not allocate this wte as it is not registered");
continue;
}
if (NUM_CLIENT_EVENTS_FREE(pwte) >= dwNumEventsToAdd) {
// increment RefCount so that it cannot be deleted
// interlocked operation as it can be decremented outside g_CS and inside only wte_CS
InterlockedExchangeAdd((PLONG)&pwte->wte_RefCount, dwIncrRefCount);
InterlockedExchangeAdd((PLONG)&pwte->wte_NumTotalEvents,
(LONG)dwNumEventsToAdd);
bFound = TRUE;
break;
}
}
if (bFound) {
//shift the wte to the end so that all server threads are balanced
RemoveEntryList(ple);
InsertTailList(pHead, ple);
return pwte;
}
else {
return NULL;
}
}
//----------------------------------------------------------------------------------//
// RegisterClientEventLocal //
// Register the wait event locally //
// event being registered by the server itself: //
// assumes lock on server //
// somewhat similar to AddClientEvent //
//----------------------------------------------------------------------------------//
DWORD
RegisterClientEventLocal (
IN PWT_EVENT_ENTRY pee,
IN PWAIT_THREAD_ENTRY pwte
)
{
// insert wait event in list of event entries //dont have to do interlocked here
pwte->wte_RefCount ++;
pwte->wte_NumTotalEvents ++;
InsertInEventsList(pee, pwte);
// insert wait event in array of event entries
if (pee->ee_bInitialState == FALSE) {
InsertInEventsArray(pee, pwte);
}
else {
pee->ee_Status = WT_STATUS_INACTIVE + WT_STATUS_FIRED; // but arrayIndex ==-1
}
return NO_ERROR;
}
//----------------------------------------------------------------------------------//
// InsertInEventsList //
// assumes lock on server //
// Insert the event in the list and increment the counters //
// NumTotalEvents has been increased during allocation of the server //
//----------------------------------------------------------------------------------//
VOID
InsertInEventsList (
IN PWT_EVENT_ENTRY pee,
IN PWAIT_THREAD_ENTRY pwte
)
{
InsertHeadList(&pwte->wteL_ClientEventEntries, &pee->ee_ServerLinks);
//
// set fields of event entry
//
pee->ee_ServerId = pwte->wte_ServerId;
pee->eeP_wte = pwte;
pee->ee_Status = WT_STATUS_REGISTERED;
//
// change fields of wait thread entry
//
//pwte->wte_RefCount++; //RefCount incremented during allocation
//pwte->wte_NumTotalEvents ++; //incremented during allocation
if (pee->ee_bHighPriority) {
pwte->wte_NumHighPriorityEvents ++;
}
return;
}
//----------------------------------------------------------------------------------//
// DeleteFromEventsList //
// NumTotalEvents/Refcount is interlockedDecremented //
//----------------------------------------------------------------------------------//
VOID
DeleteFromEventsList (
IN PWT_EVENT_ENTRY pee,
IN PWAIT_THREAD_ENTRY pwte
)
{
// remove entry from list
RemoveEntryList(&pee->ee_ServerLinks);
//
// set fields of event entry
//
pee->ee_Status = WT_STATUS_DELETED;
//
// change fields of wait thread entry
//
// pwte->wte_NumTotalEvents ++; interlocked incremented during allocation
if (pee->ee_bHighPriority) {
pwte->wte_NumHighPriorityEvents --;
}
InterlockedDecrement(&pwte->wte_NumTotalEvents);
// decremented in DeleteClientEventComplete if ee_RefCount is 0
//InterlockedDecrement(&pwte->wte_RefCount);
return;
}
//----------------------------------------------------------------------------------//
// DeleteFromEventsArray //
// pee: status is not set to inactive, but array pointer is set to -1 //
// pwte: decrement active counters //
//----------------------------------------------------------------------------------//
VOID
DeleteFromEventsArray (
IN PWT_EVENT_ENTRY pee,
IN PWAIT_THREAD_ENTRY pwte
)
{
INT iIndex;
DWORD dwCount;
iIndex = pee->ee_ArrayIndex;
dwCount = pwte->wte_NumActiveEvents + pwte->wte_LowIndex - iIndex - 1;
// shift right part towards left if its size is smaller
if (dwCount <= pwte->wte_NumActiveEvents/2) {
EventsArray_MoveOverlap (
pwte,
iIndex, //dstn
iIndex+1, //src
dwCount //count
);
}
// shift left part towards right
else {
EventsArray_MoveOverlap (
pwte,
pwte->wte_LowIndex+1, //dstn
pwte->wte_LowIndex, //src
pwte->wte_NumActiveEvents - dwCount -1 //count
);
pwte->wte_LowIndex ++;
}
// set fields of event entry
//
pee->ee_ArrayIndex = -1;
// change fields of wait thread entry
//
pwte->wte_NumActiveEvents --;
if (pee->ee_bHighPriority) {
pwte->wte_NumActiveHighPriorityEvents--;
}
}//end DeleteFromEventsArray
//----------------------------------------------------------------------------------//
// InsertInEventsArray //
// assumes lock on server:wte_CS :todo is the lock required //
// Insert the event in the events array and the map array (no checks are performed)//
// pee: status set to active, set index to array position //
// pwte: increment active counters //
//----------------------------------------------------------------------------------//
VOID
InsertInEventsArray (
IN PWT_EVENT_ENTRY pee,
IN PWAIT_THREAD_ENTRY pwte
)
{
INT iIndex;
// if the array is filled to the extreme right, then shift all events to the left end
if (EA_OVERFLOW(pwte, 1))
EventsArray_CopyLeftEnd(pwte);
//
// get index where it has to be inserted
//
if (pee->ee_bHighPriority) {
// the highPriority event has to be moved to the place right of the righmost HighPriority event
iIndex = EA_INDEX_LOW_LOW_PRIORITY_EVENT(pwte);
//copy the 1st low priority event to end+1;
if (EA_EXISTS_LOW_PRIORITY_EVENT(pwte)) {
EventsArray_Move(pwte,
EA_INDEX_HIGH_LOW_PRIORITY_EVENT(pwte)+1, //dstn
iIndex, //src
1 //count
);
}
}
else { // low priority event: insert in the end
iIndex = EA_INDEX_HIGH_LOW_PRIORITY_EVENT(pwte)+1;
}
// insert the event
EventsArray_InsertEvent(pee, pwte, iIndex);
// set fields of event entry
//
pee->ee_Status = WT_STATUS_ACTIVE;
pee->ee_ArrayIndex = iIndex;
// change fields of wait thread entry
//
pwte->wte_NumActiveEvents ++;
if (pee->ee_bHighPriority) {
pwte->wte_NumActiveHighPriorityEvents++;
}
}//end InsertInEventsArray
//----------------------------------------------------------------------------------//
// InactivateEvent //
// Remove the event from the arrays and set the inactive flag // //
// Used with bManual reset //
//----------------------------------------------------------------------------------//
VOID
InactivateEvent (
IN PWT_EVENT_ENTRY pee
)
{
DWORD dwIndex;
PWAIT_THREAD_ENTRY pwte;
dwIndex = pee->ee_ArrayIndex;
pwte = pee->eeP_wte;
// if event is not at the right end of the array, then events on its right have to be shifted
if (dwIndex != EA_INDEX_HIGH_LOW_PRIORITY_EVENT(pwte)) {
EventsArray_MoveOverlap(pwte, dwIndex, dwIndex+1,
(pwte->wte_NumActiveEvents + pwte->wte_LowIndex - dwIndex -1)
);
}
//
// change fields in event entry to make it inactive
//
pee->ee_ArrayIndex = -1;
pee->ee_Status = (pee->ee_Status&WT_STATUS_FIRED) + WT_STATUS_INACTIVE;
//
// change fields in wait thread entry
//
pwte->wte_NumActiveEvents --;
if (pee->ee_bHighPriority) {
pwte->wte_NumActiveHighPriorityEvents --;
}
return;
}
//----------------------------------------------------------------------------------//
// EventsArray_CopyLeftEnd //
// copy all the events to the left end of the array //
// sets the wte_LowIndex value //
//----------------------------------------------------------------------------------//
VOID
EventsArray_CopyLeftEnd (
IN PWAIT_THREAD_ENTRY pwte
)
{
EventsArray_Move(pwte,
0, //dstn
pwte->wte_LowIndex, //src
pwte->wte_NumActiveEvents //count
);
//
// change fields of wait thread entry
//
pwte->wte_LowIndex = 0;
return;
}
//----------------------------------------------------------------------------------//
// EventsArray_Move //
// copy dwCount events from the srcIndex to dstnIndex (no overlap) //
//----------------------------------------------------------------------------------//
VOID
EventsArray_Move (
IN PWAIT_THREAD_ENTRY pwte,
IN DWORD dwDstnIndex,
IN DWORD dwSrcIndex,
IN DWORD dwCount
)
{
PWT_EVENT_ENTRY pee;
DWORD i;
if (dwCount==0)
return;
CopyMemory( &pwte->wteA_Events[dwDstnIndex],
&pwte->wteA_Events[dwSrcIndex],
sizeof(HANDLE) * dwCount
);
CopyMemory( &pwte->wteA_EventMapper[dwDstnIndex],
&pwte->wteA_EventMapper[dwSrcIndex],
sizeof(PWT_EVENT_ENTRY) * dwCount
);
for (i=0; i<dwCount; i++) {
pee = pwte->wteA_EventMapper[dwDstnIndex + i];
pee->ee_ArrayIndex = dwDstnIndex + i;
}
return;
}
//----------------------------------------------------------------------------------//
// EventsArray_MoveOverlap //
// copy dwCount events from the srcIndex to dstnIndex (with overlap) //
//----------------------------------------------------------------------------------//
VOID
EventsArray_MoveOverlap (
IN PWAIT_THREAD_ENTRY pwte,
IN DWORD dwDstnIndex,
IN DWORD dwSrcIndex,
IN DWORD dwCount
)
{
PWT_EVENT_ENTRY pee;
DWORD i;
if (dwCount==0)
return;
MoveMemory( &pwte->wteA_Events[dwDstnIndex],
&pwte->wteA_Events[dwSrcIndex],
sizeof(HANDLE) * dwCount
);
MoveMemory( &pwte->wteA_EventMapper[dwDstnIndex],
&pwte->wteA_EventMapper[dwSrcIndex],
sizeof(PWT_EVENT_ENTRY) * dwCount
);
for (i=0; i<dwCount; i++) {
pee = pwte->wteA_EventMapper[dwDstnIndex + i];
pee->ee_ArrayIndex = dwDstnIndex + i;
}
return;
}
//----------------------------------------------------------------------------------//
// EventsArray_InsertEvent //
// Insert the event in the events array and the map array //
//----------------------------------------------------------------------------------//
VOID
EventsArray_InsertEvent (
IN PWT_EVENT_ENTRY pee,
IN PWAIT_THREAD_ENTRY pwte,
IN INT iIndex
)
{
// insert event in events array
pwte->wteA_Events[iIndex] = pee->ee_Event;
// insert pointer in map array
pwte->wteA_EventMapper[iIndex] = pee;
return;
}
//---------------------------------------------------------------------------------*//
// GetListLength //
// returns the length of the list //
// returns 0 if the list contains the header only //
//----------------------------------------------------------------------------------//
INT
GetListLength (
IN PLIST_ENTRY pHead
)
{
PLIST_ENTRY ple;
DWORD dwCount=0;
if (pHead==NULL)
return -1;
for (ple=pHead->Flink; ple!=pHead; ple=ple->Flink) {
dwCount++;
}
return dwCount;
}
VOID
PrintEvent (
PWT_EVENT_ENTRY pee,
DWORD level
)
{
PLIST_ENTRY pHead, ple;
PWT_WORK_ITEM pwi;
if (pee->ee_Event==NULL)
printf(" Event is NULL\n");
printf("-- -------------------------------------------------------------------------\n");
printf("-- <%2d><ee_bManualReset:%2d> <ee_bInitialState:%2d> <ee_Status:%2d> <ee_bHighPriority:%2d>\n",
pee->ee_EventId, pee->ee_bManualReset, pee->ee_bInitialState, pee->ee_Status,
pee->ee_bHighPriority);
printf("-- <ee_bSignalSingle:%2d> <ee_bOwnerSelf:%2d> <ee_ArrayIndex:%2d> <ee_ServerId:4%d>\n",
pee->ee_bSignalSingle, pee->ee_bOwnerSelf, pee->ee_ArrayIndex, pee->ee_ServerId);
printf("-- <ee_RefCount:%d\n", pee->ee_RefCount);
pHead = &pee->eeL_wi;
printf("-- LIST OF EVENT BINDINGS\n");
for (ple=pHead->Flink; ple!=pHead; ple=ple->Flink) {
pwi = CONTAINING_RECORD(ple, WT_WORK_ITEM, wi_ServerLinks);
printf("-- <wi_ContextSz:%lu> <wi_RunInServer:%d> <wiP_ee:%lu>\n",
pwi->wi_ContextSz, pwi->wi_RunInServer, (UINT_PTR)(PVOID)pwi->wiP_ee);
}
return;
}
VOID
PrintTimer (
PWT_TIMER_ENTRY pte,
DWORD level
)
{
LARGE_INTEGER li;
TRACE0(WAIT_TIMER,"__ ________________________________________________________________________");
TRACE4(WAIT_TIMER, "__ <TimerId:%2d><Timeout:%lu:%lu> <te_ContextSz:%lu>",
pte->te_TimerId, TIMER_HIGH(pte->te_Timeout), TIMER_LOW(pte->te_Timeout), pte->te_ContextSz);
TRACE3(WAIT_TIMER, "__ <te_RunInServer:%1d> <te_Status:%1d> <te_ServerId:%d>",
pte->te_RunInServer, pte->te_Status, pte->te_ServerId);
return;
}
//----------------------------------------------------------------------------------//
// PrintWaitThreadEntry //
// Prints the events and timers registered with this server. //
//----------------------------------------------------------------------------------//
VOID
PrintWaitThreadEntry (
PWAIT_THREAD_ENTRY pwte,
DWORD level
)
{
PLIST_ENTRY ple, pHead;
PWT_EVENT_ENTRY pee;
PWT_TIMER_ENTRY pte;
DWORD dwHigh, dwLow;
LARGE_INTEGER li;
BOOL bPrint;
DWORD i;
TRACE0(WAIT_TIMER, "\n");
TRACE0(WAIT_TIMER, "=================================================================================");
TRACE3(WAIT_TIMER, "== <wte_ServerId:%2lu> <wte_ThreadId:%2lu> <wte_NumClients:%2lu>",
pwte->wte_ServerId, pwte->wte_ThreadId, pwte->wte_NumClients);
TRACE2(WAIT_TIMER, "== <wte_Status:%2lu> <wte_RefCount:%2lu>", pwte->wte_Status, pwte->wte_RefCount);
//
// print list of events registered
//
if (!(level&DEBUGPRINT_FILTER_EVENTS)) {
TRACE0(WAIT_TIMER, "-- ");
TRACE0(WAIT_TIMER, "---------------------------------------------------------------------------------");
TRACE2(WAIT_TIMER,"-- <wte_LowIndex:%2lu> <wte_NumTotalEvents:%2lu>",
pwte->wte_LowIndex, pwte->wte_NumTotalEvents);
TRACE2(WAIT_TIMER, "-- <wte_NumActiveEvents:%2lu> <wte_NumHighPriorityEvents:%2lu>",
pwte->wte_NumActiveEvents, pwte->wte_NumHighPriorityEvents);
TRACE1(WAIT_TIMER, "-- <wte_NumActiveHighPriorityEvents:%2lu>", pwte->wte_NumActiveHighPriorityEvents);
if (level&0x2) {//dont print the initial 4 reserved events
bPrint = FALSE;
i = 0;
}
else
bPrint = TRUE;
TRACE0(WAIT_TIMER, "--");
pHead = &pwte->wteL_ClientEventEntries;
for (ple=pHead->Blink; ple!=pHead; ple=ple->Blink) {
if (!bPrint) {
if (++i==4)
bPrint=TRUE;
continue;
}
pee = CONTAINING_RECORD(ple, WT_EVENT_ENTRY, ee_ServerLinks);
PrintEvent(pee, level);
}
for (i=0; i<=10; i++) {
if (pwte->wteA_EventMapper[i]==NULL)
TRACE0(WAIT_TIMER, "--");
else
TRACE2(WAIT_TIMER, "<%d:%d>", (pwte->wteA_EventMapper[i])->ee_EventId,
(pwte->wteA_EventMapper[i])->ee_ArrayIndex);
}
}
//
// print list of timers registered
//
if (!(level&DEBUGPRINT_FILTER_TIMERS)) {
li = *(LARGE_INTEGER*)(PVOID)&pwte->wte_Timeout;
dwHigh = li.HighPart;
dwLow = li.LowPart;
TRACE0(WAIT_TIMER, "--");
TRACE0(WAIT_TIMER,"_________________________________________________________________________________");
TRACE2(WAIT_TIMER, "__ <wte_Timeout.high:%lu> <wte_Timeout.low:%lu>", dwHigh, dwLow);
pHead = &pwte->wteL_ClientTimerEntries;
for (ple=pHead->Flink; ple!=pHead; ple=ple->Flink) {
pte = CONTAINING_RECORD(ple, WT_TIMER_ENTRY, te_ServerLinks);
PrintTimer(pte, level);
}
}
return;
} //PrintWaitThreadEntry