File: //opt/saltstack/salt/lib/python3.10/site-packages/requests/__pycache__/auth.cpython-310.pyc
o
�N�g�' � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z m
Z
mZ ddlmZ ddlmZ d Zd
Zd d � ZG d
d� d�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )z]
requests.auth
~~~~~~~~~~~~~
This module contains the authentication handlers for Requests.
� N)� b64encode� )�to_native_string)�
basestring�str�urlparse)�extract_cookies_to_jar)�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�passwordZauthstr� r �A/opt/saltstack/salt/lib/python3.10/site-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__H s zAuthBase.__call__N��__name__�
__module__� __qualname__�__doc__r! r r r r r E � 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__O s
zHTTPBasicAuth.__init__c C �( t | jt|dd �k| jt|dd �kg�S �Nr r ��allr �getattrr �r �otherr r r �__eq__S �
��zHTTPBasicAuth.__eq__c C �
| |k S r) r r1 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+ r3 r6 r! r r r r r( L s r( c @ r )�
HTTPProxyAuthz=Attaches HTTP Proxy Authentication to a given Request object.c C r8 )NzProxy-Authorizationr: r r r r r! f r< zHTTPProxyAuth.__call__Nr" r r r r r= c 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+ n 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 )�hasattrrA rB �
last_nonce� nonce_count�chal�pos�
num_401_calls)r r r r �init_per_thread_statet 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 }| j� d|� d| j� �}|� d|� �}� |�}� |�}|| j j kr�| j j
d7 _
nd| j _
| j j
d�}t | j j
�� d�}||� d�7 }|t
�� � d�7 }|t�d �7 }t�|��� dd � }|dkr� |� d|� d|� ��}|s�|
||� d|� ��}n"|d k�sd |�d �v �r|� d|� d|� d |� �}|
||�}ndS || j _ d!| j� d"|� d#|� d$|� d%|� d&� }|�r?|d'|� d&�7 }|�rJ|d(|� d&�7 }|�rU|d)|� d&�7 }|�rc|d*|� d+|� d&�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_utf8ZSHAc S rQ rR )r
r r rT �sha1rV rW r r r �sha_utf8� rZ z4HTTPDigestAuth.build_digest_header.<locals>.sha_utf8zSHA-256c S rQ rR )r
r r rT �sha256rV rW r r r � sha256_utf8� rZ z7HTTPDigestAuth.build_digest_header.<locals>.sha256_utf8zSHA-512c S rQ rR )r
r r rT �sha512rV rW r r r � sha512_utf8� rZ z7HTTPDigestAuth.build_digest_header.<locals>.sha512_utf8c s � | � d|� ��S )N�:r )�s�d�Z hash_utf8r r �<lambda>� s z4HTTPDigestAuth.build_digest_header.<locals>.<lambda>�/�?ra r Z08xrS � � Zauth�,z:auth:z
username="z
", realm="z
", nonce="z", uri="z
", response="�"z
, opaque="z
, algorithm="z
, digest="z, qop="auth", nc=z
, cnonce="zDigest )rA rG �get�upperr �path�queryr r rE rF r r �time�ctime�os�urandomrT r[ rV �split)r �method�urlrK rL rM rN rO Z
_algorithmrY r\ r^ r` ZKDZentdigZp_parsedrn ZA1ZA2ZHA1ZHA2Zncvaluerb ZcnonceZrespdigZnoncebit�baser rd 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)Z is_redirectrA rI )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-authenticaterC �digest� zdigest )�flags)�countr9 ) Z status_coderA rI rH �request�body�seekr; rl �lower�re�compile�
IGNORECASEr �subrG �content�close�copyr Z_cookies�rawZprepare_cookiesrx ru rv �
connection�send�history�append)r r ry Zs_auth�patZprepZ_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 )Nr9 �responser )rJ rA rE rx ru rv r; r� �tellrH �AttributeErrorZ
register_hookr� rz rI r r r r r! s �zHTTPDigestAuth.__call__c C r, r- r. r1 r r r r3 1 r4 zHTTPDigestAuth.__eq__c C r5 r) r r1 r r r r6 9 r7 zHTTPDigestAuth.__ne__N) r# r$ r% r& r+ rJ rx rz r� r! r3 r6 r r r r r> k s
n, r> )r& rT rr r� r? rp r �base64r Z_internal_utilsr �compatr r r Zcookiesr �utilsr Z CONTENT_TYPE_FORM_URLENCODEDZCONTENT_TYPE_MULTI_PARTr r r( r= r> r r r r �<module> s& ,