ndr_ole.c File Reference

#include <stdarg.h>
#include <stdio.h>
#include <string.h>
#include "windef.h"
#include "winbase.h"
#include "winerror.h"
#include "objbase.h"
#include "ndr_misc.h"
#include "rpcndr.h"
#include "ndrtypes.h"
#include "rpcproxy.h"
#include "cpsf.h"
#include "wine/debug.h"

Include dependency graph for ndr_ole.c:

Go to the source code of this file.

Classes
struct	RpcStreamImpl

Macros
#define	COBJMACROS

Typedefs
typedef struct RpcStreamImpl	RpcStreamImpl

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (ole)
static RpcStreamImpl *	impl_from_IStream (IStream *iface)
static HRESULT WINAPI	RpcStream_QueryInterface (LPSTREAM iface, REFIID riid, LPVOID *obj)
static ULONG WINAPI	RpcStream_AddRef (LPSTREAM iface)
static ULONG WINAPI	RpcStream_Release (LPSTREAM iface)
static HRESULT WINAPI	RpcStream_Read (LPSTREAM iface, void *pv, ULONG cb, ULONG *pcbRead)
static HRESULT WINAPI	RpcStream_Write (LPSTREAM iface, const void *pv, ULONG cb, ULONG *pcbWritten)
static HRESULT WINAPI	RpcStream_Seek (LPSTREAM iface, LARGE_INTEGER move, DWORD origin, ULARGE_INTEGER *newPos)
static HRESULT WINAPI	RpcStream_SetSize (LPSTREAM iface, ULARGE_INTEGER newSize)
static HRESULT WINAPI	RpcStream_CopyTo (IStream *iface, IStream *dest, ULARGE_INTEGER len, ULARGE_INTEGER *read, ULARGE_INTEGER *written)
static HRESULT WINAPI	RpcStream_Commit (IStream *iface, DWORD flags)
static HRESULT WINAPI	RpcStream_Revert (IStream *iface)
static HRESULT WINAPI	RpcStream_LockRegion (IStream *iface, ULARGE_INTEGER offset, ULARGE_INTEGER len, DWORD locktype)
static HRESULT WINAPI	RpcStream_UnlockRegion (IStream *iface, ULARGE_INTEGER offset, ULARGE_INTEGER len, DWORD locktype)
static HRESULT WINAPI	RpcStream_Stat (IStream *iface, STATSTG *stat, DWORD flag)
static HRESULT WINAPI	RpcStream_Clone (IStream *iface, IStream **cloned)
static HRESULT	RpcStream_Create (PMIDL_STUB_MESSAGE pStubMsg, BOOL init, ULONG *size, IStream **stream)
static const IID *	get_ip_iid (PMIDL_STUB_MESSAGE pStubMsg, unsigned char *pMemory, PFORMAT_STRING pFormat)
unsigned char *WINAPI	NdrInterfacePointerMarshall (PMIDL_STUB_MESSAGE pStubMsg, unsigned char *pMemory, PFORMAT_STRING pFormat)
unsigned char *WINAPI	NdrInterfacePointerUnmarshall (PMIDL_STUB_MESSAGE pStubMsg, unsigned char **ppMemory, PFORMAT_STRING pFormat, unsigned char fMustAlloc)
void WINAPI	NdrInterfacePointerBufferSize (PMIDL_STUB_MESSAGE pStubMsg, unsigned char *pMemory, PFORMAT_STRING pFormat)
ULONG WINAPI	NdrInterfacePointerMemorySize (PMIDL_STUB_MESSAGE pStubMsg, PFORMAT_STRING pFormat)
void WINAPI	NdrInterfacePointerFree (PMIDL_STUB_MESSAGE pStubMsg, unsigned char *pMemory, PFORMAT_STRING pFormat)
void *WINAPI	NdrOleAllocate (SIZE_T Size)
void WINAPI	NdrOleFree (void *NodeToFree)
HRESULT	create_proxy (REFIID iid, IUnknown *pUnkOuter, IRpcProxyBuffer **pproxy, void **ppv)
HRESULT	create_stub (REFIID iid, IUnknown *pUnk, IRpcStubBuffer **ppstub)

Variables
static const IStreamVtbl	RpcStream_Vtbl

