o
    ÖèFh@  ã                   @  s  d dl m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
mZ d dlmZ d dlmZmZ d dlmZmZmZ d dlmZ ejZejZej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dd„Z'G dd„ dej(j)ƒZ*dS )é    )ÚannotationsN)ÚutilsÚx509)Úpkcs7)ÚhashesÚserialization)ÚecÚpaddingÚrsa)Ú_check_byteslikec                   @  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ÚNoCerts© r   r   úk/var/www/html/loop/nvenv/lib/python3.10/site-packages/cryptography/hazmat/primitives/serialization/pkcs7.pyr   &   s    r   c                   @  sP   e Zd Zdg g fd#dd	„Zd$dd„Zddœd%dd„Zd&dd„Z	d'd(d!d"„ZdS ))ÚPKCS7SignatureBuilderNÚdataúbytes | NoneÚsignersújlist[tuple[x509.Certificate, PKCS7PrivateKeyTypes, PKCS7HashTypes, padding.PSS | padding.PKCS1v15 | None]]Úadditional_certsúlist[x509.Certificate]c                 C  s   || _ || _|| _d S ©N)Ú_dataÚ_signersÚ_additional_certs)Úselfr   r   r   r   r   r   Ú__init__0   s   
zPKCS7SignatureBuilder.__init__ÚbytesÚreturnc                 C  s(   t d|ƒ | jd urtdƒ‚t|| jƒS )Nr   zdata may only be set once)r   r    Ú
ValueErrorr   r!   )r#   r   r   r   r   Úset_dataA   s   

zPKCS7SignatureBuilder.set_data)Úrsa_paddingÚcertificateúx509.CertificateÚprivate_keyÚPKCS7PrivateKeyTypesÚhash_algorithmÚPKCS7HashTypesr)   ú%padding.PSS | padding.PKCS1v15 | Nonec                C  s¤   t |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ƒ‚|d urCt |tjtjfƒs9tdƒ‚t |t	j
ƒsCtdƒ‚t| jg | j¢||||f‘ƒS )NzFhash_algorithm must be one of hashes.SHA224, SHA256, SHA384, or SHA512ú&certificate must be a x509.Certificatez.Only RSA & EC keys are supported at this time.zPadding must be PSS or PKCS1v15z&Padding is only supported for RSA keys)Ú
isinstancer   ÚSHA224ÚSHA256ÚSHA384ÚSHA512Ú	TypeErrorr   ÚCertificater
   ÚRSAPrivateKeyr   ÚEllipticCurvePrivateKeyr	   ÚPSSÚPKCS1v15r   r    r!   )r#   r*   r,   r.   r)   r   r   r   Ú
add_signerH   s<   üþ	ÿÿÿ
þþz PKCS7SignatureBuilder.add_signerc                 C  s.   t |tjƒs
tdƒ‚t| j| jg | j¢|‘ƒS )Nr1   )r2   r   r8   r7   r   r    r!   r"   )r#   r*   r   r   r   Úadd_certificates   s
   ÿz%PKCS7SignatureBuilder.add_certificateÚencodingúserialization.EncodingÚoptionsútyping.Iterable[PKCS7Options]Úbackendú
typing.Anyc                 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   )r2   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   Ú
rust_pkcs7Úsign_and_serialize)r#   r?   rA   rC   r   r   r   Úsign}   sB   
ýÿ

ÿþÿ

ÿzPKCS7SignatureBuilder.sign)r   r   r   r   r   r   )r   r%   r&   r   )
r*   r+   r,   r-   r.   r/   r)   r0   r&   r   )r*   r+   r&   r   r   )r?   r@   rA   rB   rC   rD   r&   r%   )r   r   r   r$   r(   r=   r>   rQ   r   r   r   r   r   /   s    õ
ú
+ür   r   r%   Ú	signatureÚmicalgÚstrÚ	text_modeÚboolr&   c           	      C  sì   t j ¡ }| dd¡ |jddd|d� d|_tƒ }| | ¡ |r'| 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jdd�d�}| |¡ | ¡ S )NzMIME-Versionz1.0zContent-Typezmultipart/signedzapplication/x-pkcs7-signature)ÚprotocolrS   z!This is an S/MIME signed message
z
text/plainz	smime.p7s)ÚnamezContent-Transfer-EncodingÚbase64zContent-DispositionÚ
attachment)ÚfilenameéA   )Ú
maxlinelenr   Fz
)Úlinesep)ÚmaxheaderlenÚmangle_from_Úpolicy)ÚemailÚmessageÚMessageÚ
add_headerÚpreambleÚOpenSSLMimePartÚset_payloadÚattachÚMIMEPartÚ
base64mimeÚbody_encodeÚioÚBytesIOÚ	generatorÚBytesGeneratorra   ÚcloneÚflattenÚgetvalue)	r   rR   rS   rU   ÚmÚmsg_partÚsig_partÚfpÚgr   r   r   Ú_smime_encode´   sH   
ü


ÿÿÿ
ü
ry   c                   @  s   e Zd Zddd„ZdS )rg   r&   ÚNonec                 C  s   t |  ¡ ƒr| | ¡ d S d S r   )rI   Ú	raw_itemsÚ_write_headers)r#   ro   r   r   r   r|   ç   s   ÿzOpenSSLMimePart._write_headersN)r&   rz   )r   r   r   r|   r   r   r   r   rg   ä   s    rg   )
r   r%   rR   r%   rS   rT   rU   rV   r&   r%   )+Ú
__future__r   Úemail.base64mimerb   Úemail.generatorÚemail.messageÚemail.policyrm   ÚtypingÚcryptographyr   r   Ú"cryptography.hazmat.bindings._rustr   rO   Úcryptography.hazmat.primitivesr   r   Ú)cryptography.hazmat.primitives.asymmetricr   r	   r
   Úcryptography.utilsr   Úload_pem_pkcs7_certificatesÚload_der_pkcs7_certificatesÚserialize_certificatesÚUnionr3   r4   r5   r6   r/   r9   r:   r-   ÚEnumr   r   ry   rc   rj   rg   r   r   r   r   Ú<module>   s<   ýÿ
ÿ	 
0