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 
test_IStream_invalid_operations(IStream * stream,DWORD mode)31 static void test_IStream_invalid_operations(IStream * stream, DWORD mode)
32 {
33     HRESULT ret;
34     IStream * clone;
35     ULONG refcount;
36     ULARGE_INTEGER uzero;
37     ULARGE_INTEGER uret;
38     LARGE_INTEGER zero;
39     ULONG count;
40     char data[256];
41 
42     U(uzero).HighPart = 0;
43     U(uzero).LowPart = 0;
44     U(uret).HighPart = 0;
45     U(uret).LowPart = 0;
46     U(zero).HighPart = 0;
47     U(zero).LowPart = 0;
48 
49     /* IStream::Read */
50 
51     /* IStream_Read from the COBJMACROS is undefined by shlwapi.h, replaced by the IStream_Read helper function. */
52 
53     ret = stream->lpVtbl->Read(stream, NULL, 0, &count);
54     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
55 
56     ret = stream->lpVtbl->Read(stream, data, 5, NULL);
57     ok(ret == S_FALSE || ret == S_OK, "expected S_FALSE or S_OK, got 0x%08x\n", ret);
58 
59     ret = stream->lpVtbl->Read(stream, data, 0, NULL);
60     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
61 
62     ret = stream->lpVtbl->Read(stream, data, 3, &count);
63     ok(ret == S_FALSE || ret == S_OK, "expected S_FALSE or S_OK, got 0x%08x\n", ret);
64 
65     /* IStream::Write */
66 
67     /* IStream_Write from the COBJMACROS is undefined by shlwapi.h, replaced by the IStream_Write helper function. */
68 
69     ret = stream->lpVtbl->Write(stream, NULL, 0, &count);
70     if (mode == STGM_READ)
71     {
72         ok(ret == STG_E_ACCESSDENIED /* XP */ || broken(ret == S_OK) /* Win2000 + IE5 */,
73            "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
74     }
75     else
76         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
77 
78     strcpy(data, "Hello");
79     ret = stream->lpVtbl->Write(stream, data, 5, NULL);
80     if (mode == STGM_READ)
81         ok(ret == STG_E_ACCESSDENIED,
82            "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
83     else
84         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
85 
86     strcpy(data, "Hello");
87     ret = stream->lpVtbl->Write(stream, data, 0, NULL);
88     if (mode == STGM_READ)
89         ok(ret == STG_E_ACCESSDENIED,
90            "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
91     else
92         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
93 
94     strcpy(data, "Hello");
95     ret = stream->lpVtbl->Write(stream, data, 0, &count);
96     if (mode == STGM_READ)
97         ok(ret == STG_E_ACCESSDENIED,
98            "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
99     else
100         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
101 
102     strcpy(data, "Hello");
103     ret = stream->lpVtbl->Write(stream, data, 3, &count);
104     if (mode == STGM_READ)
105         ok(ret == STG_E_ACCESSDENIED,
106            "expected STG_E_ACCESSDENIED, got 0x%08x\n", ret);
107     else
108         ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
109 
110     /* IStream::Seek */
111 
112     ret = IStream_Seek(stream, zero, STREAM_SEEK_SET, NULL);
113     ok(ret == S_OK, "expected S_OK, got 0x%08x\n", ret);
114 
115     ret = IStream_Seek(stream, zero, 20, NULL);
116     ok(ret == E_INVALIDARG,
117        "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,
170        "expected STG_E_INVALIDPOINTER or E_NOTIMPL, got 0x%08x\n", ret);
171 
172     /* IStream::Clone */
173 
174     /* Passing a NULL pointer for the second IStream::Clone param crashes on Win7 */
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 
test_stream_read_write(IStream * stream,DWORD mode)188 static void test_stream_read_write(IStream *stream, DWORD mode)
189 {
190     static const LARGE_INTEGER start;
191     HRESULT ret;
192     unsigned char buf[16];
193     DWORD written, count;
194     STATSTG statstg;
195 
196     /* IStream_Read/Write from the COBJMACROS is undefined by shlwapi.h */
197 
198     written = 0xdeadbeaf;
199     ret = stream->lpVtbl->Write(stream, "\x5e\xa7", 2, &written);
200     if (mode == STGM_WRITE || mode == STGM_READWRITE)
201     {
202         ok(ret == S_OK, "IStream_Write error %#x (access %#x)\n", ret, mode);
203         ok(written == 2, "expected 2, got %u\n", written);
204     }
205     else
206     {
207         ok(ret == STG_E_ACCESSDENIED || broken(ret == S_OK) /* win2000 */, "expected STG_E_ACCESSDENIED, got %#x (access %#x)\n", ret, mode);
208         ok(written == 0xdeadbeaf || broken(written == 2) /* win2000 */, "expected 0xdeadbeaf, got %#x\n", written);
209         written = 0;
210         if (ret == S_OK) return; /* no point in further testing */
211     }
212 
213     ret = stream->lpVtbl->Seek(stream, start, STREAM_SEEK_SET, NULL);
214     ok(ret == S_OK, "Seek error %#x\n", ret);
215 
216     count = 0xdeadbeaf;
217     ret = stream->lpVtbl->Read(stream, buf, 2, &count);
218     if (written != 0)
219     {
220         ok(ret == S_OK || broken(ret == S_FALSE) /* win2000 */, "IStream_Read error %#x (access %#x, written %u)\n", ret, mode, written);
221         if (ret == S_OK && (mode == STGM_WRITE || mode == STGM_READWRITE))
222         {
223             ok(count == 2, "expected 2, got %u\n", count);
224             ok(buf[0] == 0x5e && buf[1] == 0xa7, "expected 5ea7, got %02x%02x\n", buf[0], buf[1]);
225         }
226         else
227             ok(count == 0, "expected 0, got %u\n", count);
228     }
229     else
230     {
231         ok(ret == S_FALSE, "expected S_FALSE, got %#x (access %#x, written %u)\n", ret, mode, written);
232         ok(count == 0, "expected 0, got %u\n", count);
233     }
234 
235     ret = stream->lpVtbl->Seek(stream, start, STREAM_SEEK_SET, NULL);
236     ok(ret == S_OK, "Seek error %#x\n", ret);
237 
238     count = 0xdeadbeaf;
239     ret = stream->lpVtbl->Read(stream, buf, 0, &count);
240     ok(ret == S_OK, "IStream_Read error %#x (access %#x, written %u)\n", ret, mode, written);
241     ok(count == 0, "expected 0, got %u\n", count);
242 
243     count = 0xdeadbeaf;
244     ret = stream->lpVtbl->Read(stream, buf, sizeof(buf), &count);
245     ok(ret == S_FALSE, "expected S_FALSE, got %#x (access %#x, written %u)\n", ret, mode, written);
246     ok(count == written, "expected %u, got %u\n", written, count);
247     if (count)
248         ok(buf[0] == 0x5e && buf[1] == 0xa7, "expected 5ea7, got %02x%02x\n", buf[0], buf[1]);
249 
250     memset(&statstg, 0xff, sizeof(statstg));
251     ret = IStream_Stat(stream, &statstg, 0);
252     ok(ret == S_OK, "Stat failed, hr %#x.\n", ret);
253     ok(statstg.pwcsName != NULL, "Unexpected name %s.\n", wine_dbgstr_w(statstg.pwcsName));
254     CoTaskMemFree(statstg.pwcsName);
255 
256     memset(&statstg, 0xff, sizeof(statstg));
257     ret = IStream_Stat(stream, &statstg, STATFLAG_NONAME);
258     ok(ret == S_OK, "Stat failed, hr %#x.\n", ret);
259     ok(statstg.pwcsName == NULL, "Unexpected name %s.\n", wine_dbgstr_w(statstg.pwcsName));
260 }
261 
test_stream_qi(IStream * stream)262 static void test_stream_qi(IStream *stream)
263 {
264     IUnknown *unk;
265     HRESULT hr;
266 
267     hr = IStream_QueryInterface(stream, &IID_IStream, (void **)&unk);
268     ok(SUCCEEDED(hr), "Failed to get IStream interface, hr %#x.\n", hr);
269     IUnknown_Release(unk);
270 
271     unk = NULL;
272     hr = IStream_QueryInterface(stream, &IID_ISequentialStream, (void **)&unk);
273 todo_wine
274     ok(SUCCEEDED(hr) || broken(hr == E_NOINTERFACE) /* XP */, "Failed to get ISequentialStream interface, hr %#x.\n", hr);
275     if (unk)
276         IUnknown_Release(unk);
277 
278     hr = IStream_QueryInterface(stream, &IID_IUnknown, (void **)&unk);
279     ok(SUCCEEDED(hr), "Failed to get IUnknown interface, hr %#x.\n", hr);
280     IUnknown_Release(unk);
281 }
282 
test_SHCreateStreamOnFileA(DWORD mode,DWORD stgm)283 static void test_SHCreateStreamOnFileA(DWORD mode, DWORD stgm)
284 {
285     IStream * stream;
286     HRESULT ret;
287     ULONG refcount;
288     char test_file[MAX_PATH];
289     static const CHAR testA_txt[] = "\\testA.txt";
290 
291     trace("SHCreateStreamOnFileA: testing mode %d, STGM flags %08x\n", mode, stgm);
292 
293     /* Don't used a fixed path for the testA.txt file */
294     GetTempPathA(MAX_PATH, test_file);
295     lstrcatA(test_file, testA_txt);
296 
297     /* invalid arguments */
298 
299     stream = NULL;
300     ret = SHCreateStreamOnFileA(NULL, mode | stgm, &stream);
301     if (ret == E_INVALIDARG) /* Win98 SE */ {
302         win_skip("Not supported\n");
303         return;
304     }
305 
306     ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) /* NT */ ||
307        ret == HRESULT_FROM_WIN32(ERROR_BAD_PATHNAME) /* 9x */,
308        "SHCreateStreamOnFileA: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) "
309        "or HRESULT_FROM_WIN32(ERROR_BAD_PATHNAME), got 0x%08x\n", ret);
310     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
311 
312 if (0) /* This test crashes on WinXP SP2 */
313 {
314     ret = SHCreateStreamOnFileA(test_file, mode | stgm, NULL);
315     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
316 }
317 
318     stream = NULL;
319     ret = SHCreateStreamOnFileA(test_file, mode | STGM_CONVERT | stgm, &stream);
320     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
321     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
322 
323     stream = NULL;
324     ret = SHCreateStreamOnFileA(test_file, mode | STGM_DELETEONRELEASE | stgm, &stream);
325     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
326     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
327 
328     stream = NULL;
329     ret = SHCreateStreamOnFileA(test_file, mode | STGM_TRANSACTED | stgm, &stream);
330     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileA: expected E_INVALIDARG, got 0x%08x\n", ret);
331     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
332 
333     /* file does not exist */
334 
335     stream = NULL;
336     ret = SHCreateStreamOnFileA(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
337     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileA: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
338     ok(stream == NULL, "SHCreateStreamOnFileA: expected a NULL IStream object, got %p\n", stream);
339 
340     stream = NULL;
341     ret = SHCreateStreamOnFileA(test_file, mode | STGM_CREATE | stgm, &stream);
342     ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
343     ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
344 
345     if (stream) {
346         test_stream_qi(stream);
347         test_IStream_invalid_operations(stream, mode);
348 
349         refcount = IStream_Release(stream);
350         ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
351     }
352 
353     /* NOTE: don't delete the file, as it will be used for the file exists tests. */
354 
355     /* file exists */
356 
357     stream = NULL;
358     ret = SHCreateStreamOnFileA(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
359     ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
360     ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
361 
362     if (stream) {
363         test_IStream_invalid_operations(stream, mode);
364 
365         refcount = IStream_Release(stream);
366         ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
367     }
368 
369     stream = NULL;
370     ret = SHCreateStreamOnFileA(test_file, mode | STGM_CREATE | stgm, &stream);
371     ok(ret == S_OK, "SHCreateStreamOnFileA: expected S_OK, got 0x%08x\n", ret);
372     ok(stream != NULL, "SHCreateStreamOnFileA: expected a valid IStream object, got NULL\n");
373 
374     if (stream) {
375         BOOL delret;
376 
377         test_stream_read_write(stream, mode);
378         test_IStream_invalid_operations(stream, mode);
379 
380         refcount = IStream_Release(stream);
381         ok(refcount == 0, "SHCreateStreamOnFileA: expected 0, got %d\n", refcount);
382 
383         delret = DeleteFileA(test_file);
384         ok(delret, "SHCreateStreamOnFileA: could not delete file '%s', got error %d\n",
385            test_file, GetLastError());
386     }
387 }
388 
389 
test_SHCreateStreamOnFileW(DWORD mode,DWORD stgm)390 static void test_SHCreateStreamOnFileW(DWORD mode, DWORD stgm)
391 {
392     IStream * stream;
393     HRESULT ret;
394     ULONG refcount;
395     WCHAR test_file[MAX_PATH];
396     CHAR  test_fileA[MAX_PATH];
397     static const CHAR testW_txt[] = "\\testW.txt";
398 
399     trace("SHCreateStreamOnFileW: testing mode %d, STGM flags %08x\n", mode, stgm);
400 
401     /* Don't used a fixed path for the testW.txt file */
402     GetTempPathA(MAX_PATH, test_fileA);
403     lstrcatA(test_fileA, testW_txt);
404     MultiByteToWideChar(CP_ACP, 0, test_fileA, -1, test_file, MAX_PATH);
405 
406     /* invalid arguments */
407 
408     if (0)
409     {
410         /* Crashes on NT4 */
411         stream = NULL;
412         ret = SHCreateStreamOnFileW(NULL, mode | stgm, &stream);
413         ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) || /* XP */
414            ret == E_INVALIDARG /* Vista */,
415           "SHCreateStreamOnFileW: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) or E_INVALIDARG, got 0x%08x\n", ret);
416         ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
417     }
418 
419     if (0)
420     {
421         /* This test crashes on WinXP SP2 */
422             ret = SHCreateStreamOnFileW(test_file, mode | stgm, NULL);
423             ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
424     }
425 
426     stream = NULL;
427     ret = SHCreateStreamOnFileW(test_file, mode | STGM_CONVERT | stgm, &stream);
428     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
429     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
430 
431     stream = NULL;
432     ret = SHCreateStreamOnFileW(test_file, mode | STGM_DELETEONRELEASE | stgm, &stream);
433     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
434     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
435 
436     stream = NULL;
437     ret = SHCreateStreamOnFileW(test_file, mode | STGM_TRANSACTED | stgm, &stream);
438     ok(ret == E_INVALIDARG, "SHCreateStreamOnFileW: expected E_INVALIDARG, got 0x%08x\n", ret);
439     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
440 
441     /* file does not exist */
442 
443     stream = NULL;
444     ret = SHCreateStreamOnFileW(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
445     if (ret == E_INVALIDARG) /* Win98 SE */ {
446         win_skip("Not supported\n");
447         return;
448     }
449 
450     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), "SHCreateStreamOnFileW: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
451     ok(stream == NULL, "SHCreateStreamOnFileW: expected a NULL IStream object, got %p\n", stream);
452 
453     stream = NULL;
454     ret = SHCreateStreamOnFileW(test_file, mode | STGM_CREATE | stgm, &stream);
455     ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
456     ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
457 
458     if (stream) {
459         test_stream_qi(stream);
460         test_IStream_invalid_operations(stream, mode);
461 
462         refcount = IStream_Release(stream);
463         ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
464     }
465 
466     /* NOTE: don't delete the file, as it will be used for the file exists tests. */
467 
468     /* file exists */
469 
470     stream = NULL;
471     ret = SHCreateStreamOnFileW(test_file, mode | STGM_FAILIFTHERE | stgm, &stream);
472     ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
473     ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
474 
475     if (stream) {
476         test_IStream_invalid_operations(stream, mode);
477 
478         refcount = IStream_Release(stream);
479         ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
480     }
481 
482     stream = NULL;
483     ret = SHCreateStreamOnFileW(test_file, mode | STGM_CREATE | stgm, &stream);
484     ok(ret == S_OK, "SHCreateStreamOnFileW: expected S_OK, got 0x%08x\n", ret);
485     ok(stream != NULL, "SHCreateStreamOnFileW: expected a valid IStream object, got NULL\n");
486 
487     if (stream) {
488         BOOL delret;
489 
490         test_stream_read_write(stream, mode);
491         test_IStream_invalid_operations(stream, mode);
492 
493         refcount = IStream_Release(stream);
494         ok(refcount == 0, "SHCreateStreamOnFileW: expected 0, got %d\n", refcount);
495 
496         delret = DeleteFileA(test_fileA);
497         ok(delret, "SHCreateStreamOnFileW: could not delete the test file, got error %d\n",
498            GetLastError());
499     }
500 }
501 
502 
test_SHCreateStreamOnFileEx(DWORD mode,DWORD stgm)503 static void test_SHCreateStreamOnFileEx(DWORD mode, DWORD stgm)
504 {
505     IStream * stream;
506     IStream * template = NULL;
507     HRESULT ret;
508     ULONG refcount;
509     WCHAR test_file[MAX_PATH];
510     CHAR  test_fileA[MAX_PATH];
511     static const CHAR testEx_txt[] = "\\testEx.txt";
512     BOOL delret;
513 
514     if (winetest_debug > 1)
515         trace("SHCreateStreamOnFileEx: testing mode %d, STGM flags %08x\n", mode, stgm);
516 
517     /* Don't used a fixed path for the testEx.txt file */
518     GetTempPathA(MAX_PATH, test_fileA);
519     lstrcatA(test_fileA, testEx_txt);
520     MultiByteToWideChar(CP_ACP, 0, test_fileA, -1, test_file, MAX_PATH);
521 
522     /* invalid arguments */
523 
524     if (0)
525     {
526         /* Crashes on NT4 */
527         stream = NULL;
528         ret = SHCreateStreamOnFileEx(NULL, mode, 0, FALSE, NULL, &stream);
529         ok(ret == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) || /* XP */
530            ret == E_INVALIDARG /* Vista */,
531           "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND) or E_INVALIDARG, got 0x%08x\n", ret);
532         ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
533     }
534 
535     stream = NULL;
536     ret = SHCreateStreamOnFileEx(test_file, mode, 0, FALSE, template, &stream);
537     if (ret == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED)) {
538         win_skip("File probably locked by Anti-Virus/Spam software, trying again\n");
539         Sleep(1000);
540         ret = SHCreateStreamOnFileEx(test_file, mode, 0, FALSE, template, &stream);
541     }
542     ok( ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
543         ret == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER),
544         "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or "
545         "HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got 0x%08x\n", ret);
546 
547     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
548 
549     if (0)
550     {
551         /* This test crashes on WinXP SP2 */
552         ret = SHCreateStreamOnFileEx(test_file, mode, 0, FALSE, NULL, NULL);
553         ok(ret == E_INVALIDARG, "SHCreateStreamOnFileEx: expected E_INVALIDARG, got 0x%08x\n", ret);
554     }
555 
556     /* file does not exist */
557 
558     stream = NULL;
559     ret = SHCreateStreamOnFileEx(test_file, mode | STGM_FAILIFTHERE | stgm, 0, FALSE, NULL, &stream);
560     if ((stgm & STGM_TRANSACTED) == STGM_TRANSACTED && mode == STGM_READ) {
561         ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) /* XP */ || ret == E_INVALIDARG /* Vista */,
562           "SHCreateStreamOnFileEx: expected E_INVALIDARG or HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got 0x%08x\n", ret);
563 
564         if (ret == E_INVALIDARG) {
565             skip("SHCreateStreamOnFileEx: STGM_TRANSACTED not supported in this configuration.\n");
566             return;
567         }
568     } else {
569         ok( ret == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) ||
570             ret == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER),
571             "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or "
572             "HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got 0x%08x\n", ret);
573     }
574     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
575 
576     stream = NULL;
577     ret = SHCreateStreamOnFileEx(test_file, mode | STGM_FAILIFTHERE | stgm, 0, TRUE, NULL, &stream);
578     /* not supported on win9x */
579     if (broken(ret == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER) && stream == NULL)) {
580         skip("Not supported\n");
581         DeleteFileA(test_fileA);
582         return;
583     }
584 
585     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
586     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
587 
588     if (stream) {
589         test_stream_qi(stream);
590         test_IStream_invalid_operations(stream, mode);
591 
592         refcount = IStream_Release(stream);
593         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
594 
595         delret = DeleteFileA(test_fileA);
596         ok(delret, "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n",
597            GetLastError());
598     }
599 
600     stream = NULL;
601     ret = SHCreateStreamOnFileEx(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
602     if (ret == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED)) {
603         win_skip("File probably locked by Anti-Virus/Spam software, trying again\n");
604         Sleep(1000);
605         ret = SHCreateStreamOnFileEx(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
606     }
607     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
608     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
609 
610     if (stream) {
611         test_IStream_invalid_operations(stream, mode);
612 
613         refcount = IStream_Release(stream);
614         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
615 
616         delret = DeleteFileA(test_fileA);
617         ok(delret, "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n",
618            GetLastError());
619     }
620 
621     stream = NULL;
622     ret = SHCreateStreamOnFileEx(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
623     if (ret == HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED)) {
624         win_skip("File probably locked by Anti-Virus/Spam software, trying again\n");
625         Sleep(1000);
626         ret = SHCreateStreamOnFileEx(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
627     }
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     /* NOTE: don't delete the file, as it will be used for the file exists tests. */
639 
640     /* file exists */
641 
642     stream = NULL;
643     ret = SHCreateStreamOnFileEx(test_file, mode | STGM_FAILIFTHERE | stgm, 0, FALSE, NULL, &stream);
644     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
645     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
646 
647     if (stream) {
648         test_IStream_invalid_operations(stream, mode);
649 
650         refcount = IStream_Release(stream);
651         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
652     }
653 
654     stream = NULL;
655     ret = SHCreateStreamOnFileEx(test_file, mode | STGM_FAILIFTHERE | stgm, 0, TRUE, NULL, &stream);
656     ok(ret == HRESULT_FROM_WIN32(ERROR_FILE_EXISTS), "SHCreateStreamOnFileEx: expected HRESULT_FROM_WIN32(ERROR_FILE_EXISTS), got 0x%08x\n", ret);
657     ok(stream == NULL, "SHCreateStreamOnFileEx: expected a NULL IStream object, got %p\n", stream);
658 
659     stream = NULL;
660     ret = SHCreateStreamOnFileEx(test_file, mode | STGM_CREATE | stgm, 0, FALSE, NULL, &stream);
661     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
662     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
663 
664     if (stream) {
665         test_IStream_invalid_operations(stream, mode);
666 
667         refcount = IStream_Release(stream);
668         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
669     }
670 
671     stream = NULL;
672     ret = SHCreateStreamOnFileEx(test_file, mode | STGM_CREATE | stgm, 0, TRUE, NULL, &stream);
673     ok(ret == S_OK, "SHCreateStreamOnFileEx: expected S_OK, got 0x%08x\n", ret);
674     ok(stream != NULL, "SHCreateStreamOnFileEx: expected a valid IStream object, got NULL\n");
675 
676     if (stream) {
677         test_IStream_invalid_operations(stream, mode);
678 
679         refcount = IStream_Release(stream);
680         ok(refcount == 0, "SHCreateStreamOnFileEx: expected 0, got %d\n", refcount);
681     }
682 
683     delret = DeleteFileA(test_fileA);
684     ok(delret, "SHCreateStreamOnFileEx: could not delete the test file, got error %d\n",
685        GetLastError());
686 }
687 
688 
test_SHCreateStreamOnFileEx_CopyTo(void)689 static void test_SHCreateStreamOnFileEx_CopyTo(void)
690 {
691     HRESULT ret;
692     IStream *src, *dst;
693     WCHAR tmpPath[MAX_PATH];
694     WCHAR srcFileName[MAX_PATH];
695     WCHAR dstFileName[MAX_PATH];
696     ULARGE_INTEGER count, read, written;
697     LARGE_INTEGER distance;
698     static const char srcContents[1];
699     static const WCHAR prefix[] = { 'T', 'S', 'T', 0 };
700 
701     GetTempPathW(MAX_PATH, tmpPath);
702     ret = GetTempFileNameW(tmpPath, prefix, 0, srcFileName);
703     ok(ret != 0, "GetTempFileName failed, got error %d\n", GetLastError());
704     ret = GetTempFileNameW(tmpPath, prefix, 0, dstFileName);
705     ok(ret != 0, "GetTempFileName failed, got error %d\n", GetLastError());
706 
707     ret = SHCreateStreamOnFileEx(srcFileName, STGM_CREATE | STGM_READWRITE | STGM_DELETEONRELEASE, FILE_ATTRIBUTE_TEMPORARY, FALSE, NULL, &src);
708     ok(SUCCEEDED(ret), "SHCreateStreamOnFileEx failed with ret=0x%08x\n", ret);
709 
710     written.QuadPart = 0;
711     ret = IStream_Write(src, srcContents, sizeof(srcContents), &U(written).LowPart);
712     ok(SUCCEEDED(ret), "ISequentialStream_Write failed with ret=0x%08x\n", ret);
713 
714     distance.QuadPart = 0;
715     ret = IStream_Seek(src, distance, STREAM_SEEK_SET, &written);
716     ok(SUCCEEDED(ret), "ISequentialStream_Seek failed with ret=0x%08x\n", ret);
717 
718     ret = SHCreateStreamOnFileEx(dstFileName, STGM_CREATE | STGM_READWRITE | STGM_DELETEONRELEASE, FILE_ATTRIBUTE_TEMPORARY, FALSE, NULL, &dst);
719     ok(SUCCEEDED(ret), "SHCreateStreamOnFileEx failed with ret=0x%08x\n", ret);
720 
721     /* Test using a count larger than the source file, so that the Read operation will fall short */
722     count.QuadPart = 2;
723 
724     ret = IStream_CopyTo(src, dst, count, &read, &written);
725     ok(SUCCEEDED(ret), "CopyTo failed with ret=0x%08x\n", ret);
726 
727     ok(read.QuadPart == 1, "read does not match size: %d != 1\n", U(read).LowPart);
728     ok(written.QuadPart == 1, "written does not match size: %d != 1\n", U(written).LowPart);
729 
730     IStream_Release(dst);
731     IStream_Release(src);
732     DeleteFileW( srcFileName );
733     DeleteFileW( dstFileName );
734 }
735 
test_SHCreateMemStream(void)736 static void test_SHCreateMemStream(void)
737 {
738     static const BYTE initial[10];
739     IStream *stream, *stream2;
740     IUnknown *unk;
741     char buff[10];
742     HRESULT hr;
743 
744     stream = SHCreateMemStream(initial, 0);
745     ok(stream != NULL, "Failed to create a stream.\n");
746     IStream_Release(stream);
747 
748     stream = SHCreateMemStream(NULL, 10);
749     ok(stream != NULL, "Failed to create a stream.\n");
750     IStream_Release(stream);
751 
752     stream = SHCreateMemStream(NULL, 0);
753     ok(stream != NULL, "Failed to create a stream.\n");
754 
755     hr = IStream_QueryInterface(stream, &IID_ISequentialStream, (void **)&unk);
756 todo_wine
757     ok(hr == S_OK || broken(hr == E_NOINTERFACE) /* WinXP */, "Failed to QI, hr %#x.\n", hr);
758     if (unk)
759         IUnknown_Release(unk);
760 
761     hr = IStream_Read(stream, buff, sizeof(buff), NULL);
762 todo_wine
763     ok(hr == S_FALSE || broken(hr == S_OK) /* WinXP */, "Unexpected hr %#x.\n", hr);
764 
765     hr = IStream_Clone(stream, &stream2);
766 todo_wine
767     ok(hr == S_OK || broken(hr == E_NOTIMPL) /* < Win8 */, "Failed to clone a stream, hr %#x.\n", hr);
768     if (hr == S_OK)
769         IStream_Release(stream2);
770 
771     IStream_Release(stream);
772 }
773 
START_TEST(istream)774 START_TEST(istream)
775 {
776     static const DWORD stgm_access[] = {
777         STGM_READ,
778         STGM_WRITE,
779         STGM_READWRITE
780     };
781 
782     static const DWORD stgm_sharing[] = {
783         0,
784         STGM_SHARE_DENY_NONE,
785         STGM_SHARE_DENY_READ,
786         STGM_SHARE_DENY_WRITE,
787         STGM_SHARE_EXCLUSIVE
788     };
789 
790     static const DWORD stgm_flags[] = {
791         0,
792         STGM_CONVERT,
793         STGM_DELETEONRELEASE,
794         STGM_CONVERT | STGM_DELETEONRELEASE,
795         STGM_TRANSACTED | STGM_CONVERT,
796         STGM_TRANSACTED | STGM_DELETEONRELEASE,
797         STGM_TRANSACTED | STGM_CONVERT | STGM_DELETEONRELEASE
798     };
799 
800     int i, j, k;
801 
802     for (i = 0; i != ARRAY_SIZE(stgm_access); i++) {
803         for (j = 0; j != ARRAY_SIZE(stgm_sharing); j ++) {
804             test_SHCreateStreamOnFileA(stgm_access[i], stgm_sharing[j]);
805             test_SHCreateStreamOnFileW(stgm_access[i], stgm_sharing[j]);
806 
807             for (k = 0; k != ARRAY_SIZE(stgm_flags); k++)
808                 test_SHCreateStreamOnFileEx(stgm_access[i], stgm_sharing[j] | stgm_flags[k]);
809         }
810     }
811 
812     test_SHCreateStreamOnFileEx_CopyTo();
813     test_SHCreateMemStream();
814 }
815