diff options
author | Martin Panter <vadmium+py@gmail.com> | 2016-11-20 22:17:44 +0000 |
---|---|---|
committer | Martin Panter <vadmium+py@gmail.com> | 2016-11-20 22:17:44 +0000 |
commit | 7d6e9232b142cd85a683a195f614bc2f09751934 (patch) | |
tree | 96937a5367a436b223b9a2610c7c963760560cd5 | |
parent | a57890e5ecb164c17b79aa2713758307787eb74b (diff) | |
parent | 04b35753f7a6bcf50f4a3676aca54cc961b93c83 (diff) | |
download | cpython-git-7d6e9232b142cd85a683a195f614bc2f09751934.tar.gz |
Issue #25659: Merge ctypes fix from 3.6
-rw-r--r-- | Lib/ctypes/test/test_frombuffer.py | 8 | ||||
-rw-r--r-- | Misc/NEWS | 3 | ||||
-rw-r--r-- | Modules/_ctypes/_ctypes.c | 12 |
3 files changed, 20 insertions, 3 deletions
diff --git a/Lib/ctypes/test/test_frombuffer.py b/Lib/ctypes/test/test_frombuffer.py index 29c5a19328..7ab38f1b03 100644 --- a/Lib/ctypes/test/test_frombuffer.py +++ b/Lib/ctypes/test/test_frombuffer.py @@ -120,5 +120,13 @@ class Test(unittest.TestCase): with self.assertRaises(ValueError): (c_int * 1).from_buffer_copy(a, 16 * sizeof(c_int)) + def test_abstract(self): + self.assertRaises(TypeError, Array.from_buffer, bytearray(10)) + self.assertRaises(TypeError, Structure.from_buffer, bytearray(10)) + self.assertRaises(TypeError, Union.from_buffer, bytearray(10)) + self.assertRaises(TypeError, Array.from_buffer_copy, b"123") + self.assertRaises(TypeError, Structure.from_buffer_copy, b"123") + self.assertRaises(TypeError, Union.from_buffer_copy, b"123") + if __name__ == '__main__': unittest.main() @@ -131,6 +131,9 @@ Core and Builtins Library ------- +- Issue #25659: In ctypes, prevent a crash calling the from_buffer() and + from_buffer_copy() methods on abstract classes like Array. + - Issue #28548: In the "http.server" module, parse the protocol version if possible, to avoid using HTTP 0.9 in some error responses. diff --git a/Modules/_ctypes/_ctypes.c b/Modules/_ctypes/_ctypes.c index de1ae3d0cb..0a9201e3f4 100644 --- a/Modules/_ctypes/_ctypes.c +++ b/Modules/_ctypes/_ctypes.c @@ -463,7 +463,10 @@ CDataType_from_buffer(PyObject *type, PyObject *args) Py_ssize_t offset = 0; StgDictObject *dict = PyType_stgdict(type); - assert (dict); + if (!dict) { + PyErr_SetString(PyExc_TypeError, "abstract class"); + return NULL; + } if (!PyArg_ParseTuple(args, "O|n:from_buffer", &obj, &offset)) return NULL; @@ -531,9 +534,12 @@ CDataType_from_buffer_copy(PyObject *type, PyObject *args) Py_ssize_t offset = 0; PyObject *result; StgDictObject *dict = PyType_stgdict(type); - assert (dict); + if (!dict) { + PyErr_SetString(PyExc_TypeError, "abstract class"); + return NULL; + } - if (!PyArg_ParseTuple(args, "y*|n:from_buffer", &buffer, &offset)) + if (!PyArg_ParseTuple(args, "y*|n:from_buffer_copy", &buffer, &offset)) return NULL; if (offset < 0) { |