safe VisionBuf life time via client owner reference

Author: deanlee

msgq/visionipc/visionipc_pyx.pxd

@@ -10,6 +10,7 @@ cdef class CLContext:
 
 cdef class VisionBuf:
   cdef cppVisionBuf * buf
+  cdef object _owner
 
   @staticmethod
-  cdef create(cppVisionBuf*)
+  cdef create(cppVisionBuf*, object)

msgq/visionipc/visionipc_pyx.pyx

@@ -29,12 +29,17 @@ cpdef enum VisionStreamType:
 
 
 cdef class VisionBuf:
+
   @staticmethod
-  cdef create(cppVisionBuf * cbuf):
+  cdef create(cppVisionBuf * cbuf, object owner):
     buf = VisionBuf()
     buf.buf = cbuf
+    buf._owner = owner  # Keep a reference to the owner to manage lifetime
     return buf
 
+  def __dealloc__(self):
+    self._owner = None # explicitly release the reference
+
   @property
   def data(self):
     return np.asarray(<cnp.uint8_t[:self.buf.len]> self.buf.addr)
@@ -155,7 +160,7 @@ cdef class VisionIpcClient:
     buf = self.client.recv(&self.extra, timeout_ms)
     if not buf:
       return None
-    return VisionBuf.create(buf)
+    return VisionBuf.create(buf, self)
 
   def connect(self, bool blocking):
     return self.client.connect(blocking)