File: //snap/core18/current/usr/lib/python3/dist-packages/cryptography/__pycache__/fernet.cpython-36.pyc
3
�d Z� � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl m
Z
d dlmZmZ d dlmZmZmZ d dlmZ G dd � d e�Zd
ZG d d � d e�ZG d
d� de�Z dS )� )�absolute_import�division�print_functionN)�InvalidSignature)�default_backend)�hashes�padding)�Cipher�
algorithms�modes)�HMACc @ s e Zd ZdS )� InvalidTokenN)�__name__�
__module__� __qualname__� r r �5/usr/lib/python3/dist-packages/cryptography/fernet.pyr
s r
�< c @ s< e Zd Zd dd�Zedd� �Zdd� Zdd � Zd
d
d �ZdS )�FernetNc C sR |d krt � }tj|�}t|�dkr,td��|d d� | _|dd � | _|| _d S )N� z4Fernet key must be 32 url-safe base64-encoded bytes.� )r �base64�urlsafe_b64decode�len�
ValueError� _signing_key�_encryption_key�_backend)�self�key�backendr r r �__init__ s
zFernet.__init__c C s t jtjd��S )Nr )r �urlsafe_b64encode�os�urandom)�clsr r r � generate_key, s zFernet.generate_keyc C s$ t tj� �}tjd�}| j|||�S )Nr )�int�timer# r$ �_encrypt_from_parts)r �data� current_time�ivr r r �encrypt0 s
zFernet.encryptc C s� t |t�std��tjtjj�j� }|j |�|j
� }t tj| j �t
j|�| j�j� }|j |�|j
� }dtjd|� | | }t| jtj� | jd�} | j |� | j
� }
tj||
�S )Nzdata must be bytes.� �z>Q)r )�
isinstance�bytes� TypeErrorr �PKCS7r
�AES�
block_size�padder�update�finalizer r r �CBCr � encryptor�structZpackr r r �SHA256r r" ) r r* r+ r, r5 Z padded_datar9 �
ciphertextZ basic_parts�hZhmacr r r r) 5 s
zFernet._encrypt_from_partsc
+ C s� t |t�std��ttj� �}ytj|�}W n ttjfk
rJ t �Y nX | sbt
j |d�dkrft �yt j
d|dd� �\}W n t jk
r� t �Y nX |d k r�|| |k r�t �|t |k r�t �t| jtj� | jd�}|j|d d
� � y|j|d d � � W n tk
�r t �Y nX |dd � }|d d � }ttj| j�t j |�| j�j � } | j|�}
y|
| j � 7 }
W n t k
�r� t �Y nX t!j"tjj#�j$� } | j|
�} y| | j � 7 } W n t k
�r� t �Y nX | S )
Nztoken must be bytes.r � z>Q� � )r r � i��i��i��)%r/ r0 r1 r' r( r r �binascii�Errorr
�sixZ
indexbytesr: Zunpack�error�_MAX_CLOCK_SKEWr r r r; r r6 Zverifyr r r
r3 r r r8 � decryptorr7 r r r2 r4 �unpadder)
r �token�ttlr+ r* Z timestampr= r, r< rG Zplaintext_paddedrH Zunpaddedr r r �decryptI sP
zFernet.decrypt)N)N) r r r r! � classmethodr& r- r) rK r r r r r s
r c @ s& e Zd Zdd� Zdd� Zddd�ZdS ) � MultiFernetc C s t |�}|std��|| _d S )Nz1MultiFernet requires at least one Fernet instance)�listr �_fernets)r Zfernetsr r r r! ~ s
zMultiFernet.__init__c C s | j d j|�S )Nr )rO r- )r �msgr r r r- � s zMultiFernet.encryptNc C s: x0| j D ]&}y |j||�S tk
r, Y qX qW t�d S )N)rO rK r
)r rP rJ �fr r r rK � s
zMultiFernet.decrypt)N)r r r r! r- rK r r r r rM } s rM ) Z
__future__r r r r rB r# r: r( rD Zcryptography.exceptionsr Z cryptography.hazmat.backendsr Z cryptography.hazmat.primitivesr r Z&cryptography.hazmat.primitives.ciphersr r
r Z#cryptography.hazmat.primitives.hmacr � Exceptionr
rF �objectr rM r r r r �<module> s `