dlls/shlwapi/tests/istream.c

   1 /* Unit test suite for SHLWAPI ShCreateStreamOnFile functions.
   2  *
   3  * Copyright 2008 Reece H. Dunn
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #define COBJMACROS
  21
  22 #include <stdarg.h>
  23 #include <stdio.h>
  24
  25 #include "wine/test.h"
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "objbase.h"
  29 #include "shlwapi.h"
  30
  31
  32 /* Function pointers for the SHCreateStreamOnFile functions */
  33 static HMODULE hShlwapi;
  34 static HRESULT (WINAPI *pSHCreateStreamOnFileA)(LPCSTR file, DWORD mode, IStream **stream);
  35 static HRESULT (WINAPI *pSHCreateStreamOnFileW)(LPCWSTR file, DWORD mode, IStream **stream);
  36 static HRESULT (WINAPI *pSHCreateStreamOnFileEx)(LPCWSTR file, DWORD mode, DWORD attributes, BOOL create, IStream *template, IStream **stream);
  37
  38
  39 static void test_IStream_invalid_operations(IStream * stream, DWORD mode)
  40 {
  41     HRESULT ret;
  42     IStream * clone;
  43     ULONG refcount;
  44     ULARGE_INTEGER uzero;
  45     ULARGE_INTEGER uret;
  46     LARGE_INTEGER zero;
  47     ULONG count;
  48     char data[256];
  49
  50     U(uzero).HighPart = 0;
  51     U(uzero).LowPart = 0;
  52     U(uret).HighPart = 0;
  53     U(uret).LowPart = 0;
  54     U(zero).HighPart = 0;
  55     U(zero).LowPart = 0;
  56
  57     /* IStream::Read */
  58
  59     /* IStream_Read from the COBJMACROS is undefined by shlwapi.h, replaced by the IStream_Read helper function. */
  60
  61     ret = stream->lpVtbl->Read(stream, NULL, 0, &count);
  62     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
  63
  64     ret = stream->lpVtbl->Read(stream, data, 5, NULL);
  65     ok(ret == S_FALSE || ret == S_OK, "expected S_FALSE or S_OK, got 0x%08x\n", ret);
  66
  67     ret = stream->lpVtbl->Read(stream, data, 0, NULL);
  68     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
  69
  70     ret = stream->lpVtbl->Read(stream, data, 3, &count);
  71     ok(ret == S_FALSE || ret == S_OK, "expected S_FALSE or S_OK, got 0x%08x\n", ret);
  72
  73     /* IStream::Write */
  74
  75     /* IStream_Write from the COBJMACROS is undefined by shlwapi.h, replaced by the IStream_Write helper function. */
  76
  77     ret = stream->lpVtbl->Write(stream, NULL, 0, &count);
  78     if (mode == STGM_READ)
  79         ok(ret == STG_E_ACCESSDENIED /* XP */ || ret == S_OK /* 2000 */,
  80            "expected STG_E_ACCESSDENIED or S_OK, got 0x%08x\n", ret);
  81     else
  82         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
  83
  84     strcpy(data, "Hello");
  85     ret = stream->lpVtbl->Write(stream, data, 5, NULL);
  86     if (mode == STGM_READ)
  87         ok(ret == STG_E_ACCESSDENIED /* XP */ || ret == S_OK /* 2000 */,
  88            "expected STG_E_ACCESSDENIED or S_OK, got 0x%08x\n", ret);
  89     else
  90         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
  91
  92     strcpy(data, "Hello");
  93     ret = stream->lpVtbl->Write(stream, data, 0, NULL);
  94     if (mode == STGM_READ)
  95         ok(ret == STG_E_ACCESSDENIED /* XP */ || ret == S_OK /* 2000 */,
  96            "expected STG_E_ACCESSDENIED or S_OK, got 0x%08x\n", ret);
  97     else
  98         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
  99
 100     strcpy(data, "Hello");
 101     ret = stream->lpVtbl->Write(stream, data, 0, &count);
 102     if (mode == STGM_READ)
 103         ok(ret == STG_E_ACCESSDENIED /* XP */ || ret == S_OK /* 2000 */,
 104            "expected STG_E_ACCESSDENIED or S_OK, got 0x%08x\n", ret);
 105     else
 106         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 107
 108     strcpy(data, "Hello");
 109     ret = stream->lpVtbl->Write(stream, data, 3, &count);
 110     if (mode == STGM_READ)
 111         ok(ret == STG_E_ACCESSDENIED /* XP */ || ret == S_OK /* 2000 */,
 112            "expected STG_E_ACCESSDENIED or S_OK, got 0x%08x\n", ret);
 113     else
 114         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 115
 116     /* IStream::Seek */
 117
 118     ret = IStream_Seek(stream, zero, STREAM_SEEK_SET, NULL);
 119     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 120
 121     ret = IStream_Seek(stream, zero, 20, NULL);
 122     ok(ret == E_INVALIDARG /* XP */ || ret == S_OK /* 2000 */,
 123        "expected E_INVALIDARG or S_OK, got 0x%08x\n", ret);
 124
 125     /* IStream::CopyTo */
 126
 127     ret = IStream_CopyTo(stream, NULL, uzero, &uret, &uret);
 128     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 129
 130     clone = NULL;
 131     ret = IStream_CopyTo(stream, clone, uzero, &uret, &uret);
 132     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 133
 134     ret = IStream_CopyTo(stream, stream, uzero, &uret, &uret);
 135     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 136
 137     ret = IStream_CopyTo(stream, stream, uzero, &uret, NULL);
 138     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 139
 140     ret = IStream_CopyTo(stream, stream, uzero, NULL, &uret);
 141     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 142
 143     /* IStream::Commit */
 144
 145     ret = IStream_Commit(stream, STGC_DEFAULT);
 146     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 147
 148     /* IStream::Revert */
 149
 150     ret = IStream_Revert(stream);
 151     ok(ret == E_NOTIMPL, "expected E_NOTIMPL, got 0x%08x\n", ret);
 152
 153     /* IStream::LockRegion */
 154
 155     ret = IStream_LockRegion(stream, uzero, uzero, 0);
 156     ok(ret == E_NOTIMPL /* XP */ || ret == S_OK /* Vista */,
 157       "expected E_NOTIMPL or S_OK, got 0x%08x\n", ret);
 158
 159     /* IStream::UnlockRegion */
 160
 161     if (ret == E_NOTIMPL) /* XP */ {
 162         ret = IStream_UnlockRegion(stream, uzero, uzero, 0);
 163         ok(ret == E_NOTIMPL, "expected E_NOTIMPL, got 0x%08x\n", ret);
 164     } else /* Vista */ {
 165         ret = IStream_UnlockRegion(stream, uzero, uzero, 0);
 166         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 167
 168         ret = IStream_UnlockRegion(stream, uzero, uzero, 0);
 169         ok(ret == STG_E_LOCKVIOLATION, "expected STG_E_LOCKVIOLATION, got 0x%08x\n", ret);
 170     }
 171
 172     /* IStream::Stat */
 173
 174     ret = IStream_Stat(stream, NULL, 0);
 175     ok(ret == STG_E_INVALIDPOINTER /* XP */ || ret == E_NOTIMPL /* 2000 */,
 176        "expected STG_E_INVALIDPOINTER or E_NOTIMPL, got 0x%08x\n", ret);
 177
 178     /* IStream::Clone */
 179
 180     /* Passing a NULL pointer for the second IStream::Clone param crashes on Win7 */
 181
 182     clone = NULL;
 183     ret = IStream_Clone(stream, &clone);
 184     ok(ret == E_NOTIMPL, "expected E_NOTIMPL, got 0x%08x\n", ret);
 185     ok(clone == NULL, "expected a NULL IStream object, got %p\n", stream);
 186
 187     if (clone) {
 188         refcount = IStream_Release(clone);
 189         ok(refcount == 0, "expected 0, got %d\n", refcount);
 190     }
 191 }
 192
 193
 194 static void test_stream_read_write(IStream *stream, DWORD mode)
 195 {
 196     static const LARGE_INTEGER start;
 197     HRESULT ret;
 198     unsigned char buf[16];
 199     DWORD written, count;
 200
 201     /* IStream_Read/Write from the COBJMACROS is undefined by shlwapi.h */
 202
 203     written = 0xdeadbeaf;
 204     ret = stream->lpVtbl->Write(stream, "\x5e\xa7", 2, &written);
 205     if (mode == STGM_WRITE || mode == STGM_READWRITE)
 206     {
 207         ok(ret == S_OK, "IStream_Write error %#x (access %#x)\n", ret, mode);
 208         ok(written == 2, "expected 2, got %u\n", written);
 209     }
 210     else
 211     {
 212         ok(ret == STG_E_ACCESSDENIED || broken(ret == S_OK) /* win2000 */, "expected STG_E_ACCESSDENIED, got %#x (access %#x)\n", ret, mode);
 213         ok(written == 0xdeadbeaf || broken(written == 2) /* win2000 */, "expected 0xdeadbeaf, got %#x\n", written);
 214         written = 0;
 215         if (ret == S_OK) return; /* no point in further testing */
 216     }
 217
 218     ret = stream->lpVtbl->Seek(stream, start, STREAM_SEEK_SET, NULL);
 219     ok(ret == S_OK, "Seek error %#x\n", ret);
 220
 221     count = 0xdeadbeaf;
 222     ret = stream->lpVtbl->Read(stream, buf, 2, &count);
 223     if (written != 0)
 224     {
 225         ok(ret == S_OK, "IStream_Read error %#x (access %#x, written %u)\n", ret, mode, written);
 226         if (mode == STGM_WRITE || mode == STGM_READWRITE)
 227         {
 228             ok(count == 2, "expected 2, got %u\n", count);
 229             ok(buf[0] == 0x5e && buf[1] == 0xa7, "expected 5ea7, got %02x%02x\n", buf[0], buf[1]);
 230         }
 231         else
 232             ok(count == 0, "expected 0, got %u\n", count);
 233     }
 234     else
 235     {
 236 todo_wine
 237         ok(ret == S_FALSE, "expected S_FALSE, got %#x (access %#x, written %u)\n", ret, mode, written);
 238         ok(count == 0, "expected 0, got %u\n", count);
 239     }
 240
 241 }
 242
 243 static void test_SHCreateStreamOnFileA(DWORD mode, DWORD stgm)
 244 {
 245     IStream * stream;
 246     HRESULT ret;
 247     ULONG refcount;
 248     char test_file[MAX_PATH];
 249     static const CHAR testA_txt[] = "\\testA.txt";
 250
 251     trace("SHCreateStreamOnFileA: testing mode %d, STGM flags %08x\n", mode, stgm);
 252
 253     /* Don't used a fixed path for the testA.txt file */
 254     GetTempPathA(MAX_PATH, test_file);
 255     lstrcatA(test_file, testA_txt);
 256
 257     /* invalid arguments */
 258
 259     stream = NULL;
 260     ret = (*pSHCreateStreamOnFileA)(NULL, mode | stgm, &stream);
 261     if (ret == E_INVALIDARG) /* Win98 SE */ {
 262         win_skip("Not supported\n");
 263         return;
 264     }
 265
 266     ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) /* NT */ ||
 267        ret == HRESULT_FROM_WIN32(ERROR_BAD_PATHNAME) /* 9x */,
 268        "SHCreateStreamOnFileA: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) "
 269        "or HRESULT_FROM_WIN32(ERROR_BAD_PATHNAME), got 0x%08x\n", ret);
 270     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 271
 272 #if 0 /* This test crashes on WinXP SP2 */
 273     ret = (*pSHCreateStreamOnFileA)(test_file, mode | stgm, NULL);
 274     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
 275 #endif
 276
 277     stream = NULL;
 278     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_CONVERT | stgm, &stream);
 279     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
 280     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 281
 282     stream = NULL;
 283     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_DELETEONRELEASE | stgm, &stream);
 284     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
 285     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 286
 287     stream = NULL;
 288     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_TRANSACTED | stgm, &stream);
 289     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
 290     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 291
 292     /* file does not exist */
 293
 294     stream = NULL;
 295     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
 296     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileA: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
 297     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 298
 299     stream = NULL;
 300     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_CREATE | stgm, &stream);
 301     ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
 302     ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
 303
 304     if (stream) {
 305         test_IStream_invalid_operations(stream, mode);
 306
 307         refcount = IStream_Release(stream);
 308         ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
 309     }
 310
 311     /* NOTE: don't delete the file, as it will be used for the file exists tests. */
 312
 313     /* file exists */
 314
 315     stream = NULL;
 316     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
 317     ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
 318     ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
 319
 320     if (stream) {
 321         test_IStream_invalid_operations(stream, mode);
 322
 323         refcount = IStream_Release(stream);
 324         ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
 325     }
 326
 327     stream = NULL;
 328     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_CREATE | stgm, &stream);
 329     ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
 330     ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
 331
 332     if (stream) {
 333         BOOL delret;
 334
 335         test_stream_read_write(stream, mode);
 336         test_IStream_invalid_operations(stream, mode);
 337
 338         refcount = IStream_Release(stream);
 339         ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
 340
 341         delret = DeleteFileA(test_file);
 342         ok(delret, "SHCreateStreamOnFileA: could not delete file '%s', got error %d\n",
 343            test_file, GetLastError());
 344     }
 345 }
 346
 347
 348 static void test_SHCreateStreamOnFileW(DWORD mode, DWORD stgm)
 349 {
 350     IStream * stream;
 351     HRESULT ret;
 352     ULONG refcount;
 353     WCHAR test_file[MAX_PATH];
 354     CHAR  test_fileA[MAX_PATH];
 355     static const CHAR testW_txt[] = "\\testW.txt";
 356
 357     trace("SHCreateStreamOnFileW: testing mode %d, STGM flags %08x\n", mode, stgm);
 358
 359     /* Don't used a fixed path for the testW.txt file */
 360     GetTempPathA(MAX_PATH, test_fileA);
 361     lstrcatA(test_fileA, testW_txt);
 362     MultiByteToWideChar(CP_ACP, 0, test_fileA, -1, test_file, MAX_PATH);
 363
 364     /* invalid arguments */
 365
 366     if (0)
 367     {
 368         /* Crashes on NT4 */
 369         stream = NULL;
 370         ret = (*pSHCreateStreamOnFileW)(NULL, mode | stgm, &stream);
 371         ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) || /* XP */
 372            ret == E_INVALIDARG /* Vista */,
 373           "SHCreateStreamOnFileW: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) or E_INVALIDARG, got 0x%08x\n", ret);
 374         ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 375     }
 376
 377     if (0)
 378     {
 379         /* This test crashes on WinXP SP2 */
 380             ret = (*pSHCreateStreamOnFileW)(test_file, mode | stgm, NULL);
 381             ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
 382     }
 383
 384     stream = NULL;
 385     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_CONVERT | stgm, &stream);
 386     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
 387     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 388
 389     stream = NULL;
 390     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_DELETEONRELEASE | stgm, &stream);
 391     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
 392     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 393
 394     stream = NULL;
 395     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_TRANSACTED | stgm, &stream);
 396     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
 397     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 398
 399     /* file does not exist */
 400
 401     stream = NULL;
 402     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
 403     if (ret == E_INVALIDARG) /* Win98 SE */ {
 404         win_skip("Not supported\n");
 405         return;
 406     }
 407
 408     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileW: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
 409     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 410
 411     stream = NULL;
 412     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_CREATE | stgm, &stream);
 413     ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
 414     ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
 415
 416     if (stream) {
 417         test_IStream_invalid_operations(stream, mode);
 418
 419         refcount = IStream_Release(stream);
 420         ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
 421     }
 422
 423     /* NOTE: don't delete the file, as it will be used for the file exists tests. */
 424
 425     /* file exists */
 426
 427     stream = NULL;
 428     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
 429     ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
 430     ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
 431
 432     if (stream) {
 433         test_IStream_invalid_operations(stream, mode);
 434
 435         refcount = IStream_Release(stream);
 436         ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
 437     }
 438
 439     stream = NULL;
 440     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_CREATE | stgm, &stream);
 441     ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
 442     ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
 443
 444     if (stream) {
 445         BOOL delret;
 446
 447         test_stream_read_write(stream, mode);
 448         test_IStream_invalid_operations(stream, mode);
 449
 450         refcount = IStream_Release(stream);
 451         ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
 452
 453         delret = DeleteFileA(test_fileA);
 454         ok(delret, "SHCreateStreamOnFileW: could not delete the test file, got error %d\n",
 455            GetLastError());
 456     }
 457 }
 458
 459
 460 static void test_SHCreateStreamOnFileEx(DWORD mode, DWORD stgm)
 461 {
 462     IStream * stream;
 463     IStream * template = NULL;
 464     HRESULT ret;
 465     ULONG refcount;
 466     WCHAR test_file[MAX_PATH];
 467     CHAR  test_fileA[MAX_PATH];
 468     static const CHAR testEx_txt[] = "\\testEx.txt";
 469     BOOL delret;
 470
 471     trace("SHCreateStreamOnFileEx: testing mode %d, STGM flags %08x\n", mode, stgm);
 472
 473     /* Don't used a fixed path for the testEx.txt file */
 474     GetTempPathA(MAX_PATH, test_fileA);
 475     lstrcatA(test_fileA, testEx_txt);
 476     MultiByteToWideChar(CP_ACP, 0, test_fileA, -1, test_file, MAX_PATH);
 477
 478     /* invalid arguments */
 479
 480     if (0)
 481     {
 482         /* Crashes on NT4 */
 483         stream = NULL;
 484         ret = (*pSHCreateStreamOnFileEx)(NULL, mode, 0, FALSE, NULL, &stream);
 485         ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) || /* XP */
 486            ret == E_INVALIDARG /* Vista */,
 487           "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) or E_INVALIDARG, got 0x%08x\n", ret);
 488         ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
 489     }
 490
 491     stream = NULL;
 492     ret = (*pSHCreateStreamOnFileEx)(test_file, mode, 0, FALSE, template, &stream);
 493     if (ret == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED)) {
 494         win_skip("File probably locked by Anti-Virus/Spam software, trying again\n");
 495         Sleep(1000);
 496         ret = (*pSHCreateStreamOnFileEx)(test_file, mode, 0, FALSE, template, &stream);
 497     }
 498     ok( ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
 499         ret == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER),
 500         "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or "
 501         "HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got 0x%08x\n", ret);
 502
 503     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
 504
 505     if (0)
 506     {
 507         /* This test crashes on WinXP SP2 */
 508         ret = (*pSHCreateStreamOnFileEx)(test_file, mode, 0, FALSE, NULL, NULL);
 509         ok(ret == E_INVALIDARG, "SHCreateStreamOnFileEx: expected E_INVALIDARG, got 0x%08x\n", ret);
 510     }
 511
 512     /* file does not exist */
 513
 514     stream = NULL;
 515     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, FALSE, NULL, &stream);
 516     if ((stgm & STGM_TRANSACTED) == STGM_TRANSACTED && mode == STGM_READ) {
 517         ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) /* XP */ || ret == E_INVALIDARG /* Vista */,
 518           "SHCreateStreamOnFileEx: expected E_INVALIDARG or HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
 519
 520         if (ret == E_INVALIDARG) {
 521             skip("SHCreateStreamOnFileEx: STGM_TRANSACTED not supported in this configuration.\n");
 522             return;
 523         }
 524     } else {
 525         ok( ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
 526             ret == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER),
 527             "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or "
 528             "HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got 0x%08x\n", ret);
 529     }
 530     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
 531
 532     stream = NULL;
 533     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, TRUE, NULL, &stream);
 534     /* not supported on win9x */
 535     if (broken(ret == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER) && stream == NULL)) {
 536         skip("Not supported\n");
 537         DeleteFileA(test_fileA);
 538         return;
 539     }
 540
 541     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 542     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 543
 544     if (stream) {
 545         test_IStream_invalid_operations(stream, mode);
 546
 547         refcount = IStream_Release(stream);
 548         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 549
 550         delret = DeleteFileA(test_fileA);
 551         ok(delret, "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n",
 552            GetLastError());
 553     }
 554
 555     stream = NULL;
 556     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
 557     if (ret == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED)) {
 558         win_skip("File probably locked by Anti-Virus/Spam software, trying again\n");
 559         Sleep(1000);
 560         ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
 561     }
 562     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 563     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 564
 565     if (stream) {
 566         test_IStream_invalid_operations(stream, mode);
 567
 568         refcount = IStream_Release(stream);
 569         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 570
 571         delret = DeleteFileA(test_fileA);
 572         ok(delret, "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n",
 573            GetLastError());
 574     }
 575
 576     stream = NULL;
 577     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
 578     if (ret == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED)) {
 579         win_skip("File probably locked by Anti-Virus/Spam software, trying again\n");
 580         Sleep(1000);
 581         ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
 582     }
 583     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 584     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 585
 586     if (stream) {
 587         test_IStream_invalid_operations(stream, mode);
 588
 589         refcount = IStream_Release(stream);
 590         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 591     }
 592
 593     /* NOTE: don't delete the file, as it will be used for the file exists tests. */
 594
 595     /* file exists */
 596
 597     stream = NULL;
 598     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, FALSE, NULL, &stream);
 599     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 600     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 601
 602     if (stream) {
 603         test_IStream_invalid_operations(stream, mode);
 604
 605         refcount = IStream_Release(stream);
 606         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 607     }
 608
 609     stream = NULL;
 610     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, TRUE, NULL, &stream);
 611     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_EXISTS), "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_EXISTS), got 0x%08x\n", ret);
 612     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
 613
 614     stream = NULL;
 615     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
 616     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 617     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 618
 619     if (stream) {
 620         test_IStream_invalid_operations(stream, mode);
 621
 622         refcount = IStream_Release(stream);
 623         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 624     }
 625
 626     stream = NULL;
 627     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
 628     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 629     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 630
 631     if (stream) {
 632         test_IStream_invalid_operations(stream, mode);
 633
 634         refcount = IStream_Release(stream);
 635         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 636     }
 637
 638     delret = DeleteFileA(test_fileA);
 639     ok(delret, "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n",
 640        GetLastError());
 641 }
 642
 643
 644 START_TEST(istream)
 645 {
 646     static const DWORD stgm_access[] = {
 647         STGM_READ,
 648         STGM_WRITE,
 649         STGM_READWRITE
 650     };
 651
 652     static const DWORD stgm_sharing[] = {
 653         0,
 654         STGM_SHARE_DENY_NONE,
 655         STGM_SHARE_DENY_READ,
 656         STGM_SHARE_DENY_WRITE,
 657         STGM_SHARE_EXCLUSIVE
 658     };
 659
 660     static const DWORD stgm_flags[] = {
 661         0,
 662         STGM_CONVERT,
 663         STGM_DELETEONRELEASE,
 664         STGM_CONVERT | STGM_DELETEONRELEASE,
 665         STGM_TRANSACTED | STGM_CONVERT,
 666         STGM_TRANSACTED | STGM_DELETEONRELEASE,
 667         STGM_TRANSACTED | STGM_CONVERT | STGM_DELETEONRELEASE
 668     };
 669
 670     int i, j, k;
 671
 672     hShlwapi = LoadLibrary("shlwapi.dll");
 673
 674     pSHCreateStreamOnFileA = (void*)GetProcAddress(hShlwapi, "SHCreateStreamOnFileA");
 675     pSHCreateStreamOnFileW = (void*)GetProcAddress(hShlwapi, "SHCreateStreamOnFileW");
 676     pSHCreateStreamOnFileEx = (void*)GetProcAddress(hShlwapi, "SHCreateStreamOnFileEx");
 677
 678     if (!pSHCreateStreamOnFileA)
 679         skip("SHCreateStreamOnFileA not found.\n");
 680
 681     if (!pSHCreateStreamOnFileW)
 682         skip("SHCreateStreamOnFileW not found.\n");
 683
 684     if (!pSHCreateStreamOnFileEx)
 685         skip("SHCreateStreamOnFileEx not found.\n");
 686
 687     for (i = 0; i != sizeof(stgm_access)/sizeof(stgm_access[0]); i++) {
 688         for (j = 0; j != sizeof(stgm_sharing)/sizeof(stgm_sharing[0]); j ++) {
 689             if (pSHCreateStreamOnFileA)
 690                 test_SHCreateStreamOnFileA(stgm_access[i], stgm_sharing[j]);
 691
 692             if (pSHCreateStreamOnFileW)
 693                 test_SHCreateStreamOnFileW(stgm_access[i], stgm_sharing[j]);
 694
 695             if (pSHCreateStreamOnFileEx) {
 696                 for (k = 0; k != sizeof(stgm_flags)/sizeof(stgm_flags[0]); k++)
 697                     test_SHCreateStreamOnFileEx(stgm_access[i], stgm_sharing[j] | stgm_flags[k]);
 698             }
 699         }
 700     }
 701 }