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.
198 lines
9.6 KiB
198 lines
9.6 KiB
/*======================================================================================//
|
|
| //
|
|
|Copyright (c) 1998, 1999 Sequent Computer Systems, Incorporated //
|
|
| //
|
|
|Description: //
|
|
| //
|
|
|---------------------------------------------------------------------------------------//
|
|
| This file implements the CProcConUser class methods defined in ProcConSvc.h //
|
|
|---------------------------------------------------------------------------------------//
|
|
| //
|
|
|Created: //
|
|
| //
|
|
| Jarl McDonald 07-98 //
|
|
| //
|
|
|Revision History: //
|
|
| //
|
|
|=======================================================================================*/
|
|
#include "ProcConSvc.h"
|
|
|
|
const TCHAR *CProcConUser::PIPENAME = TEXT("\\\\.\\pipe\\ProcConPip");
|
|
|
|
// Constructor
|
|
// Note: this function runs as part of service start so keep it quick!
|
|
CProcConUser::CProcConUser( PCContext *ctxt ) :
|
|
m_cPC( *ctxt->cPC ), m_cDB( *ctxt->cDB ),
|
|
m_inBufChars( 4096 ), m_outBufChars( 65536 ),
|
|
m_clientTimeout( 5000 ), m_clientCount( 0 )
|
|
{
|
|
PCBuildAdminSecAttr( m_secAttr );
|
|
|
|
m_hConnEvent = CreateEvent( NULL, TRUE, FALSE, NULL );
|
|
if ( !m_hConnEvent )
|
|
PCLogUnExError( TEXT("PCClientConn"), TEXT("CreateEvent") );
|
|
|
|
m_olConn.hEvent = m_hConnEvent;
|
|
}
|
|
|
|
// Destructor
|
|
CProcConUser::~CProcConUser( void )
|
|
{
|
|
PCFreeSecAttr( m_secAttr );
|
|
if ( m_hConnEvent ) CloseHandle( m_hConnEvent );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------//
|
|
// Function to determine if all CProcConUser initial conditions have been met //
|
|
// Input: None //
|
|
// Returns: TRUE if ready, FALSE if not //
|
|
//--------------------------------------------------------------------------------------------//
|
|
BOOL CProcConUser::ReadyToRun( void )
|
|
{
|
|
return m_hConnEvent != NULL && m_secAttr.lpSecurityDescriptor && !m_cPC.GotShutdown();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------//
|
|
// Function to set a new timeout //
|
|
// Input: proposed new timeout //
|
|
// Returns: NT or PC error code //
|
|
// Note: lower and upper limits on timeout are a bit arbitrary (.1 secs to 30 secs OK) //
|
|
// This timeout only applies to the pipe connection timeout at clients, not to //
|
|
// transaction timeouts (which are handled by the client alone). //
|
|
//--------------------------------------------------------------------------------------------//
|
|
PCULONG32 CProcConUser::SetTimeout( PCULONG32 newTimeout )
|
|
{
|
|
if ( newTimeout >= PC_MIN_TIMEOUT && newTimeout <= PC_MAX_TIMEOUT ) {
|
|
m_clientTimeout = newTimeout;
|
|
return ERROR_SUCCESS;
|
|
}
|
|
else
|
|
return PCERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------//
|
|
// CProcConUser thread function -- this function runs in its own thread and simply offers the //
|
|
// PC named pipe to the world. Each pipe connection causes a client thread with //
|
|
// its own context to be launched and then a new pipe 'port' is created. //
|
|
// Input: None //
|
|
// Returns: 0 //
|
|
//--------------------------------------------------------------------------------------------//
|
|
PCULONG32 CProcConUser::Run( void )
|
|
{
|
|
|
|
// Wait for DB initialization to complete before establishing user service environment.
|
|
// If this isn't the first call here, the delay serves to provide some pacing.
|
|
WaitForSingleObject( m_cDB.GetDbEvent(), 1000 );
|
|
|
|
HANDLE waitList[] = { m_olConn.hEvent, m_cPC.GetShutEvent() };
|
|
HANDLE hThread = NULL;
|
|
ClientContext *context = NULL;
|
|
|
|
// User connect thread main loop -- handles all user connections to ProcCon.
|
|
// There is only one user connect thread.
|
|
// This loop runs until shutdown is signalled or until a bad NT error occurs.
|
|
for ( ; !m_cPC.GotShutdown(); ++m_clientCount ) {
|
|
|
|
// Create a new client context..
|
|
context = new ClientContext( m_clientCount, &m_cPC, &m_cDB, this, m_inBufChars, m_outBufChars );
|
|
if ( !context ) {
|
|
PCLogNoMemory( TEXT("ClientContext"), sizeof(ClientContext) );
|
|
break;
|
|
}
|
|
|
|
// Create a pipe instance...
|
|
context->hPipe = CreateNamedPipe( PIPENAME,
|
|
PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED, // read/write access, overlapped enabled
|
|
PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE | PIPE_WAIT, // message-based, blocking
|
|
PIPE_UNLIMITED_INSTANCES, // no limit
|
|
m_outBufChars, // output buffer size
|
|
m_inBufChars, // input buffer size
|
|
m_clientTimeout, // client side time-out
|
|
&m_secAttr ); // our security attr -- built in constructor
|
|
if ( context->hPipe == INVALID_HANDLE_VALUE ) {
|
|
PCLogUnExError( TEXT("PCClientConn"), TEXT("CreatePipe") );
|
|
context->hPipe = NULL;
|
|
break;
|
|
}
|
|
|
|
// Initiate pipe connect (but don't wait for a connection)...
|
|
DWORD connError = 0;
|
|
BOOL launchClient = FALSE;
|
|
ResetEvent( m_olConn.hEvent );
|
|
if ( !ConnectNamedPipe( context->hPipe, &m_olConn ) )
|
|
connError = GetLastError();
|
|
|
|
// If we have a suspended client from our last connect, release it before waiting on a new connection...
|
|
if ( hThread ) {
|
|
if ( ResumeThread( hThread ) == 0xffffffff )
|
|
PCLogUnExError( TEXT("PCClientConn"), TEXT("ResumeThread1") );
|
|
CloseHandle( hThread );
|
|
hThread = NULL;
|
|
}
|
|
|
|
// Analyze result of connect. If necessary, wait for a client or a shutdown request...
|
|
if ( !connError || connError == ERROR_PIPE_CONNECTED )
|
|
launchClient = TRUE;
|
|
else if ( connError == ERROR_IO_PENDING ) {
|
|
PCULONG32 rc = WaitForMultipleObjects( ENTRY_COUNT(waitList), waitList, FALSE, INFINITE );
|
|
|
|
// If we got a client, create a thread for it...
|
|
if ( rc - WAIT_OBJECT_0 == 0 ) {
|
|
PCULONG32 bytes;
|
|
if ( !GetOverlappedResult( context->hPipe, &m_olConn, &bytes, TRUE ) ) {
|
|
PCLogUnExError( TEXT("PCClientConn"), TEXT("ConnectPipeResult") );
|
|
break;
|
|
}
|
|
else launchClient = TRUE;
|
|
}
|
|
// If we got a shutdown request, just break out...
|
|
else if ( rc - WAIT_OBJECT_0 == 1 )
|
|
break;
|
|
else {
|
|
PCLogUnExError( TEXT("PCClientConn"), TEXT("WaitOnPipeOrShutdown") );
|
|
break;
|
|
}
|
|
}
|
|
else {
|
|
PCLogUnExError( TEXT("PCClientConn"), TEXT("ConnectPipe") );
|
|
break;
|
|
}
|
|
|
|
// If we have a good connection, start a suspended client thread.
|
|
// The thread will be released after we have a new pipe instance ready to go.
|
|
// This minimizes the interval during which a pipe instance is not available.
|
|
if ( launchClient ) {
|
|
hThread = CreateThread( NULL, 0, PCClientThread, context, CREATE_SUSPENDED, NULL );
|
|
if ( !hThread ) {
|
|
PCLogUnExError( TEXT("PCClientConn"), TEXT("CreateThread") );
|
|
break;
|
|
}
|
|
context = NULL; // Client thread deletes its context, we're done with it
|
|
}
|
|
|
|
}
|
|
|
|
// Clean up if we have a context that has not been passed to a client thread...
|
|
if ( context ) {
|
|
if ( context->hPipe ) {
|
|
CancelIo( context->hPipe );
|
|
CloseHandle( context->hPipe );
|
|
}
|
|
delete context;
|
|
}
|
|
|
|
// If we have a suspended client, release it before leaving...
|
|
if ( hThread ) {
|
|
if ( ResumeThread( hThread ) == 0xffffffff )
|
|
PCLogUnExError( TEXT("PCClientConn"), TEXT("ResumeThread2") );
|
|
CloseHandle( hThread );
|
|
}
|
|
|
|
// Note: if this return is not due to a shutdwon request, this function will be called again.
|
|
// This 'pipe restart' is the last line of recovery in case of serious pipe (or other) errors.
|
|
return 0;
|
|
}
|
|
|
|
// End of CProcConUser.cpp
|
|
//============================================================================J McDonald fecit====//
|
|
|