vd#dZddlmZdZddlZddlmZdZdZd Z d Z d Z d Z d Z dZdZdZdZddZdZdZdZddZdS)z Utilities )print_functionFN) struct_unpackc t|tr|St|}|dkr|dS|dkr!|dd|z zz}td|dS|dd|z zz}td|dS#xYw)z&Unpacks the given buffer to an integerrrA>&A> A>>Bc||dks|dkrtdttjd|S)z^ Takes an unsigned byte (1 byte) and packs it as a bytes-object. Returns string. rz int1store requires 0 <= i <= 255  1uuC;<<<T1--...c||dks|dkrtdttjd|S)z` Takes an unsigned short (2 bytes) and packs it as a bytes-object. Returns string. rz"int2store requires 0 <= i <= 65535  1uuE =>>>T1--...rc|dks|dkrtdttjd|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'MsG  1uuH @AAAT1--ac2333rc||dks|dkrtdttjd|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*Ys>  1uuJBCCCT1--...rc||dks|dkrtdttjd|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-esA  1uu(((<===T1--...rc|dks|dkrtd|dkrt}n1|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| 1uu(((<===Cxx! e! h! j! ! =  rc|dks|dkrtd|dkr"ttjd|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  1uu(((23333wwT1--... e6;tQ#7#78888 h6;tQ#7#7!#<====6;tQ#7#78888rc.|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$ af+C J rc:|ddkr |dddfSdx}}|d}|dkr|}|d|zd|d|dzfS|dkrd}n|dkrd }|d krd }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}}ABBFU a&C czzAJKK #a l"344   czz S519% & &F "## $c%!)FUNQ4F*F&G HHrc"g}dddd}t|}d}||kr||}|dkrdS|dkr|d|d z }n|d kr1|}|||d z||d zz|d |zz }ntd} ||}n#t$rYdSwxYwt||d z||d zz}|||d z|z||z|d zz|d |z|zz }||kt |S) 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_listrPsy H!! $ $E#hhG C --C C<<4 C<< OOD ! ! ! 1HCC||S1Wfa.@$@ ABBBq6z! !%LEE   44  cAgua/@%@!ABBa%#'(BBCEEEq5y6))1 --4 ??s B B"!B"c||td|K ||}n#t$rtdwxYw||dzd|d|fS|t||Std)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|ABBB  ?))C..CC ? ? ?=>> > ?C!GHH s1S5z** #t$$$ E F FFs -AcX t|d|}n#xYw||d|fS)zMRead an integer from buffer Returns a tuple (truncated buffer, int) rNrCr9s rread_intrWs= c!D&k"" J sc|std|d}|dkr |dddfS|dkr|ddt|fS|dkr(|ddtd|dddfS|d kr+|d dtd |dd d zdfS|d kr(|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&zz"_digest_buffer..Fs333! A 333rc2g|]}dt|zSr_)ordr`s rrcz"_digest_buffer..Gs#4441IA&444r)r strjoin)rs r_digest_bufferrhCsV c3  5ww33s333444 7744444 5 55rc|rF|r|dkrt|d|}nt|}t|dz|zdStt|dS)z2Debug function printing output of _digest_buffer()rz: N)rhprint)abufferprefixlimitdigests r print_bufferrpIsx '  -UQYY#GAeG$455FF#G,,F ftmf$%%%%% nW%%&&&&&r)NN)Nri)__doc__ __future__r__MYSQL_DEBUG__rcatch23rrrr#r'r*r-r0r6r<rFrPrUrWr[rhrpr8rrrusS0&%%%%% """"""$ / / / / / / 4 4 4 / / / / / /4999$2I2I2Ij&&&RGGGG*   BBB6666 ' ' ' ' ' 'r