summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Lib/test/test_ctypes/test_parameters.py52
-rw-r--r--Misc/NEWS.d/next/Library/2022-12-11-14-38-59.gh-issue-99952.IYGLzr.rst2
-rw-r--r--Modules/_ctypes/_ctypes.c3
-rw-r--r--Modules/_ctypes/_ctypes_test.c6
4 files changed, 63 insertions, 0 deletions
diff --git a/Lib/test/test_ctypes/test_parameters.py b/Lib/test/test_ctypes/test_parameters.py
index 22d290d..06cc951 100644
--- a/Lib/test/test_ctypes/test_parameters.py
+++ b/Lib/test/test_ctypes/test_parameters.py
@@ -266,6 +266,58 @@ class SimpleTypesTestCase(unittest.TestCase):
self.assertRegex(repr(c_wchar_p.from_param('hihi')), r"^<cparam 'Z' \(0x[A-Fa-f0-9]+\)>$")
self.assertRegex(repr(c_void_p.from_param(0x12)), r"^<cparam 'P' \(0x0*12\)>$")
+ @test.support.cpython_only
+ def test_from_param_result_refcount(self):
+ # Issue #99952
+ import _ctypes_test
+ from ctypes import PyDLL, c_int, c_void_p, py_object, Structure
+
+ class X(Structure):
+ """This struct size is <= sizeof(void*)."""
+ _fields_ = [("a", c_void_p)]
+
+ def __del__(self):
+ trace.append(4)
+
+ @classmethod
+ def from_param(cls, value):
+ trace.append(2)
+ return cls()
+
+ PyList_Append = PyDLL(_ctypes_test.__file__)._testfunc_pylist_append
+ PyList_Append.restype = c_int
+ PyList_Append.argtypes = [py_object, py_object, X]
+
+ trace = []
+ trace.append(1)
+ PyList_Append(trace, 3, "dummy")
+ trace.append(5)
+
+ self.assertEqual(trace, [1, 2, 3, 4, 5])
+
+ class Y(Structure):
+ """This struct size is > sizeof(void*)."""
+ _fields_ = [("a", c_void_p), ("b", c_void_p)]
+
+ def __del__(self):
+ trace.append(4)
+
+ @classmethod
+ def from_param(cls, value):
+ trace.append(2)
+ return cls()
+
+ PyList_Append = PyDLL(_ctypes_test.__file__)._testfunc_pylist_append
+ PyList_Append.restype = c_int
+ PyList_Append.argtypes = [py_object, py_object, Y]
+
+ trace = []
+ trace.append(1)
+ PyList_Append(trace, 3, "dummy")
+ trace.append(5)
+
+ self.assertEqual(trace, [1, 2, 3, 4, 5])
+
################################################################
if __name__ == '__main__':
diff --git a/Misc/NEWS.d/next/Library/2022-12-11-14-38-59.gh-issue-99952.IYGLzr.rst b/Misc/NEWS.d/next/Library/2022-12-11-14-38-59.gh-issue-99952.IYGLzr.rst
new file mode 100644
index 0000000..09ec961
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2022-12-11-14-38-59.gh-issue-99952.IYGLzr.rst
@@ -0,0 +1,2 @@
+Fix a reference undercounting issue in :class:`ctypes.Structure` with ``from_param()``
+results larger than a C pointer.
diff --git a/Modules/_ctypes/_ctypes.c b/Modules/_ctypes/_ctypes.c
index 272cafb..8690f2c 100644
--- a/Modules/_ctypes/_ctypes.c
+++ b/Modules/_ctypes/_ctypes.c
@@ -412,6 +412,7 @@ _ctypes_alloc_format_string_with_shape(int ndim, const Py_ssize_t *shape,
typedef struct {
PyObject_HEAD
void *ptr;
+ PyObject *keep; // If set, a reference to the original CDataObject.
} StructParamObject;
@@ -419,6 +420,7 @@ static void
StructParam_dealloc(PyObject *myself)
{
StructParamObject *self = (StructParamObject *)myself;
+ Py_XDECREF(self->keep);
PyMem_Free(self->ptr);
Py_TYPE(self)->tp_free(myself);
}
@@ -466,6 +468,7 @@ StructUnionType_paramfunc(CDataObject *self)
StructParamObject *struct_param = (StructParamObject *)obj;
struct_param->ptr = ptr;
+ struct_param->keep = Py_NewRef(self);
} else {
ptr = self->b_ptr;
obj = Py_NewRef(self);
diff --git a/Modules/_ctypes/_ctypes_test.c b/Modules/_ctypes/_ctypes_test.c
index e1f91b4..a8811d0 100644
--- a/Modules/_ctypes/_ctypes_test.c
+++ b/Modules/_ctypes/_ctypes_test.c
@@ -1047,6 +1047,12 @@ EXPORT(long) _test_i38748_runCallback(_test_i38748_funcType callback, int a, int
#endif
+EXPORT(int)
+_testfunc_pylist_append(PyObject *list, PyObject *item)
+{
+ return PyList_Append(list, item);
+}
+
static struct PyModuleDef_Slot _ctypes_test_slots[] = {
{0, NULL}
};