From 853f5db07bf46397ce8d9eb0d8f918f9b8f9a692 Mon Sep 17 00:00:00 2001 From: xuanyue Date: Mon, 9 Aug 2021 16:59:36 +0800 Subject: [PATCH] code check --- mindspore/lite/src/ops/populate/adder_populate.cc | 8 +++++++- .../lite/src/ops/populate/constant_of_shape_populate.cc | 2 ++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/src/ops/populate/adder_populate.cc b/mindspore/lite/src/ops/populate/adder_populate.cc index 5b41e4f5ae7..894434ef590 100644 --- a/mindspore/lite/src/ops/populate/adder_populate.cc +++ b/mindspore/lite/src/ops/populate/adder_populate.cc @@ -42,7 +42,13 @@ OpParameter *PopulateAdderParameter(const void *prim) { auto pad_list = value->pad_list(); auto dilation = value->dilation(); if (kernel_size == nullptr || stride == nullptr || pad_list == nullptr || dilation == nullptr) { - MS_LOG(ERROR) << "nullptr"; + MS_LOG(ERROR) << "exist attr is nullptr"; + free(param); + return nullptr; + } + if (kernel_size->size() < kMinShapeSizeTwo || stride->size() < kMinShapeSizeTwo || + pad_list->size() < kMinShapeSizeFour || dilation->size() < kMinShapeSizeTwo) { + MS_LOG(ERROR) << "exist attr size is invalid."; free(param); return nullptr; } diff --git a/mindspore/lite/src/ops/populate/constant_of_shape_populate.cc b/mindspore/lite/src/ops/populate/constant_of_shape_populate.cc index 25e721dee69..097b0780395 100644 --- a/mindspore/lite/src/ops/populate/constant_of_shape_populate.cc +++ b/mindspore/lite/src/ops/populate/constant_of_shape_populate.cc @@ -47,6 +47,8 @@ OpParameter *PopulateConstantOfShapeParameter(const void *prim) { param->data_type_ = static_cast(value->data_type()); if (val.empty() || val.size() > 1) { MS_LOG(ERROR) << "The value of constant of shape is empty or more than 1."; + free(param); + return nullptr; } else { switch (param->data_type_) { case kNumberTypeFloat32: