Mercurial > cpython-lru_cache-weakref

diff Modules/_cursesmodule.c @ 49856:68f3beb7eead

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
- Fix build failure of _cursesmodule.c building with -D_FORTIFY_SOURCE=2.
author Matthias Klose <doko@ubuntu.com>
date Wed, 26 Nov 2008 17:22:04 +0000
parents aa1fbe321acf
children 87fbbf22adf3
line wrap: on
line diff
--- a/Modules/_cursesmodule.c	Wed Nov 26 17:18:22 2008 +0000
+++ b/Modules/_cursesmodule.c	Wed Nov 26 17:22:04 2008 +0000
@@ -1857,6 +1857,7 @@
   int fd;
   FILE *fp;
   PyObject *data;
+  size_t datalen;
   WINDOW *win;
 
   PyCursesInitialised
@@ -1886,7 +1887,13 @@
     remove(fn);
     return NULL;
   }
-  fwrite(PyBytes_AS_STRING(data), 1, PyBytes_GET_SIZE(data), fp);
+  datalen = PyBytes_GET_SIZE(data);
+  if (fwrite(PyBytes_AS_STRING(data), 1, datalen, fp) != datalen) {
+    Py_DECREF(data);
+    fclose(fp);
+    remove(fn);
+    return PyErr_SetFromErrnoWithFilename(PyExc_IOError, fn);
+  }
   Py_DECREF(data);
   fseek(fp, 0, 0);
   win = getwin(fp);