Macro Definition Documentation

COBJMACROS

#define COBJMACROS

Definition at line 28 of file ndr_ole.c.

Typedef Documentation

RpcStreamImpl

typedef struct RpcStreamImpl RpcStreamImpl

Function Documentation

create_proxy()

HRESULT create_proxy	(	REFIID	iid,
		IUnknown *	pUnkOuter,
		IRpcProxyBuffer **	pproxy,
		void **	ppv
	)

Definition at line 408 of file ndr_ole.c.

{
    CLSID clsid;
    IPSFactoryBuffer *psfac;
    HRESULT r;
 
    r = CoGetPSClsid(iid, &clsid);
    if(FAILED(r)) return r;
 
    r = CoGetClassObject(&clsid, CLSCTX_INPROC_SERVER, NULL, &IID_IPSFactoryBuffer, (void **)&psfac);
    if(FAILED(r)) return r;
 
    r = IPSFactoryBuffer_CreateProxy(psfac, pUnkOuter, iid, pproxy, ppv);
 
    IPSFactoryBuffer_Release(psfac);
    return r;
}

Referenced by StdProxy_Construct(), and typelib_proxy_init().

create_stub()

HRESULT create_stub	(	REFIID	iid,
		IUnknown *	pUnk,
		IRpcStubBuffer **	ppstub
	)

Definition at line 430 of file ndr_ole.c.

{
    CLSID clsid;
    IPSFactoryBuffer *psfac;
    HRESULT r;
 
    r = CoGetPSClsid(iid, &clsid);
    if(FAILED(r)) return r;
 
    r = CoGetClassObject(&clsid, CLSCTX_INPROC_SERVER, NULL, &IID_IPSFactoryBuffer, (void **)&psfac);
    if(FAILED(r)) return r;
 
    r = IPSFactoryBuffer_CreateStub(psfac, iid, pUnk, ppstub);
 
    IPSFactoryBuffer_Release(psfac);
    return r;
}

Referenced by CStdStubBuffer_Delegating_Construct(), test_Connect(), test_CreateStub(), test_delegating_Invoke(), test_Disconnect(), test_Release(), and typelib_stub_init().

get_ip_iid()

static const IID * get_ip_iid ( PMIDL_STUB_MESSAGE  pStubMsg, unsigned char *  pMemory, PFORMAT_STRING  pFormat  )

static

Definition at line 257 of file ndr_ole.c.

{
  const IID *riid;
  if (!pFormat) return &IID_IUnknown;
  TRACE("format=%02x %02x\n", pFormat[0], pFormat[1]);
  if (pFormat[0] != FC_IP) FIXME("format=%d\n", pFormat[0]);
  if (pFormat[1] == FC_CONSTANT_IID) {
    riid = (const IID *)&pFormat[2];
  } else {
    ComputeConformance(pStubMsg, pMemory, pFormat+2, 0);
    riid = (const IID *)pStubMsg->MaxCount;
  }
  if (!riid) riid = &IID_IUnknown;
  TRACE("got %s\n", debugstr_guid(riid));
  return riid;
}

Referenced by NdrInterfacePointerBufferSize(), and NdrInterfacePointerMarshall().

impl_from_IStream()

static RpcStreamImpl * impl_from_IStream ( IStream *  iface )

inlinestatic

Definition at line 58 of file ndr_ole.c.

{
  return CONTAINING_RECORD(iface, RpcStreamImpl, IStream_iface);
}

Referenced by RpcStream_AddRef(), RpcStream_Clone(), RpcStream_Commit(), RpcStream_CopyTo(), RpcStream_LockRegion(), RpcStream_Read(), RpcStream_Release(), RpcStream_Revert(), RpcStream_Seek(), RpcStream_SetSize(), RpcStream_Stat(), RpcStream_UnlockRegion(), and RpcStream_Write().

NdrInterfacePointerBufferSize()

void WINAPI NdrInterfacePointerBufferSize	(	PMIDL_STUB_MESSAGE	pStubMsg,
		unsigned char *	pMemory,
		PFORMAT_STRING	pFormat
	)

Definition at line 342 of file ndr_ole.c.

