o U hL@s4dZddlmZddlmZddlmZddlZddlm 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'd(Z4d)d*Z5d+d,Z6d-d.Z7d/d0Z8d1d2Z9d3Z:dZ;d4Zd7Z?d8d9Z@d:d;ZAdd?ZCd@dAZDGdBdCdCZEdDZFGdEdFdFeEZGdGdHZHdIdJZIdKdLZJGdMdNdNZKGdOdPdPZLdQdRZMd\dSdTZNdUdVZOdWdXZPd]dZd[ZQdS)^zsz+extract_capability_names..r capabilitiesrrrextract_capability_namesrr*cCs,|dd}t|dkr|ddfSt|S)Nrrr)splitlentuple) capabilitypartsrrrr#s   r#cCr")NcSsg|]}t|qSr)r )r%krrr z'symref_capabilities..r)symrefsrrrsymref_capabilitiesrr4sdeepens unshallowsdoneswantshavecGs|dddd|DS)N cSsg|]}|dqS)r)r%arrrr1r2z"format_cmd_pkt..r)cmdargsrrrformat_cmd_pktrr<cCsR|d}|d|||dd}}|dddksJ||dddfS)Nr5rr7)findr+)line splice_atr:r;rrr parse_cmd_pkts rAcCs&|durdSdt|dd|S)zWrap data in a pkt-line. Args: data: The data to wrap, as a str or None. Returns: The data prefixed with its length in pkt-line format; if data was None, returns the flush-pkt ('0000'). Ns0000z%04xr )r,r)datarrrpkt_linesrDcGsddd|DtdS)z`Wrap a sequence of data in pkt-lines. Args: seq: An iterable of strings to wrap. r6cSsg|]}t|qSr)rD)r%srrrr1r2zpkt_seq..N)rrD)seqrrrpkt_seqsrGcsfdd|DS)zFilter refs to only include those with a given prefix. Args: refs: A list of refs. prefix: The prefix to filter by. cs,i|]\}tfddDr|qS)c3s|]}|VqdSN) startswith)r%pr0rr sz/filter_ref_prefix...)any)r%vprefixesrKr s,z%filter_ref_prefix..)items)refsrPrrOrfilter_ref_prefixsrTc@seZdZdZddddZdddZdd Zd d Zd d Zde fddZ dddZ ddZ dddZ dddZdddZddZdS)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 NreturncCs"||_||_||_||_d|_dSrH)readwrite_closereport_activity _readahead)selfrWrXcloserZrrr__init__s  zProtocol.__init__cCs|jr |dSdSrH)rYr\rrrr]s zProtocol.closecCs|SrHrr_rrr __enter__szProtocol.__enter__cCs |dSrH)r])r\exc_typeexc_valexc_tbrrr__exit__ zProtocol.__exit__c Cs|jdur |j}n|jj}d|_z2|d}|stt|d}|dks&|dkr2|jr/|ddWdS|jr;||d||d}Wn!tyQ}zt|d}~wtyc}ztt||d}~wwt |d|kr|tdt |ddd |d|S) 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'). NrBrrrWzLength of pkt read 04xz does not match length prefix ) r[rWrintrZConnectionResetErrorOSErrorrrr,)r\rWsizestrsize pkt_contentsexcrrr read_pkt_lines8    zProtocol.read_pkt_linecCs0z|}Wn tyYdSw||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)rorunread_pkt_line)r\ next_linerrreofs   z Protocol.eofcCs$|jdur tdtt||_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.)r[ ValueErrorrrDr\rCrrrrp"s zProtocol.unread_pkt_lineccs(|}|r|V|}|sdSdS)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)ro)r\pktrrr read_pkt_seq2s  zProtocol.read_pkt_seqc Cs\zt|}|||jr|t|dWdSWdSty-}ztt||d}~ww)zSends a pkt-line to the remote git process. Args: line: A string containing the data to send, without the length prefix. rXN)rDrXrZr,rjrr)r\r?rnrrrwrite_pkt_line=s zProtocol.write_pkt_linecCs<|r|tt|g|dd|dd}|sdSdS)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)rwbytes bytearray)r\channelblobrrrwrite_sidebandLs   zProtocol.write_sidebandcGs|t|g|RdS)zSend a command and some arguments to a git server. Only used for the TCP git protocol (git://). Args: cmd: The remote service to access. args: List of arguments to send to remove service. N)rwr<)r\r:r;rrrsend_cmdZs zProtocol.send_cmdcCs|}t|S)zRead a command and some arguments from the git client. Only used for the TCP git protocol (git://). Returns: A tuple of (command, [list of arguments]). )rorA)r\r?rrrread_cmdeszProtocol.read_cmd)NNrVN)__name__ __module__ __qualname____doc__r^r]r`rdroboolrrrprvrwr}r~rrrrrrUs %    rUi cs<eZdZdZddef d fdd ZddZdd ZZS) ReceivableProtocolaVariant of Protocol that allows reading up to a size without blocking. This class has a recv() method that behaves like socket.recv() in addition to a read() method. If you want to read n bytes from the wire and block until exactly n bytes (or EOF) are read, use read(n). If you want to read at most n bytes from the wire but don't care if you get less, use recv(n). Note that recv(n) will still block until at least one byte is read. NrVcs.tj|j|||d||_t|_||_dS)N)r]rZ)superr^rW_recvr_rbuf _rbufsize)r\recvrXr]rZrbufsize __class__rrr^s zReceivableProtocol.__init__c Cs |dksJ|j}|}|dt||}||kr;||||}t|_|j||jd|St|_ ||}||}|sLn1t|}||krX|sX|S||krc||~n||ksrJd|d|d||||7}~q@|||S)NrTz_recv(z ) returned z bytes) rtellseekrrWrrXrr,) r\rlbufstartbuf_lenrvleftrCnrrrrWs@           zReceivableProtocol.readcCs|dksJ|j}|}|dt|}||||}|sA||j}t||kr0|St}|||d~||_| |S)Nr) rrrrrrr,rrXrW)r\rlrrrrrCrrrrs"        zReceivableProtocol.recvr) rrrr _RBUFSIZEr^rWr __classcell__rrrrrss =rcCs4d|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 r7r5)rstripr+strip)textr)rrrextract_capabilitiessrcCs@|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 r5r Nr)rr+r,r)r split_textrrrextract_want_line_capabilitiess rcCsd|vrtSd|vr tStS)z.Extract the ack type from a capabilities list.r r )MULTI_ACK_DETAILED MULTI_ACK SINGLE_ACKr(rrrack_types rc@s0eZdZdZd d ddZd ddZd d d ZdS) 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. rxrVNcCs||_||_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)r\rXbufsizerrrr^ s zBufferedPktLineWriter.__init__cCst|}t|}|j||j}|dkr'||}|j|d||nd}||d}|j||jt|7_dS)z&Write data, wrapping it in a pkt-line.rN)rDr,rrrrXflush)r\rCr?line_lenoverrsavedrrrrXs   zBufferedPktLineWriter.writecCs*|j}|r ||d|_t|_dS)zFlush all data from the buffer.rN)rgetvaluer_lenrrtrrrr's   zBufferedPktLineWriter.flush)rxr)rrrrr^rXrrrrrrs   rc@s,eZdZdZd ddZd ddZdd ZdS) PktLineParserzBPacket line parser that hands completed packets off to a callback.rVNcCs||_t|_dSrH) handle_pktrr[)r\rrrrr^3s zPktLineParser.__init__cCs|j||j}t|dkrdSt|dkrOt|ddd}|dkr2|d|dd}n|t|krH||d|||d}nnt|dkst|_|j|dS)zAParse a fragment of data and call back for any completed packets.rBNrfr)r[rXrr,rhrr)r\rCrrlrrrparse7s        zPktLineParser.parsecCs |jS)zRead back any unused data.)r[rr_rrrget_tailJs zPktLineParser.get_tailr)rrrrr^rrrrrrr0s    rcCsddd|DS)Nr6cSsg|]}d|qS)r5rr$rrrr1Pr2z*format_capability_line..r9r(rrrformat_capability_lineOr!rcCs4|dur |d|dS|d|dt|dS)Nr5 r7)r)refshar)rrrformat_ref_lineSsrcC td|SNr5)COMMAND_SHALLOWrrrrformat_shallow_lineZrercCrr)COMMAND_UNSHALLOWrrrrformat_unshallow_line^rerr6cCs|rd|}d||dS)Nr5sACK rr)rrrrrformat_ack_linebsrrH)r6)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#r4COMMAND_DEEPENrr COMMAND_DONE COMMAND_WANT COMMAND_HAVEr<rArDrGrTrUrrrrrrrrrrrrrrrrs             i  ,