U  W[.0@s6dZddlmZmZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZmZmZdd lmZGd d d eZd d ZddZedZedZedZedZedZedZedZedZ edZ!ddZ"da#e"ddZ$Gddde j%e j&Z'e'Z(e()e(*ddd Z+d!d"Z,dS)#z Banana -- s-exp based protocol. Future Plans: This module is almost entirely stable. The same caveat applies to it as applies to L{twisted.spread.jelly}, however. Read its future plans for more details. @author: Glyph Lefkowitz )absolute_importdivisionN)BytesIO)protocol)styles)log) iterbyteslong _bytesChr)fullyQualifiedNamec@s eZdZdS) BananaErrorN)__name__ __module__ __qualname__rr7/usr/lib/python3/dist-packages/twisted/spread/banana.pyr sr cCsJ|dkr|tddS|dks(td|rF|t|d@|d?}q(dS)Nrz!can only encode positive integers)chrAssertionError)Zintegerstreamrrrint2b128s rcCs6d}d}t|D] }t|}|||7}|dK}q|S)a Convert an integer represented as a base 128 string into an L{int} or L{long}. @param st: The integer encoded in a byte string. @type st: L{bytes} @return: The integer value extracted from the byte string. @rtype: L{int} or L{long} rr)rord)steicharnrrrb1282int's    rcCs|adS)ak Set the limit on the prefix length for all Banana connections established after this call. The prefix length limit determines how many bytes of prefix a banana decoder will allow before rejecting a potential object as too large. @type limit: L{int} @param limit: The number of bytes of prefix for banana to allow when decoding. N) _PREFIX_LIMIT)limitrrrsetPrefixLimitIs r*@i c @seZdZdZddgZdZeZddZddZ d d Z d d Z d dZ ddZ dZddZddZddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5ZiZeD]\ZZeee<qdr=r6rrrconnectionMades zBanana.connectionMadecCs,|j}|r|dd|n ||dS)Nr) listStackappendrB)r5itemlrrrgotItemszBanana.gotItemc Cs|j|}|j}|j}|r|j|ks@tdt|jt|f||_d}t|D]}|tkrbq|d}qR||jkrtd|jfdS|d|}|||d}||dd} t ||jkrtd|jf|t kr t |}|t krtd| |gf| }nz|tkrdt |}|t kr0tdt | |kr\| |d}|| d|ndSn"|tkr| }t |}||n|tkr| }t |}||n|tkr| }t |}|| n|tkr| }t | }||n|tkr0| }t |}|j|} |jdkr || ntd | nV|tkrxt | d krr| d d}|td | dd dndSntd |f|rt |d d|d dkr|d} || qqd|_dS)NzThis ain't right: %s %srrz1Security precaution: more than %d bytes of prefixz9Security precaution: longer than %d bytes worth of prefixz#Security precaution: List too long.z%Security precaution: String too long.r-z"Invalid item for pb protocol {0!r}!dzInvalid Type Byte %rrDrJ)bufferrErIrreprr HIGH_BIT_SETr0r lenLISTr SIZE_LIMITrFSTRINGINTLONGINTLONGNEGNEGVOCABincomingVocabularyr9NotImplementedErrorformatFLOATstructZunpackpop) r5chunkrMrErIposZchZnumZtypebyterestrGrrr dataReceiveds $                         zBanana.dataReceivedcCs tdS)zFCalled when an expression (list, string, or int) is received. N)rZ)r5lstrrrr;szBanana.expressionReceivedrr/rrK )sNonesclasss dereferences references dictionarysfunctionsinstanceslistsmodules persistentstuples unpersistablescopyscachescachedsremoteslocalslcachesversionsloginspasswords challenges logged_ins not_logged_ins cachemessagesmessagesanswerserrorsdecrefsdecachesuncachecCs$g|_t|j|_d|_||_dS)Nr)rEcopyoutgoingVocabularyoutgoingSymbolsZoutgoingSymbolCountr<)r5r<rrr__init__/szBanana.__init__cCs,t}|||j|}|j|dS)a  Send the encoded representation of the given object: @param obj: An object to encode and send. @raise BananaError: If the given object is not an instance of one of the types supported by Banana. @return: L{None} N)r_encodewritegetvaluer@)r5rA encodeStreamvaluerrrr>6s zBanana.sendEncodedcCst|ttfr\t|tkr,tdt|ftt|||t|D]}|||qFn`t|t t fr||j ks~||j krtd|f||j krt| ||tnN|dkrt| ||tn0||jkrt|||tnt|||tnt|tr"|t|td|nt|tr|jdkrd||jkrd|j|}t|||tn>t|tkrtdt|ftt|||t||ntdtt||dS)Nz#list/tuple is too long to send (%d)z"int/long is too large to send (%d)rrLr-z$byte string is too long to send (%d)z%Banana cannot send {0} objects: {1!r}) isinstancelisttuplerPrRr rrQrintr r1r4r2rVrWr3rTrUfloatr\r]Zpackbytesr9rrXrSr[r type)r5rArelemZsymbolIDrrrrGsX                    zBanana._encode)r)r rr__doc__r=r0rRZ sizeLimitr*r7r:rBrCrIrMrbr;rrYitemskvrr>rrrrrr,]sf  L'  r,r.cCst}|t_t||S)zEncode a list s-expression.)r_ir@r>r)rcrrrrencode{s rcCs0g}|jt_zt|W5dt_t`X|dS)z) Decode a banana-encoded string. rJr)rFrr;rMrb)rrHrrrdecodesr)-rZ __future__rrrr]iorZtwisted.internetrZtwisted.persistedrZtwisted.pythonrZtwisted.python.compatrr r rZtwisted.python.reflectr Exceptionr rrrQrTrSrWr\rUrVrXrOr*r(rRZProtocolZ Ephemeralr,rrCr:rrrrrrs@