
    /Jfx                        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mZ d dl	m
Z
 d dlmZmZmZ d dlmZmZ d dlmZ  G d d          Z G d	 d
          Zd ZddZ G d d          Zd Z	 ddZd Z	 ddZd ZdS )    N)create_request_objectprepare_request_dict)OrderedDict)UnknownClientMethodErrorUnknownSignatureVersionError UnsupportedSignatureVersionError)	ArnParserdatetime2timestamp)fix_s3_hostc                       e Zd ZdZ	 ddZed             Zed             Zed             ZddZ		 	 	 	 dd	Z
d
 Zd Z	 	 ddZeZ	 	 	 ddZdS )RequestSignera0  
    An object to sign requests before they go out over the wire using
    one of the authentication mechanisms defined in ``auth.py``. This
    class fires two events scoped to a service and operation name:

    * choose-signer: Allows overriding the auth signer name.
    * before-sign: Allows mutating the request before signing.

    Together these events allow for customization of the request
    signing pipeline, including overrides, request path manipulation,
    and disabling signing per operation.


    :type service_id: botocore.model.ServiceId
    :param service_id: The service id for the service, e.g. ``S3``

    :type region_name: string
    :param region_name: Name of the service region, e.g. ``us-east-1``

    :type signing_name: string
    :param signing_name: Service signing name. This is usually the
                         same as the service name, but can differ. E.g.
                         ``emr`` vs. ``elasticmapreduce``.

    :type signature_version: string
    :param signature_version: Signature name like ``v4``.

    :type credentials: :py:class:`~botocore.credentials.Credentials`
    :param credentials: User credentials with which to sign requests.

    :type event_emitter: :py:class:`~botocore.hooks.BaseEventHooks`
    :param event_emitter: Extension mechanism to fire events.
    Nc                     || _         || _        || _        || _        || _        || _        t          j        |          | _        d S N)	_region_name_signing_name_signature_version_credentials_auth_token_service_idweakrefproxy_event_emitter)self
service_idregion_namesigning_namesignature_versioncredentialsevent_emitter
auth_tokens           S/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/botocore/signers.py__init__zRequestSigner.__init__D   sN     ()"3'%% &mM::    c                     | j         S r   )r   r   s    r!   r   zRequestSigner.region_nameX   s      r#   c                     | j         S r   )r   r%   s    r!   r   zRequestSigner.signature_version\   s    &&r#   c                     | j         S r   )r   r%   s    r!   r   zRequestSigner.signing_name`   s    !!r#   c                 .    |                      ||          S r   )sign)r   operation_namerequestkwargss       r!   handlerzRequestSigner.handlerd   s    
 yy111r#   standardc           	      N   |}|| j         }|| j        }|                     |||j                  }| j                            d                    | j                                        |          ||| j         || |           |t          j
        k    r|||d}	|||	d<   |j                            di           }
|s |
                    d          r|
d         |	d<   |
                    d	          r|
d	         |	d	<   |
                    d
          r|
d
         |	d
<   |
                    d          #|                     |	|
d         |
d                    	  | j        di |	}n*# t          $ r}|dk    rt          |          |d}~ww xY w|                    |           dS dS )a<  Sign a request before it goes out over the wire.

        :type operation_name: string
        :param operation_name: The name of the current operation, e.g.
                               ``ListBuckets``.
        :type request: AWSRequest
        :param request: The request object to be sent over the wire.

        :type region_name: str
        :param region_name: The region to sign the request for.

        :type signing_type: str
        :param signing_type: The type of signing to perform. This can be one of
            three possible values:

            * 'standard'     - This should be used for most requests.
            * 'presign-url'  - This should be used when pre-signing a request.
            * 'presign-post' - This should be used when pre-signing an S3 post.

        :type expires_in: int
        :param expires_in: The number of seconds the presigned url is valid
            for. This parameter is only valid for signing type 'presign-url'.

        :type signing_name: str
        :param signing_name: The name to use for the service when signing.
        Nzbefore-sign.{}.{})r+   r   r   r   request_signerr*   )r   r   r   expiressigningregionr   r   request_credentialsidentity_cache	cache_keyr.   r    )r   r   _choose_signercontextr   emitformatr   	hyphenizebotocoreUNSIGNEDget_resolve_identity_cacheget_auth_instancer   r   add_auth)r   r*   r+   r   signing_type
