o
    NDi                     @   s  d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z	 d dl
mZ d dlmZmZ d dlmZ d dlmZmZmZmZmZmZmZmZmZmZ d dlmZmZmZm Z  d dl!m"Z"m#Z#m$Z$ e%e&Z'd	Z(d
Z)dZ*dZ+g dZ,dZ-dZ.dd Z/dd Z0G dd dZ1G dd de1Z2G dd de1Z3G dd de1Z4G dd de1Z5G dd de5Z6G d d! d!e6Z7G d"d# d#e7Z8G d$d% d%e7Z9G d&d' d'e5Z:G d(d) d)e:Z;G d*d+ d+e5Z<G d,d- d-e1Z=G d.d/ d/e=Z>G d0d1 d1e=Z?G d2d3 d3e2Z@d4d5 ZAd6d7 ZBe3e4e4e=e>e?e<e7e9e8e@d8ZCer)d d9lDmEZE eCFeE n
eCFe5e:e6e;d: d;d<d=d>d?ZGd@dA eGH D ZIdS )B    N)Mapping
formatdate)sha1sha256)
itemgetter)
HAS_CRTMD5_AVAILABLEHTTPHeadersencodebytesensure_unicodeparse_qsquoteunquoteurlsplit
urlunsplit)NoAuthTokenErrorNoCredentialsErrorUnknownSignatureVersionError UnsupportedSignatureVersionError)is_valid_ipv6_endpoint_urlnormalize_url_pathpercent_encode_sequence@e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855i   z%Y-%m-%dT%H:%M:%SZz%Y%m%dT%H%M%SZ)expectztransfer-encodingz
user-agentzx-amzn-trace-idzUNSIGNED-PAYLOADz"STREAMING-UNSIGNED-PAYLOAD-TRAILERc                 C   s\   t | }|j}t| rd| d}ddd}|jd ur,|j||jkr,| d|j }|S )N[]P   i  )httphttps:)r   hostnamer   portgetscheme)url	url_partshostdefault_ports r)   F/var/www/Datamplify/venv/lib/python3.10/site-packages/botocore/auth.py_host_from_urlI   s   
r+   c                 C   s<   | j }t|trt|d}|S t|trt|}|S Nutf-8)data
isinstancebytesjsonloadsdecodestr)requestr.   r)   r)   r*   _get_body_as_dict\   s   


r6   c                   @   s   e Zd ZdZdZdd ZdS )
BaseSignerFc                 C   s   t d)Nadd_auth)NotImplementedErrorselfr5   r)   r)   r*   r8   m   s   zBaseSigner.add_authN)__name__
__module____qualname__REQUIRES_REGIONREQUIRES_TOKENr8   r)   r)   r)   r*   r7   i   s    r7   c                   @   s   e Zd ZdZ	 dd ZdS )TokenSignerTc                 C   
   || _ d S N)
auth_token)r;   rD   r)   r)   r*   __init__w      
zTokenSigner.__init__N)r<   r=   r>   r@   rE   r)   r)   r)   r*   rA   q   s    rA   c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )		SigV2Authz+
    Sign a request with Signature V2.
    c                 C   rB   rC   credentialsr;   rI   r)   r)   r*   rE      rF   zSigV2Auth.__init__c                 C   s
  t d t|j}|j}t|dkrd}|j d|j d| d}tj	| j
jdtd}g }t|D ])}|dkr;q4t|| }	t|ddd	}
t|	dd
d	}||
 d|  q4d|}||7 }t d| ||d t|  d}||fS )Nz$Calculating signature using v2 auth.r   /
r-   	digestmod	Signature safez-_~=&zString to sign: %s)loggerdebugr   r%   pathlenmethodnetlochmacnewrI   
secret_keyencoder   sortedr4   r   appendjoinupdatebase64	b64encodedigeststripr3   )r;   r5   paramssplitrW   string_to_signlhmacpairskeyvalue
quoted_keyquoted_valueqsb64r)   r)   r*   calc_signature   s.   


