pathod: handle keyboard interrupts and SSL errors.

2012-07-22 12:49:59 +12:00 · 2012-07-22 12:49:59 +12:00 · 30a6988392
parent 1e93e42883
commit 30a6988392
1 changed files with 21 additions and 12 deletions
--- a/33
+++ b/33
@ -49,15 +49,24 @@ if __name__ == "__main__":
    else:
        port = args.port

-    for i in range(args.repeat):
-        p = pathoc.Pathoc(args.host, port)
-        try:
-            p.connect()
-        except tcp.NetLibError, v:
-            print >> sys.stderr, str(v)
-            sys.exit(1)
-        if args.ssl:
-            p.convert_to_ssl(sni=args.sni)
-        if args.timeout:
-            p.settimeout(args.timeout)
-        p.print_requests(args.request, args.verbose, args.reqdump)
+    try:
+        for i in range(args.repeat):
+            p = pathoc.Pathoc(args.host, port)
+            try:
+                p.connect()
+            except tcp.NetLibError, v:
+                print >> sys.stderr, str(v)
+                sys.exit(1)
+            if args.ssl:
+                try:
+                    p.convert_to_ssl(sni=args.sni)
+                except tcp.NetLibError, v:
+                    print "\n>> %s"%v
+                    continue
+            if args.timeout:
+                p.settimeout(args.timeout)
+            p.print_requests(args.request, args.verbose, args.reqdump)
+            sys.stdout.flush()
+    except KeyboardInterrupt:
+        pass
+