.PhL\dZddlmZddlmZddlmZddlZddlm Z m Z dZ gd Z d Z dZd ZdZdZd ZdZd Zd Zd ZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(d Z)d!Z*d"Z+d#Z,d$Z-e e$e%e'egZ.e/ge.e&eeeee(e#eee)e*e+Z0e/ge.e"ee!eZ1d%Z2d&Z3d'Z4d(Z5d)Z6d*Z7d+Z8d,Z9d-Z:dZ;d.Zd1Z?d2Z@d3ZAd4ZBd5ZCd6ZDGd7d8ZEd9ZFGd:d;eEZGd<ZHd=ZId>ZJGd?d@ZKGdAdBZLdCZMdIdDZNdEZOdFZPdJdHZQdS)Kz"?"?@@ @ H H Q QQrc4tdztzS)N=)CAPABILITY_AGENTrrrrcapability_agentrs d "\^^ 33rc(tdz|zdz|zS)Nr:)CAPABILITY_SYMREF)from_refto_refs rcapability_symrefr#s 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-s 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_capabilitiesr9s 3 37 3 3 33rsdeepens unshallowsdoneswantshavecL|dzdd|DzS)N rcg|]}|dzS)r)r(as rr7z"format_cmd_pkt..s!.s...QXa[[...rN)rrL)seqs rpkt_seqrQs0 88..#... / /(4.. @@rcDfd|DS)zFilter refs to only include those with a given prefix. Args: refs: A list of refs. prefix: The prefix to filter by. cTi|]#\}tfdD |$S)c3BK|]}|VdSN) startswith)r(pr6s r z/filter_ref_prefix...s-0S0SQa0S0S0S0S0S0Sr)any)r(vr6prefixess @r z%filter_ref_prefix..sB T T TTQS0S0S0S0S(0S0S0S-S-S TAq T T Tr)items)refsr[s `rfilter_ref_prefixr_s( U T T TTZZ\\ T T TTrcneZdZdZdddZddZdZdZdZde fd Z dd Z d Z dd Z dd ZddZdZdS)ProtocolaClass for interacting with a remote git process over the wire. Parts of the git wire protocol use 'pkt-lines' to communicate. A pkt-line consists of the length of the line as a 4-byte hex string, followed by the payload data. The length includes the 4-byte header. The special line '0000' indicates the end of a section of input and is called a 'flush-pkt'. For details on the pkt-line format, see the cgit distribution: Documentation/technical/protocol-common.txt NreturncL||_||_||_||_d|_dSrU)readwrite_closereport_activity _readahead)selfrdreclosergs r__init__zProtocol.__init__s*   .-1rc@|jr|dSdSrU)rfris rrjzProtocol.closes' ;  KKMMMMM  rc|SrUrrms r __enter__zProtocol.__enter__s rc.|dSrU)rj)riexc_typeexc_valexc_tbs r__exit__zProtocol.__exit__s rcB|j|j}n|jj}d|_ |d}|stt|d}|dks|dkr|jr|dddS|jr||d||dz }t |dz|kr't dt |dzdd |d|S#t$r }t|d}~wt$r"}t t||d}~wwxYw) aReads a pkt-line from the remote git process. This method may read from the readahead buffer; see unread_pkt_line. Returns: The next string from the stream, without the length prefix, or None for a flush-pkt ('0000') or delim-pkt ('0001'). NrJrrrdzLength of pkt read 04xz does not match length prefix ) rhrdrintrgr0rConnectionResetErrorOSErrorr)rirdsizestrsize pkt_contentsexcs r read_pkt_linezProtocol.read_pkt_linesn ? "9DD?'D"DO d1ggG &%%w##DqyyDAII'4((F333t# 3$$T62224q>>L <  1$,,&m#l*;*;a*?mmmcgmmm $ + + +!s * 6 6 6"3s88,,# 5 6s*A C3+C D'C// D<DDc~ |}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)rrunread_pkt_line)ri 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.)rh ValueErrorrrLrirKs rrzProtocol.unread_pkt_line"s4 ? &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)r)ripkts r read_pkt_seqzProtocol.read_pkt_seq2sY   "" '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. reN)rLrergr0rzrr)rirFr~s rwrite_pkt_linezProtocol.write_pkt_line=s 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)rbytes bytearray)richannelblobs rwrite_sidebandzProtocol.write_sidebandLsj   i &:&: ; ;d6E6l J K K KD ::d##D D ADyyy  Dyy $999AtAAAAA999 IIdOOO qLG3 6 xxzzrc|dksJ|j}|}|dt|}||||z }|so||j}t ||kr|St}|||d~||_| |S)Nr) rrrrrrr0rrerd)rir|rrrrrKs rrzReceivableProtocol.recvsaxxxxj  H((**  ::dn--D4yyD   ))C IIdOOO HHQKKKDJxx~~rr) rrrr _RBUFSIZErkrdr __classcell__)rs@rrrssw  "&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/r0r)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,eZdZdZdd dZd dZd dZdS) 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. rrbNcV||_||_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)rirebufsizes rrkzBufferedPktLineWriter.__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)rLr0rrrreflush)rirKrFline_lenoverrsaveds rrezBufferedPktLineWriter.writes~~t99|h&6 199tOE J  T&5&\ * * * JJLLLLEUVV   E " rc|j}|r||d|_t |_dS)zFlush all data from the buffer.rN)rgetvaluer_lenrrs rrzBufferedPktLineWriter.flush'sEz""$$   KK    YY r)rr)rrrrrkrerrrrrrs_      # # # #rrc(eZdZdZddZddZdZdS) PktLineParserzBPacket line parser that hands completed packets off to a callback.rbNc:||_t|_dSrU) handle_pktrrh)rirs rrkzPktLineParser.__init__3s$!))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.rJNrvr)rhrerr0rxrr)rirKrr|s rparsezPktLineParser.parse7s d###o&&(( s88a<< F#hh!mms2A2w##Dqyy%%%!""gS!!AdF ,,,$%%j#hh!mm")) c"""""rc4|jS)zRead back any unused data.)rhrrms rget_tailzPktLineParser.get_tailJs'')))rr)rrrrrkrrrrrrr0sQLL$$$$####&*****rrc@dd|DS)Nrcg|]}d|zS)r;rr's rr7z*format_capability_line..Ps444!TAX444rr?r+s rformat_capability_linerOs# 8844|444 5 55rcX| |dz|zdzS|dz|zdzt|zdzS)Nr; r=)r)refshar,s rformat_ref_linerSsDTzC%''TzC%'*@*N*NNQVVVrctdz|zSNr;)COMMAND_SHALLOWrs rformat_shallow_linerZs T !C ''rctdz|zSr)COMMAND_UNSHALLOWrs rformat_unshallow_liner^s t #c ))rrc&|rd|z}d|z|zdzS)Nr;sACK rr)rrs rformat_ack_linerbs'#(? S=8 #e ++rrU)r)Rriorosrtypingrrerrorsrr TCP_GIT_PORTGIT_PROTOCOL_VERSIONS"DEFAULT_GIT_PROTOCOL_VERSION_FETCH!DEFAULT_GIT_PROTOCOL_VERSION_SENDZERO_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_WANTCAPABILITY_FETCHCAPABILITY_FILTERCAPABILITIES_REFCOMMON_CAPABILITIESsetKNOWN_UPLOAD_CAPABILITIESKNOWN_RECEIVE_CAPABILITIESDEPTH_INFINITENAK_LINErrr#r-r&r9COMMAND_DEEPENrr COMMAND_DONE COMMAND_WANT COMMAND_HAVErBrHrLrQr_rarrrrrrrrrrrrrrrrs[.CB55555555 " %&"$%!   )%/''# 5'#+#+#$=!*I'&  C   &        # * 0 "!S        RRR444???:::444    >>>''' = = =AAAUUU\#\#\#\#\#\#\#\#~  ffffffffR 4 4 4777$))))))))X********>666WWWW(((***,,,,,,r