closes bpo-28557: error message for bad raw readinto (GH-7496)
Co-authored-by: Benjamin Peterson <benjamin@python.org>
This commit is contained in:
parent
04fc4f2a46
commit
8666356280
|
@ -1587,6 +1587,22 @@ class CBufferedReaderTest(BufferedReaderTest, SizeofTest):
|
||||||
with self.assertRaisesRegex(TypeError, "BufferedReader"):
|
with self.assertRaisesRegex(TypeError, "BufferedReader"):
|
||||||
self.tp(io.BytesIO(), 1024, 1024, 1024)
|
self.tp(io.BytesIO(), 1024, 1024, 1024)
|
||||||
|
|
||||||
|
def test_bad_readinto_value(self):
|
||||||
|
rawio = io.BufferedReader(io.BytesIO(b"12"))
|
||||||
|
rawio.readinto = lambda buf: -1
|
||||||
|
bufio = self.tp(rawio)
|
||||||
|
with self.assertRaises(OSError) as cm:
|
||||||
|
bufio.readline()
|
||||||
|
self.assertIsNone(cm.exception.__cause__)
|
||||||
|
|
||||||
|
def test_bad_readinto_type(self):
|
||||||
|
rawio = io.BufferedReader(io.BytesIO(b"12"))
|
||||||
|
rawio.readinto = lambda buf: b''
|
||||||
|
bufio = self.tp(rawio)
|
||||||
|
with self.assertRaises(OSError) as cm:
|
||||||
|
bufio.readline()
|
||||||
|
self.assertIsInstance(cm.exception.__cause__, TypeError)
|
||||||
|
|
||||||
|
|
||||||
class PyBufferedReaderTest(BufferedReaderTest):
|
class PyBufferedReaderTest(BufferedReaderTest):
|
||||||
tp = pyio.BufferedReader
|
tp = pyio.BufferedReader
|
||||||
|
|
|
@ -1678,6 +1678,7 @@ Péter Szabó
|
||||||
John Szakmeister
|
John Szakmeister
|
||||||
Piotr Szczepaniak
|
Piotr Szczepaniak
|
||||||
Amir Szekely
|
Amir Szekely
|
||||||
|
David Szotten
|
||||||
Maciej Szulik
|
Maciej Szulik
|
||||||
Joel Taddei
|
Joel Taddei
|
||||||
Arfrever Frehtes Taifersar Arahesis
|
Arfrever Frehtes Taifersar Arahesis
|
||||||
|
|
|
@ -0,0 +1 @@
|
||||||
|
Improve the error message for a misbehaving ``rawio.readinto``
|
|
@ -1483,6 +1483,15 @@ _bufferedreader_raw_read(buffered *self, char *start, Py_ssize_t len)
|
||||||
}
|
}
|
||||||
n = PyNumber_AsSsize_t(res, PyExc_ValueError);
|
n = PyNumber_AsSsize_t(res, PyExc_ValueError);
|
||||||
Py_DECREF(res);
|
Py_DECREF(res);
|
||||||
|
|
||||||
|
if (n == -1 && PyErr_Occurred()) {
|
||||||
|
_PyErr_FormatFromCause(
|
||||||
|
PyExc_OSError,
|
||||||
|
"raw readinto() failed"
|
||||||
|
);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
if (n < 0 || n > len) {
|
if (n < 0 || n > len) {
|
||||||
PyErr_Format(PyExc_OSError,
|
PyErr_Format(PyExc_OSError,
|
||||||
"raw readinto() returned invalid length %zd "
|
"raw readinto() returned invalid length %zd "
|
||||||
|
|
Loading…
Reference in New Issue