Set exception if read past end of file

mdboom · mdboom · commit d7d70389427e · 2015-11-20T14:26:17.000-05:00
diff --git a/lib/matplotlib/backends/backend_agg.py b/lib/matplotlib/backends/backend_agg.py
@@ -526,6 +526,7 @@ def print_png(self, filename_or_obj, *args, **kwargs):
             close = True
         else:
             close = False
+
         try:
             _png.write_png(renderer._renderer, filename_or_obj, self.figure.dpi)
         finally:
diff --git a/src/_png.cpp b/src/_png.cpp
@@ -331,6 +331,8 @@ static void _read_png_data(PyObject *py_file_obj, png_bytep data, png_size_t len
         if (PyBytes_AsStringAndSize(result, &buffer, &bufflen) == 0) {
             if (bufflen == (Py_ssize_t)length) {
                 memcpy(data, buffer, length);
+            } else {
+                PyErr_SetString(PyExc_IOError, "read past end of file");
             }
         }
     }

Original file line number	Diff line number	Diff line change
`@@ -331,6 +331,8 @@ static void _read_png_data(PyObject *py_file_obj, png_bytep data, png_size_t len`
`331`	`331`	`if (PyBytes_AsStringAndSize(result, &buffer, &bufflen) == 0) {`
`332`	`332`	`if (bufflen == (Py_ssize_t)length) {`
`333`	`333`	`memcpy(data, buffer, length);`
	`334`	`+ } else {`
	`335`	`+ PyErr_SetString(PyExc_IOError, "read past end of file");`
`334`	`336`	`}`
`335`	`337`	`}`
`336`	`338`	`}`