File: //lib/python3/dist-packages/pymacaroons/serializers/__pycache__/binary_serializer.cpython-38.pyc
U
�%�Y01 � @ s� d dl mZ d dlZd dlmZ d dlZd dlZd dlZd dlm Z d dl
m Z m Z m
Z
d dlmZ d dlmZ edd d
g�ZG d d � d e�Zd
d� Zejjdkr�dd� Zndd� Zdd� ZdS )� )�unicode_literalsN)�
namedtuple)�urlsafe_b64encode)�convert_to_bytes�convert_to_string�
raw_b64decode)�BaseSerializer)� MacaroonSerializationException�PacketV2�
field_type�datac @ s� e Zd ZdZdZdZdZdZdZdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd!dd �Zd d � Zd d � ZdS )"�BinarySerializer� � � � r c C s t | �|���d��d�S )N�ascii�=)r �
serialize_raw�decode�rstrip)�self�macaroon� r �K/usr/lib/python3/dist-packages/pymacaroons/serializers/binary_serializer.py� serialize s ���zBinarySerializer.serializec C s* ddl m} |j|kr | �|�S | �|�S )Nr )� MACAROON_V1)�pymacaroons.macaroonr �version�
_serialize_v1�
_serialize_v2)r r r r r r r # s
z BinarySerializer.serialize_rawc C s� | � d|j�}|| � d|j�7 }|jD ]F}|| � d|j�7 }|jr&|jr&|| � d|j�7 }|| � d|j�7 }q&|| � dt�|j ��7 }|S )N� location�
identifier� cid� vid� cl� signature)
�
_packetize�location�
identifier�caveatsZ
_caveat_idZ_verification_key_idZ _location�binascii� unhexlify�signature_bytes)r r ZcombinedZcaveatr r r r ) s
�
�z BinarySerializer._serialize_v1c C s� ddl m} t� }|�|� |jd k rN| �|| jt|j�� | �|| j|j � | �|| j
� |j D ]d}|jd k r�| �|| jt|j�� | �|| j|j � |j
d k r�| �|| jt|j
�� | �|| j
� qb| �|| j
� | �|| jt�|j�� t|�S )Nr �� MACAROON_V2)r r/ � bytearray�appendr( �_append_packet� _LOCATIONr � _IDENTIFIERZidentifier_bytes�_EOSr* Zcaveat_id_bytes�verification_key_id�_VID�
_SIGNATUREr+ r, r- �bytes)r r r/ r �cr r r r ; s6
�
�
�
��z BinarySerializer._serialize_v2c C s. t |�dkrtd��t|�}t|�}| �|�S )Nr zempty macaroon)�len�
ValueErrorr r �deserialize_raw)r �
serialized�decodedr r r � deserializeU s
z BinarySerializer.deserializec C sZ ddl m} ddlm} t�|d d� �}||kr<| �|�S t|�rN| �|�S |d��d S )Nr r. �� MacaroonDeserializationExceptionr z7cannot determine data format of binary-encoded macaroon) r r/ �pymacaroons.exceptionsrB �sixZbyte2int�_deserialize_v2�
_is_ascii_hex�_deserialize_v1)r r> r/ rB �firstr r r r= \ s
�z BinarySerializer.deserialize_rawc C s. ddl m}m} ddlm} ddlm} ||d�}d}|t|�k �r*tt �
d|||| j � �d d�}||| j || � } | � | �\}
} |
dkr�| |_
n�|
d kr�| |_nt|
d
kr�|j�|| |d �� nV|
d kr�| |jd
_n@|
dkr�| |jd
_
n*|
dk�rt�| �|_n|dj|
| d���|| }q6|S )Nr )�Macaroonr ��CaveatrA �r s 4s� r! r" r# )� caveat_idr r$ ���r% r& z7Key {key} not valid key for this format. Value: {value})�key�value)r rI r �pymacaroons.caveatrK rC rB r; �int�structZunpack�PACKET_PREFIX_LENGTH� _depacketizer( r) r* r1 r6 r+ �hexlify� signature�format) r r? rI r rK rB r �indexZ
packet_length�packetrP rQ r r r rG h sT
��� �
�
��
z BinarySerializer._deserialize_v1c C s ddl m}m} ddlm} ddlm} |dd � }| �|�\}}d}t|�dkr~|d j | j
kr~|d j � d�}|dd � }t|�dks�|d j | j
kr�|d��||d j ||d �}| �|�\} }| }t|�dkrؐq�||d
�}
t|�dk�r |d j | j
k�r |d j � d�|
_|dd � }t|�dk�s@|d j | j
k�rH|d ��|d j |
_|dd � }t|�dk�r�|
jd k �r�|d ��|j�|
� q�t|�dk�r�|d
��|d j | jk�r�|d��|d j |
_|j�|
� q�| �|�\}} | j | jk�r�|d��t�| j �|_|S )Nr )rI r/ rJ rA r � zutf-8zinvalid macaroon header)r) r( r rL zno identifier in caveatz*location not allowed in first party caveatz extra fields found in caveatz invalid field found in caveatz+unexpected field found instead of signature)r rI r/ rR rK rC rB �_parse_section_v2r; r r3 r r r4 r( rN r* r1 r7 r6 �_parse_packet_v2r8 r+ rW rX ) r r> rI r/ rK rB ZsectionZlocr �restZcavr[ r r r rE � sl �
� � �� �z BinarySerializer._deserialize_v2c C s� | j d t|� t|� }t|�dd � }|dkrFtdj|||d���|�d��d�}|d t|� d }t� td t|� �||�}|S )
Nr i�� ztPacket too long for serialization. Max length is 0xFFFF (65535). Packet length: 0x{hex_length} ({length}) Key: {key})rP Z
hex_lengthZlengthr r � �
z4s%ds)
rU r; �hexr rY �zfill�encoder rT Zpack)r rP r Z packet_sizeZpacket_size_hex�headerZpacket_contentr[ r r r r' � s$ �� �zBinarySerializer._packetizec C s* |� d�d }|t|�d d� }||fS )Nr` r r rO )�splitr; )r r[ rP rQ r r r rV � s z BinarySerializer._depacketizeNc C s0 t ||� || jkr,t |t|�� |�|� d S )N)�_encode_uvarintr5 r; �extend)r r r Z packet_datar r r r2 � s
z BinarySerializer._append_packetc C sv ddl m} d}g }t|�dkr(|d��| �|�\}}|j| jkrJ||fS |j|kr\|d��|�|� |j}|}qdS )z� Parses a sequence of packets in data.
The sequence is terminated by a packet with a field type of EOS
:param data bytes to be deserialized.
:return: the rest of data and an array of packet V2
r rA rO z"section extends past end of bufferzfields out of orderN)rC rB r; r^ r r5 r1 )r r rB Zprev_field_typeZpacketsr_ r[ r r r r] � s �
z"BinarySerializer._parse_section_v2c C s� ddl m} t|�\}}||d� }|| jkr<|t|d�fS t|�\}}||d� }|t|�krh|d��||d� t||d|� �fS )aQ Parses a V2 data packet at the start of the given data.
The format of a packet is as follows:
field_type(varint) payload_len(varint) data[payload_len bytes]
apart from EOS which has no payload_en or data (it's a single zero
byte).
:param data:
:return: rest of data, PacketV2
r rA Nz%field data extends past end of buffer)rC rB �_decode_uvarintr5 r
r; )r r rB Zft�nZ payload_lenr r r r^ s
�z!BinarySerializer._parse_packet_v2)N)�__name__�
__module__� __qualname__rU r3 r4 r7 r8 r5 r r r r r@ r= rG rE r' rV r2 r] r^ r r r r r
s& 0;
r
c C sH |dk rt d��|d@ }|dL }|dkr4| �|� qD| �|dB � qdS )z7 Encodes integer into variable-length format into data.r z only support positive integer� � � N)r< r1 )r rj Z this_byter r r rg - s
rg r c C sf d}d}d}| D ]H}t |�}|d7 }|dk r@|||> B |f S ||d@ |> O }|d7 }qtd��dS �z� Decode a variable -length integer.
Reads a sequence of unsigned integer byte and decodes them into an
integer in variable-length format and returns it and the length read.
r r rp rn ro z cannot read uvarint from bufferN)�ord� Exception�r rj �shift�i�br r r ri ; s
ri c C s^ d}d}d}| D ]@}|d7 }|dk r8|||> B |f S ||d@ |> O }|d7 }qt d��dS rq )rs rt r r r ri M s
c C sL t d�| kr t d�kr$n ndS t d�| kr@t d�krHn ndS dS )N�0�9T�a�fF)rr )rw r r r rF _ s
rF )Z
__future__r r+ � collectionsr rD rT �sys�base64r Zpymacaroons.utilsr r r Z'pymacaroons.serializers.base_serializerr rC r r
r
rg � version_info�majorri rF r r r r �<module> s$