zSigV2Auth.calc_signaturec                 C   s   | j d u rt |jr|j}n|j}| j j|d< d|d< d|d< ttt |d< | j j	r4| j j	|d< | 
||\}}||d< |S )	NAWSAccessKeyId2SignatureVersion
HmacSHA256SignatureMethod	TimestampSecurityTokenrO   )rI   r   r.   rg   
access_keytimestrftimeISO8601gmtimetokenrr   )r;   r5   rg   rp   	signaturer)   r)   r*   r8      s   
zSigV2Auth.add_authN)r<   r=   r>   __doc__rE   rr   r8   r)   r)   r)   r*   rG   {   s
    rG   c                   @   s   e Zd Zdd Zdd ZdS )	SigV3Authc                 C   rB   rC   rH   rJ   r)   r)   r*   rE      rF   zSigV3Auth.__init__c                 C   s   | j d u rt d|jv r|jd= tdd|jd< | j jr-d|jv r&|jd= | j j|jd< tj| j jdt	d}|
|jd d t|  }d| j j d|d }d	|jv rb|jd	= ||jd	< d S )
NDateTusegmtX-Amz-Security-Tokenr-   rM   zAWS3-HTTPS AWSAccessKeyId=z ,Algorithm=HmacSHA256,Signature=zX-Amzn-Authorization)rI   r   headersr   r   r[   r\   r]   r^   r   rb   r   re   rf   rz   r3   )r;   r5   new_hmacencoded_signaturer   r)   r)   r*   r8      s*   



zSigV3Auth.add_authN)r<   r=   r>   rE   r8   r)   r)   r)   r*   r      s    r   c                   @   s   e Zd ZdZdZdd Zd1ddZdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0S )2	SigV4Authz+
    Sign a request with Signature V4.
    Tc                 C   s   || _ || _|| _d S rC   )rI   _region_name_service_namer;   rI   service_nameregion_namer)   r)   r*   rE      s   
zSigV4Auth.__init__Fc                 C   s<   |rt ||dt }|S t ||dt }|S r,   )r[   r\   r^   r   	hexdigestre   )r;   rl   msghexsigr)   r)   r*   _sign   s
   zSigV4Auth._signc                 C   sL   t  }|j D ]\}}| }|tvr|||< qd|vr$t|j|d< |S )zk
        Select the headers from the request that need to be included
        in the StringToSign.
        r'   )r
   r   itemslowerSIGNED_HEADERS_BLACKLISTr+   r%   )r;   r5   
header_mapnamerm   lnamer)   r)   r*   headers_to_sign   s   zSigV4Auth.headers_to_signc                 C   s"   |j r	| |j S | t|jS rC   )rg   _canonical_query_string_params_canonical_query_string_urlr   r%   r:   r)   r)   r*   canonical_query_string   s   z SigV4Auth.canonical_query_stringc                 C   s~   g }t |tr| }|D ]\}}|t|ddtt|ddf qg }t|D ]\}}|| d|  q)d|}|S )Nz-_.~rQ   rS   rT   )r/   r   r   r`   r   r4   r_   ra   )r;   rg   key_val_pairsrl   rm   sorted_key_valsr   r)   r)   r*   r     s   

z(SigV4Auth._canonical_query_string_paramsc           	      C   sv   d}|j r9g }|j dD ]}|d\}}}|||f qg }t|D ]\}}|| d|  q%d|}|S )NrP   rT   rS   )queryrh   	partitionr`   r_   ra   )	r;   partsr   r   pairrl   _rm   r   r)   r)   r*   r     s   
z%SigV4Auth._canonical_query_string_urlc                    sZ   g }t t|}|D ]}d fdd||D }|| dt|  q
d|S )a  
        Return the headers that need to be included in the StringToSign
        in their canonical form by converting all header keys to lower
        case, sorting them in alphabetical order and then joining
        them into a string, separated by newlines.
        ,c                 3   s    | ]}  |V  qd S rC   )_header_value.0vr;   r)   r*   	<genexpr>/  s    

z.SigV4Auth.canonical_headers.<locals>.<genexpr>r    rL   )r_   setra   get_allr`   r   )r;   r   r   sorted_header_namesrl   rm   r)   r   r*   canonical_headers%  s   
zSigV4Auth.canonical_headersc                 C   s   d | S )N )ra   rh   )r;   rm   r)   r)   r*   r   5  s   zSigV4Auth._header_valuec                 C   s    t dd t|D }d|S )Nc                 s   s    | ]	}|   V  qd S rC   )r   rf   )r   nr)   r)   r*   r   >  s    z+SigV4Auth.signed_headers.<locals>.<genexpr>;)r_   r   ra   )r;   r   r   r)   r)   r*   signed_headers=  s   
zSigV4Auth.signed_headersc                 C   s0   |j di }|d}t|to|ddkS )Nchecksumrequest_algorithmintrailer)contextr#   r/   dict)r;   r5   checksum_context	algorithmr)   r)   r*   _is_streaming_checksum_payloadA  s   
z(SigV4Auth._is_streaming_checksum_payloadc                 C   s   |  |rtS | |stS |j}|r>t|dr>| }t|j	t
}t }t|dD ]}|| q+| }|| |S |rFt| S tS )Nseek    )r   "STREAMING_UNSIGNED_PAYLOAD_TRAILER_should_sha256_sign_payloadUNSIGNED_PAYLOADbodyhasattrtell	functoolspartialreadPAYLOAD_BUFFERr   iterrb   r   r   EMPTY_SHA256_HASH)r;   r5   request_bodypositionread_chunksizer   chunkhex_checksumr)   r)   r*   payloadF  s&   


