WebP: Fix memory leak in test and avoid possible memory leak if size is not correct (ie. passing 0 size will leak). Min size is the WebP sign (3x32bits)

1 files changed, 7 insertions, 2 deletions

diff --git a/tests/webp/gdImageWebpEx.c b/tests/webp/gdImageWebpEx.c
index 43011de..9ed1008 100644
--- a/tests/webp/gdImageWebpEx.c
+++ b/tests/webp/gdImageWebpEx.c
@@ -9,6 +9,7 @@ int main()
 	gdImagePtr im;
 	FILE *fp1;
 	int size = 1;
+	void *data;
 
 	im = gdImageCreateTrueColor(20, 20);
 	if (!im) {
@@ -17,10 +18,14 @@ int main()
 
 	fp1 = gdTestTempFp();
 	gdImageWebpEx(im, fp1, 8);
-	gdImageWebpPtrEx(im, &size, 9);
-
+	data = gdImageWebpPtrEx(im, &size, 9);
 	fclose(fp1);
+    if (data == NULL) {
+		gdImageDestroy(im);
+        return 1;
+    }
 
+    gdFree(data);
 	gdImageDestroy(im);
 	return 0;
 }