Your IP : 216.73.216.213
B
:��].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 �H/opt/alt/python37/lib64/python3.7/site-packages/twisted/spread/banana.pyr s r c C sN | dkr|t d�� d S | dks(td��x | rH|t | d@ �� | d? } q*W d S )Nr z!can only encode positive integers� � )�chr�AssertionError)�integer�streamr r r �int2b128 s r c C s: d}d}x,t | �D ] }t|�}||| 7 }|dK }qW |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 Zx e�� D ]\ZZeee< q�W 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}
s pbs 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 )r4 r r r �connectionReady{ s zBanana.connectionReadyc C s || _ | �� d S )N)�currentDialectr5 )r4 Zdialectr r r �_selectDialect� s zBanana._selectDialectc C s� | j r| �|� nt| jr\xl|D ]$}|| jkr | �|� | �|� P q W t�d� | j� � n*|| jkrr| �|� 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.)
r6 �expressionReceived�isClient�
knownDialects� sendEncodedr7 r �msg� transportZloseConnection)r4 �objZ serverVerr r r �callExpressionReceived� s
z Banana.callExpressionReceivedc C s& | � t� d | _| js"| �| j� d S )N)r+ r) r6 r9 r; r: )r4 r r r �connectionMade� s
zBanana.connectionMadec C s, | j }|r |d d �|� n
| �|� d S )N���r )� listStack�appendr? )r4 �item�lr r r �gotItem� s zBanana.gotItem� c C s� | j | }| j}| j}�x�|�r�| j |ksDtdt| j �t|�f ��|| _ d}x@t|�D ]}|tkrfP |d }qXW || jkr�td| jf ��d S |d |� }|||d � }||d d � } t |�| jkr�td| jf ��|t
k�rt |�}|t kr�td��|�
|g f� | }�nz|tk�rjt |�}|t k�r6td��t | �|k�rb| |d � }|| d |� � nd S �n"|tk�r�| }t |�}||� �n |tk�r�| }t |�}||� n�|tk�r�| }t |�}|| � n�|tk�r�| }t |� }||� n�|tk�r6| }t |�}| j| }
| jdk�r&||
� ntd �|
���nV|tk�r~t | �d
k�rx| d
d � }|t�d | d d
� �d � nd S ntd |f ��x>|�r�t |d
d �|d
d k�r�|�� d }
||
� �q�W qW d| _ 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.s pbz"Invalid item for pb protocol {0!r}� z!dzInvalid Type Byte %rrA rG ) �bufferrB rF r �reprr � HIGH_BIT_SETr/ r �len�LISTr �
SIZE_LIMITrC �STRING�INT�LONGINT�LONGNEG�NEG�VOCAB�incomingVocabularyr6 �NotImplementedError�format�FLOAT�struct�unpack�pop) r4 �chunkrI rB rF �pos�ch�numZtypebyte�restrD r r r � dataReceived� s�
$
& zBanana.dataReceivedc C s
t � �dS )zFCalled when an expression (list, string, or int) is received.
N)rV )r4 �lstr r r r8 � s zBanana.expressionReceivedr r. � � � � r rH � �
� � �
� � � � � � � � � � � � � � � � � � ) 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 )rB �copy�outgoingVocabulary�outgoingSymbolsZoutgoingSymbolCountr9 )r4 r9 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= )r4 r> � encodeStream�valuer r r r; 6 s zBanana.sendEncodedc C s� t |ttf�r`t|�tkr,tdt|�f ��tt|�|� |t� x|D ]}| �||� qHW �nbt |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�rj|| jk�rj| 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 z!ds pbz$byte string is too long to send (%d)z%Banana cannot send {0} objects: {1!r}) �
isinstance�list�tuplerL rN r r rM r� �intr r0 r3 r1 rR rS r2 rP rQ �floatrX rY �pack�bytesr6 r� rT rO rW r �type)r4 r> r� �elemZsymbolIDr r r r� G sN
zBanana._encode)r )r
r r �__doc__r: r/ rN Z sizeLimitr+ r5 r7 r? r@ rF rI ra r8 r rU �items�k�vr� r; r� r r r r r- ] sd
L
r- s nonec C s t � }|t_t�| � |�� S )zEncode a list s-expression.)r �_ir= r; r� )rb r� r r r �encode{ s
r� c C s0 g }|j t_zt�| � W ddt_t`X |d S )z)
Decode a banana-encoded string.
NrG r )rC r� r8 ra rI )r rE r r r �decode� s r� )-r� �
__future__r r r~ rY �ior Ztwisted.internetr Ztwisted.persistedr Ztwisted.pythonr Ztwisted.python.compatr r r
r Ztwisted.python.reflectr � Exceptionr r r rM rP rO rS rX rQ rR rT rK r+ r) rN ZProtocolZ Ephemeralr- r� r@ r7 r� r� r r r r �<module>
s@