diff options
author | Hirokazu Yamamoto <ocean-city@m2.ccsnet.ne.jp> | 2009-02-28 10:31:54 (GMT) |
---|---|---|
committer | Hirokazu Yamamoto <ocean-city@m2.ccsnet.ne.jp> | 2009-02-28 10:31:54 (GMT) |
commit | f2dc885780278246d1a4572c822c4387142a84df (patch) | |
tree | ec5d9d135bb0fcaa050665b98508c26186c0d362 /Lib/test/test_mmap.py | |
parent | 7334735ee17951e4ee5555e1183ae4f269fe04af (diff) | |
download | cpython-f2dc885780278246d1a4572c822c4387142a84df.zip cpython-f2dc885780278246d1a4572c822c4387142a84df.tar.gz cpython-f2dc885780278246d1a4572c822c4387142a84df.tar.bz2 |
Issue #5386: mmap.write_byte didn't check map size, so it could cause buffer
overrun.
Diffstat (limited to 'Lib/test/test_mmap.py')
-rw-r--r-- | Lib/test/test_mmap.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/Lib/test/test_mmap.py b/Lib/test/test_mmap.py index c06b757..c8e52a6 100644 --- a/Lib/test/test_mmap.py +++ b/Lib/test/test_mmap.py @@ -467,6 +467,38 @@ class MmapTests(unittest.TestCase): self.assert_(issubclass(mmap.error, EnvironmentError)) self.assert_("mmap.error" in str(mmap.error)) + def test_io_methods(self): + data = "0123456789" + open(TESTFN, "wb").write("x"*len(data)) + f = open(TESTFN, "r+b") + m = mmap.mmap(f.fileno(), len(data)) + f.close() + # Test write_byte() + for i in xrange(len(data)): + self.assertEquals(m.tell(), i) + m.write_byte(data[i:i+1]) + self.assertEquals(m.tell(), i+1) + self.assertRaises(ValueError, m.write_byte, "x") + self.assertEquals(m[:], data) + # Test read_byte() + m.seek(0) + for i in xrange(len(data)): + self.assertEquals(m.tell(), i) + self.assertEquals(m.read_byte(), data[i:i+1]) + self.assertEquals(m.tell(), i+1) + self.assertRaises(ValueError, m.read_byte) + # Test read() + m.seek(3) + self.assertEquals(m.read(3), "345") + self.assertEquals(m.tell(), 6) + # Test write() + m.seek(3) + m.write("bar") + self.assertEquals(m.tell(), 6) + self.assertEquals(m[:], "012bar6789") + m.seek(8) + self.assertRaises(ValueError, m.write, "bar") + def test_main(): run_unittest(MmapTests) |