expires_inr   explicit_region_namer   r,   signing_contextauthes                r!   r)   zRequestSigner.signk   sB   F  ++K-L //L'/
 

 	  && **,,n  %)/) 	! 
	
 
	
 
	
  111 ,*%6 F
 %$.y!%o11)R@@O' BO,?,?,I,I B(7(A}%"">22 I)8)H~&""#899 0?)1,- ""#344@,,#$45#K0  
-t-7777/   :--:*;    G MM'"""""C 21s   E$ $
F.FFc                     ||d<   ||d<   d S )Nr5   r6   r8   )r   r,   cacher6   s       r!   rA   z%RequestSigner._resolve_identity_cache   s    #( '{r#   c                 N   ddd}|                     |d          }|                     d          p| j        }|                     di           }|                     d| j                  }|                     d| j                  }	|t          j        ur|                    |          s||z  }| j                            d		                    | j
                                        |          ||	||
          \  }
}|*|}|t          j        ur|                    |          s||z  }|S )ai  
        Allow setting the signature version via the choose-signer event.
        A value of `botocore.UNSIGNED` means no signing will be performed.

        :param operation_name: The operation to sign.
        :param signing_type: The type of signing that the signer is to be used
            for.
        :return: The signature version to sign with.
        z-presign-postz-query)presign-postpresign-url 	auth_typer2   r   r3   zchoose-signer.{}.{})r   r   r   r:   )r@   r   r   r   r>   r?   endswithr   emit_until_responser<   r   r=   )r   r*   rD   r:   signing_type_suffix_mapsuffixr   r2   r   r   r-   responses               r!   r9   zRequestSigner._choose_signer   sY    ,##
 #
 ),,\2>> $KK44O8O++i,,{{>43EFFkk(D,=>>X%666%..v66 7 ' /CC!(( **,,n  &#/ D 
 
  ( "):::)226:: ; "V+!  r#   c                 2   || j         }t          j        j                            |          }|t          |          |j        du r/d}| j        | j                                        } ||          }|S |p| j	        }	t          |dd          du r(|d         }
|d         }|
                    |          }	|d= d}|	|	                                }||d<   |j        r/| j        t          j                                        ||d<   ||d	<    |d
i |}|S )a  
        Get an auth instance which can be used to sign a request
        using the given signature version.

        :type signing_name: string
        :param signing_name: Service signing name. This is usually the
                             same as the service name, but can differ. E.g.
                             ``emr`` vs. ``elasticmapreduce``.

        :type region_name: string
        :param region_name: Name of the service region, e.g. ``us-east-1``

        :type signature_version: string
        :param signature_version: Signature name like ``v4``.

        :rtype: :py:class:`~botocore.auth.BaseSigner`
        :return: Auth instance to sign a request.
        Nr7   TREQUIRES_IDENTITY_CACHEr5   r6   r   r   service_namer8   )r   r>   rH   AUTH_TYPE_MAPSr@   r   REQUIRES_TOKENr   get_frozen_tokenr   getattrget_credentialsget_frozen_credentialsREQUIRES_REGIONr   
