/
lib64
/
python3.9
/
site-packages
/
mercurial
/
__pycache__
/
Upload File
HOME
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 &