2 * COM proxy/stub factory (CStdPSFactory) implementation
4 * Copyright 2001 Ove Kåven, TransGaming Technologies
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.
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.
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
33 #include "wine/debug.h"
37 WINE_DEFAULT_DEBUG_CHANNEL(ole);
39 static BOOL FindProxyInfo(const ProxyFileInfo **pProxyFileList, REFIID riid, const ProxyFileInfo **pProxyInfo, int *pIndex)
41 while (*pProxyFileList) {
42 if ((*pProxyFileList)->pIIDLookupRtn(riid, pIndex)) {
43 *pProxyInfo = *pProxyFileList;
44 TRACE("found: ProxyInfo %p Index %d\n", *pProxyInfo, *pIndex);
53 static HRESULT WINAPI CStdPSFactory_QueryInterface(LPPSFACTORYBUFFER iface,
57 ICOM_THIS(CStdPSFactoryBuffer,iface);
58 TRACE("(%p)->QueryInterface(%s,%p)\n",iface,debugstr_guid(riid),obj);
59 if (IsEqualGUID(&IID_IUnknown,riid) ||
60 IsEqualGUID(&IID_IPSFactoryBuffer,riid)) {
68 static ULONG WINAPI CStdPSFactory_AddRef(LPPSFACTORYBUFFER iface)
70 ICOM_THIS(CStdPSFactoryBuffer,iface);
71 TRACE("(%p)->AddRef()\n",iface);
72 return ++(This->RefCount);
75 static ULONG WINAPI CStdPSFactory_Release(LPPSFACTORYBUFFER iface)
77 ICOM_THIS(CStdPSFactoryBuffer,iface);
78 TRACE("(%p)->Release()\n",iface);
79 return --(This->RefCount);
82 static HRESULT WINAPI CStdPSFactory_CreateProxy(LPPSFACTORYBUFFER iface,
85 LPRPCPROXYBUFFER *ppProxy,
88 ICOM_THIS(CStdPSFactoryBuffer,iface);
89 const ProxyFileInfo *ProxyInfo;
91 TRACE("(%p)->CreateProxy(%p,%s,%p,%p)\n",iface,pUnkOuter,
92 debugstr_guid(riid),ppProxy,ppv);
93 if (!FindProxyInfo(This->pProxyFileList,riid,&ProxyInfo,&Index))
95 return StdProxy_Construct(riid, pUnkOuter, ProxyInfo->pNamesArray[Index],
96 ProxyInfo->pProxyVtblList[Index],
97 ProxyInfo->pStubVtblList[Index], iface, ppProxy, ppv);
100 static HRESULT WINAPI CStdPSFactory_CreateStub(LPPSFACTORYBUFFER iface,
102 LPUNKNOWN pUnkServer,
103 LPRPCSTUBBUFFER *ppStub)
105 ICOM_THIS(CStdPSFactoryBuffer,iface);
106 const ProxyFileInfo *ProxyInfo;
108 TRACE("(%p)->CreateStub(%s,%p,%p)\n",iface,debugstr_guid(riid),
110 if (!FindProxyInfo(This->pProxyFileList,riid,&ProxyInfo,&Index))
111 return E_NOINTERFACE;
112 return CStdStubBuffer_Construct(riid, pUnkServer, ProxyInfo->pNamesArray[Index],
113 ProxyInfo->pStubVtblList[Index], iface, ppStub);
116 static ICOM_VTABLE(IPSFactoryBuffer) CStdPSFactory_Vtbl =
118 ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
119 CStdPSFactory_QueryInterface,
120 CStdPSFactory_AddRef,
121 CStdPSFactory_Release,
122 CStdPSFactory_CreateProxy,
123 CStdPSFactory_CreateStub
126 /***********************************************************************
127 * NdrDllGetClassObject [RPCRT4.@]
129 HRESULT WINAPI NdrDllGetClassObject(REFCLSID rclsid, REFIID iid, LPVOID *ppv,
130 const ProxyFileInfo **pProxyFileList,
132 CStdPSFactoryBuffer *pPSFactoryBuffer)
135 if (!pPSFactoryBuffer->lpVtbl) {
136 pPSFactoryBuffer->lpVtbl = &CStdPSFactory_Vtbl;
137 pPSFactoryBuffer->RefCount = 0;
138 pPSFactoryBuffer->pProxyFileList = pProxyFileList;
140 if (IsEqualGUID(rclsid, pclsid))
141 return IPSFactoryBuffer_QueryInterface((LPPSFACTORYBUFFER)pPSFactoryBuffer, iid, ppv);
142 return CLASS_E_CLASSNOTAVAILABLE;
145 /***********************************************************************
146 * NdrDllCanUnloadNow [RPCRT4.@]
148 HRESULT WINAPI NdrDllCanUnloadNow(CStdPSFactoryBuffer *pPSFactoryBuffer)
150 return !(pPSFactoryBuffer->RefCount);
153 /***********************************************************************
154 * NdrDllRegisterProxy [RPCRT4.@]
156 HRESULT WINAPI NdrDllRegisterProxy(HMODULE hDll,
157 const ProxyFileInfo **pProxyFileList,
161 char keyname[120], module[120];
164 TRACE("(%p,%p,%s)\n", hDll, pProxyFileList, debugstr_guid(pclsid));
165 UuidToStringA((UUID*)pclsid, (unsigned char**)&clsid);
167 /* register interfaces to point to clsid */
168 while (*pProxyFileList) {
170 for (u=0; u<(*pProxyFileList)->TableSize; u++) {
171 CInterfaceStubVtbl *proxy = (*pProxyFileList)->pStubVtblList[u];
172 PCInterfaceName name = (*pProxyFileList)->pNamesArray[u];
175 TRACE("registering %s %s => %s\n", name, debugstr_guid(proxy->header.piid), clsid);
177 UuidToStringA((UUID*)proxy->header.piid, (unsigned char**)&iid);
178 snprintf(keyname, sizeof(keyname), "Interface\\{%s}", iid);
179 RpcStringFreeA((unsigned char**)&iid);
180 if (RegCreateKeyExA(HKEY_CLASSES_ROOT, keyname, 0, NULL, 0,
181 KEY_WRITE, NULL, &key, NULL) == ERROR_SUCCESS) {
183 RegSetValueExA(key, NULL, 0, REG_SZ, name, strlen(name));
184 if (RegCreateKeyExA(key, "ProxyStubClsid32", 0, NULL, 0,
185 KEY_WRITE, NULL, &subkey, NULL) == ERROR_SUCCESS) {
186 snprintf(module, sizeof(module), "{%s}", clsid);
187 RegSetValueExA(subkey, NULL, 0, REG_SZ, module, strlen(module));
196 /* register clsid to point to module */
197 snprintf(keyname, sizeof(keyname), "CLSID\\{%s}", clsid);
198 GetModuleFileNameA(hDll, module, sizeof(module));
199 TRACE("registering CLSID %s => %s\n", clsid, module);
200 if (RegCreateKeyExA(HKEY_CLASSES_ROOT, keyname, 0, NULL, 0,
201 KEY_WRITE, NULL, &key, NULL) == ERROR_SUCCESS) {
202 if (RegCreateKeyExA(key, "InProcServer32", 0, NULL, 0,
203 KEY_WRITE, NULL, &subkey, NULL) == ERROR_SUCCESS) {
204 RegSetValueExA(subkey, NULL, 0, REG_SZ, module, strlen(module));
211 RpcStringFreeA((unsigned char**)&clsid);
215 /***********************************************************************
216 * NdrDllUnregisterProxy [RPCRT4.@]
218 HRESULT WINAPI NdrDllUnregisterProxy(HMODULE hDll,
219 const ProxyFileInfo **pProxyFileList,
223 char keyname[120], module[120];
225 TRACE("(%p,%p,%s)\n", hDll, pProxyFileList, debugstr_guid(pclsid));
226 UuidToStringA((UUID*)pclsid, (unsigned char**)&clsid);
228 /* unregister interfaces */
229 while (*pProxyFileList) {
231 for (u=0; u<(*pProxyFileList)->TableSize; u++) {
232 CInterfaceStubVtbl *proxy = (*pProxyFileList)->pStubVtblList[u];
233 PCInterfaceName name = (*pProxyFileList)->pNamesArray[u];
236 TRACE("unregistering %s %s <= %s\n", name, debugstr_guid(proxy->header.piid), clsid);
238 UuidToStringA((UUID*)proxy->header.piid, (unsigned char**)&iid);
239 snprintf(keyname, sizeof(keyname), "Interface\\{%s}", iid);
240 RpcStringFreeA((unsigned char**)&iid);
241 RegDeleteKeyA(HKEY_CLASSES_ROOT, keyname);
246 /* unregister clsid */
247 snprintf(keyname, sizeof(keyname), "CLSID\\{%s}", clsid);
248 GetModuleFileNameA(hDll, module, sizeof(module));
249 TRACE("unregistering CLSID %s <= %s\n", clsid, module);
250 RegDeleteKeyA(HKEY_CLASSES_ROOT, keyname);
253 RpcStringFreeA((unsigned char**)&clsid);