{
  const IID *riid = get_ip_iid(pStubMsg, pMemory, pFormat);
  ULONG size = 0;
 
  TRACE("(%p,%p,%p)\n", pStubMsg, pMemory, pFormat);
  CoGetMarshalSizeMax(&size, riid, (IUnknown *)pMemory,
                      pStubMsg->dwDestContext, pStubMsg->pvDestContext,
                      MSHLFLAGS_NORMAL);
  TRACE("size=%ld\n", size);
  pStubMsg->BufferLength += sizeof(DWORD) + size;
}

Referenced by test_iface_ptr().

NdrInterfacePointerFree()

void WINAPI NdrInterfacePointerFree	(	PMIDL_STUB_MESSAGE	pStubMsg,
		unsigned char *	pMemory,
		PFORMAT_STRING	pFormat
	)

Definition at line 379 of file ndr_ole.c.

{
  LPUNKNOWN pUnk = (LPUNKNOWN)pMemory;
  TRACE("(%p,%p,%p)\n", pStubMsg, pMemory, pFormat);
  if (pUnk) IUnknown_Release(pUnk);
}

Referenced by test_iface_ptr().

NdrInterfacePointerMarshall()

unsigned char *WINAPI NdrInterfacePointerMarshall	(	PMIDL_STUB_MESSAGE	pStubMsg,
		unsigned char *	pMemory,
		PFORMAT_STRING	pFormat
	)

Definition at line 277 of file ndr_ole.c.

{
  const IID *riid = get_ip_iid(pStubMsg, pMemory, pFormat);
  LPSTREAM stream;
  HRESULT hr;
 
  TRACE("(%p,%p,%p)\n", pStubMsg, pMemory, pFormat);
  pStubMsg->MaxCount = 0;
  if (pStubMsg->Buffer + sizeof(DWORD) <= (unsigned char *)pStubMsg->RpcMsg->Buffer + pStubMsg->BufferLength) {
    hr = RpcStream_Create(pStubMsg, TRUE, NULL, &stream);
    if (hr == S_OK) {
      if (pMemory)
        hr = CoMarshalInterface(stream, riid, (IUnknown *)pMemory,
                                pStubMsg->dwDestContext, pStubMsg->pvDestContext,
                                MSHLFLAGS_NORMAL);
      IStream_Release(stream);
    }
 
    if (FAILED(hr))
      RpcRaiseException(hr);
  }
  return NULL;
}

Referenced by test_iface_ptr().

NdrInterfacePointerMemorySize()

ULONG WINAPI NdrInterfacePointerMemorySize	(	PMIDL_STUB_MESSAGE	pStubMsg,
		PFORMAT_STRING	pFormat
	)

Definition at line 360 of file ndr_ole.c.

{
  ULONG size;
 
  TRACE("(%p,%p)\n", pStubMsg, pFormat);
 
  size = *(ULONG *)pStubMsg->Buffer;
  pStubMsg->Buffer += 4;
  pStubMsg->MemorySize += 4;
 
  pStubMsg->Buffer += size;
 
  return pStubMsg->MemorySize;
}

NdrInterfacePointerUnmarshall()

unsigned char *WINAPI NdrInterfacePointerUnmarshall	(	PMIDL_STUB_MESSAGE	pStubMsg,
		unsigned char **	ppMemory,
		PFORMAT_STRING	pFormat,
		unsigned char	fMustAlloc
	)

Definition at line 306 of file ndr_ole.c.

{
  IUnknown **unk = (IUnknown **)ppMemory;
  LPSTREAM stream;
  HRESULT hr;
 
  TRACE("(%p,%p,%p,%d)\n", pStubMsg, ppMemory, pFormat, fMustAlloc);
 
  /* Avoid reference leaks for [in, out] pointers. */
  if (pStubMsg->IsClient && *unk)
    IUnknown_Release(*unk);
 
  *unk = NULL;
  if (pStubMsg->Buffer + sizeof(DWORD) < (unsigned char *)pStubMsg->RpcMsg->Buffer + pStubMsg->BufferLength) {
    ULONG size;
 
    hr = RpcStream_Create(pStubMsg, FALSE, &size, &stream);
    if (hr == S_OK) {
      if (size != 0)
        hr = CoUnmarshalInterface(stream, &IID_NULL, (void **)unk);
 
      IStream_Release(stream);
    }
 
    if (FAILED(hr))
        RpcRaiseException(hr);
  }
  return NULL;
}

