- Fix build failure of _cursesmodule.c building with -D_FORTIFY_SOURCE=2.

doko42 · doko42 · commit 7e1b8faacc60 · 2008-11-26T17:22:04.000Z
diff --git a/Misc/NEWS b/Misc/NEWS
@@ -19,6 +19,8 @@ Core and Builtins
 - Issue #4367: Python would segfault during compiling when the unicodedata
   module couldn't be imported and \N escapes were present.
 
+- Fix build failure of _cursesmodule.c building with -D_FORTIFY_SOURCE=2.
+
 Library
 -------
 
diff --git a/Modules/_cursesmodule.c b/Modules/_cursesmodule.c
@@ -1857,6 +1857,7 @@ PyCurses_GetWin(PyCursesWindowObject *self, PyObject *stream)
   int fd;
   FILE *fp;
   PyObject *data;
+  size_t datalen;
   WINDOW *win;
 
   PyCursesInitialised
@@ -1886,7 +1887,13 @@ PyCurses_GetWin(PyCursesWindowObject *self, PyObject *stream)
     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);
