shlwapi: Be less strict on which type of IShellFolder can be enumerated.
[wine/hramrach.git] / dlls / shlwapi / tests / istream.c
blob4a36b3ab0baee4a496db9908878974574acf8ce6
1 /* Unit test suite for SHLWAPI ShCreateStreamOnFile functions.
3 * Copyright 2008 Reece H. Dunn
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.
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.
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
20 #define COBJMACROS
22 #include <stdarg.h>
23 #include <stdio.h>
25 #include "wine/test.h"
26 #include "windef.h"
27 #include "winbase.h"
28 #include "objbase.h"
29 #include "shlwapi.h"
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);
39 static void test_IStream_invalid_operations(IStream * stream, DWORD mode)
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];
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;
57 /* IStream::Read */
59 /* IStream_Read from the COBJMACROS is undefined by shlwapi.h, replaced by the IStream_Read helper function. */
61 ret = stream->lpVtbl->Read(stream, NULL, 0, &count);
62 ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
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);
67 ret = stream->lpVtbl->Read(stream, data, 0, NULL);
68 ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
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);
73 /* IStream::Write */
75 /* IStream_Write from the COBJMACROS is undefined by shlwapi.h, replaced by the IStream_Write helper function. */
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);
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);
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);
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);
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);
116 /* IStream::Seek */
118 ret = IStream_Seek(stream, zero, STREAM_SEEK_SET, NULL);
119 ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
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);
125 /* IStream::CopyTo */
127 ret = IStream_CopyTo(stream, NULL, uzero, &uret, &uret);
128 ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
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);
134 ret = IStream_CopyTo(stream, stream, uzero, &uret, &uret);
135 ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
137 ret = IStream_CopyTo(stream, stream, uzero, &uret, NULL);
138 ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
140 ret = IStream_CopyTo(stream, stream, uzero, NULL, &uret);
141 ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
143 /* IStream::Commit */
145 ret = IStream_Commit(stream, STGC_DEFAULT);
146 ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
148 /* IStream::Revert */
150 ret = IStream_Revert(stream);
151 ok(ret == E_NOTIMPL, "expected E_NOTIMPL, got 0x%08x\n", ret);
153 /* IStream::LockRegion */
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);
159 /* IStream::UnlockRegion */
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);
168 ret = IStream_UnlockRegion(stream, uzero, uzero, 0);
169 ok(ret == STG_E_LOCKVIOLATION, "expected STG_E_LOCKVIOLATION, got 0x%08x\n", ret);
172 /* IStream::Stat */
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);
178 /* IStream::Clone */
180 /* Passing a NULL pointer for the second IStream::Clone param crashes on Win7 */
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);
187 if (clone) {
188 refcount = IStream_Release(clone);
189 ok(refcount == 0, "expected 0, got %d\n", refcount);
194 static void test_SHCreateStreamOnFileA(DWORD mode, DWORD stgm)
196 IStream * stream;
197 HRESULT ret;
198 ULONG refcount;
199 char test_file[MAX_PATH];
200 static const CHAR testA_txt[] = "\\testA.txt";
202 trace("SHCreateStreamOnFileA: testing mode %d, STGM flags %08x\n", mode, stgm);
204 /* Don't used a fixed path for the testA.txt file */
205 GetTempPathA(MAX_PATH, test_file);
206 lstrcatA(test_file, testA_txt);
208 /* invalid arguments */
210 stream = NULL;
211 ret = (*pSHCreateStreamOnFileA)(NULL, mode | stgm, &stream);
212 if (ret == E_INVALIDARG) /* Win98 SE */ {
213 win_skip("Not supported\n");
214 return;
217 ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) /* NT */ ||
218 ret == HRESULT_FROM_WIN32(ERROR_BAD_PATHNAME) /* 9x */,
219 "SHCreateStreamOnFileA: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) "
220 "or HRESULT_FROM_WIN32(ERROR_BAD_PATHNAME), got 0x%08x\n", ret);
221 ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
223 #if 0 /* This test crashes on WinXP SP2 */
224 ret = (*pSHCreateStreamOnFileA)(test_file, mode | stgm, NULL);
225 ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
226 #endif
228 stream = NULL;
229 ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_CONVERT | stgm, &stream);
230 ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
231 ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
233 stream = NULL;
234 ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_DELETEONRELEASE | stgm, &stream);
235 ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
236 ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
238 stream = NULL;
239 ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_TRANSACTED | stgm, &stream);
240 ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
241 ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
243 /* file does not exist */
245 stream = NULL;
246 ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
247 ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileA: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
248 ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
250 stream = NULL;
251 ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_CREATE | stgm, &stream);
252 ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
253 ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
255 if (stream) {
256 test_IStream_invalid_operations(stream, mode);
258 refcount = IStream_Release(stream);
259 ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
262 /* NOTE: don't delete the file, as it will be used for the file exists tests. */
264 /* file exists */
266 stream = NULL;
267 ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
268 ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
269 ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
271 if (stream) {
272 test_IStream_invalid_operations(stream, mode);
274 refcount = IStream_Release(stream);
275 ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
278 stream = NULL;
279 ret = (*pSHCreateStreamOnFileA)(test_file, mode | STGM_CREATE | stgm, &stream);
280 ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
281 ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
283 if (stream) {
284 test_IStream_invalid_operations(stream, mode);
286 refcount = IStream_Release(stream);
287 ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
289 ok(DeleteFileA(test_file), "SHCreateStreamOnFileA: could not delete file '%s', got error %d\n", test_file, GetLastError());
294 static void test_SHCreateStreamOnFileW(DWORD mode, DWORD stgm)
296 IStream * stream;
297 HRESULT ret;
298 ULONG refcount;
299 WCHAR test_file[MAX_PATH];
300 CHAR test_fileA[MAX_PATH];
301 static const CHAR testW_txt[] = "\\testW.txt";
303 trace("SHCreateStreamOnFileW: testing mode %d, STGM flags %08x\n", mode, stgm);
305 /* Don't used a fixed path for the testW.txt file */
306 GetTempPathA(MAX_PATH, test_fileA);
307 lstrcatA(test_fileA, testW_txt);
308 MultiByteToWideChar(CP_ACP, 0, test_fileA, -1, test_file, MAX_PATH);
310 /* invalid arguments */
312 if (0)
314 /* Crashes on NT4 */
315 stream = NULL;
316 ret = (*pSHCreateStreamOnFileW)(NULL, mode | stgm, &stream);
317 ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) || /* XP */
318 ret == E_INVALIDARG /* Vista */,
319 "SHCreateStreamOnFileW: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) or E_INVALIDARG, got 0x%08x\n", ret);
320 ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
323 if (0)
325 /* This test crashes on WinXP SP2 */
326 ret = (*pSHCreateStreamOnFileW)(test_file, mode | stgm, NULL);
327 ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
330 stream = NULL;
331 ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_CONVERT | stgm, &stream);
332 ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
333 ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
335 stream = NULL;
336 ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_DELETEONRELEASE | stgm, &stream);
337 ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
338 ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
340 stream = NULL;
341 ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_TRANSACTED | stgm, &stream);
342 ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
343 ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
345 /* file does not exist */
347 stream = NULL;
348 ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
349 if (ret == E_INVALIDARG) /* Win98 SE */ {
350 win_skip("Not supported\n");
351 return;
354 ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileW: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
355 ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
357 stream = NULL;
358 ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_CREATE | stgm, &stream);
359 ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
360 ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
362 if (stream) {
363 test_IStream_invalid_operations(stream, mode);
365 refcount = IStream_Release(stream);
366 ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
369 /* NOTE: don't delete the file, as it will be used for the file exists tests. */
371 /* file exists */
373 stream = NULL;
374 ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
375 ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
376 ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
378 if (stream) {
379 test_IStream_invalid_operations(stream, mode);
381 refcount = IStream_Release(stream);
382 ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
385 stream = NULL;
386 ret = (*pSHCreateStreamOnFileW)(test_file, mode | STGM_CREATE | stgm, &stream);
387 ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
388 ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
390 if (stream) {
391 test_IStream_invalid_operations(stream, mode);
393 refcount = IStream_Release(stream);
394 ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
396 ok(DeleteFileA(test_fileA),
397 "SHCreateStreamOnFileW: could not delete the test file, got error %d\n",
398 GetLastError());
403 static void test_SHCreateStreamOnFileEx(DWORD mode, DWORD stgm)
405 IStream * stream;
406 IStream * template = NULL;
407 HRESULT ret;
408 ULONG refcount;
409 WCHAR test_file[MAX_PATH];
410 CHAR test_fileA[MAX_PATH];
411 static const CHAR testEx_txt[] = "\\testEx.txt";
413 trace("SHCreateStreamOnFileEx: testing mode %d, STGM flags %08x\n", mode, stgm);
415 /* Don't used a fixed path for the testEx.txt file */
416 GetTempPathA(MAX_PATH, test_fileA);
417 lstrcatA(test_fileA, testEx_txt);
418 MultiByteToWideChar(CP_ACP, 0, test_fileA, -1, test_file, MAX_PATH);
420 /* invalid arguments */
422 if (0)
424 /* Crashes on NT4 */
425 stream = NULL;
426 ret = (*pSHCreateStreamOnFileEx)(NULL, mode, 0, FALSE, NULL, &stream);
427 ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) || /* XP */
428 ret == E_INVALIDARG /* Vista */,
429 "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) or E_INVALIDARG, got 0x%08x\n", ret);
430 ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
433 stream = NULL;
434 ret = (*pSHCreateStreamOnFileEx)(test_file, mode, 0, FALSE, template, &stream);
435 if (ret == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED)) {
436 win_skip("File probably locked by Anti-Virus/Spam software, trying again\n");
437 Sleep(1000);
438 ret = (*pSHCreateStreamOnFileEx)(test_file, mode, 0, FALSE, template, &stream);
440 ok( ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
441 ret == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER),
442 "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or "
443 "HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got 0x%08x\n", ret);
445 ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
447 if (0)
449 /* This test crashes on WinXP SP2 */
450 ret = (*pSHCreateStreamOnFileEx)(test_file, mode, 0, FALSE, NULL, NULL);
451 ok(ret == E_INVALIDARG, "SHCreateStreamOnFileEx: expected E_INVALIDARG, got 0x%08x\n", ret);
454 /* file does not exist */
456 stream = NULL;
457 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, FALSE, NULL, &stream);
458 if ((stgm & STGM_TRANSACTED) == STGM_TRANSACTED && mode == STGM_READ) {
459 ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) /* XP */ || ret == E_INVALIDARG /* Vista */,
460 "SHCreateStreamOnFileEx: expected E_INVALIDARG or HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
462 if (ret == E_INVALIDARG) {
463 skip("SHCreateStreamOnFileEx: STGM_TRANSACTED not supported in this configuration.\n");
464 return;
466 } else {
467 ok( ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
468 ret == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER),
469 "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or "
470 "HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got 0x%08x\n", ret);
472 ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
474 stream = NULL;
475 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, TRUE, NULL, &stream);
476 /* not supported on win9x */
477 if (broken(ret == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER) && stream == NULL)) {
478 skip("Not supported\n");
479 DeleteFileA(test_fileA);
480 return;
483 ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
484 ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
486 if (stream) {
487 test_IStream_invalid_operations(stream, mode);
489 refcount = IStream_Release(stream);
490 ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
492 ok(DeleteFileA(test_fileA),
493 "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n",
494 GetLastError());
497 stream = NULL;
498 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
499 if (ret == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED)) {
500 win_skip("File probably locked by Anti-Virus/Spam software, trying again\n");
501 Sleep(1000);
502 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
504 ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
505 ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
507 if (stream) {
508 test_IStream_invalid_operations(stream, mode);
510 refcount = IStream_Release(stream);
511 ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
513 ok(DeleteFileA(test_fileA),
514 "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n",
515 GetLastError());
518 stream = NULL;
519 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
520 if (ret == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED)) {
521 win_skip("File probably locked by Anti-Virus/Spam software, trying again\n");
522 Sleep(1000);
523 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
525 ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
526 ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
528 if (stream) {
529 test_IStream_invalid_operations(stream, mode);
531 refcount = IStream_Release(stream);
532 ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
535 /* NOTE: don't delete the file, as it will be used for the file exists tests. */
537 /* file exists */
539 stream = NULL;
540 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, FALSE, NULL, &stream);
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");
544 if (stream) {
545 test_IStream_invalid_operations(stream, mode);
547 refcount = IStream_Release(stream);
548 ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
551 stream = NULL;
552 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_FAILIFTHERE | stgm, 0, TRUE, NULL, &stream);
553 ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_EXISTS), "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_EXISTS), got 0x%08x\n", ret);
554 ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
556 stream = NULL;
557 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
558 ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
559 ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
561 if (stream) {
562 test_IStream_invalid_operations(stream, mode);
564 refcount = IStream_Release(stream);
565 ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
568 stream = NULL;
569 ret = (*pSHCreateStreamOnFileEx)(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
570 ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
571 ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
573 if (stream) {
574 test_IStream_invalid_operations(stream, mode);
576 refcount = IStream_Release(stream);
577 ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
580 ok(DeleteFileA(test_fileA),
581 "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n",
582 GetLastError());
586 START_TEST(istream)
588 static const DWORD stgm_access[] = {
589 STGM_READ,
590 STGM_WRITE,
591 STGM_READWRITE
594 static const DWORD stgm_sharing[] = {
596 STGM_SHARE_DENY_NONE,
597 STGM_SHARE_DENY_READ,
598 STGM_SHARE_DENY_WRITE,
599 STGM_SHARE_EXCLUSIVE
602 static const DWORD stgm_flags[] = {
604 STGM_CONVERT,
605 STGM_DELETEONRELEASE,
606 STGM_CONVERT | STGM_DELETEONRELEASE,
607 STGM_TRANSACTED | STGM_CONVERT,
608 STGM_TRANSACTED | STGM_DELETEONRELEASE,
609 STGM_TRANSACTED | STGM_CONVERT | STGM_DELETEONRELEASE
612 int i, j, k;
614 hShlwapi = GetModuleHandleA("shlwapi.dll");
616 pSHCreateStreamOnFileA = (void*)GetProcAddress(hShlwapi, "SHCreateStreamOnFileA");
617 pSHCreateStreamOnFileW = (void*)GetProcAddress(hShlwapi, "SHCreateStreamOnFileW");
618 pSHCreateStreamOnFileEx = (void*)GetProcAddress(hShlwapi, "SHCreateStreamOnFileEx");
620 if (!pSHCreateStreamOnFileA)
621 skip("SHCreateStreamOnFileA not found.\n");
623 if (!pSHCreateStreamOnFileW)
624 skip("SHCreateStreamOnFileW not found.\n");
626 if (!pSHCreateStreamOnFileEx)
627 skip("SHCreateStreamOnFileEx not found.\n");
629 for (i = 0; i != sizeof(stgm_access)/sizeof(stgm_access[0]); i++) {
630 for (j = 0; j != sizeof(stgm_sharing)/sizeof(stgm_sharing[0]); j ++) {
631 if (pSHCreateStreamOnFileA)
632 test_SHCreateStreamOnFileA(stgm_access[i], stgm_sharing[j]);
634 if (pSHCreateStreamOnFileW)
635 test_SHCreateStreamOnFileW(stgm_access[i], stgm_sharing[j]);
637 if (pSHCreateStreamOnFileEx) {
638 for (k = 0; k != sizeof(stgm_flags)/sizeof(stgm_flags[0]); k++)
639 test_SHCreateStreamOnFileEx(stgm_access[i], stgm_sharing[j] | stgm_flags[k]);