o
    cW                     @   s   d 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
 ddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlZddlmZ ddlmZ ddlZddlmZ ddlmZ eeZedddZG dd dej Z!G dd dej Z"G dd de!Z#G dd de!Z$G dd de"Z%G dd de$ej&dZ'e"j(G d d! d!e%Z)e!j(G d"d# d#e'Z*e"j(G d$d% d%e%Z+e!j(G d&d' d'e'Z,e"j(G d(d) d)e%Z-e!j(G d*d+ d+e'Z.e!j(G d,d- d-e$Z/e"j(G d.d/ d/e"Z0dS )0z&ACME Identifier Validation Challenges.    N)Any)cast)Dict)Mapping)Optional)Tuple)Type)TypeVar)Union)hashes)crypto)SSL)crypto_util)errorsGenericChallenge	Challenge)boundc                	       s`   e Zd ZU dZi Zeeed  f ed< e	dee
 deeef dee
df f fddZ  ZS )	r   zACME challenge.TYPESclsjobjreturnUnrecognizedChallengec              
      sP   z
t tt |W S  tjy' } zt| t|W  Y d }~S d }~ww N)	r   r   super	from_jsonjoseUnrecognizedTypeErrorloggerdebugr   )r   r   error	__class__ </opt/certbot/lib/python3.10/site-packages/acme/challenges.pyr   %   s   
zChallenge.from_json)__name__
__module____qualname____doc__r   r   strr   __annotations__classmethodr   r   r   r
   r   __classcell__r"   r"   r    r#   r       s   
 


c                       sH   e Zd ZU dZi Zeeed  f ed< deee	f f fddZ
  ZS )ChallengeResponsezACME challenge response.r   r   c                    s   t   }|| jd  |S r   )r   to_partial_jsonpoptype_field_nameselfr   r    r"   r#   r-   4   s   
z!ChallengeResponse.to_partial_json)r$   r%   r&   r'   r   r   r(   r   r)   r   r-   r+   r"   r"   r    r#   r,   /   s   
 "r,   c                       sx   e Zd ZU dZeeef ed< deeef ddf fddZ	deeef fddZ
edeeef dd fd	d
Z  ZS )r   al  Unrecognized challenge.

    ACME specification defines a generic framework for challenges and
    defines some standard challenges that are implemented in this
    module. However, other implementations (including peers) might
    define additional challenge types, which should be ignored if
    unrecognized.

    :ivar jobj: Original JSON decoded object.

    r   r   Nc                    s   t    t| d| d S )Nr   )r   __init__object__setattr__r0   r    r"   r#   r2   J   s   
zUnrecognizedChallenge.__init__c                 C   s   | j S r   )r   r1   r"   r"   r#   r-   N   s   z%UnrecognizedChallenge.to_partial_jsonc                 C   s   | |S r   r"   )r   r   r"   r"   r#   r   Q   s   zUnrecognizedChallenge.from_json)r$   r%   r&   r'   r   r(   r   r)   r   r2   r-   r*   r   r+   r"   r"   r    r#   r   <   s   
 $r   c                   @   sR   e Zd ZU dZdZ	 ejdejej	ej
edddZeed< edefdd	Zd
S )_TokenChallengez3Challenge with token.

    :ivar bytes token:

    g      0@tokenT)sizeminimum)encoderdecoderr   c                 C   s   d| j vo	d| j vS )zIs `token` good?

        .. todo:: acme-spec wants "It MUST NOT contain any non-ASCII
           characters", but it should also warrant that it doesn't
           contain ".." or "/"...

        s   ..   /)r7   r5   r"   r"   r#   
good_tokene   s   z_TokenChallenge.good_tokenN)r$   r%   r&   r'   
TOKEN_SIZEr   fieldencode_b64jose	functoolspartialdecode_b64joser7   bytesr)   propertyboolr=   r"   r"   r"   r#   r6   V   s   
 
r6   c                       s`   e Zd ZU dZedZeed< e	j
Zdddejdefdd	Zdeeef f fd
dZ  ZS )!KeyAuthorizationChallengeResponsez[Response to Challenges based on Key Authorization.

    :param str key_authorization:

    keyAuthorizationkey_authorizationchallKeyAuthorizationChallengeaccount_public_keyr   c                 C   s   | j d}t|dkrtd| j  dS |d |dkr,td|d |d dS t|j| j	d
 }|d	 |krItd
|d | dS dS )a%  Verify the key authorization.

        :param KeyAuthorization chall: Challenge that corresponds to
            this response.
        :param JWK account_public_key:

        :return: ``True`` iff verification of the key authorization was
            successful.
        :rtype: bool

        .   z)Key authorization (%r) is not well formedFr   r7   z8Mismatching token in key authorization: %r instead of %rhash_function   z=Mismatching thumbprint in key authorization: %r instead of %rT)rI   splitlenr   r   encoder   	b64encode
thumbprintthumbprint_hash_functiondecode)r1   rJ   rL   partsrV   r"   r"   r#   verify}   s,   z(KeyAuthorizationChallengeResponse.verifyc                    s   t   }|dd  |S )NrH   )r   r-   r.   r0   r    r"   r#   r-      s   
z1KeyAuthorizationChallengeResponse.to_partial_json)r$   r%   r&   r'   r   r?   rI   r(   r)   r   SHA256rW   JWKrF   rZ   r   r   r-   r+   r"   r"   r    r#   rG   t   s   
 " rG   c                
   @   s   e Zd ZU dZeZeed< eZe	e
 ed< e
jZdejdefddZdejde
fdd	Zejdejd
edefddZdejded
edee
ef fddZdS )rK   zChallenge based on Key Authorization.

    :param response_cls: Subclass of `KeyAuthorizationChallengeResponse`
        that will be used to generate ``response``.
    :param str typ: type of the challenge
    typresponse_clsaccount_keyr   c                 C   s&   |  dd t|j| jd  S )zZGenerate Key Authorization.

        :param JWK account_key:
        :rtype str:

        r7   rM   rO   )rT   r   rU   rV   rW   rX   r1   r_   r"   r"   r#   rI      s   z+KeyAuthorizationChallenge.key_authorizationc                 C   s   | j | |dS )zGenerate response to the challenge.

        :param JWK account_key:

        :returns: Response (initialized `response_cls`) to the challenge.
        :rtype: KeyAuthorizationChallengeResponse

        rI   )r^   rI   r`   r"   r"   r#   response   s   	z"KeyAuthorizationChallenge.responsekwargsc                 K   s   t  )a  Generate validation for the challenge.

        Subclasses must implement this method, but they are likely to
        return completely different data structures, depending on what's
        necessary to complete the challenge. Interpretation of that
        return value must be known to the caller.

        :param JWK account_key:
        :returns: Challenge-specific validation.

        )NotImplementedErrorr1   r_   rc   r"   r"   r#   
validation   s   z$KeyAuthorizationChallenge.validationargsc                 O   s"   |  || j|g|R i |fS )zGenerate response and validation.

        Convenience function that return results of `response` and
        `validation`.

        :param JWK account_key:
        :rtype: tuple

        )rb   rf   )r1   r_   rg   rc   r"   r"   r#   response_and_validation   s   z1KeyAuthorizationChallenge.response_and_validationN)r$   r%   r&   r'   NotImplementedr]   r(   r)   r^   r   rG   rW   r   r\   rI   rb   abcabstractmethodr   rf   r   rh   r"   r"   r"   r#   rK      s   
 
rK   )	metaclassc                   @   s0   e Zd ZdZdZdddedejdefdd	Z	d
S )DNS01ResponsezACME dns-01 challenge response.zdns-01rJ   DNS01domainrL   r   c                 C   s   |  ||}|std |S )a  Simple verify.

        This method no longer checks DNS records and is a simple wrapper
        around `KeyAuthorizationChallengeResponse.verify`.

        :param challenges.DNS01 chall: Corresponding challenge.
        :param str domain: Domain name being verified.
        :param JWK account_public_key: Public key for the key pair
            being authorized.

        :return: ``True`` iff verification of the key authorization was
            successful.
        :rtype: bool

        4Verification of key authorization in response failed)rZ   r   r   )r1   rJ   ro   rL   verifiedr"   r"   r#   simple_verify   s   
