ndr_contexthandle.c File Reference

#include "ndr_misc.h"
#include "rpc_assoc.h"
#include "rpcndr.h"
#include "wine/debug.h"
#include "wine/list.h"

Include dependency graph for ndr_contexthandle.c:

Go to the source code of this file.

Classes
struct	ndr_context_handle
struct	context_handle_entry

Macros
#define	NDR_CONTEXT_HANDLE_MAGIC   0x4352444e

Typedefs
typedef struct ndr_context_handle	ndr_context_handle

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (ole)
static struct context_handle_entry *	get_context_entry (NDR_CCONTEXT CContext)
static struct context_handle_entry *	context_entry_from_guid (LPCGUID uuid)
RPC_BINDING_HANDLE WINAPI	NDRCContextBinding (NDR_CCONTEXT CContext)
void WINAPI	NDRCContextMarshall (NDR_CCONTEXT CContext, void *pBuff)
RPC_STATUS WINAPI	RpcSmDestroyClientContext (void **ContextHandle)
void WINAPI	RpcSsDestroyClientContext (void **ContextHandle)
void WINAPI	RpcSsDontSerializeContext (void)
static RPC_STATUS	ndr_update_context_handle (NDR_CCONTEXT *CContext, RPC_BINDING_HANDLE hBinding, const ndr_context_handle *chi)
void WINAPI	NDRCContextUnmarshall (NDR_CCONTEXT *CContext, RPC_BINDING_HANDLE hBinding, void *pBuff, ULONG DataRepresentation)
void WINAPI	NDRSContextMarshall (NDR_SCONTEXT SContext, void *pBuff, NDR_RUNDOWN userRunDownIn)
void WINAPI	NDRSContextMarshallEx (RPC_BINDING_HANDLE hBinding, NDR_SCONTEXT SContext, void *pBuff, NDR_RUNDOWN userRunDownIn)
void WINAPI	NDRSContextMarshall2 (RPC_BINDING_HANDLE hBinding, NDR_SCONTEXT SContext, void *pBuff, NDR_RUNDOWN userRunDownIn, void *CtxGuard, ULONG Flags)
NDR_SCONTEXT WINAPI	NDRSContextUnmarshall (void *pBuff, ULONG DataRepresentation)
NDR_SCONTEXT WINAPI	NDRSContextUnmarshallEx (RPC_BINDING_HANDLE hBinding, void *pBuff, ULONG DataRepresentation)
NDR_SCONTEXT WINAPI	NDRSContextUnmarshall2 (RPC_BINDING_HANDLE hBinding, void *pBuff, ULONG DataRepresentation, void *CtxGuard, ULONG Flags)

Variables
static struct list	context_handle_list = LIST_INIT(context_handle_list)
static CRITICAL_SECTION	ndr_context_cs = { &ndr_context_debug, -1, 0, 0, 0, 0 }
static CRITICAL_SECTION_DEBUG	ndr_context_debug

Macro Definition Documentation

NDR_CONTEXT_HANDLE_MAGIC

#define NDR_CONTEXT_HANDLE_MAGIC   0x4352444e

Definition at line 31 of file ndr_contexthandle.c.

Typedef Documentation

ndr_context_handle

typedef struct ndr_context_handle ndr_context_handle

Function Documentation

context_entry_from_guid()

static struct context_handle_entry * context_entry_from_guid ( LPCGUID  uuid )

static

Definition at line 67 of file ndr_contexthandle.c.

{
    struct context_handle_entry *che;
    LIST_FOR_EACH_ENTRY(che, &context_handle_list, struct context_handle_entry, entry)
        if (IsEqualGUID(&che->wire_data.uuid, uuid))
            return che;
    return NULL;
}

Referenced by ndr_update_context_handle().

get_context_entry()

static struct context_handle_entry * get_context_entry ( NDR_CCONTEXT  CContext )

static

Definition at line 58 of file ndr_contexthandle.c.

{
    struct context_handle_entry *che = CContext;
 
    if (che->magic != NDR_CONTEXT_HANDLE_MAGIC)
        return NULL;
    return che;
}

Referenced by ndr_update_context_handle(), NDRCContextBinding(), NDRCContextMarshall(), and RpcSmDestroyClientContext().

ndr_update_context_handle()

