generate fixes

diff --git a/generate.c b/generate.c
index 861d2061e31484067c7738c668deecaa2aa3a71b..d5d457bc78cacd0f3ceca135ed690fc04e9d7af8 100644 (file)
--- a/generate.c
+++ b/generate.c
@@ -1,6 +1,7 @@
 
 #include <stdio.h>
 #include <stdlib.h>
+#include <assert.h>
 
 static const char charset[]=
   "0123456789"
@@ -25,13 +26,18 @@ static void gen(char *inbuf, int len) {
   }
 }    
 
-int main(void) {
+int main(int argc, char **argv) {
   int len;
+  int maxlen = argc>=2 ? atoi(argv[1]) : ML;
 
-  for (len=0; len<ML; len++) {
+  assert(maxlen <= ML);
+
+  for (len=0; len<maxlen; len++) {
     fprintf(stderr,"********** generating len=%d **********\n",len);
     gen(sofar, len);
   }
   if (ferror(stdout) ||
       fflush(stdout)) { perror("stdout"); abort(); }
+  fputs("generate done\n",stderr);
+  return 0;
 }