mirror of
https://github.com/reactos/reactos.git
synced 2024-08-22 03:12:51 +00:00
262 lines
8.2 KiB
C
262 lines
8.2 KiB
C
/*
|
|
* MIDL proxy/stub stuff
|
|
*
|
|
* Copyright 2002 Ove Kåven, TransGaming Technologies
|
|
*
|
|
* 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
|
|
*
|
|
* TODO:
|
|
* - figure out whether we *really* got this right
|
|
* - check for errors and throw exceptions
|
|
*/
|
|
|
|
#include <stdarg.h>
|
|
|
|
#define COBJMACROS
|
|
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "winerror.h"
|
|
|
|
#include "objbase.h"
|
|
|
|
#include "rpcproxy.h"
|
|
|
|
#include "wine/debug.h"
|
|
|
|
#include "ndr_misc.h"
|
|
#include "rpcndr.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(rpc);
|
|
|
|
/************************************************************************
|
|
* NdrClientInitializeNew [RPCRT4.@]
|
|
*/
|
|
void WINAPI NdrClientInitializeNew( PRPC_MESSAGE pRpcMessage, PMIDL_STUB_MESSAGE pStubMsg,
|
|
PMIDL_STUB_DESC pStubDesc, unsigned int ProcNum )
|
|
{
|
|
TRACE("(pRpcMessage == ^%p, pStubMsg == ^%p, pStubDesc == ^%p, ProcNum == %d)\n",
|
|
pRpcMessage, pStubMsg, pStubDesc, ProcNum);
|
|
|
|
pRpcMessage->Handle = NULL;
|
|
pRpcMessage->ProcNum = ProcNum | RPC_FLAGS_VALID_BIT;
|
|
pRpcMessage->RpcInterfaceInformation = pStubDesc->RpcInterfaceInformation;
|
|
pRpcMessage->RpcFlags = 0;
|
|
pRpcMessage->ReservedForRuntime = NULL;
|
|
pRpcMessage->DataRepresentation = NDR_LOCAL_DATA_REPRESENTATION;
|
|
|
|
pStubMsg->RpcMsg = pRpcMessage;
|
|
pStubMsg->BufferStart = NULL;
|
|
pStubMsg->BufferEnd = NULL;
|
|
pStubMsg->BufferLength = 0;
|
|
pStubMsg->IsClient = TRUE;
|
|
pStubMsg->ReuseBuffer = FALSE;
|
|
pStubMsg->pAllocAllNodesContext = NULL;
|
|
pStubMsg->pPointerQueueState = NULL;
|
|
pStubMsg->IgnoreEmbeddedPointers = 0;
|
|
pStubMsg->PointerBufferMark = NULL;
|
|
pStubMsg->CorrDespIncrement = 0;
|
|
pStubMsg->uFlags = 0;
|
|
pStubMsg->UniquePtrCount = 0;
|
|
pStubMsg->pfnAllocate = pStubDesc->pfnAllocate;
|
|
pStubMsg->pfnFree = pStubDesc->pfnFree;
|
|
pStubMsg->StackTop = NULL;
|
|
pStubMsg->StubDesc = pStubDesc;
|
|
pStubMsg->FullPtrRefId = 0;
|
|
pStubMsg->PointerLength = 0;
|
|
pStubMsg->fInDontFree = 0;
|
|
pStubMsg->fDontCallFreeInst = 0;
|
|
pStubMsg->fInOnlyParam = 0;
|
|
pStubMsg->fHasReturn = 0;
|
|
pStubMsg->fHasExtensions = 0;
|
|
pStubMsg->fHasNewCorrDesc = 0;
|
|
pStubMsg->fIsIn = 0;
|
|
pStubMsg->fIsOut = 0;
|
|
pStubMsg->fIsOicf = 0;
|
|
pStubMsg->fBufferValid = 0;
|
|
pStubMsg->fHasMemoryValidateCallback = 0;
|
|
pStubMsg->fInFree = 0;
|
|
pStubMsg->fNeedMCCP = 0;
|
|
pStubMsg->fUnused = 0;
|
|
pStubMsg->dwDestContext = MSHCTX_DIFFERENTMACHINE;
|
|
pStubMsg->pvDestContext = NULL;
|
|
pStubMsg->pRpcChannelBuffer = NULL;
|
|
pStubMsg->pArrayInfo = NULL;
|
|
pStubMsg->dwStubPhase = 0;
|
|
/* FIXME: LowStackMark */
|
|
pStubMsg->pAsyncMsg = NULL;
|
|
pStubMsg->pCorrInfo = NULL;
|
|
pStubMsg->pCorrMemory = NULL;
|
|
pStubMsg->pMemoryList = NULL;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* NdrServerInitializeNew [RPCRT4.@]
|
|
*/
|
|
unsigned char* WINAPI NdrServerInitializeNew( PRPC_MESSAGE pRpcMsg, PMIDL_STUB_MESSAGE pStubMsg,
|
|
PMIDL_STUB_DESC pStubDesc )
|
|
{
|
|
TRACE("(pRpcMsg == ^%p, pStubMsg == ^%p, pStubDesc == ^%p)\n", pRpcMsg, pStubMsg, pStubDesc);
|
|
|
|
pStubMsg->RpcMsg = pRpcMsg;
|
|
pStubMsg->Buffer = pStubMsg->BufferStart = pRpcMsg->Buffer;
|
|
pStubMsg->BufferEnd = pStubMsg->Buffer + pRpcMsg->BufferLength;
|
|
pStubMsg->BufferLength = pRpcMsg->BufferLength;
|
|
pStubMsg->IsClient = FALSE;
|
|
pStubMsg->ReuseBuffer = FALSE;
|
|
pStubMsg->pAllocAllNodesContext = NULL;
|
|
pStubMsg->pPointerQueueState = NULL;
|
|
pStubMsg->IgnoreEmbeddedPointers = 0;
|
|
pStubMsg->PointerBufferMark = NULL;
|
|
pStubMsg->CorrDespIncrement = 0;
|
|
pStubMsg->uFlags = 0;
|
|
pStubMsg->UniquePtrCount = 0;
|
|
pStubMsg->pfnAllocate = pStubDesc->pfnAllocate;
|
|
pStubMsg->pfnFree = pStubDesc->pfnFree;
|
|
pStubMsg->StackTop = NULL;
|
|
pStubMsg->StubDesc = pStubDesc;
|
|
pStubMsg->FullPtrXlatTables = NULL;
|
|
pStubMsg->FullPtrRefId = 0;
|
|
pStubMsg->PointerLength = 0;
|
|
pStubMsg->fInDontFree = 0;
|
|
pStubMsg->fDontCallFreeInst = 0;
|
|
pStubMsg->fInOnlyParam = 0;
|
|
pStubMsg->fHasReturn = 0;
|
|
pStubMsg->fHasExtensions = 0;
|
|
pStubMsg->fHasNewCorrDesc = 0;
|
|
pStubMsg->fIsIn = 0;
|
|
pStubMsg->fIsOut = 0;
|
|
pStubMsg->fIsOicf = 0;
|
|
pStubMsg->fHasMemoryValidateCallback = 0;
|
|
pStubMsg->fInFree = 0;
|
|
pStubMsg->fNeedMCCP = 0;
|
|
pStubMsg->fUnused = 0;
|
|
pStubMsg->dwDestContext = MSHCTX_DIFFERENTMACHINE;
|
|
pStubMsg->pvDestContext = NULL;
|
|
pStubMsg->pRpcChannelBuffer = NULL;
|
|
pStubMsg->pArrayInfo = NULL;
|
|
pStubMsg->dwStubPhase = 0;
|
|
/* FIXME: LowStackMark */
|
|
pStubMsg->pAsyncMsg = NULL;
|
|
pStubMsg->pCorrInfo = NULL;
|
|
pStubMsg->pCorrMemory = NULL;
|
|
pStubMsg->pMemoryList = NULL;
|
|
|
|
return NULL;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* NdrGetBuffer [RPCRT4.@]
|
|
*/
|
|
unsigned char *WINAPI NdrGetBuffer(PMIDL_STUB_MESSAGE stubmsg, ULONG buflen, RPC_BINDING_HANDLE handle)
|
|
{
|
|
RPC_STATUS status;
|
|
|
|
TRACE("(stubmsg == ^%p, buflen == %u, handle == %p)\n", stubmsg, buflen, handle);
|
|
|
|
stubmsg->RpcMsg->Handle = handle;
|
|
stubmsg->RpcMsg->BufferLength = buflen;
|
|
|
|
status = I_RpcGetBuffer(stubmsg->RpcMsg);
|
|
if (status != RPC_S_OK)
|
|
RpcRaiseException(status);
|
|
|
|
stubmsg->Buffer = stubmsg->RpcMsg->Buffer;
|
|
stubmsg->fBufferValid = TRUE;
|
|
stubmsg->BufferLength = stubmsg->RpcMsg->BufferLength;
|
|
return stubmsg->Buffer;
|
|
}
|
|
/***********************************************************************
|
|
* NdrFreeBuffer [RPCRT4.@]
|
|
*/
|
|
void WINAPI NdrFreeBuffer(PMIDL_STUB_MESSAGE pStubMsg)
|
|
{
|
|
TRACE("(pStubMsg == ^%p)\n", pStubMsg);
|
|
if (pStubMsg->fBufferValid)
|
|
{
|
|
I_RpcFreeBuffer(pStubMsg->RpcMsg);
|
|
pStubMsg->fBufferValid = FALSE;
|
|
}
|
|
}
|
|
|
|
/************************************************************************
|
|
* NdrSendReceive [RPCRT4.@]
|
|
*/
|
|
unsigned char *WINAPI NdrSendReceive( PMIDL_STUB_MESSAGE stubmsg, unsigned char *buffer )
|
|
{
|
|
RPC_STATUS status;
|
|
|
|
TRACE("(stubmsg == ^%p, buffer == ^%p)\n", stubmsg, buffer);
|
|
|
|
/* FIXME: how to handle errors? (raise exception?) */
|
|
if (!stubmsg) {
|
|
ERR("NULL stub message. No action taken.\n");
|
|
return NULL;
|
|
}
|
|
if (!stubmsg->RpcMsg) {
|
|
ERR("RPC Message not present in stub message. No action taken.\n");
|
|
return NULL;
|
|
}
|
|
|
|
/* avoid sending uninitialised parts of the buffer on the wire */
|
|
stubmsg->RpcMsg->BufferLength = buffer - (unsigned char *)stubmsg->RpcMsg->Buffer;
|
|
status = I_RpcSendReceive(stubmsg->RpcMsg);
|
|
if (status != RPC_S_OK)
|
|
RpcRaiseException(status);
|
|
|
|
stubmsg->BufferLength = stubmsg->RpcMsg->BufferLength;
|
|
stubmsg->BufferStart = stubmsg->RpcMsg->Buffer;
|
|
stubmsg->BufferEnd = stubmsg->BufferStart + stubmsg->BufferLength;
|
|
stubmsg->Buffer = stubmsg->BufferStart;
|
|
|
|
/* FIXME: is this the right return value? */
|
|
return NULL;
|
|
}
|
|
|
|
/************************************************************************
|
|
* NdrMapCommAndFaultStatus [RPCRT4.@]
|
|
*/
|
|
RPC_STATUS RPC_ENTRY NdrMapCommAndFaultStatus( PMIDL_STUB_MESSAGE pStubMsg,
|
|
ULONG *pCommStatus,
|
|
ULONG *pFaultStatus,
|
|
RPC_STATUS Status )
|
|
{
|
|
TRACE("(%p, %p, %p, %d)\n", pStubMsg, pCommStatus, pFaultStatus, Status);
|
|
|
|
switch (Status)
|
|
{
|
|
case ERROR_INVALID_HANDLE:
|
|
case RPC_S_INVALID_BINDING:
|
|
case RPC_S_UNKNOWN_IF:
|
|
case RPC_S_SERVER_UNAVAILABLE:
|
|
case RPC_S_SERVER_TOO_BUSY:
|
|
case RPC_S_CALL_FAILED_DNE:
|
|
case RPC_S_PROTOCOL_ERROR:
|
|
case RPC_S_UNSUPPORTED_TRANS_SYN:
|
|
case RPC_S_UNSUPPORTED_TYPE:
|
|
case RPC_S_PROCNUM_OUT_OF_RANGE:
|
|
case EPT_S_NOT_REGISTERED:
|
|
case RPC_S_COMM_FAILURE:
|
|
*pCommStatus = Status;
|
|
*pFaultStatus = 0;
|
|
break;
|
|
default:
|
|
*pCommStatus = 0;
|
|
*pFaultStatus = Status;
|
|
}
|
|
|
|
return RPC_S_OK;
|
|
}
|