zSigV4Auth.payloadc                 C   s   |j dsdS |jddS )Nr   Tpayload_signing_enabled)r%   
startswithr   r#   r:   r)   r)   r*   r   `  s   z%SigV4Auth._should_sha256_sign_payloadc                 C   s   |j  g}| t|jj}|| || | | |}|| 	|d  || 
| d|jv r>|jd }n| |}|| d|S )NrL   X-Amz-Content-SHA256)rY   upper_normalize_url_pathr   r%   rW   r`   r   r   r   r   r   r   ra   )r;   r5   crrW   r   body_checksumr)   r)   r*   canonical_requestj  s   





zSigV4Auth.canonical_requestc                 C   s   t t|dd}|S )Nz/~rQ   )r   r   )r;   rW   normalized_pathr)   r)   r*   r   y  s   zSigV4Auth._normalize_url_pathc                 C   sN   | j jg}||jd dd  || j || j |d d|S N	timestampr      aws4_requestrK   )rI   rz   r`   r   r   r   ra   r;   r5   scoper)   r)   r*   r   }  s   


zSigV4Auth.scopec                 C   sH   g }| |jd dd  | | j | | j | d d|S r   )r`   r   r   r   ra   r   r)   r)   r*   credential_scope  s   

zSigV4Auth.credential_scopec                 C   sH   dg}| |jd  | | | | t|d  d|S )z
        Return the canonical StringToSign as well as a dict
        containing the original version of all headers that
        were included in the StringToSign.
        AWS4-HMAC-SHA256r   r-   rL   )r`   r   r   r   r^   r   ra   )r;   r5   r   stsr)   r)   r*   ri     s
   
zSigV4Auth.string_to_signc                 C   sd   | j j}| d|  |jd dd }| || j}| || j}| |d}| j||ddS )NAWS4r   r   r   r   T)r   )rI   r]   r   r^   r   r   r   )r;   ri   r5   rl   k_datek_region	k_service	k_signingr)   r)   r*   r     s   zSigV4Auth.signaturec                 C   s   | j d u rt tj }|t|jd< | | | |}t	
d t	
d| | ||}t	
d| | ||}t	
d| | || d S )Nr   z$Calculating signature using v4 auth.zCanonicalRequest:
%szStringToSign:
%szSignature:
%s)rI   r   datetimeutcnowr|   SIGV4_TIMESTAMPr   _modify_request_before_signingr   rU   rV   ri   r   _inject_signature_to_request)r;   r5   datetime_nowr   ri   r   r)   r)   r*   r8     s   




