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