3
l�_4' � @ s� d dl mZmZmZ d dlmZ d dlmZ d dlm Z m
Z
m Z d dl m
Z
d dlmZmZmZ dd� Zd d
� Zeje�G d d � d e��Zeje�G d
d� de��Zejej�G dd� de��Zejej�G dd� de��Zejej �G dd� de��Z dS )� )�absolute_import�division�print_function)�utils)�InvalidSignature)� _calculate_digest_and_algorithm�_check_not_prehashed� _warn_sign_verify_deprecated)�hashes)�AsymmetricSignatureContext� AsymmetricVerificationContext�dsac C sx | j j|j�}| jjd|�}| jjd�}| j jd|t|�|||j�}| j|dk� | j|d � | jj|�d |d � S )Nzunsigned char[]zunsigned int *r � ) �_libZDSA_size�
_dsa_cdata�_ffi�newZDSA_sign�len�openssl_assert�buffer)�backend� private_key�dataZ sig_buf_lenZsig_bufZbuflen�res� r �/usr/lib64/python3.6/dsa.py�
_dsa_sig_sign s r c C s8 | j jd|t|�|t|�|j�}|dkr4| j� t�d S )Nr r )r Z
DSA_verifyr r Z_consume_errorsr )r �
public_key� signaturer r r r r �_dsa_sig_verify& s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_DSAVerificationContextc C s. || _ || _|| _|| _tj| j| j �| _d S )N)�_backend� _public_key�
_signature�
_algorithmr
�Hash� _hash_ctx)�selfr r r � algorithmr r r �__init__4 s
z _DSAVerificationContext.__init__c C s | j j|� d S )N)r& �update)r' r r r r r* < s z _DSAVerificationContext.updatec C s" | j j� }t| j| j| j|� d S )N)r& �finalizer r! r" r# )r' Zdata_to_verifyr r r �verify? s
z _DSAVerificationContext.verifyN)�__name__�
__module__� __qualname__r) r* r, r r r r r 2 s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_DSASignatureContextc C s( || _ || _|| _tj| j| j �| _d S )N)r! � _private_keyr$ r
r% r& )r' r r r( r r r r) I s z _DSASignatureContext.__init__c C s | j j|� d S )N)r& r* )r' r r r r r* O s z_DSASignatureContext.updatec C s | j j� }t| j| j|�S )N)r& r+ r r! r1 )r' Z data_to_signr r r r+ R s
z _DSASignatureContext.finalizeN)r- r. r/ r) r* r+ r r r r r0 G s r0 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_DSAParametersc C s || _ || _d S )N)r! r )r' r � dsa_cdatar r r r) Y s z_DSAParameters.__init__c C s� | j jjd�}| j jjd�}| j jjd�}| j jj| j|||� | j j|d | j jjk� | j j|d | j jjk� | j j|d | j jjk� tj | j j
|d �| j j
|d �| j j
|d �d�S )Nz BIGNUM **r )�p�q�g) r! r r r � DSA_get0_pqgr r �NULLr
�DSAParameterNumbers�
_bn_to_int)r' r4 r5 r6 r r r �parameter_numbers] s z _DSAParameters.parameter_numbersc C s | j j| �S )N)r! Zgenerate_dsa_private_key)r' r r r �generate_private_keyk s z#_DSAParameters.generate_private_keyN)r- r. r/ r) r; r< r r r r r2 W s r2 c @ sN e Zd Zdd� Zejd�Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dS )�_DSAPrivateKeyc C sp || _ || _|| _| j jjd�}| j jj||| j jj| j jj� | j j|d |jjk� | j jj |d �| _
d S )Nz BIGNUM **r ) r! r � _evp_pkeyr r r r7 r8 r � BN_num_bits� _key_size)r' r r3 �evp_pkeyr4 r r r r) q s z_DSAPrivateKey.__init__r@ c C s t � t|� t| j| |�S )N)r r r0 r! )r' �signature_algorithmr r r �signer s z_DSAPrivateKey.signerc C sR | j jjd�}| j jjd�}| j jjd�}| j jjd�}| j jjd�}| j jj| j|||� | j j|d | j jjk� | j j|d | j jjk� | j j|d | j jjk� | j jj| j||� | j j|d | j jjk� | j j|d | j jjk� t j
t j t j | j j
|d �| j j
|d �| j j
|d �d�| j j
|d �d�| j j
|d �d�S )Nz BIGNUM **r )r4 r5 r6 )r; �y)�public_numbers�x)r! r r r r7 r r r8 � DSA_get0_keyr
ZDSAPrivateNumbers�DSAPublicNumbersr9 r: )r' r4 r5 r6 �pub_keyZpriv_keyr r r �private_numbers� s( z _DSAPrivateKey.private_numbersc C s� | j jj| j�}| j j|| j jjk� | j jj|| j jj�}| j jj d�}| j jj
| j|| j jj� | j j|d | j jjk� | j jj |d �}| j jj ||| j jj�}| j j|dk� | j j
|�}t| j ||�S )Nz BIGNUM **r r )r! r �
DSAparams_dupr r r r8 �gc�DSA_freer rG ZBN_dupZ DSA_set0_keyZ_dsa_cdata_to_evp_pkey�
_DSAPublicKey)r' r3 rI Z pub_key_dupr rA r r r r � s z_DSAPrivateKey.public_keyc C sH | j jj| j�}| j j|| j jjk� | j jj|| j jj�}t | j |�S )N)
r! r rK r r r r8 rL rM r2 )r' r3 r r r �
parameters� s
z_DSAPrivateKey.parametersc C s | j j|||| | j| j�S )N)r! Z_private_key_bytesr> r )r' �encoding�formatZencryption_algorithmr r r �
private_bytes� s z _DSAPrivateKey.private_bytesc C s t | j||�\}}t| j| |�S )N)r r! r )r' r r( r r r �sign� s z_DSAPrivateKey.signN)
r- r. r/ r) r �read_only_property�key_sizerC rJ r rO rR rS r r r r r= o s
r= c @ sF e Zd Zdd� Zejd�Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dS )rN c C sp || _ || _|| _| j jjd�}| j jj||| j jj| j jj� | j j|d |jjk� | j jj |d �| _
d S )Nz BIGNUM **r ) r! r r> r r r r7 r8 r r? r@ )r' r r3 rA r4 r r r r) � s z_DSAPublicKey.__init__r@ c C s* t � tjd|� t|� t| j| ||�S )Nr )r r � _check_bytesr r r! )r' r rB r r r �verifier� s
z_DSAPublicKey.verifierc C s | j jjd�}| j jjd�}| j jjd�}| j jjd�}| j jj| j|||� | j j|d | j jjk� | j j|d | j jjk� | j j|d | j jjk� | j jj| j|| j jj� | j j|d | j jjk� t j
t j | j j |d �| j j |d �| j j |d �d�| j j |d �d�S )Nz BIGNUM **r )r4 r5 r6 )r; rD )
r! r r r r7 r r r8 rG r
rH r9 r: )r' r4 r5 r6 rI r r r rE � s" z _DSAPublicKey.public_numbersc C s2 | j jj| j�}| j jj|| j jj�}t| j |�S )N)r! r rK r r rL rM r2 )r' r3 r r r rO � s z_DSAPublicKey.parametersc C s | j j||| | jd �S )N)r! Z_public_key_bytesr> )r' rP rQ r r r � public_bytes� s z_DSAPublicKey.public_bytesc C s" t | j||�\}}t| j| ||�S )N)r r! r )r' r r r( r r r r, s z_DSAPublicKey.verifyN) r- r. r/ r) r rT rU rW rE rO rX r, r r r r rN � s
rN N) Z
__future__r r r Z cryptographyr Zcryptography.exceptionsr Z*cryptography.hazmat.backends.openssl.utilsr r r Z cryptography.hazmat.primitivesr
Z)cryptography.hazmat.primitives.asymmetricr r r
r r Zregister_interface�objectr r0 ZDSAParametersWithNumbersr2 Z DSAPrivateKeyWithSerializationr= Z DSAPublicKeyWithSerializationrN r r r r � s" Y