diff options
Diffstat (limited to 'libmproxy/console')
| -rw-r--r-- | libmproxy/console/flowlist.py | 5 | ||||
| -rw-r--r-- | libmproxy/console/flowview.py | 6 |
2 files changed, 11 insertions, 0 deletions
diff --git a/libmproxy/console/flowlist.py b/libmproxy/console/flowlist.py index 412a4ec2..e4c73740 100644 --- a/libmproxy/console/flowlist.py +++ b/libmproxy/console/flowlist.py @@ -125,13 +125,18 @@ class ConnectionItem(common.WWrap): self.master.currentflow = f self.master.focus_current() elif key == "r": + self.flow.backup() r = self.master.replay_request(self.flow) if r: self.master.statusbar.message(r) self.master.sync_list_view() elif key == "V": + if not self.flow.modified(): + self.master.statusbar.message("Flow not modified.") + return self.state.revert(self.flow) self.master.sync_list_view() + self.master.statusbar.message("Reverted.") elif key == "w": self.master.path_prompt( "Save flows: ", diff --git a/libmproxy/console/flowview.py b/libmproxy/console/flowview.py index 1ec410ed..024a7ee1 100644 --- a/libmproxy/console/flowview.py +++ b/libmproxy/console/flowview.py @@ -460,13 +460,18 @@ class FlowView(common.WWrap): elif key == "p": self.view_prev_flow(self.flow) elif key == "r": + self.flow.backup() r = self.master.replay_request(self.flow) if r: self.master.statusbar.message(r) self.master.refresh_flow(self.flow) elif key == "V": + if not self.flow.modified(): + self.master.statusbar.message("Flow not modified.") + return self.state.revert(self.flow) self.master.refresh_flow(self.flow) + self.master.statusbar.message("Reverted.") elif key == "W": self.master.path_prompt( "Save this flow: ", @@ -502,6 +507,7 @@ class FlowView(common.WWrap): self.master.view_flowdetails(self.flow) elif key == "z": if conn: + self.flow.backup() e = conn.headers["content-encoding"] or ["identity"] if e[0] != "identity": conn.decode() |