static RPC_STATUS ndr_update_context_handle ( NDR_CCONTEXT *  CContext, RPC_BINDING_HANDLE  hBinding, const ndr_context_handle *  chi  )

static

Definition at line 166 of file ndr_contexthandle.c.

{
    struct context_handle_entry *che = NULL;
 
    /* a null UUID means we should free the context handle */
    if (IsEqualGUID(&chi->uuid, &GUID_NULL))
    {
        if (*CContext)
        {
            che = get_context_entry(*CContext);
            if (!che)
                return RPC_X_SS_CONTEXT_MISMATCH;
            list_remove(&che->entry);
            RpcBindingFree(&che->handle);
            HeapFree(GetProcessHeap(), 0, che);
            che = NULL;
        }
    }
    /* if there's no existing entry matching the GUID, allocate one */
    else if (!(che = context_entry_from_guid(&chi->uuid)))
    {
        che = HeapAlloc(GetProcessHeap(), 0, sizeof *che);
        if (!che)
            return RPC_X_NO_MEMORY;
        che->magic = NDR_CONTEXT_HANDLE_MAGIC;
        RpcBindingCopy(hBinding, &che->handle);
        list_add_tail(&context_handle_list, &che->entry);
        che->wire_data = *chi;
    }
 
    *CContext = che;
 
    return RPC_S_OK;
}

Referenced by NDRCContextUnmarshall().

NDRCContextBinding()

RPC_BINDING_HANDLE WINAPI NDRCContextBinding

(

NDR_CCONTEXT

CContext

)

Definition at line 76 of file ndr_contexthandle.c.

{
    struct context_handle_entry *che;
    RPC_BINDING_HANDLE handle = NULL;
 
    TRACE("%p\n", CContext);
 
    EnterCriticalSection(&ndr_context_cs);
    che = get_context_entry(CContext);
    if (che)
        handle = che->handle;
    LeaveCriticalSection(&ndr_context_cs);
 
    if (!handle)
    {
        ERR("invalid handle %p\n", CContext);
        RpcRaiseException(RPC_X_SS_CONTEXT_MISMATCH);
    }
    return handle;
}

Referenced by client_get_handle(), CtxClose(), CtxHello(), m_CtxClose(), m_CtxHello(), and main().

NDRCContextMarshall()

void WINAPI NDRCContextMarshall	(	NDR_CCONTEXT	CContext,
		void *	pBuff
	)

Definition at line 97 of file ndr_contexthandle.c.

{
    struct context_handle_entry *che;
 
    TRACE("%p %p\n", CContext, pBuff);
 
    if (CContext)
    {
        EnterCriticalSection(&ndr_context_cs);
        che = get_context_entry(CContext);
        memcpy(pBuff, &che->wire_data, sizeof (ndr_context_handle));
        LeaveCriticalSection(&ndr_context_cs);
    }
    else
    {
        ndr_context_handle *wire_data = pBuff;
        wire_data->attributes = 0;
        wire_data->uuid = GUID_NULL;
    }
}

Referenced by main(), and NdrClientContextMarshall().

NDRCContextUnmarshall()

void WINAPI NDRCContextUnmarshall	(	NDR_CCONTEXT *	CContext,
		RPC_BINDING_HANDLE	hBinding,
		void *	pBuff,
		ULONG	DataRepresentation
	)

Definition at line 206 of file ndr_contexthandle.c.

{
    RPC_STATUS status;
 
    TRACE("*%p=(%p) %p %p %08x\n",
          CContext, *CContext, hBinding, pBuff, DataRepresentation);
 
    EnterCriticalSection(&ndr_context_cs);
    status = ndr_update_context_handle(CContext, hBinding, pBuff);
    LeaveCriticalSection(&ndr_context_cs);
    if (status)
        RpcRaiseException(status);
}

Referenced by m_CtxOpen2(), and NdrClientContextUnmarshall().

NDRSContextMarshall()

void WINAPI NDRSContextMarshall	(	NDR_SCONTEXT	SContext,
		void *	pBuff,
		NDR_RUNDOWN	userRunDownIn
	)

Definition at line 225 of file ndr_contexthandle.c.

