Commit 20f0ea1f authored by Jesus Cea's avatar Jesus Cea

#15676: mmap: add empty file check prior to offset check <- Previous patch was incomplete

parent 3fb774ec
...@@ -469,17 +469,14 @@ class MmapTests(unittest.TestCase): ...@@ -469,17 +469,14 @@ class MmapTests(unittest.TestCase):
def test_empty_file (self): def test_empty_file (self):
f = open (TESTFN, 'w+b') f = open (TESTFN, 'w+b')
f.close() f.close()
f = open(TESTFN, "rb") with open(TESTFN, "rb") as f :
try: try:
m = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ) m = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ)
m.close() m.close()
f.close()
self.fail("should not have been able to mmap empty file") self.fail("should not have been able to mmap empty file")
except ValueError as e: except ValueError as e:
f.close()
self.assertEqual(e.message, "cannot mmap an empty file") self.assertEqual(e.message, "cannot mmap an empty file")
except: except:
f.close()
self.fail("unexpected exception: " + str(e)) self.fail("unexpected exception: " + str(e))
def test_offset (self): def test_offset (self):
......
...@@ -1388,6 +1388,11 @@ new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict) ...@@ -1388,6 +1388,11 @@ new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict)
} }
size = (((PY_LONG_LONG) high) << 32) + low; size = (((PY_LONG_LONG) high) << 32) + low;
if (size == 0) {
PyErr_SetString(PyExc_ValueError,
"cannot mmap an empty file");
return NULL;
}
if (offset >= size) { if (offset >= size) {
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
"mmap offset is greater than file size"); "mmap offset is greater than file 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