[PATCH 3/3] amstream: Increase parent IAMMediaStream refcount in IDirectDrawMediaStream::CreateSample().

[Gijs Vermeulen]

Signed-off-by: Gijs Vermeulen 
---
 dlls/amstream/ddrawstream.c    | 3 +++
 dlls/amstream/tests/amstream.c | 8 ++++----
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/dlls/amstream/ddrawstream.c b/dlls/amstream/ddrawstream.c
index 917fcad9c9..0d3f2d0d4b 100644
--- a/dlls/amstream/ddrawstream.c
+++ b/dlls/amstream/ddrawstream.c
@@ -1048,6 +1048,8 @@ static ULONG WINAPI ddraw_sample_Release(IDirectDrawStreamSample *iface)
         --sample->parent->sample_refs;
         LeaveCriticalSection(&sample->parent->cs);

+        IAMMediaStream_Release(&sample->parent->IAMMediaStream_iface);
+
         if (sample->surface)
             IDirectDrawSurface_Release(sample->surface);
         HeapFree(GetProcessHeap(), 0, sample);
@@ -1163,6 +1165,7 @@ static HRESULT ddrawstreamsample_create(struct ddraw_stream *parent, IDirectDraw
     object->IDirectDrawStreamSample_iface.lpVtbl = &DirectDrawStreamSample_Vtbl;
     object->ref = 1;
     object->parent = parent;
+    IAMMediaStream_AddRef(&parent->IAMMediaStream_iface);
     ++parent->sample_refs;

if (surface)
diff --git a/dlls/amstream/tests/amstream.c b/dlls/amstream/tests/amstream.c
index 4149ff0cd6..e01e3e0cf9 100644
--- a/dlls/amstream/tests/amstream.c
+++ b/dlls/amstream/tests/amstream.c
@@ -5669,7 +5669,7 @@ static void test_ddrawstream_create_sample(void)
     EXPECT_REF(stream, 3);
     hr = IDirectDrawMediaStream_CreateSample(ddraw_stream, NULL, NULL, 0, &sample);
     ok(hr == S_OK, "Got hr %#x.\n", hr);
-    todo_wine EXPECT_REF(stream, 4);
+    EXPECT_REF(stream, 4);
     IDirectDrawStreamSample_Release(sample);
     EXPECT_REF(stream, 3);

@@ -5738,13 +5738,13 @@ static void test_ddrawstreamsample_get_media_stream(void)
         ok(hr == E_POINTER, "Got hr %#x.\n", hr);
     }

-    todo_wine EXPECT_REF(stream, 4);
+    EXPECT_REF(stream, 4);
     hr = IDirectDrawStreamSample_GetMediaStream(sample, &stream2);
     ok(hr == S_OK, "Got hr %#x.\n", hr);
     ok(stream2 == stream, "Expected stream %p, got %p.\n", stream, stream2);
-    todo_wine EXPECT_REF(stream, 5);
+    EXPECT_REF(stream, 5);
     IMediaStream_Release(stream2);
-    todo_wine EXPECT_REF(stream, 4);
+    EXPECT_REF(stream, 4);

IDirectDrawMediaStream_Release(ddraw_stream);
     ref = IDirectDrawStreamSample_Release(sample);