windows-server-2003/inetsrv/iis/iisrearc/iisplus/sfwp/sslconfigprov/inc/sslconfigpipe.hxx

#ifndef __SSLCONFIGPIPE__HXX_
#define __SSLCONFIGPIPE__HXX_
/*++

   Copyright    (c)    2001    Microsoft Corporation

   Module  Name :
     sslconfigpipe.cxx

   Abstract:
     SSL CONFIG PIPE implementation

     simple blocking pipe implementation
     that enables 
     - sending/receiving commands, 
     - sending/receiving response headers
     - sending/receiving arbitrary data blocks
     - implementing pipe listener that runs on dedicated thread
     - safe cleanup for thread running pipe listener

     CODEWORK: this class is actually general purpose
     It doesn't contain any specifics of ssl config
     It should eventually be named more appropriately 
     and made available for general use
 
   Author:
     Jaroslav Dunajsky      April-24-2001

   Environment:
     Win32 - User Mode

   Project:
     Stream Filter Worker Process
--*/


//
// Command structure to be sent over the named pipe
//

struct SSL_CONFIG_PIPE_COMMAND
{
    DWORD     dwCommandId;
    DWORD     dwParameter1;
    DWORD     dwParameter2;
};

//
// Each response to SSL_CONFIG_PIPE_COMMAND starts with
// SSL_CONFIG_PIPE_RESPONSE_HEADER
// if dwResponseSize is 0, it means that data will be terminated in a
// way specific to sent commmand
//

struct SSL_CONFIG_PIPE_RESPONSE_HEADER
{
    DWORD       dwCommandId;
    DWORD       dwParameter1;
    DWORD       dwParameter2;
    HRESULT     hrErrorCode;
    DWORD       dwResponseSize;
};


class SSL_CONFIG_PIPE
{
public:

    SSL_CONFIG_PIPE()
        :
        _hSslConfigPipe( INVALID_HANDLE_VALUE ),
        _hPipeListenerThread( NULL ),
        _dwCurrThreadId( 0 ),
        _dwLockRecursionLevel( 0 ),
        _CancelFlag( 0 ),
        _InitStatus( INIT_NONE )
        
    {
    }

    ~SSL_CONFIG_PIPE()
    {
    }
    
    HRESULT
    PipeInitializeServer(
        IN const WCHAR * wszPipeName
        )
    {
        return PipeInitialize( wszPipeName, 
                               TRUE /*server*/ );
    }
    
    HRESULT
    PipeInitializeClient(
        IN const WCHAR * wszPipeName
        )
    {
        return PipeInitialize( wszPipeName, 
                               FALSE /*client*/ );
    }
    
    HRESULT
    PipeTerminate(
        VOID
        );
    
    HRESULT
    PipeConnect(
        VOID
        );

    HRESULT
    PipeDisconnect(
        VOID
        );

    VOID
    PipeLock(
        VOID
        );
        
    VOID
    PipeUnlock(
        VOID
        );

    BOOL
    IsPipeLocked(
        VOID
        )
    {
        return _dwCurrThreadId == GetCurrentThreadId();
    }

    BOOL
    IsPipeInitialized(
        VOID
        )
    {
        return _InitStatus == INIT_PIPE_CONNECTED;
    }
     
    HRESULT
    PipeSendData(
        IN DWORD  cbNumberOfBytesToWrite,
        IN BYTE * pbBuffer
        );
    
    HRESULT
    PipeReceiveData(
        IN  DWORD  cbBytesToRead,
        OUT BYTE * pbBuffer
        );
    
    HRESULT
    PipeSendCommand(
        IN  SSL_CONFIG_PIPE_COMMAND * pCommand
        );         
    
    HRESULT
    PipeReceiveCommand(
        OUT SSL_CONFIG_PIPE_COMMAND * pCommand
        );

    HRESULT
    PipeSendResponseHeader(
        IN SSL_CONFIG_PIPE_RESPONSE_HEADER * ResponseHeader
        );
    
    
    HRESULT
    PipeReceiveResponseHeader(
        OUT SSL_CONFIG_PIPE_RESPONSE_HEADER * ResponseHeader
        );

    BOOL
    QueryPipeIsCancelled(
        VOID
        )
    {
        return  !!_CancelFlag;
    }

    VOID
    PipeCancel(
        VOID
        );
    
protected:
    
    virtual
    HRESULT
    PipeListener(
        VOID 
        )
    /*++

    Routine Description:
        Inheriting class may implement function that will be executed
        on dedicated thread. This class will handle proper synchronization 
        for cleanup of the thread
    
        Note: when implementing new PipeListener, do not forget to enable it
        by implementing QueryEnablePipeListener() that returns TRUE
    Arguments:

    Return Value:

        HRESULT

    --*/    
    {
        DBG_ASSERT( FALSE );
        return HRESULT_FROM_WIN32( ERROR_NOT_SUPPORTED );
    };

    virtual
    BOOL
    QueryEnablePipeListener(
        VOID
        )
    {
        return FALSE;
    }
private:

    HRESULT
    PipeInitialize(
        IN const WCHAR * wszPipeName,
        IN BOOL          fServer
        );

    HRESULT
    PipeConnectServer(
        VOID
        );

    HRESULT
    PipeDisconnectServer(
        VOID
        );

    HRESULT
    PipeConnectClient(
        VOID
        );

    HRESULT
    PipeDisconnectClient(
        VOID
        );

    HRESULT
    PipeStartListenerThread(
        VOID
        );
    
    static 
    DWORD
    PipeListenerThread(
        LPVOID ThreadParam
        );
    
    HRESULT
    PipeWaitForCompletion(
        IN  HRESULT    hrLastError,
        IN OVERLAPPED* pOverlapped,
        OUT DWORD *    pcbTransferred
        );

    HRESULT
    PipeCleanup(
        VOID
        )
    {
        if( CancelIo( _hSslConfigPipe ) )
        {
            return S_OK;
        }

        return E_FAIL;
    }

    enum INIT_STATUS 
    {
        INIT_NONE,
        INIT_PIPE_LOCK_CREATED,
        INIT_OVERLAPPED_R_CREATED,
        INIT_OVERLAPPED_W_CREATED,
        INIT_CANCEL_EVENT_CREATED,
        INIT_SERVER_END_PIPE_CREATED,
        INIT_PIPE_CONNECTED
    };
    INIT_STATUS         _InitStatus;
        
    CRITICAL_SECTION    _csPipeLock;
    DWORD               _dwLockRecursionLevel;
    DWORD               _dwCurrThreadId;
    HANDLE              _hSslConfigPipe;

    // reads and write may be happening concurrenly
    // that's why we need 2 overlapped structures
    
    // overlapped for Read
    OVERLAPPED          _OverlappedR;
    // overlapped for Write
    OVERLAPPED          _OverlappedW;

    HANDLE              _hPipeListenerThread;
    LONG                _CancelFlag;
    STRU                _strPipeName;
    // flag indicating Server side pipe
    BOOL                _fServer;
    HANDLE              _hCancelEvent;
    
    
};
   
#endif