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 <stdarg.h> 26 27 #define COBJMACROS 28 29 #include "windef.h" 30 #include "winbase.h" 31 #include "winerror.h" 32 33 #include "objbase.h" 34 35 #include "rpcproxy.h" 36 37 #include "wine/debug.h" 38 39 #include "ndr_misc.h" 40 #include "rpcndr.h" 41 42 WINE_DEFAULT_DEBUG_CHANNEL(rpc); 43 44 /************************************************************************ 45 * NdrClientInitializeNew [RPCRT4.@] 46 */ 47 void WINAPI NdrClientInitializeNew( PRPC_MESSAGE pRpcMessage, PMIDL_STUB_MESSAGE pStubMsg, 48 PMIDL_STUB_DESC pStubDesc, unsigned int ProcNum ) 49 { 50 TRACE("(pRpcMessage == ^%p, pStubMsg == ^%p, pStubDesc == ^%p, ProcNum == %d)\n", 51 pRpcMessage, pStubMsg, pStubDesc, ProcNum); 52 53 pRpcMessage->Handle = NULL; 54 pRpcMessage->ProcNum = ProcNum | RPC_FLAGS_VALID_BIT; 55 pRpcMessage->RpcInterfaceInformation = pStubDesc->RpcInterfaceInformation; 56 pRpcMessage->RpcFlags = 0; 57 pRpcMessage->ReservedForRuntime = NULL; 58 pRpcMessage->DataRepresentation = NDR_LOCAL_DATA_REPRESENTATION; 59 60 pStubMsg->RpcMsg = pRpcMessage; 61 pStubMsg->BufferStart = NULL; 62 pStubMsg->BufferEnd = NULL; 63 pStubMsg->BufferLength = 0; 64 pStubMsg->IsClient = TRUE; 65 pStubMsg->ReuseBuffer = FALSE; 66 pStubMsg->pAllocAllNodesContext = NULL; 67 pStubMsg->pPointerQueueState = NULL; 68 pStubMsg->IgnoreEmbeddedPointers = 0; 69 pStubMsg->PointerBufferMark = NULL; 70 pStubMsg->fBufferValid = 0; 71 pStubMsg->uFlags = 0; 72 pStubMsg->pfnAllocate = pStubDesc->pfnAllocate; 73 pStubMsg->pfnFree = pStubDesc->pfnFree; 74 pStubMsg->StackTop = NULL; 75 pStubMsg->StubDesc = pStubDesc; 76 pStubMsg->FullPtrRefId = 0; 77 pStubMsg->PointerLength = 0; 78 pStubMsg->fInDontFree = 0; 79 pStubMsg->fDontCallFreeInst = 0; 80 pStubMsg->fInOnlyParam = 0; 81 pStubMsg->fHasReturn = 0; 82 pStubMsg->fHasExtensions = 0; 83 pStubMsg->fHasNewCorrDesc = 0; 84 pStubMsg->fUnused = 0; 85 pStubMsg->dwDestContext = MSHCTX_DIFFERENTMACHINE; 86 pStubMsg->pvDestContext = NULL; 87 pStubMsg->pRpcChannelBuffer = NULL; 88 pStubMsg->pArrayInfo = NULL; 89 pStubMsg->dwStubPhase = 0; 90 /* FIXME: LowStackMark */ 91 pStubMsg->pAsyncMsg = NULL; 92 pStubMsg->pCorrInfo = NULL; 93 pStubMsg->pCorrMemory = NULL; 94 pStubMsg->pMemoryList = NULL; 95 } 96 97 /*********************************************************************** 98 * NdrServerInitializeNew [RPCRT4.@] 99 */ 100 unsigned char* WINAPI NdrServerInitializeNew( PRPC_MESSAGE pRpcMsg, PMIDL_STUB_MESSAGE pStubMsg, 101 PMIDL_STUB_DESC pStubDesc ) 102 { 103 TRACE("(pRpcMsg == ^%p, pStubMsg == ^%p, pStubDesc == ^%p)\n", pRpcMsg, pStubMsg, pStubDesc); 104 105 pStubMsg->RpcMsg = pRpcMsg; 106 pStubMsg->Buffer = pStubMsg->BufferStart = pRpcMsg->Buffer; 107 pStubMsg->BufferEnd = pStubMsg->Buffer + pRpcMsg->BufferLength; 108 pStubMsg->BufferLength = pRpcMsg->BufferLength; 109 pStubMsg->IsClient = FALSE; 110 pStubMsg->ReuseBuffer = FALSE; 111 pStubMsg->pAllocAllNodesContext = NULL; 112 pStubMsg->pPointerQueueState = NULL; 113 pStubMsg->IgnoreEmbeddedPointers = 0; 114 pStubMsg->PointerBufferMark = NULL; 115 pStubMsg->uFlags = 0; 116 pStubMsg->pfnAllocate = pStubDesc->pfnAllocate; 117 pStubMsg->pfnFree = pStubDesc->pfnFree; 118 pStubMsg->StackTop = NULL; 119 pStubMsg->StubDesc = pStubDesc; 120 pStubMsg->FullPtrXlatTables = NULL; 121 pStubMsg->FullPtrRefId = 0; 122 pStubMsg->PointerLength = 0; 123 pStubMsg->fInDontFree = 0; 124 pStubMsg->fDontCallFreeInst = 0; 125 pStubMsg->fInOnlyParam = 0; 126 pStubMsg->fHasReturn = 0; 127 pStubMsg->fHasExtensions = 0; 128 pStubMsg->fHasNewCorrDesc = 0; 129 pStubMsg->fUnused = 0; 130 pStubMsg->dwDestContext = MSHCTX_DIFFERENTMACHINE; 131 pStubMsg->pvDestContext = NULL; 132 pStubMsg->pRpcChannelBuffer = NULL; 133 pStubMsg->pArrayInfo = NULL; 134 pStubMsg->dwStubPhase = 0; 135 /* FIXME: LowStackMark */ 136 pStubMsg->pAsyncMsg = NULL; 137 pStubMsg->pCorrInfo = NULL; 138 pStubMsg->pCorrMemory = NULL; 139 pStubMsg->pMemoryList = NULL; 140 141 return NULL; 142 } 143 144 /*********************************************************************** 145 * NdrGetBuffer [RPCRT4.@] 146 */ 147 unsigned char *WINAPI NdrGetBuffer(PMIDL_STUB_MESSAGE stubmsg, ULONG buflen, RPC_BINDING_HANDLE handle) 148 { 149 RPC_STATUS status; 150 151 TRACE("(stubmsg == ^%p, buflen == %u, handle == %p)\n", stubmsg, buflen, handle); 152 153 stubmsg->RpcMsg->Handle = handle; 154 stubmsg->RpcMsg->BufferLength = buflen; 155 156 status = I_RpcGetBuffer(stubmsg->RpcMsg); 157 if (status != RPC_S_OK) 158 RpcRaiseException(status); 159 160 stubmsg->Buffer = stubmsg->RpcMsg->Buffer; 161 stubmsg->fBufferValid = TRUE; 162 stubmsg->BufferLength = stubmsg->RpcMsg->BufferLength; 163 return stubmsg->Buffer; 164 } 165 /*********************************************************************** 166 * NdrFreeBuffer [RPCRT4.@] 167 */ 168 void WINAPI NdrFreeBuffer(PMIDL_STUB_MESSAGE pStubMsg) 169 { 170 TRACE("(pStubMsg == ^%p)\n", pStubMsg); 171 if (pStubMsg->fBufferValid) 172 { 173 I_RpcFreeBuffer(pStubMsg->RpcMsg); 174 pStubMsg->fBufferValid = FALSE; 175 } 176 } 177 178 /************************************************************************ 179 * NdrSendReceive [RPCRT4.@] 180 */ 181 unsigned char *WINAPI NdrSendReceive( PMIDL_STUB_MESSAGE stubmsg, unsigned char *buffer ) 182 { 183 RPC_STATUS status; 184 185 TRACE("(stubmsg == ^%p, buffer == ^%p)\n", stubmsg, buffer); 186 187 /* FIXME: how to handle errors? (raise exception?) */ 188 if (!stubmsg) { 189 ERR("NULL stub message. No action taken.\n"); 190 return NULL; 191 } 192 if (!stubmsg->RpcMsg) { 193 ERR("RPC Message not present in stub message. No action taken.\n"); 194 return NULL; 195 } 196 197 stubmsg->RpcMsg->BufferLength = buffer - (unsigned char *)stubmsg->RpcMsg->Buffer; 198 status = I_RpcSendReceive(stubmsg->RpcMsg); 199 if (status != RPC_S_OK) 200 RpcRaiseException(status); 201 202 stubmsg->BufferLength = stubmsg->RpcMsg->BufferLength; 203 stubmsg->BufferStart = stubmsg->RpcMsg->Buffer; 204 stubmsg->BufferEnd = stubmsg->BufferStart + stubmsg->BufferLength; 205 stubmsg->Buffer = stubmsg->BufferStart; 206 207 /* FIXME: is this the right return value? */ 208 return NULL; 209 } 210 211 /************************************************************************ 212 * NdrMapCommAndFaultStatus [RPCRT4.@] 213 */ 214 RPC_STATUS RPC_ENTRY NdrMapCommAndFaultStatus( PMIDL_STUB_MESSAGE pStubMsg, 215 ULONG *pCommStatus, 216 ULONG *pFaultStatus, 217 RPC_STATUS Status ) 218 { 219 TRACE("(%p, %p, %p, %ld)\n", pStubMsg, pCommStatus, pFaultStatus, Status); 220 221 switch (Status) 222 { 223 case ERROR_INVALID_HANDLE: 224 case RPC_S_INVALID_BINDING: 225 case RPC_S_UNKNOWN_IF: 226 case RPC_S_SERVER_UNAVAILABLE: 227 case RPC_S_SERVER_TOO_BUSY: 228 case RPC_S_CALL_FAILED_DNE: 229 case RPC_S_PROTOCOL_ERROR: 230 case RPC_S_UNSUPPORTED_TRANS_SYN: 231 case RPC_S_UNSUPPORTED_TYPE: 232 case RPC_S_PROCNUM_OUT_OF_RANGE: 233 case EPT_S_NOT_REGISTERED: 234 case RPC_S_COMM_FAILURE: 235 *pCommStatus = Status; 236 *pFaultStatus = 0; 237 break; 238 default: 239 *pCommStatus = 0; 240 *pFaultStatus = Status; 241 } 242 243 return RPC_S_OK; 244 } 245