3
nf\k @ s\ d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ee
Z G dd d eZ d
S ) z
NTLM authenticating pool, contributed by erikcederstran
Issue #10, see: http://code.google.com/p/urllib3/issues/detail?id=10
)absolute_import) getLogger)ntlm )HTTPSConnectionPool)HTTPSConnectionc s: e Zd ZdZdZ fddZdd Zd fd
d Z ZS )
NTLMConnectionPoolzQ
Implements an NTLM authentication version of an urllib3 connection pool
Zhttpsc sL t t| j|| || _|| _|jdd}|d j | _|d | _|| _ dS )z
authurl is a random URL on the server that is protected by NTLM.
user is the Windows user, probably in the DOMAIN\username format.
pw is the password for the user.
\ r N)
superr __init__authurlrawusersplitupperdomainuserpw)selfr r r
argskwargsZ
user_parts) __class__ /usr/lib/python3.6/ntlmpool.pyr s
zNTLMConnectionPool.__init__c
C s | j d7 _ tjd| j | j| j ddi}d}d}t| j| jd}dtj| j ||< tjd | |j
d
| jd | |j }t |j
}tjd |j|j tjd | tjd
|jd d |_|| jd}d }x(|D ] } | d d dkr| dd }qW |d krtd||| f tj|\}
} tj|
| j| j| j| } d| ||< tjd | |j
d
| jd | |j }tjd |j|j tjd t |j
tjd
|j d d |jdkr|jdkrtdtd|j|jf d |_tjd |S )Nr
z3Starting NTLM HTTPS connection no. %d: https://%s%s
Connectionz
Keep-AliveZ
Authorizationzwww-authenticate)hostportzNTLM %szRequest headers: %sZGETzResponse status: %s %szResponse headers: %szResponse data: %s [...]d z, zNTLM z!Unexpected %s response header: %s i z3Server rejected request: wrong username or passwordz Wrong server response: %s %szConnection established)Znum_connectionslogdebugr r
r r r Z create_NTLM_NEGOTIATE_MESSAGEr ZrequestZ getresponsedictZ
getheadersZstatusreasonreadfpr ExceptionZ parse_NTLM_CHALLENGE_MESSAGEZ create_NTLM_AUTHENTICATE_MESSAGEr r r )
r headersZ
req_headerZ resp_headerZconnresZreshdrZauth_header_valuesZauth_header_valuesZServerChallengeZNegotiateFlagsZauth_msgr r r _new_conn' sZ
z NTLMConnectionPool._new_connN Tc s0 |d kr i }d|d< t t| j|||||||S )Nz
Keep-Aliver )r r urlopen)r methodZurlZbodyr' ZretriesZredirectZassert_same_host)r r r r, g s zNTLMConnectionPool.urlopen)NNr+ TT) __name__
__module__ __qualname____doc__schemer r* r,
__classcell__r r )r r r s @ r N)
r1 Z
__future__r Zloggingr r r Z packages.six.moves.http_clientr r. r r r r r r s