# HG changeset patch # Parent d9bb7a3ed51e86afe13836272944fbd42424c54d Issue #25498: Fix GC crash due to ctypes objects wrapping a memoryview This was a regression caused by revision 1da9630e9b7f. Based on patch by Eryksun. diff -r d9bb7a3ed51e Lib/ctypes/test/test_frombuffer.py --- a/Lib/ctypes/test/test_frombuffer.py Tue Oct 27 22:05:09 2015 -0500 +++ b/Lib/ctypes/test/test_frombuffer.py Wed Nov 11 04:27:01 2015 +0000 @@ -55,6 +55,12 @@ with self.assertRaises(ValueError): (c_int * 1).from_buffer(a, 16 * sizeof(c_int)) + def test_from_buffer_memoryview(self): + a = [c_char.from_buffer(memoryview(bytearray(b'a')))] + a.append(a) + del a + gc.collect() # Should not crash + def test_from_buffer_copy(self): a = array.array("i", range(16)) x = (c_int * 16).from_buffer_copy(a) diff -r d9bb7a3ed51e Misc/ACKS --- a/Misc/ACKS Tue Oct 27 22:05:09 2015 -0500 +++ b/Misc/ACKS Wed Nov 11 04:27:01 2015 +0000 @@ -399,6 +399,7 @@ Stoffel Erasmus Jürgen A. Erhard Michael Ernst +Eryksun Ben Escoto Andy Eskilsson André Espaze diff -r d9bb7a3ed51e Misc/NEWS --- a/Misc/NEWS Tue Oct 27 22:05:09 2015 -0500 +++ b/Misc/NEWS Wed Nov 11 04:27:01 2015 +0000 @@ -10,6 +10,10 @@ Core and Builtins ----------------- +- Issue #25498: Fix a crash when garbage-collecting ctypes objects created + by wrapping a memoryview. This was a regression made in 3.4.3. Based + on patch by Eryksun. + - Issue #25401: Optimize bytes.fromhex() and bytearray.fromhex(): they are now between 2x and 3.5x faster. diff -r d9bb7a3ed51e Modules/_ctypes/_ctypes.c --- a/Modules/_ctypes/_ctypes.c Tue Oct 27 22:05:09 2015 -0500 +++ b/Modules/_ctypes/_ctypes.c Wed Nov 11 04:27:01 2015 +0000 @@ -463,45 +463,65 @@ static PyObject * CDataType_from_buffer(PyObject *type, PyObject *args) { - Py_buffer buffer; + PyObject *obj; + PyObject *mv; + PyObject *result; + Py_buffer *buffer; Py_ssize_t offset = 0; - PyObject *result, *mv; + StgDictObject *dict = PyType_stgdict(type); assert (dict); - if (!PyArg_ParseTuple(args, "w*|n:from_buffer", &buffer, &offset)) + if (!PyArg_ParseTuple(args, "O|n:from_buffer", &obj, &offset)) return NULL; + mv = PyMemoryView_FromObject(obj); + if (mv == NULL) + return NULL; + + buffer = PyMemoryView_GET_BUFFER(mv); + + if (buffer->readonly) { + PyErr_SetString(PyExc_TypeError, + "underlying buffer is not writable"); + Py_DECREF(mv); + return NULL; + } + + if (!PyBuffer_IsContiguous(buffer, 'C')) { + PyErr_SetString(PyExc_TypeError, + "underlying buffer is not C contiguous"); + Py_DECREF(mv); + return NULL; + } + if (offset < 0) { PyErr_SetString(PyExc_ValueError, "offset cannot be negative"); - PyBuffer_Release(&buffer); + Py_DECREF(mv); return NULL; } - if (dict->size > buffer.len - offset) { + + if (dict->size > buffer->len - offset) { PyErr_Format(PyExc_ValueError, - "Buffer size too small (%zd instead of at least %zd bytes)", - buffer.len, dict->size + offset); - PyBuffer_Release(&buffer); + "Buffer size too small " + "(%zd instead of at least %zd bytes)", + buffer->len, dict->size + offset); + Py_DECREF(mv); return NULL; } - result = PyCData_AtAddress(type, (char *)buffer.buf + offset); + result = PyCData_AtAddress(type, (char *)buffer->buf + offset); if (result == NULL) { - PyBuffer_Release(&buffer); + Py_DECREF(mv); return NULL; } - mv = PyMemoryView_FromBuffer(&buffer); - if (mv == NULL) { - PyBuffer_Release(&buffer); + if (-1 == KeepRef((CDataObject *)result, -1, mv)) { + Py_DECREF(result); return NULL; } - /* Hack the memoryview so that it will release the buffer. */ - ((PyMemoryViewObject *)mv)->mbuf->master.obj = buffer.obj; - ((PyMemoryViewObject *)mv)->view.obj = buffer.obj; - if (-1 == KeepRef((CDataObject *)result, -1, mv)) - result = NULL; + return result; }