|
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�R � @ s( d dl mZ d dlZd dlZd dlZd dlZd dlZddlmZ ddl m
Z
ddl m Z m
Z
mZm Z mZmZmZmZ ddlmZ ejZejZejZdd � ZG d
d � d e�Zd d
� Z dd� Z dd� Z G dd� de
j �Z dd� Z!G dd� dej"�Z#G dd� de j$j%�Z&dd� Z'dej(fd d �Z)d d d �Z*dS )!� )�absolute_importN� )�_)�getattr)�bundle2�error�httpconnection�pycompat�statichttprepo�url�util�wireprotov1peer)�urlutilc
C sz t �|�d }|t|d � td� }g }d}t �dt| �|�D ]}|d7 }|�|t|� t �| ||| � �f� q |S )ac Encode a string value into multiple HTTP headers.
``value`` will be encoded into 1 or more HTTP headers with the names
``header-<N>`` where ``<N>`` is an integer starting at 1. Each header
name + value will be at most ``limit`` bytes long.
Returns an iterable of 2-tuples consisting of header names and
values as native strings.
z-%s�000s :
r r )r �strurl�len�xrange�append�str)�value�header�limit�fmt�valuelen�result�n�i� r �4/usr/lib/python3/dist-packages/mercurial/httppeer.py�encodevalueinheaders$ s *r c @ s8 e Zd Zdd� Zedd� �Zd
dd�Zejfdd �Z dS ) �
_multifilec G s6 |D ]}t �|d�stdt|�|��q|| _d| _d S )N� lengthsO _multifile only supports file objects that have a length but this one does not:r )r � safehasattr�
ValueError�type� _fileobjs�_index)�self�fileobjs�fr r r �__init__B s ��
z_multifile.__init__c C s t dd� | jD ��S )Nc s s � | ]}|j V qd S �N)�length��.0r) r r r � <genexpr>P s � z$_multifile.length.<locals>.<genexpr>)�sumr% �r' r r r r, N s z_multifile.lengthNc C s� |dkrd� dd� | jD ��S g }|rF| jt| j�k rF|�| j| j �|�� t|d �}||k r8| jd7 _||8 }|rF| jt| j�k sd� |�S )Nr � c s s � | ]}|� � V qd S r+ )�readr- r r r r/ T s � z"_multifile.read.<locals>.<genexpr>���r )�joinr% r& r r r3 )r' �amt�parts�gotr r r r3 R s �
z_multifile.readc C sB |t jkr td��|dkrtd��| jD ]}|�d� qd| _d S )NsP _multifile does not support anything other than os.SEEK_SET for whence on seek()r sQ _multifile only supports seeking to start, but that could be fixed if you need it)�os�SEEK_SET�NotImplementedErrorr% �seekr& )r' �offset�whencer) r r r r< ^ s
��
z_multifile.seekr+ )
�__name__�
__module__� __qualname__r* �propertyr, r3 r9 r: r< r r r r r A s
r c C s� |dkrd|d< |� dd�}|� di �}| �d| � d|fg} d}
|duo)d |v } | rc|rct�t|�� ��} |s<| }n t|t�rMt� |�}
t
|�|
_ |
}t� | �}t
| �|_ t ||�}t
| �|d
< n=|r�|d kskJ �|d �}|r{t
|�d
d�d �}
|
dkr�t�t|�� ��}t|d|
�D ]\}}|||< q�n| t|�� �7 } dt�| � }d||f }d}t�|d�r�|j }n|dur�t
|�}|dur�d|vr�d|d< t� }t� }|dur�|d�}|r�|�d� t|�d
��}|�d� d|v r�|�d� d|v �r|d��rdd � tj�tj�D �}|�d d
�|� � |�r4td �t|��d |
�p&d �}|D ] \}}|||< �q*g }|D ]}|�� �d!��rG|�|� �q8|�rUd"�t|��|d#< |t� |�||�} |du�rr| �d$| � | � d%d&| � | ||fS )'aW Make an HTTP request to run a command for a version 1 client.
``caps`` is a set of known server capabilities. The value may be
None if capabilities are not yet known.
``capablefn`` is a function to evaluate a capability.
``cmd``, ``args``, and ``data`` define the command, its arguments, and
raw data to pass to it.
s pushkeyr2 s dataNs headerss sending %s command
s cmdr � httppostargsz
X-HgArgs-Post� capabilitiess
httpheader� ,r s X-HgArgs ?%ss %s%sr! � Content-Type�application/mercurial-0.1s
httpmediatypes 0.1s partial-pulls 0.2txs 0.2s compressionc S s g | ]}|� � j�qS r )�wireprotosupport�name)r. �er r r �
<listcomp>� s ��z(makev1commandrequest.<locals>.<listcomp>s comp=%s� s X-HgProto� zx-hg�,�Varys sending %d bytes
zContent-Lengthz%d) �pop�debug�urlreq� urlencode�sorted�items�
isinstance�bytes�io�BytesIOr r, r �int�splitr r r" �set�add� compengines�supportedwireengines�
CLIENTROLEr5 �lower�
startswithr r r �add_unredirected_header) �ui�requestbuilder�caps� capablefn� repobaseurl�cmd�args�data�headers�q�
headersize�
postargsok�strargsr �argsio�
httpheader� encoded_argsr r �qs�cu�size� protoparams�
mediatypes�mt�comps� protoheaders� varyheaders�reqr r r �makev1commandrequestn s�
�
�
��
�
r~ c C s t jr| jS | �� s dS | �� S )z3Get request data, if any. If no data, returns None.N)r �ispy3rk �has_data�get_data)r} r r r �_reqdata� s
r� c
C s& | j }| jr�| �dd�r�d}||d t�|�� �t�|�� �f � d}t|�� �D ]*\}}t�|�}t�|�}|� d�rJ|du rCd}|t
|�7 }q*||d ||f � q*|dura||d | � t |�}|durt |d
d�} | du rwt
|�} ||d | � t
�� }
d} zcz|�|�} W n; tjy� } z
| jd kr�t�td
���� d} ~ w tjy� } z| � dt�|�� � � | �� td| ��d} ~ ww W | jr�| �dd�r�| r�| jnd}
||d t
�� |
|
f � n#| j�r
| �dd��r | r�| jnd}
||d t
�� |
|
f � w w w t � | � | S )zDSend a prepared HTTP request.
Returns the response object.
s devels debug.peer-requests devel-peer-request: %s
s %s %sNs X-hgarg-r s %s %ss+ %d bytes of commands arguments in headersr, s %d bytes of datai� s authorization faileds http error requesting %s
r4 s finished in %.4f seconds (%d)) rQ � debugflag�
configboolr �bytesurl�
get_method� get_full_urlrT � header_itemsrb r r� r r �timer�open�urlerr� httperror�coder �Abortr �httplib�
HTTPExceptionr � hidepassword� traceback�IOError�urlmod� wrapresponse)rd �openerr} �dbg�line�
hgargssizer r rk r, �start�res�instr� r r r � sendrequest� s� � ���
���
� ��
���
���������
r� c s e Zd Z� fdd�Z� ZS )�RedirectedRepoErrorc s t t| ��|� || _d S r+ )�superr� r* �respurl)r' �msgr� �� __class__r r r* < s
z RedirectedRepoError.__init__)r? r@ rA r* �
__classcell__r r r� r r� ; s r� c C s� d}t �|�� �}|�|�r|d t|� � }d}nd}|�d�|�d�kr4d}| js4| �td�| � z t �|� dd��} W n t
yR t �|j � dd��} Y nw t
�|�}
| �d�rdt�|�� ��| �d�s�| �d t
�|� � td
�|
| pzd |�d �f } |r�|r�t| |��t�| ��z| �d
d�d } tdd� | �d�D ��}
W n ty� t�td�|
| f ��w |
dkr�|r�tjd �|�}n-|
dkr�t � dt� |d��d }t� ||�}tj� |�}|�|�}n t�td�|
| f ��|| |fS )NFT� /s real URL is %s
z content-type� s application/hg-error� application/mercurial-s requested URL: '%s'
sI '%s' does not appear to be an hg repository:
---%%<--- (%s)
%s
---%%<---
s no content-typerM � -r c S s g | ]}t |��qS r )rZ )r. r r r r rK l s z*parsev1commandresponse.<locals>.<listcomp>� .s+ '%s' sent a broken Content-Type header (%s))r r s zlib)r � � Br s '%s' uses newer protocol %s) r r� �geturl�endswithr �rstrip�quiet�warnr � getheader�AttributeErrorrl �getr r� rb r �OutOfBandErrorr3 rQ r� � RepoErrorr[ �tupler# r r^ �decompressorreader�struct�unpack� readexactly� forwiretype)rd �baseurl�requrlrt �resp� compressible�
redirectedr� � qsdropped�proto�safeurlr� �subtype� version_info�elen�ename�enginer r r �parsev1commandresponseA sb
�
��
��� �
r� c @ s~ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
d dd�Z dd� Z dd� Z
dd� Zdd� Zd d � Zd S ) �httppeerc C s: || _ || _|| _|| _|d uod|v| _|| _|| _d S )NrC )rd �_path�_url�_caps�limitedarguments�
_urlopener�_requestbuilder)r' rd �pathr r� re rf r r r r* � s
zhttppeer.__init__c C s, | j jD ]}|�� t|ddd� �� qd S )N� close_allc S � d S r+ r r r r r �<lambda>� s z"httppeer.__del__.<locals>.<lambda>)r� �handlers�closer )r' �hr r r �__del__� s �zhttppeer.__del__c C � | j S r+ )r� r1 r r r r � � z httppeer.urlc C r� r+ r r1 r r r �local� � zhttppeer.localc C s | S r+ r r1 r r r �peer� r� z
httppeer.peerc C s dS )NTr r1 r r r �canpush� r� zhttppeer.canpushc C sT z| j j| j j| j j}}}W n
ty Y d S w | j�td�|||f � d S )NsE (sent %d HTTP requests and %d bytes; received %d bytes in responses)
)r� �
requestscount�sentbytescount�receivedbytescountr� rd �noter )r' �reqs�sent�recvr r r r� � s � ����zhttppeer.closec C r� r+ )r� r1 r r r � capabilities� r� zhttppeer.capabilitiesFc K s` t �|�}t| j| j| j| j| j||�\}}}t| j| j |�}t
| j| j||||�\| _}}|S r+ ) r � byteskwargsr~ rd r� r� �capabler� r� r� r� ) r' ri �
_compressiblerj r} ru rt r� �ctr r r � _callstream� s
�
�zhttppeer._callstreamc K s0 | j |fi |��}z |�� W |�� S |�� w r+ )r� r3 r� )r' ri rj �fpr r r �_call� s zhttppeer._callc
K sR | � d�}z|�d�}W n ty dg}Y nw |D ] }|tjv r&|} nqt�| j|d |�}t�| j|d�}ddi} zbz-| j |f|| d�|��}
|
�dd �} t
| �d
k r^t � t
d �|
��| W W |�� t�|� S tjys � tj y� } z | jd tjtjfv r�t �t
d
�| jd ��t �| jd ��d } ~ ww |�� t�|� w )Ns unbundlerE r2 � rbrF rG �rk rl �
r r� s unexpected response:r s push failed: %s)r� r[ r� r � bundletypes� writebundlerd r � httpsendfiler� r r �
ResponseErrorr r� r9 �unlinkr� r� �socketrj �errno�
ECONNRESET�EPIPEr� )
r' ri �cgrj �types�xr$ �tempnamer� rl �r�vals�errr r r � _callpush� s@
�
� ��� zhttppeer._callpushc
K s d }zst jddd�\}}t�|d��}|�d�}|r&|�|� |�d�}|sW d � n1 s0w Y t�| j|d��#}ddi} | j |f|| d �|��W d � W |d ur_t�
|� S S 1 scw Y W |d urtt�
|� d S d S |d ur�t�
|� w w )
Ns
hg-bundle-s .hg)�prefix�suffix�wbi r� rF rG r� ) r �mkstempr9 �fdopenr3 �writer r� rd r� r� )
r' ri r� rj �filename�fd�fh�d�fp_rl r r r �_calltwowaystream s0
�� �� �� � �zhttppeer._calltwowaystreamc K s | j |fddi|��S )Nr� T)r� )r' ri rj r r r �_callcompressable s zhttppeer._callcompressablec C s |�r+ r )r' � exceptionr r r �_abort r� zhttppeer._abortN)F)r? r@ rA r* r� r r� r� r� r� r� r� r� r r
r r r r r r r� � s
( r� c @ s e Zd ZdZddd�ZdS )�queuedcommandfuturez@Wraps result() on command futures to trigger submission on call.Nc C s, | � � r tjj�| |�S | j�� | �|�S r+ )�doner �futures�Futurer �
_peerexecutor� sendcommands)r' �timeoutr r r r s
zqueuedcommandfuture.resultr+ )r? r@ rA �__doc__r r r r r r s r c C s d }dd� }i }t | ||||d|�\}}} t| ||�}
zt| ||| |
dd�\} } }
W n3 ty] }
z't | ||||
jd|�\}}} t| ||�}
t| ||| |
dd�\} } }
W Y d }
~
nd }
~
ww z
|
�� }W |
�� n|
�� w | �d�szt� d| ��dt
|� � �i}| |fS ) Nc S s
t �d��)Ns should not be called)r �ProgrammingError)r� r r r r� 0 s
z!performhandshake.<locals>.capablerD F)r� r� s unexpected content-type: %s� v1capabilities) r~ r� r� r� r� r3 r� rb r r r\ r[ )rd r r� re rf r� rj r} r� rt r� r� r� rJ �rawdata�infor r r �performhandshake+ s8
� �
� ���
r c C s� t �|�}|js |jrt�td�|jp|j ��|�� \}}| �d| � |p,t �
| |�}t | |||�\}}t | |||||d �S )a- Construct an appropriate HTTP peer instance.
``opener`` is an ``url.opener`` that should be used to establish
connections, perform HTTP requests.
``requestbuilder`` is the type used for constructing HTTP requests.
It exists as an argument so extensions can override the default.
s unsupported URL component: "%s"s using %s
r )
r r �query�fragmentr r� r �authinforQ r� r� r r� ) rd r� r� re �ur r r� r r r r �makepeer` s
� �r" c C s� |r t �td���z|�d�rtjst �td���t| |�}|W S t jyQ } z$zt� | d| |�}| �
td�� |W W Y d }~S t jyL |�w d }~ww )Ns! cannot create new http repositorys https:s1 Python support for SSL and HTTPS is not installeds static-s (falling back to static-http)
) r r� r rb r� � has_httpsr" r� r
�instancer� )rd r� �create�intents�
createoptsr� �
httpexceptionr� r r r r$ | s& �
���r$ )NN)+�
__future__r r� rX r9 r� r� �i18nr r r r� r r r r
r r� r r
�utilsr r� r� rR r �objectr r~ r� r� r� r� r� �wirepeerr� r r r r �requestr" r$ r r r r �<module> s4 (
-| HG 5