4 * Copyright 2003 Robert Shearman
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #define NONAMELESSUNION
22 #define NONAMELESSSTRUCT
24 #include "quartz_private.h"
26 #include "wine/debug.h"
27 #include "wine/unicode.h"
36 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
38 static const WCHAR wszOutputPinName[] = { 'O','u','t','p','u','t',0 };
40 typedef struct AsyncReader
42 const IBaseFilterVtbl * lpVtbl;
43 const IFileSourceFilterVtbl * lpVtblFSF;
46 FILTER_INFO filterInfo;
48 CRITICAL_SECTION csFilter;
56 static const IBaseFilterVtbl AsyncReader_Vtbl;
57 static const IFileSourceFilterVtbl FileSource_Vtbl;
58 static const IAsyncReaderVtbl FileAsyncReader_Vtbl;
60 static HRESULT FileAsyncReader_Construct(HANDLE hFile, IBaseFilter * pBaseFilter, LPCRITICAL_SECTION pCritSec, IPin ** ppPin);
62 static inline AsyncReader *impl_from_IFileSourceFilter( IFileSourceFilter *iface )
64 return (AsyncReader *)((char*)iface - FIELD_OFFSET(AsyncReader, lpVtblFSF));
67 static WCHAR const mediatype_name[11] = {
68 'M', 'e', 'd', 'i', 'a', ' ', 'T', 'y', 'p', 'e', 0 };
69 static WCHAR const subtype_name[8] = {
70 'S', 'u', 'b', 't', 'y', 'p', 'e', 0 };
72 static HRESULT process_extensions(HKEY hkeyExtensions, LPCOLESTR pszFileName, GUID * majorType, GUID * minorType)
83 /* Get the part of the name that matters */
84 extension = PathFindExtensionW(pszFileName);
85 if (*extension != '.')
88 l = RegOpenKeyExW(hkeyExtensions, extension, 0, KEY_READ, &hsub);
92 size = sizeof(keying);
93 l = RegQueryValueExW(hsub, mediatype_name, NULL, NULL, (LPBYTE)keying, &size);
95 CLSIDFromString(keying, majorType);
97 size = sizeof(keying);
99 l = RegQueryValueExW(hsub, subtype_name, NULL, NULL, (LPBYTE)keying, &size);
102 CLSIDFromString(keying, minorType);
111 static unsigned char byte_from_hex_char(WCHAR wHex)
113 switch (tolowerW(wHex))
125 return (wHex - '0') & 0xf;
132 return (wHex - 'a' + 10) & 0xf;
138 static HRESULT process_pattern_string(LPCWSTR wszPatternString, IAsyncReader * pReader)
148 TRACE("\t\tPattern string: %s\n", debugstr_w(wszPatternString));
150 /* format: "offset, bytestocompare, mask, value" */
152 ulOffset = strtolW(wszPatternString, NULL, 10);
154 if (!(wszPatternString = strchrW(wszPatternString, ',')))
157 wszPatternString++; /* skip ',' */
159 ulBytes = strtolW(wszPatternString, NULL, 10);
161 pbMask = HeapAlloc(GetProcessHeap(), 0, ulBytes);
162 pbValue = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, ulBytes);
163 pbFile = HeapAlloc(GetProcessHeap(), 0, ulBytes);
165 /* default mask is match everything */
166 memset(pbMask, 0xFF, ulBytes);
168 if (!(wszPatternString = strchrW(wszPatternString, ',')))
173 wszPatternString++; /* skip ',' */
174 while (!isxdigitW(*wszPatternString) && (*wszPatternString != ',')) wszPatternString++;
176 for (strpos = 0; isxdigitW(*wszPatternString) && (strpos/2 < ulBytes); wszPatternString++, strpos++)
178 if ((strpos % 2) == 1) /* odd numbered position */
179 pbMask[strpos / 2] |= byte_from_hex_char(*wszPatternString);
181 pbMask[strpos / 2] = byte_from_hex_char(*wszPatternString) << 4;
184 if (!(wszPatternString = strchrW(wszPatternString, ',')))
187 wszPatternString++; /* skip ',' */
192 for ( ; !isxdigitW(*wszPatternString) && (*wszPatternString != ','); wszPatternString++)
195 for (strpos = 0; isxdigitW(*wszPatternString) && (strpos/2 < ulBytes); wszPatternString++, strpos++)
197 if ((strpos % 2) == 1) /* odd numbered position */
198 pbValue[strpos / 2] |= byte_from_hex_char(*wszPatternString);
200 pbValue[strpos / 2] = byte_from_hex_char(*wszPatternString) << 4;
205 hr = IAsyncReader_SyncRead(pReader, ulOffset, ulBytes, pbFile);
210 for (i = 0; i < ulBytes; i++)
211 if ((pbFile[i] & pbMask[i]) != pbValue[i])
218 HeapFree(GetProcessHeap(), 0, pbMask);
219 HeapFree(GetProcessHeap(), 0, pbValue);
220 HeapFree(GetProcessHeap(), 0, pbFile);
222 /* if we encountered no errors with this string, and there is a following tuple, then we
223 * have to match that as well to succeed */
224 if ((hr == S_OK) && (wszPatternString = strchrW(wszPatternString, ',')))
225 return process_pattern_string(wszPatternString + 1, pReader);
230 static HRESULT GetClassMediaFile(IAsyncReader * pReader, LPCOLESTR pszFileName, GUID * majorType, GUID * minorType)
232 HKEY hkeyMediaType = NULL;
236 static const WCHAR wszMediaType[] = {'M','e','d','i','a',' ','T','y','p','e',0};
238 TRACE("(%p, %s, %p, %p)\n", pReader, debugstr_w(pszFileName), majorType, minorType);
240 *majorType = GUID_NULL;
241 *minorType = GUID_NULL;
243 lRet = RegOpenKeyExW(HKEY_CLASSES_ROOT, wszMediaType, 0, KEY_READ, &hkeyMediaType);
244 hr = HRESULT_FROM_WIN32(lRet);
250 for (indexMajor = 0; !bFound; indexMajor++)
253 WCHAR wszMajorKeyName[CHARS_IN_GUID];
254 DWORD dwKeyNameLength = sizeof(wszMajorKeyName) / sizeof(wszMajorKeyName[0]);
255 static const WCHAR wszExtensions[] = {'E','x','t','e','n','s','i','o','n','s',0};
257 if (RegEnumKeyExW(hkeyMediaType, indexMajor, wszMajorKeyName, &dwKeyNameLength, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
259 if (RegOpenKeyExW(hkeyMediaType, wszMajorKeyName, 0, KEY_READ, &hkeyMajor) != ERROR_SUCCESS)
261 TRACE("%s\n", debugstr_w(wszMajorKeyName));
262 if (!strcmpW(wszExtensions, wszMajorKeyName))
264 if (process_extensions(hkeyMajor, pszFileName, majorType, minorType) == S_OK)
271 for (indexMinor = 0; !bFound; indexMinor++)
274 WCHAR wszMinorKeyName[CHARS_IN_GUID];
275 DWORD dwMinorKeyNameLen = sizeof(wszMinorKeyName) / sizeof(wszMinorKeyName[0]);
279 if (RegEnumKeyExW(hkeyMajor, indexMinor, wszMinorKeyName, &dwMinorKeyNameLen, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
282 if (RegOpenKeyExW(hkeyMajor, wszMinorKeyName, 0, KEY_READ, &hkeyMinor) != ERROR_SUCCESS)
285 TRACE("\t%s\n", debugstr_w(wszMinorKeyName));
287 if (RegQueryInfoKeyW(hkeyMinor, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, &maxValueLen, NULL, NULL) != ERROR_SUCCESS)
290 for (indexValue = 0; !bFound; indexValue++)
293 WCHAR wszValueName[14]; /* longest name we should encounter will be "Source Filter" */
294 LPWSTR wszPatternString = HeapAlloc(GetProcessHeap(), 0, maxValueLen);
295 DWORD dwValueNameLen = sizeof(wszValueName) / sizeof(wszValueName[0]); /* remember this is in chars */
296 DWORD dwDataLen = maxValueLen; /* remember this is in bytes */
297 static const WCHAR wszSourceFilter[] = {'S','o','u','r','c','e',' ','F','i','l','t','e','r',0};
300 if ((temp = RegEnumValueW(hkeyMinor, indexValue, wszValueName, &dwValueNameLen, NULL, &dwType, (LPBYTE)wszPatternString, &dwDataLen)) != ERROR_SUCCESS)
302 HeapFree(GetProcessHeap(), 0, wszPatternString);
306 /* if it is not the source filter value */
307 if (strcmpW(wszValueName, wszSourceFilter))
309 if (process_pattern_string(wszPatternString, pReader) == S_OK)
311 if (SUCCEEDED(CLSIDFromString(wszMajorKeyName, majorType)) &&
312 SUCCEEDED(CLSIDFromString(wszMinorKeyName, minorType)))
316 HeapFree(GetProcessHeap(), 0, wszPatternString);
318 CloseHandle(hkeyMinor);
321 CloseHandle(hkeyMajor);
324 CloseHandle(hkeyMediaType);
326 if (SUCCEEDED(hr) && !bFound)
328 ERR("Media class not found\n");
332 TRACE("Found file's class: major = %s, subtype = %s\n", qzdebugstr_guid(majorType), qzdebugstr_guid(minorType));
337 HRESULT AsyncReader_create(IUnknown * pUnkOuter, LPVOID * ppv)
339 AsyncReader *pAsyncRead;
342 return CLASS_E_NOAGGREGATION;
344 pAsyncRead = CoTaskMemAlloc(sizeof(AsyncReader));
347 return E_OUTOFMEMORY;
349 pAsyncRead->lpVtbl = &AsyncReader_Vtbl;
350 pAsyncRead->lpVtblFSF = &FileSource_Vtbl;
351 pAsyncRead->refCount = 1;
352 pAsyncRead->filterInfo.achName[0] = '\0';
353 pAsyncRead->filterInfo.pGraph = NULL;
354 pAsyncRead->pOutputPin = NULL;
355 pAsyncRead->lastpinchange = GetTickCount();
356 pAsyncRead->state = State_Stopped;
358 InitializeCriticalSection(&pAsyncRead->csFilter);
359 pAsyncRead->csFilter.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": AsyncReader.csFilter");
361 pAsyncRead->pszFileName = NULL;
362 pAsyncRead->pmt = NULL;
366 TRACE("-- created at %p\n", pAsyncRead);
371 /** IUnknown methods **/
373 static HRESULT WINAPI AsyncReader_QueryInterface(IBaseFilter * iface, REFIID riid, LPVOID * ppv)
375 AsyncReader *This = (AsyncReader *)iface;
377 TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
381 if (IsEqualIID(riid, &IID_IUnknown))
383 else if (IsEqualIID(riid, &IID_IPersist))
385 else if (IsEqualIID(riid, &IID_IMediaFilter))
387 else if (IsEqualIID(riid, &IID_IBaseFilter))
389 else if (IsEqualIID(riid, &IID_IFileSourceFilter))
390 *ppv = &This->lpVtblFSF;
394 IUnknown_AddRef((IUnknown *)(*ppv));
398 if (!IsEqualIID(riid, &IID_IPin) && !IsEqualIID(riid, &IID_IMediaSeeking) &&
399 !IsEqualIID(riid, &IID_IVideoWindow) && !IsEqualIID(riid, &IID_IBasicAudio))
400 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
402 return E_NOINTERFACE;
405 static ULONG WINAPI AsyncReader_AddRef(IBaseFilter * iface)
407 AsyncReader *This = (AsyncReader *)iface;
408 ULONG refCount = InterlockedIncrement(&This->refCount);
410 TRACE("(%p)->() AddRef from %d\n", This, refCount - 1);
415 static ULONG WINAPI AsyncReader_Release(IBaseFilter * iface)
417 AsyncReader *This = (AsyncReader *)iface;
418 ULONG refCount = InterlockedDecrement(&This->refCount);
420 TRACE("(%p)->() Release from %d\n", This, refCount + 1);
424 if (This->pOutputPin)
427 if(SUCCEEDED(IPin_ConnectedTo(This->pOutputPin, &pConnectedTo)))
429 IPin_Disconnect(pConnectedTo);
430 IPin_Release(pConnectedTo);
432 IPin_Disconnect(This->pOutputPin);
433 IPin_Release(This->pOutputPin);
435 This->csFilter.DebugInfo->Spare[0] = 0;
436 DeleteCriticalSection(&This->csFilter);
438 CoTaskMemFree(This->pszFileName);
440 FreeMediaType(This->pmt);
448 /** IPersist methods **/
450 static HRESULT WINAPI AsyncReader_GetClassID(IBaseFilter * iface, CLSID * pClsid)
452 TRACE("(%p)\n", pClsid);
454 *pClsid = CLSID_AsyncReader;
459 /** IMediaFilter methods **/
461 static HRESULT WINAPI AsyncReader_Stop(IBaseFilter * iface)
463 AsyncReader *This = (AsyncReader *)iface;
467 This->state = State_Stopped;
472 static HRESULT WINAPI AsyncReader_Pause(IBaseFilter * iface)
474 AsyncReader *This = (AsyncReader *)iface;
478 This->state = State_Paused;
483 static HRESULT WINAPI AsyncReader_Run(IBaseFilter * iface, REFERENCE_TIME tStart)
485 AsyncReader *This = (AsyncReader *)iface;
487 TRACE("(%x%08x)\n", (ULONG)(tStart >> 32), (ULONG)tStart);
489 This->state = State_Running;
494 static HRESULT WINAPI AsyncReader_GetState(IBaseFilter * iface, DWORD dwMilliSecsTimeout, FILTER_STATE *pState)
496 AsyncReader *This = (AsyncReader *)iface;
498 TRACE("(%u, %p)\n", dwMilliSecsTimeout, pState);
500 *pState = This->state;
505 static HRESULT WINAPI AsyncReader_SetSyncSource(IBaseFilter * iface, IReferenceClock *pClock)
507 /* AsyncReader *This = (AsyncReader *)iface;*/
509 TRACE("(%p)\n", pClock);
514 static HRESULT WINAPI AsyncReader_GetSyncSource(IBaseFilter * iface, IReferenceClock **ppClock)
516 /* AsyncReader *This = (AsyncReader *)iface;*/
518 TRACE("(%p)\n", ppClock);
523 /** IBaseFilter methods **/
525 static IPin* WINAPI AsyncReader_GetPin(IBaseFilter *iface, int pos)
527 AsyncReader *This = (AsyncReader *)iface;
529 if (pos >= 1 || !This->pOutputPin)
532 IPin_AddRef(This->pOutputPin);
533 return This->pOutputPin;
536 static LONG WINAPI AsyncReader_GetPinCount(IBaseFilter *iface)
538 AsyncReader *This = (AsyncReader *)iface;
540 if (!This->pOutputPin)
546 static LONG WINAPI AsyncReader_GetPinVersion(IBaseFilter *iface)
548 AsyncReader *This = (AsyncReader *)iface;
550 /* Our pins are almost static, not changing so setting static tick count is ok */
551 return This->lastpinchange;
554 static HRESULT WINAPI AsyncReader_EnumPins(IBaseFilter * iface, IEnumPins **ppEnum)
556 AsyncReader *This = (AsyncReader *)iface;
558 TRACE("(%p/%p)->(%p)\n", This, iface, ppEnum);
560 return EnumPins_Construct(iface, AsyncReader_GetPin, AsyncReader_GetPinCount, AsyncReader_GetPinVersion, ppEnum);
563 static HRESULT WINAPI AsyncReader_FindPin(IBaseFilter * iface, LPCWSTR Id, IPin **ppPin)
565 FIXME("(%s, %p)\n", debugstr_w(Id), ppPin);
570 static HRESULT WINAPI AsyncReader_QueryFilterInfo(IBaseFilter * iface, FILTER_INFO *pInfo)
572 AsyncReader *This = (AsyncReader *)iface;
574 TRACE("(%p)\n", pInfo);
576 strcpyW(pInfo->achName, This->filterInfo.achName);
577 pInfo->pGraph = This->filterInfo.pGraph;
580 IFilterGraph_AddRef(pInfo->pGraph);
585 static HRESULT WINAPI AsyncReader_JoinFilterGraph(IBaseFilter * iface, IFilterGraph *pGraph, LPCWSTR pName)
587 AsyncReader *This = (AsyncReader *)iface;
589 TRACE("(%p, %s)\n", pGraph, debugstr_w(pName));
592 strcpyW(This->filterInfo.achName, pName);
594 *This->filterInfo.achName = 0;
595 This->filterInfo.pGraph = pGraph; /* NOTE: do NOT increase ref. count */
600 static HRESULT WINAPI AsyncReader_QueryVendorInfo(IBaseFilter * iface, LPWSTR *pVendorInfo)
602 FIXME("(%p)\n", pVendorInfo);
607 static const IBaseFilterVtbl AsyncReader_Vtbl =
609 AsyncReader_QueryInterface,
612 AsyncReader_GetClassID,
616 AsyncReader_GetState,
617 AsyncReader_SetSyncSource,
618 AsyncReader_GetSyncSource,
619 AsyncReader_EnumPins,
621 AsyncReader_QueryFilterInfo,
622 AsyncReader_JoinFilterGraph,
623 AsyncReader_QueryVendorInfo
626 static HRESULT WINAPI FileSource_QueryInterface(IFileSourceFilter * iface, REFIID riid, LPVOID * ppv)
628 AsyncReader *This = impl_from_IFileSourceFilter(iface);
630 return IBaseFilter_QueryInterface((IFileSourceFilter*)&This->lpVtbl, riid, ppv);
633 static ULONG WINAPI FileSource_AddRef(IFileSourceFilter * iface)
635 AsyncReader *This = impl_from_IFileSourceFilter(iface);
637 return IBaseFilter_AddRef((IFileSourceFilter*)&This->lpVtbl);
640 static ULONG WINAPI FileSource_Release(IFileSourceFilter * iface)
642 AsyncReader *This = impl_from_IFileSourceFilter(iface);
644 return IBaseFilter_Release((IFileSourceFilter*)&This->lpVtbl);
647 static HRESULT WINAPI FileSource_Load(IFileSourceFilter * iface, LPCOLESTR pszFileName, const AM_MEDIA_TYPE * pmt)
651 IAsyncReader * pReader = NULL;
652 AsyncReader *This = impl_from_IFileSourceFilter(iface);
654 TRACE("(%s, %p)\n", debugstr_w(pszFileName), pmt);
657 /* FIXME: check the sharing values that native uses */
658 hFile = CreateFileW(pszFileName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL);
660 if (hFile == INVALID_HANDLE_VALUE)
662 return HRESULT_FROM_WIN32(GetLastError());
666 hr = FileAsyncReader_Construct(hFile, (IBaseFilter *)&This->lpVtbl, &This->csFilter, &This->pOutputPin);
667 This->lastpinchange = GetTickCount();
670 hr = IPin_QueryInterface(This->pOutputPin, &IID_IAsyncReader, (LPVOID *)&pReader);
672 /* store file name & media type */
675 CoTaskMemFree(This->pszFileName);
677 FreeMediaType(This->pmt);
679 This->pszFileName = CoTaskMemAlloc((strlenW(pszFileName) + 1) * sizeof(WCHAR));
680 strcpyW(This->pszFileName, pszFileName);
682 This->pmt = CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE));
685 This->pmt->bFixedSizeSamples = TRUE;
686 This->pmt->bTemporalCompression = FALSE;
687 This->pmt->cbFormat = 0;
688 This->pmt->pbFormat = NULL;
689 This->pmt->pUnk = NULL;
690 This->pmt->lSampleSize = 0;
691 This->pmt->formattype = FORMAT_None;
692 hr = GetClassMediaFile(pReader, pszFileName, &This->pmt->majortype, &This->pmt->subtype);
695 CoTaskMemFree(This->pmt);
700 CopyMediaType(This->pmt, pmt);
704 IAsyncReader_Release(pReader);
708 if (This->pOutputPin)
710 IPin_Release(This->pOutputPin);
711 This->pOutputPin = NULL;
714 CoTaskMemFree(This->pszFileName);
716 FreeMediaType(This->pmt);
717 This->pszFileName = NULL;
723 /* FIXME: check return codes */
727 static HRESULT WINAPI FileSource_GetCurFile(IFileSourceFilter * iface, LPOLESTR * ppszFileName, AM_MEDIA_TYPE * pmt)
729 AsyncReader *This = impl_from_IFileSourceFilter(iface);
731 TRACE("(%p, %p)\n", ppszFileName, pmt);
736 /* copy file name & media type if available, otherwise clear the outputs */
737 if (This->pszFileName)
739 *ppszFileName = CoTaskMemAlloc((strlenW(This->pszFileName) + 1) * sizeof(WCHAR));
740 strcpyW(*ppszFileName, This->pszFileName);
743 *ppszFileName = NULL;
748 CopyMediaType(pmt, This->pmt);
750 ZeroMemory(pmt, sizeof(*pmt));
756 static const IFileSourceFilterVtbl FileSource_Vtbl =
758 FileSource_QueryInterface,
762 FileSource_GetCurFile
766 /* the dwUserData passed back to user */
767 typedef struct DATAREQUEST
769 IMediaSample * pSample; /* sample passed to us by user */
770 DWORD_PTR dwUserData; /* user data passed to us */
771 OVERLAPPED ovl; /* our overlapped structure */
774 typedef struct FileAsyncReader
777 const struct IAsyncReaderVtbl * lpVtblAR;
781 /* Why would you need more? Every sample has its own handle */
785 CRITICAL_SECTION csList; /* critical section to prevent concurrency issues */
786 DATAREQUEST *sample_list;
788 /* Have a handle for every sample, and then one more as flushing handle */
792 static inline FileAsyncReader *impl_from_IAsyncReader( IAsyncReader *iface )
794 return (FileAsyncReader *)((char*)iface - FIELD_OFFSET(FileAsyncReader, lpVtblAR));
797 static HRESULT WINAPI FileAsyncReaderPin_QueryAccept(IPin *iface, const AM_MEDIA_TYPE *pmt)
799 FileAsyncReader *This = (FileAsyncReader *)iface;
800 AM_MEDIA_TYPE *pmt_filter = ((AsyncReader *)This->pin.pin.pinInfo.pFilter)->pmt;
802 FIXME("(%p, %p)\n", iface, pmt);
804 if (IsEqualGUID(&pmt->majortype, &pmt_filter->majortype) &&
805 IsEqualGUID(&pmt->subtype, &pmt_filter->subtype) &&
806 IsEqualGUID(&pmt->formattype, &FORMAT_None))
812 static HRESULT WINAPI FileAsyncReaderPin_GetMediaType(IPin *iface, int iPosition, AM_MEDIA_TYPE *pmt)
814 FileAsyncReader *This = (FileAsyncReader *)iface;
818 return VFW_S_NO_MORE_ITEMS;
819 CopyMediaType(pmt, ((AsyncReader *)This->pin.pin.pinInfo.pFilter)->pmt);
823 /* overridden pin functions */
825 static HRESULT WINAPI FileAsyncReaderPin_QueryInterface(IPin * iface, REFIID riid, LPVOID * ppv)
827 FileAsyncReader *This = (FileAsyncReader *)iface;
828 TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
832 if (IsEqualIID(riid, &IID_IUnknown))
834 else if (IsEqualIID(riid, &IID_IPin))
836 else if (IsEqualIID(riid, &IID_IAsyncReader))
837 *ppv = &This->lpVtblAR;
841 IUnknown_AddRef((IUnknown *)(*ppv));
845 if (!IsEqualIID(riid, &IID_IPin) && !IsEqualIID(riid, &IID_IMediaSeeking))
846 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
848 return E_NOINTERFACE;
851 static ULONG WINAPI FileAsyncReaderPin_Release(IPin * iface)
853 FileAsyncReader *This = (FileAsyncReader *)iface;
854 ULONG refCount = InterlockedDecrement(&This->pin.pin.refCount);
857 TRACE("(%p)->() Release from %d\n", This, refCount + 1);
861 CoTaskMemFree(This->sample_list);
862 if (This->handle_list)
864 for (x = 0; x <= This->samples; ++x)
865 CloseHandle(This->handle_list[x]);
866 CoTaskMemFree(This->handle_list);
868 CloseHandle(This->hFile);
869 This->csList.DebugInfo->Spare[0] = 0;
870 DeleteCriticalSection(&This->csList);
877 static HRESULT WINAPI FileAsyncReaderPin_EnumMediaTypes(IPin * iface, IEnumMediaTypes ** ppEnum)
879 TRACE("(%p)\n", ppEnum);
881 return EnumMediaTypes_Construct(iface, FileAsyncReaderPin_GetMediaType, BasePinImpl_GetMediaTypeVersion, ppEnum);
884 static const IPinVtbl FileAsyncReaderPin_Vtbl =
886 FileAsyncReaderPin_QueryInterface,
888 FileAsyncReaderPin_Release,
890 OutputPin_ReceiveConnection,
891 BasePinImpl_Disconnect,
892 BasePinImpl_ConnectedTo,
893 BasePinImpl_ConnectionMediaType,
894 BasePinImpl_QueryPinInfo,
895 BasePinImpl_QueryDirection,
897 FileAsyncReaderPin_QueryAccept,
898 FileAsyncReaderPin_EnumMediaTypes,
899 BasePinImpl_QueryInternalConnections,
900 OutputPin_EndOfStream,
901 OutputPin_BeginFlush,
906 /* Function called as a helper to IPin_Connect */
907 /* specific AM_MEDIA_TYPE - it cannot be NULL */
908 /* this differs from standard OutputPin_ConnectSpecific only in that it
909 * doesn't need the IMemInputPin interface on the receiving pin */
910 static HRESULT FileAsyncReaderPin_ConnectSpecific(IPin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
912 OutputPin *This = (OutputPin *)iface;
915 TRACE("(%p, %p)\n", pReceivePin, pmt);
916 dump_AM_MEDIA_TYPE(pmt);
918 /* FIXME: call queryacceptproc */
920 This->pin.pConnectedTo = pReceivePin;
921 IPin_AddRef(pReceivePin);
922 CopyMediaType(&This->pin.mtCurrent, pmt);
924 hr = IPin_ReceiveConnection(pReceivePin, iface, pmt);
928 IPin_Release(This->pin.pConnectedTo);
929 This->pin.pConnectedTo = NULL;
930 FreeMediaType(&This->pin.mtCurrent);
933 TRACE(" -- %x\n", hr);
937 static HRESULT FileAsyncReader_Construct(HANDLE hFile, IBaseFilter * pBaseFilter, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
943 piOutput.dir = PINDIR_OUTPUT;
944 piOutput.pFilter = pBaseFilter;
945 strcpyW(piOutput.achName, wszOutputPinName);
946 hr = OutputPin_Construct(&FileAsyncReaderPin_Vtbl, sizeof(FileAsyncReader), &piOutput, NULL, pCritSec, ppPin);
950 FileAsyncReader *pPinImpl = (FileAsyncReader *)*ppPin;
951 pPinImpl->lpVtblAR = &FileAsyncReader_Vtbl;
952 pPinImpl->hFile = hFile;
953 pPinImpl->bFlushing = FALSE;
954 pPinImpl->sample_list = NULL;
955 pPinImpl->handle_list = NULL;
956 pPinImpl->queued_number = 0;
957 pPinImpl->pin.pConnectSpecific = FileAsyncReaderPin_ConnectSpecific;
958 InitializeCriticalSection(&pPinImpl->csList);
959 pPinImpl->csList.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": FileAsyncReader.csList");
966 static HRESULT WINAPI FileAsyncReader_QueryInterface(IAsyncReader * iface, REFIID riid, LPVOID * ppv)
968 FileAsyncReader *This = impl_from_IAsyncReader(iface);
970 return IPin_QueryInterface((IPin *)This, riid, ppv);
973 static ULONG WINAPI FileAsyncReader_AddRef(IAsyncReader * iface)
975 FileAsyncReader *This = impl_from_IAsyncReader(iface);
977 return IPin_AddRef((IPin *)This);
980 static ULONG WINAPI FileAsyncReader_Release(IAsyncReader * iface)
982 FileAsyncReader *This = impl_from_IAsyncReader(iface);
984 return IPin_Release((IPin *)This);
987 #define DEF_ALIGNMENT 1
989 static HRESULT WINAPI FileAsyncReader_RequestAllocator(IAsyncReader * iface, IMemAllocator * pPreferred, ALLOCATOR_PROPERTIES * pProps, IMemAllocator ** ppActual)
991 FileAsyncReader *This = impl_from_IAsyncReader(iface);
995 TRACE("(%p, %p, %p)\n", pPreferred, pProps, ppActual);
997 if (!pProps->cbAlign || (pProps->cbAlign % DEF_ALIGNMENT) != 0)
998 pProps->cbAlign = DEF_ALIGNMENT;
1002 hr = IMemAllocator_SetProperties(pPreferred, pProps, pProps);
1003 /* FIXME: check we are still aligned */
1006 IMemAllocator_AddRef(pPreferred);
1007 *ppActual = pPreferred;
1008 TRACE("FileAsyncReader_RequestAllocator -- %x\n", hr);
1015 hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC, &IID_IMemAllocator, (LPVOID *)&pPreferred);
1019 hr = IMemAllocator_SetProperties(pPreferred, pProps, pProps);
1020 /* FIXME: check we are still aligned */
1023 *ppActual = pPreferred;
1024 TRACE("FileAsyncReader_RequestAllocator -- %x\n", hr);
1031 CoTaskMemFree(This->sample_list);
1032 if (This->handle_list)
1035 for (x = 0; x <= This->samples; ++x)
1036 CloseHandle(This->handle_list[x]);
1037 CoTaskMemFree(This->handle_list);
1040 This->samples = pProps->cBuffers;
1041 This->oldest_sample = 0;
1042 TRACE("Samples: %u\n", This->samples);
1043 This->sample_list = CoTaskMemAlloc(sizeof(This->sample_list[0]) * pProps->cBuffers);
1044 This->handle_list = CoTaskMemAlloc(sizeof(HANDLE) * pProps->cBuffers * 2);
1046 if (This->sample_list && This->handle_list)
1049 ZeroMemory(This->sample_list, sizeof(This->sample_list[0]) * pProps->cBuffers);
1050 for (x = 0; x < This->samples; ++x)
1052 This->sample_list[x].ovl.hEvent = This->handle_list[x] = CreateEventW(NULL, 0, 0, NULL);
1053 if (x + 1 < This->samples)
1054 This->handle_list[This->samples + 1 + x] = This->handle_list[x];
1056 This->handle_list[This->samples] = CreateEventW(NULL, 1, 0, NULL);
1057 This->pin.allocProps = *pProps;
1062 CoTaskMemFree(This->sample_list);
1063 CoTaskMemFree(This->handle_list);
1065 This->sample_list = NULL;
1066 This->handle_list = NULL;
1074 IMemAllocator_Release(pPreferred);
1077 TRACE("-- %x\n", hr);
1081 /* we could improve the Request/WaitForNext mechanism by allowing out of order samples.
1082 * however, this would be quite complicated to do and may be a bit error prone */
1083 static HRESULT WINAPI FileAsyncReader_Request(IAsyncReader * iface, IMediaSample * pSample, DWORD_PTR dwUser)
1086 REFERENCE_TIME Start;
1087 REFERENCE_TIME Stop;
1088 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1089 LPBYTE pBuffer = NULL;
1091 TRACE("(%p, %lx)\n", pSample, dwUser);
1096 /* get start and stop positions in bytes */
1098 hr = IMediaSample_GetTime(pSample, &Start, &Stop);
1101 hr = IMediaSample_GetPointer(pSample, &pBuffer);
1103 EnterCriticalSection(&This->csList);
1104 if (This->bFlushing)
1106 LeaveCriticalSection(&This->csList);
1107 return VFW_E_WRONG_STATE;
1112 DWORD dwLength = (DWORD) BYTES_FROM_MEDIATIME(Stop - Start);
1113 DATAREQUEST *pDataRq;
1116 /* Try to insert above the waiting sample if possible */
1117 for (x = This->oldest_sample; x < This->samples; ++x)
1119 if (!This->sample_list[x].pSample)
1123 if (x >= This->samples)
1124 for (x = 0; x < This->oldest_sample; ++x)
1126 if (!This->sample_list[x].pSample)
1130 /* There must be a sample we have found */
1131 assert(x < This->samples);
1132 ++This->queued_number;
1134 pDataRq = This->sample_list + x;
1136 pDataRq->ovl.u.s.Offset = (DWORD) BYTES_FROM_MEDIATIME(Start);
1137 pDataRq->ovl.u.s.OffsetHigh = (DWORD)(BYTES_FROM_MEDIATIME(Start) >> (sizeof(DWORD) * 8));
1138 pDataRq->dwUserData = dwUser;
1140 /* we violate traditional COM rules here by maintaining
1141 * a reference to the sample, but not calling AddRef, but
1142 * that's what MSDN says to do */
1143 pDataRq->pSample = pSample;
1145 /* this is definitely not how it is implemented on Win9x
1146 * as they do not support async reads on files, but it is
1147 * sooo much easier to use this than messing around with threads!
1149 if (!ReadFile(This->hFile, pBuffer, dwLength, NULL, &pDataRq->ovl))
1150 hr = HRESULT_FROM_WIN32(GetLastError());
1152 /* ERROR_IO_PENDING is not actually an error since this is what we want! */
1153 if (hr == HRESULT_FROM_WIN32(ERROR_IO_PENDING))
1157 LeaveCriticalSection(&This->csList);
1159 TRACE("-- %x\n", hr);
1163 static HRESULT WINAPI FileAsyncReader_WaitForNext(IAsyncReader * iface, DWORD dwTimeout, IMediaSample ** ppSample, DWORD_PTR * pdwUser)
1166 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1169 TRACE("(%u, %p, %p)\n", dwTimeout, ppSample, pdwUser);
1174 EnterCriticalSection(&This->csList);
1175 if (!This->bFlushing)
1177 LONG oldest = This->oldest_sample;
1179 if (!This->queued_number)
1181 /* It could be that nothing is queued right now, but that can be fixed */
1182 WARN("Called without samples in queue and not flushing!!\n");
1184 LeaveCriticalSection(&This->csList);
1186 /* wait for an object to read, or time out */
1187 buffer = WaitForMultipleObjectsEx(This->samples+1, This->handle_list + oldest, FALSE, dwTimeout, TRUE);
1189 EnterCriticalSection(&This->csList);
1190 if (buffer <= This->samples)
1192 /* Re-scale the buffer back to normal */
1195 /* Uh oh, we overshot the flusher handle, renormalize it back to 0..Samples-1 */
1196 if (buffer > This->samples)
1197 buffer -= This->samples + 1;
1198 assert(buffer <= This->samples);
1201 if (buffer >= This->samples)
1203 if (buffer != This->samples)
1205 FIXME("Returned: %u (%08x)\n", buffer, GetLastError());
1209 hr = VFW_E_WRONG_STATE;
1213 --This->queued_number;
1216 if (This->bFlushing && buffer == ~0)
1218 for (buffer = 0; buffer < This->samples; ++buffer)
1220 if (This->sample_list[buffer].pSample)
1222 ResetEvent(This->handle_list[buffer]);
1226 if (buffer == This->samples)
1228 assert(!This->queued_number);
1233 --This->queued_number;
1240 REFERENCE_TIME rtStart, rtStop;
1241 REFERENCE_TIME rtSampleStart, rtSampleStop;
1242 DATAREQUEST *pDataRq = This->sample_list + buffer;
1245 /* get any errors */
1246 if (!This->bFlushing && !GetOverlappedResult(This->hFile, &pDataRq->ovl, &dwBytes, FALSE))
1247 hr = HRESULT_FROM_WIN32(GetLastError());
1249 /* Return the sample no matter what so it can be destroyed */
1250 *ppSample = pDataRq->pSample;
1251 *pdwUser = pDataRq->dwUserData;
1253 if (This->bFlushing)
1254 hr = VFW_E_WRONG_STATE;
1259 /* Set the time on the sample */
1260 IMediaSample_SetActualDataLength(pDataRq->pSample, dwBytes);
1262 rtStart = (DWORD64)pDataRq->ovl.u.s.Offset + ((DWORD64)pDataRq->ovl.u.s.OffsetHigh << 32);
1263 rtStart = MEDIATIME_FROM_BYTES(rtStart);
1264 rtStop = rtStart + MEDIATIME_FROM_BYTES(dwBytes);
1266 IMediaSample_GetTime(pDataRq->pSample, &rtSampleStart, &rtSampleStop);
1267 assert(rtStart == rtSampleStart);
1268 assert(rtStop <= rtSampleStop);
1270 IMediaSample_SetTime(pDataRq->pSample, &rtStart, &rtStop);
1271 assert(rtStart == rtSampleStart);
1273 assert(rtStop == rtSampleStop);
1275 assert(rtStop == rtStart);
1277 This->sample_list[buffer].pSample = NULL;
1278 assert(This->oldest_sample < This->samples);
1280 if (buffer == This->oldest_sample)
1283 for (x = This->oldest_sample + 1; x < This->samples; ++x)
1284 if (This->sample_list[x].pSample)
1286 if (x >= This->samples)
1287 for (x = 0; x < This->oldest_sample; ++x)
1288 if (This->sample_list[x].pSample)
1290 if (This->oldest_sample == x)
1291 /* No samples found, reset to 0 */
1293 This->oldest_sample = x;
1296 LeaveCriticalSection(&This->csList);
1298 TRACE("-- %x\n", hr);
1302 static HRESULT WINAPI FileAsyncReader_SyncRead(IAsyncReader * iface, LONGLONG llPosition, LONG lLength, BYTE * pBuffer);
1304 static HRESULT WINAPI FileAsyncReader_SyncReadAligned(IAsyncReader * iface, IMediaSample * pSample)
1307 REFERENCE_TIME tStart;
1308 REFERENCE_TIME tStop;
1311 TRACE("(%p)\n", pSample);
1313 hr = IMediaSample_GetTime(pSample, &tStart, &tStop);
1316 hr = IMediaSample_GetPointer(pSample, &pBuffer);
1319 hr = FileAsyncReader_SyncRead(iface,
1320 BYTES_FROM_MEDIATIME(tStart),
1321 (LONG) BYTES_FROM_MEDIATIME(tStop - tStart),
1324 TRACE("-- %x\n", hr);
1328 static HRESULT WINAPI FileAsyncReader_SyncRead(IAsyncReader * iface, LONGLONG llPosition, LONG lLength, BYTE * pBuffer)
1332 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1334 TRACE("(%x%08x, %d, %p)\n", (ULONG)(llPosition >> 32), (ULONG)llPosition, lLength, pBuffer);
1336 ZeroMemory(&ovl, sizeof(ovl));
1338 ovl.hEvent = CreateEventW(NULL, 0, 0, NULL);
1339 /* NOTE: llPosition is the actual byte position to start reading from */
1340 ovl.u.s.Offset = (DWORD) llPosition;
1341 ovl.u.s.OffsetHigh = (DWORD) (llPosition >> (sizeof(DWORD) * 8));
1343 if (!ReadFile(This->hFile, pBuffer, lLength, NULL, &ovl))
1344 hr = HRESULT_FROM_WIN32(GetLastError());
1346 if (hr == HRESULT_FROM_WIN32(ERROR_IO_PENDING))
1353 if (!GetOverlappedResult(This->hFile, &ovl, &dwBytesRead, TRUE))
1354 hr = HRESULT_FROM_WIN32(GetLastError());
1357 CloseHandle(ovl.hEvent);
1359 TRACE("-- %x\n", hr);
1363 static HRESULT WINAPI FileAsyncReader_Length(IAsyncReader * iface, LONGLONG * pTotal, LONGLONG * pAvailable)
1367 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1369 TRACE("(%p, %p)\n", pTotal, pAvailable);
1371 if (((dwSizeLow = GetFileSize(This->hFile, &dwSizeHigh)) == -1) &&
1372 (GetLastError() != NO_ERROR))
1373 return HRESULT_FROM_WIN32(GetLastError());
1375 *pTotal = (LONGLONG)dwSizeLow | (LONGLONG)dwSizeHigh << (sizeof(DWORD) * 8);
1377 *pAvailable = *pTotal;
1382 static HRESULT WINAPI FileAsyncReader_BeginFlush(IAsyncReader * iface)
1384 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1388 EnterCriticalSection(&This->csList);
1389 This->bFlushing = TRUE;
1390 CancelIo(This->hFile);
1391 SetEvent(This->handle_list[This->samples]);
1392 LeaveCriticalSection(&This->csList);
1397 static HRESULT WINAPI FileAsyncReader_EndFlush(IAsyncReader * iface)
1399 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1404 EnterCriticalSection(&This->csList);
1405 ResetEvent(This->handle_list[This->samples]);
1406 This->bFlushing = FALSE;
1407 for (x = 0; x < This->samples; ++x)
1408 assert(!This->sample_list[x].pSample);
1410 LeaveCriticalSection(&This->csList);
1415 static const IAsyncReaderVtbl FileAsyncReader_Vtbl =
1417 FileAsyncReader_QueryInterface,
1418 FileAsyncReader_AddRef,
1419 FileAsyncReader_Release,
1420 FileAsyncReader_RequestAllocator,
1421 FileAsyncReader_Request,
1422 FileAsyncReader_WaitForNext,
1423 FileAsyncReader_SyncReadAligned,
1424 FileAsyncReader_SyncRead,
1425 FileAsyncReader_Length,
1426 FileAsyncReader_BeginFlush,
1427 FileAsyncReader_EndFlush,