Add checks on input

This commit is contained in:
David Husicka 2021-12-14 22:52:53 +01:00
parent 53a277821e
commit 22746fc3fb
1 changed files with 8 additions and 2 deletions

View File

@ -16,12 +16,18 @@ int main(int argc, char *argv[]) {
}
if (!strcmp(argv[i], "--x")) {
if (argc >= i) {
size_x = atoi(argv[i + 1]);
if(sscanf(argv[i+1], "%d", &size_x) != 1){
printf("invalid input\n");
return 1;
}
}
}
if (!strcmp(argv[i], "--y")) {
if (argc >= i) {
size_y = atoi(argv[i + 1]);
if(sscanf(argv[i+1], "%d", &size_y) != 1){
printf("invalid input\n");
return 1;
}
}
}
}