dlls/ieframe/tests/intshcut.c

   1 /*
   2  * Unit tests to document InternetShortcut's behaviour
   3  *
   4  * Copyright 2008 Damjan Jovanovic
   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 <stdarg.h>
  22 #include <stdio.h>
  23
  24 #define COBJMACROS
  25 #define NONAMELESSUNION
  26
  27 #include "windef.h"
  28 #include "winbase.h"
  29 #include "winreg.h"
  30 #include "winerror.h"
  31
  32 #include "shlobj.h"
  33 #include "shobjidl.h"
  34 #include "shlguid.h"
  35 #include "ole2.h"
  36 #include "mshtml.h"
  37 #include "initguid.h"
  38 #include "isguids.h"
  39 #include "intshcut.h"
  40
  41 #include "wine/test.h"
  42
  43 static HRESULT WINAPI Unknown_QueryInterface(IUnknown *pUnknown, REFIID riid, void **ppvObject)
  44 {
  45     if (IsEqualGUID(&IID_IUnknown, riid))
  46     {
  47         *ppvObject = pUnknown;
  48         return S_OK;
  49     }
  50     return E_NOINTERFACE;
  51 }
  52
  53 static ULONG WINAPI Unknown_AddRef(IUnknown *pUnknown)
  54 {
  55     return 2;
  56 }
  57
  58 static ULONG WINAPI Unknown_Release(IUnknown *pUnknown)
  59 {
  60     return 1;
  61 }
  62
  63 static IUnknownVtbl unknownVtbl = {
  64     Unknown_QueryInterface,
  65     Unknown_AddRef,
  66     Unknown_Release
  67 };
  68
  69 static IUnknown unknown = {
  70     &unknownVtbl
  71 };
  72
  73 static void test_Aggregability(void)
  74 {
  75     HRESULT hr;
  76     IUnknown *pUnknown = NULL;
  77
  78     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
  79     ok(hr == S_OK, "could not create instance of CLSID_InternetShortcut with IID_IUnknown, hr = 0x%x\n", hr);
  80     if (pUnknown)
  81         IUnknown_Release(pUnknown);
  82
  83     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&pUnknown);
  84     ok(hr == S_OK, "could not create instance of CLSID_InternetShortcut with IID_IUniformResourceLocatorA, hr = 0x%x\n", hr);
  85     if (pUnknown)
  86         IUnknown_Release(pUnknown);
  87
  88     hr = CoCreateInstance(&CLSID_InternetShortcut, &unknown, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
  89     ok(hr == CLASS_E_NOAGGREGATION, "aggregation didn't fail like it should, hr = 0x%x\n", hr);
  90     if (pUnknown)
  91         IUnknown_Release(pUnknown);
  92 }
  93
  94 static void can_query_interface(IUnknown *pUnknown, REFIID riid)
  95 {
  96     HRESULT hr;
  97     IUnknown *newInterface;
  98     hr = IUnknown_QueryInterface(pUnknown, riid, (void**)&newInterface);
  99     ok(hr == S_OK, "interface %s could not be queried\n", wine_dbgstr_guid(riid));
 100     if (SUCCEEDED(hr))
 101         IUnknown_Release(newInterface);
 102 }
 103
 104 static void test_QueryInterface(void)
 105 {
 106     HRESULT hr;
 107     IUnknown *pUnknown;
 108
 109     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUnknown, (void**)&pUnknown);
 110     ok(hr == S_OK, "Could not create InternetShortcut object: %08x\n", hr);
 111
 112     can_query_interface(pUnknown, &IID_IUniformResourceLocatorA);
 113     can_query_interface(pUnknown, &IID_IUniformResourceLocatorW);
 114     can_query_interface(pUnknown, &IID_IPersistFile);
 115     IUnknown_Release(pUnknown);
 116 }
 117
 118 #define test_shortcut_url(a,b) _test_shortcut_url(__LINE__,a,b)
 119 static void _test_shortcut_url(unsigned line, IUnknown *unk, const char *exurl)
 120 {
 121     IUniformResourceLocatorA *locator_a;
 122     char *url_a;
 123     HRESULT hres;
 124
 125     hres = IUnknown_QueryInterface(unk, &IID_IUniformResourceLocatorA, (void**)&locator_a);
 126     ok_(__FILE__,line)(hres == S_OK, "Could not get IUniformResourceLocatorA iface: %08x\n", hres);
 127
 128     hres = locator_a->lpVtbl->GetURL(locator_a, &url_a);
 129     ok_(__FILE__,line)(hres == S_OK, "GetURL failed: %08x\n", hres);
 130     ok_(__FILE__,line)(!strcmp(url_a, exurl), "unexpected URL, got %s, expected %s\n", url_a, exurl);
 131     CoTaskMemFree(url_a);
 132
 133     IUnknown_Release((IUnknown*)locator_a);
 134 }
 135
 136 #define check_string_transform(a,b,c,d,e) _check_string_transform(__LINE__,a,b,c,d,e)
 137 static void _check_string_transform(unsigned line, IUniformResourceLocatorA *urlA, LPCSTR input, DWORD flags,
 138         LPCSTR expectedOutput, BOOL is_todo)
 139 {
 140     CHAR *output;
 141     HRESULT hr;
 142
 143     hr = urlA->lpVtbl->SetURL(urlA, input, flags);
 144     ok_(__FILE__,line)(hr == S_OK, "SetUrl failed, hr=0x%x\n", hr);
 145     if (FAILED(hr))
 146         return;
 147
 148     output = (void*)0xdeadbeef;
 149     hr = urlA->lpVtbl->GetURL(urlA, &output);
 150     if(expectedOutput) {
 151         if(is_todo) {
 152             todo_wine
 153             ok_(__FILE__,line)(hr == S_OK, "GetUrl failed, hr=0x%x\n", hr);
 154         }else {
 155             ok_(__FILE__,line)(hr == S_OK, "GetUrl failed, hr=0x%x\n", hr);
 156         }
 157         todo_wine
 158         ok_(__FILE__,line)(!lstrcmpA(output, expectedOutput), "unexpected URL change %s -> %s (expected %s)\n",
 159             input, output, expectedOutput);
 160         CoTaskMemFree(output);
 161     }else {
 162         todo_wine
 163         ok_(__FILE__,line)(hr == S_FALSE, "GetUrl failed, hr=0x%x\n", hr);
 164         todo_wine
 165         ok_(__FILE__,line)(!output, "GetUrl returned %s\n", output);
 166     }
 167 }
 168
 169 static void test_ReadAndWriteProperties(void)
 170 {
 171     int iconIndex = 7;
 172     PROPSPEC ps[2];
 173     HRESULT hr;
 174     IUniformResourceLocatorA *urlA;
 175     IUniformResourceLocatorA *urlAFromFile;
 176     WCHAR fileNameW[MAX_PATH];
 177
 178     static const WCHAR shortcutW[] = {'t','e','s','t','s','h','o','r','t','c','u','t','.','u','r','l',0};
 179     WCHAR iconPath[] = {'f','i','l','e',':','/','/','/','C',':','/','a','r','b','i','t','r','a','r','y','/','i','c','o','n','/','p','a','t','h',0};
 180     char testurl[] = "http://some/bogus/url.html";
 181
 182     ps[0].ulKind = PRSPEC_PROPID;
 183     U(ps[0]).propid = PID_IS_ICONFILE;
 184     ps[1].ulKind = PRSPEC_PROPID;
 185     U(ps[1]).propid = PID_IS_ICONINDEX;
 186
 187     /* Make sure we have a valid temporary directory */
 188     GetTempPathW(MAX_PATH, fileNameW);
 189     lstrcatW(fileNameW, shortcutW);
 190
 191     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
 192     ok(hr == S_OK, "Could not create CLSID_InternetShortcut instance: %08x\n", hr);
 193     if (hr == S_OK)
 194     {
 195         IPersistFile *pf;
 196         IPropertyStorage *pPropStgWrite;
 197         IPropertySetStorage *pPropSetStg;
 198         PROPVARIANT pv[2];
 199
 200         /* We need to set a URL -- IPersistFile refuses to save without one. */
 201         hr = urlA->lpVtbl->SetURL(urlA, testurl, 0);
 202         ok(hr == S_OK, "Failed to set a URL.  hr=0x%x\n", hr);
 203
 204         /* Write this shortcut out to a file so that we can test reading it in again. */
 205         hr = urlA->lpVtbl->QueryInterface(urlA, &IID_IPersistFile, (void **) &pf);
 206         ok(hr == S_OK, "Failed to get the IPersistFile for writing.  hr=0x%x\n", hr);
 207
 208         hr = IPersistFile_Save(pf, fileNameW, TRUE);
 209         ok(hr == S_OK, "Failed to save via IPersistFile. hr=0x%x\n", hr);
 210
 211         IPersistFile_Release(pf);
 212
 213         pv[0].vt = VT_LPWSTR;
 214         U(pv[0]).pwszVal = (void *) iconPath;
 215         pv[1].vt = VT_I4;
 216         U(pv[1]).iVal = iconIndex;
 217         hr = urlA->lpVtbl->QueryInterface(urlA, &IID_IPropertySetStorage, (void **) &pPropSetStg);
 218         ok(hr == S_OK, "Unable to get an IPropertySetStorage, hr=0x%x\n", hr);
 219
 220         hr = IPropertySetStorage_Open(pPropSetStg, &FMTID_Intshcut, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, &pPropStgWrite);
 221         ok(hr == S_OK, "Unable to get an IPropertyStorage for writing, hr=0x%x\n", hr);
 222
 223         hr = IPropertyStorage_WriteMultiple(pPropStgWrite, 2, ps, pv, 0);
 224         ok(hr == S_OK, "Unable to set properties, hr=0x%x\n", hr);
 225
 226         hr = IPropertyStorage_Commit(pPropStgWrite, STGC_DEFAULT);
 227         ok(hr == S_OK, "Failed to commit properties, hr=0x%x\n", hr);
 228
 229         pPropStgWrite->lpVtbl->Release(pPropStgWrite);
 230         urlA->lpVtbl->Release(urlA);
 231         IPropertySetStorage_Release(pPropSetStg);
 232     }
 233
 234     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlAFromFile);
 235     ok(hr == S_OK, "Could not create CLSID_InternetShortcut instance: %08x\n", hr);
 236     if (hr == S_OK)
 237     {
 238         IPropertySetStorage *pPropSetStg;
 239         IPropertyStorage *pPropStgRead;
 240         PROPVARIANT pvread[2];
 241         IPersistFile *pf;
 242         LPSTR url = NULL;
 243
 244         /* Now read that .url file back in. */
 245         hr = urlAFromFile->lpVtbl->QueryInterface(urlAFromFile, &IID_IPersistFile, (void **) &pf);
 246         ok(hr == S_OK, "Failed to get the IPersistFile for reading.  hr=0x%x\n", hr);
 247
 248         hr = IPersistFile_Load(pf, fileNameW, 0);
 249         ok(hr == S_OK, "Failed to load via IPersistFile. hr=0x%x\n", hr);
 250         IPersistFile_Release(pf);
 251
 252
 253         hr = urlAFromFile->lpVtbl->GetURL(urlAFromFile, &url);
 254         ok(hr == S_OK, "Unable to get url from file, hr=0x%x\n", hr);
 255         ok(lstrcmpA(url, testurl) == 0, "Wrong url read from file: %s\n",url);
 256
 257
 258         hr = urlAFromFile->lpVtbl->QueryInterface(urlAFromFile, &IID_IPropertySetStorage, (void **) &pPropSetStg);
 259         ok(hr == S_OK, "Unable to get an IPropertySetStorage, hr=0x%x\n", hr);
 260
 261         hr = IPropertySetStorage_Open(pPropSetStg, &FMTID_Intshcut, STGM_READ | STGM_SHARE_EXCLUSIVE, &pPropStgRead);
 262         ok(hr == S_OK, "Unable to get an IPropertyStorage for reading, hr=0x%x\n", hr);
 263
 264         hr = IPropertyStorage_ReadMultiple(pPropStgRead, 2, ps, pvread);
 265         ok(hr == S_OK, "Unable to read properties, hr=0x%x\n", hr);
 266
 267         todo_wine /* Wine doesn't yet support setting properties after save */
 268         {
 269             ok(U(pvread[1]).iVal == iconIndex, "Read wrong icon index: %d\n", U(pvread[1]).iVal);
 270
 271             ok(lstrcmpW(U(pvread[0]).pwszVal, iconPath) == 0, "Wrong icon path read: %s\n", wine_dbgstr_w(U(pvread[0]).pwszVal));
 272         }
 273
 274         PropVariantClear(&pvread[0]);
 275         PropVariantClear(&pvread[1]);
 276         IPropertyStorage_Release(pPropStgRead);
 277         IPropertySetStorage_Release(pPropSetStg);
 278         urlAFromFile->lpVtbl->Release(urlAFromFile);
 279         DeleteFileW(fileNameW);
 280     }
 281 }
 282
 283 static void test_NullURLs(void)
 284 {
 285     LPSTR url = NULL;
 286     HRESULT hr;
 287     IUniformResourceLocatorA *urlA;
 288
 289     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
 290     ok(hr == S_OK, "Could not create InternetShortcut object: %08x\n", hr);
 291
 292     hr = urlA->lpVtbl->GetURL(urlA, &url);
 293     ok(hr == S_FALSE, "getting uninitialized URL unexpectedly failed, hr=0x%x\n", hr);
 294     ok(url == NULL, "uninitialized URL is not NULL but %s\n", url);
 295
 296     hr = urlA->lpVtbl->SetURL(urlA, NULL, 0);
 297     ok(hr == S_OK, "setting NULL URL unexpectedly failed, hr=0x%x\n", hr);
 298
 299     hr = urlA->lpVtbl->GetURL(urlA, &url);
 300     ok(hr == S_FALSE, "getting NULL URL unexpectedly failed, hr=0x%x\n", hr);
 301     ok(url == NULL, "URL unexpectedly not NULL but %s\n", url);
 302
 303     urlA->lpVtbl->Release(urlA);
 304 }
 305
 306 typedef struct {
 307     const char *data;
 308     const char *url;
 309 } load_test_t;
 310
 311 static const load_test_t load_tests[] = {
 312     {"[InternetShortcut]\n"
 313      "URL=http://www.winehq.org/\n"
 314      "HotKey=0\n"
 315      "IDList=\n"
 316      "[{000214A0-0000-0000-C000-000000000046}]\n"
 317      "Prop0=1,2\n",
 318
 319      "http://www.winehq.org/"
 320     }
 321 };
 322
 323 static void test_Load(void)
 324 {
 325     IPersistFile *persist_file;
 326     const load_test_t *test;
 327     WCHAR file_path[MAX_PATH];
 328     DWORD size;
 329     HANDLE file;
 330     HRESULT hres;
 331
 332     static const WCHAR test_urlW[] = {'t','e','s','t','.','u','r','l',0};
 333
 334     GetTempPathW(MAX_PATH, file_path);
 335     lstrcatW(file_path, test_urlW);
 336
 337     for(test = load_tests; test < load_tests + sizeof(load_tests)/sizeof(*load_tests); test++) {
 338         file = CreateFileW(file_path, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
 339         ok(file != INVALID_HANDLE_VALUE, "could not create test file\n");
 340         if(file == INVALID_HANDLE_VALUE)
 341             continue;
 342
 343         WriteFile(file, test->data, strlen(test->data), &size, NULL);
 344         CloseHandle(file);
 345
 346         hres = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IPersistFile, (void**)&persist_file);
 347         ok(hres == S_OK, "Could not create InternetShortcut instance: %08x\n", hres);
 348
 349         hres = IPersistFile_Load(persist_file, file_path, 0);
 350         ok(hres == S_OK, "Load failed: %08x\n", hres);
 351
 352         test_shortcut_url((IUnknown*)persist_file, test->url);
 353
 354         IPersistFile_Release(persist_file);
 355         DeleteFileW(file_path);
 356     }
 357 }
 358
 359 static void test_SetURLFlags(void)
 360 {
 361     HRESULT hr;
 362     IUniformResourceLocatorA *urlA;
 363
 364     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&urlA);
 365     ok(hr == S_OK, "Could not create InternetShortcut object: %08x\n", hr);
 366
 367     check_string_transform(urlA, "somerandomstring", 0, NULL, TRUE);
 368     check_string_transform(urlA, "www.winehq.org", 0, NULL, TRUE);
 369
 370     check_string_transform(urlA, "www.winehq.org", IURL_SETURL_FL_GUESS_PROTOCOL, "http://www.winehq.org/", FALSE);
 371     check_string_transform(urlA, "ftp.winehq.org", IURL_SETURL_FL_GUESS_PROTOCOL, "ftp://ftp.winehq.org/", FALSE);
 372
 373     urlA->lpVtbl->Release(urlA);
 374 }
 375
 376 static void test_InternetShortcut(void)
 377 {
 378     IUniformResourceLocatorA *url;
 379     HRESULT hres;
 380
 381     hres = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL, &IID_IUniformResourceLocatorA, (void**)&url);
 382     ok(hres == S_OK, "Could not create CLSID_InternetShortcut instance: %08x\n", hres);
 383     if(FAILED(hres))
 384         return;
 385
 386     test_Aggregability();
 387     test_QueryInterface();
 388     test_NullURLs();
 389     test_SetURLFlags();
 390     test_ReadAndWriteProperties();
 391     test_Load();
 392 }
 393
 394 START_TEST(intshcut)
 395 {
 396     OleInitialize(NULL);
 397
 398     test_InternetShortcut();
 399
 400     OleUninitialize();
 401 }