File: //lib/python3/dist-packages/twisted/spread/__pycache__/banana.cpython-310.pyc
o
�b�/ � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ G d d
� d
e�Zd d � Zd
d� ZdZdZdZdZdZdZdZdZdZdd� Z da e d� dZ G dd � d ej ej �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
� N)�BytesIO)�protocol)�styles)�log)� iterbytes)�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 sN | dkr
|d� d S | dksJ d��| r%|t | d@ f�� | d? } | sd S d S )Nr � z!can only encode positive integers� � )�bytes)�integer�streamr r r
�int2b128 s �r c C s6 d}d}t | �D ]}t|�}||| 7 }|dK }q|S )z�
Convert an integer represented as a base 128 string into an L{int}.
@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}
� 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
�setPrefixLimitK s
r' �@ i
c @ s( e Zd ZdZddgZdZeZdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dZdd� Zdd� Zi dd�dd�dd�d d �d d �d d!�d"d#�d$d%�d&d'�d(d)�d*d+�d,d-�d.d/�d0d1�d2d3�d4d5�d6d7�d8d9d:d;d<d=d>d?d@dAdBdCdDdEdF��Zi Ze�� D ]\ZZeee< q|dMdGdH�ZdIdJ� ZdKdL� ZdS )N�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' r s
zBanana.setPrefixLimitc C s dS )zPSurrogate for connectionMade
Called after protocol negotiation.
Nr �r2 r r r
�connectionReady~ s zBanana.connectionReadyc C s || _ | �� d S �N)�currentDialectr4 )r2 �dialectr r r
�_selectDialect� s zBanana._selectDialectc C s� | j r
| �|� d S | jr0|D ]}|| jv r#| �|� | �|� d S qt�d� | j� � d S || jv r<| �|� d S t�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.)
r6 �expressionReceived�isClient�
knownDialects� sendEncodedr8 r �msg� transport�loseConnection)r2 �obj� serverVerr r r
�callExpressionReceived� s&
��
�z Banana.callExpressionReceivedc C s* | � t� d | _| js| �| j� d S d S r5 )r' r% r6 r: r< r; r3 r r r
�connectionMade� s
�zBanana.connectionMadec C s. | j }|r|d d �|� d S | �|� d S )N���r )� listStack�appendrB )r2 �item�lr r r
�gotItem� s zBanana.gotItem� c C s� | j | }| j}| j}|�re| j |ks J d�t| j �t|����|| _ d}t|�D ] }|tkr1 n|d }q)|| jkrCtd| jf ��d S |d |� }|||d � }||d d � } t |�| jkrjtd| jf ��|t
kr�t |�}|t krztd��|�
|g f� | }n�|tkr�t |�}|t kr�td��t | �|kr�| |d � }|| d |� � n�d S |tkr�| }t |�}||� n{|tkr�| }t |�}||� nl|tkr�| }t |�}|| � n\|tkr�| }t |� }||� nL|tk�r | }t |�}| j| }
| jdk�r||
� n1td |
����|tk�r.t | �d
k�r,| d
d � }|t�d | d d
� �d � n d S td |����|�rct |d
d �|d
d k�rc|�� d }
||
� |�rct |d
d �|d
d k�sG|sd| _ d S )NzThis ain't right: {} {}r 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 � �!dzInvalid Type Byte rD rJ )�bufferrE rI �format�reprr � HIGH_BIT_SETr- r �len�LISTr �
SIZE_LIMITrF �STRING�INT�LONGINT�LONGNEG�NEG�VOCAB�incomingVocabularyr6 �NotImplementedError�FLOAT�struct�unpack�pop) r2 �chunkrM rE rI �pos�ch�num�typebyte�restrG r r r
� dataReceived� s�
�
�� ��
$ $��
MzBanana.dataReceivedc C s t � �)z=Called when an expression (list, string, or int) is received.)r[ )r2 �lstr r r
r9 s zBanana.expressionReceiveds Noner s classr, s dereference� s reference� s
dictionary� s function� s instancer s listrK s module� s
persistent�
s tuple� s
unpersistable� s copy�
s cache� s cached� s remote� s local� � � � � � � � � � � � � � � )s 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�outgoingSymbols�outgoingSymbolCountr: )r2 r: r r r
�__init__4 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> )r2 r@ � encodeStream�valuer r r
r< : s zBanana.sendEncodedc C s� t |ttf�r.t|�tkrtdt|�f ��tt|�|� |t� |D ]}| �||� q#d S t |t �r�|| j
k s=|| j krDtd|f ��|| j k rUt| |� |t
� d S |dk ret| |� |t� d S || jkrut||� |t� d S t||� |t� d S t |t�r�|t� |t�d|�� d S t |t�r�| jdkr�|| jv r�| j| }t||� |t� d S t|�tkr�tdt|�f ��tt|�|� |t� ||� d S td�t t |��|���)Nz#list/tuple is too long to send (%d)z int is too large to send (%d)r rL r* z$byte string is too long to send (%d)z#Banana cannot send {} objects: {!r}) �
isinstance�list�tuplerQ rS r r rR r� �intr. r1 r/ rW rX r0 rU rV �floatr\ r] �packr r6 r� rY rT rN r �type)r2 r@ r� �elem�symbolIDr r r
r� J sT �
� ��zBanana._encode)r )r r
r �__doc__r; r- rS � sizeLimitr' r4 r8 rB rC rI rM rf r9 r� rZ �items�k�vr� r< r� r r r r
r) a s� S������� �
� � �
�������%
r) r+ c C s t � }|t_t�| � |�� S )zEncode a list s-expression.)r �_ir> r< r� )rg r� r r r
�encode� s
r� c C s8 g }|j t_zt�| � W dt_t`|d S dt_t`w )z)
Decode a banana-encoded string.
rJ r )rF r� r9 rf rM )r rH r r r
�decode� s �r� )'r� r� r] �ior �twisted.internetr �twisted.persistedr �twisted.pythonr �twisted.python.compatr �twisted.python.reflectr � Exceptionr r r rR rU rT rX r\ rV rW rY rP r' r% rS �Protocol� Ephemeralr) r� rC r8 r� r� r r r r
�<module> sB