fix: add life support to handles cast to string_view (#6092)

* Fix handling of string_view to prevent GC'ing strings before they are used.

* style: pre-commit fixes

* Update cast.h

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Charles Beattie
2026-06-24 02:44:39 +01:00
committed by GitHub
parent da3bdcad04
commit 59d7cb28c1
3 changed files with 31 additions and 0 deletions

View File

@@ -526,6 +526,9 @@ struct string_caster {
return false;
}
value = StringType(buffer, static_cast<size_t>(size));
if (IsView) {
loader_life_support::add_patient(src);
}
return true;
}
@@ -603,6 +606,9 @@ private:
pybind11_fail("Unexpected PYBIND11_BYTES_AS_STRING() failure.");
}
value = StringType(bytes, (size_t) PYBIND11_BYTES_SIZE(src.ptr()));
if (IsView) {
loader_life_support::add_patient(src);
}
return true;
}
if (PyByteArray_Check(src.ptr())) {
@@ -613,6 +619,9 @@ private:
pybind11_fail("Unexpected PyByteArray_AsString() failure.");
}
value = StringType(bytearray, (size_t) PyByteArray_Size(src.ptr()));
if (IsView) {
loader_life_support::add_patient(src);
}
return true;
}

View File

@@ -582,6 +582,16 @@ TEST_SUBMODULE(stl, m) {
[](const std::list<std::string> &) { return 2; });
m.def("func_with_string_or_vector_string_arg_overload", [](const std::string &) { return 3; });
#ifdef PYBIND11_HAS_STRING_VIEW
m.def("func_with_string_views", [](const std::vector<std::string_view> &svs) {
py::list l;
for (std::string_view sv : svs) {
l.append(sv);
}
return l;
});
#endif
class Placeholder {
public:
Placeholder() { print_created(this); }

View File

@@ -28,6 +28,18 @@ def test_vector(doc):
# Test regression caused by 936: pointers to stl containers weren't castable
assert m.cast_ptr_vector() == ["lvalue", "lvalue"]
if hasattr(m, "func_with_string_views"):
def gen():
return ("a" + str(x) for x in range(10000, 10010))
expected = list(gen())
assert m.func_with_string_views(gen()) == expected
assert m.func_with_string_views(x.encode() for x in gen()) == expected
assert (
m.func_with_string_views(bytearray(x.encode()) for x in gen()) == expected
)
def test_deque():
"""std::deque <-> list"""