Fix potential memory leak when writing WebP throws (#2600)

2022-06-14 20:59:38 +09:00 · 2022-06-14 20:59:38 +09:00 · 7f46535cce
parent 0f1d8e964f
commit 7f46535cce
1 changed files with 3 additions and 0 deletions
--- a/dlib/image_saver/save_webp.cpp
+++ b/dlib/image_saver/save_webp.cpp
@ -74,7 +74,10 @@ namespace dlib {
            {
                fout.write(reinterpret_cast<char*>(output), output_size);
                if (!fout.good())
+                {
+                    WebPFree(output);
                    throw image_save_error("Error while writing WebP image to " + filename + ".");
+                }
            }
            else
            {