[Coverity] #1839950 Catch uncaught exception

Catch a `std::bad_array_new_length` that can be risen when array indexing Signed-off-by: Daekyoung Jung <[email protected]>
nnstreamer · Feb 11, 2025 · e0ec5e1 · e0ec5e1
1 parent dc06517
commit e0ec5e1
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/Applications/ReinforcementLearning/DeepQ/jni/main.cpp b/Applications/ReinforcementLearning/DeepQ/jni/main.cpp
@@ -433,8 +433,14 @@ int main(int argc, char **argv) {
            * @brief     Generate Lable with next state
            */
           for (unsigned int i = 0; i < in_Exp.size(); i++) {
-            STATE state = in_Exp[i].state;
-            STATE next_state = in_Exp[i].next_state;
+            STATE state, next_state;
+            try {
+              state = in_Exp[i].state;
+              next_state = in_Exp[i].next_state;
+            } catch (std::bad_array_new_length &e) {
+              std::cerr << "Error during state assignment" << std::endl;
+              return -1;
+            }
             std::vector<float> in(state.observation.begin(),
                                   state.observation.end());
             inbatch.push_back({{in}});