관리-도구
편집 파일: _transport.cpython-39.opt-1.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 | � ||� | �� \}}|S )z; Sends a request and waits for a response. )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