zSigV4Auth.add_authc                 C   sV   d|  | g}| |}|d| |  |d|  d||jd< |S )NzAWS4-HMAC-SHA256 Credential=zSignedHeaders=z
Signature=, Authorization)r   r   r`   r   ra   r   )r;   r5   r   auth_strr   r)   r)   r*   r     s   
z&SigV4Auth._inject_signature_to_requestc                 C   sv   d|j v r	|j d= | | | jjr"d|j v r|j d= | jj|j d< |jdds9d|j v r2|j d= t|j d< d S d S )Nr   r   r   Tr   )r   _set_necessary_date_headersrI   r   r   r#   r   r:   r)   r)   r*   r     s   



z(SigV4Auth._modify_request_before_signingc                 C   s   d|j v r.|j d= tj|jd t}ttt|	 |j d< d|j v r,|j d= d S d S d|j v r7|j d= |jd |j d< d S )Nr   r   
X-Amz-Date)
r   r   strptimer   r   r   intcalendartimegm	timetuple)r;   r5   datetime_timestampr)   r)   r*   r     s   




z%SigV4Auth._set_necessary_date_headersN)F)r<   r=   r>   r   r?   rE   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   ri   r   r8   r   r   r   r)   r)   r)   r*   r      s2    




r   c                       s0   e Zd Z fddZ fddZdd Z  ZS )S3SigV4Authc                    s2   t  | d|jv r|jd= | ||jd< d S )Nr   )superr   r   r   r:   	__class__r)   r*   r     s   
z*S3SigV4Auth._modify_request_before_signingc                    s   |j d}t|dd }|d u ri }|dd }|d ur|S d}|j di }|d}t|tr<|ddkr<|d	 }|jd
rG||jvrIdS |j ddrRdS t 	|S )Nclient_configs3r   zContent-MD5r   r   r   headerr   r   Thas_streaming_inputF)
r   r#   getattrr/   r   r%   r   r   r   r   )r;   r5   r   	s3_configsign_payloadchecksum_headerr   r   r   r)   r*   r     s&   


z'S3SigV4Auth._should_sha256_sign_payloadc                 C      |S rC   r)   r;   rW   r)   r)   r*   r        zS3SigV4Auth._normalize_url_path)r<   r=   r>   r   r   r   __classcell__r)   r)   r   r*   r     s    )r   c                       s8   e Zd ZdZ fddZ fddZ fddZ  ZS )S3ExpressAuthTc                      t  ||| || _d S rC   )r   rE   _identity_cache)r;   rI   r   r   identity_cacher   r)   r*   rE        
zS3ExpressAuth.__init__c                    s   t  | d S rC   )r   r8   r:   r   r)   r*   r8      s   zS3ExpressAuth.add_authc                    s>   t  | d|jvr| jj|jd< d|jv r|jd= d S d S )Nzx-amz-s3session-tokenr   )r   r   r   rI   r   r:   r   r)   r*   r   #  s   

z,S3ExpressAuth._modify_request_before_signing)r<   r=   r>   REQUIRES_IDENTITY_CACHErE   r8   r   r	  r)   r)   r   r*   r
    s
    r
  c                   @      e Zd ZdZdd ZdS )S3ExpressPostAuthTc                 C   N  t j  }|t|jd< i }|jdd d ur|jd }i }g }|jdd d ur;|jd }|dd d ur;|d }||d< d|d< | ||d< |jd |d< |ddi |d| |i |d|jd i | jj	d ur| jj	|d	< |d	| jj	i t
t|d
d
|d< | |d ||d< ||jd< ||jd< d S )Nr   s3-presign-post-fieldss3-presign-post-policy
conditionsr   x-amz-algorithmx-amz-credential
x-amz-dateX-Amz-S3session-Tokenr-   policyx-amz-signaturer   r   r|   r   r   r#   r   r`   rI   r   rc   rd   r1   dumpsr^   r3   r   r;   r5   r   fieldsr  r  r)   r)   r*   r8   /  s>   




zS3ExpressPostAuth.add_authN)r<   r=   r>   r  r8   r)   r)   r)   r*   r  ,  s    r  c                       sJ   e Zd ZdZdZed fdd
Zdd Zdd	 Zd
d Zdd Z	  Z
S )S3ExpressQueryAuthi,  T)expiresc                   s   t  j||||d || _d S )N)r  r   rE   _expires)r;   rI   r   r   r  r!  r   r)   r*   rE   ]  s   	
zS3ExpressQueryAuth.__init__c                 C     |j d}d}||kr|j d= | | |}d| ||jd | j|d}| jjd ur3| jj|d< t	|j
}t|jdd}d	d
 | D }|jrT||j i |_d}	|jrc|t| d|_|rkt|d }	|	 t| }
