mirror of
https://github.com/python/cpython.git
synced 2025-07-15 23:35:23 +00:00
Issue #24467: Fixed possible buffer over-read in bytearray. The bytearray
object now always allocates place for trailing null byte and it's buffer now is always null-terminated.
This commit is contained in:
commit
bc9e75ed02
3 changed files with 25 additions and 2 deletions
|
@ -1098,10 +1098,27 @@ class ByteArrayTest(BaseBytesTest, unittest.TestCase):
|
|||
for i in range(100):
|
||||
b += b"x"
|
||||
alloc = b.__alloc__()
|
||||
self.assertTrue(alloc >= len(b))
|
||||
self.assertGreater(alloc, len(b)) # including trailing null byte
|
||||
if alloc not in seq:
|
||||
seq.append(alloc)
|
||||
|
||||
def test_init_alloc(self):
|
||||
b = bytearray()
|
||||
def g():
|
||||
for i in range(1, 100):
|
||||
yield i
|
||||
a = list(b)
|
||||
self.assertEqual(a, list(range(1, len(a)+1)))
|
||||
self.assertEqual(len(b), len(a))
|
||||
self.assertLessEqual(len(b), i)
|
||||
alloc = b.__alloc__()
|
||||
self.assertGreater(alloc, len(b)) # including trailing null byte
|
||||
b.__init__(g())
|
||||
self.assertEqual(list(b), list(range(1, 100)))
|
||||
self.assertEqual(len(b), 99)
|
||||
alloc = b.__alloc__()
|
||||
self.assertGreater(alloc, len(b))
|
||||
|
||||
def test_extend(self):
|
||||
orig = b'hello'
|
||||
a = bytearray(orig)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue