Merge pull request #3 from radarhere/write-jpeg-com

Free comment when returning early
python-pillow · Dec 6, 2022 · 6ca08a4 · 6ca08a4
2 parents eddc9bd + 1d78008
commit 6ca08a4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/encode.c b/src/encode.c
@@ -1113,6 +1113,9 @@ PyImaging_JpegEncoderNew(PyObject *self, PyObject *args) {
         /* malloc check ok, length is from python parsearg */
         char *p = malloc(extra_size);  // Freed in JpegEncode, Case 6
         if (!p) {
+            if (comment) {
+                free(comment);
+            }
             return ImagingError_MemoryError();
         }
         memcpy(p, extra, extra_size);
@@ -1125,6 +1128,9 @@ PyImaging_JpegEncoderNew(PyObject *self, PyObject *args) {
         /* malloc check ok, length is from python parsearg */
         char *pp = malloc(rawExifLen);  // Freed in JpegEncode, Case 6
         if (!pp) {
+            if (comment) {
+                free(comment);
+            }
             if (extra) {
                 free(extra);
             }