diff options
Diffstat (limited to 'Modules/zipimport.c')
| -rw-r--r-- | Modules/zipimport.c | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/Modules/zipimport.c b/Modules/zipimport.c index 2feb2a827c..dad699e7e9 100644 --- a/Modules/zipimport.c +++ b/Modules/zipimport.c @@ -1089,6 +1089,11 @@ get_data(PyObject *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; + } bytes_size = compress == 0 ? data_size : data_size + 1; if (bytes_size == 0) bytes_size++; |
