From 427fffbcb82ba16dd65a4fee4000a05215e859b8 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Thu, 21 Jul 2016 00:43:31 -0700 Subject: web: fix flow content update --- mitmproxy/web/app.py | 19 ++++++------------- 1 file changed, 6 insertions(+), 13 deletions(-) diff --git a/mitmproxy/web/app.py b/mitmproxy/web/app.py index 56f7f67d..eaef887a 100644 --- a/mitmproxy/web/app.py +++ b/mitmproxy/web/app.py @@ -8,6 +8,7 @@ import re import six import tornado.websocket +import tornado.web from io import BytesIO from mitmproxy.flow import FlowWriter, FlowReader @@ -293,19 +294,11 @@ class ReplayFlow(RequestHandler): class FlowContent(RequestHandler): - def post (self, flow_id, message): - # handle request later now just change response content - - flow = self.flow - flow.backup() - content = self.request.files.values()[0][0].body - if (message == "response"): - with models.decoded(flow.response): - flow.response.content = content - elif(message == "request"): - with models.decoded(flow.request): - flow.request.content = content - self.state.update_flow(flow) + def post(self, flow_id, message): + self.flow.backup() + message = getattr(self.flow, message) + message.content = self.request.files.values()[0][0].body + self.state.update_flow(self.flow) def get(self, flow_id, message): message = getattr(self.flow, message) -- cgit v1.2.3