[VP] Fix gstreamer ximagesrc segfault

make the memcpy size smaller than surface size.
intel · Sep 3, 2024 · 722793d · 722793d
1 parent 347eff0
commit 722793d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/media_driver/linux/common/ddi/media_libva.cpp b/media_driver/linux/common/ddi/media_libva.cpp
@@ -5615,7 +5615,15 @@ VAStatus DdiMedia_PutImage(
         }
 
         //Copy data from image to temp surferce
-        MOS_STATUS eStatus = MOS_SecureMemcpy(tempSurfData, tempMediaSurface->data_size, imageData, vaimg->data_size);
+        MOS_STATUS eStatus = MOS_STATUS_SUCCESS;
+        if (tempMediaSurface->data_size >= vaimg->data_size)
+        {
+            eStatus = MOS_SecureMemcpy(tempSurfData, tempMediaSurface->data_size, imageData, vaimg->data_size);
+        }
+        else
+        {
+            eStatus = MOS_SecureMemcpy(tempSurfData, tempMediaSurface->data_size, imageData, tempMediaSurface->data_size);
+        }
         if (eStatus != MOS_STATUS_SUCCESS)
         {
             DDI_ASSERTMESSAGE("Failed to copy image to surface buffer.");