��V� � @ s� d Z d d l m Z y d d l m Z Wn" e k
rN d d l m Z Yn Xd d l m Z d d l m Z d d l
m Z e e � Z
Gd d � d e � Z d S)
z
NTLM authenticating pool, contributed by erikcederstran
Issue #10, see: http://code.google.com/p/urllib3/issues/detail?id=10
� )�absolute_import)�HTTPSConnection)� getLogger)�ntlm)�HTTPSConnectionPoolc s[ e Z d Z d Z d Z � f d d � Z d d � Z d d d d d � f d
d � Z � S) �NTLMConnectionPoolzQ
Implements an NTLM authentication version of an urllib3 connection pool
Zhttpsc sj t t | � j | | � | | _ | | _ | j d d � } | d j � | _ | d | _ | | _ d S)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__�authurl�rawuser�split�upper�domain�user�pw)�selfr r r �args�kwargsZ
user_parts)� __class__� � /ntlmpool.pyr s
zNTLMConnectionPool.__init__c
C s� | j d 7_ t j d | j | j | j � i } d | d <d } d } t d | j d | j � } d t j | j � | | <t j d
| � | j
d | j d | � | j � } t | j
� � } t j d | j | j � t j d
| � t j d | j d � � d | _ | | j d � } d } x4 | D], } | d d � d k r1| d d � } q1W| d k r�t d | | | f � � t j | � \ }
} t j |
| j | j | j | � } d | | | <t j d
| � | j
d | j d | � | j � } t j d | j | j � t j d
t | j
� � � t j d | j � d d � � | j d k r�| j d k r�t d � � t d | j | j f � � d | _ t j d � | S)Nr z3Starting NTLM HTTPS connection no. %d: https://%s%sz
Keep-Alive�
ConnectionZ
Authorizationzwww-authenticate�host�portzNTLM %szRequest headers: %s�GETzResponse 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_connections�log�debugr r r r r Z create_NTLM_NEGOTIATE_MESSAGEr
ZrequestZ getresponse�dictZ
getheaders�status�reason�read�fpr � ExceptionZ parse_NTLM_CHALLENGE_MESSAGEZ create_NTLM_AUTHENTICATE_MESSAGEr r r )
r �headersZ
req_headerZ resp_header�conn�resZreshdrZauth_header_valuesZauth_header_value�sZServerChallengeZNegotiateFlagsZauth_msgr r r � _new_conn* s\
z NTLMConnectionPool._new_connN� Tc sD | d k r i } d | d <t t | � j | | | | | | | � S)Nz
Keep-Aliver )r
r �urlopen)r �methodZurlZbodyr( ZretriesZredirectZassert_same_host)r r r r. k s
zNTLMConnectionPool.urlopen)�__name__�
__module__� __qualname__�__doc__�schemer r, r. r r )r r r s A r N)r3 Z
__future__r Z http.clientr � ImportError�httplib�loggingr r Zurllib3r r0 r r r r r r �<module> s