exceptionsNoRegionError)r   r   r   r   r4   r,   clsfrozen_tokenrH   r   rK   keyfrozen_credentialss                r!   rB   zRequestSigner.get_auth_instance   sc   4 $ $ 7m*../@AA;."3    %%L+#/@@BB3|$$DK)>T->31488D@@+,E%C//44K{# ""!,!C!C!E!E 2} 	2 ()77999$/F=!%1F>"s}}V}}r#     c                     t          |          }|                     |||d||           |                                 |j        S )a  Generates a presigned url

        :type request_dict: dict
        :param request_dict: The prepared request dictionary returned by
            ``botocore.awsrequest.prepare_request_dict()``

        :type operation_name: str
        :param operation_name: The operation being signed.

        :type expires_in: int
        :param expires_in: The number of seconds the presigned url is valid
            for. By default it expires in an hour (3600 seconds)

        :type region_name: string
        :param region_name: The region name to sign the presigned url.

        :type signing_name: str
        :param signing_name: The name to use for the service when signing.

        :returns: The presigned url
        rN   )r   r)   prepareurl)r   request_dictr*   rE   r   r   r+   s          r!   generate_presigned_urlz$RequestSigner.generate_presigned_urlC  sS    : (55			
 	
 	
 	{r#   r   NN)Nr.   NN)rf   NN)__name__
__module____qualname____doc__r"   propertyr   r   r   r-   r)   rA   r9   rB   get_authrk   r8   r#   r!   r   r   !   s&          T ; ; ; ;( ! ! X! ' ' X' " " X"2 2 2 2 \# \# \# \#|( ( (0! 0! 0!l  ? ? ? ?D !H ( ( ( ( ( (r#   r   c                   6    e Zd ZdZd ZddZd Z	 ddZd ZdS )	CloudFrontSignera  A signer to create a signed CloudFront URL.

    First you create a cloudfront signer based on a normalized RSA signer::

        import rsa
        def rsa_signer(message):
            private_key = open('private_key.pem', 'r').read()
            return rsa.sign(
                message,
                rsa.PrivateKey.load_pkcs1(private_key.encode('utf8')),
                'SHA-1')  # CloudFront requires SHA-1 hash
        cf_signer = CloudFrontSigner(key_id, rsa_signer)

    To sign with a canned policy::

        signed_url = cf_signer.generate_signed_url(
            url, date_less_than=datetime(2015, 12, 1))

    To sign with a custom policy::

        signed_url = cf_signer.generate_signed_url(url, policy=my_policy)
    c                 "    || _         || _        dS )a  Create a CloudFrontSigner.

        :type key_id: str
        :param key_id: The CloudFront Key Pair ID

        :type rsa_signer: callable
        :param rsa_signer: An RSA signer.
               Its only input parameter will be the message to be signed,
               and its output will be the signed content as a binary string.
               The hash algorithm needed by CloudFront is SHA-1.
        N)key_id
rsa_signer)r   rv   rw   s      r!   r"   zCloudFrontSigner.__init__  s     $r#   Nc                 T   |duo|du}|du o|du }|s|rd}t          |          ||                     ||          }t          |t                    r|                    d          }|!dt          t          |                    z  g}n,d|                     |                              d          z  g}| 	                    |          }|
                    d|                     |                              d           d| j         g           |                     ||          S )a  Creates a signed CloudFront URL based on given parameters.

        :type url: str
        :param url: The URL of the protected object

        :type date_less_than: datetime
        :param date_less_than: The URL will expire after that date and time

        :type policy: str
        :param policy: The custom policy, possibly built by self.build_policy()

        :rtype: str
        :return: The signed URL.
        Nz=Need to provide either date_less_than or policy, but not bothutf8z
Expires=%sz	Policy=%sz
Signature=zKey-Pair-Id=)
ValueErrorbuild_policy
isinstancestrencodeintr
   _url_b64encodedecoderw   extendrv   
_build_url)	r   ri   date_less_thanpolicyboth_args_suppliedneither_arg_suppliedrI   params	signatures	            r!   rk   z'CloudFrontSigner.generate_presigned_url  sQ    ,47NF$<N-5H&D. 	 !5 	 OAQ--%&&sN;;Ffc"" 	+]]6**F%"S);N)K)K%L%LLMFF!D$7$7$?$?$F$Fv$N$NNOFOOF++	LT00;;BB6JJLL,t{,,	
 	
 	
 sF+++r#   c                 H    d|v rdnd}||z   d                     |          z   S )N?&)join)r   base_urlextra_params	separators       r!   r   zCloudFrontSigner._build_url  s/    (??CC	)#chh|&<&<<<r#   c                 .   t          t          |                    }t          dd|ii          }|rd|vr|dz  }d|i|d<   |r#t          t          |                    }d|i|d<   d|fd	|fg}d
