[Python] validate tensors info

[platform/upstream/nnstreamer.git] / ext / nnstreamer / extra / nnstreamer_python3_helper.cc

diff --git a/ext/nnstreamer/extra/nnstreamer_python3_helper.cc b/ext/nnstreamer/extra/nnstreamer_python3_helper.cc
index c8f730e..b2abc06 100644 (file)
--- a/ext/nnstreamer/extra/nnstreamer_python3_helper.cc
+++ b/ext/nnstreamer/extra/nnstreamer_python3_helper.cc
@@ -484,6 +484,15 @@ parseTensorsInfo (PyObject *result, GstTensorsInfo *info)
     Py_SAFEDECREF (shape_dims);
   }
 
+  /* Validate output tensors info after parsing python object. */
+  if (!gst_tensors_info_validate (info)) {
+    gchar *info_str = gst_tensors_info_to_string (info);
+    Py_ERRMSG ("Failed to parse the tensors information from python script, it may include invalid tensor type or dimension value (%s).",
+        info_str);
+    g_free (info_str);
+    return -EINVAL;
+  }
+
   return 0;
 }