9f053ee96867668e1459862a54931d7234eb51a0
[reactos.git] / rostests / apitests / com / com_apitest.c
1 /*
2 * PROJECT: ReactOS api tests
3 * LICENSE: GPLv2+ - See COPYING in the top level directory
4 * PURPOSE: COM interface test
5 * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
6 */
7
8 #include "com_apitest.h"
9
10 #define NDEBUG
11 #include <debug.h>
12
13 #define myskip(c, ...) ((c) ? 0 : (skip(__VA_ARGS__), 1))
14 #define mytrace(...) do { \
15 int debug = winetest_debug; \
16 winetest_debug = 1; \
17 trace(__VA_ARGS__); \
18 winetest_debug = debug; \
19 } while (0)
20
21 typedef struct _KNOWN_INTERFACE
22 {
23 const IID *iid;
24 PCSTR name;
25 PCWSTR wname;
26 BOOLEAN noreg;
27 } KNOWN_INTERFACE;
28 typedef const KNOWN_INTERFACE *PCKNOWN_INTERFACE;
29
30 #undef ID_NAME
31 #define ID_NAME(c) &c, #c, L ## #c
32 static KNOWN_INTERFACE KnownInterfaces[] =
33 {
34 { ID_NAME(IID_IACList), TRUE },
35 { ID_NAME(IID_IACList2), TRUE },
36 { ID_NAME(IID_IADesktopP2), TRUE },
37 { ID_NAME(IID_IAccessible) },
38 { ID_NAME(IID_IAccessControl), TRUE },
39 { ID_NAME(IID_IAccessor) },
40 { ID_NAME(IID_IActiveDesktop), TRUE },
41 { ID_NAME(IID_IActiveDesktopP), TRUE },
42 { ID_NAME(IID_IActionProgress) },
43 { ID_NAME(IID_IActionProgressDialog) },
44 { ID_NAME(IID_IAddressBarParser), TRUE },
45 { ID_NAME(IID_IAddressBand), TRUE },
46 { ID_NAME(IID_IAddressEditBox), TRUE },
47 { ID_NAME(IID_IAugmentedShellFolder), TRUE },
48 { ID_NAME(IID_IAugmentedShellFolder2), TRUE },
49 { ID_NAME(IID_IAutoComplete), TRUE },
50 { ID_NAME(IID_IAutoComplete2), TRUE },
51 { ID_NAME(IID_IAutoCompleteDropDown) },
52 { ID_NAME(IID_IBandHost) },
53 { ID_NAME(IID_IBandNavigate), TRUE },
54 { ID_NAME(IID_IBandProxy), TRUE },
55 { ID_NAME(IID_IBandSite) },
56 { ID_NAME(IID_IBandSiteHelper), TRUE },
57 { ID_NAME(IID_IBanneredBar), TRUE },
58 { ID_NAME(IID_IBindCtx) },
59 { ID_NAME(IID_IBrowserFrameOptions), TRUE },
60 { ID_NAME(IID_IBrowserService) },
61 { ID_NAME(IID_IBrowserService2), TRUE },
62 { ID_NAME(IID_IBrowserService3), TRUE },
63 { ID_NAME(IID_ICDBurn) },
64 { ID_NAME(IID_ICDBurnExt) },
65 { ID_NAME(IID_IClassFactory) },
66 { ID_NAME(IID_IClassFactory2) },
67 { ID_NAME(IID_IClassFactory3), TRUE },
68 { ID_NAME(IID_IClientSecurity), TRUE },
69 { ID_NAME(IID_ICommDlgBrowser) },
70 { ID_NAME(IID_ICommDlgBrowser2) },
71 { ID_NAME(IID_ICommDlgBrowser3) },
72 { ID_NAME(IID_IComputerInfoChangeNotify), },
73 { ID_NAME(IID_IComThreadingInfo), TRUE },
74 { ID_NAME(IID_IContext), TRUE },
75 { ID_NAME(IID_IContextMenu), TRUE },
76 { ID_NAME(IID_IContextMenu2), TRUE },
77 { ID_NAME(IID_IContextMenu3), TRUE },
78 { ID_NAME(IID_IContextMenuCB), TRUE },
79 { ID_NAME(IID_IContextMenuSite) },
80 { ID_NAME(IID_IContinue) },
81 { ID_NAME(IID_ICopyHookA), TRUE },
82 { ID_NAME(IID_ICopyHookW), TRUE },
83 { ID_NAME(IID_IDVGetEnum), TRUE },
84 { ID_NAME(IID_IDataObject) },
85 //{ ID_NAME(IID_IDefViewID) }, == DefViewFrame3
86 { ID_NAME(IID_IDefViewFrame), TRUE },
87 { ID_NAME(IID_IDefViewFrame3) },
88 { ID_NAME(IID_IDefViewFrameGroup) },
89 { ID_NAME(IID_IDefViewSafety), },
90 { ID_NAME(IID_IDefViewScript), TRUE },
91 { ID_NAME(IID_IDeskBand) },
92 { ID_NAME(IID_IDeskBandEx) },
93 { ID_NAME(IID_IDeskBar), TRUE },
94 { ID_NAME(IID_IDeskBarClient), TRUE },
95 { ID_NAME(IID_IDeskMovr), TRUE },
96 { ID_NAME(IID_IDispatch) },
97 { ID_NAME(IID_IDockingWindow) },
98 { ID_NAME(IID_IDockingWindowFrame), TRUE },
99 { ID_NAME(IID_IDockingWindowSite), TRUE },
100 { ID_NAME(IID_IDocViewSite), TRUE },
101 { ID_NAME(IID_IDragSourceHelper), TRUE },
102 { ID_NAME(IID_IDropSource) },
103 { ID_NAME(IID_IDropTarget) },
104 { ID_NAME(IID_IDropTargetHelper), TRUE },
105 { ID_NAME(IID_IEnumExtraSearch) },
106 { ID_NAME(IID_IEnumGUID) },
107 { ID_NAME(IID_IEnumIDList) },
108 { ID_NAME(IID_IEnumShellItems) },
109 { ID_NAME(IID_IEnumString) },
110 { ID_NAME(IID_IEnumUnknown) },
111 { ID_NAME(IID_IEnumVARIANT) },
112 { ID_NAME(IID_IErrorLog) },
113 { ID_NAME(IID_IExplorerBrowser) },
114 { ID_NAME(IID_IExplorerToolbar), TRUE },
115 { ID_NAME(IID_IExtractIconA), TRUE },
116 { ID_NAME(IID_IExtractIconW), TRUE },
117 { ID_NAME(IID_IExtractImage) },
118 { ID_NAME(IID_IExtractImage2) },
119 { ID_NAME(IID_IFileDialog) },
120 { ID_NAME(IID_IFileDialog2), TRUE },
121 { ID_NAME(IID_IFileOpenDialog) },
122 { ID_NAME(IID_IFileSaveDialog) },
123 { ID_NAME(IID_IFileSearchBand) },
124 { ID_NAME(IID_IFileViewerA), TRUE },
125 { ID_NAME(IID_IFileViewerSite), TRUE },
126 { ID_NAME(IID_IFileViewerW), TRUE },
127 { ID_NAME(IID_IFolderBandPriv) },
128 { ID_NAME(IID_IFolderFilter) },
129 { ID_NAME(IID_IFolderFilterSite) },
130 { ID_NAME(IID_IFolderView) },
131 { ID_NAME(IID_IFolderView2) },
132 { ID_NAME(IID_IFolderViewHost), TRUE },
133 { ID_NAME(IID_IFolderViewOC) },
134 { ID_NAME(IID_IFolderViewSettings) },
135 { ID_NAME(IID_IGlobalFolderSettings), TRUE },
136 { ID_NAME(IID_IHWEventHandler) },
137 { ID_NAME(IID_IHWEventHandler2) },
138 { ID_NAME(IID_IInitializeObject), TRUE },
139 { ID_NAME(IID_IInputObject) },
140 { ID_NAME(IID_IInputObjectSite) },
141 { ID_NAME(IID_IInternalUnknown), TRUE },
142 { ID_NAME(IID_IMarshal) },
143 { ID_NAME(IID_IMenuBand), TRUE },
144 { ID_NAME(IID_IMenuPopup), TRUE },
145 { ID_NAME(IID_IModalWindow) },
146 { ID_NAME(IID_IMoniker) },
147 { ID_NAME(IID_IMultiMonitorDockingSite), TRUE },
148 { ID_NAME(IID_IMultiQI), TRUE },
149 { ID_NAME(IID_INamespaceProxy), TRUE },
150 { ID_NAME(IID_INameSpaceTreeControl), TRUE },
151 { ID_NAME(IID_INSCTree), TRUE },
152 { ID_NAME(IID_INSCTree2), TRUE },
153 { ID_NAME(IID_IObjMgr), TRUE },
154 { ID_NAME(IID_IObjectSafety) },
155 { ID_NAME(IID_IObjectWithSite) },
156 { ID_NAME(IID_IOleCommandTarget) },
157 { ID_NAME(IID_IOleInPlaceActiveObject) },
158 { ID_NAME(IID_IOleInPlaceFrame) },
159 { ID_NAME(IID_IOleInPlaceObject) },
160 { ID_NAME(IID_IOleInPlaceObjectWindowless), TRUE },
161 { ID_NAME(IID_IOleInPlaceSite) },
162 { ID_NAME(IID_IOleInPlaceSiteEx) },
163 { ID_NAME(IID_IOleInPlaceSiteWindowless), TRUE },
164 { ID_NAME(IID_IOleInPlaceUIWindow) },
165 { ID_NAME(IID_IOleObject) },
166 { ID_NAME(IID_IOleWindow) },
167 { ID_NAME(IID_IPersist) },
168 { ID_NAME(IID_IPersistFile) },
169 { ID_NAME(IID_IPersistFolder) },
170 { ID_NAME(IID_IPersistFolder2) },
171 { ID_NAME(IID_IPersistFolder3) },
172 { ID_NAME(IID_IPersistHistory) },
173 { ID_NAME(IID_IPersistIDList) },
174 { ID_NAME(IID_IPersistMemory) },
175 { ID_NAME(IID_IPersistPropertyBag) },
176 { ID_NAME(IID_IPersistPropertyBag2) },
177 { ID_NAME(IID_IPersistStorage) },
178 { ID_NAME(IID_IPersistStream) },
179 { ID_NAME(IID_IPersistStreamInit) },
180 { ID_NAME(IID_IPreviewHandler) },
181 { ID_NAME(IID_IPreviewHandlerFrame) },
182 { ID_NAME(IID_IPreviewHandlerVisuals) },
183 { ID_NAME(IID_IProgressDialog), TRUE },
184 { ID_NAME(IID_IPropertyBag) },
185 { ID_NAME(IID_IPropertyBag2) },
186 { ID_NAME(IID_IQueryAssociations), TRUE },
187 { ID_NAME(IID_IQueryInfo), TRUE },
188 { ID_NAME(IID_IRegTreeOptions), TRUE },
189 { ID_NAME(IID_IRemoteComputer) },
190 { ID_NAME(IID_IResolveShellLink) },
191 { ID_NAME(IID_IRunnableObject) },
192 { ID_NAME(IID_IServerSecurity), TRUE },
193 { ID_NAME(IID_IServiceProvider) },
194 { ID_NAME(IID_ISFHelper), TRUE },
195 { ID_NAME(IID_IShellApp), TRUE },
196 { ID_NAME(IID_IShellBrowser) },
197 { ID_NAME(IID_IShellBrowserService), TRUE },
198 { ID_NAME(IID_IShellChangeNotify), TRUE },
199 { ID_NAME(IID_IShellDesktopTray), TRUE },
200 { ID_NAME(IID_IShellDetails), TRUE },
201 { ID_NAME(IID_IShellDispatch) },
202 { ID_NAME(IID_IShellDispatch2) },
203 { ID_NAME(IID_IShellDispatch3) },
204 { ID_NAME(IID_IShellDispatch4) },
205 { ID_NAME(IID_IShellDispatch5), TRUE },
206 { ID_NAME(IID_IShellExtInit), TRUE },
207 { ID_NAME(IID_IShellPropSheetExt), TRUE },
208 { ID_NAME(IID_IShellIconOverlayIdentifier), TRUE },
209 { ID_NAME(IID_IShellFolder) },
210 { ID_NAME(IID_IShellFolder2) },
211 { ID_NAME(IID_IShellFolderBand), TRUE },
212 { ID_NAME(IID_IShellFolderSearchable), TRUE },
213 { ID_NAME(IID_IShellFolderSearchableCallback), TRUE },
214 { ID_NAME(IID_IShellFolderView), TRUE },
215 { ID_NAME(IID_IShellFolderViewCB), TRUE },
216 { ID_NAME(IID_IShellFolderViewDual) },
217 { ID_NAME(IID_IShellFolderViewDual2) },
218 { ID_NAME(IID_IShellFolderViewType), TRUE },
219 { ID_NAME(IID_IShellIcon) },
220 { ID_NAME(IID_IShellIconOverlay), TRUE },
221 { ID_NAME(IID_IShellIconOverlayIdentifier), TRUE },
222 { ID_NAME(IID_IShellImageData), TRUE },
223 { ID_NAME(IID_IShellImageDataAbort), TRUE },
224 { ID_NAME(IID_IShellImageDataFactory), TRUE },
225 { ID_NAME(IID_IShellItem) },
226 { ID_NAME(IID_IShellItem2) },
227 { ID_NAME(IID_IShellItemArray) },
228 { ID_NAME(IID_IShellItemFilter) },
229 { ID_NAME(IID_IShellLinkA) },
230 { ID_NAME(IID_IShellLinkDataList), TRUE },
231 { ID_NAME(IID_IShellLinkDual) },
232 { ID_NAME(IID_IShellLinkDual2) },
233 { ID_NAME(IID_IShellLinkW) },
234 { ID_NAME(IID_IShellMenu), TRUE },
235 { ID_NAME(IID_IShellMenu2), TRUE },
236 { ID_NAME(IID_IShellMenuAcc), TRUE },
237 { ID_NAME(IID_IShellMenuCallback), TRUE },
238 { ID_NAME(IID_IShellNameSpace) },
239 { ID_NAME(IID_IShellPropSheetExt), TRUE },
240 { ID_NAME(IID_IShellService), TRUE },
241 { ID_NAME(IID_IShellTaskScheduler), TRUE },
242 { ID_NAME(IID_IShellView) },
243 { ID_NAME(IID_IShellView2) },
244 { ID_NAME(IID_IShellView3) },
245 { ID_NAME(IID_IShellWindows) },
246 { ID_NAME(IID_IStorage) },
247 { ID_NAME(IID_IStream) },
248 { ID_NAME(IID_ISurrogate) },
249 { ID_NAME(IID_ISynchronize) },
250 { ID_NAME(IID_ISynchronizeContainer), TRUE },
251 { ID_NAME(IID_ISynchronizeEvent), TRUE },
252 { ID_NAME(IID_ISynchronizeHandle), TRUE },
253 { ID_NAME(IID_ITaskbarList) },
254 { ID_NAME(IID_ITaskbarList2) },
255 { ID_NAME(IID_ITrackShellMenu), TRUE },
256 { ID_NAME(IID_ITranslateShellChangeNotify), TRUE },
257 { ID_NAME(IID_ITrayPriv), TRUE },
258 { ID_NAME(IID_ITrayPriv2), TRUE },
259 { ID_NAME(IID_IUnknown) },
260 { ID_NAME(IID_IViewObject) },
261 { ID_NAME(IID_IViewObject2) },
262 { ID_NAME(IID_IViewObjectEx), TRUE },
263 { ID_NAME(IID_IWinEventHandler), TRUE },
264
265 { ID_NAME(IID_DFConstraint), TRUE },
266 { ID_NAME(DIID_DShellFolderViewEvents) },
267
268 { ID_NAME(IID_CDefView), TRUE },
269 { ID_NAME(IID_Folder) },
270 { ID_NAME(IID_Folder2) },
271 { ID_NAME(IID_Folder3) },
272 { ID_NAME(IID_FolderItem) },
273 { ID_NAME(IID_FolderItem2) },
274 { ID_NAME(IID_FolderItems) },
275 { ID_NAME(IID_FolderItems2) },
276 { ID_NAME(IID_FolderItems3) },
277 { ID_NAME(IID_FolderItemVerb) },
278 { ID_NAME(IID_FolderItemVerbs) },
279
280 { ID_NAME(CLSID_ShellDesktop), TRUE }
281 };
282 static const INT KnownInterfaceCount = RTL_NUMBER_OF(KnownInterfaces);
283
284 static
285 PCKNOWN_INTERFACE
286 FindInterface(
287 _In_ const IID *iid)
288 {
289 INT i;
290
291 for (i = 0; i < KnownInterfaceCount; i++)
292 if (IsEqualIID(KnownInterfaces[i].iid, iid))
293 return &KnownInterfaces[i];
294 ASSERT(i != KnownInterfaceCount);
295 return NULL;
296 }
297
298 static
299 BOOLEAN
300 IsInterfaceExpected(
301 _In_ PCCLASS_AND_INTERFACES class,
302 _In_ const IID *iid)
303 {
304 INT i;
305
306 for (i = 0; class->ifaces[i].iid; i++)
307 if (IsEqualIID(class->ifaces[i].iid, iid))
308 return TRUE;
309 return FALSE;
310 }
311
312 #define INTF_NOT_EXPOSED LONG_MAX
313 static
314 LONG
315 GetInterfaceOffset(
316 _In_ PUNKNOWN pUnk,
317 _In_ const IID *iid)
318 {
319 HRESULT hr;
320 PVOID pObj;
321 PUNKNOWN pUnk2;
322 LONG offset;
323
324 hr = IUnknown_QueryInterface(pUnk, iid, &pObj);
325 ok(hr == S_OK || hr == E_NOINTERFACE, "IUnknown::QueryInterface returned 0x%lx\n", hr);
326 if (FAILED(hr))
327 return INTF_NOT_EXPOSED;
328
329 pUnk2 = pObj;
330 offset = (LONG_PTR)pObj - (LONG_PTR)pUnk;
331 IUnknown_Release(pUnk2);
332 return offset;
333 }
334
335 static
336 VOID
337 TestModuleInterfaces(
338 _In_ PCCLASS_AND_INTERFACES ExpectedInterfaces,
339 _In_ INT ExpectedInterfaceCount)
340 {
341 HRESULT hr;
342 PVOID pObj;
343 PUNKNOWN pUnk;
344 INT iClass, iIntf;
345 PCCLASS_AND_INTERFACES class;
346
347 for (iClass = 0; iClass < ExpectedInterfaceCount; iClass++)
348 {
349 class = &ExpectedInterfaces[iClass];
350 hr = CoCreateInstance(class->clsid,
351 NULL,
352 CLSCTX_INPROC_SERVER,
353 &IID_IUnknown,
354 &pObj);
355 ok(hr == S_OK, "CoCreateInstance failed. hr=0x%lx\n", hr);
356 if (FAILED(hr))
357 {
358 skip("Failed to instantiate %s.\n", class->name);
359 continue;
360 }
361
362 pUnk = pObj;
363
364 /* Check that all expected interfaces are present and have the right offset */
365 for (iIntf = 0; class->ifaces[iIntf].iid; iIntf++)
366 {
367 PCKNOWN_INTERFACE iface = FindInterface(class->ifaces[iIntf].iid);
368 LONG offset = GetInterfaceOffset(pUnk, iface->iid);
369 if (offset == INTF_NOT_EXPOSED)
370 ok(0, "%s is missing %s (offset %ld)\n", class->name, iface->name, class->ifaces[iIntf].offset);
371 else if (class->ifaces[iIntf].offset != FARAWY)
372 {
373 #ifdef FAIL_WRONG_OFFSET
374 ok(offset == class->ifaces[iIntf].offset, "%s, %s offset is %ld, expected %ld\n", class->name, iface->name, offset, class->ifaces[iIntf].offset);
375 #else
376 if (offset != class->ifaces[iIntf].offset)
377 mytrace("%s, %s offset is %ld, expected %ld\n", class->name, iface->name, offset, class->ifaces[iIntf].offset);
378 #endif
379 }
380 }
381
382 /* Check that none other than the expected interfaces are present */
383 for (iIntf = 0; iIntf < KnownInterfaceCount; iIntf++)
384 {
385 PCKNOWN_INTERFACE iface = &KnownInterfaces[iIntf];
386 LONG offset;
387 if (IsInterfaceExpected(class, iface->iid))
388 continue;
389 offset = GetInterfaceOffset(pUnk, iface->iid);
390 #ifdef GENERATE_TABLE_ENTRIES
391 ok(offset == INTF_NOT_EXPOSED, "%s: { %s0x%x, &%s },\n", class->name, offset < 0 ? "-" : "", offset < 0 ? -offset : offset, iface->name);
392 #else
393 ok(offset == INTF_NOT_EXPOSED, "%s exposes %s (offset %ld), but shouldn't\n", class->name, iface->name, offset);
394 #endif
395 }
396
397 // TODO: do some aggregation
398
399 IUnknown_Release(pUnk);
400 }
401 }
402
403 static
404 VOID
405 TestModuleRegistry(
406 _In_ PCWSTR ModuleName,
407 _In_ PCCLASS_AND_INTERFACES ExpectedInterfaces,
408 _In_ INT ExpectedInterfaceCount)
409 {
410 INT iClass;
411 PCCLASS_AND_INTERFACES class;
412 HKEY hKeyClasses;
413 LONG result;
414
415 result = RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Classes\\CLSID", 0, KEY_ENUMERATE_SUB_KEYS, &hKeyClasses);
416 ok(result == NO_ERROR, "Failed to open classes key, error %lu\n", result);
417 if (!myskip(result == NO_ERROR, "No classes key\n"))
418 {
419 for (iClass = 0; iClass < ExpectedInterfaceCount; iClass++)
420 {
421 HKEY hKey;
422 HKEY hKeyServer;
423 NTSTATUS status;
424 UNICODE_STRING clsid;
425 DWORD type;
426 WCHAR data[100];
427 DWORD dataSize;
428 PCWSTR expectedThreadingModel;
429
430 class = &ExpectedInterfaces[iClass];
431 status = RtlStringFromGUID(class->clsid, &clsid);
432 ok(status == STATUS_SUCCESS, "Failed to convert guid to string for %s, status %lx\n", class->name, status);
433 if (myskip(NT_SUCCESS(status), "No guid string\n"))
434 continue;
435
436 result = RegOpenKeyEx(hKeyClasses, clsid.Buffer, 0, KEY_ENUMERATE_SUB_KEYS, &hKey);
437 ok(result == NO_ERROR, "Failed to open key for %s, error %lu\n", class->name, result);
438 RtlFreeUnicodeString(&clsid);
439 if (myskip(result == NO_ERROR, "No key\n"))
440 continue;
441
442 result = RegOpenKeyEx(hKey, L"InProcServer32", 0, KEY_QUERY_VALUE, &hKeyServer);
443 ok(result == NO_ERROR, "Failed to open key for %s, error %lu\n", class->name, result);
444 RegCloseKey(hKey);
445 if (myskip(result == NO_ERROR, "No key\n"))
446 continue;
447
448 dataSize = sizeof(data);
449 result = RegQueryValueEx(hKeyServer, NULL, NULL, &type, (PBYTE)data, &dataSize);
450 ok(result == NO_ERROR, "Failed to query value for %s, error %lu\n", class->name, result);
451 if (!myskip(result == NO_ERROR, "No module name\n"))
452 {
453 ok(type == REG_SZ || type == REG_EXPAND_SZ, "type %lu for %s\n", type, class->name);
454 ok(dataSize % sizeof(WCHAR) == 0, "size %lu for %s\n", dataSize, class->name);
455 ok(dataSize <= sizeof(data), "size %lu for %s\n", dataSize, class->name);
456 ok(data[dataSize / sizeof(WCHAR) - 1] == UNICODE_NULL, "Not null terminated for %s\n", class->name);
457 // TODO: Use SearchPath (or assume everything's in system32) and do a proper full path compare
458 PathStripPathW(data);
459 PathRemoveExtensionW(data);
460 ok(!wcsicmp(data, ModuleName), "Server is %ls, expected %ls for %s\n", data, ModuleName, class->name);
461 }
462
463 dataSize = sizeof(data);
464 result = RegQueryValueEx(hKeyServer, L"ThreadingModel", NULL, &type, (PBYTE)data, &dataSize);
465 ok(result == NO_ERROR, "Failed to query value for %s, error %lu\n", class->name, result);
466 if (!myskip(result == NO_ERROR, "No ThreadingModel\n"))
467 {
468 ok(type == REG_SZ || type == REG_EXPAND_SZ, "type %lu for %s\n", type, class->name);
469 ok(dataSize % sizeof(WCHAR) == 0, "size %lu for %s\n", dataSize, class->name);
470 ok(dataSize <= sizeof(data), "size %lu for %s\n", dataSize, class->name);
471 ok(data[dataSize / sizeof(WCHAR) - 1] == UNICODE_NULL, "Not null terminated for %s\n", class->name);
472 expectedThreadingModel = class->ThreadingModel;
473 if (!expectedThreadingModel)
474 expectedThreadingModel = L"Apartment";
475 ok(!wcsicmp(data, expectedThreadingModel), "Server is %ls, expected %ls for %s\n", data, expectedThreadingModel, class->name);
476 }
477
478 RegCloseKey(hKeyServer);
479 }
480 RegCloseKey(hKeyClasses);
481 }
482 }
483
484 static
485 VOID
486 TestManualInstantiation(
487 _In_ PCWSTR ModuleName,
488 _In_ PCCLASS_AND_INTERFACES ExpectedInterfaces,
489 _In_ INT ExpectedInterfaceCount)
490 {
491 INT iClass;
492 PCCLASS_AND_INTERFACES class;
493 HRESULT (__stdcall *DllGetClassObject)(REFCLSID, REFIID, PVOID *);
494
495 DllGetClassObject = (PVOID)GetProcAddress(GetModuleHandle(ModuleName), "DllGetClassObject");
496 ok(DllGetClassObject != NULL, "DllGetClassObject not found in %ls, error %lu\n", ModuleName, GetLastError());
497 if (myskip(DllGetClassObject != NULL, "No DllGetClassObject\n"))
498 return;
499
500 for (iClass = 0; iClass < ExpectedInterfaceCount; iClass++)
501 {
502 PVOID pv;
503 HRESULT hr;
504 class = &ExpectedInterfaces[iClass];
505 hr = DllGetClassObject(class->clsid, &IID_IClassFactory, &pv);
506 ok(hr == S_OK, "DllGetClassObject failed for %s, hr = 0x%lx\n", class->name, hr);
507 if (!myskip(SUCCEEDED(hr), "No class factory\n"))
508 {
509 IClassFactory *pCF = pv;
510 hr = IClassFactory_CreateInstance(pCF, NULL, &IID_IUnknown, &pv);
511 ok(hr == S_OK, "IClassFactory::CreateInstance failed for %s, hr = 0x%lx\n", class->name, hr);
512 if (!myskip(SUCCEEDED(hr), "No instance\n"))
513 {
514 IUnknown *pUnk = pv;
515 IUnknown_Release(pUnk);
516 }
517 IClassFactory_Release(pCF);
518 }
519 }
520 }
521
522 VOID
523 TestClasses(
524 _In_ PCWSTR ModuleName,
525 _In_ PCCLASS_AND_INTERFACES ExpectedInterfaces,
526 _In_ INT ExpectedInterfaceCount)
527 {
528 HRESULT hr;
529
530 hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
531 ok(hr == S_OK, "CoInitializeEx failed. hr=0x%lx\n", hr);
532 if (myskip(SUCCEEDED(hr), "Failed to initialize COM. Cannot perform tests\n"))
533 return;
534
535 TestModuleInterfaces(ExpectedInterfaces, ExpectedInterfaceCount);
536 TestModuleRegistry(ModuleName, ExpectedInterfaces, ExpectedInterfaceCount);
537 TestManualInstantiation(ModuleName, ExpectedInterfaces, ExpectedInterfaceCount);
538
539 CoUninitialize();
540 }
541
542 static
543 VOID
544 TestInterfaceRegistry(
545 _In_ PCKNOWN_INTERFACE Interfaces,
546 _In_ INT InterfaceCount)
547 {
548 INT i;
549 HKEY hKeyInterface;
550 LONG result;
551
552 result = RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Classes\\Interface", 0, KEY_ENUMERATE_SUB_KEYS, &hKeyInterface);
553 ok(result == NO_ERROR, "Failed to open interface key, error %lu\n", result);
554 if (!myskip(result == NO_ERROR, "No interface key\n"))
555 {
556 for (i = 0; i < InterfaceCount; i++)
557 {
558 HKEY hKey;
559 NTSTATUS status;
560 UNICODE_STRING iid;
561 DWORD type;
562 WCHAR data[100];
563 DWORD dataSize;
564 PCKNOWN_INTERFACE iface;
565 PCWSTR expectedName;
566
567 iface = &Interfaces[i];
568 status = RtlStringFromGUID(iface->iid, &iid);
569 ok(status == STATUS_SUCCESS, "Failed to convert guid to string for %s, status %lx\n", iface->name, status);
570 if (myskip(NT_SUCCESS(status), "No guid string\n"))
571 continue;
572
573 result = RegOpenKeyEx(hKeyInterface, iid.Buffer, 0, KEY_QUERY_VALUE, &hKey);
574 if (iface->noreg)
575 {
576 ok(result == ERROR_FILE_NOT_FOUND, "RegOpenKeyEx returned %lu for %s\n", result, iface->name);
577 }
578 else
579 {
580 ok(result == NO_ERROR, "Failed to open key for %s, error %lu\n", iface->name, result);
581 (void)myskip(result == NO_ERROR, "No key\n");
582 }
583 RtlFreeUnicodeString(&iid);
584 if (result != NO_ERROR)
585 continue;
586
587 dataSize = sizeof(data);
588 result = RegQueryValueEx(hKey, NULL, NULL, &type, (PBYTE)data, &dataSize);
589 ok(result == NO_ERROR, "Failed to query value for %s, error %lu\n", iface->name, result);
590 if (!myskip(result == NO_ERROR, "No module name\n"))
591 {
592 ok(type == REG_SZ, "type %lu for %s\n", type, iface->name);
593 ok(dataSize % sizeof(WCHAR) == 0, "size %lu for %s\n", dataSize, iface->name);
594 ok(dataSize <= sizeof(data), "size %lu for %s\n", dataSize, iface->name);
595 ok(data[dataSize / sizeof(WCHAR) - 1] == UNICODE_NULL, "Not null terminated for %s\n", iface->name);
596 expectedName = wcschr(iface->wname, L'_');
597 if (expectedName)
598 expectedName++;
599 else
600 expectedName = iface->wname;
601 ok(!wcsicmp(data, expectedName), "Name is %ls, expected %ls\n", data, expectedName);
602 }
603
604 RegCloseKey(hKey);
605 }
606 RegCloseKey(hKeyInterface);
607 }
608 }
609
610 START_TEST(interfaces)
611 {
612 TestInterfaceRegistry(KnownInterfaces, KnownInterfaceCount);
613 }