cpython: 20b77ff040b6 (original) (raw)

--- a/Lib/test/test_zipimport.py +++ b/Lib/test/test_zipimport.py @@ -71,6 +71,27 @@ class ImportHooksBaseTestCase(unittest.T support.modules_cleanup(*self.modules_before) +def _write_zip_package(zipname, files,

                  data_to_prepend=b"", compression=ZIP_STORED):[](#l1.8)

z = ZipFile(zipname, "w")
try:

   for name, (mtime, data) in files.items():[](#l1.11)

       zinfo = ZipInfo(name, time.localtime(mtime))[](#l1.12)

       zinfo.compress_type = compression[](#l1.13)

       z.writestr(zinfo, data)[](#l1.14)

finally:
```
   z.close()[](#l1.16)
```

if data_to_prepend:

   # Prepend data to the start of the zipfile[](#l1.19)

   with open(zipname, "rb") as f:[](#l1.20)

```
       zip_data = f.read()[](#l1.21)
```

   with open(zipname, "wb") as f:[](#l1.23)

       f.write(data_to_prepend)[](#l1.24)

```
       f.write(zip_data)[](#l1.25)
```

+ + class UncompressedZipImportTestCase(ImportHooksBaseTestCase): compression = ZIP_STORED @@ -83,23 +104,9 @@ class UncompressedZipImportTestCase(Impo ImportHooksBaseTestCase.setUp(self) def doTest(self, expected_ext, files, *modules, **kw):

```
   z = ZipFile(TEMP_ZIP, "w")[](#l1.35)
```

   _write_zip_package(TEMP_ZIP, files, data_to_prepend=kw.get("stuff"),[](#l1.36)

                      compression=self.compression)[](#l1.37)
   try:[](#l1.38)

       for name, (mtime, data) in files.items():[](#l1.39)

           zinfo = ZipInfo(name, time.localtime(mtime))[](#l1.40)

           zinfo.compress_type = self.compression[](#l1.41)

           z.writestr(zinfo, data)[](#l1.42)

```
       z.close()[](#l1.43)
```

       stuff = kw.get("stuff", None)[](#l1.45)

```
       if stuff is not None:[](#l1.46)
```

           # Prepend 'stuff' to the start of the zipfile[](#l1.47)

           with open(TEMP_ZIP, "rb") as f:[](#l1.48)

               data = f.read()[](#l1.49)

           with open(TEMP_ZIP, "wb") as f:[](#l1.50)

```
               f.write(stuff)[](#l1.51)
```
```
               f.write(data)[](#l1.52)
```

- sys.path.insert(0, TEMP_ZIP) mod = import(".".join(modules), globals(), locals(), @@ -114,7 +121,8 @@ class UncompressedZipImportTestCase(Impo self.assertEqual(file, os.path.join(TEMP_ZIP, *modules) + expected_ext) finally:

```
       z.close()[](#l1.61)
```

       while TEMP_ZIP in sys.path:[](#l1.62)

           sys.path.remove(TEMP_ZIP)[](#l1.63)
       os.remove(TEMP_ZIP)[](#l1.64)

def testAFakeZlib(self): @@ -422,10 +430,67 @@ class CompressedZipImportTestCase(Uncomp compression = ZIP_DEFLATED +class ZipFileModifiedAfterImportTestCase(ImportHooksBaseTestCase):

def setUp(self):

   zipimport._zip_directory_cache.clear()[](#l1.73)

   zipimport._zip_stat_cache.clear()[](#l1.74)

   ImportHooksBaseTestCase.setUp(self)[](#l1.75)

def tearDown(self):

   ImportHooksBaseTestCase.tearDown(self)[](#l1.78)

   if os.path.exists(TEMP_ZIP):[](#l1.79)

```
       os.remove(TEMP_ZIP)[](#l1.80)
```

def testZipFileChangesAfterFirstImport(self):

   """Alter the zip file after caching its index and try an import."""[](#l1.83)

   packdir = TESTPACK + os.sep[](#l1.84)

   files = {packdir + "__init__" + pyc_ext: (NOW, test_pyc),[](#l1.85)

            packdir + TESTMOD + ".py": (NOW, "test_value = 38\n"),[](#l1.86)

            "ziptest_a.py": (NOW, "test_value = 23\n"),[](#l1.87)

            "ziptest_b.py": (NOW, "test_value = 42\n"),[](#l1.88)

            "ziptest_c.py": (NOW, "test_value = 1337\n")}[](#l1.89)

```
   zipfile_path = TEMP_ZIP[](#l1.90)
```

   _write_zip_package(zipfile_path, files)[](#l1.91)

   self.assertTrue(os.path.exists(zipfile_path))[](#l1.92)

   sys.path.insert(0, zipfile_path)[](#l1.93)

   # Import something out of the zipfile and confirm it is correct.[](#l1.95)

   testmod = __import__(TESTPACK + "." + TESTMOD,[](#l1.96)

                        globals(), locals(), ["__dummy__"])[](#l1.97)

   self.assertEqual(testmod.test_value, 38)[](#l1.98)

   # Import something else out of the zipfile and confirm it is correct.[](#l1.99)

   ziptest_b = __import__("ziptest_b", globals(), locals(), ["test_value"])[](#l1.100)

   self.assertEqual(ziptest_b.test_value, 42)[](#l1.101)

   # Truncate and fill the zip file with non-zip garbage.[](#l1.103)

   with open(zipfile_path, "rb") as orig_zip_file:[](#l1.104)

       orig_zip_file_contents = orig_zip_file.read()[](#l1.105)

   with open(zipfile_path, "wb") as byebye_valid_zip_file:[](#l1.106)

       byebye_valid_zip_file.write(b"Tear down this wall!\n"*1987)[](#l1.107)

   # Now that the zipfile has been replaced, import something else from it[](#l1.108)

   # which should fail as the file contents are now garbage.[](#l1.109)

   with self.assertRaises(ImportError):[](#l1.110)

       ziptest_a = __import__("ziptest_a", globals(), locals(),[](#l1.111)

                              ["test_value"])[](#l1.112)

       self.assertEqual(ziptest_a.test_value, 23)[](#l1.113)

   # Now lets make it a valid zipfile that has some garbage at the start.[](#l1.115)

   # This alters all of the offsets within the file[](#l1.116)

   with open(zipfile_path, "wb") as new_zip_file:[](#l1.117)

       new_zip_file.write(b"X"*1991)  # The year Python was created.[](#l1.118)

       new_zip_file.write(orig_zip_file_contents)[](#l1.119)

   # Now that the zip file has been "restored" to a valid but different[](#l1.121)

   # zipfile the zipimporter should *successfully* re-read the new zip[](#l1.122)

   # file's end of file central index and be able to import from it again.[](#l1.123)

   ziptest_c = __import__("ziptest_c", globals(), locals(), ["test_value"])[](#l1.124)

   self.assertEqual(ziptest_c.test_value, 1337)[](#l1.125)

+ + class BadFileZipImportTestCase(unittest.TestCase): def assertZipFailure(self, filename):

   self.assertRaises(zipimport.ZipImportError,[](#l1.130)

                     zipimport.zipimporter, filename)[](#l1.131)

   with self.assertRaises(zipimport.ZipImportError):[](#l1.132)

       zipimport.zipimporter(filename)[](#l1.133)

def testNoFile(self): self.assertZipFailure('AdfjdkFJKDFJjdklfjs') @@ -499,6 +564,7 @@ def test_main(): UncompressedZipImportTestCase, CompressedZipImportTestCase, BadFileZipImportTestCase,

         ZipFileModifiedAfterImportTestCase,[](#l1.141)
       )[](#l1.142)

finally: support.unlink(TESTMOD)

--- a/Misc/NEWS +++ b/Misc/NEWS @@ -10,6 +10,11 @@ Release date: 2014-01-19 Core and Builtins ----------------- +- Issue #19081: When a zipimport .zip file in sys.path being imported from

is modified during the lifetime of the Python process after zipimport has
already cached the zip's table of contents we detect this and recover
rather than read bad data from the .zip (causing odd import errors). + Library -------

--- a/Modules/zipimport.c +++ b/Modules/zipimport.c @@ -49,10 +49,16 @@ struct _zipimporter { static PyObject ZipImportError; / read_directory() cache */ static PyObject *zip_directory_cache = NULL; +static PyObject zip_stat_cache = NULL; +/ posix.fstat or nt.fstat function. Used due to posixmodule.c's

- superior fstat implementation over libc's on Windows. */ +static PyObject fstat_function = NULL; / posix.fstat() or nt.fstat() / / forward decls */ -static PyObject *read_directory(PyObject *archive); -static PyObject *get_data(PyObject *archive, PyObject *toc_entry); +static FILE *fopen_rb_and_stat(PyObject *path, PyObject **py_stat_p); +static FILE *safely_reopen_archive(ZipImporter *self); +static PyObject *read_directory(FILE *fp, PyObject *archive); +static PyObject *get_data(FILE *fp, PyObject *archive, PyObject *toc_entry); static PyObject *get_module_code(ZipImporter *self, PyObject *fullname, int *p_ispackage, PyObject **p_modpath);

@@ -131,11 +137,39 @@ zipimporter_init(ZipImporter *self, PyOb files = PyDict_GetItem(zip_directory_cache, filename); if (files == NULL) {

   files = read_directory(filename);[](#l3.26)

```
   if (files == NULL)[](#l3.27)
```

```
   PyObject *zip_stat = NULL;[](#l3.28)
```

   FILE *fp = fopen_rb_and_stat(filename, &zip_stat);[](#l3.29)

```
   if (fp == NULL) {[](#l3.30)
```
```
       if (!PyErr_Occurred())[](#l3.31)
```

           PyErr_Format(ZipImportError, "can't open Zip file: %R",[](#l3.32)

                        filename);[](#l3.33)

       Py_XDECREF(zip_stat);[](#l3.35)
       goto error;[](#l3.36)

   if (PyDict_SetItem(zip_directory_cache, filename, files) != 0)[](#l3.37)

```
   }[](#l3.38)
```

```
   if (Py_VerboseFlag)[](#l3.40)
```

       PySys_FormatStderr("# zipimport: %U not cached, "[](#l3.41)

                          "reading TOC.\n", filename);[](#l3.42)

   files = read_directory(fp, filename);[](#l3.44)

```
   fclose(fp);[](#l3.45)
```
```
   if (files == NULL) {[](#l3.46)
```

       Py_XDECREF(zip_stat);[](#l3.47)
       goto error;[](#l3.48)

```
   }[](#l3.49)
```

   if (PyDict_SetItem(zip_directory_cache, filename, files) != 0) {[](#l3.50)

```
       Py_DECREF(files);[](#l3.51)
```
```
       Py_XDECREF(zip_stat);[](#l3.52)
```
```
       goto error;[](#l3.53)
```
```
   }[](#l3.54)
```

   if (zip_stat && PyDict_SetItem(zip_stat_cache, filename,[](#l3.55)

                                  zip_stat) != 0) {[](#l3.56)

```
       Py_DECREF(files);[](#l3.57)
```
```
       Py_DECREF(zip_stat);[](#l3.58)
```
```
       goto error;[](#l3.59)
```
```
   }[](#l3.60)
```
```
   Py_XDECREF(zip_stat);[](#l3.61)
```
} else Py_INCREF(files);

@@ -560,7 +594,8 @@ zipimporter_get_data(PyObject *obj, PyOb { ZipImporter *self = (ZipImporter *)obj; PyObject *path, *key;

PyObject *toc_entry;

FILE *fp;
PyObject *toc_entry, *data; Py_ssize_t path_start, path_len, len; if (!PyArg_ParseTuple(args, "U:zipimporter.get_data", &path)) @@ -588,15 +623,23 @@ zipimporter_get_data(PyObject *obj, PyOb key = PyUnicode_Substring(path, path_start, path_len); if (key == NULL) goto error;

fp = safely_reopen_archive(self);
if (fp == NULL)
```
   goto error;[](#l3.82)
```

+ toc_entry = PyDict_GetItem(self->files, key); if (toc_entry == NULL) { PyErr_SetFromErrnoWithFilenameObject(PyExc_IOError, key); Py_DECREF(key);

   fclose(fp);[](#l3.88)
   goto error;[](#l3.89)

} Py_DECREF(key); Py_DECREF(path);

return get_data(self->archive, toc_entry);

data = get_data(fp, self->archive, toc_entry);
fclose(fp);
return data;

error: Py_DECREF(path); return NULL; @@ -621,6 +664,7 @@ zipimporter_get_source(PyObject *obj, Py PyObject *toc_entry; PyObject *fullname, *subname, *path, *fullpath; enum zi_module_info mi;

FILE *fp;

if (!PyArg_ParseTuple(args, "U:zipimporter.get_source", &fullname)) return NULL; @@ -650,11 +694,18 @@ zipimporter_get_source(PyObject *obj, Py if (fullpath == NULL) return NULL;

fp = safely_reopen_archive(self);
if (fp == NULL) {
```
   Py_DECREF(fullpath);[](#l3.114)
```
```
   return NULL;[](#l3.115)
```
}

+ toc_entry = PyDict_GetItem(self->files, fullpath); Py_DECREF(fullpath); if (toc_entry != NULL) { PyObject *res, *bytes;

   bytes = get_data(self->archive, toc_entry);[](#l3.122)

   bytes = get_data(fp, self->archive, toc_entry);[](#l3.123)

   fclose(fp);[](#l3.124)
   if (bytes == NULL)[](#l3.125)
       return NULL;[](#l3.126)
   res = PyUnicode_FromStringAndSize(PyBytes_AS_STRING(bytes),[](#l3.127)

@@ -662,10 +713,10 @@ zipimporter_get_source(PyObject *obj, Py Py_DECREF(bytes); return res; }

fclose(fp);

/* we have the module, but no source */

Py_INCREF(Py_None);
return Py_None;

Py_RETURN_NONE;

} PyDoc_STRVAR(doc_find_module, @@ -831,10 +882,135 @@ get_long(unsigned char buf) { return x; } +/ Return 1 if objects a and b fail a Py_EQ test for an attr. */ +static int +compare_obj_attr_strings(PyObject *obj_a, PyObject *obj_b, char *attr_name) +{

int problem = 0;
PyObject *attr_a = PyObject_GetAttrString(obj_a, attr_name);
PyObject *attr_b = PyObject_GetAttrString(obj_b, attr_name);
if (attr_a == NULL || attr_b == NULL)
```
   problem = 1;[](#l3.153)
```
else

   problem = (PyObject_RichCompareBool(attr_a, attr_b, Py_EQ) != 1);[](#l3.155)

Py_XDECREF(attr_a);
Py_XDECREF(attr_b);
return problem;

+} + /*

read_directory(archive) -> files dict (new reference)

- Returns an open FILE * on success.
- Returns NULL on error with the Python error context set.
*/ +static FILE * +safely_reopen_archive(ZipImporter *self) +{
FILE *fp;
PyObject *stat_now = NULL;

fp = fopen_rb_and_stat(self->archive, &stat_now);
if (!fp) {

   PyErr_Format(ZipImportError,[](#l3.174)

                "zipimport: can not open file %U", self->archive);[](#l3.175)

```
   Py_XDECREF(stat_now);[](#l3.176)
```
```
   return NULL;[](#l3.177)
```
}

if (stat_now != NULL) {
```
   int problem = 0;[](#l3.181)
```
```
   PyObject *files;[](#l3.182)
```

   PyObject *prev_stat = PyDict_GetItem(zip_stat_cache, self->archive);[](#l3.183)

   /* Test stat_now vs the old cached stat on some key attributes. */[](#l3.184)

```
   if (prev_stat != NULL) {[](#l3.185)
```

       problem = compare_obj_attr_strings(prev_stat, stat_now,[](#l3.186)

                                          "st_ino");[](#l3.187)

       problem |= compare_obj_attr_strings(prev_stat, stat_now,[](#l3.188)

                                           "st_size");[](#l3.189)

       problem |= compare_obj_attr_strings(prev_stat, stat_now,[](#l3.190)

                                           "st_mtime");[](#l3.191)

```
   } else {[](#l3.192)
```
```
       if (Py_VerboseFlag)[](#l3.193)
```

           PySys_FormatStderr("# zipimport: no stat data for %U!\n",[](#l3.194)

                              self->archive);[](#l3.195)

```
       problem = 1;[](#l3.196)
```
```
   }[](#l3.197)
```

Given a path to a Zip archive, build a dict, mapping file names

```
   if (problem) {[](#l3.200)
```
```
       if (Py_VerboseFlag)[](#l3.201)
```

           PySys_FormatStderr("# zipimport: %U modified since last"[](#l3.202)

                              " import, rereading TOC.\n", self->archive);[](#l3.203)

       files = read_directory(fp, self->archive);[](#l3.204)

```
       if (files == NULL) {[](#l3.205)
```

           Py_DECREF(stat_now);[](#l3.206)

```
           fclose(fp);[](#l3.207)
```
```
           return NULL;[](#l3.208)
```
```
       }[](#l3.209)
```

       if (PyDict_SetItem(zip_directory_cache, self->archive,[](#l3.210)

                          files) != 0) {[](#l3.211)

```
           Py_DECREF(files);[](#l3.212)
```

           Py_DECREF(stat_now);[](#l3.213)

```
           fclose(fp);[](#l3.214)
```
```
           return NULL;[](#l3.215)
```
```
       }[](#l3.216)
```

       if (stat_now && PyDict_SetItem(zip_stat_cache, self->archive,[](#l3.217)

                                      stat_now) != 0) {[](#l3.218)

```
           Py_DECREF(files);[](#l3.219)
```

           Py_DECREF(stat_now);[](#l3.220)

```
           fclose(fp);[](#l3.221)
```
```
           return NULL;[](#l3.222)
```
```
       }[](#l3.223)
```

       Py_XDECREF(self->files);  /* free the old value. */[](#l3.224)

```
       self->files = files;[](#l3.225)
```
```
   } else {[](#l3.226)
```

       /* No problem, discard the new stat data. */[](#l3.227)

```
       Py_DECREF(stat_now);[](#l3.228)
```
```
   }[](#l3.229)
```
} /* stat succeeded */

return fp;

+} + +/*

fopen_rb_and_stat(path, &py_stat) -> FILE * +
Opens path in "rb" mode and populates the Python py_stat stat_result
with information about the opened file. *py_stat may not be changed
if there is no fstat_function or if fstat_function fails. +
Returns NULL and does nothing to py_stat if the open failed. +/ +static FILE * +fopen_rb_and_stat(PyObject *path, PyObject **py_stat_p) +{
FILE *fp;
assert(py_stat_p != NULL);
assert(*py_stat_p == NULL);

fp = _Py_fopen_obj(path, "rb");
if (fp == NULL) {
```
   if (!PyErr_Occurred())[](#l3.253)
```

       PyErr_Format(ZipImportError,[](#l3.254)

                    "zipimport: can not open file %U", path);[](#l3.255)

```
   return NULL;[](#l3.256)
```
}

if (fstat_function) {

   PyObject *stat_result = PyObject_CallFunction(fstat_function,[](#l3.260)

                                                 "i", fileno(fp));[](#l3.261)

   if (stat_result == NULL) {[](#l3.262)

       PyErr_Clear();  /* We can function without it. */[](#l3.263)

```
   } else {[](#l3.264)
```

       *py_stat_p = stat_result;[](#l3.265)

```
   }[](#l3.266)
```
}

return fp;

+} + +/*

read_directory(fp, archive) -> files dict (new reference) +
Given an open Zip archive, build a dict, mapping file names (local to the archive, using SEP as a separator) to toc entries. A toc_entry is a tuple: @@ -854,10 +1030,9 @@ get_long(unsigned char *buf) { data_size and file_offset are 0. */ static PyObject * -read_directory(PyObject *archive) +read_directory(FILE *fp, PyObject *archive) { PyObject *files = NULL;

FILE *fp; unsigned short flags; short compress, time, date, name_size; long crc, data_size, file_size, header_size; @@ -873,27 +1048,18 @@ read_directory(PyObject *archive) const char *charset; int bootstrap;
fp = _Py_fopen_obj(archive, "rb");
if (fp == NULL) {
```
   if (!PyErr_Occurred())[](#l3.297)
```

       PyErr_Format(ZipImportError, "can't open Zip file: %R", archive);[](#l3.298)

```
   return NULL;[](#l3.299)
```
}

assert(fp != NULL); if (fseek(fp, -22, SEEK_END) == -1) {

   fclose(fp);[](#l3.304)
   PyErr_Format(ZipImportError, "can't read Zip file: %R", archive);[](#l3.305)
   return NULL;[](#l3.306)

} header_position = ftell(fp); if (fread(endof_central_dir, 1, 22, fp) != 22) {

   fclose(fp);[](#l3.310)
   PyErr_Format(ZipImportError, "can't read Zip file: %R", archive);[](#l3.311)
   return NULL;[](#l3.312)

} if (get_long((unsigned char )endof_central_dir) != 0x06054B50) { / Bad: End of Central Dir signature */

   fclose(fp);[](#l3.316)
   PyErr_Format(ZipImportError, "not a Zip file: %R", archive);[](#l3.317)
   return NULL;[](#l3.318)

} @@ -1000,19 +1166,16 @@ read_directory(PyObject *archive) goto error; count++; }

fclose(fp); if (Py_VerboseFlag) PySys_FormatStderr("# zipimport: found %ld names in %R\n", count, archive); return files; file_error:
fclose(fp); Py_XDECREF(files); Py_XDECREF(nameobj); PyErr_Format(ZipImportError, "can't read Zip file: %R", archive); return NULL; error:
fclose(fp); Py_XDECREF(files); Py_XDECREF(nameobj); return NULL; @@ -1051,14 +1214,13 @@ get_decompress_func(void) return decompress; }

-/* Given a path to a Zip file and a toc_entry, return the (uncompressed) +/* Given a FILE* to a Zip file and a toc_entry, return the (uncompressed) data as a new reference. */ static PyObject * -get_data(PyObject *archive, PyObject *toc_entry) +get_data(FILE *fp, PyObject *archive, PyObject *toc_entry) { PyObject *raw_data, *data = NULL, *decompress; char *buf;

FILE *fp; int err; Py_ssize_t bytes_read = 0; long l; @@ -1072,17 +1234,8 @@ get_data(PyObject *archive, PyObject *to return NULL; }
fp = _Py_fopen_obj(archive, "rb");
if (!fp) {
```
   if (!PyErr_Occurred())[](#l3.363)
```

       PyErr_Format(PyExc_IOError,[](#l3.364)

          "zipimport: can not open file %U", archive);[](#l3.365)

```
   return NULL;[](#l3.366)
```
}

- /* Check to make sure the local file header is correct */ if (fseek(fp, file_offset, 0) == -1) {

   fclose(fp);[](#l3.371)
   PyErr_Format(ZipImportError, "can't read Zip file: %R", archive);[](#l3.372)
   return NULL;[](#l3.373)

} @@ -1094,11 +1247,9 @@ get_data(PyObject *archive, PyObject *to PyErr_Format(ZipImportError, "bad local file header in %U", archive);

   fclose(fp);[](#l3.379)
   return NULL;[](#l3.380)

} if (fseek(fp, file_offset + 26, 0) == -1) {

```
   fclose(fp);[](#l3.383)
   PyErr_Format(ZipImportError, "can't read Zip file: %R", archive);[](#l3.384)
   return NULL;[](#l3.385)
```
} @@ -1106,7 +1257,6 @@ get_data(PyObject *archive, PyObject to l = 30 + PyMarshal_ReadShortFromFile(fp) + PyMarshal_ReadShortFromFile(fp); / local header size */ if (PyErr_Occurred()) {
```
   fclose(fp);[](#l3.391)
   return NULL;[](#l3.392)
```
} file_offset += l; /* Start of file data */ @@ -1117,7 +1267,6 @@ get_data(PyObject *archive, PyObject *to raw_data = PyBytes_FromStringAndSize((char *)NULL, bytes_size); if (raw_data == NULL) {
```
   fclose(fp);[](#l3.399)
   return NULL;[](#l3.400)
```
} buf = PyBytes_AsString(raw_data); @@ -1126,11 +1275,9 @@ get_data(PyObject *archive, PyObject *to if (err == 0) { bytes_read = fread(buf, 1, data_size, fp); } else {

   fclose(fp);[](#l3.407)
   PyErr_Format(ZipImportError, "can't read Zip file: %R", archive);[](#l3.408)
   return NULL;[](#l3.409)

}

fclose(fp); if (err || bytes_read != data_size) { PyErr_SetString(PyExc_IOError, "zipimport: can't read data");

@@ -1351,12 +1498,12 @@ get_mtime_of_source(ZipImporter self, P / Return the code object for the module named by 'fullname' from the Zip archive as a new reference. */ static PyObject * -get_code_from_data(ZipImporter *self, int ispackage, int isbytecode, +get_code_from_data(ZipImporter *self, FILE *fp, int ispackage, int isbytecode, time_t mtime, PyObject *toc_entry) { PyObject *data, *modpath, *code;

data = get_data(self->archive, toc_entry);

data = get_data(fp, self->archive, toc_entry); if (data == NULL) return NULL;

@@ -1378,6 +1525,7 @@ get_module_code(ZipImporter *self, PyObj PyObject *code = NULL, *toc_entry, *subname; PyObject *path, *fullpath = NULL; struct st_zip_searchorder *zso;

FILE *fp;

subname = get_subname(fullname); if (subname == NULL) @@ -1388,6 +1536,12 @@ get_module_code(ZipImporter *self, PyObj if (path == NULL) return NULL;

fp = safely_reopen_archive(self);
if (fp == NULL) {
```
   Py_DECREF(path);[](#l3.444)
```
```
   return NULL;[](#l3.445)
```
}

+ for (zso = zip_searchorder; *zso->suffix; zso++) { code = NULL; @@ -1398,6 +1552,7 @@ get_module_code(ZipImporter *self, PyObj if (Py_VerboseFlag > 1) PySys_FormatStderr("# trying %U%c%U\n", self->archive, (int)SEP, fullpath); + toc_entry = PyDict_GetItem(self->files, fullpath); if (toc_entry != NULL) { time_t mtime = 0; @@ -1413,7 +1568,7 @@ get_module_code(ZipImporter *self, PyObj Py_CLEAR(fullpath); if (p_ispackage != NULL) *p_ispackage = ispackage;

       code = get_code_from_data(self, ispackage,[](#l3.463)

       code = get_code_from_data(self, fp, ispackage,[](#l3.464)
                                 isbytecode, mtime,[](#l3.465)
                                 toc_entry);[](#l3.466)
       if (code == Py_None) {[](#l3.467)

@@ -1433,6 +1588,7 @@ get_module_code(ZipImporter *self, PyObj } PyErr_Format(ZipImportError, "can't find module %R", fullname); exit:

fclose(fp); Py_DECREF(path); Py_XDECREF(fullpath); return code; @@ -1450,6 +1606,8 @@ This module exports three objects:\n[](#l3.476) subclass of ImportError, so it can be caught as ImportError, too.\n[](#l3.477)

_zip_directory_cache: a dict, mapping archive paths to zip directory\n[](#l3.478) info dicts, as used in zipimporter._files.\n[](#l3.479) +- _zip_stat_cache: a dict, mapping archive paths to stat_result\n[](#l3.480)

info for the .zip the last time anything was imported from it.\n[](#l3.481) \n[](#l3.482) It is usually not needed to use the zipimport module explicitly; it is\n[](#l3.483) used by the builtin import mechanism for sys.path items that are paths\n[](#l3.484) @@ -1509,6 +1667,7 @@ PyInit_zipimport(void) (PyObject *)&ZipImporter_Type) < 0) return NULL;

Py_XDECREF(zip_directory_cache); /* Avoid embedded interpreter leaks. */ zip_directory_cache = PyDict_New(); if (zip_directory_cache == NULL) return NULL;

@@ -1516,5 +1675,36 @@ PyInit_zipimport(void) if (PyModule_AddObject(mod, "_zip_directory_cache", zip_directory_cache) < 0) return NULL; +

Py_XDECREF(zip_stat_cache); /* Avoid embedded interpreter leaks. */
zip_stat_cache = PyDict_New();
if (zip_stat_cache == NULL)
```
   return NULL;[](#l3.501)
```
Py_INCREF(zip_stat_cache);
if (PyModule_AddObject(mod, "_zip_stat_cache", zip_stat_cache) < 0)
```
   return NULL;[](#l3.504)
```

{

   /* We cannot import "os" here as that is a .py/.pyc file that could[](#l3.507)

    * live within a zipped up standard library.  Import the posix or nt[](#l3.508)

    * builtin that provides the fstat() function we want instead. */[](#l3.509)

```
   PyObject *os_like_module;[](#l3.510)
```

   Py_CLEAR(fstat_function);  /* Avoid embedded interpreter leaks. */[](#l3.511)

   os_like_module = PyImport_ImportModule("posix");[](#l3.512)

   if (os_like_module == NULL) {[](#l3.513)

```
       PyErr_Clear();[](#l3.514)
```

       os_like_module = PyImport_ImportModule("nt");[](#l3.515)

```
   }[](#l3.516)
```

   if (os_like_module != NULL) {[](#l3.517)

       fstat_function = PyObject_GetAttrString(os_like_module, "fstat");[](#l3.518)

       Py_DECREF(os_like_module);[](#l3.519)

```
   }[](#l3.520)
```

   if (fstat_function == NULL) {[](#l3.521)

       PyErr_Clear();  /* non-fatal, we'll go on without it. */[](#l3.522)

```
       if (Py_VerboseFlag)[](#l3.523)
```

           PySys_WriteStderr("# zipimport unable to use os.fstat().\n");[](#l3.524)

```
   }[](#l3.525)
```
}

+ return mod; }