-=[ Mr. Bumblebee ]=-
_Indonesia_
S:Qc @@ sn d Z d d l m Z d d l Z d d l m 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 m
Z
d d l m Z m Z d d l m Z m Z d d l m Z m Z d Z d
Z d Z e Z Z d Z d
Z d Z d e f d YZ d e f d YZ! d e! f d YZ" d e" f d YZ# d Z$ d Z% d e& f d YZ' d e f d YZ( d e( f d YZ) d e( f d YZ* d! e! e e j+ f d" YZ, d# e, f d$ YZ- d d% Z/ d& e( f d' YZ0 d( e f d) YZ1 d* e1 f d+ YZ2 d, Z3 d- e1 e f d. YZ4 d S(/ s\ Wire-level encoding and decoding of requests and responses for the smart
client and server.
i ( t absolute_importN( t StringIO( t debugt errorst osutils( t messaget request( t log_exception_quietlyt mutter( t bdecode_as_tuplet bencodes bzr request 2
s bzr response 2
s bzr message 3 (bzr 1.6)
c C@ s | j } t | S( N( t readlinet
_decode_tuple( t from_filet req_line( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _recv_tuple3 s c C@ sY | d k s | d k r d S| d d k rB t j d | n t | d j d S( Nt is
s request %r not terminateds ( t NoneR t SmartProtocolErrort tuplet split( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR 8 s
c C@ sH d j | d } t | t k rD t d | | j d } n | S( s& Encode the tuple args to a bytestream.s s
s7 response args contain unicode, should be only bytes: %rt ascii( t joint typet unicodeR t encode( t argst joined( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt
_encode_tuple@ s t Requesterc B@ s2 e Z d Z d Z d Z d Z d Z RS( sU Abstract base class for an object that can issue requests on a smart
medium.
c G@ s t | j d S( sN Make a remote call.
:param args: the arguments of this call.
N( t NotImplementedErrort call( t selfR ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR P s c C@ s t | j d S( s Make a remote call with a body.
:param args: the arguments of this call.
:type body: str
:param body: the body to send with the request.
N( R t call_with_body_bytes( R R t body( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR! W s c C@ s t | j d S( s Make a remote call with a readv array.
:param args: the arguments of this call.
:type body: iterable of (start, length) tuples.
:param body: the readv ranges to send with this request.
N( R t call_with_body_readv_array( R R R" ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR# ` s c C@ s t | j d S( N( R t set_headers( R t headers( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR$ i s ( t __name__t
__module__t __doc__R R! R# R$ ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR K s
t SmartProtocolBasec B@ s e Z d Z d Z d Z RS( s# Methods common to client and serverc C@ s d j d t | | d f S( s! Encode body as a bulk data chunk.R s %d
s done
( R t len( R R" ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _encode_bulk_datar s c C@ sA g } x+ | D]# \ } } | j d | | f q
Wd j | S( s Serialise a readv offset list.s %d,%ds
( t appendR ( R t offsetst txtt startt length( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _serialise_offsetsv s ( R& R' R( R+ R1 ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR) m s t SmartServerRequestProtocolOnec B@ sJ e Z d Z d d d Z d Z d Z d Z d Z d Z RS( s< Server-side encoding and decoding logic for smart version 1.t /c C@ s^ | | _ | | _ | | _ d | _ t | _ d | _ t | _ d | _ d | _
| | _ d S( NR ( t _backing_transportt _root_client_patht
_jail_roott unused_datat Falset _finishedt in_buffert _has_dispatchedR R t
_body_decodert _write_func( R t backing_transportt
write_funct root_client_patht jail_root( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt __init__ s c C@ s t | t s t | n | j | 7_ | j sd | j k rI d St | _ y | j j d d \ } | _ | d 7} t | } t j | j
d t j d | j d | j
| _ | j j | | j j r | j | _ d | _ | j | j j n Wqt k
r qt j k
rn} t j d | j f } t j d t | f } | j | d St k
r} t | j t j d t | f d SXn | j r| j r| j | j 7_ d | _ d S| j d k rt | _ n | j j | j | j j | _ | j j } | j j! | | j j ru| j j" | j j sut# d
qun | j j d k r| j | j j | j | _ d | _ q| j j rt# d qn d S( sw Take bytes, and advance the internal state machine appropriately.
:param bytes: must be a byte string
s
Ni t commandsR@ RA R s bad request %rt errors" no more body, request not finisheds) no response and we have finished reading.($ t
isinstancet strt
ValueErrorR: R; t TrueR R R t SmartServerRequestHandlerR4 t request_handlersR5 R6 t
args_receivedt finished_readingR7 t _send_responset responset KeyboardInterruptR t UnknownSmartMethodR t verbt FailedSmartServerResponset ExceptionR R9 R< R t LengthPrefixedBodyDecodert accept_bytest read_pending_datat accept_bodyt end_of_bodyt AssertionError( R t bytest
first_linet req_argst errt protocol_errort failuret exceptiont body_data( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRU sp
c C@ s | j r t d n | j } | j } t | _ | j | j | | j t | | d k r t
| t s t | n | j
| } | j | n d S( s4 Send a smart server response down the output stream.s response already sentN( R9 RY R R" RH t _write_protocol_versiont _write_success_or_failure_prefixR= R R RE RF RG R+ ( R RN R R" RZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRM s
c C@ s d S( sh Write any prefixes this protocol requires.
Version one doesn't send protocol versions.
N( ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRb s c C@ s | j d S( s Write the protocol specific success/failure prefix.
For SmartServerRequestProtocolOne this is omitted but we
call is_successful to ensure that the response is valid.
N( t
is_successful( R RN ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRc s c C@ s1 | j r
d S| j d k r d S| j j Sd S( Ni i ( R9 R< R t next_read_size( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRe s
N(
R& R' R( R RB RU RM Rb Rc Re ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR2 ~ s E t SmartServerRequestProtocolTwoc B@ s5 e Z d Z e Z e Z d Z d Z d Z RS( s Version two of the server side of the smart protocol.
This prefixes responses with the value of RESPONSE_VERSION_TWO.
c C@ s- | j r | j d n
| j d d S( s3 Write the protocol specific success/failure prefix.s success
s failed
N( Rd R= ( R RN ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRc s c C@ s | j | j d S( sq Write any prefixes this protocol requires.
Version two sends the value of RESPONSE_VERSION_TWO.
N( R= t response_marker( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRb
s c C@ s | j r t d n t | _ | j | j | | j t | j | j d k r t
| j t s~ t d n | j d k r t d n | j
| j } | j | n% | j d k r t | j | j n d S( s4 Send a smart server response down the output stream.s response already sents body must be a strs' body_stream and body cannot both be setN( R9 RY RH Rb Rc R= R R R" R RE RF t body_streamR+ t _send_stream( R RN RZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRM s
(
R& R' R( t RESPONSE_VERSION_TWORg t REQUEST_VERSION_TWOt request_markerRc Rb RM ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRf s c C@ s% | d t | | | d d S( Ns chunked
s END
( t _send_chunks( t streamR? ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRi % s
c C@ s x | D]{ } t | t r? d t | | f } | | q t | t j ro | d t | j | d St j d | q Wd S( Ns %x
%ss ERR
s7 Chunks must be str or FailedSmartServerResponse, got %r( RE RF R* R RR Rm R R t BzrError( Rn R? t chunkRZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRm + s
t _NeedMoreBytesc B@ s e Z d Z d d Z RS( sc Raise this inside a _StatefulDecoder to stop decoding until more bytes
have been received.
c C@ s
| | _ d S( s Constructor.
:param count: the total number of bytes needed by the current state.
May be None if the number of bytes needed is unknown.
N( t count( R Rr ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB ? s N( R& R' R( R RB ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRq : s t _StatefulDecoderc B@ s; e Z d Z d Z d Z d Z d Z d Z RS( s Base class for writing state machines to decode byte streams.
Subclasses should provide a self.state_accept attribute that accepts bytes
and, if appropriate, updates self.state_accept to a different function.
accept_bytes will call state_accept as often as necessary to make sure the
state machine has progressed as far as possible before it returns.
See ProtocolThreeDecoder for an example subclass.
c C@ s: t | _ g | _ d | _ d | _ d | _ d | _ d S( Ni R ( R8 RL t _in_buffer_listt _in_buffer_lenR7 R t
bytes_leftt _number_needed_bytes( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB S s c C@ sv t | j d k r | j d Sd j | j } t | | j k rf t d | j t | n | g | _ | S( Ni i R s7 Length of buffer did not match expected value: %s != %s( R* Rt R Ru RY ( R R: ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _get_in_buffer[ s c C@ s` t | j d k r$ t d n t | j d | k rL | j d | S| j } | | S( sG Grab X bytes from the input_buffer.
Callers should have already checked that self._in_buffer_len is >
count. Note, this does not consume the bytes from the buffer. The
caller will still need to call _get_in_buffer() and then
_set_in_buffer() if they actually need to consume the bytes.
i sH Callers must be sure we have buffered bytes before calling _get_in_bytes( R* Rt RY Rx ( R Rr t in_buf( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt
_get_in_bytesf s c C@ s@ | d k r* | g | _ t | | _ n g | _ d | _ d S( Ni ( R Rt R* Ru ( R t new_buf( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _set_in_buffery s
c C@ s d | _ | j j | | j t | 7_ y@ | j } | j x& | | j k rl | j } | j qG WWn t k
r } | j | _ n Xd S( s
Decode as much of bytes as possible.
If 'bytes' contains too much data it will be appended to
self.unused_data.
finished_reading will be set when no more data is required. Further
data will be appended to self.unused_data.
N( R Rw Rt R, Ru R* t state_acceptRq Rr ( R RZ t
current_statet e( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRU s
( R& R' R( RB Rx Rz R| RU ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRs H s t ChunkedBodyDecoderc B@ s_ e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z RS(
s Decoder for chunked body data.
This is very similar the HTTP's chunked encoding. See the description of
streamed body data in `doc/developers/network-protocol.txt` for details.
c C@ sG t j | | j | _ d | _ t j | _ t | _
d | _ d S( N( Rs RB t _state_accept_expecting_headerR} R t chunk_in_progresst collectionst dequet chunksR8 RD t error_in_progress( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB s
c C@ s | j | j k r | j d S| j | j k rI | j d k rB d Sd SnX | j | j k r_ d S| j | j k r t d t d | j St d | j f d S( Ni i i i s chunked
s Impossible state: %r(
R} t _state_accept_reading_chunkRv t _state_accept_expecting_lengthRu t _state_accept_reading_unusedR t maxR* RY ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRe s c C@ s* y | j j SWn t k
r% d SXd S( N( R t popleftt
IndexErrorR ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt read_next_chunk s
c C@ sY | j } | j d } | d k r6 t d n | | } | j | | d | S( Ns
ii ( Rx t findRq R| ( R Ry t post line( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt
_extract_line s
c C@ sw | j | _ g | _ d | _ | j | _ | j rj t | j } | j j
t j | d | _ n t | _ d S( Ni ( Rx R7 Rt Ru R R} RD R R R R, R RR R RH RL ( R t
error_args( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR9 s c C@ sA | j } | d k r' | j | _ n t j d | f d S( Nt chunkeds Bad chunked body header: "%s"( R R R} R R ( R t prefix( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s
c C@ s} | j } | d k r8 t | _ g | _ | j d S| d k rR | j d St | d | _ d | _ | j | _
d S( Nt ERRt ENDi R ( R RH RD R R R9 t intRv R R R} ( R R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s
c C@ s | j } t | } | j | | j 7_ | j | | j | j | 8_ | j d k r d | _ | j r | j j | j n | j j | j d | _ | j
| _ n d S( Ni ( Rx R* R Rv R| R RD R R, R R R} ( R Ry t
in_buffer_len( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s c C@ s" | j | j 7_ g | _ d S( N( R7 Rx Rt ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR
s ( R& R' R( RB Re R R R9 R R R R ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s RT c B@ s_ e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z RS(
s& Decodes the length-prefixed bulk data.c C@ s; t j | | j | _ | j | _ d | _ d | _ d S( NR ( Rs RB R R} t _state_read_no_datat
state_readt _bodyt _trailer_buffer( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB s
c C@ s[ | j d k r | j d S| j | j k r= d t | j S| j | j k rS d Sd Sd S( Ni i i ( Rv R R} t _state_accept_reading_trailerR* R R ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRe s c C@ s
| j S( s. Return any pending data that has been decoded.( R ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRV , s c C@ so | j } | j d } | d k r+ d St | | | _ | j | | d | j | _ | j | _ d S( Ns
ii ( Rx R R Rv R| t _state_accept_reading_bodyR} t _state_read_body_bufferR ( R Ry R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR 0 s c C@ s | j } | j | 7_ | j t | 8_ | j d | j d k r | j d k r | j | j | _ | j | j | _ n d | _ | j | _ n d S( Ni ( Rx R Rv R* R| R R R R} ( R Ry ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR : s
c C@ sf | j | j 7_ | j d | j j d rb | j t d | _ | j | _ t | _
n d S( Ns done
( R Rx R| R t
startswithR* R7 R R} RH RL ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR G s
c C@ s& | j | j 7_ | j d d S( N( R7 Rx R| R ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR Q s c C@ s d S( NR ( ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR U s c C@ s | j } d | _ | S( NR ( R ( R t result( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR X s ( R& R' R( RB Re RV R R R R R R ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRT s
t SmartClientRequestProtocolOnec B@ s e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
e d Z d
Z
d Z d d
Z d Z d Z d Z d Z RS( s- The client-side protocol for smart version 1.c C@ s1 | | _ d | _ d | _ d | _ d | _ d S( s Construct a SmartClientRequestProtocolOne.
:param request: A SmartClientMediumRequest to serialise onto and
deserialise from.
N( t _requestR t _body_buffert _request_start_timet
_last_verbt _headers( R R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB b s
c C@ s t | | _ d S( N( t dictR ( R R% ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR$ n s c G@ s d t j k rr t d t | d d ! t | j j d d d k r` t d | j j j n t j
| _ n | j | | j j
| d | _ d S( Nt hpsss hpss call: %si it bases (to %s)i ( R t debug_flagsR t reprt getattrR t _mediumR R R t
timer_funcR t _write_argst finished_writingR ( R R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR q s
c C@ s d t j k r t d t | d d !| d t | j j d d d k rg t d | j j j n t d t | t
j | _ d t j k r t d
| q n | j
| | j | } | j j | | j j | d | _ d S(
s Make a remote call of args with body bytes 'body'.
After calling this, call read_response_tuple to find the result out.
R s hpss call w/body: %s (%r...)i ii t _paths (to %s)s %d bytest
hpssdetails hpss body content: %si N( R R R R R R R R R R* R R R R R+ RU R R ( R R R" RZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR! { s !
c C@ s d t j k rr t d t | d d ! t | j j d d d k r` t d | j j j n t j
| _ n | j | | j
| } | j | } | j j | | j j d t j k r t d t | n | d | _ d S(
s Make a remote call with a readv array.
The body is encoded with one line per readv offset pair. The numbers in
each pair are separated by a comma, and no trailing \n is emitted.
R s hpss call w/readv: %si iR s (to %s)s' %d bytes in readv requesti N( R R R R R R R R R R R R R R1 R+ RU R R* R ( R R R" t readv_bytesRZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR# s
c C@ s1 | j j | j j t j | d d S( Ni ( R R RL R RP ( R R Rn ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt call_with_body_stream s
c C@ s | j j d S( s After expecting a body, a response code may indicate one otherwise.
This method lets the domain client inform the protocol that no body
will be transmitted. This is a terminal method: after calling it the
protocol is not able to be used further.
N( R RL ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt cancel_read_body s c C@ s | j } d t j k r} | j d k r` t d t j | j t | d d ! d | _ q} t d t | d d ! n | S( NR s result: %6.3fs %si is result: %s( R R R R R R R R R ( R R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _read_response_tuple s c C@ s@ | j } | j | | j | | s< | j j n | S( sW Read a response tuple from the wire.
This should only be called once.
( R t _response_is_unknown_methodt _raise_args_if_errorR RL ( R t expect_bodyR ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt read_response_tuple s
c C@ sl d d d d d d d d d d
d d d
d d d d g } | d | k rh | j j t j | n d S( Nt norepositoryt
NoSuchFilet
FileExistst DirectoryNotEmptyt ShortReadvErrort UnicodeEncodeErrort UnicodeDecodeErrort
ReadOnlyErrort nobrancht NoSuchRevisiont nosuchrevisiont LockContentiont UnlockableTransportt
LockFailedt
TokenMismatcht ReadErrort PermissionDeniedi ( R RL R t ErrorFromSmartServer( R t result_tuplet v1_error_codes( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s(
c C@ sX | d d | j f k s2 | d d | j f k rT | j j t j | j n d S( s6 Raise UnexpectedSmartServerResponse if the response is an 'unknonwn
method' response to the request.
:param response: The response from a smart client call_expecting_body
call.
:param verb: The verb used in that call.
:raises: UnexpectedSmartServerResponse
RD s2 Generic bzr smart protocol error: bad request '%s's3 Generic bzr smart protocol error: bad request u'%s'N( R R RL R RP ( R R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s
ic C@ s | j d k r | j j | St } xP | j sz | j j | j } | d k rj t j d n | j
| q+ W| j j t | j | _ d t
j k r t d t | j j n | j j | S( s Read bytes from the body, decoding into a byte stream.
We read all bytes at once to ensure we've checked the trailer for
errors, and then feed the buffer back as read_body_bytes is called.
R s, Connection lost while reading response body.R s %d body bytes readN( R R t readRT RL R t
read_bytesRe R t ConnectionResetRU R RV R R R R* t getvalue( R Rr R< RZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt read_body_bytes s
c C@ s t | j j S( s( Receive a tuple from the medium request.( R R t read_line( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s c C@ sS | j d | j } | d k r) d S| d
k r9 d St j d | f d S( s- Return protocol version number of the server.t hellot okt 1i t 2i s bad response %rN( s okR ( s okR ( R R R R ( R t resp( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt
query_version s
c C@ s* | j t | } | j j | d S( N( Rb R R RU ( R R RZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR ! s
c C@ s d S( sh Write any prefixes this protocol requires.
Version one doesn't send protocol versions.
N( ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRb & s ( R& R' R( RB R$ R R! R# R R R R8 R R R R R R R Rb ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR ^ s"
t SmartClientRequestProtocolTwoc B@ s8 e Z d Z e Z e Z e d Z d Z d Z
RS( s Version two of the client side of the smart protocol.
This prefixes the request with the value of REQUEST_VERSION_TWO.
c C@ s | j j } | | j k r= | j j t j | n | j j } t j | } | j | | d k r t | _
| s | j j n | S| d k r t | _
| j j t j | n t j
d | d S( sW Read a response tuple from the wire.
This should only be called once.
s success
s failed
s bad protocol status %rN( R R Rg RL R t UnexpectedProtocolVersionMarkerR R R RH t response_statusR8 R R ( R R t versionR R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR 6 s$
c C@ s | j j | j d S( sp Write any prefixes this protocol requires.
Version two sends the value of REQUEST_VERSION_TWO.
N( R RU Rl ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRb O s c c@ s t } x | j s | j j | j } | d k rK t j d n | j | xV t | j d D]B } d t j k r t
| t k r t d t | n | Vqk Wq W| j j d S( s? Read bytes from the body, decoding into a byte stream.
R s, Connection lost while reading streamed body.R s %d byte chunk readN( R RL R R Re R R RU t iterR R R R R RF R R* ( R R< RZ t
body_bytes( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt read_streamed_bodyV s
!
( R& R' R( Rj Rg Rk Rl R8 R Rb R ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR - s c C@ sL t j | d t j d | d | } t | } t j | | } t | S( NRC R@ RA ( R RI RJ t ProtocolThreeResponderR t ConventionalRequestHandlert ProtocolThreeDecoder( R> R? R@ RA t request_handlert respondert message_handler( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt build_server_protocol_threek s R c B@ s e Z e Z e Z e d Z d Z d Z d Z
d Z d Z d Z
d Z d Z d Z d
Z d Z d Z d
Z RS( c C@ sp t j | t | _ | r> | j | _ t t d | _ n | j | _ d | _ t | _
| | _ | _ d S( Ni (
Rs RB R8 R; t( _state_accept_expecting_protocol_versionR} R* t MESSAGE_VERSION_THREERw t _state_accept_expecting_headerst decoding_failedR R ( R R t expect_version_marker( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB z s
c C@ s d | _ y t j | | Wn t k
r3 n t j k
r } t | j t j sd t
n | j j | j | j d nH t
k
r } t | _ t | t j r n t
| j j | n Xd S( NR ( R Rw Rs RU RO R t SmartMessageHandlerErrorRE t exc_valueRP R R R^ RS RH R R ( R RZ R` ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRU s
c C@ s | j d k r t d n t j d | j d \ } d | } | j | k rd t | n | j } | d | !} | j | | | S( Ni s !L( Ru Rq t structt unpackRz Rx R| ( R R0 t end_of_bytesRy RZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _extract_length_prefixed_bytes s
c C@ sJ | j } y t | } Wn' t k
rE t j d | f n X| S( Ns Bytes %r not bencoded( R R RG R R ( R t prefixed_bytest decoded( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _extract_prefixed_bencoded_data s
c C@ sI | j d k r t d n | j } | d } | j | d | S( Ni i ( Ru Rq Rx R| ( R Ry t one_byte( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _extract_single_byte s
c C@ s t t | j } | j } | d k ra t j | sL t j | n t t t n | j t s t j | n | j | t t | j | _
d S( Ni ( R* R Ru Rx R R R Rq R| R R} ( R t needed_bytesRy ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s c C@ sz | j } t | t k r7 t j d | f n | j | _ y | j j | Wn t j t
j n Xd S( Ns Header object %r is not a dict( R R R R R t$ _state_accept_expecting_message_partR} R t headers_receivedR t syst exc_info( R R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s c C@ s | j } | d k r' | j | _ ne | d k rB | j | _ nJ | d k r] | j | _ n/ | d k rv | j n t j d | f d S( Nt ot st bR s Bad message kind byte: %r( R t _state_accept_expecting_one_byteR} t! _state_accept_expecting_structuret _state_accept_expecting_bytest doneR R ( R t message_part_kind( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s
c C@ sO | j } | j | _ y | j j | Wn t j t j n Xd S( N( R R R} R t byte_part_receivedR R R R ( R t byte( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s c C@ sO | j } | j | _ y | j j | Wn t j t j n Xd S( N( R R R} R t bytes_part_receivedR R R R ( R R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s c C@ sO | j } | j | _ y | j j | Wn t j t j n Xd S( N( R R R} R t structure_part_receivedR R R R ( R t structure( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR
s c C@ s\ | j | _ | j d | j | _ y | j j Wn t j t
j n Xd S( N( Rx R7 R| R R R} R t end_receivedR R R R ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s
c C@ s& | j | j 7_ | j d d S( N( R7 Rx R| R ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s c C@ sP | j | j k r d S| j r# d S| j d k r@ | j | j St d d S( Ni s' don't know how many bytes are expected!( R} R R Rw R Ru RY ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRe " s ( R& R' t RESPONSE_VERSION_THREERg t REQUEST_VERSION_THREERl R8 RB RU R R R R R R R R R R R Re ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR u s !
t _ProtocolThreeEncoderc B@ s e Z e Z Z d Z d Z d Z d Z d Z d Z
d Z d Z d Z
d Z d
Z d Z d Z d
Z RS( i c C@ s g | _ d | _ | | _ d S( Ni ( t _buft _buf_lent _real_write_func( R R? ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB 7 s c C@ sH | j j | | j t | 7_ | j | j k rD | j n d S( N( R R, R R* t BUFFER_SIZEt flush( R RZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR= < s c C@ s9 | j r5 | j d j | j | j 2d | _ n d S( NR i ( R R R R ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR G s c C@ sA g } x+ | D]# \ } } | j d | | f q
Wd j | S( s Serialise a readv offset list.s %d,%ds
( R, R ( R R- R. R/ R0 ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR1 M s c C@ s | j t d S( N( R= R ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRb T s c C@ s< t | } | j t j d t | | j | d S( Ns !L( R
R= R t packR* ( R R RZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _write_prefixed_bencodeW s c C@ s | j | d S( N( R ( R R% ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _write_headers\ s c C@ sm | j d g } xF | D]> } t | t k rK | j | j d q | j | q W| j | d S( NR t utf8( R= R R R, R R ( R R t utf8_argst arg( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _write_structure_ s
c C@ s | j d | j d S( NR ( R= R ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt
_write_endi s
c C@ s= | j d | j t j d t | | j | d S( NR s !L( R= R R R* ( R RZ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _write_prefixed_bodym s
c C@ s | j d d S( Nt oC( R= ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _write_chunked_body_startr s c C@ s | j d d S( Nt oE( R= ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _write_error_statusu s c C@ s | j d d S( Nt oS( R= ( R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _write_success_statusx s i ( R& R' R Rg Rl R RB R= R R1 Rb R R R R R R R! R# ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR 2 s
R c B@ s2 e Z d Z d e d Z d Z d Z RS( c C@ sZ t j | | t | _ i t j d 6| _ d t j k rV t j
| _ d | _
n d S( Ns Software versionR ( R RB R8 t
response_sentt bzrlibt __version__R R R t threadt get_identt
_thread_idR t _response_start_time( R R? ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB ~ s c C@ s | j d k r! t j | _ n | rA d t j | j } n d } | d k r\ d } n? d t | d } t | d k r | d | d d } n t d | | j | | | f d S(
Ns %5.3fs R t i( i! i is ...s %12s: [%s] %s%s%s(
R* R R R t timet clockR R* R R) ( R t actionR t extra_bytest include_timet tt extra( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt _trace s c C@ s | j r t d | f n t | t j rZ t j d | j f } | j | d Sd t j
k r | j d t | n t
| _ | j | j | j | j | j d t | f | j d S( Ns1 send_error(%s) called, but response already sent.t
UnknownMethodR RD ( R$ RY RE R RP R RR RQ t
send_responseR R R3 RF RH Rb R R R! R R ( R R` R_ ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt
send_error s"
c C@ s | j r t d | f n t | _ | j | j | j | j r[ | j n
| j d t j
k r | j d t | j
n | j | j
| j d k r | j | j d t j
k rY| j d d t | j f | j d t qYnY| j d k rYd } } d } xt | j D] \ } } | d 7} | d k r| j t j | d } | j | Pq/t | t j r| j | j | j
Pn | t | 7} | d k r| } n | j | | j d t j
k r/| j d
d t | f | d t q/q/Wd t j
k rY| j d d
| | f | qYn | j d t j
k r| j d d d t n d S( Ns4 send_response(%r) called, but response already sent.R RN R" s %d bytesR0 i i R s
body chunkt
suppress_times body streams %d bytes %d chunkss response endR ( R$ RY RH Rb R R Rd R# R! R R R3 R R R R" R R R* Rh t _iter_with_errorsR t _translate_errorRE RR R R ( R RN Rr t num_bytest first_chunkR Rp t error_struct( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR5 s`
N( R& R' RB R R8 R3 R6 R5 ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR | s c c@ s t | } x t r y d | j f VWq t k
r> d St t f k
rW q t k
r t d t t
j d f Vd SXq Wd S( sp Handle errors from iterable.next().
Use like::
for exc_info, value in _iter_with_errors(iterable):
...
This is a safer alternative to::
try:
for value in iterable:
...
except:
...
Because the latter will catch errors from the for-loop body, not just
iterable.next()
If an error occurs, exc_info will be a exc_info tuple, and the generator
will terminate. Otherwise exc_info will be None, and value will be the
value from iterable.next(). Note that KeyboardInterrupt and SystemExit
will not be itercepted.
Ns _iter_with_errors caught error( R RH R t nextt
StopIterationRO t
SystemExitRS R R R R ( t iterablet iterator( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR8 s
t ProtocolThreeRequesterc B@ s> e Z d Z d Z d Z d Z d Z d Z RS( c C@ s2 t j | | j | | _ i | _ d | _ d S( N( R RB RU t _medium_requestR R t body_stream_started( R t medium_request( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB s c C@ s | j | _ d S( N( t copyR ( R R% ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR$ s c G@ s d t j k ro t d t | d d ! t | j j d d } | d k r] t d | n t j | _
n | j | j | j
| j | | j | j j d S( NR s hpss call: %si iR s (to %s)( R R R R R RC R R R R R Rb R R R R R ( R R R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR s
c C@ s d t j k r t d t | d d !| d t | j j d d } | d k rd t d | n t d t | t j
| _ n | j | j
| j | j | | j | | j | j j d S(
s Make a remote call of args with body bytes 'body'.
After calling this, call read_response_tuple to find the result out.
R s hpss call w/body: %s (%r...)i ii R s (to %s)s %d bytesN( R R R R R RC R R R* R R R Rb R R R R R R ( R R R" t path( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR! s !
c C@ s d t j k ro t d t | d d ! t | j j d d } | d k r] t d | n t j | _
n | j | j | j
| j | | j | } d t j k r t d t | n | j | | j | j j d S( s Make a remote call with a readv array.
The body is encoded with one line per readv offset pair. The numbers in
each pair are separated by a comma, and no trailing \n is emitted.
R s hpss call w/readv: %si iR s (to %s)s' %d bytes in readv requestN( R R R R R RC R R R R R Rb R R R R1 R* R R R ( R R R" RG R ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR# 2 s
c C@ sI d t j k rb t d | t | j j d d } | d k rP t d | n t j | _ n t
| _ | j | j
| j | j | | j t | _ x t | D]x \ } } | d k r| j | j d | j | j j | d | d | d q | j | | j q W| j | j j d S(
NR s hpss call w/body stream: %rR s (to %s)RD i i i ( s error( R R R R RC R R R R R R8 RD Rb R R R R RH R8 R! R R R ( R R Rn RG R t part( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyR H s.
( R& R' RB R$ R R! R# R ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyRB s
(5 R( t
__future__R R t cStringIOR R R R' R, R% R R R t bzrlib.smartR R t bzrlib.traceR R t bzrlib.bencodeR R
Rk Rj R R
R R R R t objectR R) R2 Rf Ri Rm RS Rq Rs R RT t ResponseHandlerR R R R R R R R8 RB ( ( ( s9 /usr/lib/python2.7/dist-packages/bzrlib/smart/protocol.pyt <module> sL
"|+ XrL ? Jd '
Copyright © 2017 || Recoded By Mr.Bumblebee