|
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\b � @ sl d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z z d dl
Z
e
j W n e y> ddl
mZ
Y nw ddlmZ ddlmZ ddlmZmZmZmZmZmZmZ ddlm Z m Z G d d
� d
e �Z G d d � d e �Z G d
d� de �Z!ddd� iZ"dd� Z#G dd� de �Z$d"dd�Z%G dd� de �Z&dd� Z'd d � Z(G d d � d e �Z)G d d!� d!e �Z*dS )#� )�absolute_importN� )�
selectors2)�_)�getattr)�encoding�error� loggingutil�pycompat� repocache�util�vfs)�cborutil�procutilc @ s4 e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd
S ) �channeledoutputz^
Write data to out in the following format:
data length (unsigned int),
data
c C � || _ || _d S �N)�out�channel)�selfr r � r �9/usr/lib/python3/dist-packages/mercurial/commandserver.py�__init__3 �
zchanneledoutput.__init__c C �
d| j S �Ns <%c-channel>�r �r r r r �name7 �
zchanneledoutput.namec C s6 |sd S | j �t�d| jt|��| � | j �� d S )N� >cI)r �write�struct�packr �len�flush)r �datar r r r! ; s zchanneledoutput.writec C � |dv rt |��t| j|�S �N)�isatty�fileno�tell�seek)�AttributeErrorr r �r �attrr r r � __getattr__B � zchanneledoutput.__getattr__N) �__name__�
__module__� __qualname__�__doc__r �propertyr r! r0 r r r r r + s
r c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
�channeledmessagea
Write encoded message and metadata to out in the following format:
data length (unsigned int),
encoded message and metadata, as a flat key-value dict.
Each message should have 'type' attribute. Messages of unknown type
should be ignored.
Tc C s t ||�| _|| _|| _d S r )r �_coutr � _encodefn)r r r �
encodename�encodefnr r r r V s
zchanneledmessage.__init__c K s0 t �|�}|d ur
||d< | j�| �|�� d S )Ns data)r
� byteskwargsr8 r! r9 )r r& �optsr r r r! [ s
zchanneledmessage.writec C s t | j|�S r )r r8 r. r r r r0 a s z channeledmessage.__getattr__N)r2 r3 r4 r5 �
structuredr r! r0 r r r r r7 H s r7 c @ s` e Zd ZdZdZdd� Zedd� �Zddd �Zd
d � Z dd d
�Z
dd� Z dd� Z e Z
dd� ZdS )�channeledinputa/
Read data from in_.
Requests for input are written to out in the following format:
channel identifier - 'I' for plain input, 'L' line based (1 byte)
how many bytes to send at most (unsigned int),
The client replies with:
data length (unsigned int), 0 meaning EOF
data
i c C s || _ || _|| _d S r )�in_r r )r r@ r r r r r r t s
zchanneledinput.__init__c C r r r r r r r r y r zchanneledinput.name���c C sP |dk r!| j }| �|| j�}|}|r | �|| j�}||7 }|s|S | �|| j�S �Nr )� maxchunksize�_readr �r �size�s�bufr r r �read} s �zchanneledinput.readc C sd |sdS |dks
J �| j �t�d||�� | j �� | j�d�}t�d|�d }|s,dS | j�|�S )N� r r � � >I)r r! r" r# r% r@ rI �unpack)r rF r �lengthr r r rD � s
zchanneledinput._readc C s^ |dk r)| j }| �|d�}|}|r'|�d�s'| �|d�}||7 }|r'|�d�r|S | �|d�S )Nr � L�
)rC rD �endswithrE r r r �readline� s � zchanneledinput.readlinec C s | S r r r r r r �__iter__� � zchanneledinput.__iter__c C s | � � }|st�|S r )rR �
StopIteration)r �lr r r �next� s zchanneledinput.nextc C r' r( )r- r r@ r. r r r r0 � r1 zchanneledinput.__getattr__N)rA )r2 r3 r4 r5 rC r r6 r rI rD rR rS rW �__next__r0 r r r r r? e s
r? s cborc C s d� t�| ��S �NrJ )�joinr � streamencode)�vr r r �<lambda>� s r] c C sD | � dd�}|D ]}t�|�}|r||f S qt�dd�|� ��)N� cmdservers message-encodingss" no supported message encodings: %s� )�
configlist�_messageencoders�getr �AbortrZ )�ui�encnames�n�fr r r �_selectmessageencoder� s
� �rh c @ sl e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z e
e d�Z
dd� ZdS )�serverzm
Listens for commands on fin, runs them and writes the output on a channel
based stream to fout.
Nc C s t �� | _|r|j| _|| _|j| _n || _d | _| _|| _t|d�| _ t|d�| _
t|d�| _ t ||d�| _
t|d�| _| j�dd�dkrV| j�� | _t| jd | j d � d | _|�d
d �d krot|�\}}t|d
||�| _|| _|�dd�| _d | _| js�t�tjtj�| _d S d S )N� d� e� o� I� rr^ � log� -)�repo�fp� uis message-outputs channel� ms shutdown-on-interrupt) r �getcwd�cwd�baseuird rq �repoui�_prereposetupsr �cdebug�cerr�coutr? �cin�cresult�config�copy� setuplogging�cmsgrh r7 �client�
configbool�_shutdown_on_interrupt�_old_inthandler�signal�SIGINT�SIG_IGN)r rd rq �fin�fout�
prereposetups�encname�encfnr r r r � s8
��zserver.__init__c C s | j s
t�tj| j� dS dS )z9release and restore resources taken during server sessionN)r� r� r� r� r r r r �cleanup s �zserver.cleanupc C s |sdS | j �|�}|st�|S rY )r� rI �EOFError)r rF r& r r r rD s z server._readc C s( t �d| �d��d }|sdS | �|�S )z[read a string from the channel
format:
data length (uint32), data
rL rK r rJ )r" rM rD )r rN r r r �_readstr s
zserver._readstrc C s | � � }|r |�d�S g S )z6read a list of NULL separated strings from the channel� )r� �split)r rG r r r � _readlist s
zserver._readlistc
C s� ddl m} | jr|�|�S z3zt�tj| j� |�|�W W t�tjtj� S tjy/ � t y7 Y nw W t�tjtj� n t�tjtj� w |j
�t d�� dS )Nr ��dispatchs
interrupted!
rA ) � r� r� r� r� r� r� r �SignalInterrupt�KeyboardInterruptrd r )r �reqr� r r r �_dispatchcommand$ s
� �$zserver._dispatchcommandc
C s* ddl m} | �� }| j�� }|g}| jr7|| j_| j�| j�}|j|_|� |� | | j_| jj
_ | j� � |D ]}|�
� t�| jd�sN|�dddd� q9|j|dd� || j| j| j| j| j| jd �}z"| �|�d
@ }| j�t�d t |��� W d |v r�t � | j � dS dS d |v r�t � | j � w w )
zireads a list of terminated arguments, executes
and writes the return code to the result channelr r� s filenors s nonttys trues
commandserverN�r� � s >is --cwd) r� r� r� rd r� rq rw rx � __class__�append�dirstate�_ui�
invalidateall�
resetstater � safehasattrr} � setconfig�requestr| r{ r� ry r� r~ r! r" r# �int�os�chdirrv ) r r� �args�copiedui�uisrx rd r� �retr r r �
runcommand> sD
�
� ��zserver.runcommandc C s | j �tj� dS )z1writes the current encoding to the result channelN)r~ r! r r r r r � getencodingp s zserver.getencodingc C sP | j �� d d� }|r$| j�|�}|r|| � |dkS t�td�| ��|dkS )NrA s unknown command %srJ )r� rR � capabilitiesrb r rc r )r �cmd�handlerr r r �serveonet s �zserver.serveone)s
runcommands getencodingc C s� dd� t| j�� }|d7 }|dtj 7 }|d7 }| jr$|d| jj 7 }|dt�� 7 }t�t d�r?|d7 }|dt �
d � 7 }| j � |� z| �
� rR | �
� sJW d S W d S ty^ Y d
S w ) Ns capabilities: r_ rP s
encoding: s message-encoding: %s
s pid: %ds getpgids pgid: %dr r )rZ �sortedr� r r� r �getpidr r� r� �getpgidr| r! r� r� )r �hellomsgr r r �serve� s*
�� ��z server.server )r2 r3 r4 r5 r r� rD r� r� r� r� r� r� r� r� r r r r ri � s
. 2
ri c C s� | � dd�}|s
dS t| �dd��}|dkr |r t�||�}n7|dkr+t�| j|�}n+t�t�|��}| � dd�}| �
dd�}t � t
j�|��}tj|t
j�|�|||d�}| h} |rg| �|j� | �|j� | D ]}
|
�d|� qidS ) z�Set up server logging facility
If cmdserver.log is '-', log messages will be sent to the given fp.
It should be the 'd' channel while a client is connected, and otherwise
is the stderr of the server process.
r^ ro Ns track-logrp s
max-log-filess max-log-size)�maxfiles�maxsize)r �setr` r �fileobjectlogger�ferrr �abspath�
expandpath� configint� configbytes�vfsmodr
r� �path�dirname�
filelogger�basename�addrw rd � setlogger) rd rq rr �logpath�tracked�loggerr� r� r
� targetuis�ur r r r� � s4
� �r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� pipeservicec C r r )rd rq )r rd rq r= r r r r � r zpipeservice.__init__c C s d S r r r r r r �init� rT zpipeservice.initc C sd | j }|�� �!\}}t|| j||�}z|�� W |�� W d � S |�� w 1 s+w Y d S r )rd �protectedfinoutri rq r� r� )r rd r� r� �svr r r �run� s �
�zpipeservice.runN)r2 r3 r4 r r� r� r r r r r� � s r� c C s t �dd� t�� d S rB )r� �setpgid�random�seedr r r r �_initworkerprocess� s r� c
C s� |� d�}|� d�}d }z�zZ||||||�}zJz|�� W n= tjy: } z| �td�|j � W Y d }~n%d }~w tyR } z
|jtjkrH� W Y d }~n
d }~w t yZ Y nw W |�
� n|�
� w W n |rq|j } nt |d�} | �
t�t�� �� � W |�� z|�� W d S ty� } z|jtjkr�� W Y d }~d S d }~ww |�� z|�� W w ty� } z
|jtjkr�� W Y d }~w d }~ww )N�rb�wbs
abort: %s
rk )�makefiler� r rc r �message�IOError�errno�EPIPEr� r� r{ r r! r �
strtolocal� traceback�
format_exc�close)
rd rq �conn�createcmdserverr� r� r� r� �instr{ r r r �
_serverequest� sZ
� �� ��
� ���� ���r� c @ sD e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dS )�unixservicehandlerz�Set of pluggable operations for unix-mode services
Almost all methods except for createcmdserver() are called in the main
process. You can't pass mutable resource back from createcmdserver().
Nc C s
|| _ d S r )rd )r rd r r r r s
zunixservicehandler.__init__c C s: t �||� |�tj� | j�td�| � | j�� d S )Ns listening at %s
) r �bindunixsocket�listen�socket� SOMAXCONNrd �statusr r% )r �sock�addressr r r �
bindsocket s z unixservicehandler.bindsocketc C s t �|� d S r )r� �unlink)r r� r r r � unlinksocket# s z unixservicehandler.unlinksocketc C � dS )z9True if server should shut down; checked per pollintervalFr r r r r �
shouldexit& s z unixservicehandler.shouldexitc C r� )z/Called when main process notices new connectionNr r r r r �
newconnection* s z unixservicehandler.newconnectionc C s t | j||||�S )zhCreate new command server instance; called in the process that
serves for the current connection)ri rd )r rq r� r� r� r� r r r r� - s z"unixservicehandler.createcmdserver) r2 r3 r4 r5 � pollintervalr r� r� r� r� r� r r r r r� s r� c @ sr e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �unixforkingservicezG
Listens on unix domain socket and forks server per connection
Nc C s� || _ || _|d | _t�td�st�td���| js"t�td���|p't |�| _
d | _ d | _ d | _
d | _t� | _d | _|�dd�}|dk rMt�td���t�||�| _tjr]t�� d S d S ) Ns addresss AF_UNIXs unsupported platforms' no socket path specified with --addressr^ s max-repo-cacher s( negative max-repo-cache size not allowed)rd rq r� r r� r� r rc r r� �_servicehandler�_sock�_mainipc�
_workeripc�_oldsigchldhandlerr� � _workerpids�_socketunlinkedr� r �
repoloader� _repoloaderr
�isdarwinr �gui)r rd rq r= r� �maxlenr r r r 8 s*
�zunixforkingservice.__init__c C s~ t � t j�| _t �t jt j�}|\| _| _| j�| j| j � t
� t d�r*t �
tj� t�tj| j�}|| _d| _| j�� d S )Ns
unblocksignalF)r� �AF_UNIXr� �
socketpair�
SOCK_DGRAMr� r� r� r� r� r r� r �
unblocksignalr� �SIGCHLD�_sigchldhandlerr� r� r� �start)r �or r r r� Q s zunixforkingservice.initc C s" | j s| j�| j� d| _ d S d S )NT)r� r� r� r� r r r r �
_unlinksocket` s
�z unixforkingservice._unlinksocketc C sN t � t j| j� | j�� | j�� | j�� | �� | j� � | �
d� d S rB ) r� r r� r� r� r� r� r
r� �stop� _reapworkersr r r r �_cleanupe s
zunixforkingservice._cleanupc C s" z | � � W | �� d S | �� w r )� _mainloopr
r r r r r� o s
zunixforkingservice.runc
C s� d}| j }t�� }|�| jtj| j� |�| jtj| j� |s*|� � r*| �
� d}z |j |j d�}W n t
yN } z|jtjkrB� g }W Y d }~nd }~ww |sU|rTnq |D ] \}}|�|j|� qWq |�� d S )NFT)�timeout)r� � selectors�DefaultSelector�registerr� �
EVENT_READ�_acceptnewconnectionr� �_handlemainipcr� r
�selectr� �OSErrorr� � ETIMEDOUTr& �fileobjr� )r �exiting�h�selector�eventsr� �key�_maskr r r r u s: � � �� � z unixforkingservice._mainloopc C s | j }z|�� \}}W n tjy( } z|jd tjkr#W Y d }~d S � d }~ww t�� }|rNz| j �
dd|� | j � |� |�
� W |�� d S |�� w z#|�� |�� | j�� | �|� |�� | j�� t�d� W d S z| j jdd� W t�d� Y d S t�d� w )Nr r^ s forked worker process (pid=%d)
T)�forcer� )r� �acceptr� r r� r� �EINTRr� �forkrd �logr� r� r� r� r� �
_runworkerr� �_exitr� )r r� r r r� �_addrr� �pidr r r r � s< �� �
z'unixforkingservice._acceptnewconnectionc
C sZ z|� d�}W n tjy$ } z|jd tjkr W Y d}~dS � d}~ww | j�|� dS )z#Process messages sent from a workeri � r N)�recvr� r r� r� r" r� �load)r r� r r� r� r r r r � s ��z!unixforkingservice._handlemainipcc C s | � tj� d S r )r r� �WNOHANG)r r� �framer r r r � s z"unixforkingservice._sigchldhandlerc
C s� | j rRz
t�d|�\}}W n+ ty8 } z |jtjkr!W Y d }~q |jtjkr(� | j �� W Y d }~d S d }~ww |dkr?d S | j� dd|� | j �
|� | j sd S d S )NrA r r^ s worker process exited (pid=%d)
) r� r� �waitpidr r� r" �ECHILD�clearrd r$ �discard)r �optionsr( �_statusr� r r r r � s"
�� �z unixforkingservice._reapworkersc C sR t � t j| j� t� | j}zt| j| j||j| j gd� W t
� � d S t
� � w )Nr� ) r� r r� r� r� r� rd rq r� �
_reposetup�gc�collect)r r� r r r r r% � s �z unixforkingservice._runworkerc sl |� � sd S G � fdd�d|j�� � |_| j|_| j�|j�}|d u r%d S |j�dd|j� t �
||� d S )Nc s e Zd Z� �fdd�Z� ZS )z8unixforkingservice._reposetup.<locals>.unixcmdserverrepoc sH t �| ��� z
| j�| j� W d S tjy# | j�dd� Y d S w )Nr^ s# failed to send repo root to master
) �superr� �
_cmdserveripc�send�rootr� r rd r$ r )r� �unixcmdserverrepor r r� � s
��z>unixforkingservice._reposetup.<locals>.unixcmdserverrepo.close)r2 r3 r4 r� �
__classcell__r �r: )r� r r: � s r: s repocaches repo from cache: %s
) �localr� r� r7 r� rb r9 rd r$ r � copycache)r rd rq �
cachedrepor r< r r3 � s
z unixforkingservice._reposetupr )r2 r3 r4 r5 r r� r
r
r� r r r r r r% r3 r r r r r� 3 s
%&
r� )NN)+�
__future__r r� r4 r� r� r� r� r" r� r � BaseSelector� ImportError�
thirdpartyr �i18nr r
r r� r r r r r r
r� �utilsr r �objectr r7 r? ra rh ri r� r� r� r� r� r� r r r r �<module> sB
� $ X�
R)%"