Kaydet (Commit) 8a0eede2 authored tarafından Antoine Pitrou's avatar Antoine Pitrou

Merged revisions 88131 via svnmerge from

svn+ssh://pythondev@svn.python.org/python/branches/py3k

........
  r88131 | antoine.pitrou | 2011-01-20 22:07:24 +0100 (jeu., 20 janv. 2011) | 6 lines

  Issue #10955: Fix a potential crash when trying to mmap() a file past its
  length.  Initial patch by Ross Lagerwall.

  This fixes a regression introduced by r88022.
........
üst 7e8fbd2b
...@@ -343,6 +343,19 @@ class MmapTests(unittest.TestCase): ...@@ -343,6 +343,19 @@ class MmapTests(unittest.TestCase):
finally: finally:
mf.close() mf.close()
def test_length_0_large_offset(self):
# Issue #10959: test mapping of a file by passing 0 for
# map length with a large offset doesn't cause a segfault.
if not hasattr(os, "stat"):
self.skipTest("needs os.stat")
with open(TESTFN, "wb") as f:
f.write(115699 * b'm') # Arbitrary character
with open(TESTFN, "w+b") as f:
self.assertRaises(ValueError, mmap.mmap, f.fileno(), 0,
offset=2147418112)
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, 'w+') f = open(TESTFN, 'w+')
......
...@@ -35,6 +35,9 @@ Core and Builtins ...@@ -35,6 +35,9 @@ Core and Builtins
Library Library
------- -------
- Issue #10955: Fix a potential crash when trying to mmap() a file past its
length. Initial patch by Ross Lagerwall.
- Issue #10898: Allow compiling the posix module when the C library defines - Issue #10898: Allow compiling the posix module when the C library defines
a symbol named FSTAT. a symbol named FSTAT.
......
...@@ -1164,6 +1164,11 @@ new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict) ...@@ -1164,6 +1164,11 @@ 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) {
if (offset >= st.st_size) {
PyErr_SetString(PyExc_ValueError,
"mmap offset is greater than file size");
return NULL;
}
map_size = st.st_size - offset; 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,
...@@ -1346,6 +1351,12 @@ new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict) ...@@ -1346,6 +1351,12 @@ new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict)
else else
m_obj->size = low; m_obj->size = low;
#endif #endif
if (offset >= m_obj->size) {
PyErr_SetString(PyExc_ValueError,
"mmap offset is greater than file size");
Py_DECREF(m_obj);
return NULL;
}
m_obj->size -= offset; m_obj->size -= offset;
} else { } else {
m_obj->size = map_size; m_obj->size = map_size;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment