mirror of https://github.com/tongzx/nt5src
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.
1299 lines
23 KiB
1299 lines
23 KiB
/*++
|
|
|
|
Copyright (c) 1998 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
atqbmon.cxx
|
|
|
|
Abstract:
|
|
|
|
ATQ Backlog Monitor
|
|
|
|
Author:
|
|
|
|
01-Dec-1998 MCourage
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
#include "isatq.hxx"
|
|
|
|
#include "atqbmon.hxx"
|
|
|
|
ATQ_BACKLOG_MONITOR * g_pAtqBacklogMonitor = NULL;
|
|
|
|
|
|
|
|
ATQ_BACKLOG_MONITOR::ATQ_BACKLOG_MONITOR(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Constructor sets up the list and the lock.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
--*/
|
|
{
|
|
INITIALIZE_CRITICAL_SECTION( &m_csLock );
|
|
InitializeListHead( &m_ListHead );
|
|
}
|
|
|
|
|
|
ATQ_BACKLOG_MONITOR::~ATQ_BACKLOG_MONITOR(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Destructor cleans up the list and the lock.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
--*/
|
|
{
|
|
LIST_ENTRY * pEntry;
|
|
ATQ_BMON_SET * pBmonSet;
|
|
|
|
//
|
|
// traverse the list and remove all sets
|
|
//
|
|
while (!IsListEmpty( &m_ListHead )) {
|
|
|
|
pEntry = RemoveHeadList( &m_ListHead );
|
|
|
|
pBmonSet = CONTAINING_RECORD( pEntry, ATQ_BMON_SET, m_SetList );
|
|
|
|
DBG_ASSERT( pBmonSet->IsEmpty() );
|
|
DBG_REQUIRE( pBmonSet->Cleanup() );
|
|
}
|
|
|
|
DeleteCriticalSection( &m_csLock );
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BACKLOG_MONITOR::AddEntry(
|
|
ATQ_BMON_ENTRY * pBmonEntry
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Adds an entry to the next available ATQ_BMON_SET.
|
|
Makes a new set if all current sets are full.
|
|
|
|
Arguments:
|
|
|
|
pBmonEntry - The entry to be added
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
LIST_ENTRY * pCurrent;
|
|
ATQ_BMON_SET * pBmonSet;
|
|
BOOL bRetval;
|
|
|
|
if (!pBmonEntry) {
|
|
return TRUE;
|
|
}
|
|
|
|
bRetval = FALSE;
|
|
|
|
Lock();
|
|
|
|
for (pCurrent = m_ListHead.Flink;
|
|
pCurrent != &m_ListHead;
|
|
pCurrent = pCurrent->Flink) {
|
|
|
|
pBmonSet = CONTAINING_RECORD( pCurrent, ATQ_BMON_SET, m_SetList );
|
|
|
|
if (pBmonSet->IsNotFull() && pBmonSet->AddEntry(pBmonEntry)) {
|
|
bRetval = TRUE;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!bRetval) {
|
|
//
|
|
// Couldn't find a set with space
|
|
// so try to make a new one.
|
|
//
|
|
pBmonSet = new ATQ_BMON_SET;
|
|
|
|
if (pBmonSet && pBmonSet->Initialize()) {
|
|
InsertHeadList( &m_ListHead, &pBmonSet->m_SetList );
|
|
bRetval = pBmonSet->AddEntry(pBmonEntry);
|
|
}
|
|
}
|
|
|
|
Unlock();
|
|
|
|
return bRetval;
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BACKLOG_MONITOR::RemoveEntry(
|
|
ATQ_BMON_ENTRY * pBmonEntry
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Removes an entry from its containing ATQ_BMON_SET.
|
|
If the set is empty it is removed from the list
|
|
of sets.
|
|
|
|
Arguments:
|
|
|
|
pBmonEntry - The entry to be removed
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
ATQ_BMON_SET * pBmonSet;
|
|
|
|
if (!pBmonEntry) {
|
|
return TRUE;
|
|
}
|
|
|
|
DBG_ASSERT( pBmonEntry && pBmonEntry->CheckSignature() );
|
|
|
|
pBmonSet = pBmonEntry->GetContainingBmonSet();
|
|
|
|
DBG_ASSERT( pBmonSet );
|
|
|
|
return pBmonSet->RemoveEntry(pBmonEntry);
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BACKLOG_MONITOR::PauseEntry(
|
|
ATQ_BMON_ENTRY * pBmonEntry
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Pauses an entry in its containing ATQ_BMON_SET.
|
|
The entry will remain in the set, but will
|
|
not send notifications.
|
|
|
|
Arguments:
|
|
|
|
pBmonEntry - The entry to be paused
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
ATQ_BMON_SET * pBmonSet;
|
|
|
|
if (!pBmonEntry) {
|
|
return TRUE;
|
|
}
|
|
|
|
DBG_ASSERT( pBmonEntry && pBmonEntry->CheckSignature() );
|
|
|
|
pBmonSet = pBmonEntry->GetContainingBmonSet();
|
|
|
|
DBG_ASSERT( pBmonSet );
|
|
|
|
return pBmonSet->PauseEntry(pBmonEntry);
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BACKLOG_MONITOR::ResumeEntry(
|
|
ATQ_BMON_ENTRY * pBmonEntry
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Undoes PauseEntry. The entry in question
|
|
will get notifications again
|
|
|
|
Arguments:
|
|
|
|
pBmonEntry - The entry to be resumed
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
ATQ_BMON_SET * pBmonSet;
|
|
|
|
if (!pBmonEntry) {
|
|
return TRUE;
|
|
}
|
|
|
|
DBG_ASSERT( pBmonEntry && pBmonEntry->CheckSignature() );
|
|
|
|
pBmonSet = pBmonEntry->GetContainingBmonSet();
|
|
|
|
DBG_ASSERT( pBmonSet );
|
|
|
|
return pBmonSet->ResumeEntry(pBmonEntry);
|
|
}
|
|
|
|
|
|
|
|
ATQ_BMON_ENTRY::ATQ_BMON_ENTRY(
|
|
SOCKET s
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Constructor sets up signature etc.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
--*/
|
|
{
|
|
DBG_ASSERT( s );
|
|
|
|
m_Signature = ATQ_BMON_ENTRY_SIGNATURE;
|
|
m_Socket = s;
|
|
m_hAddRemoveEvent = NULL;
|
|
m_BmonOpcode = BMON_INVALID;
|
|
m_pBmonSet = NULL;
|
|
m_dwErr = NO_ERROR;
|
|
}
|
|
|
|
|
|
ATQ_BMON_ENTRY::~ATQ_BMON_ENTRY(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Destructor sets up signature and closes event.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
--*/
|
|
{
|
|
DBG_ASSERT( CheckSignature() );
|
|
DBG_ASSERT( m_Socket );
|
|
|
|
m_Signature = ATQ_FREE_BMON_ENTRY_SIGNATURE;
|
|
|
|
DBG_REQUIRE( CloseHandle( m_hAddRemoveEvent ) );
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_ENTRY::InitEvent(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Sets up the AddRemove event that we use
|
|
to synchronise adding and removing of entries.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
DBG_ASSERT( CheckSignature() );
|
|
DBG_ASSERT( m_hAddRemoveEvent == NULL );
|
|
|
|
m_hAddRemoveEvent = CreateEvent(
|
|
NULL, // default security
|
|
FALSE, // do auto-reset
|
|
FALSE, // init state false
|
|
NULL // no name
|
|
);
|
|
|
|
if (m_hAddRemoveEvent == NULL) {
|
|
m_dwErr = GetLastError();
|
|
}
|
|
|
|
return (m_hAddRemoveEvent != NULL);
|
|
}
|
|
|
|
|
|
VOID
|
|
ATQ_BMON_ENTRY::SignalAddRemove(
|
|
DWORD dwError
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Signals the AddRemove event. Clients adding or
|
|
removing the entry will be blocked on its event.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
None
|
|
--*/
|
|
{
|
|
DBG_ASSERT( CheckSignature() );
|
|
DBG_ASSERT( m_hAddRemoveEvent != NULL );
|
|
|
|
m_dwErr = dwError;
|
|
DBG_REQUIRE( SetEvent(m_hAddRemoveEvent) );
|
|
}
|
|
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_ENTRY::WaitForAddRemove(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Call this function while waiting for
|
|
the entry to be added to or removed from a set.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
DWORD dwResult;
|
|
|
|
DBG_ASSERT( CheckSignature() );
|
|
DBG_ASSERT( m_hAddRemoveEvent != NULL );
|
|
|
|
//
|
|
// If the monitor thread dies, don't bother waiting for notification from
|
|
// it regarding add/removal.
|
|
//
|
|
|
|
Retry:
|
|
|
|
if ( m_pBmonSet && !m_pBmonSet->ThreadFinished() )
|
|
{
|
|
dwResult = WaitForSingleObject(m_hAddRemoveEvent, 10000 );
|
|
if ( dwResult == WAIT_TIMEOUT )
|
|
{
|
|
goto Retry;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
dwResult = WAIT_OBJECT_0;
|
|
}
|
|
|
|
DBG_ASSERT( dwResult == WAIT_OBJECT_0 );
|
|
|
|
return (dwResult == WAIT_OBJECT_0);
|
|
}
|
|
|
|
|
|
|
|
BOOL
|
|
BMON_WAKEUP_ENTRY::Callback(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
The sole purpose of the BMON_WAKEUP_ENTRY is to
|
|
get the main thread to wake up and call
|
|
SynchronizeSets. All we have to do here is
|
|
do a read so that we can wake up again in
|
|
the future.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
TRUE if successful, else FALSE
|
|
--*/
|
|
{
|
|
INT err;
|
|
DWORD dwBuff;
|
|
ATQ_BMON_SET * pBmonSet;
|
|
|
|
DBG_ASSERT( CheckSignature() );
|
|
|
|
//
|
|
// do a read to clear the wakeup signal
|
|
//
|
|
err = recvfrom(
|
|
GetSocket(), // our socket
|
|
(PCHAR) &dwBuff, // read buffer
|
|
sizeof(dwBuff), // buffer len
|
|
0, // flags
|
|
NULL, // src addr
|
|
NULL // src addr len
|
|
);
|
|
|
|
if ( err == SOCKET_ERROR ) {
|
|
|
|
DBGPRINTF((DBG_CONTEXT,
|
|
"Error %d in recvfrom\n", WSAGetLastError()));
|
|
|
|
return FALSE;
|
|
|
|
} else {
|
|
DBG_ASSERT(dwBuff == ATQ_BMON_WAKEUP_MESSAGE);
|
|
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
ATQ_BMON_SET::ATQ_BMON_SET(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Constructor sets up critsec and sets.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
--*/
|
|
{
|
|
INITIALIZE_CRITICAL_SECTION( &m_csLock );
|
|
|
|
memset(m_apEntrySet, 0, sizeof(m_apEntrySet));
|
|
FD_ZERO( &m_ListenSet );
|
|
|
|
m_SetSize = 0;
|
|
m_pWakeupEntry = NULL;
|
|
m_fCleanup = FALSE;
|
|
m_fDoSleep = FALSE;
|
|
m_fThreadFinished = FALSE;
|
|
m_dwError = NO_ERROR;
|
|
}
|
|
|
|
|
|
ATQ_BMON_SET::~ATQ_BMON_SET(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Destructor cleans up critsec.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
--*/
|
|
{
|
|
DBG_ASSERT( m_SetSize == 0 );
|
|
|
|
//
|
|
// get rid of the wakeup entry
|
|
//
|
|
DBG_REQUIRE( 0 == closesocket(m_pWakeupEntry->GetSocket()) );
|
|
delete m_pWakeupEntry;
|
|
|
|
DeleteCriticalSection(&m_csLock);
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::Initialize(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Sets up the Wakeup entry and starts
|
|
our thread.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
BOOL bRetval;
|
|
SOCKET s;
|
|
SOCKADDR_IN sockAddr;
|
|
INT err;
|
|
BMON_WAKEUP_ENTRY * pWakeup;
|
|
HANDLE hThread;
|
|
DWORD dwError = NO_ERROR;
|
|
WORD wPort = ATQ_BMON_WAKEUP_PORT;
|
|
|
|
bRetval = FALSE;
|
|
|
|
//
|
|
// set up wakeup entry
|
|
//
|
|
s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
|
|
if (s == INVALID_SOCKET) {
|
|
dwError = WSAGetLastError();
|
|
goto exit;
|
|
}
|
|
|
|
ZeroMemory(&sockAddr, sizeof(sockAddr));
|
|
sockAddr.sin_family = AF_INET;
|
|
sockAddr.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
sockAddr.sin_port = htons(wPort);
|
|
|
|
err = bind(s, (PSOCKADDR)&sockAddr, sizeof(sockAddr));
|
|
|
|
while ((err == SOCKET_ERROR)
|
|
&& (WSAGetLastError() == WSAEADDRINUSE)
|
|
&& (wPort < ATQ_BMON_WAKEUP_PORT_MAX)) {
|
|
//
|
|
// try another port
|
|
//
|
|
wPort++;
|
|
sockAddr.sin_port = htons(wPort);
|
|
|
|
err = bind(s, (PSOCKADDR)&sockAddr, sizeof(sockAddr));
|
|
}
|
|
|
|
|
|
if ( err == SOCKET_ERROR ) {
|
|
DBGERROR((DBG_CONTEXT,"Error %d in bind\n", WSAGetLastError()));
|
|
dwError = WSAGetLastError();
|
|
closesocket(s);
|
|
goto exit;
|
|
}
|
|
|
|
pWakeup = new BMON_WAKEUP_ENTRY(s);
|
|
if ( pWakeup == NULL )
|
|
{
|
|
dwError = ERROR_NOT_ENOUGH_MEMORY;
|
|
goto exit;
|
|
}
|
|
|
|
if (!pWakeup->InitEvent()) {
|
|
delete pWakeup;
|
|
pWakeup = NULL;
|
|
goto exit;
|
|
}
|
|
|
|
pWakeup->SetContainingBmonSet(this);
|
|
pWakeup->SetOpcode(BMON_WAIT);
|
|
|
|
m_pWakeupEntry = pWakeup;
|
|
m_Port = wPort;
|
|
|
|
//
|
|
// just jam this entry into the lists
|
|
//
|
|
DBG_ASSERT(m_SetSize == 0);
|
|
|
|
m_SetSize = 1;
|
|
m_apEntrySet[0] = pWakeup;
|
|
FD_SET(pWakeup->GetSocket(), &m_ListenSet);
|
|
|
|
//
|
|
// now set up our thread. from now on you
|
|
// have to Lock before manipulating the
|
|
// lists
|
|
//
|
|
m_hThread = CreateThread(
|
|
NULL, // defualt security
|
|
0, // default stack
|
|
::BmonThreadFunc, // thread func
|
|
this, // func parameter
|
|
0, // flags
|
|
NULL // discard tid
|
|
);
|
|
|
|
if (m_hThread) {
|
|
//
|
|
// it worked!
|
|
//
|
|
bRetval = TRUE;
|
|
|
|
} else {
|
|
//
|
|
// doh! clean up the wakeup entry
|
|
//
|
|
dwError = GetLastError();
|
|
|
|
closesocket( pWakeup->GetSocket() );
|
|
delete pWakeup;
|
|
|
|
m_SetSize = 0;
|
|
}
|
|
|
|
exit:
|
|
if (dwError != NO_ERROR) {
|
|
SetLastError(dwError);
|
|
}
|
|
|
|
return bRetval;
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::Cleanup(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Tells the select thread to clean up by
|
|
removing the wakeup entry.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
//
|
|
// Cleanup is done when wakup entry
|
|
// is removed.
|
|
//
|
|
// Don't use RemoveEntry, because the
|
|
// other thread will delete the wakeup
|
|
// entry.
|
|
//
|
|
DBG_ASSERT( m_pWakeupEntry && m_pWakeupEntry->CheckSignature() );
|
|
|
|
m_pWakeupEntry->SetOpcode(BMON_REMOVE);
|
|
Wakeup();
|
|
|
|
//
|
|
// Wait a reasonable amount of time for the thread to go away.
|
|
//
|
|
|
|
WaitForSingleObject( m_hThread, 10000 );
|
|
CloseHandle( m_hThread );
|
|
m_hThread = NULL;
|
|
|
|
delete this;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::IsEmpty(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This tells you if there are sockets in the
|
|
set. Note that one socket is the wakeup
|
|
socket, so a count of one means we're empty.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
TRUE on empty
|
|
--*/
|
|
{
|
|
BOOL bRetval;
|
|
|
|
Lock();
|
|
bRetval = (m_SetSize <= 1);
|
|
DBG_ASSERT( m_SetSize <= FD_SETSIZE );
|
|
Unlock();
|
|
|
|
return bRetval;
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::IsNotFull(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This tells you if there is room for more sockets
|
|
in the set. Note that one socket is the wakeup
|
|
socket.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
TRUE when space is available
|
|
--*/
|
|
{
|
|
BOOL bRetval;
|
|
|
|
Lock();
|
|
bRetval = (m_SetSize < FD_SETSIZE);
|
|
DBG_ASSERT( m_SetSize <= FD_SETSIZE );
|
|
Unlock();
|
|
|
|
return bRetval;
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::AddEntry(
|
|
ATQ_BMON_ENTRY * pBmonEntry
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Add an entry to the set
|
|
|
|
Arguments:
|
|
|
|
pBmonEntry - the entry to be added
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
DWORD i;
|
|
BOOL bAdded = FALSE;
|
|
DWORD dwError;
|
|
|
|
pBmonEntry->SetOpcode(BMON_ADD);
|
|
|
|
Lock();
|
|
|
|
for (i = 0; i < FD_SETSIZE; i++) {
|
|
if (!m_apEntrySet[i]) {
|
|
m_apEntrySet[i] = pBmonEntry;
|
|
pBmonEntry->SetContainingBmonSet(this);
|
|
bAdded = TRUE;
|
|
|
|
m_SetSize++;
|
|
DBG_ASSERT( m_SetSize <= FD_SETSIZE );
|
|
break;
|
|
}
|
|
}
|
|
|
|
Unlock();
|
|
|
|
if (bAdded) {
|
|
Wakeup();
|
|
pBmonEntry->WaitForAddRemove();
|
|
|
|
dwError = pBmonEntry->GetError();
|
|
|
|
if (dwError) {
|
|
//
|
|
// other thread will remove from list
|
|
//
|
|
SetLastError(dwError);
|
|
bAdded = FALSE;
|
|
}
|
|
}
|
|
|
|
return bAdded;
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::RemoveEntry(
|
|
ATQ_BMON_ENTRY * pBmonEntry
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Remove an entry from the set
|
|
|
|
Arguments:
|
|
|
|
pBmonEntry - the entry to be removed
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
DWORD i;
|
|
BOOL bRemoved = FALSE;
|
|
|
|
pBmonEntry->SetOpcode(BMON_REMOVE);
|
|
Wakeup();
|
|
|
|
pBmonEntry->WaitForAddRemove();
|
|
|
|
return (pBmonEntry->GetError() == NO_ERROR);
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::PauseEntry(
|
|
ATQ_BMON_ENTRY * pBmonEntry
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Pause an entry in the set. The entry's
|
|
socket will be removed from the FD_SET,
|
|
but the entry will stay.
|
|
|
|
Arguments:
|
|
|
|
pBmonEntry - the entry to be paused
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
DWORD i;
|
|
BOOL bRemoved = FALSE;
|
|
|
|
DBGPRINTF((DBG_CONTEXT,
|
|
"Pausing backlog monitor entry %p\n",
|
|
pBmonEntry));
|
|
|
|
pBmonEntry->SetOpcode(BMON_PAUSE);
|
|
Wakeup();
|
|
|
|
//
|
|
// We don't do the event stuff for pause and
|
|
// resume. It's a pain because the client will
|
|
// want to pause from within the callback
|
|
// function.
|
|
//
|
|
return (TRUE);
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::ResumeEntry(
|
|
ATQ_BMON_ENTRY * pBmonEntry
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Resume an entry in the set. The entry's
|
|
socket will be added back to the FD_SET.
|
|
|
|
Arguments:
|
|
|
|
pBmonEntry - the entry to be resumed
|
|
|
|
Return Values:
|
|
|
|
TRUE on success
|
|
--*/
|
|
{
|
|
DWORD i;
|
|
BOOL bRemoved = FALSE;
|
|
|
|
|
|
if (pBmonEntry->GetOpcode() == BMON_NOWAIT) {
|
|
DBGPRINTF((DBG_CONTEXT,
|
|
"Resuming backlog monitor entry %p\n",
|
|
pBmonEntry));
|
|
|
|
pBmonEntry->SetOpcode(BMON_RESUME);
|
|
Wakeup();
|
|
}
|
|
|
|
//
|
|
// We don't do the event stuff for pause and
|
|
// resume. It's a pain because the client will
|
|
// want to pause from within the callback
|
|
// function.
|
|
//
|
|
return (TRUE);
|
|
}
|
|
|
|
|
|
|
|
VOID
|
|
ATQ_BMON_SET::BmonThreadFunc(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function is for the set's select thread.
|
|
It calls accept with the listen set, and calls
|
|
notification functions for all sockets that
|
|
are ready.
|
|
|
|
SynchronizeSets returns false when it's time
|
|
to shut down.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
None
|
|
--*/
|
|
{
|
|
INT err;
|
|
|
|
while (SynchronizeSets()) {
|
|
|
|
err = select(
|
|
0, // junk
|
|
&m_ListenSet, // readfds
|
|
NULL, // writefds
|
|
NULL, // exceptfds
|
|
NULL // want to block
|
|
);
|
|
|
|
if (err != SOCKET_ERROR) {
|
|
DBG_ASSERT(err > 0);
|
|
|
|
if ( !NotifyEntries() )
|
|
{
|
|
//
|
|
// If we couldn't notify the entries, stop doing our thing
|
|
//
|
|
|
|
m_dwError = GetLastError();
|
|
m_dwError |= BMON_NOTIFY_ERROR;
|
|
break;
|
|
}
|
|
} else {
|
|
//
|
|
// Actually let's take the general approach that the moment the
|
|
// Backlog monitor sees trouble, it should stop. This avoids
|
|
// low memory situations where the backlog monitor just spins,
|
|
// thus becoming it's own denial of service attack.
|
|
//
|
|
|
|
m_dwError = WSAGetLastError();
|
|
m_dwError |= BMON_SELECT_ERROR;
|
|
break;
|
|
|
|
DBGPRINTF(( DBG_CONTEXT,
|
|
"Select failed with error %d\n",
|
|
WSAGetLastError()
|
|
));
|
|
|
|
|
|
}
|
|
|
|
if ( m_fDoSleep )
|
|
{
|
|
//
|
|
// Now sleep for a while. It will take time for unconnected to go away.
|
|
// We don't want to spin
|
|
//
|
|
// Of course, this means we don't do useful work on any other endpoints
|
|
// which may also be in trouble. Oh well.
|
|
//
|
|
|
|
Sleep( 5000 );
|
|
m_fDoSleep = FALSE;
|
|
}
|
|
}
|
|
|
|
m_fThreadFinished = TRUE;
|
|
}
|
|
|
|
|
|
VOID
|
|
ATQ_BMON_SET::Wakeup(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
We call this function when adding or removing
|
|
an entry. Writing to the wakeup socket wakes
|
|
up the select thread.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
None
|
|
--*/
|
|
{
|
|
SOCKADDR_IN sockAddr;
|
|
INT err;
|
|
DWORD dwBuf;
|
|
|
|
sockAddr.sin_family = AF_INET;
|
|
sockAddr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
|
|
sockAddr.sin_port = htons(m_Port);
|
|
|
|
dwBuf = ATQ_BMON_WAKEUP_MESSAGE;
|
|
|
|
err = sendto(
|
|
m_pWakeupEntry->GetSocket(),
|
|
(PCHAR)&dwBuf,
|
|
sizeof(dwBuf),
|
|
0,
|
|
(PSOCKADDR)&sockAddr,
|
|
sizeof(sockAddr)
|
|
);
|
|
|
|
if ( err == SOCKET_ERROR ) {
|
|
|
|
m_dwError = WSAGetLastError();
|
|
m_dwError |= BMON_SENDTO_ERROR;
|
|
|
|
DBGPRINTF((DBG_CONTEXT,
|
|
"Error %d in sendto\n",WSAGetLastError()));
|
|
}
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::SynchronizeSets(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function updates our FD_SET to reflect
|
|
what's in the entry set, and also notifies
|
|
entries which have been added or removed.
|
|
|
|
We also check to see if it's time to shut
|
|
down (as indicated by the removal of the
|
|
wakeup entry.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
TRUE to continue operating
|
|
FALSE on shutdown
|
|
--*/
|
|
{
|
|
DWORD i;
|
|
ATQ_BMON_ENTRY * pBmonEntry;
|
|
BOOL bRetval;
|
|
|
|
bRetval = TRUE;
|
|
|
|
Lock();
|
|
|
|
//
|
|
// clear out whatever is there now
|
|
//
|
|
FD_ZERO(&m_ListenSet);
|
|
|
|
//
|
|
// put in everything we want
|
|
//
|
|
for (i = 0; i < FD_SETSIZE; i++) {
|
|
pBmonEntry = m_apEntrySet[i];
|
|
|
|
if (pBmonEntry) {
|
|
DBG_ASSERT( pBmonEntry->CheckSignature() );
|
|
|
|
switch(pBmonEntry->GetOpcode()) {
|
|
case BMON_ADD:
|
|
FD_SET(pBmonEntry->GetSocket(), &m_ListenSet);
|
|
pBmonEntry->SetOpcode(BMON_WAIT);
|
|
pBmonEntry->SignalAddRemove(NO_ERROR);
|
|
break;
|
|
|
|
case BMON_RESUME:
|
|
FD_SET(pBmonEntry->GetSocket(), &m_ListenSet);
|
|
pBmonEntry->SetOpcode(BMON_WAIT);
|
|
break;
|
|
|
|
case BMON_PAUSE:
|
|
pBmonEntry->SetOpcode(BMON_NOWAIT);
|
|
pBmonEntry->SignalAddRemove(NO_ERROR);
|
|
break;
|
|
|
|
case BMON_REMOVE:
|
|
if (pBmonEntry == m_pWakeupEntry) {
|
|
//
|
|
// this means it's time to shut down
|
|
//
|
|
bRetval = FALSE;
|
|
}
|
|
|
|
m_apEntrySet[i] = NULL;
|
|
m_SetSize--;
|
|
pBmonEntry->SetContainingBmonSet(NULL);
|
|
pBmonEntry->SignalAddRemove(NO_ERROR);
|
|
break;
|
|
|
|
case BMON_WAIT:
|
|
FD_SET(pBmonEntry->GetSocket(), &m_ListenSet);
|
|
break;
|
|
|
|
case BMON_NOWAIT:
|
|
//
|
|
// this entry is paused, so don't do
|
|
// anything
|
|
//
|
|
break;
|
|
|
|
|
|
default:
|
|
//
|
|
// should never get here
|
|
// remove the bad entry
|
|
//
|
|
DBGPRINTF((DBG_CONTEXT,
|
|
"Invalid opcode in ATQ_BMON_ENTRY %p, %d\n",
|
|
pBmonEntry, pBmonEntry->GetOpcode()));
|
|
|
|
DBG_ASSERT(FALSE);
|
|
m_apEntrySet[i] = NULL;
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Unlock();
|
|
|
|
return bRetval;
|
|
}
|
|
|
|
|
|
BOOL
|
|
ATQ_BMON_SET::NotifyEntries(
|
|
VOID
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function looks through the entries
|
|
to see who needs to be notified and calls
|
|
their callback function.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Values:
|
|
|
|
TRUE if successful, else FALSE
|
|
--*/
|
|
{
|
|
DWORD i;
|
|
ATQ_BMON_ENTRY * pBmonEntry;
|
|
BOOL fRet = TRUE;
|
|
|
|
Lock();
|
|
|
|
for (i = 0; i < FD_SETSIZE; i++) {
|
|
pBmonEntry = m_apEntrySet[i];
|
|
|
|
if (pBmonEntry) {
|
|
if (!pBmonEntry->CheckSignature()) {
|
|
DBGPRINTF(( DBG_CONTEXT,
|
|
"ATQ_BMON_ENTRY(%p)::CheckSignature() failed. index = %d\n",
|
|
pBmonEntry, i));
|
|
|
|
DBG_ASSERT( pBmonEntry->CheckSignature() );
|
|
}
|
|
|
|
if ((pBmonEntry->GetOpcode() == BMON_WAIT)
|
|
&& (FD_ISSET(pBmonEntry->GetSocket(), &m_ListenSet))) {
|
|
|
|
if ( !pBmonEntry->Callback() )
|
|
{
|
|
fRet = FALSE;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
Unlock();
|
|
|
|
return fRet;
|
|
}
|
|
|
|
|
|
DWORD WINAPI
|
|
BmonThreadFunc(
|
|
LPVOID lpBmonSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function starts the select thread
|
|
of an ATQ_BMON_SET.
|
|
|
|
Arguments:
|
|
|
|
pBmonSet - the set to call
|
|
|
|
Return Values:
|
|
|
|
0
|
|
--*/
|
|
{
|
|
ATQ_BMON_SET * pBmonSet = (ATQ_BMON_SET *) lpBmonSet;
|
|
pBmonSet->BmonThreadFunc();
|
|
return 0;
|
|
}
|
|
|
|
|
|
//
|
|
// atqbmon.cxx
|
|
//
|
|
|