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 deZdd ZG d
d d ejZed krej 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 ddZd ddZdd 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 dt }x6t||k rL| jj|t| }|sBt ||7 }qW |jdS )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 bytearraylensZrecv
_SocketEOFdecode)selfldatar r "/usr/lib64/python3.6/_transport.py _read_all- s zTransPort._read_allc C sP |dkst |dk r tdtjtt || j| }| jjt|jd dS )z]
Sends the json formatted message by pre-appending the length
first.
Nr zMsg argument emptyzutf-8) r r
strzfillHDR_LENr
Zsendallbytesencode)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 intsocketerrorr 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 dn t
t j
dW n" t jk
r| t
t jdY 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_STREAMospathexistsaccessR_OKW_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 )methodidparams)clsz,Error while sending a message to the plug-inN)
jsondumps _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|tdS 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 \}}|dks$t|S )z;
Sends a request and waits for a response.
r/ )r8 read_respAssertionError)r r0 r7 replymsg_idr r r rpc s z
TransPort.rpcNc C s* ||||dd}| j tj|td dS )z,
Used to transmit an error.
)codemessager )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 rD 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 rD 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@ rC rE 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 errorcodeerrormsgN)r r; rC rE r. )r
Zsrvr r r r _server s
rO 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 threadingZThreadrO serverstart)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 ) N0 z z{}:""zSome text messageZDEADBEEFZtestr/ )rS r8 r<
assertTrue)r Ztctr> 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 )rM rN )rS r8 Z assertRaisesr r< rZ rA r )r Ze_msgZe_coder" r r r test_exceptions s
z _TestTransport.test_exceptionsc C s xt dddD ]}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 rZ rR sendrS r< ) r r Zpayloadr r Zwireir> r? r r r test_slow s
z_TestTransport.test_slowc C s8 | j jdd | j j \}}| j|d k | jj d S )NrL )rS r8 r< rZ rU join)r rF r? r r r tearDown s z_TestTransport.tearDownN)rG rH rI rW r\ r] rc re r r r r rP s
rP __main__)r4 r r` r' ZunittestrT Z lsm._commonr r r r Z lsm._datar r: r r6 objectr rO ZTestCaserP rG mainr r r r <module> s >
All system for education purposes only. For more tools: Telegram @jackleet