[PATCH v4 2/5] winegstreamer: Synchronize concurrent access to the media stream.

[Giovanni Mascellani]

From: Giovanni Mascellani

---
 dlls/winegstreamer/media_source.c | 57 ++++++++++++++++++++++---------
 1 file changed, 40 insertions(+), 17 deletions(-)

diff --git a/dlls/winegstreamer/media_source.c b/dlls/winegstreamer/media_source.c
index 5afe92d036b..fade993ca78 100644
--- a/dlls/winegstreamer/media_source.c
+++ b/dlls/winegstreamer/media_source.c
@@ -746,31 +746,51 @@ static HRESULT WINAPI media_stream_QueueEvent(IMFMediaStream *iface, MediaEventT
 static HRESULT WINAPI media_stream_GetMediaSource(IMFMediaStream *iface, IMFMediaSource **source)
 {
     struct media_stream *stream = impl_from_IMFMediaStream(iface);
+    HRESULT hr;

TRACE("%p, %p.\n", iface, source);

+    EnterCriticalSection(&stream->parent_source->cs);
+
     if (stream->state == STREAM_SHUTDOWN)
-        return MF_E_SHUTDOWN;
+    {
+        hr = MF_E_SHUTDOWN;
+    }
+    else
+    {
+        IMFMediaSource_AddRef(&stream->parent_source->IMFMediaSource_iface);
+        *source = &stream->parent_source->IMFMediaSource_iface;
+        hr = S_OK;
+    }

-    IMFMediaSource_AddRef(&stream->parent_source->IMFMediaSource_iface);
-    *source = &stream->parent_source->IMFMediaSource_iface;
+    LeaveCriticalSection(&stream->parent_source->cs);

-    return S_OK;
+    return hr;
 }

static HRESULT WINAPI media_stream_GetStreamDescriptor(IMFMediaStream* iface, IMFStreamDescriptor **descriptor)
 {
     struct media_stream *stream = impl_from_IMFMediaStream(iface);
+    HRESULT hr;

TRACE("%p, %p.\n", iface, descriptor);

+    EnterCriticalSection(&stream->parent_source->cs);
+
     if (stream->state == STREAM_SHUTDOWN)
-        return MF_E_SHUTDOWN;
+    {
+        hr = MF_E_SHUTDOWN;
+    }
+    else
+    {
+        IMFStreamDescriptor_AddRef(stream->descriptor);
+        *descriptor = stream->descriptor;
+        hr = S_OK;
+    }

-    IMFStreamDescriptor_AddRef(stream->descriptor);
-    *descriptor = stream->descriptor;
+    LeaveCriticalSection(&stream->parent_source->cs);

-    return S_OK;
+    return hr;
 }

static HRESULT WINAPI media_stream_RequestSample(IMFMediaStream *iface, IUnknown *token)
@@ -781,21 +801,22 @@ static HRESULT WINAPI media_stream_RequestSample(IMFMediaStream *iface, IUnknown

TRACE("%p, %p.\n", iface, token);

-    if (stream->state == STREAM_SHUTDOWN)
-        return MF_E_SHUTDOWN;
+    EnterCriticalSection(&stream->parent_source->cs);

-    if (stream->state == STREAM_INACTIVE)
+    if (stream->state == STREAM_SHUTDOWN)
+    {
+        hr = MF_E_SHUTDOWN;
+    }
+    else if (stream->state == STREAM_INACTIVE)
     {
         WARN("Stream isn't active\n");
-        return MF_E_MEDIA_SOURCE_WRONGSTATE;
+        hr = MF_E_MEDIA_SOURCE_WRONGSTATE;
     }
-
-    if (stream->eos)
+    else if (stream->eos)
     {
-        return MF_E_END_OF_STREAM;
+        hr = MF_E_END_OF_STREAM;
     }
-
-    if (SUCCEEDED(hr = source_create_async_op(SOURCE_ASYNC_REQUEST_SAMPLE, &command)))
+    else if (SUCCEEDED(hr = source_create_async_op(SOURCE_ASYNC_REQUEST_SAMPLE, &command)))
     {
         command->u.request_sample.stream = stream;
         if (token)
@@ -806,6 +827,8 @@ static HRESULT WINAPI media_stream_RequestSample(IMFMediaStream *iface, IUnknown
                 &stream->parent_source->async_commands_callback, &command->IUnknown_iface);
     }

+    LeaveCriticalSection(&stream->parent_source->cs);
+
     return hr;
 }