kylefang/mitmproxy
1
0
Fork 0
mirror of https://github.com/zhigang1992/mitmproxy.git synced 2026-04-24 04:14:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Put DumpMaster in its own file.

Browse Source 
It's going to become a more important part of the mitmproxy suite now.
This commit is contained in:
Aldo Cortesi
2010-03-01 14:58:04 +13:00
parent 0188cf8a1a
commit ba92d22e1d
3 changed files with 41 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
libmproxy/controller.py
View File

@@ -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()

39
libmproxy/dump.py Normal file
View File

@@ -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()