Rooba
/
mitmproxy
mirror of https://github.com/mitmproxy/mitmproxy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

make AuthAction generic

This commit is contained in:
Maximilian Hils 2013-12-08 01:37:45 +01:00
parent e402e3b862
commit 390f2a46c9
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
netlib/http_auth.py
View File

@ -33,6 +33,7 @@ class NullProxyAuth():
class BasicProxyAuth(NullProxyAuth): class BasicProxyAuth(NullProxyAuth):
CHALLENGE_HEADER = 'Proxy-Authenticate' CHALLENGE_HEADER = 'Proxy-Authenticate'
AUTH_HEADER = 'Proxy-Authorization' AUTH_HEADER = 'Proxy-Authorization'
def __init__(self, password_manager, realm): def __init__(self, password_manager, realm):
NullProxyAuth.__init__(self, password_manager) NullProxyAuth.__init__(self, password_manager)
self.realm = realm self.realm = realm
@ -125,11 +126,10 @@ class AuthAction(Action):
""" """
def __call__(self, parser, namespace, values, option_string=None): def __call__(self, parser, namespace, values, option_string=None):
passman = self.getPasswordManager(values) passman = self.getPasswordManager(values)
if passman: if not passman:
authenticator = BasicProxyAuth(passman, "mitmproxy") raise ArgumentTypeError("Error creating password manager for proxy authentication.")
else: authenticator = BasicProxyAuth(passman, "mitmproxy")
authenticator = NullProxyAuth(None) setattr(namespace, self.dest, authenticator)
setattr(namespace, "authenticator", authenticator)
def getPasswordManager(self, s): def getPasswordManager(self, s):
""" """