t          |          gi}t          j        |d          S )a0  A helper to build policy.

        :type resource: str
        :param resource: The URL or the stream filename of the protected object

        :type date_less_than: datetime
        :param date_less_than: The URL will expire after the time has passed

        :type date_greater_than: datetime
        :param date_greater_than: The URL will not be valid until this time

        :type ip_address: str
        :param ip_address: Use 'x.x.x.x' for an IP, or 'x.x.x.x/x' for a subnet

        :rtype: str
        :return: The policy in a compact string.
        DateLessThanzAWS:EpochTime/z/32zAWS:SourceIp	IpAddressDateGreaterThanResource	Condition	Statement),:)
separators)r   r
   r   jsondumps)	r   resourcer   date_greater_than
ip_addressmoment	conditionordered_payloadcustom_policys	            r!   r{   zCloudFrontSigner.build_policy  s    : '7788/61J KLL	 	B*$$e#
&4j%AIk" 	E+,=>>??F,;V+DI'(&1K3KL${?'C'C&DEz-J????r#   c                     t          j        |                              dd                              dd                              dd          S )N   +   -   =   _   /   ~)base64	b64encodereplace)r   datas     r!   r   zCloudFrontSigner._url_b64encode  sD     T""WT4  WT4  WT4  		
r#   rl   )	rm   rn   ro   rp   r"   rk   r   r{   r   r8   r#   r!   rt   rt   n  s         .% % %$, $, $, $,L= = =
 LP(@ (@ (@ (@T
 
 
 
 
r#   rt   c                     t           | d<   d S )Ngenerate_db_auth_token)r   class_attributesr,   s     r!   add_generate_db_auth_tokenr         1G-...r#   c                     |}|| j         j        }d|d}ddi |dd}d}| | d	| }	t          ||	           | j                            d||d
d          }
|
t          |          d         S )a  Generates an auth token used to connect to a db with IAM credentials.

    :type DBHostname: str
    :param DBHostname: The hostname of the database to connect to.

    :type Port: int
    :param Port: The port number the database is listening on.

    :type DBUsername: str
    :param DBUsername: The username to log in as.

    :type Region: str
    :param Region: The region the database is in. If None, the client
        region will be used.

    :return: A presigned url which can be used as an auth token.
    Nconnect)ActionDBUserr   rO   GET)url_pathquery_stringheadersbodymethodzhttps://r   i  zrds-db)r*   rj   r   rE   r   )metar   r   _request_signerrk   len)r   
DBHostnamePort
DBUsernameRegionr3   r   rj   schemeendpoint_urlpresigned_urls              r!   r   r     s    $ F~&  F  L F1j11411L|444(?? ! @  M V''r#   c                   $    e Zd Zd Z	 	 	 	 ddZdS )S3PostPresignerc                     || _         d S r   )r   )r   r0   s     r!   r"   zS3PostPresigner.__init__/  s    -r#   Nrf   c                    |i }|g }i }t           j                                         }|t          j        |          z   }|                    t          j        j                  |d<   g |d<   |D ]}	|d                             |	           t          |          }
||
j	        d<   ||
j	        d<   | j
                            d|
|d           |
j        |d	S )
a  Generates the url and the form fields used for a presigned s3 post

        :type request_dict: dict
        :param request_dict: The prepared request dictionary returned by
            ``botocore.awsrequest.prepare_request_dict()``

        :type fields: dict
        :param fields: A dictionary of prefilled form fields to build on top
            of.

        :type conditions: list
        :param conditions: A list of conditions to include in the policy. Each
            element can be either a list or a structure. For example:
            [
             {"acl": "public-read"},
             {"bucket": "mybucket"},
             ["starts-with", "$key", "mykey"]
            ]

        :type expires_in: int
        :param expires_in: The number of seconds the presigned post is valid
            for.

        :type region_name: string
        :param region_name: The region name to sign the presigned post to.

        :rtype: dict
        :returns: A dictionary with two elements: ``url`` and ``fields``.
            Url is the url to post to. Fields is a dictionary filled with
            the form fields and respective values to use when submitting the
            post. For example:

            {'url': 'https://mybucket.s3.amazonaws.com
             'fields': {'acl': 'public-read',
                        'key': 'mykey',
                        'signature': 'mysignature',
                        'policy': 'mybase64 encoded policy'}
            }
        N)seconds
