File: //opt/alt/python37/lib/python3.7/site-packages/botocore/__pycache__/httpchecksum.cpython-37.pyc
B
�P�e�? � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
ddl m Z m
Z
mZ ddlmZ ddlmZmZ e
r�dd 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"d4d d �Z#d d � Z$d d!� Z%d"d#� Z&d5d$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�t1�e2e+�0� �Z3d.d-d2d3gZ4dS )6aC 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 )
� BaseChecksumi c C s d S )N� )�self�chunkr r �F/opt/alt/python37/lib/python3.7/site-packages/botocore/httpchecksum.py�update2 s zBaseChecksum.updatec C s d S )Nr )r r r r �digest5 s zBaseChecksum.digestc C s | � � }t�|��d�S )N�ascii)r �base64� b64encode�decode)r �bsr r r � b64digest8 s zBaseChecksum.b64digestc s> � � � }x&t� �fdd�d�D ]}��|� q W � �|� d S )Nc s � � �j�S )N)�read� _CHUNK_SIZEr )�fileobjr r r �<lambda>> � z.BaseChecksum._handle_fileobj.<locals>.<lambda>r )�tell�iterr �seek)r r Zstart_positionr r )r r r �_handle_fileobj< s z BaseChecksum._handle_fileobjc C s, t |ttf�r| �|� n
| �|� | �� S )N)�
isinstance�bytes� bytearrayr r" r )r �bodyr r r �handleB s
zBaseChecksum.handleN) �__name__�
__module__� __qualname__r r r r r" r' r r r r r
/ s r
c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
Crc32Checksumc C s
d| _ d S )Nr )�
_int_crc32)r r r r �__init__K s zCrc32Checksum.__init__c C s t || j�d@ | _d S )Nl �� )r r, )r r r r r r N s zCrc32Checksum.updatec C s | j jddd�S )N� �big)� byteorder)r, �to_bytes)r r r r r Q s zCrc32Checksum.digestN)r( r) r* r- r r r r r r r+ J s r+ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CrtCrc32Checksumc C s
d| _ d S )Nr )r, )r r r r r- W s zCrtCrc32Checksum.__init__c C s t �|| j�}|d@ | _d S )Nl �� )�
crt_checksumsr r, )r r � new_checksumr r r r Z s zCrtCrc32Checksum.updatec C s | j jddd�S )Nr. r/ )r0 )r, r1 )r r r r r ^ s zCrtCrc32Checksum.digestN)r( r) r* r- r r r r r r r2 U s r2 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CrtCrc32cChecksumc C s
d| _ d S )Nr )� _int_crc32c)r r r r r- d s zCrtCrc32cChecksum.__init__c C s t �|| j�}|d@ | _d S )Nl �� )r3 �crc32cr6 )r r r4 r r r r g s zCrtCrc32cChecksum.updatec C s | j jddd�S )Nr. r/ )r0 )r6 r1 )r r r r r k s zCrtCrc32cChecksum.digestN)r( r) r* r- r r r r r r r5 b s r5 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� Sha1Checksumc C s t � | _d S )N)r � _checksum)r r r r r- p s zSha1Checksum.__init__c C s | j �|� d S )N)r9 r )r r r r r r s s zSha1Checksum.updatec C s
| j �� S )N)r9 r )r r r r r v s zSha1Checksum.digestN)r( r) r* r- r r r r r r r8 o s r8 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Sha256Checksumc C s t � | _d S )N)r r9 )r r r r r- { s zSha256Checksum.__init__c C s | j �|� d S )N)r9 r )r r r r r r ~ s zSha256Checksum.updatec C s
| j �� S )N)r9 r )r r r r r � s zSha256Checksum.digestN)r( r) r* r- r r r r r r r: z s r: c @ sF e Zd ZdZddd�Zdd� Zdd d
�Zdd d �Zd
d� Zdd� Z dS )�AwsChunkedWrapperi N�x-amz-checksumc C s2 || _ || _|| _| �� |d kr(| j}|| _d S )N)�_raw�_checksum_name�
_checksum_cls�_reset�_DEFAULT_CHUNK_SIZE� _chunk_size)r �raw� checksum_cls�
checksum_name�
chunk_sizer r r r- � s zAwsChunkedWrapper.__init__c C s& d| _ d| _d | _| jr"| �� | _d S )Nr F)�
_remaining� _completer9 r? )r r r r r@ � s
zAwsChunkedWrapper._resetr c C s2 |dks|dkrt 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 k r|dkrd }| j r$| js$dS |d kp8|t| j�k}x6| j sp|rp| j| �� 7 _|d kpl|t| j�k}q<W |d kr�t| j�}| jd |� }| j|d � | _|S )Nr r )rH rG �len� _make_chunk)r �sizeZwant_more_bytesZ to_returnr r r r � s
zAwsChunkedWrapper.readc C s� | j �| j�}tt|��dd � �d�}| | _| jrB| j�|� | jrv| jrv| j �d�}| j�
� �d�}d||f S d||f S )N� r s 0
%s:%s
s %s
%s
) r= r rB �hexrL �encoderH r9 r r> r )r Z raw_chunk�hex_len�name�checksumr r r rM � s z AwsChunkedWrapper._make_chunkc c s x| j s| �� V qW d S )N)rH rM )r r r r �__iter__� s zAwsChunkedWrapper.__iter__)Nr< N)r )N)
r( r) r* rA r- r@ r! r rM rU r r r r r; � s
r; c s2 e Zd Z� fdd�Zd� fdd� Zdd� Z� ZS ) �StreamingChecksumBodyc s t � �||� || _|| _d S )N)�superr- r9 � _expected)r Z
raw_stream�content_lengthrT �expected)� __class__r r r- � s z StreamingChecksumBody.__init__Nc s: t � j|d�}| j�|� |d ks.|s6|dkr6| �� |S )N)�amtr )rW r r9 r �_validate_checksum)r r\ r )r[ r r r � s
zStreamingChecksumBody.readc C s< | j �� t�| j�kr8d| j� d| j �� � �}t|d��d S )NzExpected checksum z$ did not match calculated checksum: )rI )r9 r r � b64decoderX r r )r rI r r r r] � s z(StreamingChecksumBody._validate_checksum)N)r( r) r* r- r r] �
__classcell__r r )r[ r rV � s rV c C s t | ||� t| ||� d S )N)�"resolve_request_checksum_algorithm�$resolve_response_checksum_algorithms)�request�operation_model�paramsr r r �resolve_checksum_context� s re c
C s |j }|�d�}|r�||kr�|d kr(t}|| �� }||krlts^|tkr^td|�� � d�d��td| d��d}|j r�| d �
d �r�d
}||d | d �}|d
| d kr�d S | d �di �} || d< | | d d< n6|j s�|�d��r| d �di �} d| d< | | d d< d S )NZrequestAlgorithmMemberzUsing za requires an additional dependency. You will need to pip install botocore[crt] before proceeding.)�msgz"Unsupported checksum algorithm: %s)rI �header�urlzhttps:�trailerzx-amz-checksum-%s)� algorithm�inrS rS �headers�contextrT �request_algorithmZrequestChecksumRequiredzconditional-md5) �
http_checksum�get� _SUPPORTED_CHECKSUM_ALGORITHMS�lowerr �_CRT_CHECKSUM_ALGORITHMSr �upperr Zhas_streaming_input�
startswithZhttp_checksum_required)
rb rc rd �supported_algorithmsro Zalgorithm_memberZalgorithm_nameZ
location_typerj �checksum_contextr r r r` � s:
r` c C sz | � di �� di �}|� d�}|s&d S |dkr8t| � n>|d dkrNt| � n(|d dkrdt| � ntd|d d ��d S )
Nrm rT rn zconditional-md5rk rg ri z Unknown checksum variant: %s)rI )rp r
� _apply_request_header_checksum� _apply_request_trailer_checksumr )rb rw rj r r r �apply_request_checksum, s
rz c C sd | � di �� di �}|� d�}|d }|| d kr6d S t� |d �}|� �| d �}|| d |< d S )Nrm rT rn rS rl rj r& )rp �
_CHECKSUM_CLSr' )rb rw rj �
location_namerD r r r r rx @ s
rx c C s� | � di �� di �}|� d�}|d }t� |d �}| d }| d }||krPd S d|d < d
|krr|d
d 7 < nd |d
< ||d
< t|�}|d k r�t|�|d< t|ttf�r�t�|�}t |||d�| d< d S )Nrm rT rn rS rj rl r& �chunkedzTransfer-EncodingzContent-Encodingz ,aws-chunkedz aws-chunkedz
X-Amz-Trailerz X-Amz-Decoded-Content-Length)rD rE )
rp r{ r �strr# r$ r% �io�BytesIOr; )rb rw rj r| rD rl r& rY r r r ry L s,
ry c
C s� |j }|�d�}|r�||kr�|d kr(t}dd� |�dg �D �}g }x(tD ] }||krVqH||krH|�|� qHW | d �di �} || d< | | d d< d S )NZrequestValidationModeMemberc S s h | ] }|� � �qS r )rr )�.0�ar r r � <setcomp>z s z7resolve_response_checksum_algorithms.<locals>.<setcomp>ZresponseAlgorithmsrm rT �response_algorithms)ro rp rq �_ALGORITHMS_PRIORITY_LIST�append)
rb rc rd rv ro Z mode_memberr� Zusable_algorithmsrj rw r r r ra r s
ra c C s� |d }|� di �}|� d�}|s&d S xz|D ]r}d| }||krBq,d|| krPq,|jrht| ||�|d< nt| ||�|d< |d � di �}||d< ||d d< d S W t�d |� d
�� d S ) Nrl rT r� zx-amz-checksum-%s�-r& rm Zresponse_algorithmzXSkipping checksum validation. Response did not contain one of the following algorithms: �.)rp Zhas_streaming_output�_handle_streaming_response�_handle_bytes_response�logger�info) �
http_response�responserm rc rl rw Z
algorithmsrj � header_namer r r �handle_checksum_body� s,
r� c C s6 t �|�}d| }t| j|d �d�|� |d | �S )Nzx-amz-checksum-%srl zcontent-length)r{ rp rV rC )r� r� rj rD r� r r r r� � s
r� c C sd | j }d| }t�|�}|� }|�|� |d | }|�� t�|�kr`d||�� f }t|d��|S )Nzx-amz-checksum-%srl z:Expected checksum %s did not match calculated checksum: %s)rI ) �contentr{ rp r r r r^ r r ) r� r� rj r& r� rD rT rZ rI r r r r� � s
r� )r r r r r7 )r r7 c c s | ]}|t kV qd S )N)rs )r� rS r r r � <genexpr>� s r� r r )N)N)5�__doc__r r �logging�binasciir �hashlibr r Zbotocore.compatr Zbotocore.exceptionsr r r Zbotocore.responser Zbotocore.utilsr
r Zawscrtr r3 � getLoggerr( r� r
r+ r2 r5 r8 r: r; rV re r` rz rx ry ra r� r� r� r{ rs Z_CRT_CHECKSUM_CLSr �all�keys�AssertionError�listrq r� r r r r �<module> sT
R
6 '
)