diff options
| author | Aldo Cortesi <aldo@nullcube.com> | 2011-02-16 16:43:35 +1300 |
|---|---|---|
| committer | Aldo Cortesi <aldo@nullcube.com> | 2011-02-16 16:43:35 +1300 |
| commit | 66349c97836c5885919f9fd2bce75ddd04bfc39a (patch) | |
| tree | a5d170848515d8b293de80c196760848552308ed /libmproxy | |
| parent | 27f0765cdd6bb29fe1dd00aaa24b17ae50888368 (diff) | |
| download | mitmproxy-66349c97836c5885919f9fd2bce75ddd04bfc39a.tar.gz mitmproxy-66349c97836c5885919f9fd2bce75ddd04bfc39a.tar.bz2 mitmproxy-66349c97836c5885919f9fd2bce75ddd04bfc39a.zip | |
FlowMaster bugfixes and unit tests.
Diffstat (limited to 'libmproxy')
| -rw-r--r-- | libmproxy/console.py | 2 | ||||
| -rw-r--r-- | libmproxy/flow.py | 7 | ||||
| -rw-r--r-- | libmproxy/proxy.py | 2 |
3 files changed, 3 insertions, 8 deletions
diff --git a/libmproxy/console.py b/libmproxy/console.py index 794d5483..99bf3a4e 100644 --- a/libmproxy/console.py +++ b/libmproxy/console.py @@ -1334,7 +1334,7 @@ class ConsoleMaster(flow.FlowMaster): self.process_flow(f, r) def handle_response(self, r): - f = flow.FlowMaster.handle_request(self, r) + f = flow.FlowMaster.handle_response(self, r) if f: if f.match(self.stickycookie): hid = (f.request.host, f.request.port) diff --git a/libmproxy/flow.py b/libmproxy/flow.py index e791055f..9edb9eab 100644 --- a/libmproxy/flow.py +++ b/libmproxy/flow.py @@ -315,12 +315,7 @@ class FlowMaster(controller.Master): f = self.state.add_response(r) if not f: r.ack() - else: - if f.match(self.stickycookie): - hid = (f.request.host, f.request.port) - if f.response.headers.has_key("set-cookie"): - self.stickyhosts[hid] = f.response.headers["set-cookie"] - self.process_flow(f, r) + return f class FlowWriter: diff --git a/libmproxy/proxy.py b/libmproxy/proxy.py index 6093c936..e2c25966 100644 --- a/libmproxy/proxy.py +++ b/libmproxy/proxy.py @@ -433,7 +433,7 @@ class ServerConnection: if code >= 100 and code <= 199: return self.read_response() if self.request.method == "HEAD" or code == 204 or code == 304: - content = None + content = "" else: content = read_http_body(self.rfile, self, headers, True) return Response(self.request, code, msg, headers, content) |