[Test] fix coverity 1050278, 1035696

check return result of mem map

Signed-off-by: Jaeyun Jung <jy1210.jung@samsung.com>

diff --git a/gst/nnstreamer/tensor_split/gsttensorsplit.c b/gst/nnstreamer/tensor_split/gsttensorsplit.c
index 57d2f83..4ed36f9 100644 (file)
--- a/gst/nnstreamer/tensor_split/gsttensorsplit.c
+++ b/gst/nnstreamer/tensor_split/gsttensorsplit.c
@@ -424,8 +424,8 @@ gst_get_splited_tensor (GstTensorSplit * split, GstBuffer * buffer, gint nth)
   size += gst_tensor_get_element_count (*dim) *
       tensor_element_size[split->sink_tensor_conf.info.type];
   mem = gst_allocator_alloc (NULL, size, NULL);
-  gst_memory_map (mem, &dest_info, GST_MAP_WRITE);
-  gst_buffer_map (buffer, &src_info, GST_MAP_READ);
+  g_assert (gst_memory_map (mem, &dest_info, GST_MAP_WRITE));
+  g_assert (gst_buffer_map (buffer, &src_info, GST_MAP_READ));
 
   for (i = 0; i < nth; i++) {
     dim = g_array_index (split->tensorseg, tensor_dim *, i);

diff --git a/tests/nnstreamer_sink/unittest_sink.cpp b/tests/nnstreamer_sink/unittest_sink.cpp
index 49248a5..1f1525a 100644 (file)
--- a/tests/nnstreamer_sink/unittest_sink.cpp
+++ b/tests/nnstreamer_sink/unittest_sink.cpp
@@ -341,7 +341,13 @@ _push_text_data (const guint num_buffers, const gboolean timestamps = TRUE)
     GstBuffer *buf = gst_buffer_new_allocate (NULL, 10, NULL);
     GstMapInfo info;
 
-    gst_buffer_map (buf, &info, GST_MAP_WRITE);
+    if (!gst_buffer_map (buf, &info, GST_MAP_WRITE)) {
+      _print_log ("failed to get mem map");
+      g_test_data.test_failed = failed = TRUE;
+      gst_buffer_unref (buf);
+      goto error;
+    }
+
     snprintf ((char *) info.data, 10, "%d", i);
     gst_buffer_unmap (buf, &info);
 
@@ -353,6 +359,7 @@ _push_text_data (const guint num_buffers, const gboolean timestamps = TRUE)
     if (gst_app_src_push_buffer (GST_APP_SRC (appsrc), buf) != GST_FLOW_OK) {
       _print_log ("failed to push buffer [%d]", i);
       g_test_data.test_failed = failed = TRUE;
+      gst_buffer_unref (buf);
       goto error;
     }
   }