pykms: keep Card alive until Blob is gone

1 files changed, 7 insertions, 5 deletions

diff --git a/py/pykms/pykmsbase.cpp b/py/pykms/pykmsbase.cpp
index aae5ece..96ebe7a 100644
--- a/py/pykms/pykmsbase.cpp
+++ b/py/pykms/pykmsbase.cpp
@@ -105,12 +105,14 @@ void init_pykmsbase(py::module &m)
 
 	py::class_<Blob>(m, "Blob")
 			.def("__init__", [](Blob& instance, Card& card, py::buffer buf) {
-				py::buffer_info info = buf.request();
-				if (info.ndim != 1)
-					throw std::runtime_error("Incompatible buffer dimension!");
+					py::buffer_info info = buf.request();
+					if (info.ndim != 1)
+						throw std::runtime_error("Incompatible buffer dimension!");
+
+					new (&instance) Blob(card, info.ptr, info.size * info.itemsize);
+				},
+				py::keep_alive<1, 3>())	// Keep Card alive until this is destructed
 
-				new (&instance) Blob(card, info.ptr, info.size * info.itemsize);
-			})
 			.def_property_readonly("data", &Blob::data)
 
 			// XXX pybind11 doesn't support a base object (DrmObject) with custom holder-type,