Fix wrong iterator usage. Fix potential crash.

UnickSoft · Dec 25, 2024 · f038ff4 · f038ff4
1 parent 1ee94db
commit f038ff4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/algorithm/MaxFlowPushRelableImpl.h b/algorithm/MaxFlowPushRelableImpl.h
@@ -385,7 +385,7 @@ template<class WeightTypeInterface, typename WeightType> void MaxFlowPushRelabel
 
     _flowValue.erase(last, _flowValue.end());
 
-    for(auto element = _flowValue.begin(); element != last; element++)
+    for(auto element = _flowValue.begin(); element != _flowValue.end(); element++)
     {
         element->value = 0;
         for (auto comapare = flowValueCopy.begin(); comapare != flowValueCopy.end(); comapare++)

diff --git a/common/ConsoleParams.cpp b/common/ConsoleParams.cpp
@@ -29,6 +29,10 @@ void ConsoleParams::ParseCommandLine (const std::vector<String>& params, Paramet
             {
                 commands[GetRealParamName(*param)] = *(param + 1);
             }
+            else
+            {
+                break;
+            }
         }
     }
 }
-Original file line number
+Diff line change
@@ Expand Up @@
                 {
                     commands[GetRealParamName(*param)] = *(param + 1);
                 }
+                else
+                {
+                    break;
+                }
             }
         }
     }
@@ Expand Down @@