expiration
conditionszs3-presign-post-fieldszs3-presign-post-policy	PutObjectrM   )ri   fields)datetimeutcnow	timedeltastrftimer>   rH   ISO8601appendr   r:   r   r)   ri   )r   rj   r   r   rE   r   r   datetime_nowexpire_dater   r+   s              r!   generate_presigned_postz'S3PostPresigner.generate_presigned_post2  s   ^ >FJ   (//11"X%7
%K%K%KK*33HM4IJJ|  "|# 	3 	3I< ''	2222 (554:014:01!!+~	
 	
 	
 {f555r#   )NNrf   N)rm   rn   ro   r"   r   r8   r#   r!   r   r   .  sK        . . . K6 K6 K6 K6 K6 K6r#   r   c                     t           | d<   d S )Nrk   )rk   r   s     r!   add_generate_presigned_urlr     r   r#   rf   c                    |}|}|i }|}|}dt          |           d}	| j        }
	 | j        |         }n# t          $ r t	          |          w xY w| j        j                            |          }|                     |||	          }t          j
        |                    dd                    }|                     |||	|           \  }}}|                     ||||	|d	
          }|||d<   |
                    |||          S )ax  Generate a presigned url given a client, its method, and arguments

    :type ClientMethod: string
    :param ClientMethod: The client method to presign for

    :type Params: dict
    :param Params: The parameters normally passed to
        ``ClientMethod``.

    :type ExpiresIn: int
    :param ExpiresIn: The number of seconds the presigned url is valid
        for. By default it expires in an hour (3600 seconds)

    :type HttpMethod: string
    :param HttpMethod: The http method to use on the generated url. By
        default, the http method is whatever is used in the method's model.

    :returns: The presigned url
    NTis_presign_requestuse_global_endpoint)method_name
api_paramsoperation_modelr:   BucketrO   ignore_signing_regionFr   r   r   r:   r   set_user_agent_headerr   )rj   rE   r*   )_should_use_global_endpointr   _PY_TO_OP_NAMEKeyErrorr   r   service_modelr   _emit_api_paramsr	   is_arnr@   _resolve_endpoint_ruleset_convert_to_request_dictrk   )r   ClientMethodParams	ExpiresIn
HttpMethodclient_methodr   rE   http_methodr:   r0   r*   r   bucket_is_arnr   additional_headers
propertiesrj   s                     r!   rk   rk     s   , !MF~JK":4@@ G
 )NB,]; B B B&=AAAAB i-==nMMO""' #  F
 $VZZ"%=%=>>M
 	&&#00	 	' 	 		 00'!"# 1  L !,X 00!% 1   s	   5 Ac                     t           | d<   d S )Nr   )r   r   s     r!   add_generate_presigned_postr     s    2I.///r#   c           	         |}|}|}|}	|}
