From 98b1acc51d3b58097c4ed5b607535fd7bc3dd7e4 Mon Sep 17 00:00:00 2001 From: Henrik Nordstrom Date: Fri, 12 Nov 2010 13:18:42 +0100 Subject: Add response creation to edit function on intercepted requests --- libmproxy/proxy.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) (limited to 'libmproxy/proxy.py') diff --git a/libmproxy/proxy.py b/libmproxy/proxy.py index eefa6905..44bc10e2 100644 --- a/libmproxy/proxy.py +++ b/libmproxy/proxy.py @@ -378,11 +378,17 @@ class ProxyHandler(SocketServer.StreamRequestHandler): if request is None: self.finish() return - server = ServerConnection(request) - response = server.read_response() - response = response.send(self.mqueue) + if request.is_response(): + response = request + request = False + response = response.send(self.mqueue) + else: + server = ServerConnection(request) + response = server.read_response() + response = response.send(self.mqueue) + if response is None: + server.terminate() if response is None: - server.terminate() self.finish() return self.send_response(response) -- cgit v1.2.3