Fix #8105. Add validation to mmap.mmap so invalid file descriptors

don't cause a crash on Windows.
This commit is contained in:
Brian Curtin 2010-08-01 15:26:26 +00:00
parent 0bccc185b4
commit ea47eaa395
3 changed files with 19 additions and 1 deletions

View File

@ -1,6 +1,6 @@
from test.support import TESTFN, run_unittest, import_module
import unittest
import os, re, itertools
import os, re, itertools, socket
# Skip test if we can't import mmap.
mmap = import_module('mmap')
@ -586,6 +586,17 @@ class MmapTests(unittest.TestCase):
pass
m.close()
def test_invalid_descriptor(self):
# socket file descriptors are valid, but out of range
# for _get_osfhandle, causing a crash when validating the
# parameters to _get_osfhandle.
s = socket.socket()
try:
with self.assertRaises(mmap.error):
m = mmap.mmap(s.fileno(), 10)
finally:
s.close()
def test_context_manager(self):
with mmap.mmap(-1, 10) as m:
self.assertFalse(m.closed)

View File

@ -21,6 +21,8 @@ Core and Builtins
Extensions
----------
- Issue #8105: Validate file descriptor passed to mmap.mmap on Windows.
- Issue #8046: Add context manager protocol support and .closed property
to mmap objects.

View File

@ -1236,6 +1236,11 @@ new_mmap_object(PyTypeObject *type, PyObject *args, PyObject *kwdict)
1);
*/
if (fileno != -1 && fileno != 0) {
/* Ensure that fileno is within the CRT's valid range */
if (_PyVerify_fd(fileno) == 0) {
PyErr_SetFromErrno(mmap_module_error);
return NULL;
}
fh = (HANDLE)_get_osfhandle(fileno);
if (fh==(HANDLE)-1) {
PyErr_SetFromErrno(mmap_module_error);