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 55bfb5d..83fa8f9 100644 --- a/Modules/zipimport.c +++ b/Modules/zipimport.c @@ -1111,6 +1111,11 @@ get_data(PyObject *archive, PyObject *toc_entry) } 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++; |