Check input array size is positive

2021-01-12 15:30:41 +00:00 · 2021-01-12 15:30:41 +00:00 · 435a104f6e
commit 435a104f6e
parent 903eb40d19
1 changed files with 1 additions and 1 deletions
--- a/main.cpp
+++ b/main.cpp
@ -545,7 +545,7 @@ void parseArguments(int argc, char *argv[])
    else if (!std::string("--arraysize").compare(argv[i]) ||
             !std::string("-s").compare(argv[i]))
    {
-      if (++i >= argc || !parseInt(argv[i], &ARRAY_SIZE))
+      if (++i >= argc || !parseInt(argv[i], &ARRAY_SIZE) || ARRAY_SIZE <= 0)
      {
        std::cerr << "Invalid array size." << std::endl;
        exit(EXIT_FAILURE);