From ba92d22e1d789c189f2007f8b933c520e2a593a9 Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Mon, 1 Mar 2010 14:58:04 +1300 Subject: Put DumpMaster in its own file. It's going to become a more important part of the mitmproxy suite now. --- libmproxy/dump.py | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 libmproxy/dump.py (limited to 'libmproxy/dump.py') diff --git a/libmproxy/dump.py b/libmproxy/dump.py new file mode 100644 index 00000000..83238da4 --- /dev/null +++ b/libmproxy/dump.py @@ -0,0 +1,39 @@ +import sys +import controller + +class DumpMaster(controller.Master): + """ + A simple master that just dumps to screen. + """ + def __init__(self, server, verbosity): + self.verbosity = verbosity + controller.Master.__init__(self, server) + + def run(self): + try: + return controller.Master.run(self) + except KeyboardInterrupt: + self.shutdown() + + def handle_response(self, msg): + if 0 < self.verbosity < 3: + print >> sys.stderr, ">>", + print >> sys.stderr, msg.request.short() + if self.verbosity == 1: + print >> sys.stderr, "<<", + print >> sys.stderr, msg.short() + elif self.verbosity == 2: + print >> sys.stderr, "<<" + for i in msg.assemble().splitlines(): + print >> sys.stderr, "\t", i + print >> sys.stderr, "<<" + elif self.verbosity == 3: + print >> sys.stderr, ">>" + for i in msg.request.assemble().splitlines(): + print >> sys.stderr, "\t", i + print >> sys.stderr, ">>" + print >> sys.stderr, "<<" + for i in msg.assemble().splitlines(): + print >> sys.stderr, "\t", i + print >> sys.stderr, "<<" + msg.ack() -- cgit v1.2.3