{
    TRACE("(%p %p %p)\n", SContext, pBuff, userRunDownIn);
    NDRSContextMarshall2(I_RpcGetCurrentCallHandle(), SContext, pBuff,
                         userRunDownIn, NULL, RPC_CONTEXT_HANDLE_DEFAULT_FLAGS);
}

NDRSContextMarshall2()

void WINAPI NDRSContextMarshall2	(	RPC_BINDING_HANDLE	hBinding,
		NDR_SCONTEXT	SContext,
		void *	pBuff,
		NDR_RUNDOWN	userRunDownIn,
		void *	CtxGuard,
		ULONG	Flags
	)

Definition at line 250 of file ndr_contexthandle.c.

{
    RpcBinding *binding = hBinding;
    RPC_STATUS status;
    ndr_context_handle *ndr = pBuff;
 
    TRACE("(%p %p %p %p %p %u)\n",
          hBinding, SContext, pBuff, userRunDownIn, CtxGuard, Flags);
 
    if (!binding->server || !binding->Assoc)
        RpcRaiseException(RPC_S_INVALID_BINDING);
 
    if (SContext->userContext)
    {
        status = RpcServerAssoc_UpdateContextHandle(binding->Assoc, SContext, CtxGuard, userRunDownIn);
        if (status != RPC_S_OK)
            RpcRaiseException(status);
        ndr->attributes = 0;
        RpcContextHandle_GetUuid(SContext, &ndr->uuid);
 
        RPCRT4_RemoveThreadContextHandle(SContext);
        RpcServerAssoc_ReleaseContextHandle(binding->Assoc, SContext, TRUE);
    }
    else
    {
        if (!RpcContextHandle_IsGuardCorrect(SContext, CtxGuard))
            RpcRaiseException(RPC_X_SS_CONTEXT_MISMATCH);
        memset(ndr, 0, sizeof(*ndr));
 
        RPCRT4_RemoveThreadContextHandle(SContext);
        /* Note: release the context handle twice in this case to release
         * one ref being kept around for the data and one ref for the
         * unmarshall/marshall sequence */
        if (!RpcServerAssoc_ReleaseContextHandle(binding->Assoc, SContext, TRUE))
            return; /* this is to cope with the case of the data not being valid
                     * before and so not having a further reference */
        RpcServerAssoc_ReleaseContextHandle(binding->Assoc, SContext, FALSE);
    }
}

Referenced by NDRSContextMarshall(), NDRSContextMarshallEx(), NdrServerContextMarshall(), and NdrServerContextNewMarshall().

NDRSContextMarshallEx()

void WINAPI NDRSContextMarshallEx	(	RPC_BINDING_HANDLE	hBinding,
		NDR_SCONTEXT	SContext,
		void *	pBuff,
		NDR_RUNDOWN	userRunDownIn
	)

Definition at line 237 of file ndr_contexthandle.c.

{
    TRACE("(%p %p %p %p)\n", hBinding, SContext, pBuff, userRunDownIn);
    NDRSContextMarshall2(hBinding, SContext, pBuff, userRunDownIn, NULL,
                         RPC_CONTEXT_HANDLE_DEFAULT_FLAGS);
}

NDRSContextUnmarshall()

NDR_SCONTEXT WINAPI NDRSContextUnmarshall	(	void *	pBuff,
		ULONG	DataRepresentation
	)

Definition at line 297 of file ndr_contexthandle.c.

{
    TRACE("(%p %08x)\n", pBuff, DataRepresentation);
    return NDRSContextUnmarshall2(I_RpcGetCurrentCallHandle(), pBuff,
                                  DataRepresentation, NULL,
                                  RPC_CONTEXT_HANDLE_DEFAULT_FLAGS);
}

Referenced by hello_CtxOpen().

NDRSContextUnmarshall2()

NDR_SCONTEXT WINAPI NDRSContextUnmarshall2	(	RPC_BINDING_HANDLE	hBinding,
		void *	pBuff,
		ULONG	DataRepresentation,
		void *	CtxGuard,
		ULONG	Flags
	)

Definition at line 321 of file ndr_contexthandle.c.