|}|d |d |d |
|d f}t||_
d S )Ncontent-type0application/x-www-form-urlencoded; charset=utf-8r   r   zX-Amz-AlgorithmzX-Amz-Credentialr   zX-Amz-ExpireszX-Amz-SignedHeadersr  Tkeep_blank_valuesc                 S      i | ]	\}}||d  qS r   r)   r   kr   r)   r)   r*   
<dictcomp>      zES3ExpressQueryAuth._modify_request_before_signing.<locals>.<dictcomp>rP   rT   r            r   r#   r   r   r   r   r#  rI   r   r   r%   r   r   r   rg   rb   r.   r6   r   r   )r;   r5   content_typeblocklisted_content_typer   auth_paramsr&   query_string_parts
query_dictoperation_paramsnew_query_stringpnew_url_partsr)   r)   r*   r   n  >   
z1S3ExpressQueryAuth._modify_request_before_signingc                 C      | j d| 7  _ d S Nz&X-Amz-Signature=r%   r;   r5   r   r)   r)   r*   r        z/S3ExpressQueryAuth._inject_signature_to_requestc                 C   r  rC   r)   r  r)   r)   r*   r     r  z&S3ExpressQueryAuth._normalize_url_pathc                 C      t S rC   r   r:   r)   r)   r*   r        zS3ExpressQueryAuth.payload)r<   r=   r>   DEFAULT_EXPIRESr  rE   r   r   r   r   r	  r)   r)   r   r*   r   Y  s    	Ar   c                       s4   e Zd ZdZef fdd	Zdd Zdd Z  ZS )SigV4QueryAuth  c                    r  rC   r"  )r;   rI   r   r   r!  r   r)   r*   rE     r  zSigV4QueryAuth.__init__c                 C   r$  )Nr%  r&  r   r   r'  r   Tr(  c                 S   r*  r+  r)   r,  r)   r)   r*   r.    r/  zASigV4QueryAuth._modify_request_before_signing.<locals>.<dictcomp>rP   rT   r   r0  r1  r2  r3  )r;   r5   r4  blacklisted_content_typer   r6  r&   r7  r8  r9  r:  r;  r<  r)   r)   r*   r     r=  z-SigV4QueryAuth._modify_request_before_signingc                 C   r>  r?  r@  rA  r)   r)   r*   r     rB  z+SigV4QueryAuth._inject_signature_to_request)r<   r=   r>   rF  rE   r   r   r	  r)   r)   r   r*   rG    s    ArG  c                   @   s    e Zd ZdZdd Zdd ZdS )S3SigV4QueryAutha  S3 SigV4 auth using query parameters.

    This signer will sign a request using query parameters and signature
    version 4, i.e a "presigned url" signer.

    Based off of:

    http://docs.aws.amazon.com/AmazonS3/latest/API/sigv4-query-string-auth.html

    c                 C   r  rC   r)   r  r)   r)   r*   r     r  z$S3SigV4QueryAuth._normalize_url_pathc                 C   rC  rC   rD  r:   r)   r)   r*   r   "  rE  zS3SigV4QueryAuth.payloadN)r<   r=   r>   r   r   r   r)   r)   r)   r*   rJ    s    rJ  c                   @   r  )S3SigV4PostAuthz
    Presigns a s3 post

    Implementation doc here:
    http://docs.aws.amazon.com/AmazonS3/latest/API/sigv4-UsingHTTPPOST.html
    c                 C   r  )Nr   r  r  r  r   r  r  r  x-amz-security-tokenr-   r  r  r  r  r)   r)   r*   r8   2  s:   



zS3SigV4PostAuth.add_authNr<   r=   r>   r   r8   r)   r)   r)   r*   rK  *      rK  c                   @   sx   e Zd Zg dZdddZdd Zdd Zd	d
 Zdd ZdddZ		dddZ
	dddZdd Zdd Zdd ZdS )
