mfreadwrite: Query presentation descriptor on reader creation.

Signed-off-by: Nikolay Sivov <nsivov@codeweavers.com> Signed-off-by: Alexandre Julliard <julliard@winehq.org>

mfreadwrite: Query presentation descriptor on reader creation.
fcae0bc7 · Nikolay Sivov · Alexandre Julliard · 1fb58907 · fcae0bc7
Commit fcae0bc7 authored Mar 18, 2019 by Nikolay Sivov Committed by Alexandre Julliard Mar 18, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 0 deletions

main.c dlls/mfreadwrite/main.c +8 -0

No files found.
--- a/dlls/mfreadwrite/main.c
+++ b/dlls/mfreadwrite/main.c
@@ -76,6 +76,7 @@ typedef struct source_reader
    IMFSourceReader IMFSourceReader_iface;
    LONG refcount;
    IMFMediaSource *source;
+    IMFPresentationDescriptor *descriptor;
 } srcreader;

 struct sink_writer
@@ -135,6 +136,8 @@ static ULONG WINAPI src_reader_Release(IMFSourceReader *iface)

    if (!refcount)
    {
+        if (reader->descriptor)
+            IMFPresentationDescriptor_Release(reader->descriptor);
        IMFMediaSource_Release(reader->source);
        heap_free(reader);
    }
@@ -279,7 +282,12 @@ static HRESULT create_source_reader_from_source(IMFMediaSource *source, IMFAttri
    object->source = source;
    IMFMediaSource_AddRef(object->source);

+    if (FAILED(hr = IMFMediaSource_CreatePresentationDescriptor(object->source, &object->descriptor)))
+        goto failed;
+
    hr = IMFSourceReader_QueryInterface(&object->IMFSourceReader_iface, riid, out);
+
+failed:
    IMFSourceReader_Release(&object->IMFSourceReader_iface);
    return hr;
 }