U
    *”}f  ã                   @   s˜   d dl mZmZmZ d dlmZ d dl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d	„ Zd
d„ ZG dd„ deƒZG dd„ deƒZdS )é    )Úabsolute_importÚdivisionÚprint_function)ÚEnum)Úx509)Ú_get_backend)ÚhashesÚserialization)ÚecÚrsa)Ú_check_byteslikec                 C   s   t d ƒ}| | ¡S ©N)r   Úload_pem_pkcs7_certificates©ÚdataÚbackend© r   úV/tmp/pip-unpacked-wheel-x36vw73o/cryptography/hazmat/primitives/serialization/pkcs7.pyr      s    r   c                 C   s   t d ƒ}| | ¡S r   )r   Úload_der_pkcs7_certificatesr   r   r   r   r      s    r   c                   @   s>   e Zd Zdg g fdd„Zdd„ Zdd„ Zdd	„ Zdd
d„ZdS )ÚPKCS7SignatureBuilderNc                 C   s   || _ || _|| _d S r   )Ú_dataÚ_signersÚ_additional_certs)Úselfr   ZsignersZadditional_certsr   r   r   Ú__init__   s    zPKCS7SignatureBuilder.__init__c                 C   s(   t d|ƒ | jd k	rtdƒ‚t|| jƒS )Nr   zdata may only be set once)r   r   Ú
ValueErrorr   r   )r   r   r   r   r   Úset_data    s    

zPKCS7SignatureBuilder.set_datac                 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ƒsTt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   ZRSAPrivateKeyr
   ZEllipticCurvePrivateKeyr   r   r   )r   ÚcertificateZprivate_keyZhash_algorithmr   r   r   Ú
add_signer'   s.    ûþ
ÿ 
ÿþz PKCS7SignatureBuilder.add_signerc                 C   s,   t |tjƒstdƒ‚t| j| j| j|g ƒS )Nr   )r   r   r%   r$   r   r   r   r   )r   r&   r   r   r   Úadd_certificateC   s      
ÿz%PKCS7SignatureBuilder.add_certificatec                 C   sà   t | jƒdkrtdƒ‚| jd kr(tdƒ‚t|ƒ}tdd„ |D ƒƒsJtdƒ‚|tjjtjj	tjj
fkrltdƒ‚tj|krˆtj|krˆtdƒ‚tj|kr®|tjj	tjjfkr®td	ƒ‚tj|krÊtj|krÊt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   )r   ÚPKCS7Options)Ú.0Úxr   r   r   Ú	<genexpr>Q   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	   ÚEncodingZPEMZDERZSMIMEr)   ÚTextÚDetachedSignatureÚNoAttributesÚNoCapabilitiesr   Z
pkcs7_sign)r   ÚencodingÚoptionsr   r   r   r   ÚsignK   sL    
ýÿÿþÿþÿÿþÿzPKCS7SignatureBuilder.sign)N)Ú__name__Ú
__module__Ú__qualname__r   r   r'   r(   r7   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 )r)   z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)	r8   r9   r:   r1   ÚBinaryr2   r4   r3   ZNoCertsr   r   r   r   r)   ~   s   r)   N)Ú
__future__r   r   r   Úenumr   Zcryptographyr   Zcryptography.hazmat.backendsr   Zcryptography.hazmat.primitivesr   r	   Z)cryptography.hazmat.primitives.asymmetricr
   r   Zcryptography.utilsr   r   r   Úobjectr   r)   r   r   r   r   Ú<module>   s   d