git.oblomov.eu Git - wine/blob - 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 "windef.h"
  22 #include "winbase.h"
  23 #include "winerror.h"
  24
  25 #include "wine/obj_base.h"
  26 #include "wine/obj_channel.h"
  27
  28 #include "rpcproxy.h"
  29
  30 #include "wine/debug.h"
  31
  32 #include "cpsf.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(ole);
  35
  36 #define STUB_HEADER(This) (((CInterfaceStubHeader*)((This)->lpVtbl))[-1])
  37
  38 HRESULT WINAPI CStdStubBuffer_Construct(REFIID riid,
  39                                        LPUNKNOWN pUnkServer,
  40                                        CInterfaceStubVtbl *vtbl,
  41                                        LPPSFACTORYBUFFER pPSFactory,
  42                                        LPRPCSTUBBUFFER *ppStub)
  43 {
  44   CStdStubBuffer *This;
  45
  46   TRACE("(%p,%p,%p,%p)\n", pUnkServer, vtbl, pPSFactory, ppStub);
  47   TRACE("iid=%s\n", debugstr_guid(vtbl->header.piid));
  48   TRACE("vtbl=%p\n", &vtbl->Vtbl);
  49
  50   if (!IsEqualGUID(vtbl->header.piid, riid)) {
  51     ERR("IID mismatch during stub creation\n");
  52     return RPC_E_UNEXPECTED;
  53   }
  54
  55   This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(CStdStubBuffer));
  56   if (!This) return E_OUTOFMEMORY;
  57
  58   This->lpVtbl = &vtbl->Vtbl;
  59   This->RefCount = 1;
  60   This->pvServerObject = pUnkServer;
  61   This->pPSFactory = pPSFactory;
  62   *ppStub = (LPRPCSTUBBUFFER)This;
  63
  64   IPSFactoryBuffer_AddRef(pPSFactory);
  65   return S_OK;
  66 }
  67
  68 HRESULT WINAPI CStdStubBuffer_QueryInterface(LPRPCSTUBBUFFER iface,
  69                                             REFIID riid,
  70                                             LPVOID *obj)
  71 {
  72   ICOM_THIS(CStdStubBuffer,iface);
  73   TRACE("(%p)->QueryInterface(%s,%p)\n",This,debugstr_guid(riid),obj);
  74
  75   if (IsEqualGUID(&IID_IUnknown,riid) ||
  76       IsEqualGUID(&IID_IRpcStubBuffer,riid)) {
  77     *obj = This;
  78     This->RefCount++;
  79     return S_OK;
  80   }
  81   return E_NOINTERFACE;
  82 }
  83
  84 ULONG WINAPI CStdStubBuffer_AddRef(LPRPCSTUBBUFFER iface)
  85 {
  86   ICOM_THIS(CStdStubBuffer,iface);
  87   TRACE("(%p)->AddRef()\n",This);
  88   return ++(This->RefCount);
  89 }
  90
  91 ULONG WINAPI NdrCStdStubBuffer_Release(LPRPCSTUBBUFFER iface,
  92                                       LPPSFACTORYBUFFER pPSF)
  93 {
  94   ICOM_THIS(CStdStubBuffer,iface);
  95   TRACE("(%p)->Release()\n",This);
  96
  97   if (!--(This->RefCount)) {
  98     if(This->pvServerObject)
  99         IUnknown_Release(This->pvServerObject);
 100     if(This->pPSFactory)
 101         IPSFactoryBuffer_Release(This->pPSFactory);
 102     HeapFree(GetProcessHeap(),0,This);
 103     return 0;
 104   }
 105   return This->RefCount;
 106 }
 107
 108 HRESULT WINAPI CStdStubBuffer_Connect(LPRPCSTUBBUFFER iface,
 109                                      LPUNKNOWN lpUnkServer)
 110 {
 111   ICOM_THIS(CStdStubBuffer,iface);
 112   TRACE("(%p)->Connect(%p)\n",This,lpUnkServer);
 113   This->pvServerObject = lpUnkServer;
 114   return S_OK;
 115 }
 116
 117 void WINAPI CStdStubBuffer_Disconnect(LPRPCSTUBBUFFER iface)
 118 {
 119   ICOM_THIS(CStdStubBuffer,iface);
 120   TRACE("(%p)->Disconnect()\n",This);
 121   This->pvServerObject = NULL;
 122 }
 123
 124 HRESULT WINAPI CStdStubBuffer_Invoke(LPRPCSTUBBUFFER iface,
 125                                     PRPCOLEMESSAGE pMsg,
 126                                     LPRPCCHANNELBUFFER pChannel)
 127 {
 128   ICOM_THIS(CStdStubBuffer,iface);
 129   DWORD dwPhase = STUB_UNMARSHAL;
 130   TRACE("(%p)->Invoke(%p,%p)\n",This,pMsg,pChannel);
 131
 132   STUB_HEADER(This).pDispatchTable[pMsg->iMethod](iface, pChannel, (PRPC_MESSAGE)pMsg, &dwPhase);
 133   return S_OK;
 134 }
 135
 136 LPRPCSTUBBUFFER WINAPI CStdStubBuffer_IsIIDSupported(LPRPCSTUBBUFFER iface,
 137                                                     REFIID riid)
 138 {
 139   ICOM_THIS(CStdStubBuffer,iface);
 140   TRACE("(%p)->IsIIDSupported(%s)\n",This,debugstr_guid(riid));
 141   return IsEqualGUID(STUB_HEADER(This).piid, riid) ? iface : NULL;
 142 }
 143
 144 ULONG WINAPI CStdStubBuffer_CountRefs(LPRPCSTUBBUFFER iface)
 145 {
 146   ICOM_THIS(CStdStubBuffer,iface);
 147   TRACE("(%p)->CountRefs()\n",This);
 148   return This->RefCount;
 149 }
 150
 151 HRESULT WINAPI CStdStubBuffer_DebugServerQueryInterface(LPRPCSTUBBUFFER iface,
 152                                                        LPVOID *ppv)
 153 {
 154   ICOM_THIS(CStdStubBuffer,iface);
 155   TRACE("(%p)->DebugServerQueryInterface(%p)\n",This,ppv);
 156   return S_OK;
 157 }
 158
 159 void WINAPI CStdStubBuffer_DebugServerRelease(LPRPCSTUBBUFFER iface,
 160                                              LPVOID pv)
 161 {
 162   ICOM_THIS(CStdStubBuffer,iface);
 163   TRACE("(%p)->DebugServerRelease(%p)\n",This,pv);
 164 }