o
    
c                     @   s.  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 d dlmZmZ d dlmZ dedejej fd	d
Zdedejej fddZdejej dejdefddZejejejejejf Z ejej!ej"f Z#G dd dej$Z%G dd dZ&dedede'defddZ(dS )    N)utilsx509)pkcs7)hashesserialization)ecrsa)_check_byteslikedatareturnc                 C      ddl m} || S Nr   )backend),cryptography.hazmat.backends.openssl.backendr   load_pem_pkcs7_certificatesr
   r    r   _/opt/certbot/lib/python3.10/site-packages/cryptography/hazmat/primitives/serialization/pkcs7.pyr         
r   c                 C   r   r   )r   r   load_der_pkcs7_certificatesr   r   r   r   r      r   r   certsencodingc                 C   s   t | |S N)
rust_pkcs7serialize_certificates)r   r   r   r   r   r      s   r   c                   @   s$   e Zd ZdZdZdZdZdZdZdS )PKCS7OptionszAdd text/plain MIME typez5Don't translate input data into canonical MIME formatz'Don't embed data in the PKCS7 structurezDon't embed SMIME capabilitiesz#Don't embed authenticatedAttributeszDon't embed signer certificateN)	__name__
__module____qualname__TextBinaryDetachedSignatureNoCapabilitiesNoAttributesNoCertsr   r   r   r   r   1   s    r   c                	   @   s   e Zd Zdg g fdeje dejejej	e
ef  dejej	 fddZdedd fdd	Zd
ej	de
dedd fddZd
ej	dd fddZ	ddejdeje dejdefddZdS )PKCS7SignatureBuilderNr
   signersadditional_certsc                 C   s   || _ || _|| _d S r   )_data_signers_additional_certs)selfr
   r&   r'   r   r   r   __init__;   s   
zPKCS7SignatureBuilder.__init__r   c                 C   s,   t d| | jd urtdtt|| jS )Nr
   zdata may only be set once)r	   r(   
ValueErrorr%   bytesr)   )r+   r
   r   r   r   set_dataK   s   

zPKCS7SignatureBuilder.set_datacertificateprivate_keyhash_algorithmc                 C   sn   t |tjtjtjtjtjfstdt |tj	stdt |t
jtjfs*tdt| j| j|||fg S )NzLhash_algorithm must be one of hashes.SHA1, SHA224, SHA256, SHA384, or SHA512&certificate must be a x509.Certificatez.Only RSA & EC keys are supported at this time.)
isinstancer   SHA1SHA224SHA256SHA384SHA512	TypeErrorr   Certificater   RSAPrivateKeyr   EllipticCurvePrivateKeyr%   r(   r)   )r+   r0   r1   r2   r   r   r   
add_signerR   s,   
z PKCS7SignatureBuilder.add_signerc                 C   s,   t |tjs
tdt| j| j| j|g S )Nr3   )r4   r   r;   r:   r%   r(   r)   r*   )r+   r0   r   r   r   add_certificates   s
   z%PKCS7SignatureBuilder.add_certificater   optionsr   c                 C   s   t | jdkrtd| jd u rtdt|}tdd |D s%td|tjjtjj	tjj
fvr6tdtj|v rDtj|vrDtdtj|v rW|tjj	tjjfv rWtd	tj|v retj|v retd
t| ||S )Nr   zMust have at least one signerzYou must add data to signc                 s   s    | ]}t |tV  qd S r   )r4   r   ).0xr   r   r   	<genexpr>   s    z-PKCS7SignatureBuilder.sign.<locals>.<genexpr>z*options must be from the PKCS7Options enumz1Must be PEM, DER, or SMIME from the Encoding enumzAWhen passing the Text option you must also pass DetachedSignaturez9The Text option is only available for SMIME serializationzFNoAttributes is a superset of NoCapabilities. Do not pass both values.)lenr)   r-   r(   listallr   EncodingPEMDERSMIMEr   r   r!   r#   r"   r   sign_and_serialize)r+   r   r@   r   r   r   r   sign}   sJ   
zPKCS7SignatureBuilder.signr   )r   r   r   typingOptionalr.   ListTupler   r;   _ALLOWED_PRIVATE_KEY_TYPES_ALLOWED_PKCS7_HASH_TYPESr,   r/   r>   r?   r   rG   Iterabler   AnyrL   r   r   r   r   r%   :   sV    



!
r%   	signaturemicalgc                 C   s   t j }|dd |jddd|d d|_t j }||  |dd || t j }|jddd	d
 |dd |jddd	d |t jj	|dd |d= || t
 }t jj|dd|jd}|| | S )NzMIME-Versionz1.0zContent-Typezmultipart/signedzapplication/x-pkcs7-signature)protocolrV   z!This is an S/MIME signed message
z
text/plainz	smime.p7s)namezContent-Transfer-Encodingbase64zContent-Disposition
attachment)filenameA   )
maxlinelenr   F)maxheaderlenmangle_from_policy)emailmessageMessage
add_headerpreambleMIMEPartset_payloadattach
base64mimebody_encodeioBytesIO	generatorBytesGeneratorr`   flattengetvalue)r
   rU   rV   mmsg_partsig_partfpgr   r   r   _smime_encode   s@   







rv   ))email.base64mimera   email.generatoremail.messagerk   rM   cryptographyr   r   "cryptography.hazmat.bindings._rustr   r   cryptography.hazmat.primitivesr   r   )cryptography.hazmat.primitives.asymmetricr   r   cryptography.utilsr	   r.   rO   r;   r   r   rG   r   Unionr6   r7   r8   r9   rR   r<   r=   rQ   Enumr   r%   strrv   r   r   r   r   <module>   s@   


	z