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
34 #include "wine/debug.h"
38 WINE_DEFAULT_DEBUG_CHANNEL(ole
);
40 static BOOL
FindProxyInfo(const ProxyFileInfo
**pProxyFileList
, REFIID riid
, const ProxyFileInfo
**pProxyInfo
, int *pIndex
)
42 while (*pProxyFileList
) {
43 if ((*pProxyFileList
)->pIIDLookupRtn(riid
, pIndex
)) {
44 *pProxyInfo
= *pProxyFileList
;
45 TRACE("found: ProxyInfo %p Index %d\n", *pProxyInfo
, *pIndex
);
54 static HRESULT WINAPI
CStdPSFactory_QueryInterface(LPPSFACTORYBUFFER iface
,
58 CStdPSFactoryBuffer
*This
= (CStdPSFactoryBuffer
*)iface
;
59 TRACE("(%p)->QueryInterface(%s,%p)\n",iface
,debugstr_guid(riid
),obj
);
60 if (IsEqualGUID(&IID_IUnknown
,riid
) ||
61 IsEqualGUID(&IID_IPSFactoryBuffer
,riid
)) {
69 static ULONG WINAPI
CStdPSFactory_AddRef(LPPSFACTORYBUFFER iface
)
71 CStdPSFactoryBuffer
*This
= (CStdPSFactoryBuffer
*)iface
;
72 TRACE("(%p)->AddRef()\n",iface
);
73 return ++(This
->RefCount
);
76 static ULONG WINAPI
CStdPSFactory_Release(LPPSFACTORYBUFFER iface
)
78 CStdPSFactoryBuffer
*This
= (CStdPSFactoryBuffer
*)iface
;
79 TRACE("(%p)->Release()\n",iface
);
80 return --(This
->RefCount
);
83 static HRESULT WINAPI
CStdPSFactory_CreateProxy(LPPSFACTORYBUFFER iface
,
86 LPRPCPROXYBUFFER
*ppProxy
,
89 CStdPSFactoryBuffer
*This
= (CStdPSFactoryBuffer
*)iface
;
90 const ProxyFileInfo
*ProxyInfo
;
92 TRACE("(%p)->CreateProxy(%p,%s,%p,%p)\n",iface
,pUnkOuter
,
93 debugstr_guid(riid
),ppProxy
,ppv
);
94 if (!FindProxyInfo(This
->pProxyFileList
,riid
,&ProxyInfo
,&Index
))
96 return StdProxy_Construct(riid
, pUnkOuter
, ProxyInfo
->pNamesArray
[Index
],
97 ProxyInfo
->pProxyVtblList
[Index
],
98 ProxyInfo
->pStubVtblList
[Index
], iface
, ppProxy
, ppv
);
101 static HRESULT WINAPI
CStdPSFactory_CreateStub(LPPSFACTORYBUFFER iface
,
103 LPUNKNOWN pUnkServer
,
104 LPRPCSTUBBUFFER
*ppStub
)
106 CStdPSFactoryBuffer
*This
= (CStdPSFactoryBuffer
*)iface
;
107 const ProxyFileInfo
*ProxyInfo
;
109 TRACE("(%p)->CreateStub(%s,%p,%p)\n",iface
,debugstr_guid(riid
),
111 if (!FindProxyInfo(This
->pProxyFileList
,riid
,&ProxyInfo
,&Index
))
112 return E_NOINTERFACE
;
113 return CStdStubBuffer_Construct(riid
, pUnkServer
, ProxyInfo
->pNamesArray
[Index
],
114 ProxyInfo
->pStubVtblList
[Index
], iface
, ppStub
);
117 static IPSFactoryBufferVtbl CStdPSFactory_Vtbl
=
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
[MAX_PATH
];
165 TRACE("(%p,%p,%s)\n", hDll
, pProxyFileList
, debugstr_guid(pclsid
));
166 UuidToStringA((UUID
*)pclsid
, (unsigned char**)&clsid
);
168 /* register interfaces to point to clsid */
169 while (*pProxyFileList
) {
171 for (u
=0; u
<(*pProxyFileList
)->TableSize
; u
++) {
172 CInterfaceStubVtbl
*proxy
= (*pProxyFileList
)->pStubVtblList
[u
];
173 PCInterfaceName name
= (*pProxyFileList
)->pNamesArray
[u
];
176 TRACE("registering %s %s => %s\n", name
, debugstr_guid(proxy
->header
.piid
), clsid
);
178 UuidToStringA((UUID
*)proxy
->header
.piid
, (unsigned char**)&iid
);
179 snprintf(keyname
, sizeof(keyname
), "Interface\\{%s}", iid
);
180 RpcStringFreeA((unsigned char**)&iid
);
181 if (RegCreateKeyExA(HKEY_CLASSES_ROOT
, keyname
, 0, NULL
, 0,
182 KEY_WRITE
, NULL
, &key
, NULL
) == ERROR_SUCCESS
) {
184 RegSetValueExA(key
, NULL
, 0, REG_SZ
, name
, strlen(name
));
185 if (RegCreateKeyExA(key
, "ProxyStubClsid32", 0, NULL
, 0,
186 KEY_WRITE
, NULL
, &subkey
, NULL
) == ERROR_SUCCESS
) {
187 snprintf(module
, sizeof(module
), "{%s}", clsid
);
188 RegSetValueExA(subkey
, NULL
, 0, REG_SZ
, module
, strlen(module
));
197 /* register clsid to point to module */
198 snprintf(keyname
, sizeof(keyname
), "CLSID\\{%s}", clsid
);
199 len
= GetModuleFileNameA(hDll
, module
, sizeof(module
));
200 if (len
&& len
< sizeof(module
)) {
201 TRACE("registering CLSID %s => %s\n", clsid
, module
);
202 if (RegCreateKeyExA(HKEY_CLASSES_ROOT
, keyname
, 0, NULL
, 0,
203 KEY_WRITE
, NULL
, &key
, NULL
) == ERROR_SUCCESS
) {
204 if (RegCreateKeyExA(key
, "InProcServer32", 0, NULL
, 0,
205 KEY_WRITE
, NULL
, &subkey
, NULL
) == ERROR_SUCCESS
) {
206 RegSetValueExA(subkey
, NULL
, 0, REG_SZ
, module
, strlen(module
));
214 RpcStringFreeA((unsigned char**)&clsid
);
218 /***********************************************************************
219 * NdrDllUnregisterProxy [RPCRT4.@]
221 HRESULT WINAPI
NdrDllUnregisterProxy(HMODULE hDll
,
222 const ProxyFileInfo
**pProxyFileList
,
226 char keyname
[120], module
[MAX_PATH
];
229 TRACE("(%p,%p,%s)\n", hDll
, pProxyFileList
, debugstr_guid(pclsid
));
230 UuidToStringA((UUID
*)pclsid
, (unsigned char**)&clsid
);
232 /* unregister interfaces */
233 while (*pProxyFileList
) {
235 for (u
=0; u
<(*pProxyFileList
)->TableSize
; u
++) {
236 CInterfaceStubVtbl
*proxy
= (*pProxyFileList
)->pStubVtblList
[u
];
237 PCInterfaceName name
= (*pProxyFileList
)->pNamesArray
[u
];
240 TRACE("unregistering %s %s <= %s\n", name
, debugstr_guid(proxy
->header
.piid
), clsid
);
242 UuidToStringA((UUID
*)proxy
->header
.piid
, (unsigned char**)&iid
);
243 snprintf(keyname
, sizeof(keyname
), "Interface\\{%s}", iid
);
244 RpcStringFreeA((unsigned char**)&iid
);
245 RegDeleteKeyA(HKEY_CLASSES_ROOT
, keyname
);
250 /* unregister clsid */
251 snprintf(keyname
, sizeof(keyname
), "CLSID\\{%s}", clsid
);
252 len
= GetModuleFileNameA(hDll
, module
, sizeof(module
));
253 if (len
&& len
< sizeof(module
)) {
254 TRACE("unregistering CLSID %s <= %s\n", clsid
, module
);
255 RegDeleteKeyA(HKEY_CLASSES_ROOT
, keyname
);
259 RpcStringFreeA((unsigned char**)&clsid
);