va: allocator: Fix possible memory leaks
authorMengkejiergeli Ba <mengkejiergeli.ba@intel.com>
Tue, 12 Oct 2021 09:32:30 +0000 (17:32 +0800)
committerMengkejiergeli Ba <mengkejiergeli.ba@intel.com>
Mon, 25 Oct 2021 07:03:30 +0000 (15:03 +0800)
At gst_va_dmabuf_allocator_setup_buffer_full, static code analysis tool
does not know number of objects in descriptor is always larger than 0 if
export_surface_to_dmabuf succeeds. Thus, the tool will assume buf is
allocated with mem but not released when desc.num_objects equals to 0
and raise a mem leak issue.

For gst_va_dambuf_memories_setup, we should also inform the tool that
n_planes will be larger than 0 by checking the value at very beginning.
Then, the defect similar to above will not be raised during static analysis.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/1241>

subprojects/gst-plugins-bad/sys/va/gstvaallocator.c

index dff8cfa..87d6802 100644 (file)
@@ -541,6 +541,11 @@ gst_va_dmabuf_allocator_setup_buffer_full (GstAllocator * allocator,
     goto failed;
   }
 
+  if (desc.num_objects == 0) {
+    GST_ERROR ("Failed to export surface to dmabuf");
+    goto failed;
+  }
+
   buf = gst_va_buffer_surface_new (surface, format, desc.width, desc.height);
   if (G_UNLIKELY (info)) {
     *info = self->info;
@@ -811,7 +816,8 @@ gst_va_dmabuf_memories_setup (GstVaDisplay * display, GstVideoInfo * info,
   gboolean ret;
 
   g_return_val_if_fail (GST_IS_VA_DISPLAY (display), FALSE);
-  g_return_val_if_fail (n_planes <= GST_VIDEO_MAX_PLANES, FALSE);
+  g_return_val_if_fail (n_planes > 0
+      && n_planes <= GST_VIDEO_MAX_PLANES, FALSE);
 
   format = GST_VIDEO_INFO_FORMAT (info);
   if (format == GST_VIDEO_FORMAT_UNKNOWN)