text: /*
   * 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;
  }
  