Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 18:18:25 +00:00
Code Issues Projects Releases Wiki Activity

add import capability for old flows, fix #840

Browse Source
This commit is contained in:
Maximilian Hils 2015-12-02 19:50:08 +01:00
parent b5d5e56fdb
commit 3963a2191b
2 changed files with 39 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libmproxy/flow.py
View File

@ -13,7 +13,7 @@ import urlparse
from netlib import wsgi from netlib import wsgi
from netlib.exceptions import HttpException from netlib.exceptions import HttpException
from netlib.http import CONTENT_MISSING, Headers, http1 from netlib.http import CONTENT_MISSING, Headers, http1
from . import controller, tnetstring, filt, script, version from . import controller, tnetstring, filt, script, version, flow_format_compat
from .onboarding import app from .onboarding import app
from .proxy.config import HostMatcher from .proxy.config import HostMatcher
from .protocol.http_replay import RequestReplayThread from .protocol.http_replay import RequestReplayThread
@ -1112,14 +1112,13 @@ class FlowReader:
try: try:
while True: while True:
data = tnetstring.load(self.fo) data = tnetstring.load(self.fo)
if tuple(data["version"][:2]) != version.IVERSION[:2]: try:
v = ".".join(str(i) for i in data["version"]) data = flow_format_compat.migrate_flow(data)
raise FlowReadError( except ValueError as e:
"Incompatible serialized data version: %s" % v raise FlowReadError(str(e))
)
off = self.fo.tell() off = self.fo.tell()
yield HTTPFlow.from_state(data) yield HTTPFlow.from_state(data)
except ValueError as v: except ValueError:
# Error is due to EOF # Error is due to EOF
if self.fo.tell() == off and self.fo.read() == '': if self.fo.tell() == off and self.fo.read() == '':
return return

33
libmproxy/flow_format_compat.py Normal file
View File

@ -0,0 +1,33 @@
"""
This module handles the import of mitmproxy flows generated by old versions.
"""
from __future__ import absolute_import, print_function, division
from . import version
def convert_013_014(data):
data["request"]["first_line_format"] = data["request"].pop("form_in")
data["request"]["http_version"] = "HTTP/" + ".".join(str(x) for x in data["request"].pop("httpversion"))
data["response"]["status_code"] = data["response"].pop("code")
data["response"]["body"] = data["response"].pop("content")
data["server_conn"].pop("state")
data["server_conn"]["via"] = None
data["version"] = version.IVERSION
return data
converters = {
(0, 13): convert_013_014
}
def migrate_flow(flow_data):
while True:
flow_version = tuple(flow_data["version"][:2])
if flow_version == version.IVERSION[:2]:
break
elif flow_version in converters:
flow_data = converters[flow_version](flow_data)
else:
v = ".".join(str(i) for i in flow_data["version"])
raise ValueError("Incompatible serialized data version: {}".format(v))
return flow_data