progs/demos: Bounds check input to fire.c.

This commit is contained in:
Vinson Lee 2010-01-10 00:40:46 -08:00
parent 9e98c1fbf6
commit 0c7814fe23
1 changed files with 7 additions and 1 deletions

View File

@ -726,8 +726,13 @@ main(int ac, char **av)
maxage = 1.0 / dt;
if (ac == 2)
if (ac == 2) {
np = atoi(av[1]);
if (np <= 0 || np > 1000000) {
fprintf(stderr, "Invalid input.\n");
exit(-1);
}
}
if (ac == 4) {
WIDTH = atoi(av[2]);
@ -762,6 +767,7 @@ main(int ac, char **av)
assert(np > 0);
p = (part *) malloc(sizeof(part) * np);
assert(p);
for (i = 0; i < np; i++)
setnewpart(&p[i]);