mirror of
https://github.com/reactos/reactos.git
synced 2024-08-18 17:37:02 +00:00
361 lines
12 KiB
C
361 lines
12 KiB
C
/*
|
|
* NDR data marshalling
|
|
*
|
|
* Copyright 2006 Mike McCormack (for CodeWeavers)
|
|
* Copyright 2006-2007 Robert Shearman (for CodeWeavers)
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include "ndr_misc.h"
|
|
#include "rpc_assoc.h"
|
|
#include "rpcndr.h"
|
|
|
|
#include "wine/debug.h"
|
|
#include "wine/list.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(ole);
|
|
|
|
#define NDR_CONTEXT_HANDLE_MAGIC 0x4352444e
|
|
|
|
typedef struct ndr_context_handle
|
|
{
|
|
ULONG attributes;
|
|
GUID uuid;
|
|
} ndr_context_handle;
|
|
|
|
struct context_handle_entry
|
|
{
|
|
struct list entry;
|
|
DWORD magic;
|
|
RPC_BINDING_HANDLE handle;
|
|
ndr_context_handle wire_data;
|
|
};
|
|
|
|
static struct list context_handle_list = LIST_INIT(context_handle_list);
|
|
|
|
static CRITICAL_SECTION ndr_context_cs;
|
|
static CRITICAL_SECTION_DEBUG ndr_context_debug =
|
|
{
|
|
0, 0, &ndr_context_cs,
|
|
{ &ndr_context_debug.ProcessLocksList, &ndr_context_debug.ProcessLocksList },
|
|
0, 0, { (DWORD_PTR)(__FILE__ ": ndr_context") }
|
|
};
|
|
static CRITICAL_SECTION ndr_context_cs = { &ndr_context_debug, -1, 0, 0, 0, 0 };
|
|
|
|
static struct context_handle_entry *get_context_entry(NDR_CCONTEXT CContext)
|
|
{
|
|
struct context_handle_entry *che = CContext;
|
|
|
|
if (che->magic != NDR_CONTEXT_HANDLE_MAGIC)
|
|
return NULL;
|
|
return che;
|
|
}
|
|
|
|
static struct context_handle_entry *context_entry_from_guid(LPCGUID uuid)
|
|
{
|
|
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;
|
|
}
|
|
|
|
RPC_BINDING_HANDLE WINAPI NDRCContextBinding(NDR_CCONTEXT CContext)
|
|
{
|
|
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;
|
|
}
|
|
|
|
void WINAPI NDRCContextMarshall(NDR_CCONTEXT CContext, void *pBuff)
|
|
{
|
|
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;
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
* RpcSmDestroyClientContext [RPCRT4.@]
|
|
*/
|
|
RPC_STATUS WINAPI RpcSmDestroyClientContext(void **ContextHandle)
|
|
{
|
|
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;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* RpcSsDestroyClientContext [RPCRT4.@]
|
|
*/
|
|
void WINAPI RpcSsDestroyClientContext(void **ContextHandle)
|
|
{
|
|
RPC_STATUS status = RpcSmDestroyClientContext(ContextHandle);
|
|
if (status != RPC_S_OK)
|
|
RpcRaiseException(status);
|
|
}
|
|
|
|
/***********************************************************************
|
|
* RpcSsDontSerializeContext [RPCRT4.@]
|
|
*/
|
|
void WINAPI RpcSsDontSerializeContext(void)
|
|
{
|
|
FIXME("stub\n");
|
|
}
|
|
|
|
static RPC_STATUS ndr_update_context_handle(NDR_CCONTEXT *CContext,
|
|
RPC_BINDING_HANDLE hBinding,
|
|
const ndr_context_handle *chi)
|
|
{
|
|
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;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* NDRCContextUnmarshall [RPCRT4.@]
|
|
*/
|
|
void WINAPI NDRCContextUnmarshall(NDR_CCONTEXT *CContext,
|
|
RPC_BINDING_HANDLE hBinding,
|
|
void *pBuff, ULONG DataRepresentation)
|
|
{
|
|
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);
|
|
}
|
|
|
|
/***********************************************************************
|
|
* NDRSContextMarshall [RPCRT4.@]
|
|
*/
|
|
void WINAPI NDRSContextMarshall(NDR_SCONTEXT SContext,
|
|
void *pBuff,
|
|
NDR_RUNDOWN userRunDownIn)
|
|
{
|
|
TRACE("(%p %p %p)\n", SContext, pBuff, userRunDownIn);
|
|
NDRSContextMarshall2(I_RpcGetCurrentCallHandle(), SContext, pBuff,
|
|
userRunDownIn, NULL, RPC_CONTEXT_HANDLE_DEFAULT_FLAGS);
|
|
}
|
|
|
|
/***********************************************************************
|
|
* NDRSContextMarshallEx [RPCRT4.@]
|
|
*/
|
|
void WINAPI NDRSContextMarshallEx(RPC_BINDING_HANDLE hBinding,
|
|
NDR_SCONTEXT SContext,
|
|
void *pBuff,
|
|
NDR_RUNDOWN userRunDownIn)
|
|
{
|
|
TRACE("(%p %p %p %p)\n", hBinding, SContext, pBuff, userRunDownIn);
|
|
NDRSContextMarshall2(hBinding, SContext, pBuff, userRunDownIn, NULL,
|
|
RPC_CONTEXT_HANDLE_DEFAULT_FLAGS);
|
|
}
|
|
|
|
/***********************************************************************
|
|
* NDRSContextMarshall2 [RPCRT4.@]
|
|
*/
|
|
void WINAPI NDRSContextMarshall2(RPC_BINDING_HANDLE hBinding,
|
|
NDR_SCONTEXT SContext,
|
|
void *pBuff,
|
|
NDR_RUNDOWN userRunDownIn,
|
|
void *CtxGuard, ULONG Flags)
|
|
{
|
|
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);
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
* NDRSContextUnmarshall [RPCRT4.@]
|
|
*/
|
|
NDR_SCONTEXT WINAPI NDRSContextUnmarshall(void *pBuff,
|
|
ULONG DataRepresentation)
|
|
{
|
|
TRACE("(%p %08x)\n", pBuff, DataRepresentation);
|
|
return NDRSContextUnmarshall2(I_RpcGetCurrentCallHandle(), pBuff,
|
|
DataRepresentation, NULL,
|
|
RPC_CONTEXT_HANDLE_DEFAULT_FLAGS);
|
|
}
|
|
|
|
/***********************************************************************
|
|
* NDRSContextUnmarshallEx [RPCRT4.@]
|
|
*/
|
|
NDR_SCONTEXT WINAPI NDRSContextUnmarshallEx(RPC_BINDING_HANDLE hBinding,
|
|
void *pBuff,
|
|
ULONG DataRepresentation)
|
|
{
|
|
TRACE("(%p %p %08x)\n", hBinding, pBuff, DataRepresentation);
|
|
return NDRSContextUnmarshall2(hBinding, pBuff, DataRepresentation, NULL,
|
|
RPC_CONTEXT_HANDLE_DEFAULT_FLAGS);
|
|
}
|
|
|
|
/***********************************************************************
|
|
* NDRSContextUnmarshall2 [RPCRT4.@]
|
|
*/
|
|
NDR_SCONTEXT WINAPI NDRSContextUnmarshall2(RPC_BINDING_HANDLE hBinding,
|
|
void *pBuff,
|
|
ULONG DataRepresentation,
|
|
void *CtxGuard, ULONG Flags)
|
|
{
|
|
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;
|
|
}
|