From 83450fa5d56cafe06d64fcd5f73dd9ed79c95e60 Mon Sep 17 00:00:00 2001
From: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Date: Wed, 2 Oct 2019 16:05:44 +0100
Subject: 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
---
 src/armnnCaffeParser/RecordByRecordCaffeParser.cpp | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp b/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
index 592dc5c0de..cb7943655d 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);
 }
-
-
-
-- 
cgit v1.2.1