Check input array size is positive

This commit is contained in:
Tom Deakin 2021-01-12 15:30:41 +00:00
parent 903eb40d19
commit 435a104f6e

View File

@ -545,7 +545,7 @@ void parseArguments(int argc, char *argv[])
else if (!std::string("--arraysize").compare(argv[i]) || else if (!std::string("--arraysize").compare(argv[i]) ||
!std::string("-s").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; std::cerr << "Invalid array size." << std::endl;
exit(EXIT_FAILURE); exit(EXIT_FAILURE);