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