diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2011-02-24 10:33:39 +1300 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2011-02-24 10:33:39 +1300 |
commit | 57947b328ec0faba24e4682f7e4cb9074b81b684 (patch) | |
tree | 047584b734a99c84f2a98db6f379b0fb6f98025b /libmproxy | |
parent | 3c1db00ebb2aa0596840cda6a60e2af3d11a656b (diff) | |
download | mitmproxy-57947b328ec0faba24e4682f7e4cb9074b81b684.tar.gz mitmproxy-57947b328ec0faba24e4682f7e4cb9074b81b684.tar.bz2 mitmproxy-57947b328ec0faba24e4682f7e4cb9074b81b684.zip |
Start abstracting out sticky cookie state.
Diffstat (limited to 'libmproxy')
-rw-r--r-- | libmproxy/flow.py | 25 | ||||
-rw-r--r-- | libmproxy/proxy.py | 4 |
2 files changed, 26 insertions, 3 deletions
diff --git a/libmproxy/flow.py b/libmproxy/flow.py index 42870f17..9636c3bd 100644 --- a/libmproxy/flow.py +++ b/libmproxy/flow.py @@ -2,7 +2,7 @@ This module provides more sophisticated flow tracking. These match requests with their responses, and provide filtering and interception facilities. """ -import subprocess, base64, sys, json, hashlib +import subprocess, base64, sys, json, hashlib, Cookie, cookielib, copy import proxy, threading, netstring import controller @@ -89,6 +89,29 @@ class ServerPlaybackState: return l.pop(0) +class StickyCookieState: + def __init__(self): + self.jar = {} + + def ckey(self, c): + c = copy.copy(c) + del c["expires"] + return str(c) + + def add_cookies(self, headers): + for i in headers: + c = Cookie.SimpleCookie(i) + m = c.values()[0] + self.jar[self.ckey(m)] = m + + def get_cookies(self, domain, path): + cs = [] + for i in self.jar.values(): + if cookielib.domain_match(domain, i["domain"]) and path.startswith(i.get("path", "/")): + cs.append(i) + return cs + + class Flow: def __init__(self, request): self.request = request diff --git a/libmproxy/proxy.py b/libmproxy/proxy.py index ffc9b264..100a2e34 100644 --- a/libmproxy/proxy.py +++ b/libmproxy/proxy.py @@ -85,8 +85,8 @@ def parse_url(url): else: port = 80 path = urlparse.urlunparse(('', '', path, params, query, fragment)) - if not path: - path = "/" + if not path.startswith("/"): + path = "/" + path return scheme, host, port, path |