diff --git a/undpu.c b/undpu.c index 1aa6c33..4c880aa 100644 --- a/undpu.c +++ b/undpu.c @@ -57,10 +57,12 @@ int main (int argc, char **argv){ return EXIT_FAILURE; } } - strlcpy(output,argv[1],strlen(argv[1])-4); - if((joe = fopen(output,"w"))==NULL){ - fputs("Ummm, the output file doesn't work.\n",stderr); - return EXIT_FAILURE; + if(!out){ + strlcpy(output,argv[0],strlen(argv[0])-4); + if((joe = fopen(output,"w"))==NULL){ + fputs("Ummm, the output file doesn't work.\n",stderr); + return EXIT_FAILURE; + } } while((thing=fgetc(bob))!=EOF){ fputc(thing,joe);