db74b09
From ac16bdf2d41724b5a65255d4c28fb0ec46bc42f5 Mon Sep 17 00:00:00 2001
db74b09
From: Mike Frysinger <vapier@gentoo.org>
db74b09
Date: Sat, 14 Jul 2018 13:54:08 -0400
db74b09
Subject: [PATCH] bmp: check return value in gdImageBmpPtr
db74b09
db74b09
Closes #447.
db74b09
---
db74b09
 src/gd_bmp.c | 17 ++++++++++++++---
db74b09
 1 file changed, 14 insertions(+), 3 deletions(-)
db74b09
db74b09
diff --git a/src/gd_bmp.c b/src/gd_bmp.c
db74b09
index bde0b9d3..78f40d9a 100644
db74b09
--- a/src/gd_bmp.c
db74b09
+++ b/src/gd_bmp.c
db74b09
@@ -47,6 +47,8 @@ static int bmp_read_4bit(gdImagePtr im, gdIOCtxPtr infile, bmp_info_t *info, bmp
db74b09
 static int bmp_read_8bit(gdImagePtr im, gdIOCtxPtr infile, bmp_info_t *info, bmp_hdr_t *header);
db74b09
 static int bmp_read_rle(gdImagePtr im, gdIOCtxPtr infile, bmp_info_t *info);
db74b09
 
db74b09
+static int _gdImageBmpCtx(gdImagePtr im, gdIOCtxPtr out, int compression);
db74b09
+
db74b09
 #define BMP_DEBUG(s)
db74b09
 
db74b09
 static int gdBMPPutWord(gdIOCtx *out, int w)
db74b09
@@ -87,8 +89,10 @@ BGD_DECLARE(void *) gdImageBmpPtr(gdImagePtr im, int *size, int compression)
db74b09
 	void *rv;
db74b09
 	gdIOCtx *out = gdNewDynamicCtx(2048, NULL);
db74b09
 	if (out == NULL) return NULL;
db74b09
-	gdImageBmpCtx(im, out, compression);
db74b09
-	rv = gdDPExtractData(out, size);
db74b09
+	if (!_gdImageBmpCtx(im, out, compression))
db74b09
+		rv = gdDPExtractData(out, size);
db74b09
+	else
db74b09
+		rv = NULL;
db74b09
 	out->gd_free(out);
db74b09
 	return rv;
db74b09
 }
db74b09
@@ -141,6 +145,11 @@ BGD_DECLARE(void) gdImageBmp(gdImagePtr im, FILE *outFile, int compression)
db74b09
 		compression - whether to apply RLE or not.
db74b09
 */
db74b09
 BGD_DECLARE(void) gdImageBmpCtx(gdImagePtr im, gdIOCtxPtr out, int compression)
db74b09
+{
db74b09
+	_gdImageBmpCtx(im, out, compression);
db74b09
+}
db74b09
+
db74b09
+static int _gdImageBmpCtx(gdImagePtr im, gdIOCtxPtr out, int compression)
db74b09
 {
db74b09
 	int bitmap_size = 0, info_size, total_size, padding;
db74b09
 	int i, row, xpos, pixel;
db74b09
@@ -148,6 +157,7 @@ BGD_DECLARE(void) gdImageBmpCtx(gdImagePtr im, gdIOCtxPtr out, int compression)
db74b09
 	unsigned char *uncompressed_row = NULL, *uncompressed_row_start = NULL;
db74b09
 	FILE *tmpfile_for_compression = NULL;
db74b09
 	gdIOCtxPtr out_original = NULL;
db74b09
+	int ret = 1;
db74b09
 
db74b09
 	/* No compression if its true colour or we don't support seek */
db74b09
 	if (im->trueColor) {
db74b09
@@ -325,6 +335,7 @@ BGD_DECLARE(void) gdImageBmpCtx(gdImagePtr im, gdIOCtxPtr out, int compression)
db74b09
 		out_original = NULL;
db74b09
 	}
db74b09
 
db74b09
+	ret = 0;
db74b09
 cleanup:
db74b09
 	if (tmpfile_for_compression) {
db74b09
 #ifdef _WIN32
db74b09
@@ -338,7 +349,7 @@ BGD_DECLARE(void) gdImageBmpCtx(gdImagePtr im, gdIOCtxPtr out, int compression)
db74b09
 	if (out_original) {
db74b09
 		out_original->gd_free(out_original);
db74b09
 	}
db74b09
-	return;
db74b09
+	return ret;
db74b09
 }
db74b09
 
db74b09
 static int compress_row(unsigned char *row, int length)