diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2010-03-01 14:58:04 +1300 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2010-03-01 14:58:04 +1300 |
commit | ba92d22e1d789c189f2007f8b933c520e2a593a9 (patch) | |
tree | d4838b2f122e4f0dba223fe6ea660a6ab20eee34 /libmproxy | |
parent | 0188cf8a1a9abc3b03fb1cee91ed2aa1f20b1ebd (diff) | |
download | mitmproxy-ba92d22e1d789c189f2007f8b933c520e2a593a9.tar.gz mitmproxy-ba92d22e1d789c189f2007f8b933c520e2a593a9.tar.bz2 mitmproxy-ba92d22e1d789c189f2007f8b933c520e2a593a9.zip |
Put DumpMaster in its own file.
It's going to become a more important part of the mitmproxy suite now.
Diffstat (limited to 'libmproxy')
-rw-r--r-- | libmproxy/controller.py | 36 | ||||
-rw-r--r-- | libmproxy/dump.py | 39 |
2 files changed, 39 insertions, 36 deletions
diff --git a/libmproxy/controller.py b/libmproxy/controller.py index f20d17b4..cd86151a 100644 --- a/libmproxy/controller.py +++ b/libmproxy/controller.py @@ -86,39 +86,3 @@ class Master: self.server.shutdown() -class DumpMaster(Master): - """ - A simple master that just dumps to screen. - """ - def __init__(self, server, verbosity): - self.verbosity = verbosity - Master.__init__(self, server) - - def run(self): - try: - return 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() 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() |