diff --git a/Modules/zipimport.c b/Modules/zipimport.c index 7240cb4..a139a3f 100644 --- a/Modules/zipimport.c +++ b/Modules/zipimport.c @@ -895,6 +895,11 @@ get_data(char *archive, PyObject *toc_entry) PyMarshal_ReadShortFromFile(fp); /* local header size */ file_offset += l; /* Start of file data */ + if (data_size > LONG_MAX - 1) { + fclose(fp); + PyErr_NoMemory(); + return NULL; + } raw_data = PyString_FromStringAndSize((char *)NULL, compress == 0 ? data_size : data_size + 1); if (raw_data == NULL) {