From 42b1028abfebbfaa64ba7507ff4da69737e1accb Mon Sep 17 00:00:00 2001
From: Ian Jackson <ijackson@chiark.greenend.org.uk>
Date: Fri, 4 Nov 2016 15:37:54 +0000
Subject: [PATCH] generate fixes

---
 generate.c | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/generate.c b/generate.c
index 861d206..d5d457b 100644
--- 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;
 }
-- 
2.30.2