| Current File : //usr/lib/python2.7/site-packages/lsm/_transport.pyo |
�
-�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 s& | j | | � | j � \ } } | S( s;
Sends a request and waits for a response.
( R: t read_resp( 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
R@ 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
RF R@ 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 RF 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= RA R RE RG 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= RE RG 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 ThreadRQ 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 ( RV R: R> t
assertTrue( R
t tct tR? R@ ( ( 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 RN RO ( RV R: t assertRaisesR R> R` RB 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 R` RU t sendRV R> ( R
R t payloadR R t wiret iR? R@ ( ( 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( NRM ( RV R: R R> R` RY t join( R
RH R@ ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt tearDown s ( RI RJ R[ Rc Rg Rp Rr ( ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyRR � s
t __main__( R5 R Rk R% t unittestRW t lsm._commonR R R R t lsm._dataR R< R R7 t objectR RQ t TestCaseRR RI t main( ( ( s2 /usr/lib/python2.7/site-packages/lsm/_transport.pyt <module> s � >