File: //opt/saltstack/salt/lib/python3.10/site-packages/requests/__pycache__/auth.cpython-310.pyc
o
;j�' � @ 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| �d�td� t| �} t |t�s*tjdt|��d�td� t|�}t | t�r4| �d�} t |t�r>|�d�}dtt d�
| |f��� � � }|S )z Returns a Basic Auth string.zpNon-string usernames will no longer be supported in Requests 3.0.0. Please convert the object you've passed in (zC) to a string or bytes object in the near future to avoid problems.)�categoryzpNon-string passwords will no longer be supported in Requests 3.0.0. Please convert the object you've passed in (�latin1zBasic � :) �
isinstancer �warnings�warn�DeprecationWarningr �type�encoder r �join�strip)�username�passwordZauthstr� r �A/opt/saltstack/salt/lib/python3.10/site-packages/requests/auth.py�_basic_auth_str s0
��
��
�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 r0 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* r2 r5 r r r r r r' L s r' c @ r )�
HTTPProxyAuthz=Attaches HTTP Proxy Authentication to a given Request object.c C r7 )NzProxy-Authorizationr9 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 )�hasattrr@ rA �
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 rP rQ )r
r r rS �sha1rU rV r r r �sha_utf8� rY z4HTTPDigestAuth.build_digest_header.<locals>.sha_utf8zSHA-256c S rP rQ )r
r r rS �sha256rU rV r r r � sha256_utf8� rY z7HTTPDigestAuth.build_digest_header.<locals>.sha256_utf8zSHA-512c S rP rQ )r
r r rS �sha512rU rV r r r � sha512_utf8� rY 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>�/�?r` r Z08xrR � � 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 )r@ rF �get�upperr �path�queryr r rD rE r r �time�ctime�os�urandomrS rZ rU �split)r �method�urlrJ rK rL rM rN Z
_algorithmrX r[ r] r_ ZKDZentdigZp_parsedrm ZA1ZA2ZHA1ZHA2Zncvaluera ZcnonceZrespdigZnoncebit�baser rc 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_redirectr@ rH )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-authenticaterB �digest� zdigest )�flags)�countr8 ) Z status_coder@ rH rG �request�body�seekr: rk �lower�re�compile�
IGNORECASEr �subrF �content�close�copyr Z_cookies�rawZprepare_cookiesrw rt ru �
connection�send�history�append)r r rx 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 )Nr8 �responser )rI r@ rD rw rt ru r: r �tellrG �AttributeErrorZ
register_hookr� ry rH r r r r r s �zHTTPDigestAuth.__call__c C r+ r, r- r0 r r r r2 1 r3 zHTTPDigestAuth.__eq__c C r4 r( r r0 r r r r5 9 r6 zHTTPDigestAuth.__ne__N) r" r# r$ r% r* rI rw ry r� r r2 r5 r r r r r= k s
n, r= )r% rS rq r� r> ro 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& ,