Issue #10916: mmap should not segfault when a file is mapped using 0 as

length and a non-zero offset, and an attempt to read past the end of file
is made (IndexError is raised instead).  Patch by Ross Lagerwall.

Requested by Georg.
This commit is contained in:
Antoine Pitrou 2011-01-15 16:17:07 +00:00
parent f5cf435bb5
commit 85f4615500
3 changed files with 18 additions and 1 deletions

View File

@ -320,6 +320,19 @@ class MmapTests(unittest.TestCase):
mf.close() mf.close()
f.close() f.close()
def test_length_0_offset(self):
# Issue #10916: test mapping of remainder of file by passing 0 for
# map length with an offset doesn't cause a segfault.
if not hasattr(os, "stat"):
self.skipTest("needs os.stat")
with open(TESTFN, "wb+") as f:
f.write(49152 * b'm') # Arbitrary character
with open(TESTFN, "rb") as f:
mf = mmap.mmap(f.fileno(), 0, offset=40960, access=mmap.ACCESS_READ)
self.assertRaises(IndexError, mf.__getitem__, 45000)
mf.close()
def test_move(self): def test_move(self):
# make move works everywhere (64-bit format problem earlier) # make move works everywhere (64-bit format problem earlier)
f = open(TESTFN, 'wb+') f = open(TESTFN, 'wb+')

View File

@ -43,6 +43,10 @@ Core and Builtins
Library Library
------- -------
- Issue #10916: mmap should not segfault when a file is mapped using 0 as
length and a non-zero offset, and an attempt to read past the end of file
is made (IndexError is raised instead). Patch by Ross Lagerwall.
- Issue #10907: Warn OS X 10.6 IDLE users to use ActiveState Tcl/Tk 8.5, - Issue #10907: Warn OS X 10.6 IDLE users to use ActiveState Tcl/Tk 8.5,
rather than the currently problematic Apple-supplied one, when running rather than the currently problematic Apple-supplied one, when running
with the 64-/32-bit installer variant. with the 64-/32-bit installer variant.

View File

@ -1116,7 +1116,7 @@ new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict)
# endif # endif
if (fd != -1 && fstat(fd, &st) == 0 && S_ISREG(st.st_mode)) { if (fd != -1 && fstat(fd, &st) == 0 && S_ISREG(st.st_mode)) {
if (map_size == 0) { if (map_size == 0) {
map_size = st.st_size; map_size = st.st_size - offset;
} else if ((size_t)offset + (size_t)map_size > st.st_size) { } else if ((size_t)offset + (size_t)map_size > st.st_size) {
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
"mmap length is greater than file size"); "mmap length is greater than file size");