|
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
�]LbQi � @ s� d dl mZ d dlZd dlZddlmZ ddlmZ ddlm Z ddl
m Z m Z m
ZmZmZmZmZmZmZmZmZmZmZ ddlmZm Z ej Z ej Z ed �Z ed
�Z d e e f Z!d d
� Z"dd� Z#dd� Z$dd� Z%dd� Z&e�'� Z(dfdd�Z)e)dddd �d d � �Z*e)d d dd �d!d"� �Z+e)d#dd �d$d%� �Z,e)d&d'dd �d(d)� �Z-e)d*d+dd �d,d-� �Z.g d.�Z/d/d0� Z0e)d1dd �d2d3� �Z1e)d4d5dd �d6d7� �Z
e)d8d9dd �d:d;� �Z2e)d<d=dd �d>d?� �Z3d@dA� Z4e)dBdCdd �dDdE� �Z5e)dFdd �dGdH� �Z6e)dIdd �dJdK� �Z7e)dLdMdd �dNdO� �Z8e)dPdQdd �dRdS� �Z9e)dTdUdd �dVdW� �Z:e)dXdYdd �dZd[� �Z;e)d\d]dd �d^d_� �Ze)d`dd �dadb� �Z<e)dcdFdd �ddde� �Z=dS )g� )�absolute_importN� )�_)�hex)�getattr)
�bundle2� bundlecaches� changegroup� discovery�encoding�error�exchange�pushkey�pycompat� requirements� streamclone�util�wireprototypes)�procutil�
stringutils/ incompatible Mercurial client; bundle2 requireds9 see https://www.mercurial-scm.org/wiki/IncompatibleClients %s
(%s)
c C s6 | � � D ]}|�d�r|dd� �d� S qddgS )a Returns a list of compression methods supported by the client.
Returns a list of the compression methods supported by the client
according to the protocol capabilities. If no such capability has
been announced, fallback to the default of zlib and uncompressed.
s comp=� N� ,s zlibs none)� getprotocaps�
startswith�split)�proto�cap� r �=/usr/lib/python3/dist-packages/mercurial/wireprotov1server.py�clientcompressionsupport0 s
�r c C s | j �dd�}| �|�S )z�Obtain the repo used for processing wire protocol commands.
The intent of this function is to serve as a monkeypatch point for
extensions that need commands to operate on different repo views under
specialized circumstances.
� servers view)�ui�config�filtered)�repor �command�
viewconfigr r r �getdispatchrepo@ s
r' c C s4 t | ||�} t| \}}|�|�}|| |g|�R � S �N)r' �commands�getargs)r$ r r% �func�spec�argsr r r �dispatchK s
r. c C sJ i }|D ]}||v r|| ||< ||= q|r#t j�d| d�|�f � |S )Ns, warning: %s ignored unexpected arguments %s
r )r �stderr�write�join)�cmd�keys�others�opts�kr r r �optionsT s � ��r7 c C sv | j }tj| jv }|r|�dd| �}|dur|S |�dd| �}|dur'|S |r5|�dd�}|dur5|S |�dd�S )z�Whether a bundle1 operation is allowed from the server.
Priority is:
1. server.bundle1gd.<action> (if generaldelta active)
2. server.bundle1.<action>
3. server.bundle1gd (if generaldelta active)
4. server.bundle1
r s bundle1gd.%sNs
bundle1.%ss bundle1gds bundle1)r! �requirementsmod�GENERALDELTA_REQUIREMENTr �
configbool)r$ �actionr! �gd�vr r r �bundle1allowedb s
r> � pushc s` dd� t j�� D ���dvrt�d� ��� du rd� t� t�s%t�d��� ���fdd �}|S )
a� Decorator to declare a wire protocol command.
``name`` is the name of the wire protocol command being provided.
``args`` defines the named arguments accepted by the command. It is
a space-delimited list of argument names. ``*`` denotes a special value
that says to accept all named arguments.
``permission`` defines the permission type needed to run this command.
Can be ``push`` or ``pull``. These roughly map to read-write and read-only,
respectively. Default is to assume command requires ``push`` permissions
because otherwise commands not declaring their permissions could modify
a repository that is supposed to be read-only.
c S s h | ] \}}|d dkr|�qS )s versionr r )�.0r6 r= r r r � <setcomp>� s z#wireprotocommand.<locals>.<setcomp>�r? � pullsC invalid wire protocol permission; got %s; expected "push" or "pull"N� s: arguments for version 1 commands must be declared as bytesc s0 �t v r t�d� ��tj| � ��d�t �< | S )Ns+ %s command already registered for version 1)r- �
transports�
permission)r) r �ProgrammingErrorr � commandentry)r+ �r- �namerF rE r r �register� s �
�z"wireprotocommand.<locals>.register)r �
TRANSPORTS�itemsr rG �
isinstance�bytes)rJ r- rF rK r rI r �wireprotocommand� s" ���
� rP s batchs cmds *rC )rF c sp t j}g }|�d�D ]�}|�dd�\}}i } |�d�D ]}
|
r.|
�d�\} } || �| || �< qt| \}
}t| j}|dv s@J �|�|� |r�|�� }i � |D ]$}|dkrmi }| �� D ] }||vrg| | ||< q[|� d< qO| | � |< qO|
| |g� fdd �|D ��R � }n|
| |�}t|t j�r�| S t|t j t
f�s�J �t|t j �r�|j }|� t �
|�� q
t � d�|��S )
N� ;� r r � =rB � *c s g | ]}� | �qS r r )r@ r6 ��datar r �
<listcomp>� s zbatch.<locals>.<listcomp>)r �unescapebatchargr r) rF � checkpermr3 rN �ooberror�
bytesresponserO rV �append�escapebatchargr1 )r$ r �cmdsr4 � unescapearg�res�pair�opr- �vals�a�nr= r+ r, �permr3 r6 �star�key�resultr rU r �batch� sF �
�
"
rj s betweens pairsc C sL dd� |� d�D �}g }| �|�D ] }|�t�|�d � qt�d�|��S )Nc S s g | ]}t �|d ��qS )� -)r �
decodelist)r@ �pr r r rW � s zbetween.<locals>.<listcomp>rR �
rD )r �betweenr\ r �
encodelistr[ r1 )r$ r �pairs�r�br r r ro � s
ro � branchmapc C s\ | � � }g }t�|�D ]\}}t�t�|��}t�|�}|� d||f � q t�
d� |��S )Ns %s %srn ) � branchmapr � iteritems�urlreq�quoter � fromlocalr rp r\ r[ r1 )r$ r ru �heads�branch�nodes�
branchname� branchnodesr r r ru � s
ru s branches� nodesc C sB t �|�}g }| �|�D ] }|�t �|�d � q t �d�|��S )Nrn rD )r rl �branchesr\ rp r[ r1 )r$ r r| rr rs r r r r� � s
r� s clonebundlesrD c C s t �| j�tj��S )aZ Server command for returning info for available bundles to seed clones.
Clients will parse this response and determine what bundle to fetch.
Extensions may wrap this command to filter or dynamically emit data
depending on the request. e.g. you could advertise URLs for the closest
data center given the client's IP address.
)r r[ �vfs�tryreadr �CB_MANIFEST_FILE�r$ r r r r � clonebundles s
�r� )� lookuprt � pushkey� known� getbundles unbundlehashc C s t t�}t�d|�r|�d� t�| �r=| j�dd�r |�d� t� | �}|t
j h s1|�d� n |�dd� t
|�� � | j�dd �rXt�tj| dd
��}|�d t�|� � |�d d� tj� � | j�dd
�r}|�tj� | j�dd�r}|�tj� |�| |�S )a) return a list of capabilities for a repo
This function exists to allow extensions to easily wrap capabilities
computation
- returns a lists: easy to alter
- change done here will be propagated to both `capabilities` and `hello`
command without any other action needed.
� changegroupsubsetr s preferuncompresseds stream-preferreds streams
streamreqs=%sr s experimentals bundle2-advertise)�roles bundle2=s unbundle=%ss narrows narrowservebrokenellipses)�list�
wireprotocapsr) �commandavailabler\ r �allowservergenerationr! r: �streamed_requirementsr8 �REVLOGV1_REQUIREMENTr1 �sortedr �
encodecaps� getrepocapsrw rx �bundlepriorityr � NARROWCAP� ELLIPSESCAP�addcapabilities)r$ r �caps�requiredformats�capsblobr r r �
_capabilities s&
r� s capabilitiesc C s t | |�}t�d�t|���S )NrR )r� r r[ r1 r� �r$ r r� r r r � capabilitiesE s
r� s changegroups rootsc sL t �|�}tj| || �� d�}t�| |dd�� t� fdd�d�}t j|d�S )N�� missingroots� ancestorsof� 01� servec �
� � d�S �Ni � ��readr ��cgr r �<lambda>R �
z changegroup.<locals>.<lambda>rD ��gen) r rl r
�outgoingrz �changegroupmod�makechangegroup�iter� streamres)r$ r �rootsr| r� r� r r� r r K s
� r r� s bases headsc sR t �|�}t �|�}tj| ||d�}t�| |dd�� t� fdd�d�}t j|d�S )Nr� r� r� c r� r� r� r r� r r r� \ r� z#changegroupsubset.<locals>.<lambda>rD r� )r rl r
r� r� r� r� r� )r$ r �basesrz r� r� r r� r �changegroupsubsetV s
r� �
debugwireargss one two *c C s0 t dddg|�}t�| j||fi t�|����S )Nr� s threes four)r7 r r[ �
debugwireargsr � strkwargs)r$ r �one�twor4 r5 r r r r� ` s �r� c s� dd� }| j �d�}|sdS t�| |�}t�| |�}|s dS | �� j� � j� fdd�|D �dd��� j� fd d�|D �dd��t|�} |D ]�}
|
� d
�} t
j j � | �} | r^| | vr^| | vr^qEd |
v r�z||
d �}
W n t
ys Y qEw |
�|�rzqEt� �fd d
�|
D ��r�qEt� ��fdd
�|
D ��r�qEd|
v r�z||
d �}W n t
y� Y qEw t� �fdd
�|D ��s�qE|
d }| j�d| � z | j �|�W S ty� | j�d| � Y qEw dS )a� Return a file object for the first matching pullbundle.
Pullbundles are specified in .hg/pullbundles.manifest similar to
clonebundles.
For each entry, the bundle specification is checked for compatibility:
- Client features vs the BUNDLESPEC.
- Revisions shared with the clients vs base revisions of the bundle.
A bundle can be applied only if all its base revisions are known by
the client.
- At least one leaf of the bundle's DAG is missing on the client.
- Every leaf of the bundle's DAG is part of node set the client wants.
E.g. do not send a bundle of all changes if the client wants only
one specific branch of many.
c S s dd� | � d�D �S )Nc S s h | ]}t �|��qS r )�binascii� unhexlify)r@ �hr r r rA z � z;find_pullbundle.<locals>.decodehexstring.<locals>.<setcomp>rQ )r )�sr r r �decodehexstringy s z(find_pullbundle.<locals>.decodehexstrings pullbundles.manifestNc � g | ]}� � |��qS r ��rev�r@ r� ��clr r rW � r� z#find_pullbundle.<locals>.<listcomp>T)� inclusivec r� r r� r� r� r r rW � r� s COMPRESSION� headsc 3 � � | ]
}� � |��v V qd S r( r� r� �r� �
common_ancr r � <genexpr>� � � z"find_pullbundle.<locals>.<genexpr>c 3 s, � | ]}� � |��vo� � |��vV qd S r( r� r� �r� r� � heads_ancr r r� � s
� �
�s basesc 3 r� r( r� r� r� r r r� � r� s URLs sending pullbundle "%s"
s pullbundle "%s" not accessible
)r� r� r �parseclonebundlesmanifest�filterclonebundleentries�
unfiltered� changelog� ancestorsr �getr � compengines� _bundlenames� TypeError�issubset�all�anyr! �debug�open�IOError)r$ r r5 �clheadsrz �commonr� �manifestr` � compformats�entry�comp�altcomp� bundle_heads� bundle_bases�pathr r� r �find_pullbundlei s`
�
� � �r� r� rT c
C s� t dtj�� |�}t�|�D ]O\}}tj| }|dkr#t�|�||< q|dkr1t|�d��||< q|dkr?t |�d��||< q|dkrS|dkrLd||< qt
|�||< q|d kr]t d
| ��qt | d �s|t
�|�d ��s||jd
krut�t�S tjttd��zyt | j�� �}t |�dt � ��}t |�dt � ��} | �| j� | j� dd�r�d|� � v r�t | ||||| �}
|
r�tj t �!|
�dd�W S | j� dd�r�|�dd�} | r�| s�||kr�tjt"d�t"d�d��t
j#| dfi t�$|���\} }
| �dd�}W n^ tj�yS } zPt
�|�d ���s |jd
k�r t�|j%d �W Y d }~S � t&�'| j�}d |j%fg}g }|j(d u�r9|�)d |j(f� |�*t&�+d ||�� |�,� }
d}W Y d }~nd }~ww tj |
| d�S ) Nr� r s csvr s scsvs boolean� 0Fs plains unknown getbundle option type %srC s
bundlecaps� http-v1��hintr� s commonr s
pullbundles partial-pullT)r� �prefer_uncompresseds disablefullbundles cgs% server has pull-based clones disableds/ remove --pull if specified or upgrade Mercurialr� s prefercompressedrn � message� hint� error:abort)-r7 r �GETBUNDLE_ARGUMENTSr3 r rv rl r� r �set�bool�KeyErrorr> r
�bundle2requestedr� rJ rZ �bundle2requiredr �Abort�bundle2requiredmain�bundle2requiredhintr� rz �discard�nullidr! r: r r� r� r �
filechunkiterr �getbundlechunksr� �messager �bundle20r� r\ �addpart�
bundlepart� getchunks)r$ r r4 r5 r6 r= �keytyper� rz r� �bundler �info�chunks�prefercompressed�exc�bundler�manargs�advargsr r r � getbundle� s� �
�
�
� ��
� ���r
r� c C s | � � }t�t�|�d �S )Nrn )rz r r[ rp )r$ r r� r r r rz � s rz s helloc C s t | |�j}t�d| �S )af Called as part of SSH handshake to obtain server info.
Returns a list of lines describing interesting things about the
server, in an RFC822-like format.
Currently, the only one defined is ``capabilities``, which consists of a
line of space separated tokens describing server abilities:
capabilities: <token0> <token1> <token2>
s capabilities: %s
)r� rV r r[ r� r r r �hello s r s listkeys� namespacec C s( t | �t�|���� �}t�t�|��S r( ) r� �listkeysr �tolocalrM r r[ �
pushkeymod�
encodekeys)r$ r � namespace�dr r r r s r r� � keyc
C sj zt �|�}| �|�}t|�}d}W n ty+ } z
t�|�}d}W Y d }~nd }~ww t�d||f �S )Nr r s %d %s
) r r �lookupr � Exceptionr � forcebytestrr r[ )r$ r rh r6 re rr �success�instr r r r s
��r r� s nodes *c C s* d� dd� | �t�|��D ��}t�|�S )NrD c s s � | ] }|rd p dV qdS )� 1r� Nr )r@ rs r r r r� ) s �
�zknown.<locals>.<genexpr>)r1 �knownr rl r[ )r$ r r| r4 r= r r r r ' s
�
r s protocapss capsc C s& |j tjkrt|�d��|_t�d�S )NrR s OK)rJ r �SSHV1r� r �
_protocapsr[ r� r r r � protocaps/ s
r r� s namespace key old newc C s� t |�dkr#t�|�|kr#z |�d� t�|�}W n ty" Y nw t�|�}|�� �}| �t�|�t�|�t�|�|�p?d}W d � n1 sJw Y |rU|� � nd}t
� dt |�|f �S )N� zutf-8FrD s %d
%s)
�lenr � escapestr�decoder r �UnicodeDecodeError�mayberedirectstdior �getvaluer r[ �int)r$ r r rh �old�new�outputrr r r r r 6 s(
�
�� � r s
stream_outc C s t �t�| ��S )z�If the server supports streaming clone, it advertises the "stream"
capability with a value representing the version and flags of the repo
it is serving. Client checks to see if it understands the format.
)r �streamreslegacyr �generatev1wireprotor� r r r �streamS s r. s unbundlec s$ t �|�}|�� ��}}z�t�| |d� dd� }z�|�� �| j�dd�r/�fdd�}t� ��� n3d\� �� �fd d�}t
j d
d �\}�| j� d � � t
�|t
�d
��� �D ]}� �|� qU� �d� t�| j� d �}t|tj�r�t| d�s�|jdkr�t �t�W |� W W d � S tjt t d��t� | ||d|� � �} t� | d�r�t j!| �"� d�W |� W W d � S t �#| |r�|�$� nd�W |� W W d � S |� w tj%tjtj&f�y� }
�z�t'|
dd��sMz� tj�y% }
z-t(j)�|
�*� � t(j)�+� t �#d|�r |�$� nd�W Y d }
~
W Y d }
~
W d � S d }
~
w tj&�yL t �,t
�-|
�|�r:|�$� nd� Y W Y d }
~
W d � S w t.�/| j�} t'|
dd�D ]} | �0| � �qYz�z� tj1�y� }
zrt'|
dd �}
|
d u�r�d|
�2d d�v�r�� | �3d �}|�4d |
j5� |
j6d u�r�|j4d |
j6dd � |
j7d u�r�|j4d!|
j7dd � |
j8d u�r�|j4d"|
j8dd � |
j9d u�r�|j4d#|
j9dd � |
j:d u�r�|j4d$|
j:dd � W Y d }
~
nd }
~
ww W n� tj%�y }
z&| �3d%�}|
j;d u�r|�4d&|
j;� |
j<�r|�4d'd(�=|
j<�� W Y d }
~
nWd }
~
w tj�yM }
z&d)|
j>fg}g }|
j?d u�r9|�@d*|
j?f� | �0t.�Ad+||�� W Y d }
~
n$d }
~
w tj&�yl }
z| �3d,d)tB�C|
�fg� W Y d }
~
nd }
~
ww t j!| �"� d�W Y d }
~
W d � S d }
~
ww 1 �s�w Y d S )-Ns preparing changesc S s d S r( r r r r r r� c s zunbundle.<locals>.<lambda>r s streamunbundlec s � D ]} qd S r( r )rm )�payloadr r �cleanuph s �zunbundle.<locals>.cleanup)NNc s" � r� � � �rt��� d S d S r( )�close�os�unlinkr )�fp�tempnamer r r0 s s
�s hg-unbundle-)�prefixs" redirecting incoming bundle to %s
s wb+r r? r� r� r� s addpartr� rD �duringunbundle2F�_bundle2salvagedoutputr �
_replycapsr� s errors
error:pushkeys in-reply-tor )� mandatoryr s news olds rets error:unsupportedcontents parttypes params� r� r� r� s error:pushraced)Dr rl r& r
� check_heads�
getpayloadr! r: r � chunkbufferr �mkstempr� r2 �fdopen�sysstrr0 �seek�
readbundlerN r� � cg1unpackerr> rJ rZ r� r r� r� r� �unbundle�client� safehasattrr, r �pushresr' �BundleValueError� PushRacedr r r/ �format�flush�pusherr�bytestrr r r �
PushkeyFailedr� �newpart�addparam�partidr rh r* r) �ret�parttype�paramsr1 r� r� r\ r r r )r$ r rz � their_headsr+ r0 �fdrm r� rr r r
�out�
remotecaps�part�errpartr r r )r4 r/ r5 r rE \ s�
�
��
�,�� �7��<
���M��M
� ���
� ����� ���>�rE )Nr? )>�
__future__r r� r2 �i18nr �noder r r � r r r r� r
r r r
r r r r8 r r r �utilsr r �urlerrrw r� r� r� r r' r. r7 r> � commanddictr) rP rj ro ru r� r� r� r� r� r� r� r� r
rz r r r r r r. rE r r r r �<module> s~ <�
0
-
+
F
N