Plug another leak, and finally add a test for #1174606 (read() from /dev/zero).

The leak was the reason my previous attempts at testing failed...
This commit is contained in:
Antoine Pitrou 2009-03-29 18:55:12 +00:00
parent 8e21fb2c69
commit 7d037a7b52
2 changed files with 17 additions and 0 deletions

View File

@ -505,6 +505,22 @@ def test_garbage_collection(self):
with open(support.TESTFN, "rb") as f:
self.assertEqual(f.read(), b"abcxxx")
def test_unbounded_file(self):
# Issue #1174606: reading from an unbounded stream such as /dev/zero.
zero = "/dev/zero"
if not os.path.exists(zero):
raise unittest.SkipTest("{0} does not exist".format(zero))
if sys.maxsize > 0x7FFFFFFF:
raise unittest.SkipTest("test can only run in a 32-bit address space")
if support.real_max_memuse < support._2G:
raise unittest.SkipTest("test requires at least 2GB of memory")
with open(zero, "rb", buffering=0) as f:
self.assertRaises(OverflowError, f.read)
with open(zero, "rb") as f:
self.assertRaises(OverflowError, f.read)
with open(zero, "r") as f:
self.assertRaises(OverflowError, f.read)
class CIOTest(IOTest):
pass

View File

@ -543,6 +543,7 @@ fileio_readall(PyFileIOObject *self)
PyErr_SetString(PyExc_OverflowError,
"unbounded read returned more bytes "
"than a Python string can hold ");
Py_DECREF(result);
return NULL;
}