zDNS01Response.simple_verifyN)
r$   r%   r&   r'   r]   r(   r   r\   rF   rr   r"   r"   r"   r#   rm      s     rm   c                   @   sJ   e Zd ZdZeZejZdZ	 dej	de
defddZdedefd	d
ZdS )rn   zACME dns-01 challenge._acme-challenger_   unused_kwargsr   c                 K   s$   t t| |d  S )SGenerate validation.

        :param JWK account_key:
        :rtype: str

        utf-8)r   rU   hashlibsha256rI   rT   digestrX   r1   r_   rt   r"   r"   r#   rf     s   zDNS01.validationnamec                 C   s   | j  d| S )z{Domain name for TXT validation record.

        :param str name: Domain name being validated.
        :rtype: str

        rM   )LABELr1   r{   r"   r"   r#   validation_domain_name  s   zDNS01.validation_domain_nameN)r$   r%   r&   r'   rm   r^   r]   r|   r   r\   r   r(   rf   r~   r"   r"   r"   r#   rn     s    
rn   c                   @   sL   e Zd ZdZdZdZ	 dZ	 	dddd	ed
ej	de
e dedefddZdS )HTTP01Responsez ACME http-01 challenge response.zhttp-01P   z
	 N   rJ   HTTP01ro   rL   porttimeoutr   c           
   
   C   s   |  ||std dS |dur#|| jkr#td| |d|7 }||}td|j| z
tj	|d|d}W n tj
jyV } ztd|| W Y d}~dS d}~ww d	|_td
||j|j |j| j}	| j|	krztd| j|	 dS dS )a  Simple verify.

        :param challenges.SimpleHTTP chall: Corresponding challenge.
        :param str domain: Domain name being verified.
        :param JWK account_public_key: Public key for the key pair
            being authorized.
        :param int port: Port used in the validation.
        :param int timeout: Timeout in seconds.

        :returns: ``True`` iff validation with the files currently served by the
            HTTP server is successful.
        :rtype: bool

        rp   FNz4Using non-standard port for http-01 verification: %sz:{0}zVerifying %s at %s...)rZ   r   zUnable to reach %s: %sasciizReceived %s: %s. Headers: %szEKey authorization from response (%r) doesn't match HTTP response (%r)T)rZ   r   r   PORTwarningformaturir]   requestsget
exceptionsRequestExceptionr   encodingtextheadersrstripWHITESPACE_CUTSETrI   )
r1   rJ   ro   rL   r   r   r   http_responser   challenge_responser"   r"   r#   rr   /  s:   


zHTTP01Response.simple_verify)Nr   )r$   r%   r&   r'   r]   r   r   r(   r   r\   r   intrF   rr   r"   r"   r"   r#   r     s    r   c                   @   s\   e Zd ZdZeZejZdZ	 ede	fddZ
de	de	fddZd	ejd
ede	fddZdS )r   zACME http-01 challenge.z.well-known/acme-challenger   c                 C   s   d| j  d | d S )zQPath (starting with '/') for provisioned resource.

        :rtype: str

        /r7   )URI_ROOT_PATHrT   r5   r"   r"   r#   pathq  s   zHTTP01.pathro   c                 C   s   d| | j  S )zCreate an URI to the provisioned resource.

        Forms an URI to the HTTPS server provisioned resource
        (containing :attr:`~SimpleHTTP.token`).

        :param str domain: Domain name being verified.
        :rtype: str

        zhttp://)r   )r1   ro   r"   r"   r#   r   z  s   
z
HTTP01.urir_   rt   c                 K   s
   |  |S )ru   ra   rz   r"   r"   r#   rf     s   
zHTTP01.validationN)r$   r%   r&   r'   r   r^   r]   r   rE   r(   r   r   r   r\   r   rf   r"   r"   r"   r#   r   h  s    r   c                   @   s   e Zd ZdZdZdZ	 dZdZede	fddZ
ddedeej dedeejejf fddZ				ddedee dee dejfddZdedejdefddZ				ddddedejdeej dee dee defddZd	S )TLSALPN01Responsez$ACME tls-alpn-01 challenge response.ztls-alpn-01i  s   1.3.6.1.5.5.7.1.30.1s
   acme-tls/1r   c                 C   s   t | jd S )z*Hash value stored in challenge certificaterv   )rw   rx   rI   rT   ry   r5   r"   r"   r#   h  s   zTLSALPN01Response.hN   ro   keybitsc                 C   s\   |du rt  }|t j| dt| jd }t j| jd|d}t	j
||gd|gd|fS )a  Generate tls-alpn-01 certificate.

        :param str domain: Domain verified by the challenge.
        :param OpenSSL.crypto.PKey key: Optional private key used in
            certificate generation. If not provided (``None``), then
            fresh key will be generated.
        :param int bits: Number of bits for newly generated key.

        :rtype: `tuple` of `OpenSSL.crypto.X509` and `OpenSSL.crypto.PKey`

        Ns   DER:hexT)criticalvalue)	force_san
extensions)r   PKeygenerate_keyTYPE_RSAcodecsrT   r   X509ExtensionID_PE_ACME_IDENTIFIER_V1r   gen_ss_cert)r1   ro   r   r   	der_valueacme_extensionr"   r"   r#   gen_cert  s   zTLSALPN01Response.gen_certhostr   c                 C   sL   |du rt |}td|| |du r| j}tj| || | jgdS )zProbe tls-alpn-01 challenge certificate.

        :param str domain: domain being validated, required.
        :param str host: IP address used to probe the certificate.
        :param int port: Port used to probe the certificate.

        Nz%s resolved to %s)r   r   r{   alpn_protocols)	socketgethostbynamer   r   r   r   	probe_snirT   ACME_TLS_1_PROTOCOL)r1   ro   r   r   r"   r"   r#   
probe_cert  s   	
zTLSALPN01Response.probe_certcertc                 C   s   t |}td|d| t|dks|d  | kr!dS t| D ]}|	|}|
 dkr?| }|| jk  S q'dS )a  Verify tls-alpn-01 challenge certificate.

        :param str domain: Domain name being validated.
        :param OpensSSL.crypto.X509 cert: Challenge certificate.

        :returns: Whether the certificate was successfully verified.
        :rtype: bool

        zCertificate %s. SANs: %srx   rQ   r   Fs   UNDEF)r    _pyopenssl_cert_or_req_all_namesr   r   ry   rS   lowerrangeget_extension_countget_extensionget_short_nameget_datar   )r1   ro   r   namesiextdatar"   r"   r#   verify_cert  s   

 
zTLSALPN01Response.verify_certrJ   	TLSALPN01rL   c              
   C   s~   |  ||std dS |du r9z
| j|||d}W n tjy8 } ztjt|dd W Y d}~dS d}~ww | ||S )a  Simple verify.

        Verify ``validation`` using ``account_public_key``, optionally
        probe tls-alpn-01 certificate and check using `verify_cert`.

        :param .challenges.TLSALPN01 chall: Corresponding challenge.
        :param str domain: Domain name being validated.
        :param JWK account_public_key:
        :param OpenSSL.crypto.X509 cert: Optional certificate. If not
            provided (``None``) certificate will be retrieved using
            `probe_cert`.
        :param string host: IP address used to probe the certificate.
        :param int port: Port used to probe the certificate.


        :returns: ``True`` if and only if client's control of the domain has been verified.
        :rtype: bool

        rp   FN)ro   r   r   T)exc_info)rZ   r   r   r   r   Errorr(   r   )r1   rJ   ro   rL   r   r   r   r   r"   r"   r#   rr     s   
zTLSALPN01Response.simple_verify)Nr   )NN)NNN)r$   r%   r&   r'   r]   r   r   r   rE   rD   r   r(   r   r   r   r   r   X509r   r   rF   r   r   r\   rr   r"   r"   r"   r#   r     s>    

r   c                   @   sP   e Zd ZdZeZejZdejde	de
ejejf fddZedefddZd	S )
r   zACME tls-alpn-01 challenge.r_   rc   r   c                 K   s,   t t| |j|dt t|ddS )a  Generate validation.

        :param JWK account_key:
        :param str domain: Domain verified by the challenge.
        :param OpenSSL.crypto.PKey cert_key: Optional private key used
            in certificate generation. If not provided (``None``), then
            fresh key will be generated.

        :rtype: `tuple` of `OpenSSL.crypto.X509` and `OpenSSL.crypto.PKey`

        cert_keyro   )r   ro   )r   r   rb   r   r   r(   re   r"   r"   r#   rf     s   zTLSALPN01.validationc                   C   s   t tjdot tjdS )ai  
        Check if TLS-ALPN-01 challenge is supported on this machine.
        This implies that a recent version of OpenSSL is installed (>= 1.0.2),
        or a recent cryptography version shipped with the OpenSSL library is installed.

        :returns: ``True`` if TLS-ALPN-01 is supported on this machine, ``False`` otherwise.
        :rtype: bool

        set_alpn_protosset_alpn_select_callback)hasattrr   
ConnectionContextr"   r"   r"   r#   is_supported(  s   
zTLSALPN01.is_supportedN)r$   r%   r&   r'   r   r^   r]   r   r\   r   r   r   r   r   rf   staticmethodrF   r   r"   r"   r"   r#   r     s    $r   c                	   @   s   e Zd ZdZdZdZ	 ejfdejdej	de
dejfdd	Zd
ejdejdefddZdejde
ddfddZdedefddZdS )DNSzACME "dns" challenge.dnsrs   r_   algrc   r   c                 K   s(   t jjd| jddd||d|S )zGenerate validation.

        :param .JWK account_key: Private account key.
        :param .JWA alg:

        :returns: This challenge wrapped in `.JWS`
        :rtype: .JWS

        T)	sort_keysrv   )payloadr   r   Nr"   )r   JWSsign
json_dumpsrT   )r1   r_   r   rc   r"   r"   r#   gen_validation?  s   zDNS.gen_validationrf   rL   c              
   C   s^   |j |dsdS z| | |jdkW S  tjy. } ztd| W Y d}~dS d}~ww )zwCheck validation.

        :param JWS validation:
        :param JWK account_public_key:
        :rtype: bool

        )r   Frv   z&Checking validation for DNS failed: %sN)rZ   
json_loadsr   rX   r   DeserializationErrorr   r   )r1   rf   rL   r   r"   r"   r#   check_validationN  s   
zDNS.check_validationDNSResponsec                 K   s   t | j|fi |dS )zGenerate response.

        :param .JWK account_key: Private account key.
        :param .JWA alg:

        :rtype: DNSResponse

        )rf   )r   r   re   r"   r"   r#   gen_response_  s   	zDNS.gen_responser{   c                 C   s   d | j|S )zgDomain name for TXT validation record.

        :param str name: Domain name being validated.

        z{0}.{1})r   r|   r}   r"   r"   r#   r~   j  s   zDNS.validation_domain_nameN)r$   r%   r&   r'   r]   r|   r   RS256r\   JWASignaturer   r   r   rF   r   r   r(   r~   r"   r"   r"   r#   r   7  s    
r   c                   @   sJ   e Zd ZU dZdZejdejjdZ	eje
d< dddejdefd	d
ZdS )r   z@ACME "dns" challenge response.

    :param JWS validation:

    r   rf   )r;   rJ   r   rL   r   c                 C   s   | | j|S )z~Check validation.

        :param challenges.DNS chall:
        :param JWK account_public_key:

        :rtype: bool

        )r   rf   )r1   rJ   rL   r"   r"   r#   r   ~  s   	zDNSResponse.check_validationN)r$   r%   r&   r'   r]   r   r?   r   r   rf   r)   r\   rF   r   r"   r"   r"   r#   r   s  s
   
 r   )1r'   rj   r   rA   rw   loggingr   typingr   r   r   r   r   r   r   r	   r
   cryptography.hazmat.primitivesr   josepyr   OpenSSLr   r   r   acmer   r   	getLoggerr$   r   r   TypedJSONObjectWithFieldsr   r,   r   r6   rG   ABCMetarK   registerrm   rn   r   r   r   r   r   r   r"   r"   r"   r#   <module>   s`    
2AH' %;