dGdZddlmZddlmZddlZddlmZmZdZ dZ dZ dZ d Z dZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%d Z&eee e"egZ'e(e'e!eeeee#eeee$e%g zZ)e(e'eeeegzZ*d!Z+d"Z,d#Z-d$Z.d%Z/d&Z0d'Z1d(Z2d)Z3dZ4d*Z5d+Z6d,Z7d-Z8d.Z9d/Z:d0Z;Gd1d2Zd6Z?d7Z@d8ZAGd9d:ZBGd;d<ZCd=ZDdCd>ZEd?ZFd@ZGdDdBZHdS)Ez"?"?@@ @ H H Q QQrc4tdztzS)N=)CAPABILITY_AGENTrrrrcapability_agentrrs d "\^^ 33rc(tdz|zdz|zS)Nr:)CAPABILITY_SYMREF)from_refto_refs rcapability_symrefr"vs t #h . 5 >>rcd|DS)Nc8h|]}t|dS)r)parse_capability.0cs r z+extract_capability_names..{s& 9 9 9q Q   " 9 9 9rr capabilitiess rextract_capability_namesr,zs 9 9L 9 9 99rc|dd}t|dkr |ddfSt|S)Nrrr)splitlentuple) capabilitypartss rr%r%~sA   T1 % %E 5zzQa$ <<rcd|DS)Nc g|] }t| Sr)r")r'ks r z'symref_capabilities..s 3 3 3a q ! 3 3 3rr)symrefss rsymref_capabilitiesr8s 3 37 3 3 33rsdeepens unshallowsdoneswantshavecL|dzdd|DzS)N rcg|]}|dzS)r)r'as rr6z"format_cmd_pkt..s!>L <  1$,,&O<((1,d34 $ - - -!## , 6 6 6"3s88,,# 5 6s*AC4+C D#%C44 D#DD#c~ |}n#t$rYdSwxYw||dS)zTest whether the protocol stream has reached EOF. Note that this refers to the actual stream EOF and not just a flush-pkt. Returns: True if the stream is at EOF, False otherwise. TF)rjrunread_pkt_line)rU next_lines reofz Protocol.eofsU **,,II   44  Y'''us  %%ct|jtdtt||_dS)a=Unread a single line of data into the readahead buffer. This method can be used to unread a single pkt-line into a fixed readahead buffer. Args: data: The data to unread, without the length prefix. Raises: ValueError: If more than one pkt-line is unread. Nz'Attempted to unread multiple pkt-lines.)rT ValueErrorrrKrUrJs rrlzProtocol.unread_pkt_lines4 ? &FGG G!(4..11rc#nK|}|r|V|}|dSdS)zRead a sequence of pkt-lines from the remote git process. Returns: Yields each line of data up to but not including the next flush-pkt. N)rj)rUpkts r read_pkt_seqzProtocol.read_pkt_seq sY   "" 'III$$&&C ' ' ' ' 'rc t|}|||jr%|t|ddSdS#t$r"}t t ||d}~wwxYw)zSends a pkt-line to the remote git process. Args: line: A string containing the data to send, without the length prefix. rQN)rKrQrSr/rerr)rUrEris rwrite_pkt_linezProtocol.write_pkt_lines 6D>>D JJt   # 9$$SYY88888 9 9 6 6 6"3s88,,# 5 6sAA BA;;Bc|rI|tt|g|ddz|dd}|GdSdS)zWrite multiplexed data to the sideband. Args: channel: An int specifying the channel to write to. blob: A blob of data (as a string) to send on this channel. N)rvbytes bytearray)rUchannelblobs rwrite_sidebandzProtocol.write_sideband$sj   i &:&: ; ;d6E6l J K K KD ::d##D D ADyyy  Dyy $999;tQiG999 IIdOOO qLG3 6 xxzzrc|dksJ|j}|}|dt|}||||z }|so||j}t ||kr|St}|||d~||_| |S)Nr) rrrrrrr/rrQrP)rUrgrrrrrJs rrzReceivableProtocol.recvsaxxxxj  H((**  ::dn--D4yyD   ))C IIdOOO HHQKKKDJxx~~r) rrrr _RBUFSIZErWrPr __classcell__)rs@rrrKso  "&ti"""""";;;zrrcd|vr|gfS|d\}}||dfS)zExtract a capabilities list from a string, if present. Args: text: String to extract from Returns: Tuple with text with capabilities removed and list of capabilities r<r:)rstripr.strip)textr+s rextract_capabilitiesrs] DRx,,U33D, ,$$&&,,T22 33rc|d}t|dkr|gfSd|dd|ddfS)axExtract a capabilities list from a want line, if present. Note that want lines have capabilities separated from the rest of the line by a space instead of a null byte. Thus want lines have the form: want obj-id cap1 cap2 ... Args: text: Want line to extract from Returns: Tuple with text with capabilities removed and list of capabilities r:r Nr)rr.r/r)r split_texts rextract_want_line_capabilitiesrs`$$T**J :Rx IIj!n % %z!""~ 66rc<d|vrtSd|vrtStS)z.Extract the ack type from a capabilities list.r r )MULTI_ACK_DETAILED MULTI_ACK SINGLE_ACKr*s rack_typers* ,,!!  % % rc&eZdZdZddZdZdZdS)BufferedPktLineWriteraWriter that wraps its data in pkt-lines and has an independent buffer. Consecutive calls to write() wrap the data in a pkt-line and then buffers it until enough lines have been written such that their total length (including length prefix) reach the buffer size. rxcV||_||_t|_d|_dS)zInitialize the BufferedPktLineWriter. Args: write: A write callback for the underlying writer. bufsize: The internal buffer size, including length prefixes. rN)_write_bufsizer_wbuf_buflen)rUrQbufsizes rrWzBufferedPktLineWriter.__init__s(  YY  rcpt|}t|}|j|z|jz }|dkr<||z }|j|d||nd}||d}|j||xjt|z c_dS)z&Write data, wrapping it in a pkt-line.rN)rKr/rrrrQflush)rUrJrEline_lenoverrsaveds rrQzBufferedPktLineWriter.writes~~t99|h&6 199tOE J  T&5&\ * * * JJLLLLEUVV   E " rc|j}|r||d|_t |_dS)zFlush all data from the buffer.rN)rgetvaluer_lenrrqs rrzBufferedPktLineWriter.flushsEz""$$   KK    YY rN)rx)rrrrrWrQrrrrrrsP     # # #rrc$eZdZdZdZdZdZdS) PktLineParserzBPacket line parser that hands completed packets off to a callback.c:||_t|_dSrO) handle_pktrrT)rUrs rrWzPktLineParser.__init__ s$!))rc0|j||j}t|dkrdSt|dkrt |ddd}|dkr |d|dd}n<|t|kr(||d|||d}nnt|dkt |_|j|dS)zAParse a fragment of data and call back for any completed packets.rINrbr)rTrQrr/rcrr)rUrJrrgs rparsezPktLineParser.parses d###o&&(( s88a<< F#hh!mms2A2w##Dqyy%%%!""gS!!AdF ,,,$%%j#hh!mm")) c"""""rc4|jS)zRead back any unused data.)rTrrYs rget_tailzPktLineParser.get_tail$s'')))rN)rrrrrWrrrrrrr sGLL$$$###&*****rrc@dd|DS)Nrcg|]}d|zS)r:rr&s rr6z*format_capability_line..*s444!TAX444rr>r*s rformat_capability_liner)s# 8844|444 5 55rcX| |dz|zdzS|dz|zdzt|zdzS)Nr: r<)r)refshar+s rformat_ref_liner-sLTzC%'' $J u $$\22 3  rctdz|zSNr:)COMMAND_SHALLOWrs rformat_shallow_liner7s T !C ''rctdz|zSr)COMMAND_UNSHALLOWrs rformat_unshallow_liner;s t #c ))rrc&|rd|z}d|z|zdzS)Nr:sACK rr)rrs rformat_ack_liner?s'#(? S=8 #e ++rrO)r)Iriorosrrerrorsrr TCP_GIT_PORTZERO_SHArrrSIDE_BAND_CHANNEL_DATASIDE_BAND_CHANNEL_PROGRESSSIDE_BAND_CHANNEL_FATALCAPABILITY_ATOMICCAPABILITY_DEEPEN_SINCECAPABILITY_DEEPEN_NOTCAPABILITY_DEEPEN_RELATIVECAPABILITY_DELETE_REFSCAPABILITY_INCLUDE_TAGCAPABILITY_MULTI_ACKCAPABILITY_MULTI_ACK_DETAILEDCAPABILITY_NO_DONECAPABILITY_NO_PROGRESSCAPABILITY_OFS_DELTACAPABILITY_QUIETCAPABILITY_REPORT_STATUSCAPABILITY_SHALLOWCAPABILITY_SIDE_BANDCAPABILITY_SIDE_BAND_64KCAPABILITY_THIN_PACKrr!CAPABILITY_ALLOW_TIP_SHA1_IN_WANT'CAPABILITY_ALLOW_REACHABLE_SHA1_IN_WANTCAPABILITIES_REFCOMMON_CAPABILITIESsetKNOWN_UPLOAD_CAPABILITIESKNOWN_RECEIVE_CAPABILITIESDEPTH_INFINITENAK_LINErrr"r,r%r8COMMAND_DEEPENrr COMMAND_DONE COMMAND_WANT COMMAND_HAVErArGrKrMrrrrrrrrrrrrrrrrs,CB55555555   )%/''# 5'#+#+#$=!*I'&  C%")/   !S   RRR444???:::444    >>>''' = = =\#\#\#\#\#\#\#\#~  hhhhhhhhV 4 4 4777$))))))))X********>666(((***,,,,,,r