pybind/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2026-07-16 08:41:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move code location

Browse Source
This commit is contained in:
Xuehai Pan
2026-01-23 03:04:08 +08:00
parent 740f69343c
commit d9227ce1c3
1 changed files with 25 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
include/pybind11/pybind11.h
View File

@@ -1692,30 +1692,6 @@ protected:
m_ptr = make_new_python_type(rec);
// Prevent use-after-free during interpreter shutdown. GC order is not guaranteed, so the
// internals capsule may be destroyed (resetting internals via internals_shutdown) before
// all pybind11 types are destroyed. If a type's tp_traverse/tp_clear then calls py::cast,
// it would recreate an empty internals and fail because the type registry is gone. By
// holding references to the capsules, we ensure they outlive all pybind11 types.
// We use weakrefs on the type with a cpp_function callback. When the type is destroyed,
// Python will call the callback which releases the capsule reference and the weakref.
if (PyObject *capsule = get_internals_capsule()) {
Py_INCREF(capsule);
(void) weakref(handle(m_ptr), cpp_function([](handle prevent_release) -> void {
Py_XDECREF(get_internals_capsule());
prevent_release.dec_ref();
}))
.release();
}
if (PyObject *capsule = get_local_internals_capsule()) {
Py_INCREF(capsule);
(void) weakref(handle(m_ptr), cpp_function([](handle prevent_release) -> void {
Py_XDECREF(get_local_internals_capsule());
prevent_release.dec_ref();
}))
.release();
}
/* Register supplemental type information in C++ dict */
auto *tinfo = new detail::type_info();
tinfo->type = reinterpret_cast<PyTypeObject *>(m_ptr);
@@ -1745,6 +1721,31 @@ protected:
#endif
}
// Prevent use-after-free during interpreter shutdown. GC order is not guaranteed, so
// the internals capsule may be destroyed (resetting internals via internals_shutdown)
// before all pybind11 types are destroyed. If a type's tp_traverse/tp_clear then calls
// py::cast, it would recreate an empty internals and fail because the type registry is
// gone. By holding references to the capsules, we ensure they outlive all pybind11
// types. We use weakrefs on the type with a cpp_function callback. When the type is
// destroyed, Python will call the callback which releases the capsule reference and
// the weakref.
if (PyObject *capsule = get_internals_capsule()) {
Py_INCREF(capsule);
(void) weakref(handle(m_ptr), cpp_function([](handle prevent_release) -> void {
Py_XDECREF(get_internals_capsule());
prevent_release.dec_ref();
}))
.release();
}
if (PyObject *capsule = get_local_internals_capsule()) {
Py_INCREF(capsule);
(void) weakref(handle(m_ptr), cpp_function([](handle prevent_release) -> void {
Py_XDECREF(get_local_internals_capsule());
prevent_release.dec_ref();
}))
.release();
}
PYBIND11_WARNING_PUSH
#if defined(__GNUC__) && __GNUC__ == 12
// When using GCC 12 these warnings are disabled as they trigger