diff options
Diffstat (limited to 'mitmproxy/builtins/stickyauth.py')
| -rw-r--r-- | mitmproxy/builtins/stickyauth.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/mitmproxy/builtins/stickyauth.py b/mitmproxy/builtins/stickyauth.py index 98fb65ed..18b43967 100644 --- a/mitmproxy/builtins/stickyauth.py +++ b/mitmproxy/builtins/stickyauth.py @@ -1,18 +1,17 @@ from __future__ import absolute_import, print_function, division -from mitmproxy import filt from mitmproxy import exceptions +from mitmproxy import flowfilter class StickyAuth: def __init__(self): - # Compiled filter self.flt = None self.hosts = {} def configure(self, options, updated): if options.stickyauth: - flt = filt.parse(options.stickyauth) + flt = flowfilter.parse(options.stickyauth) if not flt: raise exceptions.OptionsError( "stickyauth: invalid filter expression: %s" % options.stickyauth @@ -23,6 +22,6 @@ class StickyAuth: host = flow.request.host if "authorization" in flow.request.headers: self.hosts[host] = flow.request.headers["authorization"] - elif flow.match(self.flt): + elif flowfilter.match(self.flt, flow): if host in self.hosts: flow.request.headers["authorization"] = self.hosts[host] |