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, "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
  80     else
  81         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
  82
  83     strcpy(data, "Hello");
  84     ret = stream->lpVtbl->Write(stream, data, 5, NULL);
  85     if (mode == STGM_READ)
  86         ok(ret == STG_E_ACCESSDENIED, "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
  87     else
  88         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
  89
  90     strcpy(data, "Hello");
  91     ret = stream->lpVtbl->Write(stream, data, 0, NULL);
  92     if (mode == STGM_READ)
  93         ok(ret == STG_E_ACCESSDENIED, "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
  94     else
  95         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
  96
  97     strcpy(data, "Hello");
  98     ret = stream->lpVtbl->Write(stream, data, 0, &count);
  99     if (mode == STGM_READ)
 100         ok(ret == STG_E_ACCESSDENIED, "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
 101     else
 102         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 103
 104     strcpy(data, "Hello");
 105     ret = stream->lpVtbl->Write(stream, data, 3, &count);
 106     if (mode == STGM_READ)
 107         ok(ret == STG_E_ACCESSDENIED, "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
 108     else
 109         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 110
 111     /* IStream::Seek */
 112
 113     ret = IStream_Seek(stream, zero, STREAM_SEEK_SET, NULL);
 114     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 115
 116     ret = IStream_Seek(stream, zero, 20, NULL);
 117     ok(ret == E_INVALIDARG, "expected E_INVALIDARG, got 0x%08x\n", ret);
 118
 119     /* IStream::CopyTo */
 120
 121     ret = IStream_CopyTo(stream, NULL, uzero, &uret, &uret);
 122     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 123
 124     clone = NULL;
 125     ret = IStream_CopyTo(stream, clone, uzero, &uret, &uret);
 126     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 127
 128     ret = IStream_CopyTo(stream, stream, uzero, &uret, &uret);
 129     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 130
 131     ret = IStream_CopyTo(stream, stream, uzero, &uret, NULL);
 132     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 133
 134     ret = IStream_CopyTo(stream, stream, uzero, NULL, &uret);
 135     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 136
 137     /* IStream::Commit */
 138
 139     ret = IStream_Commit(stream, STGC_DEFAULT);
 140     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 141
 142     /* IStream::Revert */
 143
 144     ret = IStream_Revert(stream);
 145     ok(ret == E_NOTIMPL, "expected E_NOTIMPL, got 0x%08x\n", ret);
 146
 147     /* IStream::LockRegion */
 148
 149     ret = IStream_LockRegion(stream, uzero, uzero, 0);
 150     ok(ret == E_NOTIMPL /* XP */ || ret == S_OK /* Vista */,
 151       "expected E_NOTIMPL or S_OK, got 0x%08x\n", ret);
 152
 153     /* IStream::UnlockRegion */
 154
 155     if (ret == E_NOTIMPL) /* XP */ {
 156         ret = IStream_UnlockRegion(stream, uzero, uzero, 0);
 157         ok(ret == E_NOTIMPL, "expected E_NOTIMPL, got 0x%08x\n", ret);
 158     } else /* Vista */ {
 159         ret = IStream_UnlockRegion(stream, uzero, uzero, 0);
 160         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
 161
 162         ret = IStream_UnlockRegion(stream, uzero, uzero, 0);
 163         ok(ret == STG_E_LOCKVIOLATION, "expected STG_E_LOCKVIOLATION, got 0x%08x\n", ret);
 164     }
 165
 166     /* IStream::Stat */
 167
 168     ret = IStream_Stat(stream, NULL, 0);
 169     ok(ret == STG_E_INVALIDPOINTER, "expected STG_E_INVALIDPOINTER, got 0x%08x\n", ret);
 170
 171     /* IStream::Clone */
 172
 173     ret = IStream_Clone(stream, NULL);
 174     ok(ret == E_NOTIMPL, "expected E_NOTIMPL, got 0x%08x\n", ret);
 175
 176     clone = NULL;
 177     ret = IStream_Clone(stream, &clone);
 178     ok(ret == E_NOTIMPL, "expected E_NOTIMPL, got 0x%08x\n", ret);
 179     ok(clone == NULL, "expected a NULL IStream object, got %p\n", stream);
 180
 181     if (clone) {
 182         refcount = IStream_Release(clone);
 183         ok(refcount == 0, "expected 0, got %d\n", refcount);
 184     }
 185 }
 186
 187
 188 static void test_SHCreateStreamOnFileA(DWORD mode, DWORD stgm)
 189 {
 190     IStream * stream;
 191     HRESULT ret;
 192     ULONG refcount;
 193     static const char * test_file = "c:\\test.txt";
 194
 195     trace("SHCreateStreamOnFileA: testing mode %d, STGM flags %08x\n", mode, stgm);
 196
 197     /* invalid arguments */
 198
 199     stream = NULL;
 200     ret = (*pSHCreateStreamOnFileA)(NULL, mode | stgm, &stream);
 201     ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "SHCreateStreamOnFileA: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), got 0x%08x\n", ret);
 202     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 203
 204 #if 0 /* This test crashes on WinXP SP2 */
 205     ret = (*pSHCreateStreamOnFileA)(test_file, mode | stgm, NULL);
 206     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
 207 #endif
 208
 209     stream = NULL;
 210     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_CONVERT | stgm, &stream);
 211     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
 212     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 213
 214     stream = NULL;
 215     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_DELETEONRELEASE | stgm, &stream);
 216     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
 217     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 218
 219     stream = NULL;
 220     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_TRANSACTED | stgm, &stream);
 221     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
 222     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 223
 224     /* file does not exist */
 225
 226     stream = NULL;
 227     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
 228     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileA: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
 229     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
 230
 231     stream = NULL;
 232     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_CREATE | stgm, &stream);
 233     ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
 234     ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
 235
 236     if (stream) {
 237         test_IStream_invalid_operations(stream, mode);
 238
 239         refcount = IStream_Release(stream);
 240         ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
 241     }
 242
 243     /* NOTE: don't delete the file, as it will be used for the file exists tests. */
 244
 245     /* file exists */
 246
 247     stream = NULL;
 248     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
 249     ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
 250     ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
 251
 252     if (stream) {
 253         test_IStream_invalid_operations(stream, mode);
 254
 255         refcount = IStream_Release(stream);
 256         ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
 257     }
 258
 259     stream = NULL;
 260     ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_CREATE | stgm, &stream);
 261     ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
 262     ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
 263
 264     if (stream) {
 265         test_IStream_invalid_operations(stream, mode);
 266
 267         refcount = IStream_Release(stream);
 268         ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
 269
 270         ok(DeleteFileA(test_file), "SHCreateStreamOnFileA: could not delete file '%s', got error %d\n", test_file, GetLastError());
 271     }
 272 }
 273
 274
 275 static void test_SHCreateStreamOnFileW(DWORD mode, DWORD stgm)
 276 {
 277     IStream * stream;
 278     HRESULT ret;
 279     ULONG refcount;
 280     static const WCHAR test_file[] = { 'c', ':', '\\', 't', 'e', 's', 't', '.', 't', 'x', 't', '\0' };
 281
 282     trace("SHCreateStreamOnFileW: testing mode %d, STGM flags %08x\n", mode, stgm);
 283
 284     /* invalid arguments */
 285
 286     stream = NULL;
 287     ret = (*pSHCreateStreamOnFileW)(NULL, mode | stgm, &stream);
 288     ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) || /* XP */
 289        ret == E_INVALIDARG /* Vista */,
 290       "SHCreateStreamOnFileW: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) or E_INVALIDARG, got 0x%08x\n", ret);
 291     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 292
 293 #if 0 /* This test crashes on WinXP SP2 */
 294     ret = (*pSHCreateStreamOnFileW)(test_file, mode | stgm, NULL);
 295     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
 296 #endif
 297
 298     stream = NULL;
 299     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_CONVERT | stgm, &stream);
 300     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
 301     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 302
 303     stream = NULL;
 304     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_DELETEONRELEASE | stgm, &stream);
 305     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
 306     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 307
 308     stream = NULL;
 309     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_TRANSACTED | stgm, &stream);
 310     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
 311     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 312
 313     /* file does not exist */
 314
 315     stream = NULL;
 316     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
 317     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileW: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
 318     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
 319
 320     stream = NULL;
 321     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_CREATE | stgm, &stream);
 322     ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
 323     ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
 324
 325     if (stream) {
 326         test_IStream_invalid_operations(stream, mode);
 327
 328         refcount = IStream_Release(stream);
 329         ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
 330     }
 331
 332     /* NOTE: don't delete the file, as it will be used for the file exists tests. */
 333
 334     /* file exists */
 335
 336     stream = NULL;
 337     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
 338     ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
 339     ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
 340
 341     if (stream) {
 342         test_IStream_invalid_operations(stream, mode);
 343
 344         refcount = IStream_Release(stream);
 345         ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
 346     }
 347
 348     stream = NULL;
 349     ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_CREATE | stgm, &stream);
 350     ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
 351     ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
 352
 353     if (stream) {
 354         test_IStream_invalid_operations(stream, mode);
 355
 356         refcount = IStream_Release(stream);
 357         ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
 358
 359         ok(DeleteFileW(test_file), "SHCreateStreamOnFileW: could not delete the test file, got error %d\n", GetLastError());
 360     }
 361 }
 362
 363
 364 static void test_SHCreateStreamOnFileEx(DWORD mode, DWORD stgm)
 365 {
 366     IStream * stream;
 367     IStream * template = NULL;
 368     HRESULT ret;
 369     ULONG refcount;
 370     static const WCHAR test_file[] = { 'c', ':', '\\', 't', 'e', 's', 't', '.', 't', 'x', 't', '\0' };
 371
 372     trace("SHCreateStreamOnFileEx: testing mode %d, STGM flags %08x\n", mode, stgm);
 373
 374     /* invalid arguments */
 375
 376     stream = NULL;
 377     ret = (*pSHCreateStreamOnFileEx)(NULL, mode, 0, FALSE, NULL, &stream);
 378     ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) || /* XP */
 379        ret == E_INVALIDARG /* Vista */,
 380       "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) or E_INVALIDARG, got 0x%08x\n", ret);
 381     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
 382
 383     stream = NULL;
 384     ret = (*pSHCreateStreamOnFileEx)(test_file, mode, 0, FALSE, template, &stream);
 385     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
 386     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
 387
 388 #if 0 /* This test crashes on WinXP SP2 */
 389     ret = (*pSHCreateStreamOnFileEx)(test_file, mode, 0, FALSE, NULL, NULL);
 390     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileEx: expected E_INVALIDARG, got 0x%08x\n", ret);
 391 #endif
 392
 393     /* file does not exist */
 394
 395     stream = NULL;
 396     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, FALSE, NULL, &stream);
 397     if ((stgm & STGM_TRANSACTED) == STGM_TRANSACTED && mode == STGM_READ) {
 398         ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) /* XP */ || ret == E_INVALIDARG /* Vista */,
 399           "SHCreateStreamOnFileEx: expected E_INVALIDARG or HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
 400
 401         if (ret == E_INVALIDARG) {
 402             skip("SHCreateStreamOnFileEx: STGM_TRANSACTED not supported in this configuration.\n");
 403             return;
 404         }
 405     } else {
 406         ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
 407     }
 408     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
 409
 410     stream = NULL;
 411     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, TRUE, NULL, &stream);
 412     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 413     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 414
 415     if (stream) {
 416         test_IStream_invalid_operations(stream, mode);
 417
 418         refcount = IStream_Release(stream);
 419         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 420
 421         ok(DeleteFileW(test_file), "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n", GetLastError());
 422     }
 423
 424     stream = NULL;
 425     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
 426     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 427     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 428
 429     if (stream) {
 430         test_IStream_invalid_operations(stream, mode);
 431
 432         refcount = IStream_Release(stream);
 433         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 434
 435         ok(DeleteFileW(test_file), "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n", GetLastError());
 436     }
 437
 438     stream = NULL;
 439     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
 440     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 441     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 442
 443     if (stream) {
 444         test_IStream_invalid_operations(stream, mode);
 445
 446         refcount = IStream_Release(stream);
 447         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 448     }
 449
 450     /* NOTE: don't delete the file, as it will be used for the file exists tests. */
 451
 452     /* file exists */
 453
 454     stream = NULL;
 455     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, FALSE, NULL, &stream);
 456     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 457     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 458
 459     if (stream) {
 460         test_IStream_invalid_operations(stream, mode);
 461
 462         refcount = IStream_Release(stream);
 463         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 464     }
 465
 466     stream = NULL;
 467     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, TRUE, NULL, &stream);
 468     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_EXISTS), "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_EXISTS), got 0x%08x\n", ret);
 469     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
 470
 471     stream = NULL;
 472     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
 473     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 474     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 475
 476     if (stream) {
 477         test_IStream_invalid_operations(stream, mode);
 478
 479         refcount = IStream_Release(stream);
 480         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 481     }
 482
 483     stream = NULL;
 484     ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
 485     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
 486     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
 487
 488     if (stream) {
 489         test_IStream_invalid_operations(stream, mode);
 490
 491         refcount = IStream_Release(stream);
 492         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
 493     }
 494
 495     ok(DeleteFileW(test_file), "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n", GetLastError());
 496 }
 497
 498
 499 START_TEST(istream)
 500 {
 501     static const DWORD stgm_access[] = {
 502         STGM_READ,
 503         STGM_WRITE,
 504         STGM_READWRITE
 505     };
 506
 507     static const DWORD stgm_sharing[] = {
 508         0,
 509         STGM_SHARE_DENY_NONE,
 510         STGM_SHARE_DENY_READ,
 511         STGM_SHARE_DENY_WRITE,
 512         STGM_SHARE_EXCLUSIVE
 513     };
 514
 515     static const DWORD stgm_flags[] = {
 516         0,
 517         STGM_CONVERT,
 518         STGM_DELETEONRELEASE,
 519         STGM_CONVERT | STGM_DELETEONRELEASE,
 520         STGM_TRANSACTED | STGM_CONVERT,
 521         STGM_TRANSACTED | STGM_DELETEONRELEASE,
 522         STGM_TRANSACTED | STGM_CONVERT | STGM_DELETEONRELEASE
 523     };
 524
 525     int i, j, k;
 526
 527     hShlwapi = GetModuleHandleA("shlwapi.dll");
 528
 529     pSHCreateStreamOnFileA = (void*)GetProcAddress(hShlwapi, "SHCreateStreamOnFileA");
 530     pSHCreateStreamOnFileW = (void*)GetProcAddress(hShlwapi, "SHCreateStreamOnFileW");
 531     pSHCreateStreamOnFileEx = (void*)GetProcAddress(hShlwapi, "SHCreateStreamOnFileEx");
 532
 533     if (!pSHCreateStreamOnFileA)
 534         skip("SHCreateStreamOnFileA not found.\n");
 535
 536     if (!pSHCreateStreamOnFileW)
 537         skip("SHCreateStreamOnFileW not found.\n");
 538
 539     if (!pSHCreateStreamOnFileEx)
 540         skip("SHCreateStreamOnFileEx not found.\n");
 541
 542     for (i = 0; i != sizeof(stgm_access)/sizeof(stgm_access[0]); i++) {
 543         for (j = 0; j != sizeof(stgm_sharing)/sizeof(stgm_sharing[0]); j ++) {
 544             if (pSHCreateStreamOnFileA)
 545                 test_SHCreateStreamOnFileA(stgm_access[i], stgm_sharing[j]);
 546
 547             if (pSHCreateStreamOnFileW)
 548                 test_SHCreateStreamOnFileW(stgm_access[i], stgm_sharing[j]);
 549
 550             if (pSHCreateStreamOnFileEx) {
 551                 for (k = 0; k != sizeof(stgm_flags)/sizeof(stgm_flags[0]); k++)
 552                     test_SHCreateStreamOnFileEx(stgm_access[i], stgm_sharing[j] | stgm_flags[k]);
 553             }
 554         }
 555     }
 556 }