17 lines
623 B
Diff
17 lines
623 B
Diff
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) {
|