HmacV1Auth)$
accelerateaclcorsdefaultObjectAcllocationlogging
partNumberr  requestPaymenttorrent
versioning	versionIdversionswebsiteuploadsuploadIdzresponse-content-typezresponse-content-languagezresponse-expireszresponse-cache-controlzresponse-content-dispositionzresponse-content-encodingdelete	lifecycletaggingrestorestorageClassnotificationreplicationrW  	analyticsmetrics	inventoryselectzselect-typezobject-lockNc                 C   rB   rC   rH   r   r)   r)   r*   rE     rF   zHmacV1Auth.__init__c                 C   s>   t j| jjdtd}||d t| 	 
dS )Nr-   rM   )r[   r\   rI   r]   r^   r   rb   r   re   rf   r3   )r;   ri   r   r)   r)   r*   sign_string  s
   zHmacV1Auth.sign_stringc                 C   s   g d}g }d|v r|d= |   |d< |D ])}d}|D ]}| }|| d ur6||kr6|||   d}q|s>|d qd|S )N)content-md5r%  dater   FTrP   rL   )	_get_dater   r`   rf   ra   )r;   r   interesting_headershoiihfoundrl   lkr)   r)   r*   canonical_standard_headers  s"   

z%HmacV1Auth.canonical_standard_headersc                 C   s   g }i }|D ] }|  }|| d ur&|dr&ddd ||D ||< qt| }|D ]}|| d||   q/d|S )Nx-amz-r   c                 s   s    | ]}|  V  qd S rC   )rf   r   r)   r)   r*   r     s    
z6HmacV1Auth.canonical_custom_headers.<locals>.<genexpr>r    rL   )r   r   ra   r   r_   keysr`   )r;   r   ro  custom_headersrl   rr  sorted_header_keysr)   r)   r*   canonical_custom_headers  s   


z#HmacV1Auth.canonical_custom_headersc                 C   s$   t |dkr|S |d t|d fS )z(
        TODO: Do we need this?
        r0  r   )rX   r   )r;   nvr)   r)   r*   	unquote_v  s   zHmacV1Auth.unquote_vc                    s   |d ur|}n|j }|jrC|jd}dd |D } fdd|D }t|dkrC|jtdd dd |D }|d7 }|d|7 }|S )	NrT   c                 S   s   g | ]}| d dqS )rS   r0  rh   r   ar)   r)   r*   
<listcomp>  s    z1HmacV1Auth.canonical_resource.<locals>.<listcomp>c                    s$   g | ]}|d   j v r |qS r+  )QSAOfInterestrz  r|  r   r)   r*   r~    s    r   )rl   c                 S   s   g | ]}d  |qS )rS   )ra   r|  r)   r)   r*   r~    s    ?)rW   r   rh   rX   sortr   ra   )r;   rh   	auth_pathbufqsar)   r   r*   canonical_resource  s   	
zHmacV1Auth.canonical_resourcec                 C   sN   |  d }|| |d 7 }| |}|r||d 7 }|| j||d7 }|S )NrL   r  )r   rs  rx  r  )r;   rY   rh   r   r!  r  csrv  r)   r)   r*   canonical_string  s   
zHmacV1Auth.canonical_stringc                 C   sF   | j jr|d= | j j|d< | j||||d}td|  | |S )NrL  r  zStringToSign:
)rI   r   r  rU   rV   rj  )r;   rY   rh   r   r!  r  ri   r)   r)   r*   get_signature  s   
zHmacV1Auth.get_signaturec                 C   s\   | j d u rttd t|j}td|j  | j|j||j|j	d}| 
|| d S )Nz(Calculating signature using hmacv1 auth.zHTTP request method: r  )rI   r   rU   rV   r   r%   rY   r  r   r  _inject_signature)r;   r5   rh   r   r)   r)   r*   r8     s   


zHmacV1Auth.add_authc                 C   s
   t ddS )NTr   r   r   r)   r)   r*   rm    rF   zHmacV1Auth._get_datec                 C   s4   d|j v r	|j d= d| jj d| }||j d< d S )Nr   zAWS r    )r   rI   rz   )r;   r5   r   auth_headerr)   r)   r*   r    s   
zHmacV1Auth._inject_signature)NNrC   )r<   r=   r>   r  rE   rj  rs  rx  rz  r  r  r  r8   rm  r  r)   r)   r)   r*   rO  Z  s    
'
	

rO  c                   @   s0   e Zd ZdZdZefddZdd Zdd Zd	S )
HmacV1QueryAuthz
    Generates a presigned request for s3.

    Spec from this document:

    http://docs.aws.amazon.com/AmazonS3/latest/dev/RESTAuthentication.html
    #RESTAuthenticationQueryStringAuth

    rH  c                 C   s   || _ || _d S rC   )rI   r#  )r;   rI   r!  r)   r)   r*   rE     s   
