Fix some theoretical problems with png reading

mdboom · mdboom · commit 17aad94ab367 · 2015-11-18T22:10:29.000-05:00
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -1279,8 +1279,8 @@ def pilread(fname):
             from PIL import Image
         except ImportError:
             return None
-        image = Image.open(fname)
-        return pil_to_array(image)
+        with Image.open(fname) as image:
+            return pil_to_array(image)
 
     handlers = {'png': _png.read_png, }
     if format is None:
diff --git a/src/_png.cpp b/src/_png.cpp
@@ -328,10 +328,10 @@ static void _read_png_data(PyObject *py_file_obj, png_bytep data, png_size_t len
     Py_ssize_t bufflen;
     if (read_method) {
         result = PyObject_CallFunction(read_method, (char *)"i", length);
-    }
-    if (PyBytes_AsStringAndSize(result, &buffer, &bufflen) == 0) {
-        if (bufflen == (Py_ssize_t)length) {
-            memcpy(data, buffer, length);
+        if (PyBytes_AsStringAndSize(result, &buffer, &bufflen) == 0) {
+            if (bufflen == (Py_ssize_t)length) {
+                memcpy(data, buffer, length);
+            }
         }
     }
     Py_XDECREF(read_method);
