diff options
| author | Aldo Cortesi <aldo@nullcube.com> | 2014-10-25 17:58:59 +1300 |
|---|---|---|
| committer | Aldo Cortesi <aldo@nullcube.com> | 2014-10-25 17:58:59 +1300 |
| commit | fc4f9a1c7a0734a190b99265ca50d72014173859 (patch) | |
| tree | 4d529612eae9207b340a505a8163f95f0ccf25d0 /libpathod | |
| parent | 609d6eab30aac0601106993eb8c5d9a953ba402b (diff) | |
| download | mitmproxy-fc4f9a1c7a0734a190b99265ca50d72014173859.tar.gz mitmproxy-fc4f9a1c7a0734a190b99265ca50d72014173859.tar.bz2 mitmproxy-fc4f9a1c7a0734a190b99265ca50d72014173859.zip | |
pathoc -n 0 repeats forever
Diffstat (limited to 'libpathod')
| -rw-r--r-- | libpathod/cmdline.py | 2 | ||||
| -rw-r--r-- | libpathod/pathoc.py | 6 |
2 files changed, 6 insertions, 2 deletions
diff --git a/libpathod/cmdline.py b/libpathod/cmdline.py index a21fa218..f80b7aae 100644 --- a/libpathod/cmdline.py +++ b/libpathod/cmdline.py @@ -36,7 +36,7 @@ def go_pathoc(): ) parser.add_argument( "-n", dest='repeat', default=1, type=int, metavar="N", - help='Repeat N times' + help='Repeat N times. If 0 repeat for ever.' ) parser.add_argument( "-r", dest="random", action="store_true", default=False, diff --git a/libpathod/pathoc.py b/libpathod/pathoc.py index c85c207a..02d0c06d 100644 --- a/libpathod/pathoc.py +++ b/libpathod/pathoc.py @@ -224,7 +224,9 @@ class Pathoc(tcp.TCPClient): def main(args): try: - for i in range(args.repeat): + cnt = 0 + while 1: + cnt += 1 p = Pathoc( (args.host, args.port), ssl=args.ssl, @@ -258,5 +260,7 @@ def main(args): sys.stdout.flush() if ret and args.oneshot: sys.exit(0) + if cnt == args.repeat: + break except KeyboardInterrupt: pass |
