@OOf" dZddlZddlmZddlmZedeeeee e e e f Z de de fdZde deddfd Zd ej d ede defd Zd e de fdZdede fdZdede de fdZde de e e ffdZgdZy)aY tnetstring: data serialization using typed netstrings ====================================================== This is a custom Python 3 implementation of tnetstrings. Compared to other implementations, the main difference is that this implementation supports a custom unicode datatype. An ordinary tnetstring is a blob of data prefixed with its length and postfixed with its type. Here are some examples: >>> tnetstring.dumps("hello world") 11:hello world, >>> tnetstring.dumps(12345) 5:12345# >>> tnetstring.dumps([12345, True, 0]) 19:5:12345#4:true!1:0#] This module gives you the following functions: :dump: dump an object as a tnetstring to a file :dumps: dump an object as a tnetstring to a string :load: load a tnetstring-encoded object from a file :loads: load a tnetstring-encoded object from a string Note that since parsing a tnetstring requires reading all the data into memory at once, there's no efficiency gain from using the file-based versions of these functions. They're only here so you can use load() to read precisely one item from a file or socket without consuming any extra data. The tnetstrings specification explicitly states that strings are binary blobs and forbids the use of unicode at the protocol level. **This implementation decodes dictionary keys as surrogate-escaped ASCII**, all other strings are returned as plain bytes. :Copyright: (c) 2012-2013 by Ryan Kelly . :Copyright: (c) 2014 by Carlo Pires . :Copyright: (c) 2016 by Maximilian Hils . :License: MIT N)BinaryIO)Unionvaluereturncftj}t|d|dj|S)z> This function dumps a python object as a tnetstring. r) collectionsdeque_rdumpqjoin)rqs X/var/www/premiumrankchecker/venv/lib/python3.12/site-packages/mitmproxy/io/tnetstring.pydumpsr2s,',,.A Aq% 88A;r file_handlec8|jt|y)zb This function dumps a python object as a tnetstring and writes it to the given file. N)writer)rrs rdumpr>s eEl#rr sizecJ|j}| |d|dzS|dur |d|dzS|dur |d|dzSt|tr^t|j }t |}t|j }|d ||fz|d zt |z|zSt|t r^t|j }t |}t|j }|d ||fz|d zt |z|zSt|trZ|}t |}t|j }|d |||d |||d zt |z|zSt|tri|j d}t |}t|j }|d|||d |||d zt |z|zSt|ttfri|d|dzx}}t|D]}t|||}t||z j }|d |||dzt |zSt|tr~|d|dzx}}|jD]\} } t||| }t||| }!t||z j }|d |||dzt |zStd|dt!|d)a Dump value as a tnetstring, to a deque instance, last chunks first. This function generates the tnetstring representation of the given value, pushing chunks of the output onto the given deque instance. It pushes the last chunk first, then recursively generates more chunks. When passed in the current size of the string in the queue, it will return the new size of the string in the queue. Operating last-chunk-first makes it easy to calculate the size written for recursive structures without having to build their representation as a string. This is measurably faster than generating the intermediate strings, especially on deeply nested structures. s0:~Ts4:true!Fs5:false!s%s:%s#s%s:%s^,:utf8;]}zunserializable object: z ()) appendleft isinstanceintstrencodelenfloatreprbyteslisttuplereversedr dictitems ValueErrortype) r rrrdataldataspan init_sizeitemkvs rr r Fs LLE } f ax $ jax % kax E3 5z  "D 5z  " i4,&'ax#d)#e++ E5 ! E{!!#D 5z  " i4,&'ax#d)#e++ E5 !D 5z  " d  d  d  d ax#d)#e++ E3 ||F#D 5z  " d  d  d  d ax#d)#e++ ED%= ) d !8# DUO *D1dD)D *4)#$++- d  d ax#d)## E4  d !8# DKKM 'DAq1dA&D1dA&D '4)#$++- d  d ax#d)##25'DK=JKKrstringct|dS)zA This function parses a tnetstring into a python object. r)pop)r9s rloadsr<s v;q>rc||jd}|dk(r tdd}|jr@||z }t|dkDr td|jd}|jr@|dk7r td|jt |}|jdd}t ||S) zload(file) -> object This function reads a tnetstring from a file and parses it into a python object. The file must support the read() method, and this function promises not to read more data than necessary. rrznot a tnetstring: empty file z.not a tnetstring: absurdly large length prefixrz2not a tnetstring: missing or invalid length prefixr)readr0isdigitr'r$parse)rc data_lengthr2 data_types rloadrEs ACx788K ))+q { b MN N   Q  ))+  DyMNN   C , -D  #A&I D !!rrDr2c|tdk(r|S|tdk(r|jdS|tdk(r t|S|tdk(r t |S|tdk(r|d k(ry |d k(ry td ||tdk(r|rtd|y|tdk(r(g}|r"t |\}}|j ||r"|S|tdk(r*i}|r$t |\}}t |\}}|||<|r$|Std|#t$rtd|wxYw#t$rtd|wxYw)Nrrr#z+not a tnetstring: invalid integer literal: ^z)not a tnetstring: invalid float literal: !strueTsfalseFz+not a tnetstring: invalid boolean literal: ~z(not a tnetstring: invalid null literal: rr zunknown type tag: )orddecoder$r0r(r;append)rDr2lstr6dkeyvals rrArAsCI CI{{6""CI Ut9 CI S; CI 7? X J4(ST TCI GxPQ QCITJD$ JJt  CI D ICD ICAcF ))5 66? UJ4(ST T U  SHQR R Ss D' E'D?Ec |jdd\}}t|} |d|||||dzd}}}t |||fS#t$rtd|wxYw#t$rtd|wxYw)z This function parses a tnetstring into a python object. It returns a tuple giving the parsed object and a string containing any unparsed data from the end of the string. rrz4not a tnetstring: missing or invalid length prefix: Nz)not a tnetstring: invalid length prefix: )splitr$r0 IndexErrorrA)r2blengthlengthrDremains rr;r;s  4+ W O"&w-ftFQJL?Qi D !6 ))  B4( K   ODVHMNNOs AAAA6)rrrEr<r;)__doc__r typingrrr%boolr$r(r*r+r,r. TSerializablerrr r r<rErAr;__all__rrr^s(TdCsE5$tKL   5 $ $H$$PL{  PLPLMPLcPLf%M"h"="6'7S'7'7-'7T*e*mU23*0 4r