| Server IP : 80.87.202.40 / Your IP : 216.73.216.169 Web Server : Apache System : Linux rospirotorg.ru 5.14.0-539.el9.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Dec 5 22:26:13 UTC 2024 x86_64 User : bitrix ( 600) PHP Version : 8.2.27 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /usr/lib64/python3.9/site-packages/mercurial/__pycache__/ |
Upload File : |
a
�+�b@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 �7/usr/lib64/python3.9/site-packages/mercurial/sshpeer.py� _serverquote s
r Fc C sJ |rF|j sFt�|�}|rF|r"| jn| j}|�� D ]}|td�|d� q0dS )zYdisplay all data currently available on pipe as remote output.
This is non blocking.� remote: �
N)�closedr Zreadpipe�warn�status�
splitlinesr )�ui�piper r Zdisplay�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 �mainZsider r r �__init__D s zdoublepipe.__init__c C sp t | jtj�r| jjrdS | j�� | j�� g}zt�|�}W n tyR |}Y n0 | 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 �bufferedinputpipeZ hasbuffer�filenor% Zpoll�NotImplementedError)r& ZfdsZactr r r �_waitI s ��
zdoublepipe._waitc C s | � d|�S )Ns write��_call)r& �datar r r �write_ s zdoublepipe.writec C s* | � d|�}|dkr&|s&t| j| j� |S )Ns readr �r/ r r# r% �r& �size�rr r r �readb s zdoublepipe.readc C s* | � d|�}|dkr&|s&t| j| j� |S )Ns unbufferedreadr r2 r3 r r r �unbufferedreadm s zdoublepipe.unbufferedreadc C s
| � d�S )Ns readliner. �r& r r r �readlinex s zdoublepipe.readlineNc C sp |dur |r| j jr&t| j| j� dS | �� \}}|rDt| j| j� |r&t| j |�}|du rb|� S ||�S q&dS )zBcall <methname> on "main", forward output of "side" while blockingN� )r$ r r r# r% r- r )r& Zmethnamer0 Z mainreadyZ sideready�methr r r r/ { s zdoublepipe._callc C s
| j �� S r" )r$ �closer8 r r r r< � s zdoublepipe.closec C s | j jS r" )r$ r r8 r r r r � s zdoublepipe.closedc C s
| j �� S r" )r$ �flushr8 r r r r= � s zdoublepipe.flush)N)�__name__�
__module__� __qualname__�__doc__r( r- r1 r6 r7 r9 r/ r< �propertyr r= r r r r r! 3 s
r! c C s� d}|r|j sd}|�� |r0|j s0d}|�� |r||j s|d}z |D ]}| �td�|� qDW n ttfyr Y n0 |�� |r�|dur�| �d| � dS )z)Clean up pipes used by an SSH connection.FTr Ns. missing close on SSH connection created at:
%s)r r< r r �IOError�
ValueErrorZ develwarn)r ZpipeiZpipeoZpipeer Z didsomethingr r r r �
_cleanuppipes� s"
rE 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 �debugZpopen4) r �sshcmd�args� remotecmd�path�sshenv�cmd�stdin�stdout�stderr�procr r r �_makeconnection� s ��� rV 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 )Zwireprotosupport�name)�.0�er r r �
<listcomp>� s �z'_clientcapabilities.<locals>.<listcomp>s comp=%s� ,)r Z compenginesZsupportedwireenginesZ
CLIENTROLE�add�join)Z protoparams�compsr r r �_clientcapabilities� s �r_ c s � �fdd�}�� dd�}t�tt�� ��}zldd }ddd t|� |g}|rj��d
� ��d t|� � ��d � ��d
� |�d� |�� |�
� W n t y� |� Y n0 t j
} t�dt�|� �}
ddg} d} | d �r�| �r�z�|�� }
t�� dd� |
�|
�}|�r.|�d�} ��d| � W �q�| d dk�rL|
dk�rLW �q�|
�r^��d|
� | �|
� | d8 } W q� t �y� |� Y q�0 q�|� t� }| t j
k�r�t| �D ]8}
|
�d��r�|�|
d d� �d�d �� � �q�q�|�s�|� t�� dd� | |fS ) Nc s4 t �� dd� td�} ��dd�}tj| |d��d S )NT�r s# no suitable response from remote hg� uis ssherrorhint��hint)r r �configr � RepoError)�msgrc �rT r r r � badresponse� s z&_performhandshake.<locals>.badresponse� devel� debug.peer-requests %s-%s)�( 0000000000000000000000000000000000000000rk 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
r: s ^upgraded %s (.*)$s dummyi� ���Tr` r s protocol upgraded to %s
s 1
r r s
capabilities:� :) �
configboolr Zsysbytes�str�uuidZuuid4�lenrK r1 r] r= rC r �SSHV1r �compiler
Zreescaper9 r r �group�append�set�reversed�
startswith�update�split)r rR rS rT rh Z
requestlog�tokenZpairsargZ handshake� protonameZ
reupgraded�linesZ max_noiser �m�capsr rg r �_performhandshake� sf @
�
r� c @ s� e Zd Zd-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*� Zd0d+d,�ZdS )1� sshv1peerTc C sh || _ || _|| _|r4t|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.
r: r N)�_urlr Z _subprocessr! r r* �_pipeo�_pipei�_pipee�_caps�_autoreadstderrr] Zgetstackframes�
_initstack) r& r �urlrU rR rS rT r �autoreadstderrr r r r( w s zsshv1peer.__init__s batchc C s | j S r" )r� r8 r r r r� � s z
sshv1peer.urlc C s d S r" r r8 r r r �local� s zsshv1peer.localc C s | S r" r r8 r r r �peer� s zsshv1peer.peerc C s dS )NTr r8 r r r �canpush� s zsshv1peer.canpushc C s | � � d S r" ��_cleanupr8 r r r r< � s zsshv1peer.closec C s | j S r" )r� r8 r r r � capabilities� s zsshv1peer.capabilitiesc C s t | j| j� d S r" )r r r� r8 r r r �_readerr� s zsshv1peer._readerrc C s | � � |�d S r" r� )r& � exceptionr r r �_abort� s zsshv1peer._abortNc C s t | j| j| j| j|d� d S �Nr` )rE r r� r� r� )r& r r r r r� � s zsshv1peer._cleanupc C s | j | jd� d S r� )r� r� r8 r r r �__del__� s zsshv1peer.__del__Fc C s� | j jr�| j �dd�r�| j j}d}||| � t|�� �D ]\\}}t|t�sf||d |t|�f � q:t|�� �D ]"\}} ||d ||t| �f � qrq:| j �d| � | j �
d| � t j | \}
} | �
� } i }
| D ].}|dkr�||
d< �qq�|| |
|< ||= q�tt�|
��D ]v\}}| j �
d |t|�f � t|t��r|t�|�D ].\}} | j �
d |t| �f � | j �
| � �qJn | j �
|� �q| j �� |�r�| �� }t�| j|�S | jS )
Nri rj s devel-peer-request: %s
s %s: %d bytess %s-%s: %ds sending %s command
s %s
� *s %s %d
)r Z debugflagrn rK �sorted�itemsr) �dictrq r� r1 r Zcommandsrz r Z iteritemsr= �
_getamountr Z cappedreaderr� )r& rQ rM �framedZdbg�line�key�valueZdkZdvZ_func�names�keysZwireargs�k�v�amountr r r � _sendrequest� sD �
zsshv1peer._sendrequestc K s t �|�}| j|||| jv d�S �N�r� �r � byteskwargsr� �_FRAMED_COMMANDS�r& rQ rM r r r � _callstream� s
zsshv1peer._callstreamc K s t �|�}| j|||| jv d�S r� r� r� r r r �_callcompressable� s
zsshv1peer._callcompressablec K s t �|�}| j||dd��� S )NTr� )r r� r� r6 r� r r r r/ � s
zsshv1peer._callc sn | j |fi |��}|r d|fS t� fdd�d�D ]}| �|� q0| jddd� | �� }|rbd|fS | �� dfS )Nr: c s
� � d�S �Ni �r6 r ��fpr r �<lambda> r: z%sshv1peer._callpush.<locals>.<lambda>T�r= )r/ �iter� _writeframed� _readframed�r& rQ r� rM r5 �dr r� r � _callpush� s zsshv1peer._callpushc s^ | j |fi |��}|r(t�td�| ��t� fdd�d�D ]}| �|� q:| jddd� | jS )Ns unexpected remote reply: %sc s
� � d�S r� r� r r� r r r� r: z-sshv1peer._calltwowaystream.<locals>.<lambda>r: Tr� )r/ r ZAbortr r� r� r� r� r r� r �_calltwowaystream s zsshv1peer._calltwowaystreamc C s� | j �� }|dkr:| jr | �� td�}| �tj|d�� | jrH| �� z
t|�W S t yz | �t�
td�|�� Y n0 d S )Nr s check previous remote outputrb s unexpected response:) r� r9 r� r� r r� r ZOutOfBandError�intrD Z
ResponseError)r& r rf r r r r� ! s
zsshv1peer._getamountc C s | � � }|sdS | j�|�S )Nr: )r� r� r6 )r& r4 r r r r� / s zsshv1peer._readframedc C sD | j �dt|� � |r$| j �|� |r2| j �� | jr@| �� d S )Ns %d
)r� r1 rq r= r� r� )r& r0 r= r r r r� 6 s
zsshv1peer._writeframed)T)N)F)F)r> r? r@ r( r� r� r� r� r� r< r� r� r� r� r� r� r� r� r/ r� r� r� r� r� r r r r r� v s. �
"�
-r� Tc
C s� zt | |||�\}}W n& ty< t| |||dd� � Y n0 |tjkr`t| |||||||d�S t| |||dd� t�td�| ��dS )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.
Nr` )r� s# unknown version of SSH protocol: %s) r� � ExceptionrE r rr r� r re r ) r rO rU rR rS rT 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
�|�} |jp�d
}
t
�||j|j|j�} |�r |r�t�td ���d || t
�d
t|�t|
�f �f } | �d| � | j| d| d�}
|
dk�r t�td���t| || ||
| �\}}}}t| |||||�}d|�� v �r�z |jdd�tt � ��d� W n* t �y� |� � t�td���Y n0 |S )ziCreate an SSH peer.
The returned object conforms to the ``wireprotov1peer.wirepeer`` interface.
F)Z
parsequeryZ
parsefragments sshNs couldn't parse location %ss password in URL not supportedra s remotecmds sshenv� .s8 cannot create remote SSH repositories with extra optionsrF s
%s init %srG s sshpeer)Z
blockedtag�environr s could not create remote repos protocaps� )r s capability exchange failed) r r� �scheme�hostrO r re r Z checksafessh�passwdrd r� Z configitemsr Z shellenvironZsshargs�user�portrJ r rK �systemrV r� r� r/ r] r� r_ rC r� )r rO ZcreateZintentsZ
createopts�urL rN Z sshaddenvrP Z
remotepathrM rQ �resrU rR rS rT r� r r r �instancef sZ
�����
� �
r� )F)N)T)NN) Z
__future__r r rp Zi18nr r r � r r r r
r Zutilsr r
r r r �objectr! rE rV r_ r� Zwirepeerr� r� r� r r r r �<module> s$
d
K
&