mf/evr: Consistently check for shutdown flag within a lock.

dlls/mf/evr.c

@@ -1432,14 +1432,17 @@ static HRESULT WINAPI video_renderer_sink_GetPresentationClock(IMFMediaSink *ifa
 static HRESULT WINAPI video_renderer_sink_Shutdown(IMFMediaSink *iface)
 {
     struct video_renderer *renderer = impl_from_IMFMediaSink(iface);
+    HRESULT hr = S_OK;
     size_t i;
 
     TRACE("%p.\n", iface);
 
-    if (renderer->flags & EVR_SHUT_DOWN)
-        return MF_E_SHUTDOWN;
-
     EnterCriticalSection(&renderer->cs);
+
+    if (renderer->flags & EVR_SHUT_DOWN)
+        hr = MF_E_SHUTDOWN;
+    else
+    {
         renderer->flags |= EVR_SHUT_DOWN;
         /* Detach streams from the sink. */
         for (i = 0; i < renderer->stream_count; ++i)
@@ -1461,9 +1464,11 @@ static HRESULT WINAPI video_renderer_sink_Shutdown(IMFMediaSink *iface)
         IMFMediaEventQueue_Shutdown(renderer->event_queue);
         video_renderer_set_presentation_clock(renderer, NULL);
         video_renderer_release_services(renderer);
+    }
+
     LeaveCriticalSection(&renderer->cs);
 
-    return S_OK;
+    return hr;
 }
 
 static const IMFMediaSinkVtbl video_renderer_sink_vtbl =
@@ -1742,52 +1747,64 @@ static HRESULT video_renderer_initialize(struct video_renderer *renderer, IMFTra
     return hr;
 }
 
-static HRESULT WINAPI video_renderer_InitializeRenderer(IMFVideoRenderer *iface, IMFTransform *mixer,
-        IMFVideoPresenter *presenter)
+static HRESULT video_renderer_create_mixer_and_presenter(struct video_renderer *renderer,
+        IMFTransform **mixer, IMFVideoPresenter **presenter)
 {
-    struct video_renderer *renderer = impl_from_IMFVideoRenderer(iface);
     HRESULT hr;
 
-    TRACE("%p, %p, %p.\n", iface, mixer, presenter);
-
-    EnterCriticalSection(&renderer->cs);
-
-    if (renderer->flags & EVR_SHUT_DOWN)
+    if (*mixer)
     {
-        LeaveCriticalSection(&renderer->cs);
-        return MF_E_SHUTDOWN;
+        IMFTransform_AddRef(*mixer);
     }
-
-    video_renderer_uninitialize(renderer);
-
-    if (mixer)
-        IMFTransform_AddRef(mixer);
-    else if (FAILED(hr = video_renderer_create_mixer(NULL, &mixer)))
+    else if (FAILED(hr = video_renderer_create_mixer(NULL, mixer)))
     {
         WARN("Failed to create default mixer object, hr %#lx.\n", hr);
-        LeaveCriticalSection(&renderer->cs);
         return hr;
     }
 
-    if (presenter)
-        IMFVideoPresenter_AddRef(presenter);
-    else if (FAILED(hr = video_renderer_create_presenter(renderer, NULL, &presenter)))
+    if (*presenter)
+    {
+        IMFVideoPresenter_AddRef(*presenter);
+    }
+    else if (FAILED(hr = video_renderer_create_presenter(renderer, NULL, presenter)))
     {
         WARN("Failed to create default presenter, hr %#lx.\n", hr);
-        LeaveCriticalSection(&renderer->cs);
-        IMFTransform_Release(mixer);
+        IMFTransform_Release(*mixer);
         return hr;
     }
 
+    return S_OK;
+}
+
+static HRESULT WINAPI video_renderer_InitializeRenderer(IMFVideoRenderer *iface, IMFTransform *mixer,
+        IMFVideoPresenter *presenter)
+{
+    struct video_renderer *renderer = impl_from_IMFVideoRenderer(iface);
+    HRESULT hr;
+
+    TRACE("%p, %p, %p.\n", iface, mixer, presenter);
+
+    EnterCriticalSection(&renderer->cs);
+
+    if (renderer->flags & EVR_SHUT_DOWN)
+        hr = MF_E_SHUTDOWN;
+    else
+    {
+        video_renderer_uninitialize(renderer);
+
+        if (SUCCEEDED(hr = video_renderer_create_mixer_and_presenter(renderer, &mixer, &presenter)))
+        {
             /* FIXME: check clock state */
             /* FIXME: check that streams are not initialized */
 
             hr = video_renderer_initialize(renderer, mixer, presenter);
 
-    LeaveCriticalSection(&renderer->cs);
-
             IMFTransform_Release(mixer);
             IMFVideoPresenter_Release(presenter);
+        }
+    }
+
+    LeaveCriticalSection(&renderer->cs);
 
     return hr;
 }