Referenced by test_iface_ptr().

NdrOleAllocate()

void *WINAPI NdrOleAllocate

(

SIZE_T

Size

)

Definition at line 391 of file ndr_ole.c.

{
  return CoTaskMemAlloc(Size);
}

Referenced by determine_pointer_marshalling_style(), init_stub_desc(), test_conf_complex_array(), test_conf_complex_struct(), test_conformant_array(), test_conformant_string(), test_iface_ptr(), test_ndr_simple_type(), test_nonconformant_string(), test_nontrivial_pointer_types(), test_pointer_marshal(), test_simple_struct_marshal(), and test_struct_align().

NdrOleFree()

void WINAPI NdrOleFree

(

void *

NodeToFree

)

Definition at line 399 of file ndr_ole.c.

{
  CoTaskMemFree(NodeToFree);
}

Referenced by determine_pointer_marshalling_style(), init_stub_desc(), test_conf_complex_array(), test_conf_complex_struct(), test_conformant_array(), test_conformant_string(), test_iface_ptr(), test_ndr_simple_type(), test_nonconformant_string(), test_nontrivial_pointer_types(), test_pointer_marshal(), test_simple_struct_marshal(), and test_struct_align().

RpcStream_AddRef()

static ULONG WINAPI RpcStream_AddRef ( LPSTREAM  iface )

static

Definition at line 79 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  return InterlockedIncrement( &This->RefCount );
}

RpcStream_Clone()

static HRESULT WINAPI RpcStream_Clone ( IStream *  iface, IStream **  cloned  )

static

Definition at line 211 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  FIXME("(%p): stub\n", This);
  return E_NOTIMPL;
}

RpcStream_Commit()

static HRESULT WINAPI RpcStream_Commit ( IStream *  iface, DWORD  flags  )

static

Definition at line 174 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  FIXME("(%p)->(0x%08lx): stub\n", This, flags);
  return E_NOTIMPL;
}

RpcStream_CopyTo()

static HRESULT WINAPI RpcStream_CopyTo ( IStream *  iface, IStream *  dest, ULARGE_INTEGER  len, ULARGE_INTEGER *  read, ULARGE_INTEGER *  written  )

static

Definition at line 166 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  FIXME("(%p): stub\n", This);
  return E_NOTIMPL;
}

RpcStream_Create()

static HRESULT RpcStream_Create ( PMIDL_STUB_MESSAGE  pStubMsg, BOOL  init, ULONG *  size, IStream **  stream  )

static

Definition at line 236 of file ndr_ole.c.

{
  RpcStreamImpl *This;
 
  *stream = NULL;
  This = malloc(sizeof(RpcStreamImpl));
  if (!This) return E_OUTOFMEMORY;
  This->IStream_iface.lpVtbl = &RpcStream_Vtbl;
  This->RefCount = 1;
  This->pMsg = pStubMsg;
  This->size = (LPDWORD)pStubMsg->Buffer;
  This->data = pStubMsg->Buffer + sizeof(DWORD);
  This->pos = 0;
  if (init) *This->size = 0;
  TRACE("init size=%ld\n", *This->size);
 
  if (size) *size = *This->size;
  *stream = &This->IStream_iface;
  return S_OK;
}

Referenced by NdrInterfacePointerMarshall(), and NdrInterfacePointerUnmarshall().

RpcStream_LockRegion()

static HRESULT WINAPI RpcStream_LockRegion ( IStream *  iface, ULARGE_INTEGER  offset, ULARGE_INTEGER  len, DWORD  locktype  )

static

Definition at line 188 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  FIXME("(%p): stub\n", This);
  return E_NOTIMPL;
}

RpcStream_QueryInterface()

static HRESULT WINAPI RpcStream_QueryInterface ( LPSTREAM  iface, REFIID  riid, LPVOID *  obj  )

static

Definition at line 63 of file ndr_ole.c.

{
  if (IsEqualGUID(&IID_IUnknown, riid) ||
      IsEqualGUID(&IID_ISequentialStream, riid) ||
      IsEqualGUID(&IID_IStream, riid)) {
    *obj = iface;
    IStream_AddRef(iface);
    return S_OK;
  }
 
  *obj = NULL;
  return E_NOINTERFACE;
}

