File: //usr/local/lib/python3.9/site-packages/celery/security/__pycache__/serialization.cpython-39.pyc
a
X>h� � @ s� d Z ddlmZmZmZ ddlmZmZmZ ddl m
Z
ddl m Z m
Z
ddlmZmZ ddlmZ dd lmZmZ d
Zed �ZG d d
� d
�Zdddde
dfdd�ZdS )zSecure serializer.� )�dumps�loads�registry)� bytes_to_str� ensure_bytes� str_to_bytes)�DEFAULT_SECURITY_DIGEST)� b64decode� b64encode� )� Certificate� FSCertStore)�
PrivateKey)�get_digest_algorithm�reraise_errors)�SecureSerializer�
register_authz c @ sL e Zd ZdZdddedfdd�Zdd� Zdd � Zefd
d �Z efd d
�Z
dS )r zSigned serializer.N�jsonc C s& || _ || _|| _t|�| _|| _d S �N)�_key�_cert� _cert_storer �_digest� _serializer)�self�key�certZ
cert_store�digest�
serializer� r �G/usr/local/lib/python3.9/site-packages/celery/security/serialization.py�__init__ s
zSecureSerializer.__init__c C s� | j dusJ �| jdus J �tdtf��R t|| jd�\}}}t|�}| j|||| j �|| j �| j�
� d�W d � S 1 s�0 Y dS )z%Serialize data structure into string.NzUnable to serialize: {0!r}�r )� signature�signer) r r r � Exceptionr r r �_pack�signr Zget_id)r �data� content_type�content_encoding�bodyr r r � serialize s �
�zSecureSerializer.serializec C s� | j dusJ �tdtf��L | �|�}|d |d |d }}}| j | �||| j� W d � n1 sl0 Y t||d |d dd �S )
z'Deserialize data structure from string.Nz Unable to deserialize: {0!r}r# r$ r+ r) r* T)�force)r r r% �_unpack�verifyr r )r r( �payloadr# r$ r+ r r r � deserialize0 s
�
4
�z SecureSerializer.deserializec C s. |� dd� t|�t|�|||fD ��}t|�S )Nc s s | ]}t |�V qd S r )r )�.0�sr r r � <genexpr>> s z)SecureSerializer._pack.<locals>.<genexpr>)�joinr
)r r+ r) r* r$ r# �sep�fieldsr r r r& < s
��zSecureSerializer._packc C sN t t|��}|j|dd�}t |d �t |d �t|d �t|d �|d d�S )N� )�maxsplitr r � � )r$ r# r) r* r+ )r r �splitr )r r0 r6 Z raw_payload�vr r r r. D s
�zSecureSerializer._unpack) �__name__�
__module__� __qualname__�__doc__r r! r, r1 �DEFAULT_SEPARATORr& r. r r r r r s �
�
r Nr c C sJ t | ot| |d�|ot|�|o$t|�||d�}tjd|j|jddd� dS )z Register security serializer.)�passwordr" �authzapplication/datazutf-8)r) r* N)r r r r
r �registerr, r1 )r � key_passwordr �storer r r3 r r r r P s
��r )rA Zkombu.serializationr r r Zkombu.utils.encodingr r r Zcelery.app.defaultsr Zcelery.utils.serializationr r
Z certificater r
r r �utilsr r �__all__rB r r r r r r �<module> s <