web: backup flows on edit

2 files changed, 11 insertions, 0 deletions

diff --git a/libmproxy/flow.py b/libmproxy/flow.py
index 8343c183..9aa2a179 100644
--- a/libmproxy/flow.py
+++ b/libmproxy/flow.py
@@ -593,6 +593,10 @@ class State(object):
     def accept_all(self, master):
         self.flows.accept_all(master)
 
+    def backup(self, f):
+        f.backup()
+        self.update_flow(f)
+
     def revert(self, f):
         f.revert()
         self.update_flow(f)
diff --git a/libmproxy/web/app.py b/libmproxy/web/app.py
index 76160dc1..d95fb8ea 100644
--- a/libmproxy/web/app.py
+++ b/libmproxy/web/app.py
@@ -119,6 +119,7 @@ class FlowHandler(RequestHandler):
 
     def put(self, flow_id):
         flow = self.flow
+        self.state.backup(flow)
         for a, b in self.json.iteritems():
 
             if a == "request":
@@ -130,6 +131,8 @@ class FlowHandler(RequestHandler):
                         request.port = int(v)
                     elif k == "httpversion":
                         request.httpversion = tuple(int(x) for x in v)
+                    elif k == "headers":
+                        request.headers.load_state(v)
                     else:
                         print "Warning: Unknown update {}.{}: {}".format(a, k, v)
 
@@ -142,6 +145,10 @@ class FlowHandler(RequestHandler):
                         response.code = int(v)
                     elif k == "httpversion":
                         response.httpversion = tuple(int(x) for x in v)
+                    elif k == "headers":
+                        response.headers.load_state(v)
+                    else:
+                        print "Warning: Unknown update {}.{}: {}".format(a, k, v)
             else:
                 print "Warning: Unknown update {}: {}".format(a, b)
         self.state.update_flow(flow)