Rémi Bernon : winegstreamer: Check WMA sample size before wrapping the sample.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 2c4fbd15316d46c1b856bd38670deda745167e41
URL:    https://gitlab.winehq.org/wine/wine/-/commit/2c4fbd15316d46c1b856bd38670deda...

Author: Rémi Bernon <rbernon(a)codeweavers.com>
Date:   Wed Sep 14 21:59:51 2022 +0200

winegstreamer: Check WMA sample size before wrapping the sample.

---

 dlls/winegstreamer/wma_decoder.c | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/dlls/winegstreamer/wma_decoder.c b/dlls/winegstreamer/wma_decoder.c
index 2951d7cac4d..4374f5bdcdf 100644
--- a/dlls/winegstreamer/wma_decoder.c
+++ b/dlls/winegstreamer/wma_decoder.c
@@ -531,6 +531,7 @@ static HRESULT WINAPI transform_ProcessInput(IMFTransform *iface, DWORD id, IMFS
     struct wma_decoder *decoder = impl_from_IMFTransform(iface);
     struct wg_sample *wg_sample;
     MFT_INPUT_STREAM_INFO info;
+    DWORD total_length;
     HRESULT hr;
 
     TRACE("iface %p, id %lu, sample %p, flags %#lx.\n", iface, id, sample, flags);
@@ -538,18 +539,16 @@ static HRESULT WINAPI transform_ProcessInput(IMFTransform *iface, DWORD id, IMFS
     if (!decoder->wg_transform)
         return MF_E_TRANSFORM_TYPE_NOT_SET;
 
-    if (FAILED(hr = IMFTransform_GetInputStreamInfo(iface, 0, &info)))
-        return hr;
-
-    if (FAILED(hr = wg_sample_create_mf(sample, &wg_sample)))
+    if (FAILED(hr = IMFTransform_GetInputStreamInfo(iface, 0, &info))
+            || FAILED(hr = IMFSample_GetTotalLength(sample, &total_length)))
         return hr;
 
     /* WMA transform uses fixed size input samples and ignores samples with invalid sizes */
-    if (wg_sample->size % info.cbSize)
-    {
-        wg_sample_release(wg_sample);
+    if (total_length % info.cbSize)
         return S_OK;
-    }
+
+    if (FAILED(hr = wg_sample_create_mf(sample, &wg_sample)))
+        return hr;
 
     return wg_transform_push_mf(decoder->wg_transform, wg_sample, decoder->wg_sample_queue);
 }