Sindbad~EG File Manager
3
�.'e� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dlm Z
d dl m Z
d dl mZ G dd� de�Zdd � ZG d
d � d ej�Zed kr�ej� 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 sX |dk rt d��t� }x6t|�|k rL| jj|t|� �}|sBt� �||7 }qW |jd�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!zutf-8)�
ValueError� bytearray�len�sZrecv�
_SocketEOF�decode)�self�l�data�r� r �"/usr/lib64/python3.6/_transport.py� _read_all- s zTransPort._read_allc C sP |dkst |�dk r td��tjtt |��| j�| }| jjt|jd��� dS )z]
Sends the json formatted message by pre-appending the length
first.
Nr zMsg argument emptyzutf-8) r r
�str�zfill�HDR_LENr
Zsendall�bytes�encode)r �msgr
r r r � _send_msg? s zTransPort._send_msgc C sZ y | j | j�}| j t|��}W n6 tjk
rT } zttjdt|���W Y dd}~X nX |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-inN) r r �int�socket�errorr r �TRANSPORT_COMMUNICATIONr )r r r �er r r � _recv_msgM s zTransPort._recv_msgc C s
|| _ d S )N)r
)r Zsocket_descriptorr r r �__init__\ s zTransPort.__init__c C s� yZt j t jt j�}tjj| �rLtj| tjtjB �r>|j | � qXt
t j d��n t
t j
d��W n" t jk
r| t
t jd��Y nX |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_socket_ s
zTransPort.get_socketc C s | j j� dS )z@
Closes the transport and the underlying socket
N)r
�close)r r r r r. w s zTransPort.closec C sd y(|d|d�}t j|td�}| j|� W n6 tjk
r^ } zttjdt |���W Y dd}~X nX 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 r0 �argsr r Zser r r �send_req} s zTransPort.send_reqc C s" | j � }t|�r tj|td�S dS )zG
Reads a message and returns the parsed version of it.
)r3 N)r# r r4 �loads� _DataDecoder)r r r r r �read_req� s zTransPort.read_reqc C s | j ||� | j� \}}|S )z;
Sends a request and waits for a response.
)r8 � read_resp)r r0 r7 �reply�msg_idr r r �rpc� s z
TransPort.rpcNc C s* ||||d�d�}| j tj|td�� dS )z,
Used to transmit an error.
)�code�messager )r1 r )r3 N)r r4 r5 r6 )r r> Z
error_coder r r" r r r �
send_error� s zTransPort.send_errorr/ c C s" ||d�}| j tj|td�� dS )z-
Used to transmit a response
)r1 �result)r3 N)r r4 r5 r6 )r rC r> r r r r � send_resp� s
zTransPort.send_respc C sD | j � }tj|td�}d|kr.|d |d fS |d }tf |��d S )N)r3 rC r1 r )r# r4 r9 r: r )r r �respr" r r r r< � s zTransPort.read_resp)N)r/ )�__name__�
__module__� __qualname__�__doc__r r r r# r$ � staticmethodr- r. r8 r; r? rB rD r< r r r r r s
r c C s� t | �}|j� }zjxV|d dkrh|d dkrP|j|d |d d |d d � n|j|d � |j� }qW |j|d � W d| j� X dS ) z)
Test echo server for test case.
r0 �doner r1 r2 � errorcode�errormsgN)r r; rB rD r. )r
Zsrvr r r r �_server� s
rN c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �_TestTransportc C sF t jt jt j�\| _| _t| j�| _tj t
| jfd�| _ | j j � d S )N)�targetr7 )
r Z
socketpairr% r& �cr
r �client� threadingZThreadrN �server�start)r r r r �setUp� s
z_TestTransport.setUpc C sZ ddddddg}xD|D ]<}| j jd|� | j j� \}}| j|dk� | j||k� qW d S ) N�0� z z{}:""zSome text messageZDEADBEEFZtestr/ )rR r8 r< �
assertTrue)r Ztc�tr= r> r r r � test_simple� s
z_TestTransport.test_simplec C s� d}d}| j jd||d�� | jt| j j� y"| j jd||d�� | j j� W nB tk
r� } z&| j|j|k� | j|j|k� W Y d d }~X nX d S )NzTest error messager/ r )rL rM )rR r8 Z assertRaisesr r< rY r@ r )r Ze_msgZe_coder" r r r �test_exceptions� s
z _TestTransport.test_exceptionsc C s� x�t ddd�D ]�}d| }dd|d�}tj|td�}tjt|�tj�| }| j t|�dk� x|D ]}| j
j |� qbW | j j
� \}}| j ||k� qW d S ) Nr i r �xZdripr/ )r0 r1 r2 )r3 )�ranger4 r5 r6 �stringr r r r rY rQ �sendrR r< ) r r Zpayloadr r Zwire�ir= r> r r r � test_slow� s
z_TestTransport.test_slowc C s8 | j jdd � | j j� \}}| j|d k� | jj� d S )NrK )rR r8 r< rY rT �join)r rE r> r r r �tearDown s z_TestTransport.tearDownN)rF rG rH rV r[ r\ rb rd r r r r rO � s
rO �__main__)r4 r r_ r' ZunittestrS Z lsm._commonr r r r Z lsm._datar r: r r6 �objectr rN ZTestCaserO rF �mainr r r r �<module> s >
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists