o
    Ç `½3  ã                   @   sp  d dl mZmZmZ d dlZd dlZd dlmZ d dlZd dl	m
Z
 d dlmZ d dlmZmZmZ e ¡ e ¡ e ¡ e ¡ e ¡ dœZG dd	„ d	eƒZG d
d„ deƒZdd„ eD ƒZejejejejejfZdd„ ZG dd„ deƒZdd„ eD ƒZdd„ Zdd„ ZG dd„ de ƒZ!G dd„ de ƒZ"G dd„ de ƒZ#e $ej%¡G dd„ de ƒƒZ&e $ej%¡G dd „ d e ƒƒZ'dS )!é    )Úabsolute_importÚdivisionÚprint_functionN)ÚEnum)Úx509)Úhashes)Ú_EARLIEST_UTC_TIMEÚ_convert_to_naive_utc_timeÚ_reject_duplicate_extension)z1.3.14.3.2.26z2.16.840.1.101.3.4.2.4z2.16.840.1.101.3.4.2.1z2.16.840.1.101.3.4.2.2z2.16.840.1.101.3.4.2.3c                   @   s   e Zd ZdZdZdS )ÚOCSPResponderEncodingzBy HashzBy NameN)Ú__name__Ú
__module__Ú__qualname__ZHASHÚNAME© r   r   úlhome/ych/rk3568/buildroot/output/rockchip_rk3568/host/lib/python3.10/site-packages/cryptography/x509/ocsp.pyr      s    r   c                   @   s$   e Zd ZdZdZdZdZdZdZdS )ÚOCSPResponseStatusr   é   é   é   é   é   N)	r   r   r   Ú
SUCCESSFULZMALFORMED_REQUESTZINTERNAL_ERRORZ	TRY_LATERZSIG_REQUIREDZUNAUTHORIZEDr   r   r   r   r   $   s    r   c                 C   ó   i | ]}|j |“qS r   ©Úvalue©Ú.0Úxr   r   r   Ú
<dictcomp>-   ó    r   c                 C   s   t | tƒs	tdƒ‚d S )Nz9Algorithm must be SHA1, SHA224, SHA256, SHA384, or SHA512)Ú
isinstanceÚ_ALLOWED_HASHESÚ
ValueError)Ú	algorithmr   r   r   Ú_verify_algorithm7   s
   
ÿÿr%   c                   @   s   e Zd ZdZdZdZdS )ÚOCSPCertStatusr   r   r   N)r   r   r   ZGOODÚREVOKEDÚUNKNOWNr   r   r   r   r&   >   s    r&   c                 C   r   r   r   r   r   r   r   r   D   r    c                 C   ó   ddl m} | | ¡S ©Nr   ©Úbackend)Ú,cryptography.hazmat.backends.openssl.backendr,   Úload_der_ocsp_request©Údatar,   r   r   r   r.   G   ó   
r.   c                 C   r)   r*   )r-   r,   Úload_der_ocsp_responser/   r   r   r   r2   M   r1   r2   c                   @   s2   e Zd Zdg fdd„Zdd„ Zdd„ Zdd	„ ZdS )
ÚOCSPRequestBuilderNc                 C   s   || _ || _d S ©N)Ú_requestÚ_extensions)ÚselfZrequestÚ
extensionsr   r   r   Ú__init__T   s   
zOCSPRequestBuilder.__init__c                 C   sL   | j d ur	tdƒ‚t|ƒ t|tjƒrt|tjƒstdƒ‚t|||f| jƒS )Nz.Only one certificate can be added to a requestú%cert and issuer must be a Certificate)	r5   r#   r%   r!   r   ÚCertificateÚ	TypeErrorr3   r6   )r7   ÚcertÚissuerr$   r   r   r   Úadd_certificateX   s   
ÿz"OCSPRequestBuilder.add_certificatec                 C   sD   t |tjƒs
tdƒ‚t |j||¡}t|| jƒ t| j	| j|g ƒS ©Nz"extension must be an ExtensionType)
r!   r   ÚExtensionTyper<   Ú	ExtensionÚoidr
   r6   r3   r5   ©r7   Ú	extensionÚcriticalr   r   r   Úadd_extensiond   s   ÿz OCSPRequestBuilder.add_extensionc                 C   s(   ddl m} | jd u rtdƒ‚| | ¡S )Nr   r+   z*You must add a certificate before building)r-   r,   r5   r#   Zcreate_ocsp_request)r7   r,   r   r   r   Úbuildo   s   

zOCSPRequestBuilder.build)r   r   r   r9   r?   rG   rH   r   r   r   r   r3   S   s
    r3   c                   @   s   e Zd Zdd„ ZdS )Ú_SingleResponsec	           	      C   s  t |tjƒrt |tjƒstdƒ‚t|ƒ t |tjƒstdƒ‚|d ur,t |tjƒs,tdƒ‚|| _|| _|| _|| _	|| _
t |tƒsDtdƒ‚|tjurZ|d urQtdƒ‚|d urYtdƒ‚n$t |tjƒsdtdƒ‚t|ƒ}|tk rptdƒ‚|d ur~t |tjƒs~td	ƒ‚|| _|| _|| _d S )
Nr:   z%this_update must be a datetime objectz-next_update must be a datetime object or Nonez8cert_status must be an item from the OCSPCertStatus enumzBrevocation_time can only be provided if the certificate is revokedzDrevocation_reason can only be provided if the certificate is revokedz)revocation_time must be a datetime objectz7The revocation_time must be on or after 1950 January 1.zCrevocation_reason must be an item from the ReasonFlags enum or None)r!   r   r;   r<   r%   ÚdatetimeZ_certZ_issuerZ
_algorithmZ_this_updateZ_next_updater&   r'   r#   r	   r   ZReasonFlagsZ_cert_statusZ_revocation_timeZ_revocation_reason)	r7   r=   r>   r$   Úcert_statusÚthis_updateÚnext_updateÚrevocation_timeÚrevocation_reasonr   r   r   r9   y   s\   ÿ
ÿ
ÿ
ÿÿÿÿ
ÿÿ
z_SingleResponse.__init__N)r   r   r   r9   r   r   r   r   rI   x   s    rI   c                   @   sR   e Zd Zdddg fdd„Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Ze	dd„ ƒZ
dS )ÚOCSPResponseBuilderNc                 C   s   || _ || _|| _|| _d S r4   )Ú	_responseÚ_responder_idÚ_certsr6   )r7   ÚresponseÚresponder_idÚcertsr8   r   r   r   r9   ¿   s   
zOCSPResponseBuilder.__init__c	           
   	   C   s<   | j d ur	tdƒ‚t||||||||ƒ}	t|	| j| j| jƒS )Nz#Only one response per OCSPResponse.)rQ   r#   rI   rP   rR   rS   r6   )
r7   r=   r>   r$   rK   rL   rM   rN   rO   Z
singlerespr   r   r   Úadd_responseÇ   s$   
ø
üz OCSPResponseBuilder.add_responsec                 C   sP   | j d ur	tdƒ‚t|tjƒstdƒ‚t|tƒstdƒ‚t| j||f| j	| j
ƒS )Nz!responder_id can only be set oncez$responder_cert must be a Certificatez6encoding must be an element from OCSPResponderEncoding)rR   r#   r!   r   r;   r<   r   rP   rQ   rS   r6   )r7   ÚencodingZresponder_certr   r   r   rU   æ   s   

ÿüz OCSPResponseBuilder.responder_idc                 C   s\   | j d ur	tdƒ‚t|ƒ}t|ƒdkrtdƒ‚tdd„ |D ƒƒs$tdƒ‚t| j| j|| j	ƒS )Nz!certificates may only be set oncer   zcerts must not be an empty listc                 s   s    | ]	}t |tjƒV  qd S r4   )r!   r   r;   r   r   r   r   Ú	<genexpr>ý   s   € z3OCSPResponseBuilder.certificates.<locals>.<genexpr>z$certs must be a list of Certificates)
rS   r#   ÚlistÚlenÚallr<   rP   rQ   rR   r6   )r7   rV   r   r   r   Úcertificates÷   s   
üz OCSPResponseBuilder.certificatesc                 C   sL   t |tjƒs
tdƒ‚t |j||¡}t|| jƒ t| j	| j
| j| j|g ƒS r@   )r!   r   rA   r<   rB   rC   r
   r6   rP   rQ   rR   rS   rD   r   r   r   rG     s   
üz!OCSPResponseBuilder.add_extensionc                 C   sB   ddl m} | jd u rtdƒ‚| jd u rtdƒ‚| tj| ||¡S )Nr   r+   z&You must add a response before signingz*You must add a responder_id before signing)r-   r,   rQ   r#   rR   Úcreate_ocsp_responser   r   )r7   Zprivate_keyr$   r,   r   r   r   Úsign  s   


ÿzOCSPResponseBuilder.signc                 C   s@   ddl m} t|tƒstdƒ‚|tju rtdƒ‚| |d d d ¡S )Nr   r+   z7response_status must be an item from OCSPResponseStatusz$response_status cannot be SUCCESSFUL)r-   r,   r!   r   r<   r   r#   r^   )ÚclsÚresponse_statusr,   r   r   r   Úbuild_unsuccessful   s   
ÿ
z&OCSPResponseBuilder.build_unsuccessful)r   r   r   r9   rW   rU   r]   rG   r_   Úclassmethodrb   r   r   r   r   rP   ¾   s    
ÿrP   c                   @   s`   e Zd Zejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZej	d	d
„ ƒZ
ejdd„ ƒZdS )ÚOCSPRequestc                 C   ó   dS ©z3
        The hash of the issuer public key
        Nr   ©r7   r   r   r   Úissuer_key_hash0  ó    zOCSPRequest.issuer_key_hashc                 C   re   ©z-
        The hash of the issuer name
        Nr   rg   r   r   r   Úissuer_name_hash6  ri   zOCSPRequest.issuer_name_hashc                 C   re   ©zK
        The hash algorithm used in the issuer name and key hashes
        Nr   rg   r   r   r   Úhash_algorithm<  ri   zOCSPRequest.hash_algorithmc                 C   re   ©zM
        The serial number of the cert whose status is being checked
        Nr   rg   r   r   r   Úserial_numberB  ri   zOCSPRequest.serial_numberc                 C   re   )z/
        Serializes the request to DER
        Nr   )r7   rX   r   r   r   Úpublic_bytesH  ri   zOCSPRequest.public_bytesc                 C   re   )zP
        The list of request extensions. Not single request extensions.
        Nr   rg   r   r   r   r8   N  ri   zOCSPRequest.extensionsN)r   r   r   ÚabcÚabstractpropertyrh   rk   rm   ro   Úabstractmethodrp   r8   r   r   r   r   rd   .  s    




rd   c                   @   s$  e Zd Zejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejd	d
„ ƒZ	ejdd„ ƒZ
ejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd „ ƒZejd!d"„ ƒZejd#d$„ ƒZejd%d&„ ƒZejd'd(„ ƒZd)S )*ÚOCSPResponsec                 C   re   )zm
        The status of the response. This is a value from the OCSPResponseStatus
        enumeration
        Nr   rg   r   r   r   ra   W  ri   zOCSPResponse.response_statusc                 C   re   )zA
        The ObjectIdentifier of the signature algorithm
        Nr   rg   r   r   r   Úsignature_algorithm_oid^  ri   z$OCSPResponse.signature_algorithm_oidc                 C   re   )zX
        Returns a HashAlgorithm corresponding to the type of the digest signed
        Nr   rg   r   r   r   Úsignature_hash_algorithmd  ri   z%OCSPResponse.signature_hash_algorithmc                 C   re   )z%
        The signature bytes
        Nr   rg   r   r   r   Ú	signaturej  ri   zOCSPResponse.signaturec                 C   re   )z+
        The tbsResponseData bytes
        Nr   rg   r   r   r   Útbs_response_bytesp  ri   zOCSPResponse.tbs_response_bytesc                 C   re   )z»
        A list of certificates used to help build a chain to verify the OCSP
        response. This situation occurs when the OCSP responder uses a delegate
        certificate.
        Nr   rg   r   r   r   r]   v  ri   zOCSPResponse.certificatesc                 C   re   )z2
        The responder's key hash or None
        Nr   rg   r   r   r   Úresponder_key_hash~  ri   zOCSPResponse.responder_key_hashc                 C   re   )z.
        The responder's Name or None
        Nr   rg   r   r   r   Úresponder_name„  ri   zOCSPResponse.responder_namec                 C   re   )z4
        The time the response was produced
        Nr   rg   r   r   r   Úproduced_atŠ  ri   zOCSPResponse.produced_atc                 C   re   )zY
        The status of the certificate (an element from the OCSPCertStatus enum)
        Nr   rg   r   r   r   Úcertificate_status  ri   zOCSPResponse.certificate_statusc                 C   re   )z^
        The date of when the certificate was revoked or None if not
        revoked.
        Nr   rg   r   r   r   rN   –  ri   zOCSPResponse.revocation_timec                 C   re   )zi
        The reason the certificate was revoked or None if not specified or
        not revoked.
        Nr   rg   r   r   r   rO     ri   zOCSPResponse.revocation_reasonc                 C   re   )z
        The most recent time at which the status being indicated is known by
        the responder to have been correct
        Nr   rg   r   r   r   rL   ¤  ri   zOCSPResponse.this_updatec                 C   re   )zC
        The time when newer information will be available
        Nr   rg   r   r   r   rM   «  ri   zOCSPResponse.next_updatec                 C   re   rf   r   rg   r   r   r   rh   ±  ri   zOCSPResponse.issuer_key_hashc                 C   re   rj   r   rg   r   r   r   rk   ·  ri   zOCSPResponse.issuer_name_hashc                 C   re   rl   r   rg   r   r   r   rm   ½  ri   zOCSPResponse.hash_algorithmc                 C   re   rn   r   rg   r   r   r   ro   Ã  ri   zOCSPResponse.serial_numberc                 C   re   )zR
        The list of response extensions. Not single response extensions.
        Nr   rg   r   r   r   r8   É  ri   zOCSPResponse.extensionsc                 C   re   )zR
        The list of single response extensions. Not response extensions.
        Nr   rg   r   r   r   Úsingle_extensionsÏ  ri   zOCSPResponse.single_extensionsN)r   r   r   rq   rr   ra   ru   rv   rw   rx   r]   ry   rz   r{   r|   rN   rO   rL   rM   rh   rk   rm   ro   r8   r}   r   r   r   r   rt   U  sR    


















rt   )(Ú
__future__r   r   r   rq   rJ   Úenumr   ZsixÚcryptographyr   Zcryptography.hazmat.primitivesr   Zcryptography.x509.baser   r	   r
   ZSHA1ZSHA224ZSHA256ZSHA384ZSHA512Z_OIDS_TO_HASHr   r   Z_RESPONSE_STATUS_TO_ENUMr"   r%   r&   Z_CERT_STATUS_TO_ENUMr.   r2   Úobjectr3   rI   rP   Zadd_metaclassÚABCMetard   rt   r   r   r   r   Ú<module>   sF   û		û	%F
p
&