windowscodecs: Fix IWICBitmapEncoder::SetPalette for a not initialized case in PNG encoder.

Signed-off-by: Vincent Povirk <vincent@codeweavers.com> Signed-off-by: Alexandre Julliard <julliard@winehq.org>

windowscodecs: Fix IWICBitmapEncoder::SetPalette for a not initialized case in PNG encoder.
eb3eff90 · Dmitry Timoshkov · Alexandre Julliard · 2ea95f89 · eb3eff90
Commit eb3eff90 authored Sep 05, 2017 by Dmitry Timoshkov Committed by Alexandre Julliard Sep 06, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 13 additions and 3 deletions

pngformat.c dlls/windowscodecs/pngformat.c +13 -3

No files found.
--- a/dlls/windowscodecs/pngformat.c
+++ b/dlls/windowscodecs/pngformat.c
@@ -1952,10 +1952,20 @@ static HRESULT WINAPI PngEncoder_SetColorContexts(IWICBitmapEncoder *iface,
    return E_NOTIMPL;
 }

-static HRESULT WINAPI PngEncoder_SetPalette(IWICBitmapEncoder *iface, IWICPalette *pIPalette)
+static HRESULT WINAPI PngEncoder_SetPalette(IWICBitmapEncoder *iface, IWICPalette *palette)
 {
-    TRACE("(%p,%p)\n", iface, pIPalette);
-    return WINCODEC_ERR_UNSUPPORTEDOPERATION;
+    PngEncoder *This = impl_from_IWICBitmapEncoder(iface);
+    HRESULT hr;
+
+    TRACE("(%p,%p)\n", iface, palette);
+
+    EnterCriticalSection(&This->lock);
+
+    hr = This->stream ? WINCODEC_ERR_UNSUPPORTEDOPERATION : WINCODEC_ERR_NOTINITIALIZED;
+
+    LeaveCriticalSection(&This->lock);
+
+    return hr;
 }

 static HRESULT WINAPI PngEncoder_SetThumbnail(IWICBitmapEncoder *iface, IWICBitmapSource *pIThumbnail)