File: //usr/lib/python3/dist-packages/twisted/spread/__pycache__/banana.cpython-38.pyc
U
W[.0 � @ s6 d Z ddlmZmZ ddlZddlZddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZmZmZ dd lmZ G d
d � d e�Zd d
� Zdd� Zed�Zed�Zed�Zed�Z ed�Z ed�Z ed�Z ed�Z ed�Z!dd� Z"da#e"d� dZ$G d d � d e j%e j&�Z'e'� Z(e(�)� e(�*d � d d � 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_import�divisionN)�BytesIO)�protocol)�styles)�log)� iterbytes�long� _bytesChr)�fullyQualifiedNamec @ s e Zd ZdS )� BananaErrorN)�__name__�
__module__� __qualname__� r r �7/usr/lib/python3/dist-packages/twisted/spread/banana.pyr s r c C sJ | dkr|t d�� d S | dks(td��| rF|t | d@ �� | d? } q(d S )Nr z!can only encode positive integers� � )�chr�AssertionError)Zinteger�streamr r r �int2b128 s r c C s6 d}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}
� r r )r �ord)�st�e�i�char�nr r r �b1282int' s
r � � � � � � � � c C s | a dS )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)�limitr r r �setPrefixLimitI s
r* �@ i
c @ s� e Zd ZdZddgZdZeZdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dZdd� Zdd� Zddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5� Zi Ze�� D ]\ZZeee< q�d<d6d7�Zd8d9� Zd:d;� ZdS )=�Bananaz�
L{Banana} implements the I{Banana} s-expression protocol, client and
server.
@ivar knownDialects: These are the profiles supported by this Banana
implementation.
@type knownDialects: L{list} of L{bytes}
� pb� noneNc C s< || _ d|d d | _d| _d| _d|d d | _dS )zt
Set the prefix limit for decoding done by this protocol instance.
@see: L{setPrefixLimit}
� r r i �i���N)� prefixLimit�_smallestLongInt� _smallestInt� _largestInt�_largestLongInt)�selfr) r r r r* n s
zBanana.setPrefixLimitc C s dS )zPSurrogate for connectionMade
Called after protocol negotiation.
Nr �r5 r r r �connectionReady{ s zBanana.connectionReadyc C s || _ | �� d S �N)�currentDialectr7 )r5 Zdialectr r r �_selectDialect� s zBanana._selectDialectc C s� | j r| �|� nr| jrZ|D ]&}|| jkr | �|� | �|� q�q t�d� | j� � n*|| jkrp| �|� nt�d� | j� � d S )NzSThe client doesn't speak any of the protocols offered by the server: disconnecting.zYThe client selected a protocol the server didn't suggest and doesn't know: disconnecting.)
r9 �expressionReceived�isClient�
knownDialects� sendEncodedr: r �msg� transportZloseConnection)r5 �objZ serverVerr r r �callExpressionReceived� s
z Banana.callExpressionReceivedc C s& | � t� d | _| js"| �| j� d S r8 )r* r( r9 r< r> r= r6 r r r �connectionMade� s
zBanana.connectionMadec C s, | j }|r |d d �|� n
| �|� d S )N���r )� listStack�appendrB )r5 �item�lr r r �gotItem� s zBanana.gotItem� c C s� | j | }| j}| j}|�r�| j |ks@tdt| j �t|�f ��|| _ d}t|�D ]}|tkrb q�|d }qR|| jkr�td| jf ��d S |d |� }|||d � }||d d � } t |�| jkr�td| jf ��|t
k�r t |�}|t kr�td��|�
|g f� | }�nz|tk�rdt |�}|t k�r0td��t | �|k�r\| |d � }|| d |� � nd S �n"|tk�r�| }t |�}||� �n |tk�r�| }t |�}||� n�|tk�r�| }t |�}|| � n�|tk�r�| }t |� }||� n�|tk�r0| }t |�}| j| }
| jdk�r ||
� ntd �|
���nV|tk�rxt | �d
k�rr| d
d � }|t�d | d d
� �d � nd S ntd |f ��|rt |d
d �|d
d kr|�� d }
||
� �q�qd| _ d S )NzThis ain't right: %s %sr r z1Security 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 %rrD rJ ) �bufferrE rI r �reprr � HIGH_BIT_SETr0 r �len�LISTr �
SIZE_LIMITrF �STRING�INT�LONGINT�LONGNEG�NEG�VOCAB�incomingVocabularyr9 �NotImplementedError�format�FLOAT�structZunpack�pop) r5 �chunkrM rE rI �posZchZnumZtypebyte�restrG r r r � dataReceived� s�
$
�
zBanana.dataReceivedc C s
t � �dS )zFCalled when an expression (list, string, or int) is received.
N)rZ )r5 �lstr r r r; � s zBanana.expressionReceivedr r/ � � � � r rK � �
� � �
� � � � � � � � � � � � � � � � � � ) s Nones classs dereferences references
dictionarys functions instances lists modules
persistents tuples
unpersistables copys caches cacheds remotes locals lcaches versions logins passwords challenges logged_ins
not_logged_ins cachemessages messages answers errors decrefs decaches uncachec C s$ g | _ t�| j�| _d| _|| _d S )Nr )rE �copy�outgoingVocabulary�outgoingSymbolsZoutgoingSymbolCountr< )r5 r< r r r �__init__/ s zBanana.__init__c C s, 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 �_encode�write�getvaluer@ )r5 rA � encodeStream�valuer r r r> 6 s zBanana.sendEncodedc C s� t |ttf�r\t|�tkr,tdt|�f ��tt|�|� |t� |D ]}| �||� qF�n`t |t t
f�r�|| j k s~|| j kr�td|f ��|| j
k r�t| |� |t� nN|dk r�t| |� |t� n0|| jkr�t||� |t� nt||� |t� n�t |t��r"|t� |t�d|�� n�t |t��r�| jdk�rd|| jk�rd| j| }t||� |t� n>t|�tk�r�tdt|�f ��tt|�|� |t� ||� ntd� t t |��|���d S )Nz#list/tuple is too long to send (%d)z"int/long is too large to send (%d)r rL r- z$byte string is too long to send (%d)z%Banana cannot send {0} objects: {1!r}) �
isinstance�list�tuplerP rR r r rQ r� �intr r1 r4 r2 rV rW r3 rT rU �floatr\ r] Zpack�bytesr9 r� rX rS r[ r �type)r5 rA r� �elemZsymbolIDr r r r� G sX ��
�
�zBanana._encode)r )r
r r �__doc__r= r0 rR Z sizeLimitr* r7 r: rB rC rI rM rb r; r� rY �items�k�vr� r> r� r r r r r, ] sf
L�'
r, r. c C s t � }|t_t�| � |�� S )zEncode a list s-expression.)r �_ir@ r> r� )rc r� r r r �encode{ s
r� c C s0 g }|j t_zt�| � W 5 dt_t`X |d S )z)
Decode a banana-encoded string.
rJ r )rF r� r; rM rb )r rH r r r �decode� s r� )-r� Z
__future__r r r r] �ior Ztwisted.internetr Ztwisted.persistedr Ztwisted.pythonr Ztwisted.python.compatr r r
r Ztwisted.python.reflectr � Exceptionr r r rQ rT rS rW r\ rU rV rX rO r* r( rR ZProtocolZ Ephemeralr, r� rC r: r� r� r r r r �<module> s@