zHmacV1QueryAuth.__init__c                 C   s   t tt t| j S rC   )r4   r   r{   r#  r   r)   r)   r*   rm    s   zHmacV1QueryAuth._get_datec           	      C   s   i }| j j|d< ||d< |jD ]"}| }|dkr!|jd |d< q|ds*|dv r1|j| ||< qt|}t|j}|d rH|d  d| }|d	 |d
 |d ||d f}t||_d S )Nrs   rO   r   Expiresrt  )rk  r%     rT   r   r0  r1  r2  )	rI   rz   r   r   r   r   r   r%   r   )	r;   r5   r   r8  
header_keyrr  r:  r;  r<  r)   r)   r*   r    s    

z!HmacV1QueryAuth._inject_signatureN)r<   r=   r>   r   rF  rE   rm  r  r)   r)   r)   r*   r    s    
r  c                   @   r  )HmacV1PostAuthz
    Generates a presigned post for s3.

    Spec from this document:

    http://docs.aws.amazon.com/AmazonS3/latest/dev/UsingHTTPPOST.html
    c                 C   s   i }|j dd d ur|j d }i }g }|j dd d ur.|j d }|dd d ur.|d }||d< | jj|d< | jjd urM| jj|d< |d| jji tt	|
dd|d< | |d |d< ||j d< ||j d< d S )	Nr  r  r  rs   rL  r-   r  r   )r   r#   rI   rz   r   r`   rc   rd   r1   r  r^   r3   rj  )r;   r5   r  r  r  r)   r)   r*   r8   A  s,   


zHmacV1PostAuth.add_authNrM  r)   r)   r)   r*   r  8  s    r  c                   @   r  )
BearerAuthz
    Performs bearer token authorization by placing the bearer token in the
    Authorization header as specified by Section 2.1 of RFC 6750.

    https://datatracker.ietf.org/doc/html/rfc6750#section-2.1
    c                 C   s>   | j d u rt d| j j }d|jv r|jd= ||jd< d S )NzBearer r   )rD   r   r   r   )r;   r5   r  r)   r)   r*   r8   h  s   

zBearerAuth.add_authNrM  r)   r)   r)   r*   r  `  rN  r  c                 C   sR   | D ]!}|dkrt |   S |t v rt | }|tv r|  S qt|dt| d)Nsmithy.api#noAuthsignature_version)AUTH_TYPE_TO_SIGNATURE_VERSIONAUTH_TYPE_MAPSr   r   )
auth_trait	auth_typer  r)   r)   r*   resolve_auth_typer  s   

r  c                    s   dd |D  dd | D }|rt dd|  |   } fddt|D }|D ]}|dkr9t|   S t|}|tv rF|  S q-tdt	 d)	Nc                 S   s   g | ]	}| d d qS #r{  r   r$   r)   r)   r*   r~    r/  z2resolve_auth_scheme_preference.<locals>.<listcomp>c                 S   s   g | ]}|t vr|qS r)   )AUTH_PREF_TO_SIGNATURE_VERSIONr  r)   r)   r*   r~    
    z-Unsupported auth schemes in preference list: r   c                    s   g | ]}| v r|qS r)   r)   r  service_supportedr)   r*   r~    r  noAuthr  )
rU   rV   ra   r   fromkeysr  r#   r  r   r_   )preference_listauth_optionsunsupportedcombinedprioritized_schemesr$   sig_versionr)   r  r*   resolve_auth_scheme_preference  s,   

r  )v2v3v3httpsr   zs3-queryzs3-presign-postzs3v4-presign-postzv4-s3expresszv4-s3express-queryzv4-s3express-presign-postbearer)CRT_AUTH_TYPE_MAPS)v4zv4-querys3v4z
s3v4-queryr  v4ar  none)zaws.auth#sigv4zaws.auth#sigv4azsmithy.api#httpBearerAuthr  c                 C   s    i | ]\}}| d d |qS r  r{  )r   auth_schemer  r)   r)   r*   r.    s    r.  )Jrc   r   r   r   r[   r1   rU  r{   collections.abcr   email.utilsr   hashlibr   r   operatorr   botocore.compatr   r	   r
   r   r   r   r   r   r   r   botocore.exceptionsr   r   r   r   botocore.utilsr   r   r   	getLoggerr<   rU   r   r   r}   r   r   r   r   r+   r6   r7   rA   rG   r   r   r   r
  r  r   rG  rJ  rK  rO  r  r  r  r  r  r  botocore.crt.authr  rb   r  r   r  r)   r)   r)   r*   <module>   s   0

=  6-hQ0 *5(!

