|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/mercurial/__pycache__/ |
Upload File : |
o
�]Lb@Z � @ s� d dl mZ d dlZd dlZddlmZ ddlmZ ddlm Z mZm
Z
m Z m Z m
Z
ddlmZmZmZ dd � Zd d d �ZG d
d� de�Zdd� Zd dd�Zdd� Zdd� ZG dd� de j�Zd dd�Z d!d d �Z dS )"� )�absolute_importN� )�_)�getattr)�error�pycompat�util�wireprototypes�wireprotov1peer�wireprotov1server)�procutil�
stringutil�urlutilc C s( | s| S t �d| �r | S d| �dd� S )z=quote a string for the remote shell ... which we assume is shs [a-zA-Z0-9@%_+=:,./-]*$s '%s'� 's '\'')�re�match�replace)�s� r �3/usr/lib/python3/dist-packages/mercurial/sshpeer.py� _serverquote s
r Fc C sV |r#|j s%t�|�}|r'|r| jn| j}|�� D ]}|td�|d� qdS dS dS dS )zYdisplay all data currently available on pipe as remote output.
This is non blocking.� remote: �
N)�closedr �readpipe�warn�status�
splitlinesr )�ui�piper r �display�lr r r �_forwardoutput' s
�r" c @ sf e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z ddd�Z
dd� Z e dd� �Z
dd� ZdS )�
doublepipea3 Operate 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.c C s || _ || _|| _d S �N)�_ui�_main�_side)�selfr �main�sider r r �__init__D s
zdoublepipe.__init__c C sp t | jtj�r
| jjr
dS | j�� | j�� g}zt�|�}W n ty) |}Y nw | j�� |v | j�� |v fS )z�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`)
)TT) �
isinstancer& r �bufferedinputpipe� hasbuffer�filenor' �poll�NotImplementedError)r( �fds�actr r r �_waitI s �� � zdoublepipe._waitc C s | � d|�S )Ns write��_call)r( �datar r r �write_ � zdoublepipe.writec C �* | � d|�}|dkr|st| j| j� |S )Ns readr �r6 r" r% r' �r( �size�rr r r �readb � zdoublepipe.readc C r: )Ns unbufferedreadr r; r< r r r �unbufferedreadm r@ zdoublepipe.unbufferedreadc C s
| � d�S )Ns readliner5 �r( r r r �readlinex �
zdoublepipe.readlineNc C sn |dur|r
| j jrt| j| j� dS | �� \}}|r#t| j| j� |r6t| j |�}|du r2|� S ||�S q)zBcall <methname> on "main", forward output of "side" while blockingN� )r& r r" r% r' r4 r )r( �methnamer7 � mainready� sideready�methr r r r6 { s �zdoublepipe._callc C �
| j �� S r$ )r& �closerB r r r rK � rD zdoublepipe.closec C s | j jS r$ )r& r rB r r r r � s zdoublepipe.closedc C rJ r$ )r& �flushrB r r r rL � rD zdoublepipe.flushr$ )�__name__�
__module__� __qualname__�__doc__r+ r4 r8 r? rA rC r6 rK �propertyr rL r r r r r# 3 s
r# c C s� d}|r
|j s
d}|�� |r|j sd}|�� |r>|j s>d}z|D ]
}| �td�|� q"W n ttfy9 Y nw |�� |rM|durO| �d| � dS dS dS )z)Clean up pipes used by an SSH connection.FTr Ns. missing close on SSH connection created at:
%s)r rK r r �IOError�
ValueError� develwarn)r �pipei�pipeo�pipeer � didsomethingr! r r r �
_cleanuppipes� s(
�� �rY c C sV d||t �dt|�t|�f �f }| �d| � t j|d|d�\}}} }
|
||| fS )z�Create an SSH connection to a server.
Returns a tuple of (process, stdin, stdout, stderr) for the
spawned process.
� %s %s %ss %s -R %s serve --stdio� running %s
r )�bufsize�env)r �
shellquoter �debug�popen4) r �sshcmd�args� remotecmd�path�sshenv�cmd�stdin�stdout�stderr�procr r r �_makeconnection� s ��� rk c C s6 dh} dd� t j�t j�D �}| �dd�|� � | S )zwReturn list of capabilities of this client.
Returns a list of capabilities that are supported by this client.
s partial-pullc S s g | ]}|� � j�qS r )�wireprotosupport�name)�.0�er r r �
<listcomp>� s ��z'_clientcapabilities.<locals>.<listcomp>s comp=%s� ,)r � compengines�supportedwireengines�
CLIENTROLE�add�join)� protoparams�compsr r r �_clientcapabilities� s �ry c s� � �fdd�}�� dd�}t�tt�� ��}z6dd }ddd t|� |g}|r5��d
� ��d t|� � ��d � ��d
� |�d� |�� |�
� W n t yX |� Y nw t j
} t�dt�|� �}
ddg} d} | d r�| r�z?|�� }
t�� dd� |
�|
�}|r�|�d�} ��d| � W n4| d dkr�|
dkr�W n(|
r���d|
� | �|
� | d8 } W n t y� |� Y nw | d r�| sr|� t� }| t j
kr�t| �D ]}
|
�d�r�|�|
d d� �d�d �� � nq�|s�|� t�� dd� | |fS ) Nc s0 t �� dd� td�} ��dd�}tj| |d��)NT�r s# no suitable response from remote hg� uis ssherrorhint��hint)r" r �configr � RepoError)�msgr} �ri r r r � badresponse� s z&_performhandshake.<locals>.badresponse� devel� debug.peer-requests %s-%s)�( 0000000000000000000000000000000000000000r� s hello
s between
s pairs %d
s" devel-peer-request: hello+between
s& devel-peer-request: pairs: %d bytes
s sending hello command
s sending between command
rE s ^upgraded %s (.*)$s dummyi� ���Trz r s protocol upgraded to %s
s 1
r r s
capabilities:� :) �
configboolr �sysbytes�str�uuid�uuid4�lenr_ r8 rv rL rR r �SSHV1r �compiler
�reescaperC r" r �group�append�set�reversed�
startswith�update�split)r rg rh ri r� �
requestlog�token�pairsarg� handshake� protoname�
reupgraded�lines� max_noiser! �m�capsr r� r �_performhandshake� sn @
�
�
� �
�
r� c @ s� e Zd Z d-dd�ZdhZdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z d.dd�Z
dd� Zd/dd �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd/d+d,�ZdS )0� sshv1peerTc C sh || _ || _|| _|rt|t�|�|�}t|||�}|| _|| _|| _|| _ || _
d� t� d��| _
dS )a� 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.
rE r N)�_urlr � _subprocessr# r r- �_pipeo�_pipei�_pipee�_caps�_autoreadstderrrv �getstackframes�
_initstack) r( r �urlrj rg rh ri r� �autoreadstderrr r r r+ w s zsshv1peer.__init__s batchc C � | j S r$ )r� rB r r r r� � � z
sshv1peer.urlc C s d S r$ r rB r r r �local� � zsshv1peer.localc C s | S r$ r rB r r r �peer� r� zsshv1peer.peerc C s dS )NTr rB r r r �canpush� r� zsshv1peer.canpushc C s | � � d S r$ ��_cleanuprB r r r rK � r9 zsshv1peer.closec C r� r$ )r� rB r r r � capabilities� r� zsshv1peer.capabilitiesc C s t | j| j� d S r$ )r" r r� rB r r r �_readerr� � zsshv1peer._readerrc C s | � � |�r$ r� )r( � exceptionr r r �_abort� s zsshv1peer._abortNc C s t | j| j| j| j|d� d S �Nrz )rY r r� r� r� )r( r r r r r� � s zsshv1peer._cleanupc C s | j | jd� d S r� )r� r� rB r r r �__del__� r� zsshv1peer.__del__Fc C s� | j jrL| j �dd�rL| j j}d}||| � t|�� �D ].\}}t|t�s3||d |t|�f � q t|�� �D ]\}} ||d ||t| �f � q9q | j �d| � | j �
d| � t j | \}
} | �
� } i }
| D ]}|dkrw||
d< n
|| |
|< ||= qktt�|
��D ]8\}}| j �
d |t|�f � t|t�r�t�|�D ]\}} | j �
d |t| �f � | j �
| � q�q�| j �
|� q�| j �� |r�| �� }t�| j|�S | jS )
Nr� r� s devel-peer-request: %s
s %s: %d bytess %s-%s: %ds sending %s command
s %s
� *s %s %d
)r � debugflagr� r_ �sorted�itemsr, �dictr� r� r8 r �commandsr� r � iteritemsrL �
_getamountr � cappedreaderr� )r( rf rb �framed�dbg�line�key�value�dk�dv�_func�names�keys�wireargs�k�v�amountr r r � _sendrequest� sH �
�
�
zsshv1peer._sendrequestc K � t �|�}| j|||| jv d�S �N�r� �r � byteskwargsr� �_FRAMED_COMMANDS�r( rf rb r r r � _callstream� �
zsshv1peer._callstreamc K r� r� r� r� r r r �_callcompressable� r� zsshv1peer._callcompressablec K s t �|�}| j||dd��� S )NTr� )r r� r� r? r� r r r r6 � s
zsshv1peer._callc sn | j |fi |��}|rd|fS t� fdd�d�D ]}| �|� q| jddd� | �� }|r1d|fS | �� dfS )NrE c �
� � d�S �Ni �r? r ��fpr r �<lambda> �
z%sshv1peer._callpush.<locals>.<lambda>T�rL )r6 �iter� _writeframed� _readframed�r( rf r� rb r> �dr r� r � _callpush� s zsshv1peer._callpushc s^ | j |fi |��}|rt�td�| ��t� fdd�d�D ]}| �|� q | jddd� | jS )Ns unexpected remote reply: %sc r� r� r� r r� r r r� r� z-sshv1peer._calltwowaystream.<locals>.<lambda>rE Tr� )r6 r �Abortr r� r� r� r� r r� r �_calltwowaystream s zsshv1peer._calltwowaystreamc C s~ | j �� }|dkr | jr| �� td�}| �tj|d�� | jr$| �� zt|�W S t y> | �t�
td�|�� Y d S w )Nr s check previous remote outputr| s unexpected response:) r� rC r� r� r r� r �OutOfBandError�intrS �
ResponseError)r( r! r� r r r r� ! s
�zsshv1peer._getamountc C s | � � }|sdS | j�|�S )NrE )r� r� r? )r( r= r r r r� / s zsshv1peer._readframedc C sH | j �dt|� � |r| j �|� |r| j �� | jr"| �� d S d S )Ns %d
)r� r8 r� rL r� r� )r( r7 rL r r r r� 6 s
�zsshv1peer._writeframed�Tr$ �F)rM rN rO r+ r� r� r� r� r� rK r� r� r� r� r� r� r� r� r6 r� r� r� r� r� r r r r r� v s0
�"�
-r� Tc
C s� z t | |||�\}}W n ty t| |||dd� � w |tjkr.t| |||||||d�S t| |||dd� t�td�| ��)a� 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.
Nrz )r� s# unknown version of SSH protocol: %s) r� � ExceptionrY r r� r� r r r ) r rd rj rg rh ri r� r� r� r r r �makepeer@ s* �
�
�r� c C s� t j|ddd�}|jdks|jr|jdu r t�td�| ��t �|� |j dur/t�td���| �
dd�}| �
dd�}t | � d ��}t
�|�} |jpKd
}
t
�||j|j|j�} |r�|rbt�td ���d || t
�d
t|�t|
�f �f } | �d| � | j| d| d�}
|
dkr�t�td���t| || ||
| �\}}}}t| |||||�}d|�� v r�z|jdd�tt � ��d� W |S t y� |� � t�td���w |S )ziCreate an SSH peer.
The returned object conforms to the ``wireprotov1peer.wirepeer`` interface.
F)�
parsequery�
parsefragments sshNs couldn't parse location %ss password in URL not supportedr{ s remotecmds sshenv� .s8 cannot create remote SSH repositories with extra optionsrZ s
%s init %sr[ s sshpeer)�
blockedtag�environr s could not create remote repos protocaps� )r� s capability exchange failed) r r� �scheme�hostrd r r r � checksafessh�passwdr~ r� � configitemsr � shellenviron�sshargs�user�portr^ r r_ �systemrk r� r� r6 rv r� ry rR r� )r rd �create�intents�
createopts�ura rc � sshaddenvre �
remotepathrb rf �resrj rg rh ri r� r r r �instancef s^
����� � � ��r r� r$ r� )NN) �
__future__r r r� �i18nr r r � r r r r
r �utilsr r
r r r" �objectr# rY rk ry r� �wirepeerr� r� r r r r r �<module> s&
d
K&