�
��fg� � � � d Z ddlmZ ddlZddlmZ ddlmZ ddlmZ ddl m
Z
ej d e � � ee
� � Z G d
� d e� � ZdS ) z
NTLM authenticating pool, contributed by erikcederstran
Issue #10, see: http://code.google.com/p/urllib3/issues/detail?id=10
� )�absolute_importN)� getLogger)�ntlm� )�HTTPSConnectionPool)�HTTPSConnectiona# The 'urllib3.contrib.ntlmpool' module is deprecated and will be removed in urllib3 v2.0 release, urllib3 is not able to support it properly due to reasons listed in issue: https://github.com/urllib3/urllib3/issues/2282. If you are a user of this module please comment in the mentioned issue.c �B � � e Zd ZdZdZ� fd�Zd� Z d � fd� Z� xZS )
�NTLMConnectionPoolzQ
Implements an NTLM authentication version of an urllib3 connection pool
�httpsc � �� t t | � � j |i |�� || _ || _ |� dd� � }|d � � � | _ |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__�authurl�rawuser�split�upper�domain�user�pw)�selfr r r �args�kwargs�
user_parts� __class__s ��:/usr/lib/python3/dist-packages/urllib3/contrib/ntlmpool.pyr zNTLMConnectionPool.__init__" sv �� � 1�� �$�'�'�0�$�A�&�A�A�A� �� ��� ��Z�Z��a�(�(�
� ��m�)�)�+�+�� � �q�M�� ������ c � � | xj dz
c_ t � d| j | j | j � � ddi}d}d}t | j | j �� � }dt j | j � � z ||< t � d |� � |�
d
| j d |� � |� � � }t |�
� � � � }t � d |j |j � � t � d |� � t � d
|� d� � � � d |_ || � d� � }d }|D ]} | d d� dk r
| dd � }�|�t' d|�d|| ��� � �t j |� � \ }
} t j |
| j | j | j | � � } d| z ||< t � d |� � |�
d
| j d |� � |� � � }t � d |j |j � � t � d t |�
� � � � � � t � d
|� � � d d� � � |j dk r9|j dk rt' d� � �t' d|j �d|j ��� � �d |_ t � d� � |S )Nr z3Starting NTLM HTTPS connection no. %d: https://%s%s�
Connection�
Keep-Alive�
Authorizationzwww-authenticate)�host�portzNTLM %szRequest headers: %s�GETzResponse status: %s %szResponse headers: %szResponse data: %s [...]�d z, � zNTLM z Unexpected z response header: �� i� z3Server rejected request: wrong username or passwordzWrong server response: � zConnection established)�num_connections�log�debugr# r r r$ r � create_NTLM_NEGOTIATE_MESSAGEr �request� getresponse�dict�
getheaders�status�reason�read�fpr � Exception� parse_NTLM_CHALLENGE_MESSAGE� create_NTLM_AUTHENTICATE_MESSAGEr r r )
r �headers�
req_header� resp_header�conn�res�reshdr�auth_header_values�auth_header_value�s�ServerChallenge�NegotiateFlags�auth_msgs
r � _new_connz NTLMConnectionPool._new_conn0 s$ � �
� � ��!� � � � � � A� � � �I� �L�
�
�
� � �.��$�
�(� � �D�I�D�I�>�>�>�� (�$�*L� �L�+
� +
�
��
�� � � �'��1�1�1� � � �U�D�L�$��8�8�8�� � � � ���c�n�n�&�&�'�'�� � � �*�C�J��
�C�C�C� � � �(�&�1�1�1� � � �+�S�X�X�c�]�]�;�;�;� ��� $�K� 0� 6� 6�t� <� <�� ��#� *� *�A���!��u�� � �$%�a�b�b�E�!�� � $��)�7B�{�{�F�;�DW�DW�X�� �
�
+/�*K� �+
� +
�'�� � �8� �T�Y�� �T�W�n�
�
�� (�(� 2��
�� � � �'��1�1�1� � � �U�D�L�$��8�8�8�� � � � �� � � �*�C�J��
�C�C�C� � � �(�$�s�~�~�/?�/?�*@�*@�A�A�A� � � �+�S�X�X�Z�Z����-=�>�>�>� �:�� � ��z�S� � � � U�V�V�V��)�c�j�j�j�#�*�*� U�V�V� V���� � � �*�+�+�+�� r N� Tc �r �� |�i }d|d< t t | � � � |||||||� � S )Nr! r )r r
�urlopen) r �method�url�bodyr9 �retries�redirect�assert_same_hostr s �r rH zNTLMConnectionPool.urlopens sL �� � �?��G� ,�� � ��'��.�.�6�6� �C��w���;K�
�
�
r )NNrF TT) �__name__�
__module__� __qualname__�__doc__�schemer rE rH �
__classcell__)r s @r r
r
s� �� � � � � �� � �F� � � � � � A� A� A�N ���� �
�
�
�
�
�
�
�
�
�
r r
)rR �
__future__r �warnings�loggingr r � r �six.moves.http_clientr �warn�DeprecationWarningrO r+ r
� r r �<module>r] s� ��� �
'� &� &� &� &� &� ���� � � � � � � � � � � � � "� "� "� "� "� "� 1� 1� 1� 1� 1� 1�
��
�N� � � � � �i�����g
� g
� g
� g
� g
�,� g
� g
� g
� g
� g
r |