>GdZddlmZddlmZddlZddlZddlmZm Z dZ dZ dZ dZ d ZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&d Z'ee e!e#egZ(e)e(e"eeeee$eeee%e&g zZ*e)e(eeeegzZ+d!Z,d"Z-d#Z.d$Z/d%Z0d&Z1d'Z2dZ3d(Z4d)Z5d*Z6d+Z7Gd,d-e8Z9d.Z:d/Z;d0Z<Gd1d2e8Z=d3Z>Gd4d5e=Z?d6Z@d7ZAd8ZBGd9d:e8ZCGd;d>rc&td|DS)Nc38K|]}t|dyw)rN)parse_capability).0cs r z+extract_capability_names..|s<|!"1%|s)set capabilitiess rextract_capability_namesr({s <|< <rN)__name__ __module__ __qualname____doc__r:r=r@rrrr5r5sE  rr5c\|dzdj|Dcgc]}|dz c}zScc}w)N r)join)cmdargsas rformat_cmd_pktrLs0 :t!|jr|jyyr0)rZr<s rr@zProtocol.closes ;; KKM rc|Sr0rr<s r __enter__zProtocol.__enter__s rc$|jyr0)r@)r9exc_typeexc_valexc_tbs r__exit__zProtocol.__exit__s  rc|j |j}n|jj}d|_ |d}|s tt|d}|dk(r|jr|j ddy|jr|j |d||dz }t |dz|k7rt dt |dz|fz|S#tj$r}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'). NrTrr7z9Length of pkt read %04x does not match length prefix %04x) r\r7rintr[r+rsocketerror)r9r7sizestrsize pkt_contentses r read_pkt_linezProtocol.read_pkt_lines ?? "99D??''D"DO 1gG%''w#Dqy''((F3##$$T62q>L< 1$,&O<(1,d34 || &"1% % &sAC<)CC7' C22C7cf |j}|j|y#t$rYywxYw)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)rnrunread_pkt_line)r9 next_lines reofz Protocol.eofs< **,I Y'  s $ 00cd|j tdtt||_y)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\ ValueErrorrrVr9rUs rrpzProtocol.unread_pkt_lines* ?? &FG G!(4.1rc#`K|j}|r||j}|ryyw)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)rn)r9pkts r read_pkt_seqzProtocol.read_pkt_seqs1   "I$$&Cs)..c t|}|j||jr|jt|dyy#tj $r}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. r8N)rVr8r[r+rhrir)r9rPrms rwrite_pkt_linezProtocol.write_pkt_line"s] &D>D JJt ##$$SY8$|| &"1% % &sAAA* A%%A*c0Gddt}||S)z5Return a writable file-like object for this protocol.c$eZdZdZdZdZdZy))Protocol.write_file..ProtocolFilec ||_d|_yNr)_proto_offset)r9protos rr:z2Protocol.write_file..ProtocolFile.__init__5s#   rcv|jj||xjt|z c_yr0)rr8rr+rus rr8z/Protocol.write_file..ProtocolFile.write9s& !!$' D ) rc|jSr0)rr<s rr=z.Protocol.write_file..ProtocolFile.tell=s ||#rcyr0rr<s rr@z/Protocol.write_file..ProtocolFile.close@srN)rArBrCr:r8r=r@rrrr5r}4s ! * $ rr5)object)r9r5s r write_filezProtocol.write_file1s 6 D!!rcn|r3|jtt|g|ddz|dd}|r2yy)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)rzbytes bytearray)r9channelblobs rwrite_sidebandzProtocol.write_sidebandEs<   i &: ;d6El J KD ::d#DD ADy  Dy $9 G;tQiG G9 IIdO qLG36 xxzrc|dkDsJ|j}|j}|jdt|j}|j|||z }|s_|j |j }t ||k(r|St}|j||jd~||_|j|Sr) rr=rrrrr+rr8r7)r9rkrrrrrUs rrzReceivableProtocol.recvsaxxjj  H((* ::dnn-D4yD  )C IIdO HHQKDJxx~r) rArBrCrD _RBUFSIZEr:r7r __classcell__)rs@rrrls! "&ti";zrrcd|vr|gfS|jjd\}}||jjdfS)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 rGrF)rstripr*strip)textr's rextract_capabilitiesrsO DRx,,U3D, ,$$&,,T2 33rc|jjd}t|dkr|gfSdj|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 rFr Nr)rr*r+rH)r split_texts rextract_want_line_capabilitiesrsO$$T*J :Rx IIj!n %z!"~ 66rc6d|vrtSd|vrtStS)z.Extract the ack type from a capabilities list.r r )MULTI_ACK_DETAILED MULTI_ACK SINGLE_ACKr&s rack_typers$ ,!!  % rc$eZdZdZddZdZdZy)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. cL||_||_t|_d|_y)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)r9r8bufsizes rr:zBufferedPktLineWriter.__init__s"  Y  rcZt|}t|}|j|z|jz }|dk\r4||z }|jj |d||j nd}||d}|jj ||xjt|z c_y)z&Write data, wrapping it in a pkt-line.rN)rVr+rrrr8flush)r9rUrPline_lenoverrsaveds rr8zBufferedPktLineWriter.writes~t9||h&6 19tOE JJ  T&5\ * JJLEUV   E " rc|jj}|r|j|d|_t |_y)zFlush all data from the buffer.rN)rgetvaluer_lenrrus rrzBufferedPktLineWriter.flush"s3zz""$  KK  Y rN)r)rArBrCrDr:r8rrrrrrs  #rrc"eZdZdZdZdZdZy) PktLineParserzBPacket line parser that hands completed packets off to a callback.c0||_t|_yr0) handle_pktrr\)r9rs rr:zPktLineParser.__init__.s$!)rc|jj||jj}t|dkryt|dk\rct |ddd}|dk(r|j d|dd}n)|t|kr|j |d|||d}nnt|dk\rct |_|jj|y)zAParse a fragment of data and call back for any completed packets.rTNrfr)r\r8rr+rgrr)r9rUrrks rparsezPktLineParser.parse2s d#oo&&( s8a< #h!ms2Aw#Dqy%!"gS!Ad ,$%j#h!m") c"rc6|jjS)zRead back any unused data.)r\rr<s rget_tailzPktLineParser.get_tailEs''))rN)rArBrCrDr:rrrrrrr+sL$#&*rr)ErDiorosrrhrdulwich.errorsrr 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_CAPABILITIESr%KNOWN_UPLOAD_CAPABILITIESKNOWN_RECEIVE_CAPABILITIESrrrr(r!r3COMMAND_DEEPENCOMMAND_SHALLOWCOMMAND_UNSHALLOW COMMAND_DONE COMMAND_WANT COMMAND_HAVErr5rLrRrVrXrrrrrrrrrrrs,C    )%/''# 5'#+#+#$=!*I'&  %")/   !  F4?=4      6  >' =n#vn#b  hhV 47$)F)X*F*r