advapi32: Fix the failing QueryServiceConfig2 test on platforms win2k3 and vista.
[wine/multimedia.git] / dlls / quartz / filesource.c
blob73c92b2f4814c31092d498ca1a7bd6d358562cf4
1 /*
2 * File Source Filter
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"
28 #include "pin.h"
29 #include "uuids.h"
30 #include "vfwmsgs.h"
31 #include "winbase.h"
32 #include "winreg.h"
33 #include <assert.h>
35 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
37 static const WCHAR wszOutputPinName[] = { 'O','u','t','p','u','t',0 };
39 typedef struct AsyncReader
41 const IBaseFilterVtbl * lpVtbl;
42 const IFileSourceFilterVtbl * lpVtblFSF;
44 LONG refCount;
45 FILTER_INFO filterInfo;
46 FILTER_STATE state;
47 CRITICAL_SECTION csFilter;
49 IPin * pOutputPin;
50 LPOLESTR pszFileName;
51 AM_MEDIA_TYPE * pmt;
52 } AsyncReader;
54 static const IBaseFilterVtbl AsyncReader_Vtbl;
55 static const IFileSourceFilterVtbl FileSource_Vtbl;
56 static const IAsyncReaderVtbl FileAsyncReader_Vtbl;
58 static HRESULT FileAsyncReader_Construct(HANDLE hFile, IBaseFilter * pBaseFilter, LPCRITICAL_SECTION pCritSec, IPin ** ppPin);
60 static inline AsyncReader *impl_from_IFileSourceFilter( IFileSourceFilter *iface )
62 return (AsyncReader *)((char*)iface - FIELD_OFFSET(AsyncReader, lpVtblFSF));
65 static HRESULT process_extensions(HKEY hkeyExtensions, LPCOLESTR pszFileName, GUID * majorType, GUID * minorType)
67 /* FIXME: implement */
68 return E_NOTIMPL;
71 static unsigned char byte_from_hex_char(WCHAR wHex)
73 switch (tolowerW(wHex))
75 case '0':
76 case '1':
77 case '2':
78 case '3':
79 case '4':
80 case '5':
81 case '6':
82 case '7':
83 case '8':
84 case '9':
85 return (wHex - '0') & 0xf;
86 case 'a':
87 case 'b':
88 case 'c':
89 case 'd':
90 case 'e':
91 case 'f':
92 return (wHex - 'a' + 10) & 0xf;
93 default:
94 return 0;
98 static HRESULT process_pattern_string(LPCWSTR wszPatternString, IAsyncReader * pReader)
100 ULONG ulOffset;
101 ULONG ulBytes;
102 BYTE * pbMask;
103 BYTE * pbValue;
104 BYTE * pbFile;
105 HRESULT hr = S_OK;
106 ULONG strpos;
108 TRACE("\t\tPattern string: %s\n", debugstr_w(wszPatternString));
110 /* format: "offset, bytestocompare, mask, value" */
112 ulOffset = strtolW(wszPatternString, NULL, 10);
114 if (!(wszPatternString = strchrW(wszPatternString, ',')))
115 return E_INVALIDARG;
117 wszPatternString++; /* skip ',' */
119 ulBytes = strtolW(wszPatternString, NULL, 10);
121 pbMask = HeapAlloc(GetProcessHeap(), 0, ulBytes);
122 pbValue = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, ulBytes);
123 pbFile = HeapAlloc(GetProcessHeap(), 0, ulBytes);
125 /* default mask is match everything */
126 memset(pbMask, 0xFF, ulBytes);
128 if (!(wszPatternString = strchrW(wszPatternString, ',')))
129 hr = E_INVALIDARG;
131 wszPatternString++; /* skip ',' */
133 if (hr == S_OK)
135 for ( ; !isxdigitW(*wszPatternString) && (*wszPatternString != ','); wszPatternString++)
138 for (strpos = 0; isxdigitW(*wszPatternString) && (strpos/2 < ulBytes); wszPatternString++, strpos++)
140 if ((strpos % 2) == 1) /* odd numbered position */
141 pbMask[strpos / 2] |= byte_from_hex_char(*wszPatternString);
142 else
143 pbMask[strpos / 2] = byte_from_hex_char(*wszPatternString) << 4;
146 if (!(wszPatternString = strchrW(wszPatternString, ',')))
147 hr = E_INVALIDARG;
149 wszPatternString++; /* skip ',' */
152 if (hr == S_OK)
154 for ( ; !isxdigitW(*wszPatternString) && (*wszPatternString != ','); wszPatternString++)
157 for (strpos = 0; isxdigitW(*wszPatternString) && (strpos/2 < ulBytes); wszPatternString++, strpos++)
159 if ((strpos % 2) == 1) /* odd numbered position */
160 pbValue[strpos / 2] |= byte_from_hex_char(*wszPatternString);
161 else
162 pbValue[strpos / 2] = byte_from_hex_char(*wszPatternString) << 4;
166 if (hr == S_OK)
167 hr = IAsyncReader_SyncRead(pReader, ulOffset, ulBytes, pbFile);
169 if (hr == S_OK)
171 ULONG i;
172 for (i = 0; i < ulBytes; i++)
173 if ((pbFile[i] & pbMask[i]) != pbValue[i])
175 hr = S_FALSE;
176 break;
180 HeapFree(GetProcessHeap(), 0, pbMask);
181 HeapFree(GetProcessHeap(), 0, pbValue);
182 HeapFree(GetProcessHeap(), 0, pbFile);
184 /* if we encountered no errors with this string, and there is a following tuple, then we
185 * have to match that as well to succeed */
186 if ((hr == S_OK) && (wszPatternString = strchrW(wszPatternString, ',')))
187 return process_pattern_string(wszPatternString + 1, pReader);
188 else
189 return hr;
192 static HRESULT GetClassMediaFile(IAsyncReader * pReader, LPCOLESTR pszFileName, GUID * majorType, GUID * minorType)
194 HKEY hkeyMediaType = NULL;
195 LONG lRet;
196 HRESULT hr = S_OK;
197 BOOL bFound = FALSE;
198 static const WCHAR wszMediaType[] = {'M','e','d','i','a',' ','T','y','p','e',0};
200 CopyMemory(majorType, &GUID_NULL, sizeof(*majorType));
201 CopyMemory(minorType, &GUID_NULL, sizeof(*minorType));
203 lRet = RegOpenKeyExW(HKEY_CLASSES_ROOT, wszMediaType, 0, KEY_READ, &hkeyMediaType);
204 hr = HRESULT_FROM_WIN32(lRet);
206 if (SUCCEEDED(hr))
208 DWORD indexMajor;
210 for (indexMajor = 0; !bFound; indexMajor++)
212 HKEY hkeyMajor;
213 WCHAR wszMajorKeyName[CHARS_IN_GUID];
214 DWORD dwKeyNameLength = sizeof(wszMajorKeyName) / sizeof(wszMajorKeyName[0]);
215 static const WCHAR wszExtensions[] = {'E','x','t','e','n','s','i','o','n','s',0};
217 if (RegEnumKeyExW(hkeyMediaType, indexMajor, wszMajorKeyName, &dwKeyNameLength, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
218 break;
219 if (RegOpenKeyExW(hkeyMediaType, wszMajorKeyName, 0, KEY_READ, &hkeyMajor) != ERROR_SUCCESS)
220 break;
221 TRACE("%s\n", debugstr_w(wszMajorKeyName));
222 if (!strcmpW(wszExtensions, wszMajorKeyName))
224 if (process_extensions(hkeyMajor, pszFileName, majorType, minorType) == S_OK)
225 bFound = TRUE;
227 else
229 DWORD indexMinor;
231 for (indexMinor = 0; !bFound; indexMinor++)
233 HKEY hkeyMinor;
234 WCHAR wszMinorKeyName[CHARS_IN_GUID];
235 DWORD dwMinorKeyNameLen = sizeof(wszMinorKeyName) / sizeof(wszMinorKeyName[0]);
236 DWORD maxValueLen;
237 DWORD indexValue;
239 if (RegEnumKeyExW(hkeyMajor, indexMinor, wszMinorKeyName, &dwMinorKeyNameLen, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
240 break;
242 if (RegOpenKeyExW(hkeyMajor, wszMinorKeyName, 0, KEY_READ, &hkeyMinor) != ERROR_SUCCESS)
243 break;
245 TRACE("\t%s\n", debugstr_w(wszMinorKeyName));
247 if (RegQueryInfoKeyW(hkeyMinor, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, &maxValueLen, NULL, NULL) != ERROR_SUCCESS)
248 break;
250 for (indexValue = 0; !bFound; indexValue++)
252 DWORD dwType;
253 WCHAR wszValueName[14]; /* longest name we should encounter will be "Source Filter" */
254 LPWSTR wszPatternString = HeapAlloc(GetProcessHeap(), 0, maxValueLen);
255 DWORD dwValueNameLen = sizeof(wszValueName) / sizeof(wszValueName[0]); /* remember this is in chars */
256 DWORD dwDataLen = maxValueLen; /* remember this is in bytes */
257 static const WCHAR wszSourceFilter[] = {'S','o','u','r','c','e',' ','F','i','l','t','e','r',0};
258 LONG temp;
260 if ((temp = RegEnumValueW(hkeyMinor, indexValue, wszValueName, &dwValueNameLen, NULL, &dwType, (LPBYTE)wszPatternString, &dwDataLen)) != ERROR_SUCCESS)
262 HeapFree(GetProcessHeap(), 0, wszPatternString);
263 break;
266 /* if it is not the source filter value */
267 if (strcmpW(wszValueName, wszSourceFilter))
269 if (process_pattern_string(wszPatternString, pReader) == S_OK)
271 if (SUCCEEDED(CLSIDFromString(wszMajorKeyName, majorType)) &&
272 SUCCEEDED(CLSIDFromString(wszMinorKeyName, minorType)))
273 bFound = TRUE;
276 HeapFree(GetProcessHeap(), 0, wszPatternString);
278 CloseHandle(hkeyMinor);
281 CloseHandle(hkeyMajor);
284 CloseHandle(hkeyMediaType);
286 if (SUCCEEDED(hr) && !bFound)
288 ERR("Media class not found\n");
289 hr = E_FAIL;
291 else if (bFound)
292 TRACE("Found file's class: major = %s, subtype = %s\n", qzdebugstr_guid(majorType), qzdebugstr_guid(minorType));
294 return hr;
297 HRESULT AsyncReader_create(IUnknown * pUnkOuter, LPVOID * ppv)
299 AsyncReader *pAsyncRead;
301 if( pUnkOuter )
302 return CLASS_E_NOAGGREGATION;
304 pAsyncRead = CoTaskMemAlloc(sizeof(AsyncReader));
306 if (!pAsyncRead)
307 return E_OUTOFMEMORY;
309 pAsyncRead->lpVtbl = &AsyncReader_Vtbl;
310 pAsyncRead->lpVtblFSF = &FileSource_Vtbl;
311 pAsyncRead->refCount = 1;
312 pAsyncRead->filterInfo.achName[0] = '\0';
313 pAsyncRead->filterInfo.pGraph = NULL;
314 pAsyncRead->pOutputPin = NULL;
316 InitializeCriticalSection(&pAsyncRead->csFilter);
317 pAsyncRead->csFilter.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": AsyncReader.csFilter");
319 pAsyncRead->pszFileName = NULL;
320 pAsyncRead->pmt = NULL;
322 *ppv = (LPVOID)pAsyncRead;
324 TRACE("-- created at %p\n", pAsyncRead);
326 return S_OK;
329 /** IUnknown methods **/
331 static HRESULT WINAPI AsyncReader_QueryInterface(IBaseFilter * iface, REFIID riid, LPVOID * ppv)
333 AsyncReader *This = (AsyncReader *)iface;
335 TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
337 *ppv = NULL;
339 if (IsEqualIID(riid, &IID_IUnknown))
340 *ppv = (LPVOID)This;
341 else if (IsEqualIID(riid, &IID_IPersist))
342 *ppv = (LPVOID)This;
343 else if (IsEqualIID(riid, &IID_IMediaFilter))
344 *ppv = (LPVOID)This;
345 else if (IsEqualIID(riid, &IID_IBaseFilter))
346 *ppv = (LPVOID)This;
347 else if (IsEqualIID(riid, &IID_IFileSourceFilter))
348 *ppv = (LPVOID)(&This->lpVtblFSF);
350 if (*ppv)
352 IUnknown_AddRef((IUnknown *)(*ppv));
353 return S_OK;
356 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
358 return E_NOINTERFACE;
361 static ULONG WINAPI AsyncReader_AddRef(IBaseFilter * iface)
363 AsyncReader *This = (AsyncReader *)iface;
364 ULONG refCount = InterlockedIncrement(&This->refCount);
366 TRACE("(%p)->() AddRef from %d\n", This, refCount - 1);
368 return refCount;
371 static ULONG WINAPI AsyncReader_Release(IBaseFilter * iface)
373 AsyncReader *This = (AsyncReader *)iface;
374 ULONG refCount = InterlockedDecrement(&This->refCount);
376 TRACE("(%p)->() Release from %d\n", This, refCount + 1);
378 if (!refCount)
380 if (This->pOutputPin)
382 IPin *pConnectedTo;
383 if(SUCCEEDED(IPin_ConnectedTo(This->pOutputPin, &pConnectedTo)))
385 IPin_Disconnect(pConnectedTo);
386 IPin_Release(pConnectedTo);
388 IPin_Disconnect(This->pOutputPin);
389 IPin_Release(This->pOutputPin);
391 This->csFilter.DebugInfo->Spare[0] = 0;
392 DeleteCriticalSection(&This->csFilter);
393 This->lpVtbl = NULL;
394 CoTaskMemFree(This);
395 return 0;
397 else
398 return refCount;
401 /** IPersist methods **/
403 static HRESULT WINAPI AsyncReader_GetClassID(IBaseFilter * iface, CLSID * pClsid)
405 TRACE("(%p)\n", pClsid);
407 *pClsid = CLSID_AsyncReader;
409 return S_OK;
412 /** IMediaFilter methods **/
414 static HRESULT WINAPI AsyncReader_Stop(IBaseFilter * iface)
416 AsyncReader *This = (AsyncReader *)iface;
418 TRACE("()\n");
420 This->state = State_Stopped;
422 return S_OK;
425 static HRESULT WINAPI AsyncReader_Pause(IBaseFilter * iface)
427 AsyncReader *This = (AsyncReader *)iface;
429 TRACE("()\n");
431 This->state = State_Paused;
433 return S_OK;
436 static HRESULT WINAPI AsyncReader_Run(IBaseFilter * iface, REFERENCE_TIME tStart)
438 AsyncReader *This = (AsyncReader *)iface;
440 TRACE("(%x%08x)\n", (ULONG)(tStart >> 32), (ULONG)tStart);
442 This->state = State_Running;
444 return S_OK;
447 static HRESULT WINAPI AsyncReader_GetState(IBaseFilter * iface, DWORD dwMilliSecsTimeout, FILTER_STATE *pState)
449 AsyncReader *This = (AsyncReader *)iface;
451 TRACE("(%u, %p)\n", dwMilliSecsTimeout, pState);
453 *pState = This->state;
455 return S_OK;
458 static HRESULT WINAPI AsyncReader_SetSyncSource(IBaseFilter * iface, IReferenceClock *pClock)
460 /* AsyncReader *This = (AsyncReader *)iface;*/
462 TRACE("(%p)\n", pClock);
464 return S_OK;
467 static HRESULT WINAPI AsyncReader_GetSyncSource(IBaseFilter * iface, IReferenceClock **ppClock)
469 /* AsyncReader *This = (AsyncReader *)iface;*/
471 TRACE("(%p)\n", ppClock);
473 return S_OK;
476 /** IBaseFilter methods **/
478 static HRESULT WINAPI AsyncReader_EnumPins(IBaseFilter * iface, IEnumPins **ppEnum)
480 ENUMPINDETAILS epd;
481 AsyncReader *This = (AsyncReader *)iface;
483 TRACE("(%p)\n", ppEnum);
485 epd.cPins = This->pOutputPin ? 1 : 0;
486 epd.ppPins = &This->pOutputPin;
487 return IEnumPinsImpl_Construct(&epd, ppEnum);
490 static HRESULT WINAPI AsyncReader_FindPin(IBaseFilter * iface, LPCWSTR Id, IPin **ppPin)
492 FIXME("(%s, %p)\n", debugstr_w(Id), ppPin);
494 return E_NOTIMPL;
497 static HRESULT WINAPI AsyncReader_QueryFilterInfo(IBaseFilter * iface, FILTER_INFO *pInfo)
499 AsyncReader *This = (AsyncReader *)iface;
501 TRACE("(%p)\n", pInfo);
503 strcpyW(pInfo->achName, This->filterInfo.achName);
504 pInfo->pGraph = This->filterInfo.pGraph;
506 if (pInfo->pGraph)
507 IFilterGraph_AddRef(pInfo->pGraph);
509 return S_OK;
512 static HRESULT WINAPI AsyncReader_JoinFilterGraph(IBaseFilter * iface, IFilterGraph *pGraph, LPCWSTR pName)
514 AsyncReader *This = (AsyncReader *)iface;
516 TRACE("(%p, %s)\n", pGraph, debugstr_w(pName));
518 if (pName)
519 strcpyW(This->filterInfo.achName, pName);
520 else
521 *This->filterInfo.achName = 0;
522 This->filterInfo.pGraph = pGraph; /* NOTE: do NOT increase ref. count */
524 return S_OK;
527 static HRESULT WINAPI AsyncReader_QueryVendorInfo(IBaseFilter * iface, LPWSTR *pVendorInfo)
529 FIXME("(%p)\n", pVendorInfo);
531 return E_NOTIMPL;
534 static const IBaseFilterVtbl AsyncReader_Vtbl =
536 AsyncReader_QueryInterface,
537 AsyncReader_AddRef,
538 AsyncReader_Release,
539 AsyncReader_GetClassID,
540 AsyncReader_Stop,
541 AsyncReader_Pause,
542 AsyncReader_Run,
543 AsyncReader_GetState,
544 AsyncReader_SetSyncSource,
545 AsyncReader_GetSyncSource,
546 AsyncReader_EnumPins,
547 AsyncReader_FindPin,
548 AsyncReader_QueryFilterInfo,
549 AsyncReader_JoinFilterGraph,
550 AsyncReader_QueryVendorInfo
553 static HRESULT WINAPI FileSource_QueryInterface(IFileSourceFilter * iface, REFIID riid, LPVOID * ppv)
555 AsyncReader *This = impl_from_IFileSourceFilter(iface);
557 return IBaseFilter_QueryInterface((IFileSourceFilter*)&This->lpVtbl, riid, ppv);
560 static ULONG WINAPI FileSource_AddRef(IFileSourceFilter * iface)
562 AsyncReader *This = impl_from_IFileSourceFilter(iface);
564 return IBaseFilter_AddRef((IFileSourceFilter*)&This->lpVtbl);
567 static ULONG WINAPI FileSource_Release(IFileSourceFilter * iface)
569 AsyncReader *This = impl_from_IFileSourceFilter(iface);
571 return IBaseFilter_Release((IFileSourceFilter*)&This->lpVtbl);
574 static HRESULT WINAPI FileSource_Load(IFileSourceFilter * iface, LPCOLESTR pszFileName, const AM_MEDIA_TYPE * pmt)
576 HRESULT hr;
577 HANDLE hFile;
578 IAsyncReader * pReader = NULL;
579 AsyncReader *This = impl_from_IFileSourceFilter(iface);
581 TRACE("(%s, %p)\n", debugstr_w(pszFileName), pmt);
583 /* open file */
584 /* FIXME: check the sharing values that native uses */
585 hFile = CreateFileW(pszFileName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL);
587 if (hFile == INVALID_HANDLE_VALUE)
589 return HRESULT_FROM_WIN32(GetLastError());
592 /* create pin */
593 hr = FileAsyncReader_Construct(hFile, (IBaseFilter *)&This->lpVtbl, &This->csFilter, &This->pOutputPin);
595 if (SUCCEEDED(hr))
596 hr = IPin_QueryInterface(This->pOutputPin, &IID_IAsyncReader, (LPVOID *)&pReader);
598 /* store file name & media type */
599 if (SUCCEEDED(hr))
601 This->pszFileName = CoTaskMemAlloc((strlenW(pszFileName) + 1) * sizeof(WCHAR));
602 strcpyW(This->pszFileName, pszFileName);
603 This->pmt = CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE));
604 if (!pmt)
606 This->pmt->bFixedSizeSamples = TRUE;
607 This->pmt->bTemporalCompression = FALSE;
608 This->pmt->cbFormat = 0;
609 This->pmt->pbFormat = NULL;
610 This->pmt->pUnk = NULL;
611 This->pmt->lSampleSize = 0;
612 memcpy(&This->pmt->formattype, &FORMAT_None, sizeof(FORMAT_None));
613 hr = GetClassMediaFile(pReader, pszFileName, &This->pmt->majortype, &This->pmt->subtype);
614 if (FAILED(hr))
616 CoTaskMemFree(This->pmt);
617 This->pmt = NULL;
620 else
621 CopyMediaType(This->pmt, pmt);
624 if (pReader)
625 IAsyncReader_Release(pReader);
627 if (FAILED(hr))
629 if (This->pOutputPin)
631 IPin_Release(This->pOutputPin);
632 This->pOutputPin = NULL;
635 CoTaskMemFree(This->pszFileName);
636 This->pszFileName = NULL;
638 CloseHandle(hFile);
641 /* FIXME: check return codes */
642 return hr;
645 static HRESULT WINAPI FileSource_GetCurFile(IFileSourceFilter * iface, LPOLESTR * ppszFileName, AM_MEDIA_TYPE * pmt)
647 AsyncReader *This = impl_from_IFileSourceFilter(iface);
649 TRACE("(%p, %p)\n", ppszFileName, pmt);
651 if (!ppszFileName)
652 return E_POINTER;
654 /* copy file name & media type if available, otherwise clear the outputs */
655 if (This->pszFileName)
657 *ppszFileName = CoTaskMemAlloc((strlenW(This->pszFileName) + 1) * sizeof(WCHAR));
658 strcpyW(*ppszFileName, This->pszFileName);
660 else
661 *ppszFileName = NULL;
663 if (pmt)
665 if (This->pmt)
666 CopyMediaType(pmt, This->pmt);
667 else
668 ZeroMemory(pmt, sizeof(*pmt));
671 return S_OK;
674 static const IFileSourceFilterVtbl FileSource_Vtbl =
676 FileSource_QueryInterface,
677 FileSource_AddRef,
678 FileSource_Release,
679 FileSource_Load,
680 FileSource_GetCurFile
684 /* the dwUserData passed back to user */
685 typedef struct DATAREQUEST
687 IMediaSample * pSample; /* sample passed to us by user */
688 DWORD_PTR dwUserData; /* user data passed to us */
689 OVERLAPPED ovl; /* our overlapped structure */
691 struct DATAREQUEST * pNext; /* next data request in list */
692 } DATAREQUEST;
694 static void queue(DATAREQUEST * pHead, DATAREQUEST * pItem)
696 DATAREQUEST * pCurrent;
697 for (pCurrent = pHead; pCurrent->pNext; pCurrent = pCurrent->pNext)
699 pCurrent->pNext = pItem;
702 typedef struct FileAsyncReader
704 OutputPin pin;
705 const struct IAsyncReaderVtbl * lpVtblAR;
707 HANDLE hFile;
708 HANDLE hEvent;
709 BOOL bFlushing;
710 DATAREQUEST * pHead; /* head of data request list */
711 CRITICAL_SECTION csList; /* critical section to protect operations on list */
712 } FileAsyncReader;
714 static inline FileAsyncReader *impl_from_IAsyncReader( IAsyncReader *iface )
716 return (FileAsyncReader *)((char*)iface - FIELD_OFFSET(FileAsyncReader, lpVtblAR));
719 static HRESULT AcceptProcAFR(LPVOID iface, const AM_MEDIA_TYPE *pmt)
721 AsyncReader *This = (AsyncReader *)iface;
723 FIXME("(%p, %p)\n", iface, pmt);
725 if (IsEqualGUID(&pmt->majortype, &This->pmt->majortype) &&
726 IsEqualGUID(&pmt->subtype, &This->pmt->subtype) &&
727 IsEqualGUID(&pmt->formattype, &FORMAT_None))
728 return S_OK;
730 return S_FALSE;
733 /* overriden pin functions */
735 static HRESULT WINAPI FileAsyncReaderPin_QueryInterface(IPin * iface, REFIID riid, LPVOID * ppv)
737 FileAsyncReader *This = (FileAsyncReader *)iface;
738 TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
740 *ppv = NULL;
742 if (IsEqualIID(riid, &IID_IUnknown))
743 *ppv = (LPVOID)This;
744 else if (IsEqualIID(riid, &IID_IPin))
745 *ppv = (LPVOID)This;
746 else if (IsEqualIID(riid, &IID_IAsyncReader))
747 *ppv = (LPVOID)&This->lpVtblAR;
749 if (*ppv)
751 IUnknown_AddRef((IUnknown *)(*ppv));
752 return S_OK;
755 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
757 return E_NOINTERFACE;
760 static ULONG WINAPI FileAsyncReaderPin_Release(IPin * iface)
762 FileAsyncReader *This = (FileAsyncReader *)iface;
763 ULONG refCount = InterlockedDecrement(&This->pin.pin.refCount);
765 TRACE("(%p)->() Release from %d\n", This, refCount + 1);
767 if (!refCount)
769 DATAREQUEST * pCurrent;
770 DATAREQUEST * pNext;
771 for (pCurrent = This->pHead; pCurrent; pCurrent = pNext)
773 pNext = pCurrent->pNext;
774 CoTaskMemFree(pCurrent);
776 CloseHandle(This->hFile);
777 CloseHandle(This->hEvent);
778 This->csList.DebugInfo->Spare[0] = 0;
779 DeleteCriticalSection(&This->csList);
780 CoTaskMemFree(This);
781 return 0;
783 return refCount;
786 static HRESULT WINAPI FileAsyncReaderPin_EnumMediaTypes(IPin * iface, IEnumMediaTypes ** ppEnum)
788 ENUMMEDIADETAILS emd;
789 FileAsyncReader *This = (FileAsyncReader *)iface;
791 TRACE("(%p)\n", ppEnum);
793 emd.cMediaTypes = 1;
794 emd.pMediaTypes = ((AsyncReader *)This->pin.pin.pinInfo.pFilter)->pmt;
796 return IEnumMediaTypesImpl_Construct(&emd, ppEnum);
799 static const IPinVtbl FileAsyncReaderPin_Vtbl =
801 FileAsyncReaderPin_QueryInterface,
802 IPinImpl_AddRef,
803 FileAsyncReaderPin_Release,
804 OutputPin_Connect,
805 OutputPin_ReceiveConnection,
806 IPinImpl_Disconnect,
807 IPinImpl_ConnectedTo,
808 IPinImpl_ConnectionMediaType,
809 IPinImpl_QueryPinInfo,
810 IPinImpl_QueryDirection,
811 IPinImpl_QueryId,
812 IPinImpl_QueryAccept,
813 FileAsyncReaderPin_EnumMediaTypes,
814 IPinImpl_QueryInternalConnections,
815 OutputPin_EndOfStream,
816 OutputPin_BeginFlush,
817 OutputPin_EndFlush,
818 OutputPin_NewSegment
821 /* Function called as a helper to IPin_Connect */
822 /* specific AM_MEDIA_TYPE - it cannot be NULL */
823 /* this differs from standard OutputPin_ConnectSpecific only in that it
824 * doesn't need the IMemInputPin interface on the receiving pin */
825 static HRESULT FileAsyncReaderPin_ConnectSpecific(IPin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
827 OutputPin *This = (OutputPin *)iface;
828 HRESULT hr;
830 TRACE("(%p, %p)\n", pReceivePin, pmt);
831 dump_AM_MEDIA_TYPE(pmt);
833 /* FIXME: call queryacceptproc */
835 This->pin.pConnectedTo = pReceivePin;
836 IPin_AddRef(pReceivePin);
837 CopyMediaType(&This->pin.mtCurrent, pmt);
839 hr = IPin_ReceiveConnection(pReceivePin, iface, pmt);
841 if (FAILED(hr))
843 IPin_Release(This->pin.pConnectedTo);
844 This->pin.pConnectedTo = NULL;
845 FreeMediaType(&This->pin.mtCurrent);
848 TRACE(" -- %x\n", hr);
849 return hr;
852 static HRESULT FileAsyncReader_Construct(HANDLE hFile, IBaseFilter * pBaseFilter, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
854 FileAsyncReader * pPinImpl;
855 PIN_INFO piOutput;
857 *ppPin = NULL;
859 pPinImpl = CoTaskMemAlloc(sizeof(*pPinImpl));
861 if (!pPinImpl)
862 return E_OUTOFMEMORY;
864 piOutput.dir = PINDIR_OUTPUT;
865 piOutput.pFilter = pBaseFilter;
866 strcpyW(piOutput.achName, wszOutputPinName);
868 if (SUCCEEDED(OutputPin_Init(&piOutput, NULL, pBaseFilter, AcceptProcAFR, pCritSec, &pPinImpl->pin)))
870 pPinImpl->pin.pin.lpVtbl = &FileAsyncReaderPin_Vtbl;
871 pPinImpl->lpVtblAR = &FileAsyncReader_Vtbl;
872 pPinImpl->hFile = hFile;
873 pPinImpl->hEvent = CreateEventW(NULL, 0, 0, NULL);
874 pPinImpl->bFlushing = FALSE;
875 pPinImpl->pHead = NULL;
876 pPinImpl->pin.pConnectSpecific = FileAsyncReaderPin_ConnectSpecific;
877 InitializeCriticalSection(&pPinImpl->csList);
878 pPinImpl->csList.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": FileAsyncReader.csList");
880 *ppPin = (IPin *)(&pPinImpl->pin.pin.lpVtbl);
881 return S_OK;
884 CoTaskMemFree(pPinImpl);
885 return E_FAIL;
888 /* IAsyncReader */
890 static HRESULT WINAPI FileAsyncReader_QueryInterface(IAsyncReader * iface, REFIID riid, LPVOID * ppv)
892 FileAsyncReader *This = impl_from_IAsyncReader(iface);
894 return IPin_QueryInterface((IPin *)This, riid, ppv);
897 static ULONG WINAPI FileAsyncReader_AddRef(IAsyncReader * iface)
899 FileAsyncReader *This = impl_from_IAsyncReader(iface);
901 return IPin_AddRef((IPin *)This);
904 static ULONG WINAPI FileAsyncReader_Release(IAsyncReader * iface)
906 FileAsyncReader *This = impl_from_IAsyncReader(iface);
908 return IPin_Release((IPin *)This);
911 #define DEF_ALIGNMENT 1
913 static HRESULT WINAPI FileAsyncReader_RequestAllocator(IAsyncReader * iface, IMemAllocator * pPreferred, ALLOCATOR_PROPERTIES * pProps, IMemAllocator ** ppActual)
915 HRESULT hr = S_OK;
917 TRACE("(%p, %p, %p)\n", pPreferred, pProps, ppActual);
919 if (!pProps->cbAlign || (pProps->cbAlign % DEF_ALIGNMENT) != 0)
920 pProps->cbAlign = DEF_ALIGNMENT;
922 if (pPreferred)
924 ALLOCATOR_PROPERTIES PropsActual;
925 hr = IMemAllocator_SetProperties(pPreferred, pProps, &PropsActual);
926 /* FIXME: check we are still aligned */
927 if (SUCCEEDED(hr))
929 IMemAllocator_AddRef(pPreferred);
930 *ppActual = pPreferred;
931 TRACE("FileAsyncReader_RequestAllocator -- %x\n", hr);
932 return S_OK;
936 pPreferred = NULL;
938 hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC, &IID_IMemAllocator, (LPVOID *)&pPreferred);
940 if (SUCCEEDED(hr))
942 ALLOCATOR_PROPERTIES PropsActual;
943 hr = IMemAllocator_SetProperties(pPreferred, pProps, &PropsActual);
944 /* FIXME: check we are still aligned */
945 if (SUCCEEDED(hr))
947 *ppActual = pPreferred;
948 TRACE("FileAsyncReader_RequestAllocator -- %x\n", hr);
949 return S_OK;
953 if (FAILED(hr))
955 *ppActual = NULL;
956 if (pPreferred)
957 IMemAllocator_Release(pPreferred);
960 TRACE("-- %x\n", hr);
961 return hr;
964 /* we could improve the Request/WaitForNext mechanism by allowing out of order samples.
965 * however, this would be quite complicated to do and may be a bit error prone */
966 static HRESULT WINAPI FileAsyncReader_Request(IAsyncReader * iface, IMediaSample * pSample, DWORD_PTR dwUser)
968 REFERENCE_TIME Start;
969 REFERENCE_TIME Stop;
970 DATAREQUEST * pDataRq;
971 BYTE * pBuffer;
972 HRESULT hr = S_OK;
973 FileAsyncReader *This = impl_from_IAsyncReader(iface);
975 TRACE("(%p, %lx)\n", pSample, dwUser);
977 /* check flushing state */
978 if (This->bFlushing)
979 return VFW_E_WRONG_STATE;
981 if (!(pDataRq = CoTaskMemAlloc(sizeof(*pDataRq))))
982 hr = E_OUTOFMEMORY;
984 /* get start and stop positions in bytes */
985 if (SUCCEEDED(hr))
986 hr = IMediaSample_GetTime(pSample, &Start, &Stop);
988 if (SUCCEEDED(hr))
989 hr = IMediaSample_GetPointer(pSample, &pBuffer);
991 if (SUCCEEDED(hr))
993 DWORD dwLength = (DWORD) BYTES_FROM_MEDIATIME(Stop - Start);
995 pDataRq->ovl.u.s.Offset = (DWORD) BYTES_FROM_MEDIATIME(Start);
996 pDataRq->ovl.u.s.OffsetHigh = (DWORD)(BYTES_FROM_MEDIATIME(Start) >> (sizeof(DWORD) * 8));
997 pDataRq->ovl.hEvent = This->hEvent;
998 pDataRq->dwUserData = dwUser;
999 pDataRq->pNext = NULL;
1000 /* we violate traditional COM rules here by maintaining
1001 * a reference to the sample, but not calling AddRef, but
1002 * that's what MSDN says to do */
1003 pDataRq->pSample = pSample;
1005 EnterCriticalSection(&This->csList);
1007 if (This->pHead)
1008 /* adds data request to end of list */
1009 queue(This->pHead, pDataRq);
1010 else
1011 This->pHead = pDataRq;
1013 LeaveCriticalSection(&This->csList);
1015 /* this is definitely not how it is implemented on Win9x
1016 * as they do not support async reads on files, but it is
1017 * sooo much easier to use this than messing around with threads!
1019 if (!ReadFile(This->hFile, pBuffer, dwLength, NULL, &pDataRq->ovl))
1020 hr = HRESULT_FROM_WIN32(GetLastError());
1022 /* ERROR_IO_PENDING is not actually an error since this is what we want! */
1023 if (hr == HRESULT_FROM_WIN32(ERROR_IO_PENDING))
1024 hr = S_OK;
1027 if (FAILED(hr) && pDataRq)
1029 EnterCriticalSection(&This->csList);
1031 DATAREQUEST * pCurrent;
1032 for (pCurrent = This->pHead; pCurrent && pCurrent->pNext; pCurrent = pCurrent->pNext)
1033 if (pCurrent->pNext == pDataRq)
1035 pCurrent->pNext = pDataRq->pNext;
1036 break;
1039 LeaveCriticalSection(&This->csList);
1040 CoTaskMemFree(pDataRq);
1043 TRACE("-- %x\n", hr);
1044 return hr;
1047 static HRESULT WINAPI FileAsyncReader_WaitForNext(IAsyncReader * iface, DWORD dwTimeout, IMediaSample ** ppSample, DWORD_PTR * pdwUser)
1049 HRESULT hr = S_OK;
1050 DWORD dwBytes = 0;
1051 DATAREQUEST * pDataRq = NULL;
1052 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1054 TRACE("(%u, %p, %p)\n", dwTimeout, ppSample, pdwUser);
1056 /* FIXME: we could do with improving this by waiting for an array of event handles
1057 * and then determining which one finished and removing that from the list, otherwise
1058 * we will end up waiting for longer than we should do, if a later request finishes
1059 * before an earlier one */
1061 *ppSample = NULL;
1062 *pdwUser = 0;
1064 /* we return immediately if flushing */
1065 if (This->bFlushing)
1066 hr = VFW_E_WRONG_STATE;
1068 if (SUCCEEDED(hr))
1070 /* wait for the read to finish or timeout */
1071 if (WaitForSingleObject(This->hEvent, dwTimeout) == WAIT_TIMEOUT)
1072 hr = VFW_E_TIMEOUT;
1074 if (SUCCEEDED(hr))
1076 EnterCriticalSection(&This->csList);
1078 pDataRq = This->pHead;
1079 if (pDataRq == NULL)
1080 hr = E_FAIL;
1081 else
1082 This->pHead = pDataRq->pNext;
1084 LeaveCriticalSection(&This->csList);
1087 if (SUCCEEDED(hr))
1089 /* get any errors */
1090 if (!GetOverlappedResult(This->hFile, &pDataRq->ovl, &dwBytes, FALSE))
1091 hr = HRESULT_FROM_WIN32(GetLastError());
1094 if (SUCCEEDED(hr))
1096 IMediaSample_SetActualDataLength(pDataRq->pSample, dwBytes);
1097 *ppSample = pDataRq->pSample;
1098 *pdwUser = pDataRq->dwUserData;
1101 /* no need to close event handle since we will close it when the pin is destroyed */
1102 CoTaskMemFree(pDataRq);
1104 TRACE("-- %x\n", hr);
1105 return hr;
1108 static HRESULT WINAPI FileAsyncReader_SyncRead(IAsyncReader * iface, LONGLONG llPosition, LONG lLength, BYTE * pBuffer);
1110 static HRESULT WINAPI FileAsyncReader_SyncReadAligned(IAsyncReader * iface, IMediaSample * pSample)
1112 BYTE * pBuffer;
1113 REFERENCE_TIME tStart;
1114 REFERENCE_TIME tStop;
1115 HRESULT hr;
1117 TRACE("(%p)\n", pSample);
1119 hr = IMediaSample_GetTime(pSample, &tStart, &tStop);
1121 if (SUCCEEDED(hr))
1122 hr = IMediaSample_GetPointer(pSample, &pBuffer);
1124 if (SUCCEEDED(hr))
1125 hr = FileAsyncReader_SyncRead(iface,
1126 BYTES_FROM_MEDIATIME(tStart),
1127 (LONG) BYTES_FROM_MEDIATIME(tStop - tStart),
1128 pBuffer);
1130 TRACE("-- %x\n", hr);
1131 return hr;
1134 static HRESULT WINAPI FileAsyncReader_SyncRead(IAsyncReader * iface, LONGLONG llPosition, LONG lLength, BYTE * pBuffer)
1136 OVERLAPPED ovl;
1137 HRESULT hr = S_OK;
1138 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1140 TRACE("(%x%08x, %d, %p)\n", (ULONG)(llPosition >> 32), (ULONG)llPosition, lLength, pBuffer);
1142 ZeroMemory(&ovl, sizeof(ovl));
1144 ovl.hEvent = CreateEventW(NULL, 0, 0, NULL);
1145 /* NOTE: llPosition is the actual byte position to start reading from */
1146 ovl.u.s.Offset = (DWORD) llPosition;
1147 ovl.u.s.OffsetHigh = (DWORD) (llPosition >> (sizeof(DWORD) * 8));
1149 if (!ReadFile(This->hFile, pBuffer, lLength, NULL, &ovl))
1150 hr = HRESULT_FROM_WIN32(GetLastError());
1152 if (hr == HRESULT_FROM_WIN32(ERROR_IO_PENDING))
1153 hr = S_OK;
1155 if (SUCCEEDED(hr))
1157 DWORD dwBytesRead;
1159 if (!GetOverlappedResult(This->hFile, &ovl, &dwBytesRead, TRUE))
1160 hr = HRESULT_FROM_WIN32(GetLastError());
1163 CloseHandle(ovl.hEvent);
1165 TRACE("-- %x\n", hr);
1166 return hr;
1169 static HRESULT WINAPI FileAsyncReader_Length(IAsyncReader * iface, LONGLONG * pTotal, LONGLONG * pAvailable)
1171 DWORD dwSizeLow;
1172 DWORD dwSizeHigh;
1173 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1175 TRACE("(%p, %p)\n", pTotal, pAvailable);
1177 if (((dwSizeLow = GetFileSize(This->hFile, &dwSizeHigh)) == -1) &&
1178 (GetLastError() != NO_ERROR))
1179 return HRESULT_FROM_WIN32(GetLastError());
1181 *pTotal = (LONGLONG)dwSizeLow | (LONGLONG)dwSizeHigh << (sizeof(DWORD) * 8);
1183 *pAvailable = *pTotal;
1185 return S_OK;
1188 static HRESULT WINAPI FileAsyncReader_BeginFlush(IAsyncReader * iface)
1190 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1192 TRACE("()\n");
1194 This->bFlushing = TRUE;
1195 CancelIo(This->hFile);
1196 SetEvent(This->hEvent);
1198 /* FIXME: free list */
1200 return S_OK;
1203 static HRESULT WINAPI FileAsyncReader_EndFlush(IAsyncReader * iface)
1205 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1207 TRACE("()\n");
1209 This->bFlushing = FALSE;
1211 return S_OK;
1214 static const IAsyncReaderVtbl FileAsyncReader_Vtbl =
1216 FileAsyncReader_QueryInterface,
1217 FileAsyncReader_AddRef,
1218 FileAsyncReader_Release,
1219 FileAsyncReader_RequestAllocator,
1220 FileAsyncReader_Request,
1221 FileAsyncReader_WaitForNext,
1222 FileAsyncReader_SyncReadAligned,
1223 FileAsyncReader_SyncRead,
1224 FileAsyncReader_Length,
1225 FileAsyncReader_BeginFlush,
1226 FileAsyncReader_EndFlush,