From 0b53c4566f2e2d7592271dbe1d82e1d3948b2610 Mon Sep 17 00:00:00 2001 From: MyungJoo Ham Date: Wed, 19 Sep 2018 17:10:52 +0900 Subject: [PATCH] [Transform/Transpose] Check the validity of dimensions If it is < 0 or >= 4, it is not valid and breaks the buffer. Supposed to Fix #552 ps. @jijoongmoon : You do not have unit testcases for transpose! CC: Jijoong Moon Signed-off-by: MyungJoo Ham --- gst/tensor_transform/tensor_transform.c | 2 ++ 1 file changed, 2 insertions(+) diff --git a/gst/tensor_transform/tensor_transform.c b/gst/tensor_transform/tensor_transform.c index ca3c230..9abbd99 100644 --- a/gst/tensor_transform/tensor_transform.c +++ b/gst/tensor_transform/tensor_transform.c @@ -965,6 +965,8 @@ gst_tensor_dimension_conversion (GstTensor_Transform * filter, } } else { for (i = 0; i < NNS_TENSOR_RANK_LIMIT; i++) { + g_assert (filter->data_transpose.trans_order[i] >= 0 && + filter->data_transpose.trans_order[i] < NNS_TENSOR_RANK_LIMIT); destDim[filter->data_transpose.trans_order[i]] = srcDim[i]; } } -- 2.7.4