[CABMAN]
[reactos.git] / reactos / dll / directx / quartz / filesource.c
1 /*
2 * File Source Filter
3 *
4 * Copyright 2003 Robert Shearman
5 *
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.
10 *
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.
15 *
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
19 */
20
21 #define NONAMELESSUNION
22 #define NONAMELESSSTRUCT
23
24 #include "quartz_private.h"
25
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 <shlwapi.h>
34 #include <assert.h>
35
36 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
37
38 static const WCHAR wszOutputPinName[] = { 'O','u','t','p','u','t',0 };
39
40 typedef struct AsyncReader
41 {
42 BaseFilter filter;
43 IFileSourceFilter IFileSourceFilter_iface;
44 IAMFilterMiscFlags IAMFilterMiscFlags_iface;
45
46 IPin * pOutputPin;
47 LPOLESTR pszFileName;
48 AM_MEDIA_TYPE * pmt;
49 } AsyncReader;
50
51 static inline AsyncReader *impl_from_BaseFilter(BaseFilter *iface)
52 {
53 return CONTAINING_RECORD(iface, AsyncReader, filter);
54 }
55
56 static inline AsyncReader *impl_from_IBaseFilter(IBaseFilter *iface)
57 {
58 return CONTAINING_RECORD(iface, AsyncReader, filter.IBaseFilter_iface);
59 }
60
61 static inline AsyncReader *impl_from_IFileSourceFilter(IFileSourceFilter *iface)
62 {
63 return CONTAINING_RECORD(iface, AsyncReader, IFileSourceFilter_iface);
64 }
65
66 static inline AsyncReader *impl_from_IAMFilterMiscFlags(IAMFilterMiscFlags *iface)
67 {
68 return CONTAINING_RECORD(iface, AsyncReader, IAMFilterMiscFlags_iface);
69 }
70
71 static const IBaseFilterVtbl AsyncReader_Vtbl;
72 static const IFileSourceFilterVtbl FileSource_Vtbl;
73 static const IAsyncReaderVtbl FileAsyncReader_Vtbl;
74 static const IAMFilterMiscFlagsVtbl IAMFilterMiscFlags_Vtbl;
75
76 static HRESULT FileAsyncReader_Construct(HANDLE hFile, IBaseFilter * pBaseFilter, LPCRITICAL_SECTION pCritSec, IPin ** ppPin);
77
78 static const WCHAR mediatype_name[] = {
79 'M', 'e', 'd', 'i', 'a', ' ', 'T', 'y', 'p', 'e', 0 };
80 static const WCHAR subtype_name[] = {
81 'S', 'u', 'b', 't', 'y', 'p', 'e', 0 };
82 static const WCHAR source_filter_name[] = {
83 'S','o','u','r','c','e',' ','F','i','l','t','e','r',0};
84
85 static HRESULT process_extensions(HKEY hkeyExtensions, LPCOLESTR pszFileName, GUID * majorType, GUID * minorType, GUID * sourceFilter)
86 {
87 WCHAR *extension;
88 LONG l;
89 HKEY hsub;
90 WCHAR keying[39];
91 DWORD size;
92
93 if (!pszFileName)
94 return E_POINTER;
95
96 /* Get the part of the name that matters */
97 extension = PathFindExtensionW(pszFileName);
98 if (*extension != '.')
99 return E_FAIL;
100
101 l = RegOpenKeyExW(hkeyExtensions, extension, 0, KEY_READ, &hsub);
102 if (l)
103 return E_FAIL;
104
105 if (majorType)
106 {
107 size = sizeof(keying);
108 l = RegQueryValueExW(hsub, mediatype_name, NULL, NULL, (LPBYTE)keying, &size);
109 if (!l)
110 CLSIDFromString(keying, majorType);
111 }
112
113 if (minorType)
114 {
115 size = sizeof(keying);
116 if (!l)
117 l = RegQueryValueExW(hsub, subtype_name, NULL, NULL, (LPBYTE)keying, &size);
118 if (!l)
119 CLSIDFromString(keying, minorType);
120 }
121
122 if (sourceFilter)
123 {
124 size = sizeof(keying);
125 if (!l)
126 l = RegQueryValueExW(hsub, source_filter_name, NULL, NULL, (LPBYTE)keying, &size);
127 if (!l)
128 CLSIDFromString(keying, sourceFilter);
129 }
130
131 RegCloseKey(hsub);
132
133 if (!l)
134 return S_OK;
135 return E_FAIL;
136 }
137
138 static unsigned char byte_from_hex_char(WCHAR wHex)
139 {
140 switch (tolowerW(wHex))
141 {
142 case '0':
143 case '1':
144 case '2':
145 case '3':
146 case '4':
147 case '5':
148 case '6':
149 case '7':
150 case '8':
151 case '9':
152 return (wHex - '0') & 0xf;
153 case 'a':
154 case 'b':
155 case 'c':
156 case 'd':
157 case 'e':
158 case 'f':
159 return (wHex - 'a' + 10) & 0xf;
160 default:
161 return 0;
162 }
163 }
164
165 static HRESULT process_pattern_string(LPCWSTR wszPatternString, IAsyncReader * pReader)
166 {
167 ULONG ulOffset;
168 ULONG ulBytes;
169 BYTE * pbMask;
170 BYTE * pbValue;
171 BYTE * pbFile;
172 HRESULT hr = S_OK;
173 ULONG strpos;
174
175 TRACE("\t\tPattern string: %s\n", debugstr_w(wszPatternString));
176
177 /* format: "offset, bytestocompare, mask, value" */
178
179 ulOffset = strtolW(wszPatternString, NULL, 10);
180
181 if (!(wszPatternString = strchrW(wszPatternString, ',')))
182 return E_INVALIDARG;
183
184 wszPatternString++; /* skip ',' */
185
186 ulBytes = strtolW(wszPatternString, NULL, 10);
187
188 pbMask = HeapAlloc(GetProcessHeap(), 0, ulBytes);
189 pbValue = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, ulBytes);
190 pbFile = HeapAlloc(GetProcessHeap(), 0, ulBytes);
191
192 /* default mask is match everything */
193 memset(pbMask, 0xFF, ulBytes);
194
195 if (!(wszPatternString = strchrW(wszPatternString, ',')))
196 hr = E_INVALIDARG;
197
198 if (hr == S_OK)
199 {
200 wszPatternString++; /* skip ',' */
201 while (!isxdigitW(*wszPatternString) && (*wszPatternString != ',')) wszPatternString++;
202
203 for (strpos = 0; isxdigitW(*wszPatternString) && (strpos/2 < ulBytes); wszPatternString++, strpos++)
204 {
205 if ((strpos % 2) == 1) /* odd numbered position */
206 pbMask[strpos / 2] |= byte_from_hex_char(*wszPatternString);
207 else
208 pbMask[strpos / 2] = byte_from_hex_char(*wszPatternString) << 4;
209 }
210
211 if (!(wszPatternString = strchrW(wszPatternString, ',')))
212 hr = E_INVALIDARG;
213 else
214 wszPatternString++; /* skip ',' */
215 }
216
217 if (hr == S_OK)
218 {
219 for ( ; !isxdigitW(*wszPatternString) && (*wszPatternString != ','); wszPatternString++)
220 ;
221
222 for (strpos = 0; isxdigitW(*wszPatternString) && (strpos/2 < ulBytes); wszPatternString++, strpos++)
223 {
224 if ((strpos % 2) == 1) /* odd numbered position */
225 pbValue[strpos / 2] |= byte_from_hex_char(*wszPatternString);
226 else
227 pbValue[strpos / 2] = byte_from_hex_char(*wszPatternString) << 4;
228 }
229 }
230
231 if (hr == S_OK)
232 hr = IAsyncReader_SyncRead(pReader, ulOffset, ulBytes, pbFile);
233
234 if (hr == S_OK)
235 {
236 ULONG i;
237 for (i = 0; i < ulBytes; i++)
238 if ((pbFile[i] & pbMask[i]) != pbValue[i])
239 {
240 hr = S_FALSE;
241 break;
242 }
243 }
244
245 HeapFree(GetProcessHeap(), 0, pbMask);
246 HeapFree(GetProcessHeap(), 0, pbValue);
247 HeapFree(GetProcessHeap(), 0, pbFile);
248
249 /* if we encountered no errors with this string, and there is a following tuple, then we
250 * have to match that as well to succeed */
251 if ((hr == S_OK) && (wszPatternString = strchrW(wszPatternString, ',')))
252 return process_pattern_string(wszPatternString + 1, pReader);
253 else
254 return hr;
255 }
256
257 HRESULT GetClassMediaFile(IAsyncReader * pReader, LPCOLESTR pszFileName, GUID * majorType, GUID * minorType, GUID * sourceFilter)
258 {
259 HKEY hkeyMediaType = NULL;
260 LONG lRet;
261 HRESULT hr = S_OK;
262 BOOL bFound = FALSE;
263 static const WCHAR wszMediaType[] = {'M','e','d','i','a',' ','T','y','p','e',0};
264
265 TRACE("(%p, %s, %p, %p)\n", pReader, debugstr_w(pszFileName), majorType, minorType);
266
267 if(majorType)
268 *majorType = GUID_NULL;
269 if(minorType)
270 *minorType = GUID_NULL;
271 if(sourceFilter)
272 *sourceFilter = GUID_NULL;
273
274 lRet = RegOpenKeyExW(HKEY_CLASSES_ROOT, wszMediaType, 0, KEY_READ, &hkeyMediaType);
275 hr = HRESULT_FROM_WIN32(lRet);
276
277 if (SUCCEEDED(hr))
278 {
279 DWORD indexMajor;
280
281 for (indexMajor = 0; !bFound; indexMajor++)
282 {
283 HKEY hkeyMajor;
284 WCHAR wszMajorKeyName[CHARS_IN_GUID];
285 DWORD dwKeyNameLength = sizeof(wszMajorKeyName) / sizeof(wszMajorKeyName[0]);
286 static const WCHAR wszExtensions[] = {'E','x','t','e','n','s','i','o','n','s',0};
287
288 if (RegEnumKeyExW(hkeyMediaType, indexMajor, wszMajorKeyName, &dwKeyNameLength, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
289 break;
290 if (RegOpenKeyExW(hkeyMediaType, wszMajorKeyName, 0, KEY_READ, &hkeyMajor) != ERROR_SUCCESS)
291 break;
292 TRACE("%s\n", debugstr_w(wszMajorKeyName));
293 if (!strcmpW(wszExtensions, wszMajorKeyName))
294 {
295 if (process_extensions(hkeyMajor, pszFileName, majorType, minorType, sourceFilter) == S_OK)
296 bFound = TRUE;
297 }
298 /* We need a reader interface to check bytes */
299 else if (pReader)
300 {
301 DWORD indexMinor;
302
303 for (indexMinor = 0; !bFound; indexMinor++)
304 {
305 HKEY hkeyMinor;
306 WCHAR wszMinorKeyName[CHARS_IN_GUID];
307 DWORD dwMinorKeyNameLen = sizeof(wszMinorKeyName) / sizeof(wszMinorKeyName[0]);
308 WCHAR wszSourceFilterKeyName[CHARS_IN_GUID];
309 DWORD dwSourceFilterKeyNameLen = sizeof(wszSourceFilterKeyName);
310 DWORD maxValueLen;
311 DWORD indexValue;
312
313 if (RegEnumKeyExW(hkeyMajor, indexMinor, wszMinorKeyName, &dwMinorKeyNameLen, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
314 break;
315
316 if (RegOpenKeyExW(hkeyMajor, wszMinorKeyName, 0, KEY_READ, &hkeyMinor) != ERROR_SUCCESS)
317 break;
318
319 TRACE("\t%s\n", debugstr_w(wszMinorKeyName));
320
321 if (RegQueryInfoKeyW(hkeyMinor, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, &maxValueLen, NULL, NULL) != ERROR_SUCCESS)
322 break;
323
324 for (indexValue = 0; !bFound; indexValue++)
325 {
326 DWORD dwType;
327 WCHAR wszValueName[14]; /* longest name we should encounter will be "Source Filter" */
328 LPWSTR wszPatternString = HeapAlloc(GetProcessHeap(), 0, maxValueLen);
329 DWORD dwValueNameLen = sizeof(wszValueName) / sizeof(wszValueName[0]); /* remember this is in chars */
330 DWORD dwDataLen = maxValueLen; /* remember this is in bytes */
331
332 if (RegEnumValueW(hkeyMinor, indexValue, wszValueName, &dwValueNameLen, NULL, &dwType, (LPBYTE)wszPatternString, &dwDataLen) != ERROR_SUCCESS)
333 {
334 HeapFree(GetProcessHeap(), 0, wszPatternString);
335 break;
336 }
337
338 if (strcmpW(wszValueName, source_filter_name)==0) {
339 HeapFree(GetProcessHeap(), 0, wszPatternString);
340 continue;
341 }
342
343 /* if it is not the source filter value */
344 if (process_pattern_string(wszPatternString, pReader) == S_OK)
345 {
346 HeapFree(GetProcessHeap(), 0, wszPatternString);
347 if (majorType && FAILED(CLSIDFromString(wszMajorKeyName, majorType)))
348 break;
349 if (minorType && FAILED(CLSIDFromString(wszMinorKeyName, minorType)))
350 break;
351 if (sourceFilter)
352 {
353 /* Look up the source filter key */
354 if (RegQueryValueExW(hkeyMinor, source_filter_name, NULL, NULL, (LPBYTE)wszSourceFilterKeyName, &dwSourceFilterKeyNameLen))
355 break;
356 if (FAILED(CLSIDFromString(wszSourceFilterKeyName, sourceFilter)))
357 break;
358 }
359 bFound = TRUE;
360 } else
361 HeapFree(GetProcessHeap(), 0, wszPatternString);
362 }
363 CloseHandle(hkeyMinor);
364 }
365 }
366 CloseHandle(hkeyMajor);
367 }
368 }
369 CloseHandle(hkeyMediaType);
370
371 if (SUCCEEDED(hr) && !bFound)
372 {
373 ERR("Media class not found\n");
374 hr = E_FAIL;
375 }
376 else if (bFound)
377 {
378 TRACE("Found file's class:\n");
379 if(majorType)
380 TRACE("\tmajor = %s\n", qzdebugstr_guid(majorType));
381 if(minorType)
382 TRACE("\tsubtype = %s\n", qzdebugstr_guid(minorType));
383 if(sourceFilter)
384 TRACE("\tsource filter = %s\n", qzdebugstr_guid(sourceFilter));
385 }
386
387 return hr;
388 }
389
390 static IPin* WINAPI AsyncReader_GetPin(BaseFilter *iface, int pos)
391 {
392 AsyncReader *This = impl_from_BaseFilter(iface);
393
394 if (pos >= 1 || !This->pOutputPin)
395 return NULL;
396
397 IPin_AddRef(This->pOutputPin);
398 return This->pOutputPin;
399 }
400
401 static LONG WINAPI AsyncReader_GetPinCount(BaseFilter *iface)
402 {
403 AsyncReader *This = impl_from_BaseFilter(iface);
404
405 if (!This->pOutputPin)
406 return 0;
407 else
408 return 1;
409 }
410
411 static const BaseFilterFuncTable BaseFuncTable = {
412 AsyncReader_GetPin,
413 AsyncReader_GetPinCount
414 };
415
416 HRESULT AsyncReader_create(IUnknown * pUnkOuter, LPVOID * ppv)
417 {
418 AsyncReader *pAsyncRead;
419
420 if( pUnkOuter )
421 return CLASS_E_NOAGGREGATION;
422
423 pAsyncRead = CoTaskMemAlloc(sizeof(AsyncReader));
424
425 if (!pAsyncRead)
426 return E_OUTOFMEMORY;
427
428 BaseFilter_Init(&pAsyncRead->filter, &AsyncReader_Vtbl, &CLSID_AsyncReader, (DWORD_PTR)(__FILE__ ": AsyncReader.csFilter"), &BaseFuncTable);
429
430 pAsyncRead->IFileSourceFilter_iface.lpVtbl = &FileSource_Vtbl;
431 pAsyncRead->IAMFilterMiscFlags_iface.lpVtbl = &IAMFilterMiscFlags_Vtbl;
432 pAsyncRead->pOutputPin = NULL;
433
434 pAsyncRead->pszFileName = NULL;
435 pAsyncRead->pmt = NULL;
436
437 *ppv = pAsyncRead;
438
439 TRACE("-- created at %p\n", pAsyncRead);
440
441 return S_OK;
442 }
443
444 /** IUnknown methods **/
445
446 static HRESULT WINAPI AsyncReader_QueryInterface(IBaseFilter * iface, REFIID riid, LPVOID * ppv)
447 {
448 AsyncReader *This = impl_from_IBaseFilter(iface);
449
450 TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
451
452 *ppv = NULL;
453
454 if (IsEqualIID(riid, &IID_IUnknown))
455 *ppv = This;
456 else if (IsEqualIID(riid, &IID_IPersist))
457 *ppv = This;
458 else if (IsEqualIID(riid, &IID_IMediaFilter))
459 *ppv = This;
460 else if (IsEqualIID(riid, &IID_IBaseFilter))
461 *ppv = This;
462 else if (IsEqualIID(riid, &IID_IFileSourceFilter))
463 *ppv = &This->IFileSourceFilter_iface;
464 else if (IsEqualIID(riid, &IID_IAMFilterMiscFlags))
465 *ppv = &This->IAMFilterMiscFlags_iface;
466
467 if (*ppv)
468 {
469 IUnknown_AddRef((IUnknown *)(*ppv));
470 return S_OK;
471 }
472
473 if (!IsEqualIID(riid, &IID_IPin) && !IsEqualIID(riid, &IID_IMediaSeeking) &&
474 !IsEqualIID(riid, &IID_IVideoWindow) && !IsEqualIID(riid, &IID_IBasicAudio))
475 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
476
477 return E_NOINTERFACE;
478 }
479
480 static ULONG WINAPI AsyncReader_Release(IBaseFilter * iface)
481 {
482 AsyncReader *This = impl_from_IBaseFilter(iface);
483 ULONG refCount = BaseFilterImpl_Release(iface);
484
485 TRACE("(%p)->() Release from %d\n", This, refCount + 1);
486
487 if (!refCount)
488 {
489 if (This->pOutputPin)
490 {
491 IPin *pConnectedTo;
492 if(SUCCEEDED(IPin_ConnectedTo(This->pOutputPin, &pConnectedTo)))
493 {
494 IPin_Disconnect(pConnectedTo);
495 IPin_Release(pConnectedTo);
496 }
497 IPin_Disconnect(This->pOutputPin);
498 IPin_Release(This->pOutputPin);
499 }
500 CoTaskMemFree(This->pszFileName);
501 if (This->pmt)
502 FreeMediaType(This->pmt);
503 CoTaskMemFree(This);
504 return 0;
505 }
506 else
507 return refCount;
508 }
509
510 /** IMediaFilter methods **/
511
512 static HRESULT WINAPI AsyncReader_Stop(IBaseFilter * iface)
513 {
514 AsyncReader *This = impl_from_IBaseFilter(iface);
515
516 TRACE("()\n");
517
518 This->filter.state = State_Stopped;
519
520 return S_OK;
521 }
522
523 static HRESULT WINAPI AsyncReader_Pause(IBaseFilter * iface)
524 {
525 AsyncReader *This = impl_from_IBaseFilter(iface);
526
527 TRACE("()\n");
528
529 This->filter.state = State_Paused;
530
531 return S_OK;
532 }
533
534 static HRESULT WINAPI AsyncReader_Run(IBaseFilter * iface, REFERENCE_TIME tStart)
535 {
536 AsyncReader *This = impl_from_IBaseFilter(iface);
537
538 TRACE("(%x%08x)\n", (ULONG)(tStart >> 32), (ULONG)tStart);
539
540 This->filter.state = State_Running;
541
542 return S_OK;
543 }
544
545 /** IBaseFilter methods **/
546
547 static HRESULT WINAPI AsyncReader_FindPin(IBaseFilter * iface, LPCWSTR Id, IPin **ppPin)
548 {
549 AsyncReader *This = impl_from_IBaseFilter(iface);
550 TRACE("(%s, %p)\n", debugstr_w(Id), ppPin);
551
552 if (!Id || !ppPin)
553 return E_POINTER;
554
555 if (strcmpW(Id, wszOutputPinName))
556 {
557 *ppPin = NULL;
558 return VFW_E_NOT_FOUND;
559 }
560
561 *ppPin = This->pOutputPin;
562 IPin_AddRef(*ppPin);
563 return S_OK;
564 }
565
566 static const IBaseFilterVtbl AsyncReader_Vtbl =
567 {
568 AsyncReader_QueryInterface,
569 BaseFilterImpl_AddRef,
570 AsyncReader_Release,
571 BaseFilterImpl_GetClassID,
572 AsyncReader_Stop,
573 AsyncReader_Pause,
574 AsyncReader_Run,
575 BaseFilterImpl_GetState,
576 BaseFilterImpl_SetSyncSource,
577 BaseFilterImpl_GetSyncSource,
578 BaseFilterImpl_EnumPins,
579 AsyncReader_FindPin,
580 BaseFilterImpl_QueryFilterInfo,
581 BaseFilterImpl_JoinFilterGraph,
582 BaseFilterImpl_QueryVendorInfo
583 };
584
585 static HRESULT WINAPI FileSource_QueryInterface(IFileSourceFilter * iface, REFIID riid, LPVOID * ppv)
586 {
587 AsyncReader *This = impl_from_IFileSourceFilter(iface);
588
589 return IBaseFilter_QueryInterface(&This->filter.IBaseFilter_iface, riid, ppv);
590 }
591
592 static ULONG WINAPI FileSource_AddRef(IFileSourceFilter * iface)
593 {
594 AsyncReader *This = impl_from_IFileSourceFilter(iface);
595
596 return IBaseFilter_AddRef(&This->filter.IBaseFilter_iface);
597 }
598
599 static ULONG WINAPI FileSource_Release(IFileSourceFilter * iface)
600 {
601 AsyncReader *This = impl_from_IFileSourceFilter(iface);
602
603 return IBaseFilter_Release(&This->filter.IBaseFilter_iface);
604 }
605
606 static HRESULT WINAPI FileSource_Load(IFileSourceFilter * iface, LPCOLESTR pszFileName, const AM_MEDIA_TYPE * pmt)
607 {
608 HRESULT hr;
609 HANDLE hFile;
610 IAsyncReader * pReader = NULL;
611 AsyncReader *This = impl_from_IFileSourceFilter(iface);
612
613 TRACE("(%s, %p)\n", debugstr_w(pszFileName), pmt);
614
615 /* open file */
616 /* FIXME: check the sharing values that native uses */
617 hFile = CreateFileW(pszFileName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL);
618
619 if (hFile == INVALID_HANDLE_VALUE)
620 {
621 return HRESULT_FROM_WIN32(GetLastError());
622 }
623
624 /* create pin */
625 hr = FileAsyncReader_Construct(hFile, &This->filter.IBaseFilter_iface, &This->filter.csFilter, &This->pOutputPin);
626 BaseFilterImpl_IncrementPinVersion(&This->filter);
627
628 if (SUCCEEDED(hr))
629 hr = IPin_QueryInterface(This->pOutputPin, &IID_IAsyncReader, (LPVOID *)&pReader);
630
631 /* store file name & media type */
632 if (SUCCEEDED(hr))
633 {
634 CoTaskMemFree(This->pszFileName);
635 if (This->pmt)
636 FreeMediaType(This->pmt);
637
638 This->pszFileName = CoTaskMemAlloc((strlenW(pszFileName) + 1) * sizeof(WCHAR));
639 strcpyW(This->pszFileName, pszFileName);
640
641 This->pmt = CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE));
642 if (!pmt)
643 {
644 This->pmt->bFixedSizeSamples = TRUE;
645 This->pmt->bTemporalCompression = FALSE;
646 This->pmt->cbFormat = 0;
647 This->pmt->pbFormat = NULL;
648 This->pmt->pUnk = NULL;
649 This->pmt->lSampleSize = 0;
650 This->pmt->formattype = FORMAT_None;
651 hr = GetClassMediaFile(pReader, pszFileName, &This->pmt->majortype, &This->pmt->subtype, NULL);
652 if (FAILED(hr))
653 {
654 CoTaskMemFree(This->pmt);
655 This->pmt = NULL;
656 }
657 }
658 else
659 CopyMediaType(This->pmt, pmt);
660 }
661
662 if (pReader)
663 IAsyncReader_Release(pReader);
664
665 if (FAILED(hr))
666 {
667 if (This->pOutputPin)
668 {
669 IPin_Release(This->pOutputPin);
670 This->pOutputPin = NULL;
671 }
672
673 CoTaskMemFree(This->pszFileName);
674 if (This->pmt)
675 FreeMediaType(This->pmt);
676 This->pszFileName = NULL;
677 This->pmt = NULL;
678
679 CloseHandle(hFile);
680 }
681
682 /* FIXME: check return codes */
683 return hr;
684 }
685
686 static HRESULT WINAPI FileSource_GetCurFile(IFileSourceFilter * iface, LPOLESTR * ppszFileName, AM_MEDIA_TYPE * pmt)
687 {
688 AsyncReader *This = impl_from_IFileSourceFilter(iface);
689
690 TRACE("(%p, %p)\n", ppszFileName, pmt);
691
692 if (!ppszFileName)
693 return E_POINTER;
694
695 /* copy file name & media type if available, otherwise clear the outputs */
696 if (This->pszFileName)
697 {
698 *ppszFileName = CoTaskMemAlloc((strlenW(This->pszFileName) + 1) * sizeof(WCHAR));
699 strcpyW(*ppszFileName, This->pszFileName);
700 }
701 else
702 *ppszFileName = NULL;
703
704 if (pmt)
705 {
706 if (This->pmt)
707 CopyMediaType(pmt, This->pmt);
708 else
709 ZeroMemory(pmt, sizeof(*pmt));
710 }
711
712 return S_OK;
713 }
714
715 static const IFileSourceFilterVtbl FileSource_Vtbl =
716 {
717 FileSource_QueryInterface,
718 FileSource_AddRef,
719 FileSource_Release,
720 FileSource_Load,
721 FileSource_GetCurFile
722 };
723
724
725 /* the dwUserData passed back to user */
726 typedef struct DATAREQUEST
727 {
728 IMediaSample * pSample; /* sample passed to us by user */
729 DWORD_PTR dwUserData; /* user data passed to us */
730 OVERLAPPED ovl; /* our overlapped structure */
731 } DATAREQUEST;
732
733 typedef struct FileAsyncReader
734 {
735 BaseOutputPin pin;
736 IAsyncReader IAsyncReader_iface;
737
738 ALLOCATOR_PROPERTIES allocProps;
739 HANDLE hFile;
740 BOOL bFlushing;
741 /* Why would you need more? Every sample has its own handle */
742 LONG queued_number;
743 LONG samples;
744 LONG oldest_sample;
745 CRITICAL_SECTION csList; /* critical section to prevent concurrency issues */
746 DATAREQUEST *sample_list;
747
748 /* Have a handle for every sample, and then one more as flushing handle */
749 HANDLE *handle_list;
750 } FileAsyncReader;
751
752 static inline FileAsyncReader *impl_from_IPin(IPin *iface)
753 {
754 return CONTAINING_RECORD(iface, FileAsyncReader, pin.pin.IPin_iface);
755 }
756
757 static inline FileAsyncReader *impl_from_BasePin(BasePin *iface)
758 {
759 return CONTAINING_RECORD(iface, FileAsyncReader, pin.pin);
760 }
761
762 static inline FileAsyncReader *impl_from_BaseOutputPin(BaseOutputPin *iface)
763 {
764 return CONTAINING_RECORD(iface, FileAsyncReader, pin);
765 }
766
767 static inline BaseOutputPin *impl_BaseOututPin_from_BasePin(BasePin *iface)
768 {
769 return CONTAINING_RECORD(iface, BaseOutputPin, pin);
770 }
771
772 static inline FileAsyncReader *impl_from_IAsyncReader(IAsyncReader *iface)
773 {
774 return CONTAINING_RECORD(iface, FileAsyncReader, IAsyncReader_iface);
775 }
776
777 static HRESULT WINAPI FileAsyncReaderPin_QueryAccept(IPin *iface, const AM_MEDIA_TYPE *pmt)
778 {
779 FileAsyncReader *This = impl_from_IPin(iface);
780 AM_MEDIA_TYPE *pmt_filter = impl_from_IBaseFilter(This->pin.pin.pinInfo.pFilter)->pmt;
781
782 FIXME("(%p, %p)\n", iface, pmt);
783
784 if (IsEqualGUID(&pmt->majortype, &pmt_filter->majortype) &&
785 IsEqualGUID(&pmt->subtype, &pmt_filter->subtype) &&
786 IsEqualGUID(&pmt->formattype, &FORMAT_None))
787 return S_OK;
788
789 return S_FALSE;
790 }
791
792 static HRESULT WINAPI FileAsyncReaderPin_GetMediaType(BasePin *iface, int iPosition, AM_MEDIA_TYPE *pmt)
793 {
794 FileAsyncReader *This = impl_from_BasePin(iface);
795 if (iPosition < 0)
796 return E_INVALIDARG;
797 if (iPosition > 0)
798 return VFW_S_NO_MORE_ITEMS;
799 CopyMediaType(pmt, impl_from_IBaseFilter(This->pin.pin.pinInfo.pFilter)->pmt);
800 return S_OK;
801 }
802
803 /* overridden pin functions */
804
805 static HRESULT WINAPI FileAsyncReaderPin_QueryInterface(IPin * iface, REFIID riid, LPVOID * ppv)
806 {
807 FileAsyncReader *This = impl_from_IPin(iface);
808 TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
809
810 *ppv = NULL;
811
812 if (IsEqualIID(riid, &IID_IUnknown))
813 *ppv = This;
814 else if (IsEqualIID(riid, &IID_IPin))
815 *ppv = This;
816 else if (IsEqualIID(riid, &IID_IAsyncReader))
817 *ppv = &This->IAsyncReader_iface;
818
819 if (*ppv)
820 {
821 IUnknown_AddRef((IUnknown *)(*ppv));
822 return S_OK;
823 }
824
825 if (!IsEqualIID(riid, &IID_IPin) && !IsEqualIID(riid, &IID_IMediaSeeking))
826 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
827
828 return E_NOINTERFACE;
829 }
830
831 static ULONG WINAPI FileAsyncReaderPin_Release(IPin * iface)
832 {
833 FileAsyncReader *This = impl_from_IPin(iface);
834 ULONG refCount = InterlockedDecrement(&This->pin.pin.refCount);
835 int x;
836
837 TRACE("(%p)->() Release from %d\n", This, refCount + 1);
838
839 if (!refCount)
840 {
841 CoTaskMemFree(This->sample_list);
842 if (This->handle_list)
843 {
844 for (x = 0; x <= This->samples; ++x)
845 CloseHandle(This->handle_list[x]);
846 CoTaskMemFree(This->handle_list);
847 }
848 CloseHandle(This->hFile);
849 This->csList.DebugInfo->Spare[0] = 0;
850 DeleteCriticalSection(&This->csList);
851 CoTaskMemFree(This);
852 return 0;
853 }
854 return refCount;
855 }
856
857 static const IPinVtbl FileAsyncReaderPin_Vtbl =
858 {
859 FileAsyncReaderPin_QueryInterface,
860 BasePinImpl_AddRef,
861 FileAsyncReaderPin_Release,
862 BaseOutputPinImpl_Connect,
863 BaseOutputPinImpl_ReceiveConnection,
864 BasePinImpl_Disconnect,
865 BasePinImpl_ConnectedTo,
866 BasePinImpl_ConnectionMediaType,
867 BasePinImpl_QueryPinInfo,
868 BasePinImpl_QueryDirection,
869 BasePinImpl_QueryId,
870 FileAsyncReaderPin_QueryAccept,
871 BasePinImpl_EnumMediaTypes,
872 BasePinImpl_QueryInternalConnections,
873 BaseOutputPinImpl_EndOfStream,
874 BaseOutputPinImpl_BeginFlush,
875 BaseOutputPinImpl_EndFlush,
876 BasePinImpl_NewSegment
877 };
878
879 /* Function called as a helper to IPin_Connect */
880 /* specific AM_MEDIA_TYPE - it cannot be NULL */
881 /* this differs from standard OutputPin_AttemptConnection only in that it
882 * doesn't need the IMemInputPin interface on the receiving pin */
883 static HRESULT WINAPI FileAsyncReaderPin_AttemptConnection(BasePin * iface, IPin * pReceivePin, const AM_MEDIA_TYPE * pmt)
884 {
885 BaseOutputPin *This = impl_BaseOututPin_from_BasePin(iface);
886 HRESULT hr;
887
888 TRACE("(%p, %p)\n", pReceivePin, pmt);
889 dump_AM_MEDIA_TYPE(pmt);
890
891 /* FIXME: call queryacceptproc */
892
893 This->pin.pConnectedTo = pReceivePin;
894 IPin_AddRef(pReceivePin);
895 CopyMediaType(&This->pin.mtCurrent, pmt);
896
897 hr = IPin_ReceiveConnection(pReceivePin, &iface->IPin_iface, pmt);
898
899 if (FAILED(hr))
900 {
901 IPin_Release(This->pin.pConnectedTo);
902 This->pin.pConnectedTo = NULL;
903 FreeMediaType(&This->pin.mtCurrent);
904 }
905
906 TRACE(" -- %x\n", hr);
907 return hr;
908 }
909
910 static HRESULT WINAPI FileAsyncReaderPin_DecideBufferSize(BaseOutputPin *iface, IMemAllocator *pAlloc, ALLOCATOR_PROPERTIES *ppropInputRequest)
911 {
912 FileAsyncReader *This = impl_from_BaseOutputPin(iface);
913 ALLOCATOR_PROPERTIES actual;
914
915 if (ppropInputRequest->cbAlign && ppropInputRequest->cbAlign != This->allocProps.cbAlign)
916 FIXME("Requested Buffer cbAlign mismatch %i,%i\n",This->allocProps.cbAlign, ppropInputRequest->cbAlign);
917 if (ppropInputRequest->cbPrefix)
918 FIXME("Requested Buffer cbPrefix mismatch %i,%i\n",This->allocProps.cbPrefix, ppropInputRequest->cbPrefix);
919 if (ppropInputRequest->cbBuffer)
920 FIXME("Requested Buffer cbBuffer mismatch %i,%i\n",This->allocProps.cbBuffer, ppropInputRequest->cbBuffer);
921 if (ppropInputRequest->cBuffers)
922 FIXME("Requested Buffer cBuffers mismatch %i,%i\n",This->allocProps.cBuffers, ppropInputRequest->cBuffers);
923
924 return IMemAllocator_SetProperties(pAlloc, &This->allocProps, &actual);
925 }
926
927 static const BasePinFuncTable output_BaseFuncTable = {
928 NULL,
929 FileAsyncReaderPin_AttemptConnection,
930 BasePinImpl_GetMediaTypeVersion,
931 FileAsyncReaderPin_GetMediaType
932 };
933
934 static const BaseOutputPinFuncTable output_BaseOutputFuncTable = {
935 FileAsyncReaderPin_DecideBufferSize,
936 BaseOutputPinImpl_DecideAllocator,
937 BaseOutputPinImpl_BreakConnect
938 };
939
940 static HRESULT FileAsyncReader_Construct(HANDLE hFile, IBaseFilter * pBaseFilter, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
941 {
942 PIN_INFO piOutput;
943 HRESULT hr;
944
945 *ppPin = NULL;
946 piOutput.dir = PINDIR_OUTPUT;
947 piOutput.pFilter = pBaseFilter;
948 strcpyW(piOutput.achName, wszOutputPinName);
949 hr = BaseOutputPin_Construct(&FileAsyncReaderPin_Vtbl, sizeof(FileAsyncReader), &piOutput, &output_BaseFuncTable, &output_BaseOutputFuncTable, pCritSec, ppPin);
950
951 if (SUCCEEDED(hr))
952 {
953 FileAsyncReader *pPinImpl = (FileAsyncReader *)*ppPin;
954 pPinImpl->IAsyncReader_iface.lpVtbl = &FileAsyncReader_Vtbl;
955 pPinImpl->hFile = hFile;
956 pPinImpl->bFlushing = FALSE;
957 pPinImpl->sample_list = NULL;
958 pPinImpl->handle_list = NULL;
959 pPinImpl->queued_number = 0;
960 InitializeCriticalSection(&pPinImpl->csList);
961 pPinImpl->csList.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": FileAsyncReader.csList");
962 }
963 return hr;
964 }
965
966 /* IAsyncReader */
967
968 static HRESULT WINAPI FileAsyncReader_QueryInterface(IAsyncReader * iface, REFIID riid, LPVOID * ppv)
969 {
970 FileAsyncReader *This = impl_from_IAsyncReader(iface);
971
972 return IPin_QueryInterface(&This->pin.pin.IPin_iface, riid, ppv);
973 }
974
975 static ULONG WINAPI FileAsyncReader_AddRef(IAsyncReader * iface)
976 {
977 FileAsyncReader *This = impl_from_IAsyncReader(iface);
978
979 return IPin_AddRef(&This->pin.pin.IPin_iface);
980 }
981
982 static ULONG WINAPI FileAsyncReader_Release(IAsyncReader * iface)
983 {
984 FileAsyncReader *This = impl_from_IAsyncReader(iface);
985
986 return IPin_Release(&This->pin.pin.IPin_iface);
987 }
988
989 #define DEF_ALIGNMENT 1
990
991 static HRESULT WINAPI FileAsyncReader_RequestAllocator(IAsyncReader * iface, IMemAllocator * pPreferred, ALLOCATOR_PROPERTIES * pProps, IMemAllocator ** ppActual)
992 {
993 FileAsyncReader *This = impl_from_IAsyncReader(iface);
994
995 HRESULT hr = S_OK;
996
997 TRACE("(%p, %p, %p)\n", pPreferred, pProps, ppActual);
998
999 if (!pProps->cbAlign || (pProps->cbAlign % DEF_ALIGNMENT) != 0)
1000 pProps->cbAlign = DEF_ALIGNMENT;
1001
1002 if (pPreferred)
1003 {
1004 hr = IMemAllocator_SetProperties(pPreferred, pProps, pProps);
1005 /* FIXME: check we are still aligned */
1006 if (SUCCEEDED(hr))
1007 {
1008 IMemAllocator_AddRef(pPreferred);
1009 *ppActual = pPreferred;
1010 TRACE("FileAsyncReader_RequestAllocator -- %x\n", hr);
1011 goto done;
1012 }
1013 }
1014
1015 pPreferred = NULL;
1016
1017 hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC, &IID_IMemAllocator, (LPVOID *)&pPreferred);
1018
1019 if (SUCCEEDED(hr))
1020 {
1021 hr = IMemAllocator_SetProperties(pPreferred, pProps, pProps);
1022 /* FIXME: check we are still aligned */
1023 if (SUCCEEDED(hr))
1024 {
1025 *ppActual = pPreferred;
1026 TRACE("FileAsyncReader_RequestAllocator -- %x\n", hr);
1027 }
1028 }
1029
1030 done:
1031 if (SUCCEEDED(hr))
1032 {
1033 CoTaskMemFree(This->sample_list);
1034 if (This->handle_list)
1035 {
1036 int x;
1037 for (x = 0; x <= This->samples; ++x)
1038 CloseHandle(This->handle_list[x]);
1039 CoTaskMemFree(This->handle_list);
1040 }
1041
1042 This->samples = pProps->cBuffers;
1043 This->oldest_sample = 0;
1044 TRACE("Samples: %u\n", This->samples);
1045 This->sample_list = CoTaskMemAlloc(sizeof(This->sample_list[0]) * pProps->cBuffers);
1046 This->handle_list = CoTaskMemAlloc(sizeof(HANDLE) * pProps->cBuffers * 2);
1047
1048 if (This->sample_list && This->handle_list)
1049 {
1050 int x;
1051 ZeroMemory(This->sample_list, sizeof(This->sample_list[0]) * pProps->cBuffers);
1052 for (x = 0; x < This->samples; ++x)
1053 {
1054 This->sample_list[x].ovl.hEvent = This->handle_list[x] = CreateEventW(NULL, 0, 0, NULL);
1055 if (x + 1 < This->samples)
1056 This->handle_list[This->samples + 1 + x] = This->handle_list[x];
1057 }
1058 This->handle_list[This->samples] = CreateEventW(NULL, 1, 0, NULL);
1059 This->allocProps = *pProps;
1060 }
1061 else
1062 {
1063 hr = E_OUTOFMEMORY;
1064 CoTaskMemFree(This->sample_list);
1065 CoTaskMemFree(This->handle_list);
1066 This->samples = 0;
1067 This->sample_list = NULL;
1068 This->handle_list = NULL;
1069 }
1070 }
1071
1072 if (FAILED(hr))
1073 {
1074 *ppActual = NULL;
1075 if (pPreferred)
1076 IMemAllocator_Release(pPreferred);
1077 }
1078
1079 TRACE("-- %x\n", hr);
1080 return hr;
1081 }
1082
1083 /* we could improve the Request/WaitForNext mechanism by allowing out of order samples.
1084 * however, this would be quite complicated to do and may be a bit error prone */
1085 static HRESULT WINAPI FileAsyncReader_Request(IAsyncReader * iface, IMediaSample * pSample, DWORD_PTR dwUser)
1086 {
1087 HRESULT hr = S_OK;
1088 REFERENCE_TIME Start;
1089 REFERENCE_TIME Stop;
1090 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1091 LPBYTE pBuffer = NULL;
1092
1093 TRACE("(%p, %lx)\n", pSample, dwUser);
1094
1095 if (!pSample)
1096 return E_POINTER;
1097
1098 /* get start and stop positions in bytes */
1099 if (SUCCEEDED(hr))
1100 hr = IMediaSample_GetTime(pSample, &Start, &Stop);
1101
1102 if (SUCCEEDED(hr))
1103 hr = IMediaSample_GetPointer(pSample, &pBuffer);
1104
1105 EnterCriticalSection(&This->csList);
1106 if (This->bFlushing)
1107 {
1108 LeaveCriticalSection(&This->csList);
1109 return VFW_E_WRONG_STATE;
1110 }
1111
1112 if (SUCCEEDED(hr))
1113 {
1114 DWORD dwLength = (DWORD) BYTES_FROM_MEDIATIME(Stop - Start);
1115 DATAREQUEST *pDataRq;
1116 int x;
1117
1118 /* Try to insert above the waiting sample if possible */
1119 for (x = This->oldest_sample; x < This->samples; ++x)
1120 {
1121 if (!This->sample_list[x].pSample)
1122 break;
1123 }
1124
1125 if (x >= This->samples)
1126 for (x = 0; x < This->oldest_sample; ++x)
1127 {
1128 if (!This->sample_list[x].pSample)
1129 break;
1130 }
1131
1132 /* There must be a sample we have found */
1133 assert(x < This->samples);
1134 ++This->queued_number;
1135
1136 pDataRq = This->sample_list + x;
1137
1138 pDataRq->ovl.u.s.Offset = (DWORD) BYTES_FROM_MEDIATIME(Start);
1139 pDataRq->ovl.u.s.OffsetHigh = (DWORD)(BYTES_FROM_MEDIATIME(Start) >> (sizeof(DWORD) * 8));
1140 pDataRq->dwUserData = dwUser;
1141
1142 /* we violate traditional COM rules here by maintaining
1143 * a reference to the sample, but not calling AddRef, but
1144 * that's what MSDN says to do */
1145 pDataRq->pSample = pSample;
1146
1147 /* this is definitely not how it is implemented on Win9x
1148 * as they do not support async reads on files, but it is
1149 * sooo much easier to use this than messing around with threads!
1150 */
1151 if (!ReadFile(This->hFile, pBuffer, dwLength, NULL, &pDataRq->ovl))
1152 hr = HRESULT_FROM_WIN32(GetLastError());
1153
1154 /* ERROR_IO_PENDING is not actually an error since this is what we want! */
1155 if (hr == HRESULT_FROM_WIN32(ERROR_IO_PENDING))
1156 hr = S_OK;
1157 }
1158
1159 LeaveCriticalSection(&This->csList);
1160
1161 TRACE("-- %x\n", hr);
1162 return hr;
1163 }
1164
1165 static HRESULT WINAPI FileAsyncReader_WaitForNext(IAsyncReader * iface, DWORD dwTimeout, IMediaSample ** ppSample, DWORD_PTR * pdwUser)
1166 {
1167 HRESULT hr = S_OK;
1168 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1169 DWORD buffer = ~0;
1170
1171 TRACE("(%u, %p, %p)\n", dwTimeout, ppSample, pdwUser);
1172
1173 *ppSample = NULL;
1174 *pdwUser = 0;
1175
1176 EnterCriticalSection(&This->csList);
1177 if (!This->bFlushing)
1178 {
1179 LONG oldest = This->oldest_sample;
1180
1181 if (!This->queued_number)
1182 {
1183 /* It could be that nothing is queued right now, but that can be fixed */
1184 WARN("Called without samples in queue and not flushing!!\n");
1185 }
1186 LeaveCriticalSection(&This->csList);
1187
1188 /* wait for an object to read, or time out */
1189 buffer = WaitForMultipleObjectsEx(This->samples+1, This->handle_list + oldest, FALSE, dwTimeout, TRUE);
1190
1191 EnterCriticalSection(&This->csList);
1192 if (buffer <= This->samples)
1193 {
1194 /* Re-scale the buffer back to normal */
1195 buffer += oldest;
1196
1197 /* Uh oh, we overshot the flusher handle, renormalize it back to 0..Samples-1 */
1198 if (buffer > This->samples)
1199 buffer -= This->samples + 1;
1200 assert(buffer <= This->samples);
1201 }
1202
1203 if (buffer >= This->samples)
1204 {
1205 if (buffer != This->samples)
1206 {
1207 FIXME("Returned: %u (%08x)\n", buffer, GetLastError());
1208 hr = VFW_E_TIMEOUT;
1209 }
1210 else
1211 hr = VFW_E_WRONG_STATE;
1212 buffer = ~0;
1213 }
1214 else
1215 --This->queued_number;
1216 }
1217
1218 if (This->bFlushing && buffer == ~0)
1219 {
1220 for (buffer = 0; buffer < This->samples; ++buffer)
1221 {
1222 if (This->sample_list[buffer].pSample)
1223 {
1224 ResetEvent(This->handle_list[buffer]);
1225 break;
1226 }
1227 }
1228 if (buffer == This->samples)
1229 {
1230 assert(!This->queued_number);
1231 hr = VFW_E_TIMEOUT;
1232 }
1233 else
1234 {
1235 --This->queued_number;
1236 hr = S_OK;
1237 }
1238 }
1239
1240 if (SUCCEEDED(hr))
1241 {
1242 REFERENCE_TIME rtStart, rtStop;
1243 REFERENCE_TIME rtSampleStart, rtSampleStop;
1244 DATAREQUEST *pDataRq = This->sample_list + buffer;
1245 DWORD dwBytes = 0;
1246
1247 /* get any errors */
1248 if (!This->bFlushing && !GetOverlappedResult(This->hFile, &pDataRq->ovl, &dwBytes, FALSE))
1249 hr = HRESULT_FROM_WIN32(GetLastError());
1250
1251 /* Return the sample no matter what so it can be destroyed */
1252 *ppSample = pDataRq->pSample;
1253 *pdwUser = pDataRq->dwUserData;
1254
1255 if (This->bFlushing)
1256 hr = VFW_E_WRONG_STATE;
1257
1258 if (FAILED(hr))
1259 dwBytes = 0;
1260
1261 /* Set the time on the sample */
1262 IMediaSample_SetActualDataLength(pDataRq->pSample, dwBytes);
1263
1264 rtStart = (DWORD64)pDataRq->ovl.u.s.Offset + ((DWORD64)pDataRq->ovl.u.s.OffsetHigh << 32);
1265 rtStart = MEDIATIME_FROM_BYTES(rtStart);
1266 rtStop = rtStart + MEDIATIME_FROM_BYTES(dwBytes);
1267
1268 IMediaSample_GetTime(pDataRq->pSample, &rtSampleStart, &rtSampleStop);
1269 assert(rtStart == rtSampleStart);
1270 assert(rtStop <= rtSampleStop);
1271
1272 IMediaSample_SetTime(pDataRq->pSample, &rtStart, &rtStop);
1273 assert(rtStart == rtSampleStart);
1274 if (hr == S_OK)
1275 assert(rtStop == rtSampleStop);
1276 else
1277 assert(rtStop == rtStart);
1278
1279 This->sample_list[buffer].pSample = NULL;
1280 assert(This->oldest_sample < This->samples);
1281
1282 if (buffer == This->oldest_sample)
1283 {
1284 LONG x;
1285 for (x = This->oldest_sample + 1; x < This->samples; ++x)
1286 if (This->sample_list[x].pSample)
1287 break;
1288 if (x >= This->samples)
1289 for (x = 0; x < This->oldest_sample; ++x)
1290 if (This->sample_list[x].pSample)
1291 break;
1292 if (This->oldest_sample == x)
1293 /* No samples found, reset to 0 */
1294 x = 0;
1295 This->oldest_sample = x;
1296 }
1297 }
1298 LeaveCriticalSection(&This->csList);
1299
1300 TRACE("-- %x\n", hr);
1301 return hr;
1302 }
1303
1304 static HRESULT WINAPI FileAsyncReader_SyncRead(IAsyncReader * iface, LONGLONG llPosition, LONG lLength, BYTE * pBuffer);
1305
1306 static HRESULT WINAPI FileAsyncReader_SyncReadAligned(IAsyncReader * iface, IMediaSample * pSample)
1307 {
1308 BYTE * pBuffer;
1309 REFERENCE_TIME tStart;
1310 REFERENCE_TIME tStop;
1311 HRESULT hr;
1312
1313 TRACE("(%p)\n", pSample);
1314
1315 hr = IMediaSample_GetTime(pSample, &tStart, &tStop);
1316
1317 if (SUCCEEDED(hr))
1318 hr = IMediaSample_GetPointer(pSample, &pBuffer);
1319
1320 if (SUCCEEDED(hr))
1321 hr = FileAsyncReader_SyncRead(iface,
1322 BYTES_FROM_MEDIATIME(tStart),
1323 (LONG) BYTES_FROM_MEDIATIME(tStop - tStart),
1324 pBuffer);
1325
1326 TRACE("-- %x\n", hr);
1327 return hr;
1328 }
1329
1330 static HRESULT WINAPI FileAsyncReader_SyncRead(IAsyncReader * iface, LONGLONG llPosition, LONG lLength, BYTE * pBuffer)
1331 {
1332 OVERLAPPED ovl;
1333 HRESULT hr = S_OK;
1334 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1335
1336 TRACE("(%x%08x, %d, %p)\n", (ULONG)(llPosition >> 32), (ULONG)llPosition, lLength, pBuffer);
1337
1338 ZeroMemory(&ovl, sizeof(ovl));
1339
1340 ovl.hEvent = CreateEventW(NULL, 0, 0, NULL);
1341 /* NOTE: llPosition is the actual byte position to start reading from */
1342 ovl.u.s.Offset = (DWORD) llPosition;
1343 ovl.u.s.OffsetHigh = (DWORD) (llPosition >> (sizeof(DWORD) * 8));
1344
1345 if (!ReadFile(This->hFile, pBuffer, lLength, NULL, &ovl))
1346 hr = HRESULT_FROM_WIN32(GetLastError());
1347
1348 if (hr == HRESULT_FROM_WIN32(ERROR_IO_PENDING))
1349 hr = S_OK;
1350
1351 if (SUCCEEDED(hr))
1352 {
1353 DWORD dwBytesRead;
1354
1355 if (!GetOverlappedResult(This->hFile, &ovl, &dwBytesRead, TRUE))
1356 hr = HRESULT_FROM_WIN32(GetLastError());
1357 }
1358
1359 CloseHandle(ovl.hEvent);
1360
1361 TRACE("-- %x\n", hr);
1362 return hr;
1363 }
1364
1365 static HRESULT WINAPI FileAsyncReader_Length(IAsyncReader * iface, LONGLONG * pTotal, LONGLONG * pAvailable)
1366 {
1367 DWORD dwSizeLow;
1368 DWORD dwSizeHigh;
1369 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1370
1371 TRACE("(%p, %p)\n", pTotal, pAvailable);
1372
1373 if (((dwSizeLow = GetFileSize(This->hFile, &dwSizeHigh)) == -1) &&
1374 (GetLastError() != NO_ERROR))
1375 return HRESULT_FROM_WIN32(GetLastError());
1376
1377 *pTotal = (LONGLONG)dwSizeLow | (LONGLONG)dwSizeHigh << (sizeof(DWORD) * 8);
1378
1379 *pAvailable = *pTotal;
1380
1381 return S_OK;
1382 }
1383
1384 static HRESULT WINAPI FileAsyncReader_BeginFlush(IAsyncReader * iface)
1385 {
1386 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1387
1388 TRACE("()\n");
1389
1390 EnterCriticalSection(&This->csList);
1391 This->bFlushing = TRUE;
1392 CancelIo(This->hFile);
1393 SetEvent(This->handle_list[This->samples]);
1394 LeaveCriticalSection(&This->csList);
1395
1396 return S_OK;
1397 }
1398
1399 static HRESULT WINAPI FileAsyncReader_EndFlush(IAsyncReader * iface)
1400 {
1401 FileAsyncReader *This = impl_from_IAsyncReader(iface);
1402 int x;
1403
1404 TRACE("()\n");
1405
1406 EnterCriticalSection(&This->csList);
1407 ResetEvent(This->handle_list[This->samples]);
1408 This->bFlushing = FALSE;
1409 for (x = 0; x < This->samples; ++x)
1410 assert(!This->sample_list[x].pSample);
1411
1412 LeaveCriticalSection(&This->csList);
1413
1414 return S_OK;
1415 }
1416
1417 static const IAsyncReaderVtbl FileAsyncReader_Vtbl =
1418 {
1419 FileAsyncReader_QueryInterface,
1420 FileAsyncReader_AddRef,
1421 FileAsyncReader_Release,
1422 FileAsyncReader_RequestAllocator,
1423 FileAsyncReader_Request,
1424 FileAsyncReader_WaitForNext,
1425 FileAsyncReader_SyncReadAligned,
1426 FileAsyncReader_SyncRead,
1427 FileAsyncReader_Length,
1428 FileAsyncReader_BeginFlush,
1429 FileAsyncReader_EndFlush,
1430 };
1431
1432
1433 static HRESULT WINAPI AMFilterMiscFlags_QueryInterface(IAMFilterMiscFlags *iface, REFIID riid, void **ppv) {
1434 AsyncReader *This = impl_from_IAMFilterMiscFlags(iface);
1435 return IUnknown_QueryInterface((IUnknown*)This, riid, ppv);
1436 }
1437
1438 static ULONG WINAPI AMFilterMiscFlags_AddRef(IAMFilterMiscFlags *iface) {
1439 AsyncReader *This = impl_from_IAMFilterMiscFlags(iface);
1440 return IUnknown_AddRef((IUnknown*)This);
1441 }
1442
1443 static ULONG WINAPI AMFilterMiscFlags_Release(IAMFilterMiscFlags *iface) {
1444 AsyncReader *This = impl_from_IAMFilterMiscFlags(iface);
1445 return IUnknown_Release((IUnknown*)This);
1446 }
1447
1448 static ULONG WINAPI AMFilterMiscFlags_GetMiscFlags(IAMFilterMiscFlags *iface) {
1449 return AM_FILTER_MISC_FLAGS_IS_SOURCE;
1450 }
1451
1452 static const IAMFilterMiscFlagsVtbl IAMFilterMiscFlags_Vtbl = {
1453 AMFilterMiscFlags_QueryInterface,
1454 AMFilterMiscFlags_AddRef,
1455 AMFilterMiscFlags_Release,
1456 AMFilterMiscFlags_GetMiscFlags
1457 };