handle an error case to GetBackendCapacity correctly 10/233110/1
authorInki Dae <inki.dae@samsung.com>
Tue, 12 May 2020 06:16:20 +0000 (15:16 +0900)
committerInki Dae <inki.dae@samsung.com>
Tue, 12 May 2020 07:41:23 +0000 (16:41 +0900)
Change-Id: I86f633f9b8b43a375b80e662d081a5537890e796
Signed-off-by: Inki Dae <inki.dae@samsung.com>
src/inference_engine_common_impl.cpp

index a5b03a8dd511365abce8b7c1b4b9ca4876c34a57..d49e074c82de00e06728bbe7245e0902bcc0a80e 100755 (executable)
@@ -358,6 +358,11 @@ int InferenceEngineCommon::SetOutputLayerProperty(inference_engine_layer_propert
 
 int InferenceEngineCommon::GetBackendCapacity(inference_engine_capacity *capacity)
 {
+       if (capacity == nullptr) {
+               LOGE("Given inference_engine_capacity object is invalid.");
+               return INFERENCE_ENGINE_ERROR_INVALID_PARAMETER;
+       }
+
     return mBackendHandle->GetBackendCapacity(capacity);
 }