ó ½-'Nc@sÞdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZmZdefd „ƒYZd efd „ƒYZd efd „ƒYZdS(sLow-level server communication.iÿÿÿÿN(tfetch(t create_file(tbpickle(t format_delta(t SERVER_APItVERSIONt HTTPTransportcBsGeZdZddd„Zd„Zd„Zd„Zded„Z RS(s=Transport makes a request to exchange message data over HTTP.cCs||_||_||_dS(s- @param url: URL of the remote Landscape server message system. @param pubkey: SSH public key used for secure communication. @param payload_recorder: PayloadRecorder used for recording exchanges with the server. If C{None}, exchanges will not be recorded. N(t_urlt_pubkeyt_payload_recorder(tselfturltpubkeytpayload_recorder((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyt__init__s  cCs|jS(s)Get the URL of the remote message system.(R(R ((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pytget_urlscCs ||_dS(s)Set the URL of the remote message system.N(R(R R ((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pytset_url"sc Csri|d6dtd6dd6}|r2||d/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyt_curl&s     c Cs„tj|ƒ}|jdk r1|jj|ƒny”tjƒ}tjƒjƒtj krwtj dt j |ƒƒn|j |||ƒ\}}tjdt|ƒt|ƒttjƒ|ƒƒWntjd|jƒdSX|jtjƒ}|dkrtjd|fƒdSyJtj|ƒ} tjƒjƒtj krctj dt j | ƒƒnWntjd|ƒdSX| S( s”Exchange message data with the server. @param payload: The object to send, it must be L{bpickle}-compatible. @param computer_id: The computer ID to send the message as (see also L{Identity}). @type: C{dict} @return: The server's response to sent message or C{None} in case of error. @note: This code is thread safe (HOPEFULLY). sSending payload: %ss*Sent %d bytes and received %d bytes in %s.s"Error contacting the server at %s.iÈs'Server returned non-expected result: %dsReceived payload: %ss Server returned invalid data: %rN(RtdumpsR tNonetsavettimetloggingt getLoggertgetEffectiveLeveltDEBUGtdebugtpprinttpformatRtinfotlenRt exceptionRtgetinfoRt RESPONSE_CODEterrortloads( R RRRtspayloadt start_timetcurlyRtcodetresponse((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pytexchange0s6    N( t__name__t __module__t__doc__RRRRRRR4(((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyRs    tPayloadRecordercBs;eZdZd„Zd„Zd„Zd„Zd„ZRS(sã L{PayloadRecorder} records client exchanges with the server to disk for later playback. Exchange payloads will be stored one per file, where the file name is the elapsed time since the client was started. cCsQtjƒ|_||_d|_|jdk rM|j|jƒ|jƒndS(sP @param destination_dir - The directory to record exchanges in. iÿÿÿÿN(R t _time_offsett_destination_dirt_last_payload_timeRt_create_destination_dirt_delete_old_payloads(R tdestination_dir((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyRhs   cCs&tjj|ƒs"tj|ƒndS(s}Create the destination directory if it does not exist. @param destination_dir: The directory to be created. N(tostpathtexiststmkdir(R R>((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyR<sscCs[xTtj|jƒD]@}tjj|j|ƒ}tjj|ƒrtj|ƒqqWdS(s5Delete payloads lying around from a previous session.N(R?tlistdirR:R@tjointisfiletunlink(R tfilenamet file_path((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyR={scCs/|jƒ}ttjj|j|ƒ|ƒdS(saPersist the given payload to disk. @param payload: The payload to be persisted. N(tget_payload_filenameRR?R@RDR:(R Rt payload_name((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyR‚s cCsTtjƒ|j}d|j}d|}||krC|d}n||_d|S(sr Generate a payload filename. This method ensures that payloads will have a unique name. s%.3fgü©ñÒMbP?(R R9R;(R t payload_timetlast_payload_timetthis_payload_time((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyRI‹s     (R5R6R7RR<R=RRI(((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyR8_s    t FakeTransportcBsAeZdZdddd„Zd„Zd„Zded„ZRS(s$Fake transport for testing purposes.cCs^||_||_g|_g|_d|_d|_d|_d|_i|_ ||_ dS(Ni( RR tpayloadst responsest_current_responsetnext_expected_sequenceRRRtextraR(R R R R ((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyRœs         cCs|jS(N(R(R ((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyR¨scCs ||_dS(N(R(R R ((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyR«scCs®|jj|ƒ||_||_|jt|jdgƒƒ7_|jt|jƒkr}|j|j}|jd7_ng}i|jd6|d6}|j |j ƒ|S(Ntmessagesisnext-expected-sequence( ROtappendRRRRR)tgetRQRPtupdateRS(R RRRR3tresult((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyR4®s  !  N( R5R6R7RRRRRR4(((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyRN™s   (R7R?R R!R&Rtlandscape.lib.fetchRtlandscape.lib.fsRt landscape.libRt landscape.logRt landscapeRRtobjectRR8RN(((s>/usr/lib/python2.7/dist-packages/landscape/broker/transport.pyts     O: