3
گau @ s d Z ddlZddlZddlZddlmZ ddlZddljZ ddl
Z
ddl Z ddl Z ddl
Z
ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddl Z ddl Z ddl Z ddl Z ddl!m"Z" ddl#m$Z$ ddl%m&Z& ddl'm(Z( ddl'm)Z) ddl'm*Z* ddl'm+Z+ ddl,m-Z- e
j.e/Z0dZ1dZ2G dd dZ3G d d d e3Z4G d d d e3Z5G d d! d!Z6G d"d# d#Z7G d$d% d%Z8eee8e j9e/ e j9e/< dS )&zACME client API. N) parsedate_tz)
ModuleType)Any)Callable)cast)Dict)Iterable)List)Optional)Set)Text)Tuple)Union) HTTPAdapter)parse_header_links)SourceAddressAdapter) crypto_util)errors)jws)messages)VersionedLEACMEMixin- zapplication/pkix-certc @ s* e Zd ZdZejdeddddZed%e j
e e e e ej
ddd Zej
ejej
d
d d Zeee j
d
ddZd&ej
e ej ej
dddZej
ej
dddZejejdddZd'e j
e ej e e ejdddZeje j
ejd d d Zee j
ee j d d d!Z e j ee dd"d#d$Z dS )(
ClientBasezACME client base object.
:ivar messages.Directory directory:
:ivar .ClientNetwork net: Client network.
:ivar int acme_version: ACME protocol version. 1 or 2.
ClientNetworkN) directorynet acme_versionreturnc C s || _ || _|| _dS )zInitialize.
:param .messages.Directory directory: Directory Resource
:param .ClientNetwork net: Client network.
:param int acme_version: ACME protocol version. 1 or 2.
N)r r r )selfr r r r /usr/lib/python3.6/client.py__init__8 s zClientBase.__init__)responseuriterms_of_servicer c C s> d|j kr|j d d }tjtjj|j |jjd||dS )Nzterms-of-serviceurlLocation)bodyr# r$ )linksr RegistrationResource Registration from_jsonjsonheadersget)clsr" r# r$ r r r _regr_from_responseD s
z ClientBase._regr_from_response)regrr' r c C s" | j |j|}| j||j|jdS )N)r# r$ )_postr# r0 r$ )r r1 r' r" r r r _send_recv_regrP s zClientBase._send_recv_regr)argskwargsr c O s< |j d| j t| jdr.|j dt| jd | jj||S )zBWrapper around self.net.post that adds the acme_version.
r newNonce
new_nonce_url)
setdefaultr hasattrr getattrr post)r r4 r5 r r r r2 ^ s zClientBase._post)r1 updater c C s< |dkr|j n|}tjf t|}| j||d}|| j_|S )aK Update registration.
:param messages.RegistrationResource regr: Registration Resource.
:param messages.Registration update: Updated body of the
resource. If not provided, body will be taken from `regr`.
:returns: Updated Registration Resource.
:rtype: `.RegistrationResource`
N)r' )r' r UpdateRegistrationdictr3 r account)r r1 r< r' Z updated_regrr r r update_registrationg s
z ClientBase.update_registration)r1 r c C s | j |tjjdddS )zDeactivate registration.
:param messages.RegistrationResource regr: The Registration Resource
to be deactivated.
:returns: The Registration resource that was deactivated.
:rtype: `.RegistrationResource`
deactivatedN)statusZcontact)r@ r r* r+ )r r1 r r r deactivate_registrationz s z"ClientBase.deactivate_registration)authzrr c C s. t jdd}| j|j|}| j||jj|jS )a Deactivate authorization.
:param messages.AuthorizationResource authzr: The Authorization resource
to be deactivated.
:returns: The Authorization resource that was deactivated.
:rtype: `.AuthorizationResource`
rA )rB )r ZUpdateAuthorizationr2 r# _authzr_from_responser'
identifier)r rD r' r" r r r deactivate_authorization s z#ClientBase.deactivate_authorization)r" rF r# r c C sF t jt jj|j |jjd|d}|d k rB|jj|krBt j
||S )Nr& )r' r# ) r AuthorizationResourceZ
Authorizationr+ r, r- r. r' rF r UnexpectedUpdate)r r" rF r# rD r r r rE s
z ClientBase._authzr_from_response)challbr" r c C sv | j |j|}y|jd d }W n tk
r> tjdY nX tj|tjj |j
d}|j|jkrrtj |j|S )ah Answer challenge.
:param challb: Challenge Resource body.
:type challb: `.ChallengeBody`
:param response: Corresponding Challenge response
:type response: `.challenges.ChallengeResponse`
:returns: Challenge Resource with updated body.
:rtype: `.ChallengeResource`
:raises .UnexpectedUpdate:
upr% z"up" Link header missing)
authzr_urir' ) r2 r# r( KeyErrorr ClientErrorr ChallengeResource
ChallengeBodyr+ r, rI )r rJ r" rL Zchallrr r r answer_challenge s zClientBase.answer_challenge)r" defaultr c C s |j jdt|}y t|}W nv tk
r t|}|dk ry4tj|d dk rX|d nd}tj|dd | S ttfk
r Y nX |}Y nX tjj tj|d S ) a Compute next `poll` time based on response ``Retry-After`` header.
Handles integers and various datestring formats per
https://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.37
:param requests.Response response: Response from `poll`.
:param int default: Default value (in seconds), used when
``Retry-After`` header is not present or invalid.
:returns: Time point when next `poll` should be performed.
:rtype: `datetime.datetime`
z Retry-AfterN r )secondsrV )
r- r. strint
ValueErrorr datetime timedelta
OverflowErrornow)r/ r" rR retry_afterrU whenZtz_secsr r r r^ s
zClientBase.retry_after)certrsnr% r c C s0 | j |tj||d}|jtjkr,tjddS )a. Revoke certificate.
:param .ComparableX509 cert: `OpenSSL.crypto.X509` wrapped in
`.ComparableX509`
:param int rsn: Reason code for certificate revocation.
:param str url: ACME URL to post to
:raises .ClientError: If revocation is unsuccessful.
) certificatereasonz0Successful revocation must return HTTP OK statusN)r2 r
Revocation status_code http_clientZOKr rN )r r` ra r% r" r r r _revoke s
zClientBase._revoke)NN)N)NN)!__name__
__module__ __qualname____doc__r DirectoryrX r! classmethodrequestsResponser
rW r) r0 r* r3 r r2 r@ rC rH rG
IdentifierrE rP rO rQ rZ r^ joseComparableX509rg r r r r r 1 s.
$ r c
s e Zd ZdZejddfejejej e
e d dd fdd
Z d0e ej
ejdd d
Zejejd d d
Zejejd ddZd1eje e ejdddZd2ee e ejdddZejeej ejdddZejeeje j f dddZ d3ejeej e e eejeejd f f d d d!Z eee j ejf d"d#d$Z!ejejd%d&d'Z"ejejd%d(d)Z#d4eje e$ej d*d+d,Z%eje dd-d.d/Z& Z'S )5ClientaF ACME client for a v1 API.
.. deprecated:: 1.18.0
Use :class:`ClientV2` instead.
.. todo::
Clean up raised error types hierarchy, document, and handle (wrap)
instances of `.DeserializationError` raised in `from_json()`.
:ivar messages.Directory directory:
:ivar key: `josepy.JWK` (private)
:ivar alg: `josepy.JWASignature`
:ivar bool verify_ssl: Verify SSL certificates?
:ivar .ClientNetwork net: Client network. Useful for testing. If not
supplied, it will be initialized using `key`, `alg` and
`verify_ssl`.
TNr )r keyalg
verify_sslr r c sR || _ |dkr t|||d}t|tr s z4Client.poll_and_request_issuance..c S s i | ]
}||qS r r )r rD r r r
s z4Client.poll_and_request_issuance..zSleeping for %d secondsrS )rR c s s | ]}|j jtjkV qd S )N)r' rB r STATUS_INVALID)r rD r r r s z3Client.poll_and_request_issuance..c 3 s | ]} | V qd S )Nr )r rD )updatedr r r s ) r} collections defaultdictrX set enumerateheapqheapifyheappoprZ r] rU r r timesleepr r' rB r STATUS_VALIDr heappushr^ addanyvaluesr Z PollErrortupler )r r r r r ZattemptsZ exhaustedZwaitingr_ r rD r] rU r r" Zupdated_authzrsr )r r poll_and_request_issuance s8
z Client.poll_and_request_issuance)r# r c C s8 t }| jj|d|i|d}|tjtjjtjj|j fS )zReturns certificate from URI.
:param str uri: URI of certificate
:returns: tuple of the form
(response, :class:`josepy.util.ComparableX509`)
:rtype: tuple
r )r- r )
r r r. rq rr r r r r r )r r# r r" r r r _get_cert s
zClient._get_cert)certrr c C sL | j |j\}}d|jkr$tjd|jd |jkr@tj|j|j|dS )zCheck for new cert.
:param certr: Certificate Resource
:type certr: `.CertificateResource`
:returns: Updated Certificate Resource.
:rtype: `.CertificateResource`
r& zLocation header missing)r' )r r# r- r rN rI textr< )r r r" r` r r r
check_cert s
zClient.check_certc C s
| j |S )zRefresh certificate.
:param certr: Certificate Resource
:type certr: `.CertificateResource`
:returns: Updated Certificate Resource.
:rtype: `.CertificateResource`
)r )r r r r r refresh" s
zClient.refresh)r
max_lengthr c C sl g }|j }xD|dk rNt||k rN| j|\}}|jjdi jd}|j| q W |dk rhtjdj||S )a Fetch chain for certificate.
:param .CertificateResource certr: Certificate Resource
:param int max_length: Maximum allowed length of the chain.
Note that each element in the certificate requires new
``HTTP GET`` request, and the length of the chain is
controlled by the ACME CA.
:raises errors.Error: if recursion exceeds `max_length`
:returns: Certificate chain for the Certificate Resource. It is
a list ordered so that the first element is a signer of the
certificate from Certificate Resource. Will be empty if
``cert_chain_uri`` is ``None``.
:rtype: `list` of `OpenSSL.crypto.X509` wrapped in `.ComparableX509`
NrK r% z'Recursion limit reached. Didn't get {0}) r lenr r( r. appendr Errorformat)r r r chainr# r" r` r r r fetch_chain1 s zClient.fetch_chain)r` ra r c C s | j ||| jtttj dS )a Revoke certificate.
:param .ComparableX509 cert: `OpenSSL.crypto.X509` wrapped in
`.ComparableX509`
:param int rsn: Reason code for certificate revocation.
:raises .ClientError: If revocation is unsuccessful.
N)rg r r rW r rd )r r` ra r r r revokeO s z
Client.revoke)N)N)N)r r )r )(rh ri rj rk rq RS256r rl JWK JWASignatureboolr
r! r{ r) r~ r r rp rW rH r r rr r r r r
rn ro r rX r r r r r r r
__classcell__r r )ry r rs s2 *
% J rs c sZ e Zd ZdZejddd fdd Zejejddd Z ejejd
d d Z
d-eje ej ejd
fdd
Z
ejejd
ddZeejdddZejeejejf dddZd.eje ej ejdddZejejejddd Zd/ejejeejd d d Ze j e dd!d"d#Z ed$d%d&Z e!e!ejd'd(d)Z"eje#e$e# d*d+d,Z% Z&S )0ClientV2zuACME client for a v2 API.
:ivar messages.Directory directory:
:ivar .ClientNetwork net: Client network.
r N)r r r c s t j||dd dS )zInitialize.
:param .messages.Directory directory: Directory Resource
:param .ClientNetwork net: Client network.
)r r r N)rx r! )r r r )ry r r r! d s zClientV2.__init__) new_accountr c C sL | j | jd |}|jdkr6d|jkr6tj|jd | j|}|| j_|S )zRegister.
:param .NewRegistration new_account:
:raises .ConflictError: in case the account already exists
:returns: Registration Resource.
:rtype: `.RegistrationResource`
newAccount r& ) r2 r re r- r
ConflictErrorr0 r r? )r r r" r1 r r r r l s
zClientV2.new_account)r1 r c C s6 || j _| j|jd}| j||j|jd| j _| j jS )zQuery server about registration.
:param messages.RegistrationResource regr: Existing Registration
Resource.
N)r# r$ )r r? r2 r# r0 r$ )r r1 r" r r r r s
zClientV2.query_registration)r1 r< r c s | j |}t j||S )aK Update registration.
:param messages.RegistrationResource regr: Registration Resource.
:param messages.Registration update: Updated body of the
resource. If not provided, body will be taken from `regr`.
:returns: Updated Registration Resource.
:rtype: `.RegistrationResource`
)_get_v2_accountrx r@ )r r1 r< new_regr)ry r r r@ s
z ClientV2.update_registrationc C sJ d | j _|jjdd}| j| jd |}|jd }|j|d}|| j _|S )NT)Zonly_return_existingr r& )r# )r r? r' r< r2 r r- )r r1 Zonly_existing_regr" Z updated_urir r r r r s
zClientV2._get_v2_account)csr_pemr c
C s t jjt jj|}tj|}tj|}g }x"|D ]}|jtj tj
|d q0W x"|D ]}|jtj tj |d qTW tj |d}| j
| jd |} tjj| j }
g } x(|
jD ] } | j| j| j| | d qW tj|
| jjd| |dS )zRequest a new Order object from the server.
:param bytes csr_pem: A CSR in PEM format.
:returns: The newly created order.
:rtype: OrderResource
)r r ) identifiersZnewOrder)r# r& )r' r# authorizationsr )r r load_certificate_request FILETYPE_PEMr _pyopenssl_cert_or_req_all_namesZ _pyopenssl_cert_or_req_san_ipr r rp r Z
IDENTIFIER_IPZNewOrderr2 r Orderr+ r, r rE _post_as_get
OrderResourcer- r. )
r r r dnsNamesZipNamesr nameZipsorderr" r' r r% r r r new_order s*
zClientV2.new_order)rD r c C s( | j |j}| j||jj|j}||fS )a Poll Authorization Resource for status.
:param authzr: Authorization Resource
:type authzr: `.AuthorizationResource`
:returns: Updated Authorization Resource and HTTP response.
:rtype: (`.AuthorizationResource`, `requests.Response`)
)r r# rE r' rF )r rD r" r r r r r s z
ClientV2.poll)orderrdeadliner c C s6 |dkr t j j t jdd }| j||}| j||S )ad Poll authorizations and finalize the order.
If no deadline is provided, this method will timeout after 90
seconds.
:param messages.OrderResource orderr: order to finalize
:param datetime.datetime deadline: when to stop polling and timeout
:returns: finalized order
:rtype: messages.OrderResource
NZ )rU )rZ r] r[ poll_authorizationsfinalize_order)r r r r r r poll_and_finalize s zClientV2.poll_and_finalizec C s g }x\|j jD ]P}xJtjj |k r\| j| j||d}|j jtjkrP|j | P t
j d qW qW t |t |j jk r~t
j g }x@|D ]8}|j jtjkrx$|j jD ]}|jdk r|j | qW qW |rt
j||j|dS )z Poll Order Resource for status.)r# rS N)r )r' r rZ r] rE r rB r ZSTATUS_PENDINGr r r r r TimeoutErrorr Z
challengeserrorZValidationErrorr< )r r r Z responsesr% rD ZfailedZchallr r r r s&
z ClientV2.poll_authorizationsF)r r fetch_alternative_chainsr c s t jjt jj|j}tjtj|d} j |j
j | xt j j
|k rtjd j|j}tjj|j }|jdk rtj|j|jdk r8 j|j}|j||jd}|rֈ j|d} fdd| D }
|j|
d}|S q8W tj dS ) a{ Finalize an order and obtain a certificate.
:param messages.OrderResource orderr: order to finalize
:param datetime.datetime deadline: when to stop polling and timeout
:param bool fetch_alternative_chains: whether to also fetch alternative
certificate chains
:returns: finalized order
:rtype: messages.OrderResource
)r rS N)r'
fullchain_pemZ alternatec s g | ]} j |jqS r )r r )r r% )r r r r % s z+ClientV2.finalize_order..)Zalternative_fullchains_pem) r r r r r r r rq rr r2 r' finalizerZ r] r r r r# r r+ r, r r Z
IssuanceErrorrb r< r
_get_linksr ) r r r r r Z wrapped_csrr" r' Zcertificate_responseZalt_chains_urlsZ
alt_chainsr )r r r s&
zClientV2.finalize_order)r` ra r c C s | j ||| jd dS )a Revoke certificate.
:param .ComparableX509 cert: `OpenSSL.crypto.X509` wrapped in
`.ComparableX509`
:param int rsn: Reason code for certificate revocation.
:raises .ClientError: If revocation is unsuccessful.
Z
revokeCertN)rg r )r r` ra r r r r * s zClientV2.revoke)r c C s t | jdo| jjjS )zGChecks if ACME server requires External Account Binding authentication.meta)r9 r r external_account_required)r r r r r 7 s z"ClientV2.external_account_required)r4 r5 r c O s( |dd d |dd }| j ||S )z
Send GET request using the POST-as-GET protocol.
:param args:
:param kwargs:
:return:
NrS )N)r2 )r r4 r5 Znew_argsr r r r ; s zClientV2._post_as_get)r"
relation_typer c s. d|j krg S t|j d } fdd|D S )z
Retrieves all Link URIs of relation_type from the response.
:param requests.Response response: The requests HTTP response.
:param str relation_type: The relation type to filter by.
ZLinkc s0 g | ](}d |krd|kr|d kr|d qS )Zrelr% r )r l)r r r r P s z'ClientV2._get_links..)r- r )r r" r r( r )r r r E s
zClientV2._get_links)N)N)F)'rh ri rj rk r rl r! r{ r) r r r
r* r@ r bytesr r rH r
rn ro r rZ r r r r rq rr rX r r r r rW r r r r r )ry r r ] s& #
r c @ s e Zd ZdZdejeddddZeeddd Z d e
j e e
egdf e
jd
d d Zee
jd
ddZd e
jejee
jdddZejeddddZe
jedddZeddd ZdS ) BackwardsCompatibleClientV2a ACME client wrapper that tends towards V2-style calls, but
supports V1 servers.
.. deprecated:: 1.18.0
Use :class:`ClientV2` instead.
.. note:: While this class handles the majority of the differences
between versions of the ACME protocol, if you need to support an
ACME server based on version 3 or older of the IETF ACME draft
that uses combinations in authorizations (or lack thereof) to
signal that the client needs to complete something other than
any single challenge in the authorization to make it valid, the
user of this class needs to understand and handle these
differences themselves. This does not apply to either of Let's
Encrypt's endpoints where successfully completing any challenge
in an authorization will make it valid.
:ivar int acme_version: 1 or 2, corresponding to the Let's Encrypt endpoint
:ivar .ClientBase client: either Client or ClientV2
r N)r rt serverr c C sT t jj|j|j }| j|| _| | jdkrBt|||d| _nt ||d| _d S )NrS )rt r )r )
r rl r+ r. r, _acme_version_from_directoryr rs clientr )r r rt r r r r r r! j s
z$BackwardsCompatibleClientV2.__init__)r r c C s t | j|S )N)r: r )r r r r r __getattr__s s z'BackwardsCompatibleClientV2.__getattr__)r1 check_tos_cbr c s t dd fdd }| jdkrVtt| j}|j|}|jdk rR||j |j|S |S tt| j}d|j j
kr||j j
j |j dd}|j |S dS ) zCombined register and agree_tos for V1, new_account for V2
:param .NewRegistration regr:
:param callable check_tos_cb: callback that raises an error if
the check does not work
N)tosr c s d k r | d S )Nr )r )r r r _assess_tos s zDBackwardsCompatibleClientV2.new_account_and_tos.._assess_tosrS r$ T)Zterms_of_service_agreed)
rW r r rs r r~ r$ r r r r r< r )r r1 r r client_v1Z client_v2r )r r new_account_and_tosv s
z/BackwardsCompatibleClientV2.new_account_and_tos)r r c C st | j dkrbtt| j}tjjtjj|}tj |}g }x |D ]}|j
|j | q.) r r rv r8 r r r Zrequestrn
exceptionsZRequestExceptionrematchrW groupsrY r r- r. r base64Z b64encoder encodingr re joinitems)r r r% r4 r5 r" eZ err_regexmhostpathZ_err_noZerr_msgZ
debug_contentr r r
_send_request{ s8
$
zClientNetwork._send_request)r4 r5 r c O s | j d||S )a Send HEAD request without checking the response.
Note, that `_check_response` is not called, as it is expected
that status code other than successfully 2xx will be returned, or
messages2.Error will be raised by the server.
HEAD)r( )r' )r r4 r5 r r r head s zClientNetwork.head)r% r r5 r c K s | j | jd|f||dS )z$Send GET request and check response.ZGET)r )r r' )r r% r r5 r r r r. s zClientNetwork.get)r" r c C s | j |jkrz|j| j }ytjjd j|}W n0 tjk
r^ } ztj ||W Y d d }~X nX t
j d| | j j
| n
tj|d S )Nr zStoring nonce: %s)REPLAY_NONCE_HEADERr- r ZHeader_fieldsr rq r r ZBadNoncer r r r Z MissingNonce)r r" r Z
decoded_noncer r r r
_add_nonce s zClientNetwork._add_nonce)r% r7 r c C sL | j sBtjd |d kr$| j|}n| j| j|d d}| j| | j j S )NzRequesting fresh nonce)r )r r r r) r r, pop)r r% r7 r" r r r
_get_nonce s
zClientNetwork._get_noncec O sZ y | j ||S tjk
rT } z*|jdkrBtjd| | j ||S W Y dd}~X nX dS )zPOST object wrapped in `.JWS` and check response.
If the server responded with a badNonce error, the request will
be retried once.
ZbadNoncez Retrying request after error:
%sN)
_post_oncer r coder r )r r4 r5 r r r r r; s
zClientNetwork.postrS )r% r r r r5 r c K sh |j dd }| j|| j||||}|jdd|i | jd|fd|i|}| j||d}| j| |S )Nr7 r- z Content-Typer r )r )r- r r. r8 r' r r, ) r r% r r r r5 r7 r r" r r r r/ s
zClientNetwork._post_once)N)&rh ri rj rk r ZJOSE_CONTENT_TYPEr r* rq r DEFAULT_NETWORK_TIMEOUTr r
r r) r r rW rX r r
r! r ZJSONDeSerializabler
r rm rn ro r r r' r) r. r, r. r; r/ r r r r r s. > :K
r c @ sd e Zd ZdZeddddZeedddZeedd d
d Z eddd d
Z
e e dddZ dS )_ClientDeprecationModulez
Internal class delegating to a module, and displaying warnings when attributes
related to deprecated attributes in the acme.client module.
N)moduler c C s || j d< d S )N_module)__dict__)r r3 r r r r! s z!_ClientDeprecationModule.__init__)attrr c C s* |dkr t jdj|tdd t| j|S )Nrs r zHThe {0} attribute in acme.client is deprecated and will be removed soon.r )
stacklevel)rs r )warningswarnr DeprecationWarningr: r4 )r r6 r r r r s
z$_ClientDeprecationModule.__getattr__)r6 r r c C s t | j|| d S )N)setattrr4 )r r6 r r r r __setattr__ s z$_ClientDeprecationModule.__setattr__c C s t | j| d S )N)delattrr4 )r r6 r r r __delattr__ s z$_ClientDeprecationModule.__delattr__)r c C s dgt | j S )Nr4 )dirr4 )r r r r __dir__ s z _ClientDeprecationModule.__dir__)
rh ri rj rk r r! rW r r r< r> r r@ r r r r r2 s r2 ):rk r r rZ Z email.utilsr r Z http.clientr rf Zloggingr sysr typesr Ztypingr r r r r r r
r r r
r r8 Zjosepyrq r rn Zrequests.adaptersr Zrequests.utilsr Z!requests_toolbelt.adapters.sourcer Zacmer r r r Z acme.mixinsr Z getLoggerrh r r1 r r rs r r r r2 modulesr r r r sf
F i x