RpcStream_Read()

static HRESULT WINAPI RpcStream_Read ( LPSTREAM  iface, void *  pv, ULONG  cb, ULONG *  pcbRead  )

static

Definition at line 97 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  HRESULT hr = S_OK;
  if (This->pos + cb > *This->size)
  {
    cb = *This->size - This->pos;
    hr = S_FALSE;
  }
  if (cb) {
    memcpy(pv, This->data + This->pos, cb);
    This->pos += cb;
  }
  if (pcbRead) *pcbRead = cb;
  return hr;
}

RpcStream_Release()

static ULONG WINAPI RpcStream_Release ( LPSTREAM  iface )

static

Definition at line 85 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  ULONG ref = InterlockedDecrement( &This->RefCount );
  if (!ref) {
    TRACE("size=%ld\n", *This->size);
    This->pMsg->Buffer = This->data + *This->size;
    free(This);
  }
  return ref;
}

RpcStream_Revert()

static HRESULT WINAPI RpcStream_Revert ( IStream *  iface )

static

Definition at line 181 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  FIXME("(%p): stub\n", This);
  return E_NOTIMPL;
}

RpcStream_Seek()

static HRESULT WINAPI RpcStream_Seek ( LPSTREAM  iface, LARGE_INTEGER  move, DWORD  origin, ULARGE_INTEGER *  newPos  )

static

Definition at line 132 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  switch (origin) {
  case STREAM_SEEK_SET:
    This->pos = move.LowPart;
    break;
  case STREAM_SEEK_CUR:
    This->pos = This->pos + move.LowPart;
    break;
  case STREAM_SEEK_END:
    This->pos = *This->size + move.LowPart;
    break;
  default:
    return STG_E_INVALIDFUNCTION;
  }
  if (newPos) {
    newPos->u.LowPart = This->pos;
    newPos->u.HighPart = 0;
  }
  return S_OK;
}

RpcStream_SetSize()

static HRESULT WINAPI RpcStream_SetSize ( LPSTREAM  iface, ULARGE_INTEGER  newSize  )

static

Definition at line 158 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  *This->size = newSize.LowPart;
  return S_OK;
}

RpcStream_Stat()

static HRESULT WINAPI RpcStream_Stat ( IStream *  iface, STATSTG *  stat, DWORD  flag  )

static

Definition at line 204 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  FIXME("(%p): stub\n", This);
  return E_NOTIMPL;
}

RpcStream_UnlockRegion()

static HRESULT WINAPI RpcStream_UnlockRegion ( IStream *  iface, ULARGE_INTEGER  offset, ULARGE_INTEGER  len, DWORD  locktype  )

static

Definition at line 196 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  FIXME("(%p): stub\n", This);
  return E_NOTIMPL;
}

RpcStream_Write()

static HRESULT WINAPI RpcStream_Write ( LPSTREAM  iface, const void *  pv, ULONG  cb, ULONG *  pcbWritten  )

static

Definition at line 117 of file ndr_ole.c.

{
  RpcStreamImpl *This = impl_from_IStream(iface);
  if (This->data + cb > (unsigned char *)This->pMsg->RpcMsg->Buffer + This->pMsg->BufferLength)
    return STG_E_MEDIUMFULL;
  memcpy(This->data + This->pos, pv, cb);
  This->pos += cb;
  if (This->pos > *This->size) *This->size = This->pos;
  if (pcbWritten) *pcbWritten = cb;
  return S_OK;
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

ole

)

Variable Documentation

RpcStream_Vtbl

const IStreamVtbl RpcStream_Vtbl

static

Initial value:

=
{
  RpcStream_QueryInterface,
  RpcStream_AddRef,
  RpcStream_Release,
  RpcStream_Read,
  RpcStream_Write,
  RpcStream_Seek,
  RpcStream_SetSize,
  RpcStream_CopyTo,
  RpcStream_Commit,
  RpcStream_Revert,
  RpcStream_LockRegion,
  RpcStream_UnlockRegion,
  RpcStream_Stat,
  RpcStream_Clone
}

Definition at line 218 of file ndr_ole.c.

Referenced by RpcStream_Create().