Nikolay Sivov : mf: Add missing IMFMediaEventGenerator query support for sample grabber stream.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: c17994c21809303dbc98c1f71b11fbe26999f3ad
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=c17994c21809303dbc98c1f71b11fbe26999f3ad

Author: Nikolay Sivov <nsivov at codeweavers.com>
Date:   Thu Apr 23 15:03:18 2020 +0300

mf: Add missing IMFMediaEventGenerator query support for sample grabber stream.

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

---

 dlls/mf/samplegrabber.c | 1 +
 dlls/mf/tests/mf.c      | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/dlls/mf/samplegrabber.c b/dlls/mf/samplegrabber.c
index 1919ce6220..21f89c0ad0 100644
--- a/dlls/mf/samplegrabber.c
+++ b/dlls/mf/samplegrabber.c
@@ -141,6 +141,7 @@ static HRESULT WINAPI sample_grabber_stream_QueryInterface(IMFStreamSink *iface,
     TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), obj);
 
     if (IsEqualIID(riid, &IID_IMFStreamSink) ||
+            IsEqualIID(riid, &IID_IMFMediaEventGenerator) ||
             IsEqualIID(riid, &IID_IUnknown))
     {
         *obj = &grabber->IMFStreamSink_iface;
diff --git a/dlls/mf/tests/mf.c b/dlls/mf/tests/mf.c
index ae9f58a4bc..5b300cd90f 100644
--- a/dlls/mf/tests/mf.c
+++ b/dlls/mf/tests/mf.c
@@ -2028,6 +2028,10 @@ static void test_sample_grabber(void)
     hr = IMFMediaSink_GetStreamSinkByIndex(sink, 0, &stream);
     ok(hr == S_OK, "Failed to get sink stream, hr %#x.\n", hr);
 
+    hr = IMFStreamSink_QueryInterface(stream, &IID_IMFMediaEventGenerator, (void **)&unk);
+    ok(hr == S_OK, "Failed to get interface, hr %#x.\n", hr);
+    IUnknown_Release(unk);
+
     hr = IMFStreamSink_GetIdentifier(stream, &id);
     ok(hr == S_OK, "Failed to get stream id, hr %#x.\n", hr);
     ok(id == 0, "Unexpected id %#x.\n", id);