From 3e014090c7bcfdf650d0c48df54f757f4432ee5f Mon Sep 17 00:00:00 2001 From: Jaeyun Jung Date: Mon, 18 Dec 2023 19:11:33 +0900 Subject: [PATCH] [CodeClean] fix svace issue Check null ptr when getting n-th tensor info. Signed-off-by: Jaeyun Jung --- c/src/ml-api-inference-internal.c | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/c/src/ml-api-inference-internal.c b/c/src/ml-api-inference-internal.c index efd78d3..8de3378 100644 --- a/c/src/ml-api-inference-internal.c +++ b/c/src/ml-api-inference-internal.c @@ -28,6 +28,10 @@ gst_info_is_extended (const GstTensorsInfo * gst_info) for (i = 0; i < gst_info->num_tensors; i++) { _info = gst_tensors_info_get_nth_info ((GstTensorsInfo *) gst_info, i); + if (!_info) + _ml_error_report_return (FALSE, + "The parameter, gst_info, has invalid number of tensors. The max number of tensors is " + NNS_TENSOR_SIZE_LIMIT_STR); if (_info->dimension[ML_TENSOR_RANK_LIMIT_PREV] > 0) return TRUE; @@ -55,8 +59,8 @@ _ml_tensors_info_create_from_gst (ml_tensors_info_h * ml_info, is_extended = gst_info_is_extended (gst_info); if (is_extended) - _ml_error_report_return_continue_iferr (ml_tensors_info_create_extended - (ml_info), + _ml_error_report_return_continue_iferr + (ml_tensors_info_create_extended (ml_info), "The call to ml_tensors_info_create_extended has failed with %d.", _ERRNO); else -- 2.7.4