U %Y01@sddlmZddlZddlmZddlZddlZddlZddlm Z ddl m Z m Z m Z ddlmZddlmZedd d gZGd d d eZd dZejjdkrddZnddZddZdS))unicode_literalsN) namedtuple)urlsafe_b64encode)convert_to_bytesconvert_to_string raw_b64decode)BaseSerializer)MacaroonSerializationExceptionPacketV2 field_typedatac@seZdZdZdZdZdZdZdZddZ dd Z d d Z d d Z ddZ ddZddZddZddZddZd!ddZddZdd ZdS)"BinarySerializerrcCst||ddS)Nascii=)r serialize_rawdecoderstrip)selfmacaroonrK/usr/lib/python3/dist-packages/pymacaroons/serializers/binary_serializer.py serializeszBinarySerializer.serializecCs*ddlm}|j|kr ||S||S)Nr) MACAROON_V1)pymacaroons.macaroonrversion _serialize_v1 _serialize_v2)rrrrrrr#s   zBinarySerializer.serialize_rawcCs|d|j}||d|j7}|jD]F}||d|j7}|jr&|jr&||d|j7}||d|j7}q&||dt|j 7}|S)Nlocation identifiercidvidcl signature) _packetizelocation identifiercaveatsZ _caveat_idZ_verification_key_idZ _locationbinascii unhexlifysignature_bytes)rrZcombinedZcaveatrrrr)s   zBinarySerializer._serialize_v1cCsddlm}t}|||jdk rN|||jt|j|||j|j |||j |j D]d}|jdk r|||jt|j|||j|j |j dk r|||jt|j |||j qb|||j |||jt|jt|S)Nr MACAROON_V2)rr/ bytearrayappendr(_append_packet _LOCATIONr _IDENTIFIERZidentifier_bytes_EOSr*Zcaveat_id_bytesverification_key_id_VID _SIGNATUREr+r,r-bytes)rrr/r crrrr ;s6          zBinarySerializer._serialize_v2cCs.t|dkrtdt|}t|}||S)Nrzempty macaroon)len ValueErrorrrdeserialize_raw)r serializeddecodedrrr deserializeUs  zBinarySerializer.deserializecCsZddlm}ddlm}t|dd}||kr<||St|rN||S|ddS)Nrr. MacaroonDeserializationExceptionrz7cannot determine data format of binary-encoded macaroon) rr/pymacaroons.exceptionsrBsixZbyte2int_deserialize_v2 _is_ascii_hex_deserialize_v1)rr>r/rBfirstrrrr=\s    z BinarySerializer.deserialize_rawc Cs.ddlm}m}ddlm}ddlm}||d}d}|t|kr*tt d||||j dd}|||j ||} | | \} } | dkr| |_ n| d kr| |_nt| d kr|j|| |d nV| d kr| |jd _n@| dkr| |jd _ n*| dkrt| |_n|dj| | d||}q6|S)Nr)MacaroonrCaveatrArs4sr!r"r#) caveat_idrr$r%r&z7Key {key} not valid key for this format. Value: {value})keyvalue)rrIrpymacaroons.caveatrKrCrBr;intstructZunpackPACKET_PREFIX_LENGTH _depacketizer(r)r*r1r6r+hexlify signatureformat) rr?rIrrKrBrindexZ packet_lengthpacketrPrQrrrrGhsT       z BinarySerializer._deserialize_v1c Csddlm}m}ddlm}ddlm}|dd}||\}}d}t|dkr~|dj |j kr~|dj d}|dd}t|dks|dj |j kr|d||dj ||d }||\} }| }t|dkrؐq||d } t|dkr |dj |j kr |dj d| _|dd}t|dks@|dj |j krH|d |dj | _|dd}t|dkr| jdk r|d |j| qt|dkr|d |dj |jkr|d|dj | _|j| q||\}} | j |jkr|dt| j |_|S)Nr)rIr/rJrArzutf-8zinvalid macaroon header)r)r(rrLzno identifier in caveatz*location not allowed in first party caveatzextra fields found in caveatzinvalid field found in caveatz+unexpected field found instead of signature)rrIr/rRrKrCrB_parse_section_v2r;r r3r rr4r(rNr*r1r7r6_parse_packet_v2r8r+rWrX) rr>rIr/rKrBZsectionZlocrrestZcavr[rrrrEsl              z BinarySerializer._deserialize_v2cCs|jdt|t|}t|dd}|dkrFtdj|||d|dd}|dt|d}t td t|||}|S) NriztPacket too long for serialization. Max length is 0xFFFF (65535). Packet length: 0x{hex_length} ({length}) Key: {key})rPZ hex_lengthZlengthrr  z4s%ds) rUr;hexr rYzfillencoderrTZpack)rrPr Z packet_sizeZpacket_size_hexheaderZpacket_contentr[rrrr's$ zBinarySerializer._packetizecCs*|dd}|t|dd}||fS)Nr`rrrO)splitr;)rr[rPrQrrrrVszBinarySerializer._depacketizeNcCs0t||||jkr,t|t|||dS)N)_encode_uvarintr5r;extend)rr r Z packet_datarrrr2s  zBinarySerializer._append_packetcCsvddlm}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 rrArOz"section extends past end of bufferzfields out of orderN)rCrBr;r^r r5r1)rr rBZprev_field_typeZpacketsr_r[rrrr]s     z"BinarySerializer._parse_section_v2cCsddlm}t|\}}||d}||jkr<|t|dfSt|\}}||d}|t|krh|d||dt||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 rrANz%field data extends past end of buffer)rCrB_decode_uvarintr5r r;)rr rBZftnZ payload_lenrrrr^s       z!BinarySerializer._parse_packet_v2)N)__name__ __module__ __qualname__rUr3r4r7r8r5rrrr r@r=rGrEr'rVr2r]r^rrrrr s& 0; r cCsH|dkrtd|d@}|dL}|dkr4||qD||dBqdS)z7 Encodes integer into variable-length format into data.rzonly support positive integerN)r<r1)r rjZ this_byterrrrg-s rgrcCsfd}d}d}|D]H}t|}|d7}|dkr@|||>B|fS||d@|>O}|d7}qtddSz 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. rrrprnrozcannot read uvarint from bufferN)ord Exceptionr rjshiftibrrrri;s ricCs^d}d}d}|D]@}|d7}|dkr8|||>B|fS||d@|>O}|d7}qtddSrq)rsrtrrrriMs cCsLtd|krtdkr$nndStd|kr@tdkrHnndSdS)N09TafF)rr)rwrrrrF_s   rF)Z __future__rr+ collectionsrrDrTsysbase64rZpymacaroons.utilsrrrZ'pymacaroons.serializers.base_serializerrrCr r r rg version_infomajorrirFrrrrs$