[PATCH 3/7] mfsrcsnk: Implement IMFMediaSource_Shutdown for the media sources.

[Rémi Bernon]

From: Rémi Bernon

---
 dlls/mfsrcsnk/media_source.c | 39 ++++++++++++++++++++++++++++++++++--
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/dlls/mfsrcsnk/media_source.c b/dlls/mfsrcsnk/media_source.c
index 8b6a015073f..1329787c546 100644
--- a/dlls/mfsrcsnk/media_source.c
+++ b/dlls/mfsrcsnk/media_source.c
@@ -29,8 +29,15 @@ struct media_source
     IMFMediaSource IMFMediaSource_iface;
     LONG refcount;

+    CRITICAL_SECTION cs;
     IMFMediaEventQueue *queue;
     IMFByteStream *stream;
+
+    enum
+    {
+        SOURCE_STOPPED,
+        SOURCE_SHUTDOWN,
+    } state;
 };

static struct media_source *media_source_from_IMFMediaSource(IMFMediaSource *iface)
@@ -75,7 +82,14 @@ static ULONG WINAPI media_source_Release(IMFMediaSource *iface)

if (!refcount)
     {
+        IMFMediaSource_Shutdown(iface);
+
         IMFMediaEventQueue_Release(source->queue);
+        IMFByteStream_Release(source->stream);
+
+        source->cs.DebugInfo->Spare[0] = 0;
+        DeleteCriticalSection(&source->cs);
+
         free(source);
     }

@@ -153,8 +167,24 @@ static HRESULT WINAPI media_source_Pause(IMFMediaSource *iface)
 static HRESULT WINAPI media_source_Shutdown(IMFMediaSource *iface)
 {
     struct media_source *source = media_source_from_IMFMediaSource(iface);
-    FIXME("source %p\n", source);
-    return E_NOTIMPL;
+
+    TRACE("source %p\n", source);
+
+    EnterCriticalSection(&source->cs);
+
+    if (source->state == SOURCE_SHUTDOWN)
+    {
+        LeaveCriticalSection(&source->cs);
+        return MF_E_SHUTDOWN;
+    }
+    source->state = SOURCE_SHUTDOWN;
+
+    IMFMediaEventQueue_Shutdown(source->queue);
+    IMFByteStream_Close(source->stream);
+
+    LeaveCriticalSection(&source->cs);
+
+    return S_OK;
 }

static const IMFMediaSourceVtbl media_source_vtbl =
@@ -192,6 +222,11 @@ static HRESULT media_source_create(const WCHAR *url, IMFByteStream *stream, IMFM
         return hr;
     }

+    IMFByteStream_AddRef((source->stream = stream));
+
+    InitializeCriticalSectionEx(&source->cs, 0, RTL_CRITICAL_SECTION_FLAG_FORCE_DEBUG_INFO);
+    source->cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": cs");
+
     *out = &source->IMFMediaSource_iface;
     TRACE("created source %p\n", source);
     return S_OK;