File: //usr/local/CyberCP/lib/python3.10/site-packages/botocore/__pycache__/httpchecksum.cpython-310.pyc
o
�ho? � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
ddl m Z m
Z
mZ ddlmZ ddlmZmZ e
rCdd lmZ ndZe�e�ZG d
d � d �ZG d d
� d
e�ZG dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd� d�Z G dd� de�Z!dd� Z" d3d d �Z#d d � Z$d d!� Z%d"d#� Z& d3d$d%�Z'd&d'� Z(d(d)� Z)d*d+� Z*ee e d,�Z+d-d.gZ,e
r�e e d/�Z-e+�.e-� e/d0d1� e-�0� D ��s�J �e1e+�0� �Z2g d2�Z3dS )4aB The interfaces in this module are not intended for public use.
This module defines interfaces for applying checksums to HTTP requests within
the context of botocore. This involves both resolving the checksum to be used
based on client configuration and environment, as well as application of the
checksum to the request.
� N)�crc32)�sha1�sha256)�HAS_CRT)�AwsChunkedWrapperError�FlexibleChecksumError�MissingDependencyException)�
StreamingBody)�conditionally_calculate_md5�determine_content_length)� checksumsc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� BaseChecksum� c C � d S �N� ��self�chunkr r �H/usr/local/CyberCP/lib/python3.10/site-packages/botocore/httpchecksum.py�update3 � zBaseChecksum.updatec C r r r �r r r r �digest6 r zBaseChecksum.digestc C s | � � }t�|��d�S )N�ascii)r �base64� b64encode�decode)r �bsr r r � b64digest9 s zBaseChecksum.b64digestc s: � � � }t� �fdd�d�D ]}��|� q� �|� d S )Nc s � � �j�S r )�read� _CHUNK_SIZEr ��fileobjr r r �<lambda>? s z.BaseChecksum._handle_fileobj.<locals>.<lambda>� )�tell�iterr �seek)r r# �start_positionr r r" r �_handle_fileobj= s z BaseChecksum._handle_fileobjc C s2 t |ttf�r| �|� | �� S | �|� | �� S r )�
isinstance�bytes� bytearrayr r* r )r �bodyr r r �handleC s
�zBaseChecksum.handleN) �__name__�
__module__� __qualname__r! r r r r* r/ r r r r r
0 s r
c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�
Crc32Checksumc C �
d| _ d S �Nr ��
_int_crc32r r r r �__init__L �
zCrc32Checksum.__init__c C s t || j�d@ | _d S �Nl �� )r r8 r r r r r O s zCrc32Checksum.updatec C � | j jddd�S �N� �big)� byteorder�r8 �to_bytesr r r r r R � zCrc32Checksum.digestN�r0 r1 r2 r9 r r r r r r r4 K � r4 c @ r3 )�CrtCrc32Checksumc C r5 r6 r7 r r r r r9 X r: zCrtCrc32Checksum.__init__c C � t �|| j�}|d@ | _d S r; )�
crt_checksumsr r8 �r r � new_checksumr r r r [ � zCrtCrc32Checksum.updatec C r< r= rA r r r r r _ rC zCrtCrc32Checksum.digestNrD r r r r rF V � rF c @ r3 )�CrtCrc32cChecksumc C r5 r6 )� _int_crc32cr r r r r9 e r: zCrtCrc32cChecksum.__init__c C rG r; )rH �crc32crN rI r r r r h rK zCrtCrc32cChecksum.updatec C r< r= )rN rB r r r r r l rC zCrtCrc32cChecksum.digestNrD r r r r rM c rL rM c @ r3 )� Sha1Checksumc C � t � | _d S r )r � _checksumr r r r r9 q � zSha1Checksum.__init__c C � | j �|� d S r �rR r r r r r r t rC zSha1Checksum.updatec C �
| j �� S r �rR r r r r r r w r: zSha1Checksum.digestNrD r r r r rP p rE rP c @ r3 )�Sha256Checksumc C rQ r )r rR r r r r r9 | rS zSha256Checksum.__init__c C rT r rU r r r r r rC zSha256Checksum.updatec C rV r rW r r r r r � r: zSha256Checksum.digestNrD r r r r rX { rE rX c @ sL e Zd ZdZ ddd�Zdd� Zdd d
�Zdd d �Zd
d� Zdd� Z dS )�AwsChunkedWrapperr N�x-amz-checksumc C s2 || _ || _|| _| �� |d u r| j}|| _d S r )�_raw�_checksum_name�
_checksum_cls�_reset�_DEFAULT_CHUNK_SIZE� _chunk_size)r �raw� checksum_cls�
checksum_name�
chunk_sizer r r r9 � s
zAwsChunkedWrapper.__init__c C s* d| _ d| _d | _| jr| �� | _d S d S )Nr% F)�
_remaining� _completerR r] r r r r r^ � s �zAwsChunkedWrapper._resetr c C s2 |dks|dkr
t dd��| �� | j�d� d S )Nr z Can only seek to start of stream�� error_msg)r r^ r[ r( )r �offset�whencer r r r( � s �zAwsChunkedWrapper.seekc C s� |d ur
|dkr
d }| j r| jsdS |d u p |t| j�k}| j s;|r;| j| �� 7 _|d u p5|t| j�k}| j s;|s"|d u rDt| j�}| jd |� }| j|d � | _|S )Nr r% )rf re �len� _make_chunk)r �size�want_more_bytes� to_returnr r r r � s
�
zAwsChunkedWrapper.readc C s� | j �| j�}tt|��dd � �d�}| | _| jr!| j�|� | jr;| jr;| j �d�}| j�
� �d�}d||f S d||f S )N� r s 0
%s:%s
s %s
%s
) r[ r r` �hexrk �encoderf rR r r\ r )r � raw_chunk�hex_len�name�checksumr r r rl � s z AwsChunkedWrapper._make_chunkc c s � | j s| �� V | j rd S d S r )rf rl r r r r �__iter__� s �
�zAwsChunkedWrapper.__iter__)NrZ N)r r )
r0 r1 r2 r_ r9 r^ r( r rl rw r r r r rY � s
�
rY c s2 e Zd Z� fdd�Zd� fdd� Zdd� Z� ZS ) �StreamingChecksumBodyc s t � �||� || _|| _d S r )�superr9 rR � _expected)r �
raw_stream�content_lengthrv �expected�� __class__r r r9 � s
z StreamingChecksumBody.__init__Nc s: t � j|d�}| j�|� |d u s|s|dkr| �� |S )N)�amtr )ry r rR r �_validate_checksum)r r� r r~ r r r � s
zStreamingChecksumBody.readc C s< | j �� t�| j�kr d| j� d| j �� � �}t|d��d S )N�Expected checksum �$ did not match calculated checksum: rg )rR r r � b64decoderz r r )r rh r r r r� � s
��
�z(StreamingChecksumBody._validate_checksumr )r0 r1 r2 r9 r r� �
__classcell__r r r~ r rx � s rx c C s t | ||� t| ||� d S r )�"resolve_request_checksum_algorithm�$resolve_response_checksum_algorithms)�request�operation_model�paramsr r r �resolve_checksum_context� s r� c
C s |j }|�d�}|rl||v rl|d u rt}|| �� }||vr7ts/|tv r/td|�� � d�d��td|� �d��d}|j rE| d �
d �rEd
}||d |� �d �}|d
| d v rXd S | d �di �} || d< | | d d< d S |j st|�d�r�| d �di �} d| d< | | d d< d S d S )N�requestAlgorithmMemberzUsing za requires an additional dependency. You will need to pip install botocore[crt] before proceeding.)�msgz Unsupported checksum algorithm: rg �header�urlzhttps:�trailer�x-amz-checksum-)� algorithm�inru ru �headers�contextrv �request_algorithm�requestChecksumRequired�conditional-md5) �
http_checksum�get� _SUPPORTED_CHECKSUM_ALGORITHMS�lowerr �_CRT_CHECKSUM_ALGORITHMSr �upperr �has_streaming_input�
startswith�http_checksum_required)
r� r� r� �supported_algorithmsr� �algorithm_member�algorithm_name�
location_typer� �checksum_contextr r r r� � sD
���
��r� c C s~ | � di �� di �}|� d�}|sd S |dkr t| � d S |d dkr)t| � d S |d dkr5t| � d S td�|d �d ��)
Nr� rv r� r� r� r� r� z Unknown checksum variant: {}rg )r� r
� _apply_request_header_checksum� _apply_request_trailer_checksumr �format)r� r� r� r r r �apply_request_checksum- s
�r� c C sd | � di �� di �}|� d�}|d }|| d v rd S t� |d �}|� �| d �}|| d |< d S )Nr� rv r� ru r� r� r. )r� �
_CHECKSUM_CLSr/ )r� r� r� �
location_namerb r r r r r� A s
r� c C s� | � di �� di �}|� d�}|d }t� |d �}| d }| d }||v r(d S d|d < d
|v r9|d
d 7 < nd |d
< ||d
< t|�}|d urOt|�|d< t|ttf�r[t�|�}t |||d�| d< d S )Nr� rv r� ru r� r� r. �chunkedzTransfer-EncodingzContent-Encodingz ,aws-chunkedz aws-chunkedz
X-Amz-Trailerz X-Amz-Decoded-Content-Length)rb rc )
r� r� r �strr+ r, r- �io�BytesIOrY )r� r� r� r� rb r� r. r| r r r r� M s.
�r� c
C s� |j }|�d�}|rH||v rJ|d u rt}dd� |�dg �D �}g }tD ]}||vr*q#||v r3|�|� q#| d �di �} || d< | | d d< d S d S d S )N�requestValidationModeMemberc S s h | ]}|� � �qS r )r� )�.0�ar r r � <setcomp>{ s �z7resolve_response_checksum_algorithms.<locals>.<setcomp>�responseAlgorithmsr� rv �response_algorithms)r� r� r� �_ALGORITHMS_PRIORITY_LIST�append)
r� r� r� r� r� � mode_memberr� �usable_algorithmsr� r� r r r r� s s&
�
��r� c C s� |d }|� di �}|� d�}|sd S |D ];}d|� �}||vr!qd|| v r(q|jr4t| ||�|d< nt| ||�|d< |d � di �}||d< ||d d< d S t�d |� d
�� d S ) Nr� rv r� r� �-r. r� �response_algorithmzXSkipping checksum validation. Response did not contain one of the following algorithms: �.)r� �has_streaming_output�_handle_streaming_response�_handle_bytes_response�logger�info) �
http_response�responser� r� r� r� �
algorithmsr� � header_namer r r �handle_checksum_body� s6
�� ��r� c C s8 t �|�}d|� �}t| j|d �d�|� |d | �S )Nr� r� zcontent-length)r� r� rx ra )r� r� r� rb r� r r r r� � s
�r� c C sj | j }d|� �}t�|�}|� }|�|� |d | }|�� t�|�kr3d|� d|�� � �}t|d��|S )Nr� r� r� r� rg ) �contentr� r� r r r r� r r ) r� r� r� r. r� rb rv r} rh r r r r� � s
��
r� )r r r r rO )r rO c c s � | ]}|t v V qd S r )r� )r� ru r r r � <genexpr>� s �
�r� )rO r r r r )4�__doc__r r� �logging�binasciir �hashlibr r �botocore.compatr �botocore.exceptionsr r r �botocore.responser �botocore.utilsr
r �awscrtr rH � getLoggerr0 r� r
r4 rF rM rP rX rY rx r� r� r� r� r� r� r� r� r� r� r� �_CRT_CHECKSUM_CLSr �all�keys�listr� r� r r r r �<module> s`
R
�: '
�) ��
