Skip to content

Commit

Permalink
bpo-39033: Fix NameError in zipimport during hash validation (pythonG…
Browse files Browse the repository at this point in the history
…H-17588)


Patch by Karthikeyan Singaravelan.
  • Loading branch information
tirkarthi authored and shihai1991 committed Jan 31, 2020
1 parent d6e64b3 commit 2cd1637
Show file tree
Hide file tree
Showing 5 changed files with 286 additions and 269 deletions.
16 changes: 16 additions & 0 deletions Lib/test/test_zipimport.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
import struct
import time
import unittest
import unittest.mock

from test import support

Expand Down Expand Up @@ -204,6 +205,21 @@ def check(mod):
self.assertEqual(mod.state, 'old')
self.doTest(None, files, TESTMOD, call=check)

@unittest.mock.patch('_imp.check_hash_based_pycs', 'always')
def test_checked_hash_based_change_pyc(self):
source = b"state = 'old'"
source_hash = importlib.util.source_hash(source)
bytecode = importlib._bootstrap_external._code_to_hash_pyc(
compile(source, "???", "exec"),
source_hash,
False,
)
files = {TESTMOD + ".py": (NOW, "state = 'new'"),
TESTMOD + ".pyc": (NOW - 20, bytecode)}
def check(mod):
self.assertEqual(mod.state, 'new')
self.doTest(None, files, TESTMOD, call=check)

def testEmptyPy(self):
files = {TESTMOD + ".py": (NOW, "")}
self.doTest(None, files, TESTMOD)
Expand Down
2 changes: 1 addition & 1 deletion Lib/zipimport.py
Original file line number Diff line number Diff line change
Expand Up @@ -608,7 +608,7 @@ def _unmarshal_code(self, pathname, fullpath, fullname, data):
)

try:
_boostrap_external._validate_hash_pyc(
_bootstrap_external._validate_hash_pyc(
data, source_hash, fullname, exc_details)
except ImportError:
return None
Expand Down
1 change: 1 addition & 0 deletions Misc/ACKS
Original file line number Diff line number Diff line change
Expand Up @@ -1901,3 +1901,4 @@ Tim Hopper
Dan Lidral-Porter
Ngalim Siregar
Tim Gates
Karthikeyan Singaravelan
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Fix :exc:`NameError` in :mod:`zipimport`. Patch by Karthikeyan Singaravelan.
Loading

0 comments on commit 2cd1637

Please sign in to comment.