File: //lib64/python3.9/site-packages/lsm/__pycache__/_transport.cpython-39.pyc
a
V��d� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dlmZ d dl
m Z d dl
m
Z G dd� de�Zdd � ZG d
d � d ej�Zed kr�e�� dS )
� N)�LsmError� ErrorNumber)� SocketEOF)� DataDecoder)� DataEncoderc @ s| e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Ze d d � �Z
d
d� Z dd� Z dd� Z
dd� Zd dd�Zd dd�Zdd � ZdS ) � TransPorta�
Provides wire serialization by using json. Loosely conforms to json-rpc,
however a length header was added so that we would have the ability to use
non sax like json parsers, which are more abundant.
<Zero padded 10 digit number [1..2**32] for the length followed by
valid json.
Notes:
id field (json-rpc) is present but currently not being used.
This is available to be expanded on later.
�
c C sT |dk rt d��t� }t|�|k rJ| j�|t|� �}|s@t� �||7 }q|�d�S )z�
Reads l number of bytes before returning. Will raise a SocketEOF
if socket returns zero bytes (i.e. socket no longer connected)
� z Trying to read less than 1 byte!�utf-8)�
ValueError� bytearray�len�sZrecv�
_SocketEOF�decode)�self�l�data�r� r �4/usr/lib64/python3.9/site-packages/lsm/_transport.py� _read_all# s
zTransPort._read_allc C sP |du st |�dk r td��t�tt |��| j�| }| j�t|�d��� dS )z]
Sends the json formatted message by pre-appending the length
first.
Nr zMsg argument emptyr
) r
r �str�zfill�HDR_LENr Zsendall�bytes�encode)r �msgr r r r � _send_msg5 s zTransPort._send_msgc
C sx z | � | j�}| � t|��}W nT tjyV } z ttjdt|���W Y d}~n&d}~0 t yr ttjd��Y n0 |S )zk
Reads header first to get the length and then the remaining
bytes of the message.
z.Error while reading a message from the plug-inNz3Error while reading a message from the plug-in, EOF)
r r �int�socket�errorr r �TRANSPORT_COMMUNICATIONr r )r Z num_bytesr �er r r � _recv_msgC s � �
zTransPort._recv_msgc C s
|| _ d S )N)r )r Zsocket_descriptorr r r �__init__V s zTransPort.__init__c C s� zZt � t jt j�}tj�| �rLt�| tjtjB �r>|� | � qXt
t j d��n t
t j
d��W n t jyz t
t jd��Y n0 |S )zE
Returns a connected socket from the passed in path.
z-Permissions are incorrect for IPC socket filez Plug-in appears to not existz*Unable to connect to lsmd, daemon started?)r �AF_UNIX� SOCK_STREAM�os�path�exists�access�R_OK�W_OKZconnectr r ZPLUGIN_SOCKET_PERMISSIONZPLUGIN_NOT_EXISTr! ZPLUGIN_IPC_FAIL)r) r r r r �
get_socketY s" ���
zTransPort.get_socketc C s | j �� dS )z@
Closes the transport and the underlying socket
N)r �close�r r r r r/ r s zTransPort.closec
C sf z(|d|d�}t j|td�}| �|� W n8 tjy` } z ttjdt |���W Y d}~n
d}~0 0 dS )z�
Sends a request given a method and arguments.
Note: arguments must be in the form that can be automatically
serialized to json
�d ��method�id�params��clsz,Error while sending a message to the plug-inN)
�json�dumps� _DataEncoderr r r! r r r" r )r r3 �argsr r Zser r r �send_reqx s �zTransPort.send_reqc C s" | � � }t|�r tj|td�S dS )zG
Reads a message and returns the parsed version of it.
r6 N)r$ r
r8 �loads� _DataDecoder)r r r r r �read_req� s zTransPort.read_reqc C s( | � ||� | �� \}}|dks$J �|S )z;
Sends a request and waits for a response.
r1 )r< � read_resp)r r3 r; �reply�msg_idr r r �rpc� s z
TransPort.rpcNc C s* ||||d�d�}| � tj|td�� dS )z,
Used to transmit an error.
)�code�messager )r4 r! r6 N�r r8 r9 r: )r rB Z
error_coder r r# r r r �
send_error� s ��zTransPort.send_errorr1 c C s" ||d�}| � tj|td�� dS )z-
Used to transmit a response
)r4 �resultr6 NrF )r rH rB r r r r � send_resp� s
zTransPort.send_respc C sH | � � }tj|td�}d|v r.|d |d fS |d }tf i |���d S )Nr6 rH r4 r! )r$ r8 r= r> r )r r �respr# r r r r@ � s zTransPort.read_resp)N)r1 )�__name__�
__module__� __qualname__�__doc__r r r r$ r% � staticmethodr. r/ r< r? rC rG rI r@ r r r r r s
r c C s� t | �}|�� }zn|d dkrf|d dkrN|�|d |d d |d d � n|�|d � |�� }q|�|d � W | �� n
| �� 0 dS ) z)
Test echo server for test case.
r3 �doner! r4 r5 � errorcode�errormsgN)r r? rG rI r/ )r Zsrvr r r r �_server� s
�
rS c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �_TestTransportc C sF t �t jt j�\| _| _t| j�| _tj t
| jfd�| _ | j � � d S )N)�targetr; )
r Z
socketpairr& r' �cr r �client� threading�ThreadrS �server�startr0 r r r �setUp� s � z_TestTransport.setUpc C sN g d�}|D ]<}| j �d|� | j �� \}}| �|dk� | �||k� q d S )N)�0� z z{}:""zSome text messageZDEADBEEF�testr1 )rW r< r@ �
assertTrue)r Ztc�trA rB r r r � test_simple� s z_TestTransport.test_simplec
C s� d}d}| j �d||d�� | �t| j j� z"| j �d||d�� | j �� W nD ty� } z,| �|j|k� | �|j|k� W Y d }~n
d }~0 0 d S )NzTest error messager1 r! )rQ rR )rW r< Z assertRaisesr r@ r` rD r )r Ze_msgZe_coder# r r r �test_exceptions� s �z _TestTransport.test_exceptionsc
C s� t ddd�D ]�}d| }dd|d�}tj|td�}tt|��}|�tj�| }| � t|�dk� |D ]}| j
� t |d �� qd| j
�� \}} | � ||k� q d S )
Nr i r �xZdripr1 r2 r6 r
)�ranger8 r9 r: r r
r r r r` rV �sendr rW r@ )
r r Zpayloadr r ZhdrZwire�irA rB r r r � test_slow� s z_TestTransport.test_slowc C sB | j �dd � | j �� \}}| �|d u � | j�� | j �� d S )NrP )rW r< r@ r` rZ �joinr/ )r rJ rB r r r �tearDown
s
z_TestTransport.tearDownN)rK rL rM r\ rb rc rh rj r r r r rT � s
rT �__main__)r8 r r( ZunittestrX Z lsm._commonr r r r Z lsm._datar r> r r: �objectr rS ZTestCaserT rK �mainr r r r �<module> s 'C