OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

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

This commit is contained in:
Adrià Arrufat 2022-06-14 20:59:38 +09:00 committed by GitHub
parent 0f1d8e964f
commit 7f46535cce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dlib/image_saver/save_webp.cpp
View File

@ -74,8 +74,11 @@ namespace dlib {
{ {
fout.write(reinterpret_cast<char*>(output), output_size); fout.write(reinterpret_cast<char*>(output), output_size);
if (!fout.good()) if (!fout.good())
{
WebPFree(output);
throw image_save_error("Error while writing WebP image to " + filename + "."); throw image_save_error("Error while writing WebP image to " + filename + ".");
} }
}
else else
{ {
throw image_save_error("Error while encoding WebP image to " + filename + "."); throw image_save_error("Error while encoding WebP image to " + filename + ".");