/*++

Copyright (C) 1996-2001 Microsoft Corporation

Module Name:

Abstract:

History:

--*/


#ifndef __MSMQRECV_H__
#define __MSMQRECV_H__

#include <unk.h>
#include <wstring.h>
#include <sync.h>
#include <wmimsg.h>
#include <comutl.h>
#include "msmqhdlr.h"
#include "msmqrecv.h"
#include "msmqcomn.h"

/**************************************************************************
  CMsgMsmqReceiver
***************************************************************************/

class CMsgMsmqReceiver 
: public CUnkBase<IWmiMessageReceiver,&IID_IWmiMessageReceiver>
{
    CCritSec m_cs;
    CMsmqApi m_Api;
    DWORD m_dwFlags;
    WString m_wsEndpoint;
    QUEUEHANDLE m_hQueue;
    PVOID m_pSvcId;
    CWbemPtr<IWmiMessageService> m_pSvc;
    CWbemPtr<IWmiMessageSendReceive> m_pRecv;
    CWbemPtr<CMsgMsmqHandler> m_pHndlr;

    HRESULT EnsureReceiver();
    void* GetInterface( REFIID riid );
        
public:

    CMsgMsmqReceiver( CLifeControl* pCtl );
    ~CMsgMsmqReceiver();

    //
    // called by Rcv Sinks on error. 
    //
    HRESULT HandleError( HRESULT hr );
    HRESULT HandleReceiveError( HRESULT hr );

    STDMETHOD(Close)(); 

    STDMETHOD(Open)( LPCWSTR wszEndpoint,
                     DWORD dwFlags,
                     WMIMSG_RCVR_AUTH_INFOP pAuthInfo,
                     IWmiMessageSendReceive* pRcv );
};

/*************************************************************************
  CMsgSimpleRcvSink - Receive operation is not safe.  If something bad 
  happens after receiving the message but before delivering to the handler, 
  the message will be lost.  Supports Overlapped I/O.
**************************************************************************/

class CMsgSimpleRcvSink
: public CUnkBase<IWmiMessageReceiverSink, &IID_IWmiMessageReceiverSink>
{
    //
    // does not hold ref counts on these because of circular ref.
    //
    CMsgMsmqHandler* m_pHndlr;
    CMsgMsmqReceiver* m_pRcvr;

    CMsgSimpleRcvSink( CLifeControl* pCtl,
                       CMsgMsmqHandler* pHndlr,
                       CMsgMsmqReceiver* pRcvr )
    : CUnkBase<IWmiMessageReceiverSink, &IID_IWmiMessageReceiverSink>( pCtl ),
      m_pHndlr( pHndlr ), m_pRcvr( pRcvr ) {}

public: 

    STDMETHOD(Receive)( PVOID pOverlapped );
    STDMETHOD(Notify)( PVOID pOverlapped );

    static HRESULT Create( CLifeControl* pControl,
                           CMsgMsmqHandler* pHndlr,
                           CMsgMsmqReceiver* pRcvr,
                           IWmiMessageReceiverSink** ppRvcrSink );
};

/*************************************************************************
  CMsgSafeRcvSink - safely reads messages off a queue.  It first peeks at
  messages, delivers to the handler on completion, and then removes the 
  message.  Supports Overlapped I/O.  This implementation does NOT support 
  multiple instances per queue.  
**************************************************************************/

class CMsgSafeRcvSink
: public CUnkBase<IWmiMessageReceiverSink, &IID_IWmiMessageReceiverSink>
{
    //
    // does not hold ref counts on these because of circular ref.
    //
    CMsgMsmqHandler* m_pHndlr;
    CMsgMsmqReceiver* m_pRcvr;

    CMsgSafeRcvSink( CLifeControl* pCtl,
                     CMsgMsmqHandler* pHndlr,
                     CMsgMsmqReceiver* pRcvr )

    : CUnkBase<IWmiMessageReceiverSink, &IID_IWmiMessageReceiverSink>( pCtl ),
      m_pHndlr( pHndlr ), m_pRcvr( pRcvr ) {}

public: 

    STDMETHOD(Receive)( PVOID pOverlapped );
    STDMETHOD(Notify)( PVOID pOverlapped );  

    static HRESULT Create( CLifeControl* pControl,
                           CMsgMsmqHandler* pHndlr,
                           CMsgMsmqReceiver* pRcvr,
                           IWmiMessageReceiverSink** ppRvcrSink );
};

#endif // __MSMQRECV_H__