diff --git a/dpu.c b/dpu.c index aa6fc80..d699d85 100755 --- a/dpu.c +++ b/dpu.c @@ -1,3 +1,4 @@ +#include #include #include #include @@ -7,7 +8,13 @@ int main (int argc, char **argv){ FILE *bob,*joe,*moe; int thing, factor, arg; if(argc >= 3){ - factor = atoi (argv[1]); + char *end_char; + factor = strtol(argv[1], &end_char, 0); + if(!*(argv[1]) || *end_char) + { + fputs("I can haz number?\n", stderr); + return EXIT_FAILURE; + } arg = 2; }else{ factor = 9;