CasperSecurity
o
Ba�_�' � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
m Z m Z ddl
mZ ddlmZ ddlmZ d Zd
Zd d � ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )z]
requests.auth
~~~~~~~~~~~~~
This module contains the authentication handlers for Requests.
� N)� b64encode� )�urlparse�str�
basestring)�extract_cookies_to_jar)�to_native_string)�parse_dict_headerz!application/x-www-form-urlencodedzmultipart/form-datac C s� t | t�stjd�| �td� t| �} t |t�s(tjd�t|��td� t|�}t | t�r2| �d�} t |t�r<|�d�}dt t
d� | |f��� � � }|S )z Returns a Basic Auth string.z�Non-string usernames will no longer be supported in Requests 3.0.0. Please convert the object you've passed in ({!r}) to a string or bytes object in the near future to avoid problems.)�categoryz�Non-string passwords will no longer be supported in Requests 3.0.0. Please convert the object you've passed in ({!r}) to a string or bytes object in the near future to avoid problems.�latin1zBasic � :)
�
isinstancer �warnings�warn�format�DeprecationWarningr �type�encoder r �join�strip)�username�password�authstr� r �//usr/lib/python3/dist-packages/requests/auth.py�_basic_auth_str s,
�
�
�r c @ � e Zd ZdZdd� ZdS )�AuthBasez4Base class that all auth implementations derive fromc C s t d��)Nz Auth hooks must be callable.)�NotImplementedError��self�rr r r �__call__K s zAuthBase.__call__N��__name__�
__module__� __qualname__�__doc__r" r r r r r H � r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
HTTPBasicAuthz?Attaches HTTP Basic Authentication to the given Request object.c C s || _ || _d S �N)r r �r r r r r r �__init__R s
zHTTPBasicAuth.__init__c C �( t | jt|dd �k| jt|dd �kg�S �Nr r ��allr �getattrr �r �otherr r r �__eq__V � �zHTTPBasicAuth.__eq__c C �
| |k S r* r r2 r r r �__ne__\ �
zHTTPBasicAuth.__ne__c C � t | j| j�|jd< |S )N�
Authorization�r r r �headersr r r r r" _ � zHTTPBasicAuth.__call__N)r$ r% r& r' r, r4 r7 r" r r r r r) O s r) c @ r )�
HTTPProxyAuthz=Attaches HTTP Proxy Authentication to a given Request object.c C r9 )NzProxy-Authorizationr; r r r r r" g r= zHTTPProxyAuth.__call__Nr# r r r r r> d r( r> c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�HTTPDigestAuthz@Attaches HTTP Digest Authentication to the given Request object.c C s || _ || _t�� | _d S r* )r r � threading�local�
_thread_localr+ r r r r, o s zHTTPDigestAuth.__init__c C sD t | jd�s d| j_d| j_d| j_i | j_d | j_d | j_d S d S )N�initT� r )�hasattrrB rC �
last_nonce� nonce_count�chal�pos�
num_401_calls)r r r r �init_per_thread_stateu s �z$HTTPDigestAuth.init_per_thread_statec s� | j jd }| j jd }| j j�d�}| j j�d�}| j j�d�}d� |du r*d}n|�� }|dks6|dkr=d d
� } | � n |d krHd d
� }
|
� n|dkrSdd� } | � n
|dkr]dd� } | � � fdd�}
� du ridS d}t|�}|jpsd}|jr~|d|j 7 }d| j|| jf }d||f }� |�}� |�}|| j j kr�| j j
d7 _
nd| j _
d| j j
}t | j j
�� d �}||� d �7 }|t
�� � d �7 }|t�d �7 }t�|��� dd � }|dkr� d|||f �}|s�|
|d||f �}n |d ks�d |�d �v �r
d!|||d |f }|
||�}ndS || j _ d"| j||||f }|�r&|d#| 7 }|�r/|d$| 7 }|�r8|d%| 7 }|�rC|d&||f 7 }d'| S )(z
:rtype: str
�realm�nonce�qop� algorithm�opaqueN�MD5zMD5-SESSc S �" t | t�r
| �d�} t�| ��� S �N�utf-8)r
r r �hashlib�md5� hexdigest��xr r r �md5_utf8� �
z4HTTPDigestAuth.build_digest_header.<locals>.md5_utf8�SHAc S rR rS )r
r r rU �sha1rW rX r r r �sha_utf8� r[ z4HTTPDigestAuth.build_digest_header.<locals>.sha_utf8zSHA-256c S rR rS )r
r r rU �sha256rW rX r r r � sha256_utf8� r[ z7HTTPDigestAuth.build_digest_header.<locals>.sha256_utf8zSHA-512c S rR rS )r
r r rU �sha512rW rX r r r � sha512_utf8� r[ z7HTTPDigestAuth.build_digest_header.<locals>.sha512_utf8c s � d| |f �S )N�%s:%sr )�s�d�� hash_utf8r r �<lambda>� s z4HTTPDigestAuth.build_digest_header.<locals>.<lambda>�/�?z%s:%s:%src r z%08xrT � � �auth�,z%s:%s:%s:%s:%sz>username="%s", realm="%s", nonce="%s", uri="%s", response="%s"z
, opaque="%s"z, algorithm="%s"z
, digest="%s"z , qop="auth", nc=%s, cnonce="%s"z Digest %s)rB rH �get�upperr �path�queryr r rF rG r r �time�ctime�os�urandomrU r] rW �split)r �method�urlrL rM rN rO rP �
_algorithmrZ r^ r` rb �KD�entdig�p_parsedrq �A1�A2�HA1�HA2�ncvaluerd �cnonce�respdig�noncebit�baser rf r �build_digest_header s�
� � z"HTTPDigestAuth.build_digest_headerc K s |j r d| j_dS dS )z)Reset num_401_calls counter on redirects.r N)� is_redirectrB rJ )r r! �kwargsr r r �handle_redirect� s �z HTTPDigestAuth.handle_redirectc K s" d|j kr dk sn d| j_|S | jjdur!|jj�| jj� |j�dd�}d|� � v r�| jjdk r�| j jd7 _t
j d t
j d
�}t
|jd|dd ��| j_|j |�� |j�� }t|j|j|j� |�|j� | �|j|j�|jd < |jj|fi |��}|j � |� ||_|S d| j_|S )
zo
Takes the given response and tries digest-auth, if needed.
:rtype: requests.Response
i� i� r Nzwww-authenticaterD �digest� zdigest )�flags)�countr: ) � status_coderB rJ rI �request�body�seekr< ro �lower�re�compile�
IGNORECASEr �subrH �content�close�copyr �_cookies�raw�prepare_cookiesr� rx ry �
connection�send�history�append)r r! r� �s_auth�pat�prep�_rr r r �
handle_401� s0
� zHTTPDigestAuth.handle_401c C s| | � � | jjr| �|j|j�|jd< z |j�� | j_ W n
t
y) d | j_ Y nw |� d| j � |� d| j
� d| j_|S )Nr: �responser )rK rB rF r� rx ry r< r� �tellrI �AttributeError�
register_hookr� r� rJ r r r r r" s �zHTTPDigestAuth.__call__c C r- r. r/ r2 r r r r4 * r5 zHTTPDigestAuth.__eq__c C r6 r* r r2 r r r r7 0 r8 zHTTPDigestAuth.__ne__N) r$ r% r& r' r, rK r� r� r� r" r4 r7 r r r r r? l s
f, r? )r' ru r� rs rU r@ r �base64r �compatr r r �cookiesr �_internal_utilsr �utilsr � CONTENT_TYPE_FORM_URLENCODED�CONTENT_TYPE_MULTI_PARTr �objectr r) r> r? r r r r �<module> s& ,