EVOLUTION-MANAGER
Edit File: _transport.cpython-36.pyc
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� \}}|dks$t�|S )z; Sends a request and waits for a response. r/ )r8 � read_resp�AssertionError)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 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 � errorcode�errormsgN)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 �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/ )rS 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 )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� 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 rZ rR �sendrS 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 )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 >