관리-도구
편집 파일: _transport.pyc
� -�P`c @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z d d l m Z d d l m Z d d l m Z d e f d � � YZ d � Z d e j f d � � YZ e d k r� e j � n d S( i����N( t LsmErrort ErrorNumber( t SocketEOF( t DataDecoder( t DataEncodert TransPortc B s� e Z d Z d Z d � Z d � Z d � Z d � Z e d � � Z d � Z d � Z d � Z d � Z d d � Z d d � Z d � Z RS( s� 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. i c C s� | d k r t d � � n t � } xN t | � | k rt | j j | t | � � } | sg t � � n | | 7} q' W| j d � S( s� Reads l number of bytes before returning. Will raise a SocketEOF if socket returns zero bytes (i.e. socket no longer connected) i s Trying to read less than 1 byte!s utf-8( t ValueErrort bytearrayt lent st recvt _SocketEOFt decode( t selft lt datat r( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt _read_all- s c C su | d k s t | � d k r- t d � � n t j t t | � � | j � | } | j j t | j d � � � d S( s] Sends the json formatted message by pre-appending the length first. i s Msg argument emptys utf-8N( t NoneR R t strt zfillt HDR_LENR t sendallt bytest encode( R t msgR ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt _send_msg? s %c C sc y+ | j | j � } | j t | � � } Wn1 t j k r^ } t t j d t | � � � n X| S( sk Reads header first to get the length and then the remaining bytes of the message. s. Error while reading a message from the plug-in( R R t intt sockett errorR R t TRANSPORT_COMMUNICATIONR ( R R R t e( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt _recv_msgM s c C s | | _ d S( N( R ( R t socket_descriptor( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt __init__\ s c C s� y� t j t j t j � } t j j | � rn t j | t j t j B� rY | j | � q� t t j d � � n t t j d � � Wn& t j k r� t t j d � � n X| S( sE Returns a connected socket from the passed in path. s- Permissions are incorrect for IPC socket files Plug-in appears to not exists* Unable to connect to lsmd, daemon started?( R t AF_UNIXt SOCK_STREAMt ost patht existst accesst R_OKt W_OKt connectR R t PLUGIN_SOCKET_PERMISSIONt PLUGIN_NOT_EXISTR t PLUGIN_IPC_FAIL( R& R ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt get_socket_ s c C s | j j � d S( s@ Closes the transport and the underlying socket N( R t close( R ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyR0 w s c C sy yA i | d 6d d 6| d 6} t j | d t �} | j | � Wn1 t j k rt } t t j d t | � � � n Xd S( s� Sends a request given a method and arguments. Note: arguments must be in the form that can be automatically serialized to json t methodid t idt paramst clss, Error while sending a message to the plug-inN( t jsont dumpst _DataEncoderR R R R R R R ( R R1 t argsR R t se( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt send_req} s c C s/ | j � } t | � r+ t j | d t �Sd S( sG Reads a message and returns the parsed version of it. R4 N( R R R5 t loadst _DataDecoder( R R ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt read_req� s c C s8 | j | | � | j � \ } } | d k s4 t � | S( s; Sends a request and waits for a response. id ( R: t read_respt AssertionError( R R1 R8 t replyt msg_id( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt rpc� s c C sI i | d 6i | d 6| d 6| d 6d 6} | j t j | d t �� d S( s, Used to transmit an error. R2 t codet messageR R R4 N( R R5 R6 R7 ( R RA t error_codeR R R ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt send_error� s id c C s4 i | d 6| d 6} | j t j | d t �� d S( s- Used to transmit a response R2 t resultR4 N( R R5 R6 R7 ( R RG RA R ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt send_resp� s c C sY | j � } t j | d t �} d | k r? | d | d f S| d } t | � � d S( NR4 RG R2 R ( R R5 R; R< R ( R R t respR ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyR> � s N( t __name__t __module__t __doc__R R R R R" t staticmethodR/ R0 R: R= RB R RF RH R> ( ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyR s c C s� t | � } | j � } z� xk | d d k r� | d d k rh | j | d | d d | d d � n | j | d � | j � } q W| j | d � Wd | j � Xd S( s) Test echo server for test case. R1 t doneR R2 R3 t errorcodet errormsgN( R R= RF RH R0 ( R t srvR ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt _server� s t _TestTransportc B s5 e Z d � Z d � Z d � Z d � Z d � Z RS( c C sh t j t j t j � \ | _ | _ t | j � | _ t j d t d | j f � | _ | j j � d S( Nt targetR8 ( R t socketpairR# R$ t cR R t clientt threadingt ThreadRR t servert start( R ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt setUp� s !c C s{ d d d d d d g } x\ | D]T } | j j d | � | j j � \ } } | j | d k � | j | | k � q Wd S( Nt 0t s s {}:""s Some text messaget DEADBEEFt testid ( RW R: R> t assertTrue( R t tct tR@ RA ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt test_simple� s c C s� d } d } | j j d i | d 6| d 6� | j t | j j � y2 | j j d i | d 6| d 6� | j j � Wn? t k r� } | j | j | k � | j | j | k � n Xd S( Ns Test error messageid R RO RP ( RW R: t assertRaisesR R> Ra RC R ( R t e_msgt e_codeR ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt test_exceptions� s ! c C s� x� t d d d � D]� } d | } i d d 6d d 6| d 6} t j | d t �} t j t | � t j � | } | j t | � d k � x | D] } | j j | � q� W| j j � \ } } | j | | k � q Wd S( Ni i i t xt dripR1 id R2 R3 R4 ( t rangeR5 R6 R7 t stringR R R R Ra RV t sendRW R> ( R R t payloadR R t wiret iR@ RA ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt test_slow� s c C sL | j j d d � | j j � \ } } | j | d k � | j j � d S( NRN ( RW R: R R> Ra RZ t join( R RI RA ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt tearDown s ( RJ RK R\ Rd Rh Rq Rs ( ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyRS � s t __main__( R5 R Rl R% t unittestRX t lsm._commonR R R R t lsm._dataR R< R R7 t objectR RR t TestCaseRS RJ t main( ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt <module> s � >