Fixes #224

FLAMEGPU · Apr 8, 2020 · caeb660 · caeb660
1 parent 23af5f7
commit caeb660
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/flamegpu/sim/Simulation.cu b/src/flamegpu/sim/Simulation.cu
@@ -84,16 +84,28 @@ int Simulation::checkArgs(int argc, const char** argv) {
         std::transform(arg.begin(), arg.end(), arg.begin(), [](unsigned char c) { return std::use_facet< std::ctype<char>>(std::locale()).tolower(c); });
         // -in <string>, Specifies the input state file
         if (arg.compare("--in") == 0 || arg.compare("-i") == 0) {
+            if (i + 1 >= argc) {
+                fprintf(stderr, "%s requires a trailing argument\n", arg.c_str());
+                return false;
+            }
             config.xml_input_file = std::string(argv[++i]);
             continue;
         }
         // -steps <uint>, The number of steps to be executed
         if (arg.compare("--steps") == 0 || arg.compare("-s") == 0) {
+            if (i + 1 >= argc) {
+                fprintf(stderr, "%s requires a trailing argument\n", arg.c_str());
+                return false;
+            }
             config.steps = static_cast<int>(strtoul(argv[++i], nullptr, 0));
             continue;
         }
         // -random <uint>, Uses the specified random seed, defaults to clock
         if (arg.compare("--random") == 0 || arg.compare("-r") == 0) {
+            if (i + 1 >= argc) {
+                fprintf(stderr, "%s requires a trailing argument\n", arg.c_str());
+                return false;
+            }
             // Reinitialise RandomManager state
             config.random_seed = static_cast<unsigned int>(strtoul(argv[++i], nullptr, 0));
             continue;