dll/win32/rpcrt4/ndr_clientserver.c

   1 /*
   2  * MIDL proxy/stub stuff
   3  *
   4  * Copyright 2002 Ove Kåven, TransGaming Technologies
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  *
  20  * TODO:
  21  *  - figure out whether we *really* got this right
  22  *  - check for errors and throw exceptions
  23  */
  24
  25 #include "precomp.h"
  26
  27 WINE_DEFAULT_DEBUG_CHANNEL(rpc);
  28
  29 /************************************************************************
  30  *             NdrClientInitializeNew [RPCRT4.@]
  31  */
  32 void WINAPI NdrClientInitializeNew( PRPC_MESSAGE pRpcMessage, PMIDL_STUB_MESSAGE pStubMsg,
  33                                     PMIDL_STUB_DESC pStubDesc, unsigned int ProcNum )
  34 {
  35   TRACE("(pRpcMessage == ^%p, pStubMsg == ^%p, pStubDesc == ^%p, ProcNum == %d)\n",
  36     pRpcMessage, pStubMsg, pStubDesc, ProcNum);
  37
  38   pRpcMessage->Handle = NULL;
  39   pRpcMessage->ProcNum = ProcNum | RPC_FLAGS_VALID_BIT;
  40   pRpcMessage->RpcInterfaceInformation = pStubDesc->RpcInterfaceInformation;
  41   pRpcMessage->RpcFlags = 0;
  42   pRpcMessage->ReservedForRuntime = NULL;
  43   pRpcMessage->DataRepresentation = NDR_LOCAL_DATA_REPRESENTATION;
  44
  45   pStubMsg->RpcMsg = pRpcMessage;
  46   pStubMsg->BufferStart = NULL;
  47   pStubMsg->BufferEnd = NULL;
  48   pStubMsg->BufferLength = 0;
  49   pStubMsg->IsClient = TRUE;
  50   pStubMsg->ReuseBuffer = FALSE;
  51   pStubMsg->pAllocAllNodesContext = NULL;
  52   pStubMsg->pPointerQueueState = NULL;
  53   pStubMsg->IgnoreEmbeddedPointers = 0;
  54   pStubMsg->PointerBufferMark = NULL;
  55   pStubMsg->CorrDespIncrement = 0;
  56   pStubMsg->uFlags = 0;
  57   pStubMsg->UniquePtrCount = 0;
  58   pStubMsg->pfnAllocate = pStubDesc->pfnAllocate;
  59   pStubMsg->pfnFree = pStubDesc->pfnFree;
  60   pStubMsg->StackTop = NULL;
  61   pStubMsg->StubDesc = pStubDesc;
  62   pStubMsg->FullPtrRefId = 0;
  63   pStubMsg->PointerLength = 0;
  64   pStubMsg->fInDontFree = 0;
  65   pStubMsg->fDontCallFreeInst = 0;
  66   pStubMsg->fInOnlyParam = 0;
  67   pStubMsg->fHasReturn = 0;
  68   pStubMsg->fHasExtensions = 0;
  69   pStubMsg->fHasNewCorrDesc = 0;
  70   pStubMsg->fIsIn = 0;
  71   pStubMsg->fIsOut = 0;
  72   pStubMsg->fIsOicf = 0;
  73   pStubMsg->fBufferValid = 0;
  74   pStubMsg->fHasMemoryValidateCallback = 0;
  75   pStubMsg->fInFree = 0;
  76   pStubMsg->fNeedMCCP = 0;
  77   pStubMsg->fUnused = 0;
  78   pStubMsg->dwDestContext = MSHCTX_DIFFERENTMACHINE;
  79   pStubMsg->pvDestContext = NULL;
  80   pStubMsg->pRpcChannelBuffer = NULL;
  81   pStubMsg->pArrayInfo = NULL;
  82   pStubMsg->dwStubPhase = 0;
  83   /* FIXME: LowStackMark */
  84   pStubMsg->pAsyncMsg = NULL;
  85   pStubMsg->pCorrInfo = NULL;
  86   pStubMsg->pCorrMemory = NULL;
  87   pStubMsg->pMemoryList = NULL;
  88 }
  89
  90 /***********************************************************************
  91  *             NdrServerInitializeNew [RPCRT4.@]
  92  */
  93 unsigned char* WINAPI NdrServerInitializeNew( PRPC_MESSAGE pRpcMsg, PMIDL_STUB_MESSAGE pStubMsg,
  94                                               PMIDL_STUB_DESC pStubDesc )
  95 {
  96   TRACE("(pRpcMsg == ^%p, pStubMsg == ^%p, pStubDesc == ^%p)\n", pRpcMsg, pStubMsg, pStubDesc);
  97
  98   pStubMsg->RpcMsg = pRpcMsg;
  99   pStubMsg->Buffer = pStubMsg->BufferStart = pRpcMsg->Buffer;
 100   pStubMsg->BufferEnd = pStubMsg->Buffer + pRpcMsg->BufferLength;
 101   pStubMsg->BufferLength = pRpcMsg->BufferLength;
 102   pStubMsg->IsClient = FALSE;
 103   pStubMsg->ReuseBuffer = FALSE;
 104   pStubMsg->pAllocAllNodesContext = NULL;
 105   pStubMsg->pPointerQueueState = NULL;
 106   pStubMsg->IgnoreEmbeddedPointers = 0;
 107   pStubMsg->PointerBufferMark = NULL;
 108   pStubMsg->CorrDespIncrement = 0;
 109   pStubMsg->uFlags = 0;
 110   pStubMsg->UniquePtrCount = 0;
 111   pStubMsg->pfnAllocate = pStubDesc->pfnAllocate;
 112   pStubMsg->pfnFree = pStubDesc->pfnFree;
 113   pStubMsg->StackTop = NULL;
 114   pStubMsg->StubDesc = pStubDesc;
 115   pStubMsg->FullPtrXlatTables = NULL;
 116   pStubMsg->FullPtrRefId = 0;
 117   pStubMsg->PointerLength = 0;
 118   pStubMsg->fInDontFree = 0;
 119   pStubMsg->fDontCallFreeInst = 0;
 120   pStubMsg->fInOnlyParam = 0;
 121   pStubMsg->fHasReturn = 0;
 122   pStubMsg->fHasExtensions = 0;
 123   pStubMsg->fHasNewCorrDesc = 0;
 124   pStubMsg->fIsIn = 0;
 125   pStubMsg->fIsOut = 0;
 126   pStubMsg->fIsOicf = 0;
 127   pStubMsg->fHasMemoryValidateCallback = 0;
 128   pStubMsg->fInFree = 0;
 129   pStubMsg->fNeedMCCP = 0;
 130   pStubMsg->fUnused = 0;
 131   pStubMsg->dwDestContext = MSHCTX_DIFFERENTMACHINE;
 132   pStubMsg->pvDestContext = NULL;
 133   pStubMsg->pRpcChannelBuffer = NULL;
 134   pStubMsg->pArrayInfo = NULL;
 135   pStubMsg->dwStubPhase = 0;
 136   /* FIXME: LowStackMark */
 137   pStubMsg->pAsyncMsg = NULL;
 138   pStubMsg->pCorrInfo = NULL;
 139   pStubMsg->pCorrMemory = NULL;
 140   pStubMsg->pMemoryList = NULL;
 141
 142   return NULL;
 143 }
 144
 145 /***********************************************************************
 146  *           NdrGetBuffer [RPCRT4.@]
 147  */
 148 unsigned char *WINAPI NdrGetBuffer(PMIDL_STUB_MESSAGE stubmsg, ULONG buflen, RPC_BINDING_HANDLE handle)
 149 {
 150   RPC_STATUS status;
 151
 152   TRACE("(stubmsg == ^%p, buflen == %u, handle == %p)\n", stubmsg, buflen, handle);
 153
 154   stubmsg->RpcMsg->Handle = handle;
 155   stubmsg->RpcMsg->BufferLength = buflen;
 156
 157   status = I_RpcGetBuffer(stubmsg->RpcMsg);
 158   if (status != RPC_S_OK)
 159     RpcRaiseException(status);
 160
 161   stubmsg->Buffer = stubmsg->RpcMsg->Buffer;
 162   stubmsg->fBufferValid = TRUE;
 163   stubmsg->BufferLength = stubmsg->RpcMsg->BufferLength;
 164   return stubmsg->Buffer;
 165 }
 166 /***********************************************************************
 167  *           NdrFreeBuffer [RPCRT4.@]
 168  */
 169 void WINAPI NdrFreeBuffer(PMIDL_STUB_MESSAGE pStubMsg)
 170 {
 171   TRACE("(pStubMsg == ^%p)\n", pStubMsg);
 172   if (pStubMsg->fBufferValid)
 173   {
 174     I_RpcFreeBuffer(pStubMsg->RpcMsg);
 175     pStubMsg->fBufferValid = FALSE;
 176   }
 177 }
 178
 179 /************************************************************************
 180  *           NdrSendReceive [RPCRT4.@]
 181  */
 182 unsigned char *WINAPI NdrSendReceive( PMIDL_STUB_MESSAGE stubmsg, unsigned char *buffer  )
 183 {
 184   RPC_STATUS status;
 185
 186   TRACE("(stubmsg == ^%p, buffer == ^%p)\n", stubmsg, buffer);
 187
 188   /* FIXME: how to handle errors? (raise exception?) */
 189   if (!stubmsg) {
 190     ERR("NULL stub message.  No action taken.\n");
 191     return NULL;
 192   }
 193   if (!stubmsg->RpcMsg) {
 194     ERR("RPC Message not present in stub message.  No action taken.\n");
 195     return NULL;
 196   }
 197
 198   /* avoid sending uninitialised parts of the buffer on the wire */
 199   stubmsg->RpcMsg->BufferLength = buffer - (unsigned char *)stubmsg->RpcMsg->Buffer;
 200   status = I_RpcSendReceive(stubmsg->RpcMsg);
 201   if (status != RPC_S_OK)
 202     RpcRaiseException(status);
 203
 204   stubmsg->BufferLength = stubmsg->RpcMsg->BufferLength;
 205   stubmsg->BufferStart = stubmsg->RpcMsg->Buffer;
 206   stubmsg->BufferEnd = stubmsg->BufferStart + stubmsg->BufferLength;
 207   stubmsg->Buffer = stubmsg->BufferStart;
 208
 209   /* FIXME: is this the right return value? */
 210   return NULL;
 211 }
 212
 213 /************************************************************************
 214  *           NdrMapCommAndFaultStatus [RPCRT4.@]
 215  */
 216 RPC_STATUS RPC_ENTRY NdrMapCommAndFaultStatus( PMIDL_STUB_MESSAGE pStubMsg,
 217                                                ULONG *pCommStatus,
 218                                                ULONG *pFaultStatus,
 219                                                RPC_STATUS Status )
 220 {
 221     TRACE("(%p, %p, %p, %d)\n", pStubMsg, pCommStatus, pFaultStatus, Status);
 222
 223     switch (Status)
 224     {
 225     case ERROR_INVALID_HANDLE:
 226     case RPC_S_INVALID_BINDING:
 227     case RPC_S_UNKNOWN_IF:
 228     case RPC_S_SERVER_UNAVAILABLE:
 229     case RPC_S_SERVER_TOO_BUSY:
 230     case RPC_S_CALL_FAILED_DNE:
 231     case RPC_S_PROTOCOL_ERROR:
 232     case RPC_S_UNSUPPORTED_TRANS_SYN:
 233     case RPC_S_UNSUPPORTED_TYPE:
 234     case RPC_S_PROCNUM_OUT_OF_RANGE:
 235     case EPT_S_NOT_REGISTERED:
 236     case RPC_S_COMM_FAILURE:
 237         *pCommStatus = Status;
 238         *pFaultStatus = 0;
 239         break;
 240     default:
 241         *pCommStatus = 0;
 242         *pFaultStatus = Status;
 243     }
 244
 245     return RPC_S_OK;
 246 }