From: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Date: Wed, 2 Oct 2019 15:05:44 +0000 (+0100)
Subject: IVGCVSW-3941 Fix bug around file opening in RecordByRecordCaffeParser
X-Git-Tag: submit/tizen/20200316.035456~203
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=83450fa5d56cafe06d64fcd5f73dd9ed79c95e60;p=platform%2Fupstream%2Farmnn.git

IVGCVSW-3941 Fix bug around file opening in RecordByRecordCaffeParser

* Added error check after attempting to create input file stream and
  throw FileNotFoundException in case of failure

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I46b77e21c5330bb60e1190ddbefa6473e43085d9
---

diff --git a/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp b/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
index 592dc5c..cb79436 100644
--- a/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
+++ b/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
@@ -474,8 +474,12 @@ armnn::INetworkPtr RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(
     }
     m_RequestedOutputs = requestedOutputs;
 
-    //FILE * fp = fopen(graphFile, "rb");
     std::ifstream ifs(graphFile, std::ifstream::in|std::ifstream::binary);
+    if (ifs.fail())
+    {
+        throw armnn::FileNotFoundException("Failed to open graph file '" + std::string(graphFile) + "'");
+    }
+
     std::vector<LayerParameterInfo> layerInfo;
     NetParameterInfo netParameterInfo;
     while(true)
@@ -727,6 +731,3 @@ armnn::INetworkPtr RecordByRecordCaffeParser::LoadLayers(std::ifstream& ifs,
 
     return move(m_Network);
 }
-
-
-