{
    RpcBinding *binding = hBinding;
    NDR_SCONTEXT SContext;
    RPC_STATUS status;
    const ndr_context_handle *context_ndr = pBuff;
 
    TRACE("(%p %p %08x %p %u)\n",
          hBinding, pBuff, DataRepresentation, CtxGuard, Flags);
 
    if (!binding->server || !binding->Assoc)
        RpcRaiseException(RPC_S_INVALID_BINDING);
 
    if (!pBuff || (!context_ndr->attributes &&
                   UuidIsNil((UUID *)&context_ndr->uuid, &status)))
        status = RpcServerAssoc_AllocateContextHandle(binding->Assoc, CtxGuard,
                                                      &SContext);
    else
    {
        if (context_ndr->attributes)
        {
            ERR("non-null attributes 0x%x\n", context_ndr->attributes);
            status = RPC_X_SS_CONTEXT_MISMATCH;
        }
        else
            status = RpcServerAssoc_FindContextHandle(binding->Assoc,
                                                      &context_ndr->uuid,
                                                      CtxGuard, Flags,
                                                      &SContext);
    }
 
    if (status != RPC_S_OK)
        RpcRaiseException(status);
 
    RPCRT4_PushThreadContextHandle(SContext);
    return SContext;
}

Referenced by NdrContextHandleInitialize(), NDRSContextUnmarshall(), NDRSContextUnmarshallEx(), NdrServerContextNewUnmarshall(), and NdrServerContextUnmarshall().

NDRSContextUnmarshallEx()

NDR_SCONTEXT WINAPI NDRSContextUnmarshallEx	(	RPC_BINDING_HANDLE	hBinding,
		void *	pBuff,
		ULONG	DataRepresentation
	)

Definition at line 309 of file ndr_contexthandle.c.

{
    TRACE("(%p %p %08x)\n", hBinding, pBuff, DataRepresentation);
    return NDRSContextUnmarshall2(hBinding, pBuff, DataRepresentation, NULL,
                                  RPC_CONTEXT_HANDLE_DEFAULT_FLAGS);
}

RpcSmDestroyClientContext()

RPC_STATUS WINAPI RpcSmDestroyClientContext

(

void **

ContextHandle

)

Definition at line 121 of file ndr_contexthandle.c.

{
    RPC_STATUS status = RPC_X_SS_CONTEXT_MISMATCH;
    struct context_handle_entry *che = NULL;
 
    TRACE("(%p)\n", ContextHandle);
 
    EnterCriticalSection(&ndr_context_cs);
    che = get_context_entry(*ContextHandle);
    *ContextHandle = NULL;
    if (che)
    {
        status = RPC_S_OK;
        list_remove(&che->entry);
    }
 
    LeaveCriticalSection(&ndr_context_cs);
 
    if (che)
    {
        RpcBindingFree(&che->handle);
        HeapFree(GetProcessHeap(), 0, che);
    }
 
    return status;
}

Referenced by RpcSsDestroyClientContext().

RpcSsDestroyClientContext()

void WINAPI RpcSsDestroyClientContext

(

void **

ContextHandle

)

Definition at line 151 of file ndr_contexthandle.c.

{
    RPC_STATUS status = RpcSmDestroyClientContext(ContextHandle);
    if (status != RPC_S_OK)
        RpcRaiseException(status);
}

RpcSsDontSerializeContext()

void WINAPI RpcSsDontSerializeContext

(

void

)

Definition at line 161 of file ndr_contexthandle.c.

{
    FIXME("stub\n");
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

ole

)

Variable Documentation

context_handle_list

struct list context_handle_list = LIST_INIT(context_handle_list)

static

Definition at line 47 of file ndr_contexthandle.c.

ndr_context_cs

static CRITICAL_SECTION ndr_context_cs = { &ndr_context_debug, -1, 0, 0, 0, 0 }

static

Definition at line 49 of file ndr_contexthandle.c.

Referenced by NDRCContextBinding(), NDRCContextMarshall(), NDRCContextUnmarshall(), and RpcSmDestroyClientContext().

ndr_context_debug

CRITICAL_SECTION_DEBUG ndr_context_debug

static

Initial value:

=
{
    0, 0, &ndr_context_cs,
    { &ndr_context_debug.ProcessLocksList, &ndr_context_debug.ProcessLocksList },
      0, 0, { (DWORD_PTR)(__FILE__ ": ndr_context") }
}

Definition at line 50 of file ndr_contexthandle.c.