--memo-limit - limit failed attempts to find new generated requests to N

libpathod/cmdline.py

@@ -34,6 +34,10 @@ def go_pathoc():
         metavar = "HOST:PORT",
         help="Issue an HTTP CONNECT to connect to the specified host."
     )
+    parser.add_argument(
+        "--memo-limit", dest='memolimit', default=5000, type=int, metavar="N",
+        help='Stop if we do not find a valid request after N attempts.'
+    )
     parser.add_argument(
         "-m", dest='memo', action="store_true", default=False,
         help="""

libpathod/pathoc.py

@@ -222,9 +222,14 @@ class Pathoc(tcp.TCPClient):
 
 def main(args):
     memo = set([])
+    trycount = 0
     try:
         cnt = 0
         while 1:
+            if trycount > args.memolimit:
+                print >> sys.stderr, "Memo limit exceeded..."
+                return
+
             cnt += 1
             if args.random:
                 playlist = [random.choice(args.requests)]
@@ -251,8 +256,10 @@ def main(args):
                         newlist.append(spec)
                 playlist = newlist
             if not playlist:
+                trycount += 1
                 continue
 
+            trycount = 0
             try:
                 p.connect(args.connect_to)
             except (tcp.NetLibError, PathocError), v: