windows-nt/Source/XPSP1/NT/com/ole32/olethunk/ole16/compobj/rpc16.cxx

380 lines
10 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
//+-------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1992 - 1994.
//
// File: rpc16.cxx
//
// Contents: Maps the 16-bit IRpcChannel implementation to a 16-bit
// implementation of IRpcChannelBuffer.
//
// This is required to support custom interface marshalling.
//
// History: 30-Mar-94 BobDay Adapted from OLELRPC.CPP
//
//--------------------------------------------------------------------------
#include <headers.cxx>
#pragma hdrstop
#include <ole2ver.h>
#include <ole2sp.h>
#include "rpc16.hxx"
interface IRpcChannel : public IUnknown
{
public:
virtual HRESULT STDMETHODCALLTYPE GetStream
(
REFIID riid,
int iMethod,
BOOL fSend,
BOOL fNoWait,
DWORD size,
IStream FAR * FAR *ppIStream
) = 0;
virtual HRESULT STDMETHODCALLTYPE Call
(
IStream FAR * pIStream
) = 0;
virtual HRESULT STDMETHODCALLTYPE GetDestCtx
(
DWORD *pdwDestContext,
void **ppvDestContext
) = 0;
virtual HRESULT STDMETHODCALLTYPE IsConnected
(
void
) = 0;
};
//
// 16-bit IRpcChannel interface, stream-based
//
// This is the interface seen by the 16-bit proxy implementations
//
class CRpcChannel : public IRpcChannel
{
public:
HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid,void **ppvObject);
ULONG STDMETHODCALLTYPE AddRef(void);
ULONG STDMETHODCALLTYPE Release(void);
STDMETHOD(GetStream)(REFIID iid, int iMethod, BOOL fSend,
BOOL fNoWait, DWORD size, IStream FAR* FAR* ppIStream);
STDMETHOD(Call)(IStream FAR* pIStream);
STDMETHOD(GetDestCtx)(DWORD FAR* lpdwDestCtx, LPVOID FAR* lplpvDestCtx);
STDMETHOD(IsConnected)(void);
CRpcChannel FAR *CRpcChannel::Create( CRpcChannelBuffer FAR *prcb );
private:
CRpcChannel::CRpcChannel( CRpcChannelBuffer FAR *prcb );
CRpcChannel::~CRpcChannel( void );
ULONG m_refs; // Reference count
CRpcChannelBuffer FAR * m_prcb; // ChannelBuffer to talk through
};
//+-------------------------------------------------------------------------
//
// Implementation of CRpcChannel
//
//--------------------------------------------------------------------------
//+---------------------------------------------------------------------------
//
// Function: Create, public
//
// Synopsis: Creates a 16-bit CRpcChannel and gives it the
// IRpcChannelBuffer that in needs to call.
//
// Arguments: [prcb] - IRpcChannelBuffer to call when called
//
// Returns: IRpcChannel *, Null on failure
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
CRpcChannel FAR *CRpcChannel::Create( CRpcChannelBuffer FAR *prcb )
{
CRpcChannel FAR *pRC;
pRC = new CRpcChannel( prcb );
return( pRC );
}
//+---------------------------------------------------------------------------
//
// Function: Constructor, private
//
// Synopsis: Creates a 16-bit CRpcChannel and gives it the
// IRpcChannelBuffer that in needs to call.
//
// Arguments: [prcb] - IRpcChannelBuffer to call when called
//
// Returns: IRpcChannel *, Null on failure
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
CRpcChannel::CRpcChannel( CRpcChannelBuffer FAR *prcb )
{
m_refs = 1;
m_prcb = prcb; // Save it and addref it
if ( m_prcb != NULL )
{
m_prcb->AddRef();
}
}
//+---------------------------------------------------------------------------
//
// Function: Destructor, private
//
// Synopsis: Destroys the CRpcChannel object
//
// Arguments: none
//
// Returns: nothing
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
CRpcChannel::~CRpcChannel( void )
{
if ( m_prcb != NULL )
{
m_prcb->Release(); // Release it and zero it
m_prcb = NULL;
}
}
//+---------------------------------------------------------------------------
//
// Function: QueryInterface, public
//
// Synopsis: Allows querying for other interfaces supported by this
// object
//
// Arguments: [iidInterface] - iid to query for
// [ppv] - interface pointer returned
//
// Returns: nothing
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
STDMETHODIMP CRpcChannel::QueryInterface(
REFIID iidInterface,
void FAR* FAR* ppvObj )
{
HRESULT hresult;
// Two interfaces supported: IUnknown, IRpcChannel
if (iidInterface == IID_IUnknown || iidInterface == IID_IRpcChannel)
{
m_refs++; // A pointer to this object is returned
*ppvObj = this;
hresult = NOERROR;
}
else
{
*ppvObj = NULL;
hresult = ResultFromScode(E_NOINTERFACE);
}
return hresult;
}
//+---------------------------------------------------------------------------
//
// Function: AddRef, public
//
// Synopsis: Increments object reference count, called when object is
// referenced by another pointer/client.
//
// Arguments: none
//
// Returns: new reference count
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
STDMETHODIMP_(ULONG) CRpcChannel::AddRef(void)
{
return ++m_refs;
}
//+---------------------------------------------------------------------------
//
// Function: Release, public
//
// Synopsis: Decrements object reference count and handles necessary
// self-destruction and sub-releasing, called when object is
// no longer referenced by another pointer/client.
//
// Arguments: none
//
// Returns: new reference count
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
STDMETHODIMP_(ULONG) CRpcChannel::Release(void)
{
if (--m_refs != 0) // Object still alive?
{
return m_refs;
}
delete this; // Suicide
return 0;
}
//+---------------------------------------------------------------------------
//
// Function: GetStream, public
//
// Synopsis: Create an IStream onto which the client can write in
// preparation for making an RPC call with the stream contents
// as parameters.
//
// Arguments: [iid] - IID of interface being RPC'd
// [iMethod] - Method # within interface
// [fSend] - Should we use SendMessage?
// [fNoWait] - Should we be asynchronous?
// [size] - Initial size of stream
// [ppIStream] - Output IStream
//
// Returns: HRESULT for success/failure
// [ppIStream] - IStream for writing parameters
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
STDMETHODIMP CRpcChannel::GetStream(
REFIID iid,
int iMethod,
BOOL fSend,
BOOL fNoWait,
DWORD size,
IStream FAR* FAR* ppIStream )
{
CPkt FAR* pCPkt;
// no point in allowing this to succeed if we are not connected
if ( IsConnected() != NOERROR )
{
// Connection terminated (server died or disconnectd)
return ResultFromScode(RPC_E_CONNECTION_TERMINATED);
}
pCPkt = CPkt::CreateForCall(this,iid,iMethod,fSend,fNoWait,size);
if (pCPkt == NULL)
{
*ppIStream = NULL;
return ResultFromScode(E_OUTOFMEMORY);
}
if (pCPkt->QueryInterface(IID_IStream,(void FAR* FAR*) ppIStream)
!= NOERROR)
{
pCPkt->Release();
return ResultFromScode(E_OUTOFMEMORY);
}
pCPkt->SetRpcChannelBuffer( m_prcb );
pCPkt->Release();
return NOERROR;
}
//+---------------------------------------------------------------------------
//
// Function: Call, public
//
// Synopsis: Sends a call to a remote procedure. Previously, all of the
// parameters for the procedure were serialized into the IStream
// which was passed in.
//
// Arguments: [pIStream] - IStream for parameters to procedure
//
// Returns: HRESULT for success/failure of call or procedure
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
STDMETHODIMP CRpcChannel::Call(
IStream FAR * pIStream )
{
HRESULT hresult;
CPkt FAR * pCPkt;
hresult = pIStream->QueryInterface(IID_CPkt,(void FAR* FAR*) &pCPkt);
if (hresult != NOERROR)
{
return ResultFromScode(E_INVALIDARG);
}
return pCPkt->CallRpcChannelBuffer();
}
//+---------------------------------------------------------------------------
//
// Function: GetDestCtx, public
//
// Synopsis: According to the code in OLELRPC.CPP, this is all this code
// does, presumably, it might be expanded in the future.
//
// Arguments:
//
// Returns:
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
STDMETHODIMP CRpcChannel::GetDestCtx(
DWORD FAR * lpdwDestCtx,
LPVOID FAR * lplpvDestCtx )
{
*lpdwDestCtx = NULL;
if (lplpvDestCtx)
{
*lplpvDestCtx = NULL;
}
return NOERROR;
}
//+---------------------------------------------------------------------------
//
// Function: IsConnected, public
//
// Synopsis: According to the code in OLELRPC.CPP, this is all this code
// does, presumably, it might be expanded in the future.
//
// Arguments: none
//
// Returns: HRESULT indicating connection status
//
// History: 30-Mar-94 BobDay Created
//
//----------------------------------------------------------------------------
STDMETHODIMP CRpcChannel::IsConnected( void )
{
if ( m_prcb == NULL )
{
return ResultFromScode(E_UNEXPECTED);
}
return m_prcb->IsConnected();
}