check nullptr in TargetMalloc and flatbuffers, test=develop

Author: Shixiaowei02

lite/backends/host/target_wrapper.cc

@@ -24,9 +24,7 @@ const int MALLOC_ALIGN = 64;
 void* TargetWrapper<TARGET(kHost)>::Malloc(size_t size) {
   size_t offset = sizeof(void*) + MALLOC_ALIGN - 1;
   char* p = static_cast<char*>(malloc(offset + size));
-  if (!p) {
-    return nullptr;
-  }
+  CHECK(p);
   void* r = reinterpret_cast<void*>(reinterpret_cast<size_t>(p + offset) &
                                     (~(MALLOC_ALIGN - 1)));
   static_cast<void**>(r)[-1] = p;
@@ -41,6 +39,8 @@ void TargetWrapper<TARGET(kHost)>::MemcpySync(void* dst,
                                               const void* src,
                                               size_t size,
                                               IoDirection dir) {
+  CHECK(dst);
+  CHECK(src);
   memcpy(dst, src, size);
 }
 

lite/model_parser/flatbuffers/io.cc

@@ -61,11 +61,13 @@ void SetParamWithTensor(const std::string& name,
 }
 
 void SetTensorWithParam(lite::Tensor* tensor, const ParamDescReadAPI& param) {
+  CHECK(tensor);
   tensor->Resize(param.Dim());
   tensor->set_precision(lite::ConvertPrecisionType(param.GetDataType()));
-  std::memcpy(tensor->mutable_data(param.byte_size()),
-              param.GetData(),
-              param.byte_size());
+  auto* dst = tensor->mutable_data(param.byte_size());
+  CHECK(dst);
+  CHECK(param.GetData());
+  std::memcpy(dst, param.GetData(), param.byte_size());
 }
 
 void SetCombinedParamsWithScope(const lite::Scope& scope,
@@ -82,9 +84,11 @@ void SetScopeWithCombinedParams(lite::Scope* scope,
                                 const CombinedParamsDescReadAPI& params) {
   CHECK(scope);
   for (size_t i = 0; i < params.GetParamsSize(); ++i) {
-    const auto& param = *params.GetParamDesc(i);
-    auto* tensor = scope->Var(param.Name())->GetMutable<lite::Tensor>();
-    SetTensorWithParam(tensor, param);
+    const auto* param = params.GetParamDesc(i);
+    CHECK(param);
+    auto* tensor = scope->Var(param->Name())->GetMutable<lite::Tensor>();
+    CHECK(tensor);
+    SetTensorWithParam(tensor, *param);
     tensor->set_persistable(true);
   }
 }

lite/model_parser/flatbuffers/param_desc.h

@@ -36,11 +36,17 @@ class ParamDescView : public ParamDescReadAPI {
     CHECK(desc_->variable_type() ==
           proto::ParamDesc_::VariableDesc_LoDTensorDesc);
     tensor_desc_ = desc_->variable_as<proto::ParamDesc_::LoDTensorDesc>();
+    CHECK(tensor_desc_);
+    CHECK(tensor_desc_->data());
+  }
+  std::string Name() const override {
+    CHECK(desc_->name());
+    return desc_->name()->c_str();
   }
-  std::string Name() const override { return desc_->name()->c_str(); }
 
   std::vector<int64_t> Dim() const override {
-    const auto& dims = tensor_desc_->dim();
+    const auto* dims = tensor_desc_->dim();
+    CHECK(dims);
     std::vector<int64_t> dims_vec;
     dims_vec.resize(dims->size());
     for (size_t i = 0; i < dims->size(); ++i) {
@@ -86,6 +92,8 @@ class CombinedParamsDescView : public CombinedParamsDescReadAPI {
 
   void InitParams() {
     desc_ = proto::GetCombinedParamsDesc(buf_.data());
+    CHECK(desc_);
+    CHECK(desc_->params());
     size_t params_size = desc_->params()->size();
     params_.resize(params_size);
     for (size_t idx = 0; idx < params_size; ++idx) {