mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-30 03:14:22 +00:00
87 lines
2.0 KiB
Python
87 lines
2.0 KiB
Python
import datetime
|
|
|
|
import six
|
|
|
|
import netlib.utils
|
|
import netlib.tcp
|
|
import netlib.http
|
|
from netlib import strutils
|
|
|
|
TIMEFMT = '%d-%m-%y %H:%M:%S'
|
|
|
|
|
|
def write_raw(fp, lines):
|
|
if fp:
|
|
fp.write(
|
|
"%s: " % datetime.datetime.now().strftime(TIMEFMT)
|
|
)
|
|
for i in lines:
|
|
fp.write(i)
|
|
fp.write("\n")
|
|
fp.flush()
|
|
|
|
|
|
class LogCtx(object):
|
|
|
|
def __init__(self, fp, hex, rfile, wfile):
|
|
self.lines = []
|
|
self.fp = fp
|
|
self.suppressed = False
|
|
self.hex = hex
|
|
self.rfile, self.wfile = rfile, wfile
|
|
|
|
def __enter__(self):
|
|
if self.wfile:
|
|
self.wfile.start_log()
|
|
if self.rfile:
|
|
self.rfile.start_log()
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
wlog = self.wfile.get_log() if self.wfile else None
|
|
rlog = self.rfile.get_log() if self.rfile else None
|
|
if self.suppressed or not self.fp:
|
|
return
|
|
if wlog:
|
|
self("Bytes written:")
|
|
self.dump(wlog, self.hex)
|
|
if rlog:
|
|
self("Bytes read:")
|
|
self.dump(rlog, self.hex)
|
|
if self.lines:
|
|
write_raw(
|
|
self.fp,
|
|
[
|
|
"\n".join(self.lines),
|
|
]
|
|
)
|
|
if exc_value:
|
|
six.reraise(exc_type, exc_value, traceback)
|
|
|
|
def suppress(self):
|
|
self.suppressed = True
|
|
|
|
def dump(self, data, hexdump):
|
|
if hexdump:
|
|
for line in netlib.utils.hexdump(data):
|
|
self("\t%s %s %s" % line)
|
|
else:
|
|
for i in strutils.clean_bin(data).split("\n"):
|
|
self("\t%s" % i)
|
|
|
|
def __call__(self, line):
|
|
self.lines.append(line)
|
|
|
|
|
|
class ConnectionLogger:
|
|
def __init__(self, fp, hex, rfile, wfile):
|
|
self.fp = fp
|
|
self.hex = hex
|
|
self.rfile, self.wfile = rfile, wfile
|
|
|
|
def ctx(self):
|
|
return LogCtx(self.fp, self.hex, self.rfile, self.wfile)
|
|
|
|
def write(self, lines):
|
|
write_raw(self.fp, lines)
|