kernel32/tests: Add a test to check some fields in fake dlls.
[wine.git] / dlls / windowscodecs / stream.c
blob366d11b1116ab5b3c274b8c81e2e6fc9b914f0ea
1 /*
2 * Copyright 2009 Tony Wasserka
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 #include "wine/debug.h"
21 #define COBJMACROS
22 #include "windef.h"
23 #include "winbase.h"
24 #include "winreg.h"
25 #include "objbase.h"
26 #include "shlwapi.h"
27 #include "wincodecs_private.h"
29 WINE_DEFAULT_DEBUG_CHANNEL(wincodecs);
31 /******************************************
32 * StreamOnMemory implementation
34 * Used by IWICStream_InitializeFromMemory
37 typedef struct StreamOnMemory {
38 IStream IStream_iface;
39 LONG ref;
41 BYTE *pbMemory;
42 DWORD dwMemsize;
43 DWORD dwCurPos;
45 CRITICAL_SECTION lock; /* must be held when pbMemory or dwCurPos is accessed */
46 } StreamOnMemory;
48 static inline StreamOnMemory *StreamOnMemory_from_IStream(IStream *iface)
50 return CONTAINING_RECORD(iface, StreamOnMemory, IStream_iface);
53 static HRESULT WINAPI StreamOnMemory_QueryInterface(IStream *iface,
54 REFIID iid, void **ppv)
56 TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
58 if (!ppv) return E_INVALIDARG;
60 if (IsEqualIID(&IID_IUnknown, iid) || IsEqualIID(&IID_IStream, iid) ||
61 IsEqualIID(&IID_ISequentialStream, iid))
63 *ppv = iface;
64 IUnknown_AddRef((IUnknown*)*ppv);
65 return S_OK;
67 else
69 *ppv = NULL;
70 return E_NOINTERFACE;
74 static ULONG WINAPI StreamOnMemory_AddRef(IStream *iface)
76 StreamOnMemory *This = StreamOnMemory_from_IStream(iface);
77 ULONG ref = InterlockedIncrement(&This->ref);
79 TRACE("(%p) refcount=%u\n", iface, ref);
81 return ref;
84 static ULONG WINAPI StreamOnMemory_Release(IStream *iface)
86 StreamOnMemory *This = StreamOnMemory_from_IStream(iface);
87 ULONG ref = InterlockedDecrement(&This->ref);
89 TRACE("(%p) refcount=%u\n", iface, ref);
91 if (ref == 0) {
92 This->lock.DebugInfo->Spare[0] = 0;
93 DeleteCriticalSection(&This->lock);
94 HeapFree(GetProcessHeap(), 0, This);
96 return ref;
99 static HRESULT WINAPI StreamOnMemory_Read(IStream *iface,
100 void *pv, ULONG cb, ULONG *pcbRead)
102 StreamOnMemory *This = StreamOnMemory_from_IStream(iface);
103 ULONG uBytesRead;
105 TRACE("(%p, %p, %u, %p)\n", This, pv, cb, pcbRead);
107 if (!pv) return E_INVALIDARG;
109 EnterCriticalSection(&This->lock);
110 uBytesRead = min(cb, This->dwMemsize - This->dwCurPos);
111 memmove(pv, This->pbMemory + This->dwCurPos, uBytesRead);
112 This->dwCurPos += uBytesRead;
113 LeaveCriticalSection(&This->lock);
115 if (pcbRead) *pcbRead = uBytesRead;
117 return S_OK;
120 static HRESULT WINAPI StreamOnMemory_Write(IStream *iface,
121 void const *pv, ULONG cb, ULONG *pcbWritten)
123 StreamOnMemory *This = StreamOnMemory_from_IStream(iface);
124 HRESULT hr;
126 TRACE("(%p, %p, %u, %p)\n", This, pv, cb, pcbWritten);
128 if (!pv) return E_INVALIDARG;
130 EnterCriticalSection(&This->lock);
131 if (cb > This->dwMemsize - This->dwCurPos) {
132 hr = STG_E_MEDIUMFULL;
134 else {
135 memmove(This->pbMemory + This->dwCurPos, pv, cb);
136 This->dwCurPos += cb;
137 hr = S_OK;
138 if (pcbWritten) *pcbWritten = cb;
140 LeaveCriticalSection(&This->lock);
142 return hr;
145 static HRESULT WINAPI StreamOnMemory_Seek(IStream *iface,
146 LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
148 StreamOnMemory *This = StreamOnMemory_from_IStream(iface);
149 LARGE_INTEGER NewPosition;
150 HRESULT hr=S_OK;
152 TRACE("(%p, %s, %d, %p)\n", This, wine_dbgstr_longlong(dlibMove.QuadPart), dwOrigin, plibNewPosition);
154 EnterCriticalSection(&This->lock);
155 if (dwOrigin == STREAM_SEEK_SET) NewPosition.QuadPart = dlibMove.QuadPart;
156 else if (dwOrigin == STREAM_SEEK_CUR) NewPosition.QuadPart = This->dwCurPos + dlibMove.QuadPart;
157 else if (dwOrigin == STREAM_SEEK_END) NewPosition.QuadPart = This->dwMemsize + dlibMove.QuadPart;
158 else hr = E_INVALIDARG;
160 if (SUCCEEDED(hr)) {
161 if (NewPosition.u.HighPart) hr = HRESULT_FROM_WIN32(ERROR_ARITHMETIC_OVERFLOW);
162 else if (NewPosition.QuadPart > This->dwMemsize) hr = E_INVALIDARG;
163 else if (NewPosition.QuadPart < 0) hr = E_INVALIDARG;
166 if (SUCCEEDED(hr)) {
167 This->dwCurPos = NewPosition.u.LowPart;
169 if(plibNewPosition) plibNewPosition->QuadPart = This->dwCurPos;
171 LeaveCriticalSection(&This->lock);
173 return hr;
176 /* SetSize isn't implemented in the native windowscodecs DLL either */
177 static HRESULT WINAPI StreamOnMemory_SetSize(IStream *iface,
178 ULARGE_INTEGER libNewSize)
180 TRACE("(%p, %s)\n", iface, wine_dbgstr_longlong(libNewSize.QuadPart));
181 return E_NOTIMPL;
184 /* CopyTo isn't implemented in the native windowscodecs DLL either */
185 static HRESULT WINAPI StreamOnMemory_CopyTo(IStream *iface,
186 IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
188 TRACE("(%p, %p, %s, %p, %p)\n", iface, pstm, wine_dbgstr_longlong(cb.QuadPart), pcbRead, pcbWritten);
189 return E_NOTIMPL;
192 /* Commit isn't implemented in the native windowscodecs DLL either */
193 static HRESULT WINAPI StreamOnMemory_Commit(IStream *iface,
194 DWORD grfCommitFlags)
196 TRACE("(%p, %#x)\n", iface, grfCommitFlags);
197 return E_NOTIMPL;
200 /* Revert isn't implemented in the native windowscodecs DLL either */
201 static HRESULT WINAPI StreamOnMemory_Revert(IStream *iface)
203 TRACE("(%p)\n", iface);
204 return E_NOTIMPL;
207 /* LockRegion isn't implemented in the native windowscodecs DLL either */
208 static HRESULT WINAPI StreamOnMemory_LockRegion(IStream *iface,
209 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
211 TRACE("(%p, %s, %s, %d)\n", iface, wine_dbgstr_longlong(libOffset.QuadPart),
212 wine_dbgstr_longlong(cb.QuadPart), dwLockType);
213 return E_NOTIMPL;
216 /* UnlockRegion isn't implemented in the native windowscodecs DLL either */
217 static HRESULT WINAPI StreamOnMemory_UnlockRegion(IStream *iface,
218 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
220 TRACE("(%p, %s, %s, %d)\n", iface, wine_dbgstr_longlong(libOffset.QuadPart),
221 wine_dbgstr_longlong(cb.QuadPart), dwLockType);
222 return E_NOTIMPL;
225 static HRESULT WINAPI StreamOnMemory_Stat(IStream *iface,
226 STATSTG *pstatstg, DWORD grfStatFlag)
228 StreamOnMemory *This = StreamOnMemory_from_IStream(iface);
229 TRACE("(%p, %p, %#x)\n", This, pstatstg, grfStatFlag);
231 if (!pstatstg) return E_INVALIDARG;
233 ZeroMemory(pstatstg, sizeof(STATSTG));
234 pstatstg->type = STGTY_STREAM;
235 pstatstg->cbSize.QuadPart = This->dwMemsize;
237 return S_OK;
240 /* Clone isn't implemented in the native windowscodecs DLL either */
241 static HRESULT WINAPI StreamOnMemory_Clone(IStream *iface,
242 IStream **ppstm)
244 TRACE("(%p, %p)\n", iface, ppstm);
245 return E_NOTIMPL;
249 static const IStreamVtbl StreamOnMemory_Vtbl =
251 /*** IUnknown methods ***/
252 StreamOnMemory_QueryInterface,
253 StreamOnMemory_AddRef,
254 StreamOnMemory_Release,
255 /*** ISequentialStream methods ***/
256 StreamOnMemory_Read,
257 StreamOnMemory_Write,
258 /*** IStream methods ***/
259 StreamOnMemory_Seek,
260 StreamOnMemory_SetSize,
261 StreamOnMemory_CopyTo,
262 StreamOnMemory_Commit,
263 StreamOnMemory_Revert,
264 StreamOnMemory_LockRegion,
265 StreamOnMemory_UnlockRegion,
266 StreamOnMemory_Stat,
267 StreamOnMemory_Clone,
270 /******************************************
271 * StreamOnFileHandle implementation (internal)
274 typedef struct StreamOnFileHandle {
275 IStream IStream_iface;
276 LONG ref;
278 HANDLE map;
279 void *mem;
280 IWICStream *stream;
281 } StreamOnFileHandle;
283 static inline StreamOnFileHandle *StreamOnFileHandle_from_IStream(IStream *iface)
285 return CONTAINING_RECORD(iface, StreamOnFileHandle, IStream_iface);
288 static HRESULT WINAPI StreamOnFileHandle_QueryInterface(IStream *iface,
289 REFIID iid, void **ppv)
291 TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
293 if (!ppv) return E_INVALIDARG;
295 if (IsEqualIID(&IID_IUnknown, iid) || IsEqualIID(&IID_IStream, iid) ||
296 IsEqualIID(&IID_ISequentialStream, iid))
298 *ppv = iface;
299 IUnknown_AddRef((IUnknown*)*ppv);
300 return S_OK;
302 else
304 *ppv = NULL;
305 return E_NOINTERFACE;
309 static ULONG WINAPI StreamOnFileHandle_AddRef(IStream *iface)
311 StreamOnFileHandle *This = StreamOnFileHandle_from_IStream(iface);
312 ULONG ref = InterlockedIncrement(&This->ref);
314 TRACE("(%p) refcount=%u\n", iface, ref);
316 return ref;
319 static ULONG WINAPI StreamOnFileHandle_Release(IStream *iface)
321 StreamOnFileHandle *This = StreamOnFileHandle_from_IStream(iface);
322 ULONG ref = InterlockedDecrement(&This->ref);
324 TRACE("(%p) refcount=%u\n", iface, ref);
326 if (ref == 0) {
327 IWICStream_Release(This->stream);
328 UnmapViewOfFile(This->mem);
329 CloseHandle(This->map);
330 HeapFree(GetProcessHeap(), 0, This);
332 return ref;
335 static HRESULT WINAPI StreamOnFileHandle_Read(IStream *iface,
336 void *pv, ULONG cb, ULONG *pcbRead)
338 StreamOnFileHandle *This = StreamOnFileHandle_from_IStream(iface);
339 TRACE("(%p, %p, %u, %p)\n", This, pv, cb, pcbRead);
341 return IWICStream_Read(This->stream, pv, cb, pcbRead);
344 static HRESULT WINAPI StreamOnFileHandle_Write(IStream *iface,
345 void const *pv, ULONG cb, ULONG *pcbWritten)
347 ERR("(%p, %p, %u, %p)\n", iface, pv, cb, pcbWritten);
348 return HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED);
351 static HRESULT WINAPI StreamOnFileHandle_Seek(IStream *iface,
352 LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
354 StreamOnFileHandle *This = StreamOnFileHandle_from_IStream(iface);
355 TRACE("(%p, %s, %d, %p)\n", This, wine_dbgstr_longlong(dlibMove.QuadPart), dwOrigin, plibNewPosition);
357 return IWICStream_Seek(This->stream, dlibMove, dwOrigin, plibNewPosition);
360 static HRESULT WINAPI StreamOnFileHandle_SetSize(IStream *iface,
361 ULARGE_INTEGER libNewSize)
363 TRACE("(%p, %s)\n", iface, wine_dbgstr_longlong(libNewSize.QuadPart));
364 return E_NOTIMPL;
367 static HRESULT WINAPI StreamOnFileHandle_CopyTo(IStream *iface,
368 IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
370 TRACE("(%p, %p, %s, %p, %p)\n", iface, pstm, wine_dbgstr_longlong(cb.QuadPart), pcbRead, pcbWritten);
371 return E_NOTIMPL;
374 static HRESULT WINAPI StreamOnFileHandle_Commit(IStream *iface,
375 DWORD grfCommitFlags)
377 TRACE("(%p, %#x)\n", iface, grfCommitFlags);
378 return E_NOTIMPL;
381 static HRESULT WINAPI StreamOnFileHandle_Revert(IStream *iface)
383 TRACE("(%p)\n", iface);
384 return E_NOTIMPL;
387 static HRESULT WINAPI StreamOnFileHandle_LockRegion(IStream *iface,
388 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
390 TRACE("(%p, %s, %s, %d)\n", iface, wine_dbgstr_longlong(libOffset.QuadPart),
391 wine_dbgstr_longlong(cb.QuadPart), dwLockType);
392 return E_NOTIMPL;
395 static HRESULT WINAPI StreamOnFileHandle_UnlockRegion(IStream *iface,
396 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
398 TRACE("(%p, %s, %s, %d)\n", iface, wine_dbgstr_longlong(libOffset.QuadPart),
399 wine_dbgstr_longlong(cb.QuadPart), dwLockType);
400 return E_NOTIMPL;
403 static HRESULT WINAPI StreamOnFileHandle_Stat(IStream *iface,
404 STATSTG *pstatstg, DWORD grfStatFlag)
406 StreamOnFileHandle *This = StreamOnFileHandle_from_IStream(iface);
407 TRACE("(%p, %p, %#x)\n", This, pstatstg, grfStatFlag);
409 return IWICStream_Stat(This->stream, pstatstg, grfStatFlag);
412 static HRESULT WINAPI StreamOnFileHandle_Clone(IStream *iface,
413 IStream **ppstm)
415 TRACE("(%p, %p)\n", iface, ppstm);
416 return E_NOTIMPL;
419 static const IStreamVtbl StreamOnFileHandle_Vtbl =
421 /*** IUnknown methods ***/
422 StreamOnFileHandle_QueryInterface,
423 StreamOnFileHandle_AddRef,
424 StreamOnFileHandle_Release,
425 /*** ISequentialStream methods ***/
426 StreamOnFileHandle_Read,
427 StreamOnFileHandle_Write,
428 /*** IStream methods ***/
429 StreamOnFileHandle_Seek,
430 StreamOnFileHandle_SetSize,
431 StreamOnFileHandle_CopyTo,
432 StreamOnFileHandle_Commit,
433 StreamOnFileHandle_Revert,
434 StreamOnFileHandle_LockRegion,
435 StreamOnFileHandle_UnlockRegion,
436 StreamOnFileHandle_Stat,
437 StreamOnFileHandle_Clone,
440 /******************************************
441 * StreamOnStreamRange implementation
443 * Used by IWICStream_InitializeFromIStreamRegion
446 typedef struct StreamOnStreamRange {
447 IStream IStream_iface;
448 LONG ref;
450 IStream *stream;
451 ULARGE_INTEGER pos;
452 ULARGE_INTEGER offset;
453 ULARGE_INTEGER max_size;
455 CRITICAL_SECTION lock;
456 } StreamOnStreamRange;
458 static inline StreamOnStreamRange *StreamOnStreamRange_from_IStream(IStream *iface)
460 return CONTAINING_RECORD(iface, StreamOnStreamRange, IStream_iface);
463 static HRESULT WINAPI StreamOnStreamRange_QueryInterface(IStream *iface,
464 REFIID iid, void **ppv)
466 TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
468 if (!ppv) return E_INVALIDARG;
470 if (IsEqualIID(&IID_IUnknown, iid) || IsEqualIID(&IID_IStream, iid) ||
471 IsEqualIID(&IID_ISequentialStream, iid))
473 *ppv = iface;
474 IUnknown_AddRef((IUnknown*)*ppv);
475 return S_OK;
477 else
479 *ppv = NULL;
480 return E_NOINTERFACE;
484 static ULONG WINAPI StreamOnStreamRange_AddRef(IStream *iface)
486 StreamOnStreamRange *This = StreamOnStreamRange_from_IStream(iface);
487 ULONG ref = InterlockedIncrement(&This->ref);
489 TRACE("(%p) refcount=%u\n", iface, ref);
491 return ref;
494 static ULONG WINAPI StreamOnStreamRange_Release(IStream *iface)
496 StreamOnStreamRange *This = StreamOnStreamRange_from_IStream(iface);
497 ULONG ref = InterlockedDecrement(&This->ref);
499 TRACE("(%p) refcount=%u\n", iface, ref);
501 if (ref == 0) {
502 This->lock.DebugInfo->Spare[0] = 0;
503 DeleteCriticalSection(&This->lock);
504 IStream_Release(This->stream);
505 HeapFree(GetProcessHeap(), 0, This);
507 return ref;
510 static HRESULT WINAPI StreamOnStreamRange_Read(IStream *iface,
511 void *pv, ULONG cb, ULONG *pcbRead)
513 StreamOnStreamRange *This = StreamOnStreamRange_from_IStream(iface);
514 ULONG uBytesRead=0;
515 HRESULT hr;
516 ULARGE_INTEGER OldPosition;
517 LARGE_INTEGER SetPosition;
519 TRACE("(%p, %p, %u, %p)\n", This, pv, cb, pcbRead);
521 if (!pv) return E_INVALIDARG;
523 EnterCriticalSection(&This->lock);
524 SetPosition.QuadPart = 0;
525 hr = IStream_Seek(This->stream, SetPosition, STREAM_SEEK_CUR, &OldPosition);
526 if (SUCCEEDED(hr))
528 SetPosition.QuadPart = This->pos.QuadPart + This->offset.QuadPart;
529 hr = IStream_Seek(This->stream, SetPosition, STREAM_SEEK_SET, NULL);
531 if (SUCCEEDED(hr))
533 if (This->pos.QuadPart + cb > This->max_size.QuadPart)
535 /* This would read past the end of the stream. */
536 if (This->pos.QuadPart > This->max_size.QuadPart)
537 cb = 0;
538 else
539 cb = This->max_size.QuadPart - This->pos.QuadPart;
541 hr = IStream_Read(This->stream, pv, cb, &uBytesRead);
542 SetPosition.QuadPart = OldPosition.QuadPart;
543 IStream_Seek(This->stream, SetPosition, STREAM_SEEK_SET, NULL);
545 if (SUCCEEDED(hr))
546 This->pos.QuadPart += uBytesRead;
547 LeaveCriticalSection(&This->lock);
549 if (SUCCEEDED(hr) && pcbRead) *pcbRead = uBytesRead;
551 return hr;
554 static HRESULT WINAPI StreamOnStreamRange_Write(IStream *iface,
555 void const *pv, ULONG cb, ULONG *pcbWritten)
557 StreamOnStreamRange *This = StreamOnStreamRange_from_IStream(iface);
558 HRESULT hr;
559 ULARGE_INTEGER OldPosition;
560 LARGE_INTEGER SetPosition;
561 ULONG uBytesWritten=0;
562 TRACE("(%p, %p, %u, %p)\n", This, pv, cb, pcbWritten);
564 if (!pv) return E_INVALIDARG;
566 EnterCriticalSection(&This->lock);
567 SetPosition.QuadPart = 0;
568 hr = IStream_Seek(This->stream, SetPosition, STREAM_SEEK_CUR, &OldPosition);
569 if (SUCCEEDED(hr))
571 SetPosition.QuadPart = This->pos.QuadPart + This->offset.QuadPart;
572 hr = IStream_Seek(This->stream, SetPosition, STREAM_SEEK_SET, NULL);
574 if (SUCCEEDED(hr))
576 if (This->pos.QuadPart + cb > This->max_size.QuadPart)
578 /* This would read past the end of the stream. */
579 if (This->pos.QuadPart > This->max_size.QuadPart)
580 cb = 0;
581 else
582 cb = This->max_size.QuadPart - This->pos.QuadPart;
584 hr = IStream_Write(This->stream, pv, cb, &uBytesWritten);
585 SetPosition.QuadPart = OldPosition.QuadPart;
586 IStream_Seek(This->stream, SetPosition, STREAM_SEEK_SET, NULL);
588 if (SUCCEEDED(hr))
589 This->pos.QuadPart += uBytesWritten;
590 LeaveCriticalSection(&This->lock);
592 if (SUCCEEDED(hr) && pcbWritten) *pcbWritten = uBytesWritten;
594 return hr;
597 static HRESULT WINAPI StreamOnStreamRange_Seek(IStream *iface,
598 LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
600 StreamOnStreamRange *This = StreamOnStreamRange_from_IStream(iface);
601 ULARGE_INTEGER NewPosition, actual_size;
602 HRESULT hr=S_OK;
603 STATSTG statstg;
604 TRACE("(%p, %s, %d, %p)\n", This, wine_dbgstr_longlong(dlibMove.QuadPart), dwOrigin, plibNewPosition);
606 EnterCriticalSection(&This->lock);
607 actual_size = This->max_size;
608 if (dwOrigin == STREAM_SEEK_SET)
609 NewPosition.QuadPart = dlibMove.QuadPart;
610 else if (dwOrigin == STREAM_SEEK_CUR)
611 NewPosition.QuadPart = This->pos.QuadPart + dlibMove.QuadPart;
612 else if (dwOrigin == STREAM_SEEK_END)
614 hr = IStream_Stat(This->stream, &statstg, STATFLAG_NONAME);
615 if (SUCCEEDED(hr))
617 if (This->max_size.QuadPart + This->offset.QuadPart > statstg.cbSize.QuadPart)
618 actual_size.QuadPart = statstg.cbSize.QuadPart - This->offset.QuadPart;
619 NewPosition.QuadPart = dlibMove.QuadPart + actual_size.QuadPart;
622 else hr = E_INVALIDARG;
624 if (SUCCEEDED(hr) && (NewPosition.u.HighPart != 0 || NewPosition.QuadPart > actual_size.QuadPart))
625 hr = WINCODEC_ERR_VALUEOUTOFRANGE;
627 if (SUCCEEDED(hr)) {
628 This->pos.QuadPart = NewPosition.QuadPart;
630 if(plibNewPosition) plibNewPosition->QuadPart = This->pos.QuadPart;
632 LeaveCriticalSection(&This->lock);
634 return hr;
637 /* SetSize isn't implemented in the native windowscodecs DLL either */
638 static HRESULT WINAPI StreamOnStreamRange_SetSize(IStream *iface,
639 ULARGE_INTEGER libNewSize)
641 TRACE("(%p, %s)\n", iface, wine_dbgstr_longlong(libNewSize.QuadPart));
642 return E_NOTIMPL;
645 /* CopyTo isn't implemented in the native windowscodecs DLL either */
646 static HRESULT WINAPI StreamOnStreamRange_CopyTo(IStream *iface,
647 IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
649 TRACE("(%p, %p, %s, %p, %p)\n", iface, pstm, wine_dbgstr_longlong(cb.QuadPart),
650 pcbRead, pcbWritten);
651 return E_NOTIMPL;
654 /* Commit isn't implemented in the native windowscodecs DLL either */
655 static HRESULT WINAPI StreamOnStreamRange_Commit(IStream *iface,
656 DWORD grfCommitFlags)
658 TRACE("(%p, %#x)\n", iface, grfCommitFlags);
659 return E_NOTIMPL;
662 /* Revert isn't implemented in the native windowscodecs DLL either */
663 static HRESULT WINAPI StreamOnStreamRange_Revert(IStream *iface)
665 TRACE("(%p)\n", iface);
666 return E_NOTIMPL;
669 /* LockRegion isn't implemented in the native windowscodecs DLL either */
670 static HRESULT WINAPI StreamOnStreamRange_LockRegion(IStream *iface,
671 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
673 TRACE("(%p, %s, %s, %d)\n", iface, wine_dbgstr_longlong(libOffset.QuadPart),
674 wine_dbgstr_longlong(cb.QuadPart), dwLockType);
675 return E_NOTIMPL;
678 /* UnlockRegion isn't implemented in the native windowscodecs DLL either */
679 static HRESULT WINAPI StreamOnStreamRange_UnlockRegion(IStream *iface,
680 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
682 TRACE("(%p, %s, %s, %d)\n", iface, wine_dbgstr_longlong(libOffset.QuadPart),
683 wine_dbgstr_longlong(cb.QuadPart), dwLockType);
684 return E_NOTIMPL;
687 static HRESULT WINAPI StreamOnStreamRange_Stat(IStream *iface,
688 STATSTG *pstatstg, DWORD grfStatFlag)
690 StreamOnStreamRange *This = StreamOnStreamRange_from_IStream(iface);
691 HRESULT hr;
692 TRACE("(%p, %p, %#x)\n", This, pstatstg, grfStatFlag);
694 if (!pstatstg) return E_INVALIDARG;
696 EnterCriticalSection(&This->lock);
697 hr = IStream_Stat(This->stream, pstatstg, grfStatFlag);
698 if (SUCCEEDED(hr))
700 pstatstg->cbSize.QuadPart -= This->offset.QuadPart;
701 if (This->max_size.QuadPart < pstatstg->cbSize.QuadPart)
702 pstatstg->cbSize.QuadPart = This->max_size.QuadPart;
705 LeaveCriticalSection(&This->lock);
707 return hr;
710 /* Clone isn't implemented in the native windowscodecs DLL either */
711 static HRESULT WINAPI StreamOnStreamRange_Clone(IStream *iface,
712 IStream **ppstm)
714 TRACE("(%p, %p)\n", iface, ppstm);
715 return E_NOTIMPL;
718 static const IStreamVtbl StreamOnStreamRange_Vtbl =
720 /*** IUnknown methods ***/
721 StreamOnStreamRange_QueryInterface,
722 StreamOnStreamRange_AddRef,
723 StreamOnStreamRange_Release,
724 /*** ISequentialStream methods ***/
725 StreamOnStreamRange_Read,
726 StreamOnStreamRange_Write,
727 /*** IStream methods ***/
728 StreamOnStreamRange_Seek,
729 StreamOnStreamRange_SetSize,
730 StreamOnStreamRange_CopyTo,
731 StreamOnStreamRange_Commit,
732 StreamOnStreamRange_Revert,
733 StreamOnStreamRange_LockRegion,
734 StreamOnStreamRange_UnlockRegion,
735 StreamOnStreamRange_Stat,
736 StreamOnStreamRange_Clone,
740 /******************************************
741 * IWICStream implementation
744 typedef struct IWICStreamImpl
746 IWICStream IWICStream_iface;
747 LONG ref;
749 IStream *pStream;
750 } IWICStreamImpl;
752 static inline IWICStreamImpl *impl_from_IWICStream(IWICStream *iface)
754 return CONTAINING_RECORD(iface, IWICStreamImpl, IWICStream_iface);
757 static HRESULT WINAPI IWICStreamImpl_QueryInterface(IWICStream *iface,
758 REFIID iid, void **ppv)
760 IWICStreamImpl *This = impl_from_IWICStream(iface);
761 TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
763 if (!ppv) return E_INVALIDARG;
765 if (IsEqualIID(&IID_IUnknown, iid) || IsEqualIID(&IID_IStream, iid) ||
766 IsEqualIID(&IID_ISequentialStream, iid) || IsEqualIID(&IID_IWICStream, iid))
768 *ppv = &This->IWICStream_iface;
769 IUnknown_AddRef((IUnknown*)*ppv);
770 return S_OK;
772 else
774 *ppv = NULL;
775 return E_NOINTERFACE;
779 static ULONG WINAPI IWICStreamImpl_AddRef(IWICStream *iface)
781 IWICStreamImpl *This = impl_from_IWICStream(iface);
782 ULONG ref = InterlockedIncrement(&This->ref);
784 TRACE("(%p) refcount=%u\n", iface, ref);
786 return ref;
789 static ULONG WINAPI IWICStreamImpl_Release(IWICStream *iface)
791 IWICStreamImpl *This = impl_from_IWICStream(iface);
792 ULONG ref = InterlockedDecrement(&This->ref);
794 TRACE("(%p) refcount=%u\n", iface, ref);
796 if (ref == 0) {
797 if (This->pStream) IStream_Release(This->pStream);
798 HeapFree(GetProcessHeap(), 0, This);
800 return ref;
803 static HRESULT WINAPI IWICStreamImpl_Read(IWICStream *iface,
804 void *pv, ULONG cb, ULONG *pcbRead)
806 IWICStreamImpl *This = impl_from_IWICStream(iface);
807 TRACE("(%p, %p, %u, %p)\n", This, pv, cb, pcbRead);
809 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
810 return IStream_Read(This->pStream, pv, cb, pcbRead);
813 static HRESULT WINAPI IWICStreamImpl_Write(IWICStream *iface,
814 void const *pv, ULONG cb, ULONG *pcbWritten)
816 IWICStreamImpl *This = impl_from_IWICStream(iface);
817 TRACE("(%p, %p, %u, %p)\n", This, pv, cb, pcbWritten);
819 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
820 return IStream_Write(This->pStream, pv, cb, pcbWritten);
823 static HRESULT WINAPI IWICStreamImpl_Seek(IWICStream *iface,
824 LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
826 IWICStreamImpl *This = impl_from_IWICStream(iface);
827 TRACE("(%p, %s, %d, %p)\n", This, wine_dbgstr_longlong(dlibMove.QuadPart),
828 dwOrigin, plibNewPosition);
830 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
831 return IStream_Seek(This->pStream, dlibMove, dwOrigin, plibNewPosition);
834 static HRESULT WINAPI IWICStreamImpl_SetSize(IWICStream *iface,
835 ULARGE_INTEGER libNewSize)
837 IWICStreamImpl *This = impl_from_IWICStream(iface);
838 TRACE("(%p, %s)\n", This, wine_dbgstr_longlong(libNewSize.QuadPart));
840 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
841 return IStream_SetSize(This->pStream, libNewSize);
844 static HRESULT WINAPI IWICStreamImpl_CopyTo(IWICStream *iface,
845 IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
847 IWICStreamImpl *This = impl_from_IWICStream(iface);
848 TRACE("(%p, %p, %s, %p, %p)\n", This, pstm, wine_dbgstr_longlong(cb.QuadPart), pcbRead, pcbWritten);
850 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
851 return IStream_CopyTo(This->pStream, pstm, cb, pcbRead, pcbWritten);
854 static HRESULT WINAPI IWICStreamImpl_Commit(IWICStream *iface,
855 DWORD grfCommitFlags)
857 IWICStreamImpl *This = impl_from_IWICStream(iface);
858 TRACE("(%p, %#x)\n", This, grfCommitFlags);
860 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
861 return IStream_Commit(This->pStream, grfCommitFlags);
864 static HRESULT WINAPI IWICStreamImpl_Revert(IWICStream *iface)
866 IWICStreamImpl *This = impl_from_IWICStream(iface);
867 TRACE("(%p)\n", This);
869 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
870 return IStream_Revert(This->pStream);
873 static HRESULT WINAPI IWICStreamImpl_LockRegion(IWICStream *iface,
874 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
876 IWICStreamImpl *This = impl_from_IWICStream(iface);
877 TRACE("(%p, %s, %s, %d)\n", This, wine_dbgstr_longlong(libOffset.QuadPart),
878 wine_dbgstr_longlong(cb.QuadPart), dwLockType);
880 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
881 return IStream_LockRegion(This->pStream, libOffset, cb, dwLockType);
884 static HRESULT WINAPI IWICStreamImpl_UnlockRegion(IWICStream *iface,
885 ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
887 IWICStreamImpl *This = impl_from_IWICStream(iface);
888 TRACE("(%p, %s, %s, %d)\n", This, wine_dbgstr_longlong(libOffset.QuadPart),
889 wine_dbgstr_longlong(cb.QuadPart), dwLockType);
891 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
892 return IStream_UnlockRegion(This->pStream, libOffset, cb, dwLockType);
895 static HRESULT WINAPI IWICStreamImpl_Stat(IWICStream *iface,
896 STATSTG *pstatstg, DWORD grfStatFlag)
898 IWICStreamImpl *This = impl_from_IWICStream(iface);
899 TRACE("(%p, %p, %#x)\n", This, pstatstg, grfStatFlag);
901 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
902 return IStream_Stat(This->pStream, pstatstg, grfStatFlag);
905 static HRESULT WINAPI IWICStreamImpl_Clone(IWICStream *iface,
906 IStream **ppstm)
908 IWICStreamImpl *This = impl_from_IWICStream(iface);
909 TRACE("(%p, %p)\n", This, ppstm);
911 if (!This->pStream) return WINCODEC_ERR_NOTINITIALIZED;
912 return IStream_Clone(This->pStream, ppstm);
915 static HRESULT WINAPI IWICStreamImpl_InitializeFromIStream(IWICStream *iface, IStream *stream)
917 IWICStreamImpl *This = impl_from_IWICStream(iface);
918 HRESULT hr = S_OK;
920 TRACE("(%p, %p)\n", iface, stream);
922 if (!stream) return E_INVALIDARG;
923 if (This->pStream) return WINCODEC_ERR_WRONGSTATE;
925 IStream_AddRef(stream);
927 if (InterlockedCompareExchangePointer((void **)&This->pStream, stream, NULL))
929 /* Some other thread set the stream first. */
930 IStream_Release(stream);
931 hr = WINCODEC_ERR_WRONGSTATE;
934 return hr;
937 static HRESULT WINAPI IWICStreamImpl_InitializeFromFilename(IWICStream *iface,
938 LPCWSTR wzFileName, DWORD dwDesiredAccess)
940 IWICStreamImpl *This = impl_from_IWICStream(iface);
941 HRESULT hr;
942 DWORD dwMode;
943 IStream *stream;
945 TRACE("(%p, %s, %u)\n", iface, debugstr_w(wzFileName), dwDesiredAccess);
947 if (This->pStream) return WINCODEC_ERR_WRONGSTATE;
949 if(dwDesiredAccess & GENERIC_WRITE)
950 dwMode = STGM_SHARE_DENY_WRITE | STGM_WRITE | STGM_CREATE;
951 else if(dwDesiredAccess & GENERIC_READ)
952 dwMode = STGM_SHARE_DENY_WRITE | STGM_READ | STGM_FAILIFTHERE;
953 else
954 return E_INVALIDARG;
956 hr = SHCreateStreamOnFileW(wzFileName, dwMode, &stream);
958 if (SUCCEEDED(hr))
960 if (InterlockedCompareExchangePointer((void**)&This->pStream, stream, NULL))
962 /* Some other thread set the stream first. */
963 IStream_Release(stream);
964 hr = WINCODEC_ERR_WRONGSTATE;
968 return hr;
971 /******************************************
972 * IWICStream_InitializeFromMemory
974 * Initializes the internal IStream object to retrieve its data from a memory chunk.
976 * PARAMS
977 * pbBuffer [I] pointer to the memory chunk
978 * cbBufferSize [I] number of bytes to use from the memory chunk
980 * RETURNS
981 * SUCCESS: S_OK
982 * FAILURE: E_INVALIDARG, if pbBuffer is NULL
983 * E_OUTOFMEMORY, if we run out of memory
984 * WINCODEC_ERR_WRONGSTATE, if the IStream object has already been initialized before
987 static HRESULT WINAPI IWICStreamImpl_InitializeFromMemory(IWICStream *iface,
988 BYTE *pbBuffer, DWORD cbBufferSize)
990 IWICStreamImpl *This = impl_from_IWICStream(iface);
991 StreamOnMemory *pObject;
992 TRACE("(%p, %p, %u)\n", iface, pbBuffer, cbBufferSize);
994 if (!pbBuffer) return E_INVALIDARG;
995 if (This->pStream) return WINCODEC_ERR_WRONGSTATE;
997 pObject = HeapAlloc(GetProcessHeap(), 0, sizeof(StreamOnMemory));
998 if (!pObject) return E_OUTOFMEMORY;
1000 pObject->IStream_iface.lpVtbl = &StreamOnMemory_Vtbl;
1001 pObject->ref = 1;
1002 pObject->pbMemory = pbBuffer;
1003 pObject->dwMemsize = cbBufferSize;
1004 pObject->dwCurPos = 0;
1005 InitializeCriticalSection(&pObject->lock);
1006 pObject->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": StreamOnMemory.lock");
1008 if (InterlockedCompareExchangePointer((void**)&This->pStream, pObject, NULL))
1010 /* Some other thread set the stream first. */
1011 IStream_Release(&pObject->IStream_iface);
1012 return WINCODEC_ERR_WRONGSTATE;
1015 return S_OK;
1018 static HRESULT map_file(HANDLE file, HANDLE *map, void **mem, LARGE_INTEGER *size)
1020 *map = NULL;
1021 if (!GetFileSizeEx(file, size)) return HRESULT_FROM_WIN32(GetLastError());
1022 if (size->u.HighPart)
1024 WARN("file too large\n");
1025 return E_FAIL;
1027 if (!(*map = CreateFileMappingW(file, NULL, PAGE_READONLY, 0, size->u.LowPart, NULL)))
1029 return HRESULT_FROM_WIN32(GetLastError());
1031 if (!(*mem = MapViewOfFile(*map, FILE_MAP_READ, 0, 0, size->u.LowPart)))
1033 CloseHandle(*map);
1034 return HRESULT_FROM_WIN32(GetLastError());
1036 return S_OK;
1039 HRESULT stream_initialize_from_filehandle(IWICStream *iface, HANDLE file)
1041 IWICStreamImpl *This = impl_from_IWICStream(iface);
1042 StreamOnFileHandle *pObject;
1043 IWICStream *stream = NULL;
1044 HANDLE map;
1045 void *mem;
1046 LARGE_INTEGER size;
1047 HRESULT hr;
1048 TRACE("(%p,%p)\n", iface, file);
1050 if (This->pStream) return WINCODEC_ERR_WRONGSTATE;
1052 hr = map_file(file, &map, &mem, &size);
1053 if (FAILED(hr)) return hr;
1055 hr = StreamImpl_Create(&stream);
1056 if (FAILED(hr)) goto error;
1058 hr = IWICStreamImpl_InitializeFromMemory(stream, mem, size.u.LowPart);
1059 if (FAILED(hr)) goto error;
1061 pObject = HeapAlloc(GetProcessHeap(), 0, sizeof(StreamOnFileHandle));
1062 if (!pObject)
1064 hr = E_OUTOFMEMORY;
1065 goto error;
1067 pObject->IStream_iface.lpVtbl = &StreamOnFileHandle_Vtbl;
1068 pObject->ref = 1;
1069 pObject->map = map;
1070 pObject->mem = mem;
1071 pObject->stream = stream;
1073 if (InterlockedCompareExchangePointer((void**)&This->pStream, pObject, NULL))
1075 /* Some other thread set the stream first. */
1076 IStream_Release(&pObject->IStream_iface);
1077 return WINCODEC_ERR_WRONGSTATE;
1079 return S_OK;
1081 error:
1082 if (stream) IWICStream_Release(stream);
1083 UnmapViewOfFile(mem);
1084 CloseHandle(map);
1085 return hr;
1088 static HRESULT WINAPI IWICStreamImpl_InitializeFromIStreamRegion(IWICStream *iface,
1089 IStream *pIStream, ULARGE_INTEGER ulOffset, ULARGE_INTEGER ulMaxSize)
1091 IWICStreamImpl *This = impl_from_IWICStream(iface);
1092 StreamOnStreamRange *pObject;
1094 TRACE("(%p,%p,%s,%s)\n", iface, pIStream, wine_dbgstr_longlong(ulOffset.QuadPart),
1095 wine_dbgstr_longlong(ulMaxSize.QuadPart));
1097 if (!pIStream) return E_INVALIDARG;
1098 if (This->pStream) return WINCODEC_ERR_WRONGSTATE;
1100 pObject = HeapAlloc(GetProcessHeap(), 0, sizeof(StreamOnStreamRange));
1101 if (!pObject) return E_OUTOFMEMORY;
1103 pObject->IStream_iface.lpVtbl = &StreamOnStreamRange_Vtbl;
1104 pObject->ref = 1;
1105 IStream_AddRef(pIStream);
1106 pObject->stream = pIStream;
1107 pObject->pos.QuadPart = 0;
1108 pObject->offset = ulOffset;
1109 pObject->max_size = ulMaxSize;
1110 InitializeCriticalSection(&pObject->lock);
1111 pObject->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": StreamOnStreamRange.lock");
1113 if (InterlockedCompareExchangePointer((void**)&This->pStream, pObject, NULL))
1115 /* Some other thread set the stream first. */
1116 IStream_Release(&pObject->IStream_iface);
1117 return WINCODEC_ERR_WRONGSTATE;
1120 return S_OK;
1124 static const IWICStreamVtbl WICStream_Vtbl =
1126 /*** IUnknown methods ***/
1127 IWICStreamImpl_QueryInterface,
1128 IWICStreamImpl_AddRef,
1129 IWICStreamImpl_Release,
1130 /*** ISequentialStream methods ***/
1131 IWICStreamImpl_Read,
1132 IWICStreamImpl_Write,
1133 /*** IStream methods ***/
1134 IWICStreamImpl_Seek,
1135 IWICStreamImpl_SetSize,
1136 IWICStreamImpl_CopyTo,
1137 IWICStreamImpl_Commit,
1138 IWICStreamImpl_Revert,
1139 IWICStreamImpl_LockRegion,
1140 IWICStreamImpl_UnlockRegion,
1141 IWICStreamImpl_Stat,
1142 IWICStreamImpl_Clone,
1143 /*** IWICStream methods ***/
1144 IWICStreamImpl_InitializeFromIStream,
1145 IWICStreamImpl_InitializeFromFilename,
1146 IWICStreamImpl_InitializeFromMemory,
1147 IWICStreamImpl_InitializeFromIStreamRegion,
1150 HRESULT StreamImpl_Create(IWICStream **stream)
1152 IWICStreamImpl *pObject;
1154 if( !stream ) return E_INVALIDARG;
1156 pObject = HeapAlloc(GetProcessHeap(), 0, sizeof(IWICStreamImpl));
1157 if( !pObject ) {
1158 *stream = NULL;
1159 return E_OUTOFMEMORY;
1162 pObject->IWICStream_iface.lpVtbl = &WICStream_Vtbl;
1163 pObject->ref = 1;
1164 pObject->pStream = NULL;
1166 *stream = &pObject->IWICStream_iface;
1168 return S_OK;