dlls/rpcrt4/cstub.c

   1 /*
   2  * COM stub (CStdStubBuffer) implementation
   3  *
   4  * Copyright 2001 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winerror.h"
  28
  29 #include "objbase.h"
  30
  31 #include "rpcproxy.h"
  32
  33 #include "wine/debug.h"
  34
  35 #include "cpsf.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(ole);
  38
  39 #define STUB_HEADER(This) (((CInterfaceStubHeader*)((This)->lpVtbl))[-1])
  40
  41 HRESULT WINAPI CStdStubBuffer_Construct(REFIID riid,
  42                                        LPUNKNOWN pUnkServer,
  43                                        PCInterfaceName name,
  44                                        CInterfaceStubVtbl *vtbl,
  45                                        LPPSFACTORYBUFFER pPSFactory,
  46                                        LPRPCSTUBBUFFER *ppStub)
  47 {
  48   CStdStubBuffer *This;
  49
  50   TRACE("(%p,%p,%p,%p) %s\n", pUnkServer, vtbl, pPSFactory, ppStub, name);
  51   TRACE("iid=%s\n", debugstr_guid(vtbl->header.piid));
  52   TRACE("vtbl=%p\n", &vtbl->Vtbl);
  53
  54   if (!IsEqualGUID(vtbl->header.piid, riid)) {
  55     ERR("IID mismatch during stub creation\n");
  56     return RPC_E_UNEXPECTED;
  57   }
  58
  59   This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(CStdStubBuffer));
  60   if (!This) return E_OUTOFMEMORY;
  61
  62   This->lpVtbl = &vtbl->Vtbl;
  63   This->RefCount = 1;
  64   This->pvServerObject = pUnkServer;
  65   This->pPSFactory = pPSFactory;
  66   *ppStub = (LPRPCSTUBBUFFER)This;
  67
  68   IUnknown_AddRef(This->pvServerObject);
  69   IPSFactoryBuffer_AddRef(pPSFactory);
  70   return S_OK;
  71 }
  72
  73 HRESULT WINAPI CStdStubBuffer_QueryInterface(LPRPCSTUBBUFFER iface,
  74                                             REFIID riid,
  75                                             LPVOID *obj)
  76 {
  77   CStdStubBuffer *This = (CStdStubBuffer *)iface;
  78   TRACE("(%p)->QueryInterface(%s,%p)\n",This,debugstr_guid(riid),obj);
  79
  80   if (IsEqualGUID(&IID_IUnknown,riid) ||
  81       IsEqualGUID(&IID_IRpcStubBuffer,riid)) {
  82     *obj = This;
  83     This->RefCount++;
  84     return S_OK;
  85   }
  86   return E_NOINTERFACE;
  87 }
  88
  89 ULONG WINAPI CStdStubBuffer_AddRef(LPRPCSTUBBUFFER iface)
  90 {
  91   CStdStubBuffer *This = (CStdStubBuffer *)iface;
  92   TRACE("(%p)->AddRef()\n",This);
  93   return ++(This->RefCount);
  94 }
  95
  96 ULONG WINAPI NdrCStdStubBuffer_Release(LPRPCSTUBBUFFER iface,
  97                                       LPPSFACTORYBUFFER pPSF)
  98 {
  99   CStdStubBuffer *This = (CStdStubBuffer *)iface;
 100   TRACE("(%p)->Release()\n",This);
 101
 102   if (!--(This->RefCount)) {
 103     IRpcStubBuffer_Disconnect(iface);
 104     if(This->pPSFactory)
 105         IPSFactoryBuffer_Release(This->pPSFactory);
 106     HeapFree(GetProcessHeap(),0,This);
 107     return 0;
 108   }
 109   return This->RefCount;
 110 }
 111
 112 ULONG WINAPI NdrCStdStubBuffer2_Release(LPRPCSTUBBUFFER iface,
 113                                         LPPSFACTORYBUFFER pPSF)
 114 {
 115     FIXME("Not implemented\n");
 116     return 0;
 117 }
 118
 119 HRESULT WINAPI CStdStubBuffer_Connect(LPRPCSTUBBUFFER iface,
 120                                      LPUNKNOWN lpUnkServer)
 121 {
 122   CStdStubBuffer *This = (CStdStubBuffer *)iface;
 123   TRACE("(%p)->Connect(%p)\n",This,lpUnkServer);
 124   This->pvServerObject = lpUnkServer;
 125   return S_OK;
 126 }
 127
 128 void WINAPI CStdStubBuffer_Disconnect(LPRPCSTUBBUFFER iface)
 129 {
 130   CStdStubBuffer *This = (CStdStubBuffer *)iface;
 131   TRACE("(%p)->Disconnect()\n",This);
 132   if (This->pvServerObject)
 133   {
 134     IUnknown_Release(This->pvServerObject);
 135     This->pvServerObject = NULL;
 136   }
 137 }
 138
 139 HRESULT WINAPI CStdStubBuffer_Invoke(LPRPCSTUBBUFFER iface,
 140                                     PRPCOLEMESSAGE pMsg,
 141                                     LPRPCCHANNELBUFFER pChannel)
 142 {
 143   CStdStubBuffer *This = (CStdStubBuffer *)iface;
 144   DWORD dwPhase = STUB_UNMARSHAL;
 145   TRACE("(%p)->Invoke(%p,%p)\n",This,pMsg,pChannel);
 146
 147   if (STUB_HEADER(This).pDispatchTable)
 148     STUB_HEADER(This).pDispatchTable[pMsg->iMethod](iface, pChannel, (PRPC_MESSAGE)pMsg, &dwPhase);
 149   else /* pure interpreted */
 150     NdrStubCall2(iface, pChannel, (PRPC_MESSAGE)pMsg, &dwPhase);
 151   return S_OK;
 152 }
 153
 154 LPRPCSTUBBUFFER WINAPI CStdStubBuffer_IsIIDSupported(LPRPCSTUBBUFFER iface,
 155                                                     REFIID riid)
 156 {
 157   CStdStubBuffer *This = (CStdStubBuffer *)iface;
 158   TRACE("(%p)->IsIIDSupported(%s)\n",This,debugstr_guid(riid));
 159   return IsEqualGUID(STUB_HEADER(This).piid, riid) ? iface : NULL;
 160 }
 161
 162 ULONG WINAPI CStdStubBuffer_CountRefs(LPRPCSTUBBUFFER iface)
 163 {
 164   CStdStubBuffer *This = (CStdStubBuffer *)iface;
 165   TRACE("(%p)->CountRefs()\n",This);
 166   return This->RefCount;
 167 }
 168
 169 HRESULT WINAPI CStdStubBuffer_DebugServerQueryInterface(LPRPCSTUBBUFFER iface,
 170                                                        LPVOID *ppv)
 171 {
 172   CStdStubBuffer *This = (CStdStubBuffer *)iface;
 173   TRACE("(%p)->DebugServerQueryInterface(%p)\n",This,ppv);
 174   return S_OK;
 175 }
 176
 177 void WINAPI CStdStubBuffer_DebugServerRelease(LPRPCSTUBBUFFER iface,
 178                                              LPVOID pv)
 179 {
 180   CStdStubBuffer *This = (CStdStubBuffer *)iface;
 181   TRACE("(%p)->DebugServerRelease(%p)\n",This,pv);
 182 }
 183
 184 const MIDL_SERVER_INFO *CStdStubBuffer_GetServerInfo(IRpcStubBuffer *iface)
 185 {
 186   CStdStubBuffer *This = (CStdStubBuffer *)iface;
 187   return STUB_HEADER(This).pServerInfo;
 188 }