Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-29 19:08:44 +00:00
Code Issues Projects Releases Wiki Activity

Client playback shortcut in mitmproxy can now be used to interrupt a current client playback.

Browse Source
This commit is contained in:
Aldo Cortesi 2011-03-18 09:38:51 +13:00
parent 4893e5e5a4
commit f97c144869
2 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
libmproxy/console.py
View File

@ -1298,11 +1298,22 @@ class ConsoleMaster(flow.FlowMaster):
if k == "?": if k == "?":
self.view_help() self.view_help()
elif k == "c": elif k == "c":
self.path_prompt( if not self.client_playback:
"Client replay: ", self.path_prompt(
self.state.last_saveload, "Client replay: ",
self.client_playback_path self.state.last_saveload,
) self.client_playback_path
)
else:
self.prompt_onekey(
"Stop current client replay?",
(
("yes", "y"),
("no", "n"),
),
self.stop_client_playback_prompt,
)
k = None k = None
elif k == "l": elif k == "l":
self.prompt("Limit: ", self.state.limit_txt, self.set_limit) self.prompt("Limit: ", self.state.limit_txt, self.set_limit)
@ -1384,6 +1395,10 @@ class ConsoleMaster(flow.FlowMaster):
except (Stop, KeyboardInterrupt): except (Stop, KeyboardInterrupt):
pass pass
def stop_client_playback_prompt(self, a):
if a != "n":
self.stop_client_playback()
def quit(self, a): def quit(self, a):
if a != "n": if a != "n":
raise Stop raise Stop

3
libmproxy/flow.py
View File

@ -464,6 +464,9 @@ class FlowMaster(controller.Master):
""" """
self.client_playback = ClientPlaybackState(flows, exit) self.client_playback = ClientPlaybackState(flows, exit)
def stop_client_playback(self):
self.client_playback = None
def start_server_playback(self, flows, kill, headers, exit): def start_server_playback(self, flows, kill, headers, exit):
""" """
flows: A list of flows. flows: A list of flows.