user32/tests: Fix the listbox delete test on NT4.
[wine] / dlls / shell32 / tests / shelllink.c
1 /*
2  * Unit tests for shelllinks
3  *
4  * Copyright 2004 Mike McCormack
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  * This is a test program for the SHGet{Special}Folder{Path|Location} functions
20  * of shell32, that get either a filesystem path or a LPITEMIDLIST (shell
21  * namespace) path for a given folder (CSIDL value).
22  *
23  */
24
25 #define COBJMACROS
26
27 #include "initguid.h"
28 #include "windows.h"
29 #include "shlguid.h"
30 #include "shobjidl.h"
31 #include "shlobj.h"
32 #include "wine/test.h"
33
34 #include "shell32_test.h"
35
36 #ifndef SLDF_HAS_LOGO3ID
37 #  define SLDF_HAS_LOGO3ID 0x00000800 /* not available in the Vista SDK */
38 #endif
39
40 typedef void (WINAPI *fnILFree)(LPITEMIDLIST);
41 typedef BOOL (WINAPI *fnILIsEqual)(LPCITEMIDLIST, LPCITEMIDLIST);
42 typedef HRESULT (WINAPI *fnSHILCreateFromPath)(LPCWSTR, LPITEMIDLIST *,DWORD*);
43 typedef HRESULT (WINAPI *fnSHDefExtractIconA)(LPCSTR, int, UINT, HICON*, HICON*, UINT);
44
45 static fnILFree pILFree;
46 static fnILIsEqual pILIsEqual;
47 static fnSHILCreateFromPath pSHILCreateFromPath;
48 static fnSHDefExtractIconA pSHDefExtractIconA;
49
50 static DWORD (WINAPI *pGetLongPathNameA)(LPCSTR, LPSTR, DWORD);
51
52 static const GUID _IID_IShellLinkDataList = {
53     0x45e2b4ae, 0xb1c3, 0x11d0,
54     { 0xb9, 0x2f, 0x00, 0xa0, 0xc9, 0x03, 0x12, 0xe1 }
55 };
56
57 static const WCHAR notafile[]= { 'C',':','\\','n','o','n','e','x','i','s','t','e','n','t','\\','f','i','l','e',0 };
58
59
60 /* For some reason SHILCreateFromPath does not work on Win98 and
61  * SHSimpleIDListFromPathA does not work on NT4. But if we call both we
62  * get what we want on all platforms.
63  */
64 static LPITEMIDLIST (WINAPI *pSHSimpleIDListFromPathAW)(LPCVOID);
65
66 static LPITEMIDLIST path_to_pidl(const char* path)
67 {
68     LPITEMIDLIST pidl;
69
70     if (!pSHSimpleIDListFromPathAW)
71     {
72         HMODULE hdll=GetModuleHandleA("shell32.dll");
73         pSHSimpleIDListFromPathAW=(void*)GetProcAddress(hdll, (char*)162);
74         if (!pSHSimpleIDListFromPathAW)
75             win_skip("SHSimpleIDListFromPathAW not found in shell32.dll\n");
76     }
77
78     pidl=NULL;
79     /* pSHSimpleIDListFromPathAW maps to A on non NT platforms */
80     if (pSHSimpleIDListFromPathAW && (GetVersion() & 0x80000000))
81         pidl=pSHSimpleIDListFromPathAW(path);
82
83     if (!pidl)
84     {
85         WCHAR* pathW;
86         HRESULT r;
87         int len;
88
89         len=MultiByteToWideChar(CP_ACP, 0, path, -1, NULL, 0);
90         pathW=HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
91         MultiByteToWideChar(CP_ACP, 0, path, -1, pathW, len);
92
93         r=pSHILCreateFromPath(pathW, &pidl, NULL);
94         ok(SUCCEEDED(r), "SHILCreateFromPath failed (0x%08x)\n", r);
95         HeapFree(GetProcessHeap(), 0, pathW);
96     }
97     return pidl;
98 }
99
100
101 /*
102  * Test manipulation of an IShellLink's properties.
103  */
104
105 static void test_get_set(void)
106 {
107     HRESULT r;
108     IShellLinkA *sl;
109     char mypath[MAX_PATH];
110     char buffer[INFOTIPSIZE];
111     LPITEMIDLIST pidl, tmp_pidl;
112     const char * str;
113     int i;
114     WORD w;
115
116     r = CoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
117                          &IID_IShellLinkA, (LPVOID*)&sl);
118     ok(SUCCEEDED(r), "no IID_IShellLinkA (0x%08x)\n", r);
119     if (FAILED(r))
120         return;
121
122     /* Test Getting / Setting the description */
123     strcpy(buffer,"garbage");
124     r = IShellLinkA_GetDescription(sl, buffer, sizeof(buffer));
125     ok(SUCCEEDED(r), "GetDescription failed (0x%08x)\n", r);
126     ok(*buffer=='\0', "GetDescription returned '%s'\n", buffer);
127
128     str="Some description";
129     r = IShellLinkA_SetDescription(sl, str);
130     ok(SUCCEEDED(r), "SetDescription failed (0x%08x)\n", r);
131
132     strcpy(buffer,"garbage");
133     r = IShellLinkA_GetDescription(sl, buffer, sizeof(buffer));
134     ok(SUCCEEDED(r), "GetDescription failed (0x%08x)\n", r);
135     ok(lstrcmp(buffer,str)==0, "GetDescription returned '%s'\n", buffer);
136
137     /* Test Getting / Setting the work directory */
138     strcpy(buffer,"garbage");
139     r = IShellLinkA_GetWorkingDirectory(sl, buffer, sizeof(buffer));
140     ok(SUCCEEDED(r), "GetWorkingDirectory failed (0x%08x)\n", r);
141     ok(*buffer=='\0', "GetWorkingDirectory returned '%s'\n", buffer);
142
143     str="c:\\nonexistent\\directory";
144     r = IShellLinkA_SetWorkingDirectory(sl, str);
145     ok(SUCCEEDED(r), "SetWorkingDirectory failed (0x%08x)\n", r);
146
147     strcpy(buffer,"garbage");
148     r = IShellLinkA_GetWorkingDirectory(sl, buffer, sizeof(buffer));
149     ok(SUCCEEDED(r), "GetWorkingDirectory failed (0x%08x)\n", r);
150     ok(lstrcmpi(buffer,str)==0, "GetWorkingDirectory returned '%s'\n", buffer);
151
152     /* Test Getting / Setting the path */
153     strcpy(buffer,"garbage");
154     r = IShellLinkA_GetPath(sl, buffer, sizeof(buffer), NULL, SLGP_RAWPATH);
155     ok(SUCCEEDED(r), "GetPath failed (0x%08x)\n", r);
156     ok(*buffer=='\0', "GetPath returned '%s'\n", buffer);
157
158     r = IShellLinkA_SetPath(sl, "");
159     ok(r==S_OK, "SetPath failed (0x%08x)\n", r);
160
161     strcpy(buffer,"garbage");
162     r = IShellLinkA_GetPath(sl, buffer, sizeof(buffer), NULL, SLGP_RAWPATH);
163     ok(SUCCEEDED(r), "GetPath failed (0x%08x)\n", r);
164     ok(*buffer=='\0', "GetPath returned '%s'\n", buffer);
165
166     /* Win98 returns S_FALSE, but WinXP returns S_OK */
167     str="c:\\nonexistent\\file";
168     r = IShellLinkA_SetPath(sl, str);
169     ok(r==S_FALSE || r==S_OK, "SetPath failed (0x%08x)\n", r);
170
171     strcpy(buffer,"garbage");
172     r = IShellLinkA_GetPath(sl, buffer, sizeof(buffer), NULL, SLGP_RAWPATH);
173     ok(SUCCEEDED(r), "GetPath failed (0x%08x)\n", r);
174     ok(lstrcmpi(buffer,str)==0, "GetPath returned '%s'\n", buffer);
175
176     /* Get some real path to play with */
177     GetWindowsDirectoryA( mypath, sizeof(mypath)-12 );
178     strcat(mypath, "\\regedit.exe");
179
180     /* Test the interaction of SetPath and SetIDList */
181     tmp_pidl=NULL;
182     r = IShellLinkA_GetIDList(sl, &tmp_pidl);
183     ok(SUCCEEDED(r), "GetIDList failed (0x%08x)\n", r);
184     if (SUCCEEDED(r))
185     {
186         BOOL ret;
187
188         strcpy(buffer,"garbage");
189         ret = SHGetPathFromIDListA(tmp_pidl, buffer);
190         todo_wine {
191         ok(ret, "SHGetPathFromIDListA failed\n");
192         }
193         if (ret)
194             ok(lstrcmpi(buffer,str)==0, "GetIDList returned '%s'\n", buffer);
195     }
196
197     pidl=path_to_pidl(mypath);
198     ok(pidl!=NULL, "path_to_pidl returned a NULL pidl\n");
199
200     if (pidl)
201     {
202         r = IShellLinkA_SetIDList(sl, pidl);
203         ok(SUCCEEDED(r), "SetIDList failed (0x%08x)\n", r);
204
205         tmp_pidl=NULL;
206         r = IShellLinkA_GetIDList(sl, &tmp_pidl);
207         ok(SUCCEEDED(r), "GetIDList failed (0x%08x)\n", r);
208         ok(tmp_pidl && pILIsEqual(pidl, tmp_pidl),
209            "GetIDList returned an incorrect pidl\n");
210
211         /* tmp_pidl is owned by IShellLink so we don't free it */
212         pILFree(pidl);
213
214         strcpy(buffer,"garbage");
215         r = IShellLinkA_GetPath(sl, buffer, sizeof(buffer), NULL, SLGP_RAWPATH);
216         ok(SUCCEEDED(r), "GetPath failed (0x%08x)\n", r);
217         todo_wine
218         ok(lstrcmpi(buffer, mypath)==0, "GetPath returned '%s'\n", buffer);
219
220     }
221
222     /* test path with quotes (IShellLinkA_SetPath returns S_FALSE on W2K and below and S_OK on XP and above */
223     r = IShellLinkA_SetPath(sl, "\"c:\\nonexistent\\file\"");
224     ok(r==S_FALSE || r == S_OK, "SetPath failed (0x%08x)\n", r);
225
226     r = IShellLinkA_GetPath(sl, buffer, sizeof(buffer), NULL, SLGP_RAWPATH);
227     ok(r==S_OK, "GetPath failed (0x%08x)\n", r);
228     ok(!lstrcmp(buffer, "C:\\nonexistent\\file") ||
229        broken(!lstrcmp(buffer, "C:\\\"c:\\nonexistent\\file\"")), /* NT4 */
230        "case doesn't match\n");
231
232     r = IShellLinkA_SetPath(sl, "\"c:\\foo");
233     ok(r==S_FALSE || r == S_OK || r == E_INVALIDARG /* Vista */, "SetPath failed (0x%08x)\n", r);
234
235     r = IShellLinkA_SetPath(sl, "\"\"c:\\foo");
236     ok(r==S_FALSE || r == S_OK || r == E_INVALIDARG /* Vista */, "SetPath failed (0x%08x)\n", r);
237
238     r = IShellLinkA_SetPath(sl, "c:\\foo\"");
239     ok(r==S_FALSE || r == S_OK || r == E_INVALIDARG /* Vista */, "SetPath failed (0x%08x)\n", r);
240
241     r = IShellLinkA_SetPath(sl, "\"\"c:\\foo\"");
242     ok(r==S_FALSE || r == S_OK || r == E_INVALIDARG /* Vista */, "SetPath failed (0x%08x)\n", r);
243
244     r = IShellLinkA_SetPath(sl, "\"\"c:\\foo\"\"");
245     ok(r==S_FALSE || r == S_OK || r == E_INVALIDARG /* Vista */, "SetPath failed (0x%08x)\n", r);
246
247     /* Test Getting / Setting the arguments */
248     strcpy(buffer,"garbage");
249     r = IShellLinkA_GetArguments(sl, buffer, sizeof(buffer));
250     ok(SUCCEEDED(r), "GetArguments failed (0x%08x)\n", r);
251     ok(*buffer=='\0', "GetArguments returned '%s'\n", buffer);
252
253     str="param1 \"spaced param2\"";
254     r = IShellLinkA_SetArguments(sl, str);
255     ok(SUCCEEDED(r), "SetArguments failed (0x%08x)\n", r);
256
257     strcpy(buffer,"garbage");
258     r = IShellLinkA_GetArguments(sl, buffer, sizeof(buffer));
259     ok(SUCCEEDED(r), "GetArguments failed (0x%08x)\n", r);
260     ok(lstrcmp(buffer,str)==0, "GetArguments returned '%s'\n", buffer);
261
262     /* Test Getting / Setting showcmd */
263     i=0xdeadbeef;
264     r = IShellLinkA_GetShowCmd(sl, &i);
265     ok(SUCCEEDED(r), "GetShowCmd failed (0x%08x)\n", r);
266     ok(i==SW_SHOWNORMAL, "GetShowCmd returned %d\n", i);
267
268     r = IShellLinkA_SetShowCmd(sl, SW_SHOWMAXIMIZED);
269     ok(SUCCEEDED(r), "SetShowCmd failed (0x%08x)\n", r);
270
271     i=0xdeadbeef;
272     r = IShellLinkA_GetShowCmd(sl, &i);
273     ok(SUCCEEDED(r), "GetShowCmd failed (0x%08x)\n", r);
274     ok(i==SW_SHOWMAXIMIZED, "GetShowCmd returned %d'\n", i);
275
276     /* Test Getting / Setting the icon */
277     i=0xdeadbeef;
278     strcpy(buffer,"garbage");
279     r = IShellLinkA_GetIconLocation(sl, buffer, sizeof(buffer), &i);
280     todo_wine {
281     ok(SUCCEEDED(r), "GetIconLocation failed (0x%08x)\n", r);
282     }
283     ok(*buffer=='\0', "GetIconLocation returned '%s'\n", buffer);
284     ok(i==0, "GetIconLocation returned %d\n", i);
285
286     str="c:\\nonexistent\\file";
287     r = IShellLinkA_SetIconLocation(sl, str, 0xbabecafe);
288     ok(SUCCEEDED(r), "SetIconLocation failed (0x%08x)\n", r);
289
290     i=0xdeadbeef;
291     r = IShellLinkA_GetIconLocation(sl, buffer, sizeof(buffer), &i);
292     ok(SUCCEEDED(r), "GetIconLocation failed (0x%08x)\n", r);
293     ok(lstrcmpi(buffer,str)==0, "GetIconLocation returned '%s'\n", buffer);
294     ok(i==0xbabecafe, "GetIconLocation returned %d'\n", i);
295
296     /* Test Getting / Setting the hot key */
297     w=0xbeef;
298     r = IShellLinkA_GetHotkey(sl, &w);
299     ok(SUCCEEDED(r), "GetHotkey failed (0x%08x)\n", r);
300     ok(w==0, "GetHotkey returned %d\n", w);
301
302     r = IShellLinkA_SetHotkey(sl, 0x5678);
303     ok(SUCCEEDED(r), "SetHotkey failed (0x%08x)\n", r);
304
305     w=0xbeef;
306     r = IShellLinkA_GetHotkey(sl, &w);
307     ok(SUCCEEDED(r), "GetHotkey failed (0x%08x)\n", r);
308     ok(w==0x5678, "GetHotkey returned %d'\n", w);
309
310     IShellLinkA_Release(sl);
311 }
312
313
314 /*
315  * Test saving and loading .lnk files
316  */
317
318 #define lok                   ok_(__FILE__, line)
319 #define lok_todo_4(todo_flag,a,b,c,d) \
320     if ((todo & todo_flag) == 0) lok((a), (b), (c), (d)); \
321     else todo_wine lok((a), (b), (c), (d));
322 #define lok_todo_2(todo_flag,a,b) \
323     if ((todo & todo_flag) == 0) lok((a), (b)); \
324     else todo_wine lok((a), (b));
325 #define check_lnk(a,b,c)        check_lnk_(__LINE__, (a), (b), (c))
326
327 void create_lnk_(int line, const WCHAR* path, lnk_desc_t* desc, int save_fails)
328 {
329     HRESULT r;
330     IShellLinkA *sl;
331     IPersistFile *pf;
332
333     r = CoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
334                          &IID_IShellLinkA, (LPVOID*)&sl);
335     lok(SUCCEEDED(r), "no IID_IShellLinkA (0x%08x)\n", r);
336     if (FAILED(r))
337         return;
338
339     if (desc->description)
340     {
341         r = IShellLinkA_SetDescription(sl, desc->description);
342         lok(SUCCEEDED(r), "SetDescription failed (0x%08x)\n", r);
343     }
344     if (desc->workdir)
345     {
346         r = IShellLinkA_SetWorkingDirectory(sl, desc->workdir);
347         lok(SUCCEEDED(r), "SetWorkingDirectory failed (0x%08x)\n", r);
348     }
349     if (desc->path)
350     {
351         r = IShellLinkA_SetPath(sl, desc->path);
352         lok(SUCCEEDED(r), "SetPath failed (0x%08x)\n", r);
353     }
354     if (desc->pidl)
355     {
356         r = IShellLinkA_SetIDList(sl, desc->pidl);
357         lok(SUCCEEDED(r), "SetIDList failed (0x%08x)\n", r);
358     }
359     if (desc->arguments)
360     {
361         r = IShellLinkA_SetArguments(sl, desc->arguments);
362         lok(SUCCEEDED(r), "SetArguments failed (0x%08x)\n", r);
363     }
364     if (desc->showcmd)
365     {
366         r = IShellLinkA_SetShowCmd(sl, desc->showcmd);
367         lok(SUCCEEDED(r), "SetShowCmd failed (0x%08x)\n", r);
368     }
369     if (desc->icon)
370     {
371         r = IShellLinkA_SetIconLocation(sl, desc->icon, desc->icon_id);
372         lok(SUCCEEDED(r), "SetIconLocation failed (0x%08x)\n", r);
373     }
374     if (desc->hotkey)
375     {
376         r = IShellLinkA_SetHotkey(sl, desc->hotkey);
377         lok(SUCCEEDED(r), "SetHotkey failed (0x%08x)\n", r);
378     }
379
380     r = IShellLinkW_QueryInterface(sl, &IID_IPersistFile, (LPVOID*)&pf);
381     lok(SUCCEEDED(r), "no IID_IPersistFile (0x%08x)\n", r);
382     if (SUCCEEDED(r))
383     {
384         r = IPersistFile_Save(pf, path, TRUE);
385         if (save_fails)
386         {
387             todo_wine {
388             lok(SUCCEEDED(r), "save failed (0x%08x)\n", r);
389             }
390         }
391         else
392         {
393             lok(SUCCEEDED(r), "save failed (0x%08x)\n", r);
394         }
395         IPersistFile_Release(pf);
396     }
397
398     IShellLinkA_Release(sl);
399 }
400
401 static void check_lnk_(int line, const WCHAR* path, lnk_desc_t* desc, int todo)
402 {
403     HRESULT r;
404     IShellLinkA *sl;
405     IPersistFile *pf;
406     char buffer[INFOTIPSIZE];
407
408     r = CoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
409                          &IID_IShellLinkA, (LPVOID*)&sl);
410     lok(SUCCEEDED(r), "no IID_IShellLinkA (0x%08x)\n", r);
411     if (FAILED(r))
412         return;
413
414     r = IShellLinkA_QueryInterface(sl, &IID_IPersistFile, (LPVOID*)&pf);
415     lok(SUCCEEDED(r), "no IID_IPersistFile (0x%08x)\n", r);
416     if (FAILED(r))
417     {
418         IShellLinkA_Release(sl);
419         return;
420     }
421
422     r = IPersistFile_Load(pf, path, STGM_READ);
423     lok(SUCCEEDED(r), "load failed (0x%08x)\n", r);
424     IPersistFile_Release(pf);
425     if (FAILED(r))
426     {
427         IShellLinkA_Release(sl);
428         return;
429     }
430
431     if (desc->description)
432     {
433         strcpy(buffer,"garbage");
434         r = IShellLinkA_GetDescription(sl, buffer, sizeof(buffer));
435         lok(SUCCEEDED(r), "GetDescription failed (0x%08x)\n", r);
436         lok_todo_4(0x1, lstrcmp(buffer, desc->description)==0,
437            "GetDescription returned '%s' instead of '%s'\n",
438            buffer, desc->description);
439     }
440     if (desc->workdir)
441     {
442         strcpy(buffer,"garbage");
443         r = IShellLinkA_GetWorkingDirectory(sl, buffer, sizeof(buffer));
444         lok(SUCCEEDED(r), "GetWorkingDirectory failed (0x%08x)\n", r);
445         lok_todo_4(0x2, lstrcmpi(buffer, desc->workdir)==0,
446            "GetWorkingDirectory returned '%s' instead of '%s'\n",
447            buffer, desc->workdir);
448     }
449     if (desc->path)
450     {
451         strcpy(buffer,"garbage");
452         r = IShellLinkA_GetPath(sl, buffer, sizeof(buffer), NULL, SLGP_RAWPATH);
453         lok(SUCCEEDED(r), "GetPath failed (0x%08x)\n", r);
454         lok_todo_4(0x4, lstrcmpi(buffer, desc->path)==0,
455            "GetPath returned '%s' instead of '%s'\n",
456            buffer, desc->path);
457     }
458     if (desc->pidl)
459     {
460         LPITEMIDLIST pidl=NULL;
461         r = IShellLinkA_GetIDList(sl, &pidl);
462         lok(SUCCEEDED(r), "GetIDList failed (0x%08x)\n", r);
463         lok_todo_2(0x8, pILIsEqual(pidl, desc->pidl),
464            "GetIDList returned an incorrect pidl\n");
465     }
466     if (desc->showcmd)
467     {
468         int i=0xdeadbeef;
469         r = IShellLinkA_GetShowCmd(sl, &i);
470         lok(SUCCEEDED(r), "GetShowCmd failed (0x%08x)\n", r);
471         lok_todo_4(0x10, i==desc->showcmd,
472            "GetShowCmd returned 0x%0x instead of 0x%0x\n",
473            i, desc->showcmd);
474     }
475     if (desc->icon)
476     {
477         int i=0xdeadbeef;
478         strcpy(buffer,"garbage");
479         r = IShellLinkA_GetIconLocation(sl, buffer, sizeof(buffer), &i);
480         lok(SUCCEEDED(r), "GetIconLocation failed (0x%08x)\n", r);
481         lok_todo_4(0x20, lstrcmpi(buffer, desc->icon)==0,
482            "GetIconLocation returned '%s' instead of '%s'\n",
483            buffer, desc->icon);
484         lok_todo_4(0x20, i==desc->icon_id,
485            "GetIconLocation returned 0x%0x instead of 0x%0x\n",
486            i, desc->icon_id);
487     }
488     if (desc->hotkey)
489     {
490         WORD i=0xbeef;
491         r = IShellLinkA_GetHotkey(sl, &i);
492         lok(SUCCEEDED(r), "GetHotkey failed (0x%08x)\n", r);
493         lok_todo_4(0x40, i==desc->hotkey,
494            "GetHotkey returned 0x%04x instead of 0x%04x\n",
495            i, desc->hotkey);
496     }
497
498     IShellLinkA_Release(sl);
499 }
500
501 static void test_load_save(void)
502 {
503     WCHAR lnkfile[MAX_PATH];
504     char lnkfileA[MAX_PATH];
505     static const char lnkfileA_name[] = "\\test.lnk";
506
507     lnk_desc_t desc;
508     char mypath[MAX_PATH];
509     char mydir[MAX_PATH];
510     char realpath[MAX_PATH];
511     char* p;
512     HANDLE hf;
513     DWORD r;
514
515     if (!pGetLongPathNameA)
516     {
517         win_skip("GetLongPathNameA is not available\n");
518         return;
519     }
520
521     /* Don't used a fixed path for the test.lnk file */
522     GetTempPathA(MAX_PATH, lnkfileA);
523     lstrcatA(lnkfileA, lnkfileA_name);
524     MultiByteToWideChar(CP_ACP, 0, lnkfileA, -1, lnkfile, MAX_PATH);
525
526     /* Save an empty .lnk file */
527     memset(&desc, 0, sizeof(desc));
528     create_lnk(lnkfile, &desc, 0);
529
530     /* It should come back as a bunch of empty strings */
531     desc.description="";
532     desc.workdir="";
533     desc.path="";
534     desc.arguments="";
535     desc.icon="";
536     check_lnk(lnkfile, &desc, 0x0);
537
538     /* Point a .lnk file to nonexistent files */
539     desc.description="";
540     desc.workdir="c:\\Nonexitent\\work\\directory";
541     desc.path="c:\\nonexistent\\path";
542     desc.pidl=NULL;
543     desc.arguments="";
544     desc.showcmd=0;
545     desc.icon="c:\\nonexistent\\icon\\file";
546     desc.icon_id=1234;
547     desc.hotkey=0;
548     create_lnk(lnkfile, &desc, 0);
549     check_lnk(lnkfile, &desc, 0x0);
550
551     r=GetModuleFileName(NULL, mypath, sizeof(mypath));
552     ok(r<sizeof(mypath), "GetModuleFileName failed (%d)\n", r);
553     strcpy(mydir, mypath);
554     p=strrchr(mydir, '\\');
555     if (p)
556         *p='\0';
557
558     /* IShellLink returns path in long form */
559     if (!pGetLongPathNameA(mypath, realpath, MAX_PATH)) strcpy( realpath, mypath );
560
561     /* Overwrite the existing lnk file and point it to existing files */
562     desc.description="test 2";
563     desc.workdir=mydir;
564     desc.path=realpath;
565     desc.pidl=NULL;
566     desc.arguments="/option1 /option2 \"Some string\"";
567     desc.showcmd=SW_SHOWNORMAL;
568     desc.icon=mypath;
569     desc.icon_id=0;
570     desc.hotkey=0x1234;
571     create_lnk(lnkfile, &desc, 0);
572     check_lnk(lnkfile, &desc, 0x0);
573
574     /* Overwrite the existing lnk file and test link to a command on the path */
575     desc.description="command on path";
576     desc.workdir=mypath;
577     desc.path="rundll32.exe";
578     desc.pidl=NULL;
579     desc.arguments="/option1 /option2 \"Some string\"";
580     desc.showcmd=SW_SHOWNORMAL;
581     desc.icon=mypath;
582     desc.icon_id=0;
583     desc.hotkey=0x1234;
584     create_lnk(lnkfile, &desc, 0);
585     /* Check that link is created to proper location */
586     SearchPathA( NULL, desc.path, NULL, MAX_PATH, realpath, NULL);
587     desc.path=realpath;
588     check_lnk(lnkfile, &desc, 0x0);
589
590     /* Create a temporary non-executable file */
591     r=GetTempPath(sizeof(mypath), mypath);
592     ok(r<sizeof(mypath), "GetTempPath failed (%d), err %d\n", r, GetLastError());
593     r=pGetLongPathNameA(mypath, mydir, sizeof(mydir));
594     ok(r<sizeof(mydir), "GetLongPathName failed (%d), err %d\n", r, GetLastError());
595     p=strrchr(mydir, '\\');
596     if (p)
597         *p='\0';
598
599     strcpy(mypath, mydir);
600     strcat(mypath, "\\test.txt");
601     hf = CreateFile(mypath, GENERIC_WRITE, 0, NULL,
602                     CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
603     CloseHandle(hf);
604
605     /* Overwrite the existing lnk file and test link to an existing non-executable file */
606     desc.description="non-executable file";
607     desc.workdir=mydir;
608     desc.path=mypath;
609     desc.pidl=NULL;
610     desc.arguments="";
611     desc.showcmd=SW_SHOWNORMAL;
612     desc.icon=mypath;
613     desc.icon_id=0;
614     desc.hotkey=0x1234;
615     create_lnk(lnkfile, &desc, 0);
616     check_lnk(lnkfile, &desc, 0x0);
617
618     r = DeleteFileA(mypath);
619     ok(r, "failed to delete file %s (%d)\n", mypath, GetLastError());
620
621     /* FIXME: Also test saving a .lnk pointing to a pidl that cannot be
622      * represented as a path.
623      */
624
625     /* DeleteFileW is not implemented on Win9x */
626     r=DeleteFileA(lnkfileA);
627     ok(r, "failed to delete link '%s' (%d)\n", lnkfileA, GetLastError());
628 }
629
630 static void test_datalink(void)
631 {
632     static const WCHAR lnk[] = {
633       ':',':','{','9','d','b','1','1','8','6','f','-','4','0','d','f','-','1',
634       '1','d','1','-','a','a','8','c','-','0','0','c','0','4','f','b','6','7',
635       '8','6','3','}',':','{','0','0','0','1','0','4','0','9','-','7','8','E',
636       '1','-','1','1','D','2','-','B','6','0','F','-','0','0','6','0','9','7',
637       'C','9','9','8','E','7','}',':',':','{','9','d','b','1','1','8','6','e',
638       '-','4','0','d','f','-','1','1','d','1','-','a','a','8','c','-','0','0',
639       'c','0','4','f','b','6','7','8','6','3','}',':','2','6',',','!','!','g',
640       'x','s','f','(','N','g',']','q','F','`','H','{','L','s','A','C','C','E',
641       'S','S','F','i','l','e','s','>','p','l','T',']','j','I','{','j','f','(',
642       '=','1','&','L','[','-','8','1','-',']',':',':',0 };
643     static const WCHAR comp[] = {
644       '2','6',',','!','!','g','x','s','f','(','N','g',']','q','F','`','H','{',
645       'L','s','A','C','C','E','S','S','F','i','l','e','s','>','p','l','T',']',
646       'j','I','{','j','f','(','=','1','&','L','[','-','8','1','-',']',0 };
647     IShellLinkDataList *dl = NULL;
648     IShellLinkW *sl = NULL;
649     HRESULT r;
650     DWORD flags = 0;
651     EXP_DARWIN_LINK *dar;
652
653     r = CoCreateInstance( &CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
654                             &IID_IShellLinkW, (LPVOID*)&sl );
655     ok( r == S_OK ||
656         broken(r == E_NOINTERFACE), /* Win9x */
657         "CoCreateInstance failed (0x%08x)\n", r);
658     if (!sl)
659     {
660         win_skip("no shelllink\n");
661         return;
662     }
663
664     r = IShellLinkW_QueryInterface( sl, &_IID_IShellLinkDataList, (LPVOID*) &dl );
665     ok( r == S_OK ||
666         broken(r == E_NOINTERFACE), /* NT4 */
667         "IShellLinkW_QueryInterface failed (0x%08x)\n", r);
668
669     if (!dl)
670     {
671         win_skip("no datalink interface\n");
672         IShellLinkW_Release( sl );
673         return;
674     }
675
676     flags = 0;
677     r = IShellLinkDataList_GetFlags( dl, &flags );
678     ok( r == S_OK, "GetFlags failed\n");
679     ok( flags == 0, "GetFlags returned wrong flags\n");
680
681     dar = (void*)-1;
682     r = IShellLinkDataList_CopyDataBlock( dl, EXP_DARWIN_ID_SIG, (LPVOID*) &dar );
683     ok( r == E_FAIL, "CopyDataBlock failed\n");
684     ok( dar == NULL, "should be null\n");
685
686     r = IShellLinkW_SetPath(sl, lnk);
687     ok(r == S_OK, "set path failed\n");
688
689     /*
690      * The following crashes:
691      * r = IShellLinkDataList_GetFlags( dl, NULL );
692      */
693
694     flags = 0;
695     r = IShellLinkDataList_GetFlags( dl, &flags );
696     ok( r == S_OK, "GetFlags failed\n");
697     ok( flags == (SLDF_HAS_DARWINID|SLDF_HAS_LOGO3ID),
698         "GetFlags returned wrong flags\n");
699
700     dar = NULL;
701     r = IShellLinkDataList_CopyDataBlock( dl, EXP_DARWIN_ID_SIG, (LPVOID*) &dar );
702     ok( r == S_OK, "CopyDataBlock failed\n");
703
704     ok( dar && ((DATABLOCK_HEADER*)dar)->dwSignature == EXP_DARWIN_ID_SIG, "signature wrong\n");
705     ok( dar && 0==lstrcmpW(dar->szwDarwinID, comp ), "signature wrong\n");
706
707     LocalFree( dar );
708
709     IUnknown_Release( dl );
710     IShellLinkW_Release( sl );
711 }
712
713 static void test_shdefextracticon(void)
714 {
715     HICON hiconlarge=NULL, hiconsmall=NULL;
716     HRESULT res;
717
718     if (!pSHDefExtractIconA)
719     {
720         win_skip("SHDefExtractIconA is unavailable\n");
721         return;
722     }
723
724     res = pSHDefExtractIconA("shell32.dll", 0, 0, &hiconlarge, &hiconsmall, MAKELONG(16,24));
725     ok(SUCCEEDED(res), "SHDefExtractIconA failed, res=%x\n", res);
726     ok(hiconlarge != NULL, "got null hiconlarge\n");
727     ok(hiconsmall != NULL, "got null hiconsmall\n");
728     DestroyIcon(hiconlarge);
729     DestroyIcon(hiconsmall);
730
731     hiconsmall = NULL;
732     res = pSHDefExtractIconA("shell32.dll", 0, 0, NULL, &hiconsmall, MAKELONG(16,24));
733     ok(SUCCEEDED(res), "SHDefExtractIconA failed, res=%x\n", res);
734     ok(hiconsmall != NULL, "got null hiconsmall\n");
735     DestroyIcon(hiconsmall);
736
737     res = pSHDefExtractIconA("shell32.dll", 0, 0, NULL, NULL, MAKELONG(16,24));
738     ok(SUCCEEDED(res), "SHDefExtractIconA failed, res=%x\n", res);
739 }
740
741 START_TEST(shelllink)
742 {
743     HRESULT r;
744     HMODULE hmod = GetModuleHandleA("shell32.dll");
745     HMODULE hkernel32 = GetModuleHandleA("kernel32.dll");
746
747     pILFree = (fnILFree) GetProcAddress(hmod, (LPSTR)155);
748     pILIsEqual = (fnILIsEqual) GetProcAddress(hmod, (LPSTR)21);
749     pSHILCreateFromPath = (fnSHILCreateFromPath) GetProcAddress(hmod, (LPSTR)28);
750     pSHDefExtractIconA = (fnSHDefExtractIconA) GetProcAddress(hmod, "SHDefExtractIconA");
751
752     pGetLongPathNameA = (void *)GetProcAddress(hkernel32, "GetLongPathNameA");
753
754     r = CoInitialize(NULL);
755     ok(SUCCEEDED(r), "CoInitialize failed (0x%08x)\n", r);
756     if (FAILED(r))
757         return;
758
759     test_get_set();
760     test_load_save();
761     test_datalink();
762     test_shdefextracticon();
763
764     CoUninitialize();
765 }