|i }n|                                 }|	g }	dt          |           d}t          | j                  }| j        j                            d          }|                     d|i||          }t          j	        |
                    dd                    }|                     ||||           \  }}}|                     |||||d	
          }|	                    d|i           |                    d          r/|	                    dd|dt          d                    g           n|	                    d|i           ||d<   |                    |||	|
          S )a	  Builds the url and the form fields used for a presigned s3 post

    :type Bucket: string
    :param Bucket: The name of the bucket to presign the post to. Note that
        bucket related conditions should not be included in the
        ``conditions`` parameter.

    :type Key: string
    :param Key: Key name, optionally add ${filename} to the end to
        attach the submitted filename. Note that key related conditions and
        fields are filled out for you and should not be included in the
        ``Fields`` or ``Conditions`` parameter.

    :type Fields: dict
    :param Fields: A dictionary of prefilled form fields to build on top
        of. Elements that may be included are acl, Cache-Control,
        Content-Type, Content-Disposition, Content-Encoding, Expires,
        success_action_redirect, redirect, success_action_status,
        and x-amz-meta-.

        Note that if a particular element is included in the fields
        dictionary it will not be automatically added to the conditions
        list. You must specify a condition for the element as well.

    :type Conditions: list
    :param Conditions: A list of conditions to include in the policy. Each
        element can be either a list or a structure. For example:

        [
         {"acl": "public-read"},
         ["content-length-range", 2, 5],
         ["starts-with", "$success_action_redirect", ""]
        ]

        Conditions that are included may pertain to acl,
        content-length-range, Cache-Control, Content-Type,
        Content-Disposition, Content-Encoding, Expires,
        success_action_redirect, redirect, success_action_status,
        and/or x-amz-meta-.

        Note that if you include a condition, you must specify
        the a valid value in the fields dictionary as well. A value will
        not be added automatically to the fields dictionary based on the
        conditions.

    :type ExpiresIn: int
    :param ExpiresIn: The number of seconds the presigned post
        is valid for.

    :rtype: dict
    :returns: A dictionary with two elements: ``url`` and ``fields``.
        Url is the url to post to. Fields is a dictionary filled with
        the form fields and respective values to use when submitting the
        post. For example:

        {'url': 'https://mybucket.s3.amazonaws.com
         'fields': {'acl': 'public-read',
                    'key': 'mykey',
                    'signature': 'mysignature',
                    'policy': 'mybase64 encoded policy'}
        }
    NTr   CreateBucketr   r   rO   r   Fr   bucketz${filename}zstarts-withz$keyrd   )rj   r   r   rE   )copyr   r   r   r   r   r   r   r	   r   r@   r   r   r   rQ   r   r   )r   r   KeyFields
Conditionsr   r   rd   r   r   rE   r:   post_presignerr   r   r   r   r   r   rj   s                       r!   r   r     s   B F
CFJJ~
 #:4@@ G
 %T%9::N i-==nMMO""f%' #  F
 $VZZ"%=%=>>M
 	&&#00	 	' 	 		 00'!"# 1  L x())) ||M"" (=&#6K]9K9K8K6K2LMNNNN5#,''' F5M11!	 2   r#   c                    | j         j        dk    rdS | j         j        j        }|rc|                    dd          rdS |                    d          dk    r| j         j        j        dk    rdS |                    d          dk    rdS d	S )
NawsFuse_dualstack_endpointus_east_1_regional_endpointregionalz	us-east-1addressing_stylevirtualT)r   	partitionconfigs3r@   r   )client	s3_configs     r!   r   r   _  s    {%%u"%I 	==1599 	5MM788JFF".+==5==+,,	9954r#   r   )Nrf   N)NNrf   )r   r   r   r   r>   botocore.authbotocore.awsrequestr   r   botocore.compatr   botocore.exceptionsr   r   r   botocore.utilsr	   r
   r   r   rt   r   r   r   r   rk   r   r   r   r8   r#   r!   <module>r     s            K K K K K K K K ' ' ' ' ' '         
 9 8 8 8 8 8 8 8 ' & & & & &J J J J J J J JZ
C
 C
 C
 C
 C
 C
 C
 C
LH H H3( 3( 3( 3(lO6 O6 O6 O6 O6 O6 O6 O6dH H H
 AEL L L L^J J J
 @DE E E EP    r#   