Dh#dZddlmZdZddlZddlmZdZdZd Z d Z d Z d Z d Z dZdZdZdZddZdZdZdZddZy)z Utilities )print_functionFN) struct_unpackc t|tr|St|}|dk(r|dS|dkr|dd|z zz}td|dS|dd|z zz}td|dS#xYw)z&Unpacks the given buffer to an integerrr>T1-..rcp|dks|dkDr tdttjd|ddS)zb Takes an unsigned integer (3 bytes) and packs it as a bytes-object. Returns string. rz%int3store requires 0 <= i <= 16777215r rrs r int3storer'Ms:  1uH @AAT1-a233rcj|dks|dkDr tdttjd|S)zb Takes an unsigned integer (4 bytes) and packs it as a bytes-object. Returns string. rz'int4store requires 0 <= i <= 4294967295r rrs r int4storer*Ys3  1uJBCCT1-..rcj|dks|dkDr tdttjd|S)zZ Takes an unsigned integer (8 bytes) and packs it as string. Returns string. rz!int8store requires 0 <= i <= 2^64r rrs r int8storer-es5  1u((<==T1-..rc|dks|dkDr td|dkrt}n*|dkrt}n|dkrt}n|dkrt}nt }||S)z Takes an unsigned integers and packs it as a bytes-object. This function uses int1store, int2store, int3store, int4store or int8store depending on the integer value. returns string. rr,z!intstore requires 0 <= i <= 2^64rr!r%r))rrr#r'r*r-)r formed_strings rintstorer0qs_ 1u((<==Cx! e! h! j! !  rcZ|dks|dkDr td|dkrttjd|S|dkr"dttjd|zS|d kr%d ttjd |dd zSd ttjd|zS)z| Takes an unsigned integer and packs it as bytes, with the information of how much bytes the encoded int takes. rr,zRequires 0 <= i <= 2^64rr!r"r%r r&r rrs rlc_intr6s  1u((2333wT1-.. e6;;tQ#7888 h6;;tQ#7!#<===6;;tQ#7888rc|d|}||d|fS)zi Reads bytes from a buffer. Returns a tuple with buffer less the read bytes, and the bytes. rNrsizeress r read_bytesr<s a+C J rc|ddk(r|dddfSdx}}|d}|dkr|}|d|zd|d|dzfS|dk(rd}n|dk(rd }|d k(rd }t|d|dz}|||zdzd||dz||zdzfS) u Takes a buffer and reads a length coded string from the start. This is how Length coded strings work If the string is 250 bytes long or smaller, then it looks like this: <-- 1b --> +----------+------------------------- | length | a string goes here +----------+------------------------- If the string is bigger than 250, then it looks like this: <- 1b -><- 2/3/8 -> +------+-----------+------------------------- | type | length | a string goes here +------+-----------+------------------------- if type == ü: length is code in next 2 bytes elif type == ý: length is code in next 3 bytes elif type == þ: length is code in next 8 bytes NULL has a special value. If the buffer starts with û then it's a NULL and we return None as value. Returns a tuple (trucated buffer, bytes). rr2rNr&r r)rrlsizefsts rread_lc_stringrFs@ 1v}ABFU a&C czAJK #a "344   cz S519% &F "# $c%!)FUNQ4F&G HHrcg}dddd}t|}d}||kr||}|dk(ry|dk(r|jd|d z }nz|d kr(|}|j||d z||d zz|d |zz }nMd} ||}t||d z||d zz}|j||d z|z||z|d zz|d |z|zz }||krt |S#t$rYywxYw) zXReads all length encoded strings from the given buffer Returns a list of bytes r@r&r )r?rArBrrNr2rr>)rappendKeyErrorrtuple)rbyteslstsizesbuf_lenposfirstrrDs rread_lc_string_listrPs0 H!! $E#hG C -C C< C< OOD ! 1HC|S1Wfa.@ ABq6z! !%LE!cAgua/@!ABa%#'(BCEq5y6))1 -4 ?   s2C CCc| | td| |j|}||dzd|d|fS| t||Std#t$r tdwxYw)zr Reads a string up until a character or for a given size. Returns a tuple (trucated buffer, string). Nz&read_string() needs either end or sizezend byte not present in bufferrrz.read_string() needs either end or size (weird))rindexr<)rendr:idxs r read_stringrUs  {t|ABB  ?))C.CC!GH s1Sz** #t$$ E FF  ?=> > ?s A Ac: t|d|}||d|fS#xYw)zMRead an integer from buffer Returns a tuple (truncated buffer, int) rNrCr9s rread_intrWs4 c!Dk" J  sc,|s td|d}|dk(r|dddfS|dkr|ddt|fS|dk(r|ddtd|dddfS|d k(r|d dtd |dd d zdfS|d k(r|ddtd|dddfStd)z Takes a buffer and reads an length code string from the start. Returns a tuple with buffer less the integer and the integer read. z Empty buffer.rr2rNr?r&zroso0& "$ / / 4 / /49$2Ij&RG* B66 'r