CasperSecurity
o
�%�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� ZejjdkrVdd� 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 ]#}|| � d|j�7 }|jr6|jr6|| � d|j�7 }|| � d|j�7 }q|| � dt�|j ��7 }|S )N� location�
identifier� cid� vid� cl� signature)
�
_packetize�location�
identifier�caveats�
_caveat_id�_verification_key_id� _location�binascii� unhexlify�signature_bytes)r r �combined�caveatr r r r ) s
��
�z BinarySerializer._serialize_v1c C s� ddl m} t� }|�|� |jd ur'| �|| jt|j�� | �|| j|j � | �|| j
� |j D ]2}|jd urC| �|| jt|j�� | �|| j|j � |j
d ur\| �|| jt|j
�� | �|| j
� q1| �|| j
� | �|| jt�|j�� t|�S )Nr �� MACAROON_V2)r r4 � bytearray�appendr( �_append_packet� _LOCATIONr � _IDENTIFIER�identifier_bytes�_EOSr* �caveat_id_bytes�verification_key_id�_VID�
_SIGNATUREr. r/ r0 �bytes)r r r4 r �cr r r r ; s6
�
�
�
��z BinarySerializer._serialize_v2c C s. t |�dkr
td��t|�}t|�}| �|�S )Nr zempty macaroon)�len�
ValueErrorr r �deserialize_raw)r �
serialized�decodedr r r � deserializeU s
z BinarySerializer.deserializec C sV ddl m} ddlm} t�|d d� �}||kr | �|�S t|�r'| �|�S |d��)Nr r3 �� MacaroonDeserializationExceptionr z7cannot determine data format of binary-encoded macaroon) r r4 �pymacaroons.exceptionsrI �six�byte2int�_deserialize_v2�
_is_ascii_hex�_deserialize_v1)r rE r4 rI �firstr r r rD \ s
�z BinarySerializer.deserialize_rawc C s4 ddl m}m} ddlm} ddlm} ||d�}d}|t|�k r�tt �
d|||| j � �d d�}||| j || � } | � | �\}
} |
dkrM| |_
nA|
d krU| |_n9|
d
krd|j�|| |d �� n*|
d kro| |jd
_n |
dkrz| |jd
_
n|
dkr�t�| �|_n |dj|
| d���|| }|t|�k s!|S )Nr )�Macaroonr ��CaveatrH �r s 4s� r! r" r# )� caveat_idr r$ ���r% r& z7Key {key} not valid key for this format. Value: {value})�key�value)r rQ r �pymacaroons.caveatrS rJ rI rB �int�struct�unpack�PACKET_PREFIX_LENGTH� _depacketizer( r) r* r6 r= r. �hexlify� signature�format) r rF rQ r rS rI r �index�
packet_length�packetrX rY r r r rO 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|�dksM|d j | j
krQ|d��||d j ||d �} | �|�\} }| }t|�dkrlny||d �}
t|�dkr�|d j | j
kr�|d j � d�|
_|dd � }t|�dks�|d j | j
kr�|d ��|d j |
_|dd � }t|�dkr�|
jd ur�|d
��|j�|
� q[t|�dkr�|d��|d j | jkr�|d��|d j |
_|j�|
� q\| �|�\}} | j | jkr�|d��t�| j �|_|S )Nr )rQ r4 rR rH r � zutf-8zinvalid macaroon header)r) r( r TrT 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 rQ r4 rZ rS rJ rI �_parse_section_v2rB r r8 r r r9 r( rV r* r6 r> r= �_parse_packet_v2r? r. r` ra ) r rE rQ r4 rS rI �section�locr �rest�cavre r r r rM � sp �
�
� �� �! �z BinarySerializer._deserialize_v2c C s� | j d t|� t|� }t|�dd � }|dkr#tdj|||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})rX �
hex_length�lengthr r � �
z4s%ds)
r^ rB �hexr rb �zfill�encoder r\ �pack)r rX r � packet_size�packet_size_hex�header�packet_contentre r r r r' � s& �� �zBinarySerializer._packetizec C s* |� d�d }|t|�d d� }||fS )Nro r r rW )�splitrB )r re rX rY r r r r_ � s z BinarySerializer._depacketizeNc C s4 t ||� || jkrt |t|�� |�|� d S d S �N)�_encode_uvarintr; rB �extend)r r r � packet_datar r r r7 � s
�z BinarySerializer._append_packetc C st ddl m} d}g } t|�dkr|d��| �|�\}}|j| jkr&||fS |j|kr/|d��|�|� |j}|}q )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 rH rW Tz"section extends past end of bufferzfields out of order)rJ rI rB rh r r; r6 )r r rI �prev_field_type�packetsrk re r r r rg � s" �
�z"BinarySerializer._parse_section_v2c C s� ddl m} t|�\}}||d� }|| jkr |t|d�fS t|�\}}||d� }|t|�kr4|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 rH Nz%field data extends past end of buffer)rJ rI �_decode_uvarintr; r
rB )r r rI �ft�n� payload_lenr r r rh s
� z!BinarySerializer._parse_packet_v2rz )�__name__�
__module__� __qualname__r^ r8 r9 r> r? r; r r r r rG rD rO rM r' r_ r7 rg rh r r r r r
s( 0;
r
c C sH |dk rt d�� |d@ }|dL }|dkr | �|� dS | �|dB � q )z7 Encodes integer into variable-length format into data.r z only support positive integerT� � � N)rC r6 )r r� � this_byter r r r{ - s
�r{ r c C sb d}d}d}| D ]$}t |�}|d7 }|dk r |||> B |f S ||d@ |> O }|d7 }qtd���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 r� r� r� z cannot read uvarint from buffer)�ord� Exception�r r� �shift�i�br r r r� ; s
r� c C sZ d}d}d}| D ] }|d7 }|dk r |||> B |f S ||d@ |> O }|d7 }qt d��r� )r� r� r r r r� M s
c C sH t d�| krt d�krdS t d�| kr t d�kr"dS dS dS )N�0�9T�a�fF)r� )r� r r r rN _ s ��rN )�
__future__r r. � collectionsr rK r\ �sys�base64r �pymacaroons.utilsr r r �'pymacaroons.serializers.base_serializerr rJ r r
r
r{ � version_info�majorr� rN r r r r �<module> s&