3
l�_ � @ s� d dl mZmZmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z d dl
mZmZ d dlmZ dd � Zd
d � ZG d d
� d
e�ZG dd� de�ZdS )� )�absolute_import�division�print_function)�Enum)�x509)� _get_backend)�hashes�
serialization)�ec�rsa)�_check_byteslikec C s t d �}|j| �S )N)r �load_pem_pkcs7_certificates)�data�backend� r � /usr/lib64/python3.6/pkcs7.pyr
s r
c C s t d �}|j| �S )N)r �load_der_pkcs7_certificates)r r r r r r s r c @ s> e Zd Zdg g fdd�Zdd� Zdd� Zdd � Zd d
d �ZdS )
�PKCS7SignatureBuilderNc C s || _ || _|| _d S )N)�_data�_signers�_additional_certs)�selfr ZsignersZadditional_certsr r r �__init__ s z PKCS7SignatureBuilder.__init__c C s( t d|� | jd k r td��t|| j�S )Nr zdata may only be set once)r r �
ValueErrorr r )r r r r r �set_data s
z PKCS7SignatureBuilder.set_datac C sn t |tjtjtjtjtjf�s&td��t |tj �s:td��t |t
j t j
f�sTtd��t| j| j|||fg �S )NzLhash_algorithm must be one of hashes.SHA1, SHA224, SHA256, SHA384, or SHA512z&certificate must be a x509.Certificatez.Only RSA & EC keys are supported at this time.)�
isinstancer ZSHA1ZSHA224ZSHA256ZSHA384ZSHA512� TypeErrorr � Certificater Z
RSAPrivateKeyr
ZEllipticCurvePrivateKeyr r r )r � certificateZ private_keyZhash_algorithmr r r �
add_signer' s"
z PKCS7SignatureBuilder.add_signerc C s, t |tj�std��t| j| j| j|g �S )Nz&certificate must be a x509.Certificate)r r r r r r r r )r r r r r �add_certificateC s z%PKCS7SignatureBuilder.add_certificatec C s� t | j�dkrtd��| jd kr(td��t|�}tdd� |D ��sJtd��|tjjtjj tjj
fkrltd��t j |kr�t j
|kr�td��t j |kr�|tjj tjjfkr�td ��t j|kr�t j|kr�td
��t|�}|j| ||�S ) Nr z Must have at least one signerzYou must add data to signc s s | ]}t |t�V qd S )N)r � PKCS7Options)�.0�xr r r � Q s z-PKCS7SignatureBuilder.sign..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 ZEncodingZPEMZDERZSMIMEr! �Text�DetachedSignature� NoAttributes�NoCapabilitiesr Z
pkcs7_sign)r �encodingZoptionsr r r r �signK s8
zPKCS7SignatureBuilder.sign)N)�__name__�
__module__� __qualname__r r r r r- r r r r r s
r c @ s$ e Zd ZdZdZdZdZdZdZdS )r! zAdd 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) r. r/ r0 r( ZBinaryr) r+ r* ZNoCertsr r r r r! ~ s r! N)Z
__future__r r r �enumr Z cryptographyr Z cryptography.hazmat.backendsr Z cryptography.hazmat.primitivesr r Z)cryptography.hazmat.primitives.asymmetricr
r Zcryptography.utilsr r
r �objectr r! r r r r � s d