mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-04 08:59:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

Issue #25021: Correctly make sure that product.__setstate__ does not access

Browse source 
invalid memory.
This commit is contained in:
Kristján Valur Jónsson 2015-09-12 15:20:54 +00:00
parent a82f77fb00
commit 102764a1f6
2 changed files with 20 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
Lib/test/test_itertools.py
View file

@ -959,6 +959,16 @@ class TestBasicOps(unittest.TestCase):
self.assertEqual(list(copy.deepcopy(product(*args))), result)
self.pickletest(product(*args))
def test_product_issue_25021(self):
# test that indices are properly clamped to the length of the tuples
p = product((1, 2),(3,))
p.__setstate__((0, 0x1000)) # will access tuple element 1 if not clamped
self.assertEqual(next(p), (2, 3))
# test that empty tuple in the list will result in an immediate StopIteration
p = product((1, 2), (), (3,))
p.__setstate__((0, 0, 0x1000)) # will access tuple element 1 if not clamped
self.assertRaises(StopIteration, next, p)
def test_repeat(self):
self.assertEqual(list(repeat(object='a', times=3)), ['a', 'a', 'a'])
self.assertEqual(lzip(range(3),repeat('a')),