- Fix build failure of _cursesmodule.c building with -D_FORTIFY_SOURCE=2.
diff --git a/Modules/_cursesmodule.c b/Modules/_cursesmodule.c
index 4f99da5..3764f48 100644
--- a/Modules/_cursesmodule.c
+++ b/Modules/_cursesmodule.c
@@ -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);