File: //usr/lib/python3/dist-packages/mercurial/__pycache__/wireprotoserver.cpython-310.pyc
o
�]Lb�E � @ s( d dl mZ d dlZd dlZd dlZddlmZ ddlmZm Z m
Z
m Z m Z m
Z
ddlm Z ddlmZmZ e jZe jZe jZdZd Zd
Zd Ze jZd d
� Ze� e j �G dd� de ��Z dd� Z dd� Z!dd� Z"dd� Z#dd� Z$dd� Z%d d � Z&e� e j �G d d � d e ��Z'd d!� Z(G d"d#� d#e �Z)dS )$� )�absolute_importN� )�_)�encoding�error�pycompat�util�wireprototypes�wireprotov1server)r )� compression�
stringutil�� s application/mercurial-0.1s application/mercurial-0.2s application/hg-errorc C sL g }d} | j �d||f �}|du rn
|�t�|�� |d7 }qd�|�S )zjDecode a long value from multiple HTTP request headers.
Returns the value as a bytes, not a str.
r Ts %s-%dN� )�headers�get�appendr �bytesurl�join)�req� headerprefix�chunks�i�v� r �;/usr/lib/python3/dist-packages/mercurial/wireprotoserver.py�decodevaluefromheaders* s �
r c @ sf e Zd Zdd� Zedd� �Zdd� Zdd� Zd d
� Zd d � Z e
j d
d� �Z dd� Z
dd� Zdd� ZdS )�httpv1protocolhandlerc C s || _ || _|| _d | _d S �N)�_req�_ui�
_checkperm�
_protocaps)�selfr �ui� checkpermr r r �__init__= s
z httpv1protocolhandler.__init__c C s dS )Ns http-v1r �r" r r r �nameC s zhttpv1protocolhandler.namec s� | � � }i � |�� }|D ],}|dkr0i }|�� D ]}|dkr*||vr*|| d ||< q|� d< q || d � |< q � fdd�|D �S )N� *� cmdr c � g | ]}� | �qS r r ��.0�k��datar r �
<listcomp>T � z1httpv1protocolhandler.getargs.<locals>.<listcomp>)�_args�split�keys)r" �args� knownargsr4 r- �star�keyr r. r �getargsG s �
z httpv1protocolhandler.getargsc C sj | j j�� }t| j j�dd��}|r#|�tj| j j �
|�dd�� |S t | j d�}|�tj|dd�� |S )N�
X-HgArgs-Postr T)�keep_blank_valuess X-HgArg) r �qsparams�
asdictoflists�intr r �update�urlreq�parseqs�bodyfh�readr )r" r5 �postlen�argvaluer r r r2 V s �� zhttpv1protocolhandler._argsc C s, | j d u rt| jd�}t|�d��| _ | j S )Ns X-HgProto� )r! r r �setr3 )r" �valuer r r � getprotocapse s
z"httpv1protocolhandler.getprotocapsc C s: t | jjd �}|t | jj�dd��8 }tj| jj|d�S )Ns Content-Lengthr: r )�limit)r> r r r r �
filechunkiterrB )r" �lengthr r r �
getpayloadk s z httpv1protocolhandler.getpayloadc c sZ � | j j}| j j}t�� }z|| j _|| j _|V W || j _|| j _d S || j _|| j _w r )r �fout�ferrr �stringio)r" �oldout�olderr�outr r r �mayberedirectstdiot s � �
z(httpv1protocolhandler.mayberedirectstdioc C s. d| j jt�| j jp d�t�| j jpd�f S )Ns remote:%s:%s:%sr )r � urlschemer@ �quote�
remotehost�
remoteuserr& r r r �client� s
�z httpv1protocolhandler.clientc C s~ |� d� |� d|j�dd� � |j�dd�r |� d� |� d� t�|jtj�}|r=d�d d
� |D ��}|� d | � |S ) N� batchs
httpheader=%d� servers maxhttpheaderlens experimentals httppostargss httpmediatype=0.1rx,0.1tx,0.2tx� ,c s s � | ] }t �|�� j�V qd S r )r@ rV �wireprotosupportr' )r, �er r r � <genexpr>� s �
�z8httpv1protocolhandler.addcapabilities.<locals>.<genexpr>s compression=%s) r r# � configint�
configboolr �supportedcompenginesr �
SERVERROLEr )r" �repo�caps� compengines� comptypesr r r �addcapabilities� s
�
�
�z%httpv1protocolhandler.addcapabilitiesc C s
| � |�S r )r �r" �permr r r r$ � �
z httpv1protocolhandler.checkpermN)�__name__�
__module__� __qualname__r% �propertyr' r9 r2 rI rM �
contextlib�contextmanagerrT rY rh r$ r r r r r ; s
r c C s
| t jv S r )r
�commands)�cmdr r r �iscmd� rk rt c
s ddl m} �j}d�jvrdS �jd }t|�sdS �jr2|�d�|_t|j d< |�
dd � d S t �|j � ��fd
d ��}z t
|�|||� W d S |jy� } z)|j D ] \} }
|
|j | < qV|�|jt�|��|_|�
dt�|� � W Y d }~d S d }~ww )
a� Possibly process a wire protocol request.
If the current request is a wire protocol request, the request is
processed by this function.
``req`` is a ``parsedrequest`` instance.
``res`` is a ``wsgiresponse`` instance.
Returns a bool indicating if the request was serviced. If set, the caller
should stop processing the request, as a response has already been issued.
r ��commonr) Fi� � Content-Type� 0
%s
s Not FoundTc s � ��| �S r r )rj �r$ �rctxr r r �<lambda>� s z#handlewsgirequest.<locals>.<lambda>N)�hgwebrv rd r< rt � dispatchpath�
statusmessage�status�HGTYPEr � setbodybytesr r# � _callhttp�
ErrorResponse�coder � forcebytestr) rz r �resr$ � hgwebcommonrd rs �protor^ r- r r ry r �handlewsgirequest� s8
� � � ��r� c C s� d|� � v r@|rtt�� i fS t�|�}t�| tj�D ]$}|� � j
|v r?i }| � dd|�
� �}|dur8||d< t||f S qd| � dd�i}t t
jd |fS )z�Determine the appropriate response type and compression settings.
Returns a tuple of (mediatype, compengine, engineopts).
s 0.2r[ s %slevelNs levels zliblevels zlib)rI �HGTYPE2r � _noopenginer
�clientcompressionsupportr rb rc r] r' r` r� r rf )r# r� �prefer_uncompressed� compformats�engine�opts�levelr r r �_httpresponsetype� s
��r� c s� ddl m� dd� }d� �fdd� }tj�||�s#|tttd�� d S |�tj| j � t�
| ||�}t |t �rA|tt
|d� d S t |tj�rQ|tt
|jd� d S t |tj�ra|tt
|jd � d S t |tj�r�|j}t| j||j�\} }
} |
�|| �}| tkr�|||
| �}|t| |d � d S t |tj�r�d
|j|jf }|tt
|d� d S t |tj �r�d |j }d �_ |tt
|d� d S t |tj �r�|tt|j d� d S t �!d
|��)Nr ru c s sX � |� � j}dt|� k rdk sJ � J �t�dt|��V |V | D ]}|V q$d S )Nr � � B)r] r' �len�struct�pack)�genr� �
engineoptsr' �chunkr r r � genversion2 s �
�z _callhttp.<locals>.genversion2c sR | t krd�_n� �| ��_|�jd< |d ur ��|� |d ur'��|� d S d S )Ns 200 Script output followsrw )�HTTP_OKr r~ r r� �
setbodygen)r� � contenttype� bodybytes�bodygen�r� r� r r � setresponse' s
�z _callhttp.<locals>.setresponses: requested wire protocol command is not available over HTTP)r� )r� s %d
%srx Ts hgweb.protocol internal failure)NN)"r| rv r
rr �commandavailabler� � HGERRTYPEr r$ �
permission�dispatch�
isinstance�bytesr� r �
bytesresponser/ �streamreslegacyr� � streamresr� r# r� �compressstreamr� �pushresr� �output�pusherr�drain�ooberror�messager �ProgrammingError) rd r r� r� rs r� r� �rspr� � mediatyper� r� r r� r r� sN
��
�
r� c C s( | � dt|� � | � |� | �� dS )z-Send a bytes response for protocol version 1.s %d
N)�writer� �flush)rN rH r r r �_sshv1respondbytesd s
r� c C s& | j }|jD ]}||� q| �� d S r )r� r� r� )rN �sourcer� r� r r r �_sshv1respondstreamk s
r� c C s, |� d| � |�� | � d� | �� d S )Ns %s
-
�
)r� r� )rN rO r� r r r �_sshv1respondooberrorr s
r� c @ sb e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd
d � Z e
j d d
� �Z dd� Z
dd� Zdd� ZdS )�sshv1protocolhandlerz<Handler for requests services via version 1 of SSH protocol.c C s || _ || _|| _t� | _d S r )r �_fin�_foutrG r! )r" r# �finrN r r r r% } s z sshv1protocolhandler.__init__c C s t jS r )r �SSHV1r& r r r r' � s zsshv1protocolhandler.namec
s� i � |� � }t�t|��D ]Z}| j�� d d� }|� � \}}||vr+t�td�| ��|dkr[i }t�t |��D ] }| j�� d d� }|� � \}}| j�
t |��} | ||< q8|� d< q
| j�
t |��} | � |< q
� fdd�|D �S )N���s unexpected parameter %rr( c r* r r r+ r. r r r0 � r1 z0sshv1protocolhandler.getargs.<locals>.<listcomp>) r3 r �xranger� r� �readliner �Abortr r> rC )
r" r5 r4 �n�argline�arg�lr7 r- �valr r. r r9 � s$
z sshv1protocolhandler.getargsc C s | j S r )r! r& r r r rI � s z!sshv1protocolhandler.getprotocapsc c sH � t | jd� t| j�� �}|r"| j�|�V t| j�� �}|sd S d S )Nr )r� r� r> r� r� rC )r" �countr r r rM � s � �z sshv1protocolhandler.getpayloadc c s � d V d S r r r& r r r rT � s �
z'sshv1protocolhandler.mayberedirectstdioc C s" t j�dd��dd�d }d| S )Ns
SSH_CLIENTr rF r r s remote:ssh:)r �environr r3 )r" rY r r r rY � s zsshv1protocolhandler.clientc C s$ | j tjkr |�d� |�d� |S )Ns protocapsrZ )r' r r� r )r" rd re r r r rh � s
z$sshv1protocolhandler.addcapabilitiesc C s d S r r ri r r r r$ � s z sshv1protocolhandler.checkpermN)rl rm rn �__doc__r% ro r' r9 rI rM rp rq rT rY rh r$ r r r r r� y s
r� c
C sv d}t | ||�}|�� s�|dkr�|�� d d� }|s d}qtj�||�}|s,t|d� qt�|||�} |jj �
� |jj �
� t | t
�rJt|| � nit | tj�rWt|| j� n\t | tj�rct|| � nPt | tj�rot|| � nDt | tj�r�t|d� t|d| j � n0t | tj�r�t|| j� n#t | tj�r�t|| j | j� nt�d| ��|dkr�d S t�d| ��|�� r d S d S )Ns protov1-servingr� s shutdownr s %ds6 unhandled response type from wire protocol command: %ss unhandled ssh server state: %s) r� �is_setr� r
rr r� r� r� r# rN r� rO r� r� r r� r/ r� r� r� r� r� r� r� r� r� r r� )
r# rd r� rN �ev�stater� �request� availabler� r r r �
_runsshserver� sV �
����r� c @ s& e Zd Zddd�Zdd� Zdd� ZdS ) � sshserverNc C sT || _ || _|�� \| _| _|r(tj|| jddd�| _tj||jddd�|_d S d S )N� oT)�logdata� e)r �_repo�
protectfinoutr� r� r �makeloggingfileobjectrO )r" r# rd �logfhr r r r%
s
�
��zsshserver.__init__c C s$ | � t�� � | j�| j| j� d S r )�
serveuntil� threading�Eventr �
restorefinoutr� r� r& r r r �
serve_forever s zsshserver.serve_foreverc C s t | j| j| j| j|� dS )z%Serve until a threading.Event is set.N)r� r r� r� r� )r" r� r r r r� s zsshserver.serveuntilr )rl rm rn r% r� r� r r r r r� s
r� )*�
__future__r rp r� r� �i18nr � r r r r r r
�
interfaces�
interfaceutil�utilsr r rP �urlerrr@ r� r� r� r� r� r � implementer�baseprotocolhandler�objectr rt r� r� r� r� r� r� r� r� r� r r r r �<module> s<
nF$L
GK