1 /*
2  * Misc tests
3  *
4  * Copyright 2006 Paul Vriens
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20 
21 #include <stdio.h>
22 #include <windows.h>
23 #include <commctrl.h>
24 
25 #include "wine/test.h"
26 #include "v6util.h"
27 
28 static PVOID (WINAPI * pAlloc)(LONG);
29 static PVOID (WINAPI * pReAlloc)(PVOID, LONG);
30 static BOOL (WINAPI * pFree)(PVOID);
31 static LONG (WINAPI * pGetSize)(PVOID);
32 
33 static INT (WINAPI * pStr_GetPtrA)(LPCSTR, LPSTR, INT);
34 static BOOL (WINAPI * pStr_SetPtrA)(LPSTR, LPCSTR);
35 static INT (WINAPI * pStr_GetPtrW)(LPCWSTR, LPWSTR, INT);
36 static BOOL (WINAPI * pStr_SetPtrW)(LPWSTR, LPCWSTR);
37 
38 static HMODULE hComctl32 = 0;
39 
40 static char testicon_data[] =
41 {
42     0x00, 0x00, 0x01, 0x00, 0x01, 0x00, 0x02, 0x02, 0x00, 0x00, 0x01, 0x00,
43     0x20, 0x00, 0x40, 0x00, 0x00, 0x00, 0x16, 0x00, 0x00, 0x00, 0x28, 0x00,
44     0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x01, 0x00,
45     0x20, 0x00, 0x00, 0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x12, 0x0b,
46     0x00, 0x00, 0x12, 0x0b, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
47     0x00, 0x00, 0xde, 0xde, 0xde, 0xff, 0xde, 0xde, 0xde, 0xff, 0xde, 0xde,
48     0xde, 0xff, 0xde, 0xde, 0xde, 0xff, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
49     0x00, 0x00
50 };
51 
52 #define COMCTL32_GET_PROC(ordinal, func) \
53     p ## func = (void*)GetProcAddress(hComctl32, (LPSTR)ordinal); \
54     if(!p ## func) { \
55       trace("GetProcAddress(%d)(%s) failed\n", ordinal, #func); \
56       FreeLibrary(hComctl32); \
57     }
58 
59 static BOOL InitFunctionPtrs(void)
60 {
61     hComctl32 = LoadLibraryA("comctl32.dll");
62 
63     if(!hComctl32)
64     {
65         trace("Could not load comctl32.dll\n");
66         return FALSE;
67     }
68 
69     COMCTL32_GET_PROC(71, Alloc);
70     COMCTL32_GET_PROC(72, ReAlloc);
71     COMCTL32_GET_PROC(73, Free);
72     COMCTL32_GET_PROC(74, GetSize);
73 
74     COMCTL32_GET_PROC(233, Str_GetPtrA)
75     COMCTL32_GET_PROC(234, Str_SetPtrA)
76     COMCTL32_GET_PROC(235, Str_GetPtrW)
77     COMCTL32_GET_PROC(236, Str_SetPtrW)
78 
79     return TRUE;
80 }
81 
82 static void test_GetPtrAW(void)
83 {
84     if (pStr_GetPtrA)
85     {
86         static const char source[] = "Just a source string";
87         static const char desttest[] = "Just a destination string";
88         static char dest[MAX_PATH];
89         int sourcelen;
90         int destsize = MAX_PATH;
91         int count;
92 
93         sourcelen = strlen(source) + 1;
94 
95         count = pStr_GetPtrA(NULL, NULL, 0);
96         ok (count == 0, "Expected count to be 0, it was %d\n", count);
97 
98         if (0)
99         {
100             /* Crashes on W98, NT4, W2K, XP, W2K3
101              * Our implementation also crashes and we should probably leave
102              * it like that.
103              */
104             count = pStr_GetPtrA(NULL, NULL, destsize);
105             trace("count : %d\n", count);
106         }
107 
108         count = pStr_GetPtrA(source, NULL, 0);
109         ok (count == sourcelen ||
110             broken(count == sourcelen - 1), /* win9x */
111             "Expected count to be %d, it was %d\n", sourcelen, count);
112 
113         strcpy(dest, desttest);
114         count = pStr_GetPtrA(source, dest, 0);
115         ok (count == sourcelen ||
116             broken(count == 0), /* win9x */
117             "Expected count to be %d, it was %d\n", sourcelen, count);
118         ok (!lstrcmpA(dest, desttest) ||
119             broken(!lstrcmpA(dest, "")), /* Win7 */
120             "Expected destination to not have changed\n");
121 
122         count = pStr_GetPtrA(source, NULL, destsize);
123         ok (count == sourcelen ||
124             broken(count == sourcelen - 1), /* win9x */
125             "Expected count to be %d, it was %d\n", sourcelen, count);
126 
127         count = pStr_GetPtrA(source, dest, destsize);
128         ok (count == sourcelen ||
129             broken(count == sourcelen - 1), /* win9x */
130             "Expected count to be %d, it was %d\n", sourcelen, count);
131         ok (!lstrcmpA(source, dest), "Expected source and destination to be the same\n");
132 
133         strcpy(dest, desttest);
134         count = pStr_GetPtrA(NULL, dest, destsize);
135         ok (count == 0, "Expected count to be 0, it was %d\n", count);
136         ok (dest[0] == '\0', "Expected destination to be cut-off and 0 terminated\n");
137 
138         destsize = 15;
139         count = pStr_GetPtrA(source, dest, destsize);
140         ok (count == 15 ||
141             broken(count == 14), /* win9x */
142             "Expected count to be 15, it was %d\n", count);
143         ok (!memcmp(source, dest, 14), "Expected first part of source and destination to be the same\n");
144         ok (dest[14] == '\0', "Expected destination to be cut-off and 0 terminated\n");
145     }
146 }
147 
148 static void test_Alloc(void)
149 {
150     PCHAR p;
151     BOOL res;
152     DWORD size, min;
153 
154     /* allocate size 0 */
155     p = pAlloc(0);
156     ok(p != NULL, "Expected non-NULL ptr\n");
157 
158     /* get the minimum size */
159     min = pGetSize(p);
160 
161     /* free the block */
162     res = pFree(p);
163     ok(res == TRUE, "Expected TRUE, got %d\n", res);
164 
165     /* allocate size 1 */
166     p = pAlloc(1);
167     ok(p != NULL, "Expected non-NULL ptr\n");
168 
169     /* get the allocated size */
170     size = pGetSize(p);
171     ok(size == 1 ||
172        broken(size == min), /* win9x */
173        "Expected 1, got %d\n", size);
174 
175     /* reallocate the block */
176     p = pReAlloc(p, 2);
177     ok(p != NULL, "Expected non-NULL ptr\n");
178 
179     /* get the new size */
180     size = pGetSize(p);
181     ok(size == 2 ||
182        broken(size == min), /* win9x */
183        "Expected 2, got %d\n", size);
184 
185     /* free the block */
186     res = pFree(p);
187     ok(res == TRUE, "Expected TRUE, got %d\n", res);
188 
189     /* free a NULL ptr */
190     res = pFree(NULL);
191     ok(res == TRUE ||
192        broken(res == FALSE), /* win9x */
193        "Expected TRUE, got %d\n", res);
194 
195     /* reallocate a NULL ptr */
196     p = pReAlloc(NULL, 2);
197     ok(p != NULL, "Expected non-NULL ptr\n");
198 
199     res = pFree(p);
200     ok(res == TRUE, "Expected TRUE, got %d\n", res);
201 }
202 
203 static void test_LoadIconWithScaleDown(void)
204 {
205     static const WCHAR nonexisting_fileW[] = {'n','o','n','e','x','i','s','t','i','n','g','.','i','c','o',0};
206     static const WCHAR nonexisting_resourceW[] = {'N','o','n','e','x','i','s','t','i','n','g',0};
207     static const WCHAR prefixW[] = {'I','C','O',0};
208     HRESULT (WINAPI *pLoadIconMetric)(HINSTANCE, const WCHAR *, int, HICON *);
209     HRESULT (WINAPI *pLoadIconWithScaleDown)(HINSTANCE, const WCHAR *, int, int, HICON *);
210     WCHAR tmp_path[MAX_PATH], icon_path[MAX_PATH];
211     ICONINFO info;
212     HMODULE hinst;
213     HANDLE handle;
214     DWORD written;
215     HRESULT hr;
216     BITMAP bmp;
217     HICON icon;
218     void *ptr;
219     int bytes;
220     BOOL res;
221 
222     hinst = LoadLibraryA("comctl32.dll");
223     pLoadIconMetric        = (void *)GetProcAddress(hinst, "LoadIconMetric");
224     pLoadIconWithScaleDown = (void *)GetProcAddress(hinst, "LoadIconWithScaleDown");
225     if (!pLoadIconMetric || !pLoadIconWithScaleDown)
226     {
227         win_skip("LoadIconMetric or pLoadIconWithScaleDown not exported by name\n");
228         FreeLibrary(hinst);
229         return;
230     }
231 
232     GetTempPathW(MAX_PATH, tmp_path);
233     GetTempFileNameW(tmp_path, prefixW, 0, icon_path);
234     handle = CreateFileW(icon_path, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
235                          FILE_ATTRIBUTE_NORMAL, NULL);
236     ok(handle != INVALID_HANDLE_VALUE, "CreateFileW failed with error %u\n", GetLastError());
237     res = WriteFile(handle, testicon_data, sizeof(testicon_data), &written, NULL);
238     ok(res && written == sizeof(testicon_data), "Failed to write icon file\n");
239     CloseHandle(handle);
240 
241     /* test ordinals */
242     ptr = GetProcAddress(hinst, (const char *)380);
243     ok(ptr == pLoadIconMetric,
244        "got wrong pointer for ordinal 380, %p expected %p\n", ptr, pLoadIconMetric);
245 
246     ptr = GetProcAddress(hinst, (const char *)381);
247     ok(ptr == pLoadIconWithScaleDown,
248        "got wrong pointer for ordinal 381, %p expected %p\n", ptr, pLoadIconWithScaleDown);
249 
250     /* invalid arguments */
251     icon = (HICON)0x1234;
252     hr = pLoadIconMetric(NULL, (LPWSTR)IDI_APPLICATION, 0x100, &icon);
253     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %x\n", hr);
254     ok(icon == NULL, "Expected NULL, got %p\n", icon);
255 
256     icon = (HICON)0x1234;
257     hr = pLoadIconMetric(NULL, NULL, LIM_LARGE, &icon);
258     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %x\n", hr);
259     ok(icon == NULL, "Expected NULL, got %p\n", icon);
260 
261     icon = (HICON)0x1234;
262     hr = pLoadIconWithScaleDown(NULL, NULL, 32, 32, &icon);
263     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %x\n", hr);
264     ok(icon == NULL, "Expected NULL, got %p\n", icon);
265 
266     /* non-existing filename */
267     hr = pLoadIconMetric(NULL, nonexisting_fileW, LIM_LARGE, &icon);
268     ok(hr == HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND) || hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) /* Win7 */,
269        "Expected HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND), got %x\n", hr);
270 
271     hr = pLoadIconWithScaleDown(NULL, nonexisting_fileW, 32, 32, &icon);
272     todo_wine
273     ok(hr == HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND),
274        "Expected HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND), got %x\n", hr);
275 
276     /* non-existing resource name */
277     hr = pLoadIconMetric(hinst, nonexisting_resourceW, LIM_LARGE, &icon);
278     ok(hr == HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND),
279        "Expected HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND), got %x\n", hr);
280 
281     hr = pLoadIconWithScaleDown(hinst, nonexisting_resourceW, 32, 32, &icon);
282     ok(hr == HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND),
283        "Expected HRESULT_FROM_WIN32(ERROR_RESOURCE_TYPE_NOT_FOUND), got %x\n", hr);
284 
285     /* load icon using predefined identifier */
286     hr = pLoadIconMetric(NULL, (LPWSTR)IDI_APPLICATION, LIM_SMALL, &icon);
287     ok(hr == S_OK, "Expected S_OK, got %x\n", hr);
288     res = GetIconInfo(icon, &info);
289     ok(res, "Failed to get icon info, error %u\n", GetLastError());
290     bytes = GetObjectA(info.hbmColor, sizeof(bmp), &bmp);
291     ok(bytes > 0, "Failed to get bitmap info for icon\n");
292     ok(bmp.bmWidth  == GetSystemMetrics(SM_CXSMICON), "Wrong icon width %d\n", bmp.bmWidth);
293     ok(bmp.bmHeight == GetSystemMetrics(SM_CYSMICON), "Wrong icon height %d\n", bmp.bmHeight);
294     DestroyIcon(icon);
295 
296     hr = pLoadIconMetric(NULL, (LPWSTR)IDI_APPLICATION, LIM_LARGE, &icon);
297     ok(hr == S_OK, "Expected S_OK, got %x\n", hr);
298     res = GetIconInfo(icon, &info);
299     ok(res, "Failed to get icon info, error %u\n", GetLastError());
300     bytes = GetObjectA(info.hbmColor, sizeof(bmp), &bmp);
301     ok(bytes > 0, "Failed to get bitmap info for icon\n");
302     ok(bmp.bmWidth  == GetSystemMetrics(SM_CXICON), "Wrong icon width %d\n", bmp.bmWidth);
303     ok(bmp.bmHeight == GetSystemMetrics(SM_CYICON), "Wrong icon height %d\n", bmp.bmHeight);
304     DestroyIcon(icon);
305 
306     hr = pLoadIconWithScaleDown(NULL, (LPWSTR)IDI_APPLICATION, 42, 42, &icon);
307     ok(hr == S_OK, "Expected S_OK, got %x\n", hr);
308     res = GetIconInfo(icon, &info);
309     ok(res, "Failed to get icon info, error %u\n", GetLastError());
310     bytes = GetObjectA(info.hbmColor, sizeof(bmp), &bmp);
311     ok(bytes > 0, "Failed to get bitmap info for icon\n");
312     ok(bmp.bmWidth  == 42, "Wrong icon width %d\n", bmp.bmWidth);
313     ok(bmp.bmHeight == 42, "Wrong icon height %d\n", bmp.bmHeight);
314     DestroyIcon(icon);
315 
316     /* load icon from file */
317     hr = pLoadIconMetric(NULL, icon_path, LIM_SMALL, &icon);
318     ok(hr == S_OK, "Expected S_OK, got %x\n", hr);
319     res = GetIconInfo(icon, &info);
320     ok(res, "Failed to get icon info, error %u\n", GetLastError());
321     bytes = GetObjectA(info.hbmColor, sizeof(bmp), &bmp);
322     ok(bytes > 0, "Failed to get bitmap info for icon\n");
323     ok(bmp.bmWidth  == GetSystemMetrics(SM_CXSMICON), "Wrong icon width %d\n", bmp.bmWidth);
324     ok(bmp.bmHeight == GetSystemMetrics(SM_CYSMICON), "Wrong icon height %d\n", bmp.bmHeight);
325     DestroyIcon(icon);
326 
327     hr = pLoadIconWithScaleDown(NULL, icon_path, 42, 42, &icon);
328     ok(hr == S_OK, "Expected S_OK, got %x\n", hr);
329     res = GetIconInfo(icon, &info);
330     ok(res, "Failed to get icon info, error %u\n", GetLastError());
331     bytes = GetObjectA(info.hbmColor, sizeof(bmp), &bmp);
332     ok(bytes > 0, "Failed to get bitmap info for icon\n");
333     ok(bmp.bmWidth  == 42, "Wrong icon width %d\n", bmp.bmWidth);
334     ok(bmp.bmHeight == 42, "Wrong icon height %d\n", bmp.bmHeight);
335     DestroyIcon(icon);
336 
337     DeleteFileW(icon_path);
338     FreeLibrary(hinst);
339 }
340 
341 static void check_class( const char *name, int must_exist, UINT style, UINT ignore, BOOL v6 )
342 {
343     WNDCLASSA wc;
344 
345     if (GetClassInfoA( 0, name, &wc ))
346     {
347         char buff[64];
348         HWND hwnd;
349 
350 todo_wine_if(!strcmp(name, "SysLink") && !must_exist && !v6)
351         ok( must_exist, "System class %s should %sexist\n", name, must_exist ? "" : "NOT " );
352         if (!must_exist) return;
353 
354 todo_wine_if(!strcmp(name, "ScrollBar") || (!strcmp(name, "tooltips_class32") && v6))
355         ok( !(~wc.style & style & ~ignore), "System class %s is missing bits %x (%08x/%08x)\n",
356             name, ~wc.style & style, wc.style, style );
357 todo_wine_if((!strcmp(name, "tooltips_class32") && v6) || !strcmp(name, "SysLink"))
358         ok( !(wc.style & ~style), "System class %s has extra bits %x (%08x/%08x)\n",
359             name, wc.style & ~style, wc.style, style );
360         ok( !wc.hInstance, "System class %s has hInstance %p\n", name, wc.hInstance );
361 
362         hwnd = CreateWindowA(name, 0, 0, 0, 0, 0, 0, 0, NULL, GetModuleHandleA(NULL), 0);
363         ok( hwnd != NULL, "Failed to create window for class %s.\n", name );
364         GetClassNameA(hwnd, buff, ARRAY_SIZE(buff));
365         ok( !strcmp(name, buff), "Unexpected class name %s, expected %s.\n", buff, name );
366         DestroyWindow(hwnd);
367     }
368     else
369         ok( !must_exist, "System class %s does not exist\n", name );
370 }
371 
372 /* test styles of system classes */
373 static void test_builtin_classes(void)
374 {
375     /* check style bits */
376     check_class( "Button",     1, CS_PARENTDC | CS_DBLCLKS | CS_HREDRAW | CS_VREDRAW | CS_GLOBALCLASS, 0, FALSE );
377     check_class( "ComboBox",   1, CS_PARENTDC | CS_DBLCLKS | CS_HREDRAW | CS_VREDRAW | CS_GLOBALCLASS, 0, FALSE );
378     check_class( "Edit",       1, CS_PARENTDC | CS_DBLCLKS | CS_GLOBALCLASS, 0, FALSE );
379     check_class( "ListBox",    1, CS_PARENTDC | CS_DBLCLKS | CS_GLOBALCLASS, 0, FALSE );
380     check_class( "ScrollBar",  1, CS_PARENTDC | CS_DBLCLKS | CS_HREDRAW | CS_VREDRAW | CS_GLOBALCLASS, 0, FALSE );
381     check_class( "Static",     1, CS_PARENTDC | CS_DBLCLKS | CS_GLOBALCLASS, 0, FALSE );
382     check_class( "ComboLBox",  1, CS_SAVEBITS | CS_DBLCLKS | CS_DROPSHADOW | CS_GLOBALCLASS, CS_DROPSHADOW, FALSE );
383 }
384 
385 static void test_comctl32_classes(BOOL v6)
386 {
387     check_class(ANIMATE_CLASSA,      1, CS_DBLCLKS | CS_GLOBALCLASS, 0, FALSE);
388     check_class(WC_COMBOBOXEXA,      1, CS_GLOBALCLASS, 0, FALSE);
389     check_class(DATETIMEPICK_CLASSA, 1, CS_GLOBALCLASS, 0, FALSE);
390     check_class(WC_HEADERA,          1, CS_DBLCLKS | CS_GLOBALCLASS, 0, FALSE);
391     check_class(HOTKEY_CLASSA,       1, CS_GLOBALCLASS, 0, FALSE);
392     check_class(WC_IPADDRESSA,       1, CS_DBLCLKS | CS_HREDRAW | CS_VREDRAW | CS_GLOBALCLASS, 0, FALSE);
393     check_class(WC_LISTVIEWA,        1, CS_DBLCLKS | CS_GLOBALCLASS, 0, FALSE);
394     check_class(MONTHCAL_CLASSA,     1, CS_GLOBALCLASS, 0, FALSE);
395     check_class(WC_NATIVEFONTCTLA,   1, CS_GLOBALCLASS, 0, FALSE);
396     check_class(WC_PAGESCROLLERA,    1, CS_GLOBALCLASS, 0, FALSE);
397     check_class(PROGRESS_CLASSA,     1, CS_HREDRAW | CS_VREDRAW | CS_GLOBALCLASS, 0, FALSE);
398     check_class(REBARCLASSNAMEA,     1, CS_DBLCLKS | CS_GLOBALCLASS, 0, FALSE);
399     check_class(STATUSCLASSNAMEA,    1, CS_DBLCLKS | CS_VREDRAW | CS_GLOBALCLASS, 0, FALSE);
400     check_class(WC_TABCONTROLA,      1, CS_DBLCLKS | CS_HREDRAW | CS_VREDRAW | CS_GLOBALCLASS, 0, FALSE);
401     check_class(TOOLBARCLASSNAMEA,   1, CS_DBLCLKS | CS_GLOBALCLASS, 0, FALSE);
402     if (v6)
403         check_class(TOOLTIPS_CLASSA, 1, CS_DBLCLKS | CS_HREDRAW | CS_VREDRAW | CS_GLOBALCLASS | CS_DROPSHADOW, CS_SAVEBITS | CS_HREDRAW | CS_VREDRAW /* XP */, TRUE);
404     else
405         check_class(TOOLTIPS_CLASSA, 1, CS_DBLCLKS | CS_GLOBALCLASS | CS_SAVEBITS, CS_HREDRAW | CS_VREDRAW /* XP */, FALSE);
406     check_class(TRACKBAR_CLASSA,     1, CS_GLOBALCLASS, 0, FALSE);
407     check_class(WC_TREEVIEWA,        1, CS_DBLCLKS | CS_GLOBALCLASS, 0, FALSE);
408     check_class(UPDOWN_CLASSA,       1, CS_HREDRAW | CS_VREDRAW | CS_GLOBALCLASS, 0, FALSE);
409     check_class("SysLink", v6, CS_GLOBALCLASS, 0, FALSE);
410 }
411 
412 START_TEST(misc)
413 {
414     ULONG_PTR ctx_cookie;
415     HANDLE hCtx;
416 
417     if(!InitFunctionPtrs())
418         return;
419 
420     test_GetPtrAW();
421     test_Alloc();
422 
423     test_comctl32_classes(FALSE);
424 
425     if (!load_v6_module(&ctx_cookie, &hCtx))
426         return;
427 
428     test_comctl32_classes(TRUE);
429     test_builtin_classes();
430     test_LoadIconWithScaleDown();
431 
432     unload_v6_module(ctx_cookie, hCtx);
433 }
434