From: Donghak PARK <donghak.park@samsung.com>
Date: Thu, 6 Feb 2025 01:19:14 +0000 (+0900)
Subject: [Coverity] Fix Coverity issue at KNN Application
X-Git-Tag: accepted/tizen/7.0/unified/20250213.155121~6
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=59d234eb0aee8858491372ef94a5b51b3c18eefc;p=platform%2Fcore%2Fml%2Fnntrainer.git

[Coverity] Fix Coverity issue at KNN Application

Uncaught exception (UNCAUGHT_EXCEPT)
- root_function: In function main(int, char **) an exception of type std::bad_cast is thrown and never caught
- to fix issue, add try catch statement at writefile

**Self evaluation:**
1. Build test:	 [X]Passed [ ]Failed [ ]Skipped
2. Run test:	 [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: Donghak PARK <donghak.park@samsung.com>
---

diff --git a/Applications/KNN/jni/main.cpp b/Applications/KNN/jni/main.cpp
index 8f74c10a..e42dd24f 100644
--- a/Applications/KNN/jni/main.cpp
+++ b/Applications/KNN/jni/main.cpp
@@ -207,11 +207,18 @@ int main(int argc, char *argv[]) {
       std::string out_file = "/sdcard/Transfer-Learning/";
       out_file += total_label[i] + std::to_string(j + 1) + ".txt";
       printf("%s\n", out_file.c_str());
-      std::ofstream writeFile(out_file.data());
-      if (writeFile.is_open()) {
-        for (int k = 0; k < 128; k++)
-          writeFile << out[i][j][k] << std::endl;
-        writeFile.close();
+      try {
+        std::ofstream writeFile(out_file.data());
+        if (writeFile.is_open()) {
+          for (int k = 0; k < 128; k++)
+            writeFile << out[i][j][k] << std::endl;
+          writeFile.close();
+        } else {
+          throw std::runtime_error("Failed to open file : " + out_file);
+        }
+      } catch (const std::exception &e) {
+        std::cerr << "Exception caught while writing to file : " << e.what()
+                  << std::endl;
       }
     }
   }