�
1�3\c@@s
dd
lm
Z

ddlZddlZddlmZ
ddlmZmZm	Z	m
Z
mZmZm
Z

ddlmZmZ
d�Zd�Zd	ef
d
��YZd�Zdd�
Zd
�Zd�Zdejf
d��YZdef
d��YZed�
Zddd�ZdS(i(
tabsolute_importNi
(
t
_(terrortpycompattutiltwireprotoservertwireprototypestwireprotov1peertwireprotov1server(tprocutilt
stringutilc

C@s4|s
|Stj
d
|�r |Sd|jdd�S(s=quote a string for the remote shell ... which we assume is shs[a-zA-Z0-9@%_+=:,./-]*$s'%s't
's'\''(tretmatchtreplace(
t
s((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_serverquotes




cC@sU|
rQtj
|
�
}|rQx0|j�D]}|jtd
�
|d�
q(WqQndS(sYdisplay all data currently available on pipe as remote output.

    This is non blocking.sremote: s

N(R	treadpipet
splitlineststatusR
(tuitpipeRt
l((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_forwardoutput$s




t
doublepipecB@sbeZ
dZd
�Zd�Zd�Zd�Zd�Zd�Zd
d�
Z
d�Zd	�ZRS(s3Operate a side-channel pipe in addition of a main one

    The side-channel pipe contains server output to be forwarded to the user
    input. The double pipe will behave as the "main" pipe, but will ensure the
    content of the "side" pipe is properly processed while we wait for blocking
    call on the "main" pipe.

    If large amounts of data are read from "main", the forward will cease after
    the first bytes start to appear. This simplifies the implementation
    without affecting actual output of sshpeer too much as we rarely issue
    large read for data not yet emitted by the server.

    The main pipe is expected to be a 'bufferedinputpipe' from the util module
    that handle all the os specific bits. This class lives in this module
    because it focus on behavior specific to the ssh protocol.cC@s|
|_||_
||_dS(
N(t_uit_maint_side(tselfRtmaintside((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt__init__?s
	
	
c
C@s�t|j
tj�r+|j
jr+ttfS|j
j�|jj�g}
ytj|
�
}Wnt	k
ru


|
}n
X|j
j�|k|jj�|kfS(
s�wait until some data are available on main or side

        return a pair of boolean (ismainready, issideready)

        (This will only wait for data if the setup is supported by `util.poll`)
        (
t
isinstanceRRtbufferedinputpipet	hasbuffertTruetfilenoRtpolltNotImplementedError(Rtfdstact((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_waitDs







cC@s|jd
|
�S(Ntwrite(
t_call(Rtdata((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR*Xs
cC@s?|jd
|
�}|
dkr;|r;t
|j|j�
n|S(Ntreadi(R+RRR(Rtsizet
r((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR-[s


cC@s?|jd
|
�}|
dkr;|r;t
|j|j�
n|S(Ntunbufferedreadi(R+RRR(RR.R/((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR0fs


c

C@s
|jd
�
S(Ntreadline(
R+(
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR1qs
cC@s�|dk	r|s|j
jr6t|j|j�
d
Sxptr�|j�\}}|rmt|j|j�
n|r9t|j
|
�}|dkr�|�S||�
Sq9q9WdS(sKcall <methname> on "main", forward output of "side" while blocking
        tN(	tNoneRtclosedRRRR#R)tgetattr(RtmethnameR,t	mainreadyt	sidereadytmeth((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR+ts


	






c


C@s
|jj
�S(
N(Rtclose(
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR:�s
c


C@s
|jj
�S(
N(Rtflush(
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR;�s
N(
t__name__t
__module__t__doc__RR)R*R-R0R1R3R+R:R;(((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR.s							cC@s�|r|j�
n|
r&|
j�
n|r~y+x$|D]}|j
td
�
|�
q6WWnttfk
rp


n
X|j�
ndS(s)Clean up pipes used by an SSH connection.sremote: N(R:RR
tIOErrort
ValueError(RtpipeitpipeotpipeeR((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt
_cleanuppipes�s









cC@s�d
|
|tj
dt|�
t|�
f�
f}|jd|�

tj|�
}tj|ddd|�
\}}}	}
|
|||	fS(s�Create an SSH connection to a server.

    Returns a tuple of (process, stdin, stdout, stderr) for the
    spawned process.
    s%s %s %ss%s -R %s serve --stdiosrunning %s
tbufsizeitenv(R	t
shellquoteRtdebugtquotecommandtpopen4(Rtsshcmdtargst	remotecmdtpathtsshenvtcmdtstdintstdouttstderrtproc((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_makeconnection�s


	
 
'cC@sXd
h
}gtj
jtj�
D]}
|
j�j^q}|jddj|�
�

|S(swReturn list of capabilities of this client.

    Returns a list of capabilities that are supported by this client.
    spartial-pullscomp=%st
,(Rtcompenginestsupportedwireenginest
CLIENTROLEtwireprotosupporttnametaddtjoin(tprotoparamst
etcomps((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_clientcapabilities�s
	

.

c@s���
fd
�}�
jdd�}t
jttj��
�
}dtjfg
}tj	j
|�
}y�dddddf}dd	d
t|�
|g}	�
jdd�r��
jd
||f�

|	j
dd||f�
n|r

�
jd�

�
jdt|�
�

n�
jd�

�
jd�

|
jdj|	�
�

|
j�
Wntk
rb



|�
n
Xtj}
tjdtj|�
�
}ddg}d}
x�|drq|
rqy�|j�}t�
��
|j|�
}|r|jd�
}
�
jd|
�

Pn|ddkr"|dkr"Pn|r;�
jd|�
n|j|�

|
d8}
Wq�
tk
rm


|�
q�
Xq�
W|�
t�}|
tjkr�x�t|�
D]=}|jd�
r�|j |d j!d�
dj!��

Pq�q�Wn�|
tjkr�|j�}yt"|�
}Wnt#k
r'


|�
n
X|j$|�
}|jd �
sP|�
n�
jd!|�

|j |j!d�
dj!��

|j$d�

n|s�|�
nt�
��
|
|fS("Nc@sDt�
��
t
d
�
}�
jdd�}
tj|d|
�

�
dS(Ns#no suitable response from remote hgRtssherrorhintthint(RR
tconfigRt	RepoError(tmsgRc(RSR(s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pytbadresponse�s


tdevelsdebug.peer-requesttprotos%s-%st
0i(shello
sbetween
s	pairs %d
texperimentalssshpeer.advertise-v2ssending upgrade request: %s %s
isupgrade %s %s
s"devel-peer-request: hello+between
s&devel-peer-request:   pairs: %d bytes
ssending hello command
ssending between command
R2s^upgraded %s (.*)$tdummyi�
i����i
sprotocol upgraded to %s
s1
s

sremote: s
capabilities:t
:scapabilities: sremote: %s
(%t
configboolRtsysbyteststrtuuidtuuid4RtSSHV2Rturlreqt	urlencodetlenRHtinsertR*R]R;R?RtSSHV1RtcompileR
treescapeR1RR
tgrouptappendtsettreversedt
startswithtupdatetsplittintR@R-(RRQRRRSRgt
requestlogttokentupgradecapstpairsargt	handshaket	protonamet
reupgradedtlinest	max_noiseRt
mtcapstlinetvaluelentcapsline((RSRs7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_performhandshake�s�
<




	












	


















	

$









 


t	sshv1peercB@s�eZ
ed�
Zd
h
Zd�Zd�Zd�Zd�Zd�Z	d�Z
d�Zd	�Zd
�Z
e
Zed�
Zd�Zd
�Zd�Zd�Zd�Zd�Zd�Zed�
ZRS(c		C@s�||_|
|_
||_|rQt|
tj|�
|�}t|
||�}n||_||_||_||_	||_
d
S(s�
Create a peer from an existing SSH connection.

        ``proc`` is a handle on the underlying SSH process.
        ``stdin``, ``stdout``, and ``stderr`` are handles on the stdio
        pipes for that process.
        ``caps`` is a set of capabilities supported by the remote.
        ``autoreadstderr`` denotes whether to automatically read from
        stderr and to forward its output.
        N(t_urlRt_subprocessRRR!t_pipeot_pipeit_pipeet_capst_autoreadstderr(	RRturlRTRQRRRSR�tautoreadstderr((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyRo
s	
		

	
	
	
	
tbatchc


C@s|jS(
N(
R�(
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR��
s
c


C@sdS(
N(
R3(
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pytlocal�
s
c


C@s|S(
N((
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pytpeer�
s
c


C@stS(
N(
R#(
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pytcanpush�
s
c


C@sdS(
N((
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR:�
s
c


C@s|jS(
N(
R�(
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pytcapabilities�
s
c

C@st|j
|j�
dS(
N(RRR�(
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_readerr�
s
c
C@s|j�
|
�
dS(
N(
t_cleanup(Rt	exception((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_abort�
s


c

C@s#t|j
|j|j|j�
dS(
N(RDRR�R�R�(
R((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR��
s
cC@sS|jj
r�|jjd
d�r�|jj}d}|||
�

x�t|j��
D]�\}}t|t�s�||d|t|�
f�

qTxAt|j��
D]-\}}	||d||t|	�
f�

q�WqTWn|jjd|
�

|j	j
d|
�

tj|
\}
}|j
�}i}
x=|D]5}|dkrO
||
d<Pq/
|||
|<||=q/
Wx�t|
j��
D]�\}}|j	j
d	|t|�
f�

t|t�rx]|j�D]<\}}	|j	j
d	|t|	�
f�

|j	j
|	�

q�
Wq{
|j	j
|�

q{
W|j	j�
|rL|j�}tj|j|�S|jS(
NRhsdebug.peer-requestsdevel-peer-request: %s
s  %s: %d bytess  %s-%s: %dssending %s command
s%s
t
*s%s %d
(Rt	debugflagRnRHtsortedtitemsR tdictRvR�R*RtcommandsR�t	iteritemsR;t
_getamountRtcappedreaderR�(RRPRLtframedtdbgR�tkeytvaluetdktdvt_functnamestkeystwireargst
kt
vtamount((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_sendrequest�
sB







!
,













 


 




cK@s.tj
|�
}|j|
|d
|
|jk�
S(NR�(RtbyteskwargsR�t_FRAMED_COMMANDS(RRPRL((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_callstream�
s

cK@s.tj
|�
}|j|
|d
|
|jk�
S(NR�(RR�R�R�(RRPRL((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_callcompressable�
s

cK@s+tj
|�
}|j|
|d
t�
j�S(NR�(RR�R�R#R-(RRPRL((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR+�
s

c@s�|j|
|�
}|r"d
|fSx-t
�f
d�d
�D]}|j|�

q;W|jd
dt�


|j�}|r�d
|fS|j�d
fS(NR2c@s
�jd
�
S(Ni(
R-((
tfp(s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt<lambda>�
R2R;(R+titert_writeframedR#t_readframed(RRPR�RLR/t
d((
R�s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt	_callpush�
s








c@s~|j|
|�
}|r4t
jtd
�
|�
�
nx-t�f
d�d�D]}|j|�

qMW|jddt�


|jS(Nsunexpected remote reply: %sc@s
�jd
�
S(Ni(
R-((
R�(s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR�R2R2R;(R+RtAbortR
R�R�R#R�(RRPR�RLR/R�((
R�s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt_calltwowaystreams


c
C@s�|jj
�}
|
d
krY|jr1|j�
ntd�
}|jtjd|�
�

n|jro|j�
nyt|
�
SWn0t	k
r�


|jtj
td�
|
��

n
XdS(Ns

scheck previous remote outputRcsunexpected response:(R�R1R�R�R
R�RtOutOfBandErrorR�R@t
ResponseError(RRRf((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR�s


	




	






c
C@s&|j�}
|
sd
S|j
j|
�
S(NR2(R�R�R-(RR.((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR�#s


cC@sc|jj
d
t|
�
�

|
r3|jj
|
�

n|rI|jj�
n|jr_|j�
ndS(Ns%d
(R�R*RvR;R�R�(RR,R;((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR�*s





	
(R<R=R#RR�R�R�R�R�R:R�R�R�R�t__del__tFalseR�R�R�R+R�R�R�R�R�(((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR�n
s*										,							t	sshv2peerc
B@seZ
dZRS(
s9A peer that speakers version 2 of the transport protocol.(R<R=R>(((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyR�3s
c	
C@s�yt||||�\}}Wn't
k
rH


t||||�
�n
X|tjkrzt||
|||||d
|�
S|tjkr�t||
|||||d
|�
St||||�
tj	t
d�
|�
�
dS(s�
Make a peer instance from existing pipes.

    ``path`` and ``proc`` are stored on the eventual peer instance and may
    not be used for anything meaningful.

    ``stdin``, ``stdout``, and ``stderr`` are the pipes connected to the
    SSH server's stdio handles.

    This function is factored out to allow creating peers that don't
    actually spawn a new process. It is useful for starting SSH protocol
    servers and clients via non-standard means, which can be useful for
    testing.
    R�s#unknown version of SSH protocol: %sN(R�t	ExceptionRDRRxR�RsR�RReR
(	RRNRTRQRRRSR�R�R�((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pytmakepeer9s











cC@sktj
|
d
tdt�
}|jdksC|jsC|jdkr_tjt	d�
|
�
�
ntj
|
�

|jdk	r�tjt	d�
�
�
n|jdd�}|jdd�}t
|jd�
�
}tj|�
}	|jp�d	}
tj||j|j|j�}|r�
|r/
tjt	d
�
�
�
nd||tjdt|�
t|
�
f�
f}|jd
|�

|j|ddd|	�
}
|
dkr�
tjt	d�
�
�
q�
nt|||||
|	�\}}}}t||
||||�}d|j�krgy)|jdddjtt��
�
�


Wqgtk
rc


|j �
tjt	d�
�
�
qgXn|S(siCreate an SSH peer.

    The returned object conforms to the ``wireprotov1peer.wirepeer`` interface.
    t
parsequeryt
parsefragmenttsshscouldn't parse location %sspassword in URL not supportedRRMROt
.s8cannot create remote SSH repositories with extra optionss%s %s %ss
%s init %ssrunning %s
t
blockedtagtsshpeertenvironiscould not create remote repot	protocapsR�t
 scapability exchange failedN(!RR�R�tschemethostRNR3RReR
tchecksafesshtpasswdRdR�tconfigitemsR	tshellenvirontsshargstusertportRGRRHtsystemRUR�R�R+R]R�RaR?R�(RRNtcreatetintentst
createoptst
uRKRMt	sshaddenvROt
remotepathRLRPtresRTRQRRRSR�((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pytinstanceXsB
(






!
	
	
 












( t
__future__RRRqti18nR
R2RRRRRRRtutilsR	R
RRtobjectRRDR3RURaR�twirepeerR�R�R#R�R�(((s7/usr/lib64/python2.7/site-packages/mercurial/sshpeer.pyt<module>s 

4			
^			��