Google
Edit File: fernet.cpython-39.pyc
a �a � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl m Z m Z d dl mZmZmZ d dlmZ G dd� de�Zd ZG d d � d e�ZG d d � d e�ZdS )� N)�utils)�InvalidSignature)�hashes�padding)�Cipher� algorithms�modes)�HMACc @ s e Zd ZdS )� InvalidTokenN)�__name__� __module__� __qualname__� r r �9/usr/lib64/python3.9/site-packages/cryptography/fernet.pyr s r �< c @ s e Zd Zd!ejeef ejd�dd�Ze ed�dd��Z eed�d d �Z ee ed �d d �Z ee eed�dd�Zd"eeje ed�dd�Zee e ed�dd�Zee d�dd�Zeeeje ef d�dd��Zedd�d d �Zee ejeje e f ed �d d �ZdS )#�FernetN)�key�backendc C s> t �|�}t|�dkr td��|d d� | _|dd � | _d S )N� z4Fernet key must be 32 url-safe base64-encoded bytes.� )�base64�urlsafe_b64decode�len� ValueError� _signing_key�_encryption_key)�selfr r r r r �__init__ s �zFernet.__init__)�returnc C s t �t�d��S )Nr )r �urlsafe_b64encode�os�urandom)�clsr r r � generate_key* s zFernet.generate_key)�datar c C s | � |tt�� ��S �N��encrypt_at_time�int�time)r r$ r r r �encrypt. s zFernet.encrypt)r$ � current_timer c C s t �d�}| �|||�S )Nr )r r! �_encrypt_from_parts)r r$ r+ �ivr r r r' 1 s zFernet.encrypt_at_time)r$ r+ r- r c C s� t �d|� t�tjj��� }|�|�|� � }t t�| j �t � |���� }|�|�|� � }dt�d|� | | }t| jt�� �} | �|� | � � } t�|| �S )Nr$ � ��>Q)r � _check_bytesr �PKCS7r �AES� block_size�padder�update�finalizer r r �CBC� encryptor�struct�packr r r �SHA256r r ) r r$ r+ r- r4 Z padded_datar8 � ciphertextZ basic_parts�hZhmacr r r r, 5 s �� zFernet._encrypt_from_parts)�token�ttlr c C s: t �|�\}}|d u r d }n|tt�� �f}| �|||�S r% )r �_get_unverified_token_datar( r) � _decrypt_data)r r> r? � timestampr$ � time_infor r r �decryptK s zFernet.decrypt)r> r? r+ r c C s0 |d u rt d��t�|�\}}| �||||f�S )Nz6decrypt_at_time() can only be used with a non-None ttl)r r r@ rA )r r> r? r+ rB r$ r r r �decrypt_at_timeS s �zFernet.decrypt_at_time)r> r c C s t �|�\}}| �|� |S r% )r r@ �_verify_signature)r r> rB r$ r r r �extract_timestamp] s zFernet.extract_timestampc C s� t �d| � zt�| �}W n ttjfy6 t�Y n0 |rH|d dkrLt�zt� d|dd� �\}W n tj y~ t�Y n0 ||fS )Nr> r � r/ � � ) r r0 r r � TypeError�binascii�Errorr r9 �unpack�error)r> r$ rB r r r r@ c s z!Fernet._get_unverified_token_datac C sT t | jt�� �}|�|d d� � z|�|dd � � W n tyN t�Y n0 d S )N���)r r r r; r5 Zverifyr r )r r$ r= r r r rF t s zFernet._verify_signature)r$ rB rC r c C s� |d ur0|\}}|| |k r t �|t |k r0t �| �|� |dd� }|dd� }tt�| j�t�|��� � }|� |�} z| |� � 7 } W n t y� t �Y n0 t �tjj��� } | � | �} z| | � � 7 } W n t y� t �Y n0 | S )NrJ � rP )r �_MAX_CLOCK_SKEWrF r r r2 r r r7 � decryptorr5 r6 r r r1 r3 �unpadder) r r$ rB rC r? r+ r- r< rS Zplaintext_paddedrT Zunpaddedr r r rA | s0 � zFernet._decrypt_data)N)N)r r r �typing�Union�bytes�str�Anyr � classmethodr# r* r( r' r, �OptionalrD rE rG � staticmethod�Tupler@ rF rA r r r r r s2 � � � � �r c @ s� e Zd Zeje d�dd�Zeed�dd�Zee ed�dd �Z eed�d d �Z deej e ed �dd�Z ee e ed�dd�Zd S )� MultiFernet)�fernetsc C s t |�}|std��|| _d S )Nz1MultiFernet requires at least one Fernet instance)�listr �_fernets)r r_ r r r r � s �zMultiFernet.__init__)�msgr c C s | � |tt�� ��S r% r& )r rb r r r r* � s zMultiFernet.encrypt)rb r+ r c C s | j d �||�S )Nr )ra r' )r rb r+ r r r r' � s zMultiFernet.encrypt_at_timec C sh t �|�\}}| jD ]0}z|�||d �}W qJW q tyB Y q0 qt�t�d�}| jd �|||�S )Nr r )r r@ ra rA r r r! r, )r rb rB r$ �f�pr- r r r �rotate� s zMultiFernet.rotateN)rb r? r c C s: | j D ]*}z|�||�W S ty. Y q0 qt�d S r% )ra rD r )r rb r? rc r r r rD � s zMultiFernet.decrypt)rb r? r+ r c C s< | j D ],}z|�|||�W S ty0 Y q0 qt�d S r% )ra rE r )r rb r? r+ rc r r r rE � s zMultiFernet.decrypt_at_time)N)r r r rU �Iterabler r rW r* r( r' re r[ rD rE r r r r r^ � s �r^ )r rL r r9 r) rU Z cryptographyr Zcryptography.exceptionsr Z cryptography.hazmat.primitivesr r Z&cryptography.hazmat.primitives.ciphersr r r Z#cryptography.hazmat.primitives.hmacr � Exceptionr rR �objectr r^ r r r r �<module> s