/
lib64
/
python3.9
/
site-packages
/
mercurial
/
__pycache__
/
Upload File
HOME
a �+�b\b � @ sl d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z zd dl Z e j W n e y| ddl mZ Y n0 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 s || _ || _d S �N)�out�channel)�selfr r � r �=/usr/lib64/python3.9/site-packages/mercurial/commandserver.py�__init__3 s zchanneledoutput.__init__c C s d| j S �Ns <%c-channel>�r �r r r r �name7 s 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 s |dv rt |��t| j|�S �N)�isatty�fileno�tell�seek)�AttributeErrorr r �r �attrr r r � __getattr__B s zchanneledoutput.__getattr__N) �__name__� __module__� __qualname__�__doc__r �propertyr r r+ 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 Z encodenameZencodefnr r r r V s zchanneledmessage.__init__c K s0 t �|�}|d ur||d< | j�| �|�� d S )Ns data)r Z byteskwargsr2 r r3 )r r" �optsr r r r [ s zchanneledmessage.writec C s t | j|�S r )r r2 r) r r r r+ a s z channeledmessage.__getattr__N)r, r- r. r/ Z structuredr r r+ r r r r r1 H s r1 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 r6 r r r r r r t s zchanneledinput.__init__c C s d| j S r r r r r r r y s zchanneledinput.name���c C sR |dk r@| j }| �|| j�}|}|r<| �|| j�}||7 }q |S | �|| j�S d S �Nr )� maxchunksize�_readr �r �size�s�bufr r r �read} s zchanneledinput.readc C sh |sdS |dksJ �| j �t�d||�� | j �� | j�d�}t�d|�d }|sXdS | j�|�S d S )N� r r � � >I)r r r r r! r6 r? �unpack)r r< r �lengthr r r r: � s zchanneledinput._readc C sV |dk rF| j }| �|d�}|}|rB|�d�sB| �|d�}||7 }q |S | �|d�S d S )Nr � L� )r9 r: �endswithr; r r r �readline� s zchanneledinput.readlinec C s | S r r r r r r �__iter__� s zchanneledinput.__iter__c C s | � � }|st�|S r )rH � StopIteration)r �lr r r �next� s zchanneledinput.nextc C s |dv rt |��t| j|�S r# )r( r r6 r) r r r r+ � s zchanneledinput.__getattr__N)r7 )r7 )r, r- r. r/ r9 r r0 r r? r: rH rI rL �__next__r+ r r r r r5 e s r5 s cborc C s d� t�| ��S �Nr@ )�joinr Z streamencode)�vr r r �<lambda>� r@ rQ c C sH | � dd�}|D ] }t�|�}|r||f S qt�dd�|� ��d S )N� cmdservers message-encodingss" no supported message encodings: %s� )� configlist�_messageencoders�getr �AbortrO )�uiZencnames�n�fr r r �_selectmessageencoder� s �r[ 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�dkr�| j�� | _t| jd | j d � d | _|�d d �d kr�t|�\}}t|d ||�| _|| _|�dd�| _d | _| j�st�tjtj�| _d S )N� d� e� o� I� rrR � log� -)�repo�fp� uis message-outputs channel� ms shutdown-on-interrupt) r �getcwd�cwd�baseuirX rd �repoui�_prereposetupsr Zcdebug�cerr�coutr5 �cin�cresult�config�copy� setuplogging�cmsgr[ r1 �clientZ configbool�_shutdown_on_interrupt�_old_inthandler�signal�SIGINT�SIG_IGN)r rX rd �fin�fout� prereposetupsZencnameZencfnr r r r � s6 �zserver.__init__c C s | j st�tj| j� dS )z9release and restore resources taken during server sessionN)rv rx ry rw r r r r �cleanup s zserver.cleanupc C s |sdS | j �|�}|s t�|S rN )ru r? �EOFError)r r< r" r r r r: s z server._readc C s( t �d| �d��d }|s dS | �|�S )z[read a string from the channel format: data length (uint32), data rB rA r r@ )r rC r: )r rD r r r �_readstr s zserver._readstrc C s | � � }|r|�d�S g S dS )z6read a list of NULL separated strings from the channel� N)r� �split)r r= r r r � _readlist s zserver._readlistc C s� ddl m} | jr |�|�S zjz.t�tj| j� |�|�W W t�tjtj� S tjyb � Y n t yr Y n0 W t�tjtj� nt�tjtj� 0 |j �t d�� dS )Nr ��dispatchs interrupted! r7 ) � r� rv rx ry rw rz r ZSignalInterrupt�KeyboardInterruptrX r )r �reqr� r r r �_dispatchcommand$ s � $zserver._dispatchcommandc C s* ddl m} | �� }| j�� }|g}| jrn|| j_| j�| j�}|j|_|� |� | | j_| jj _ | j� � |D ]*}|� � t�| jd�sr|�dddd� qr|j|dd� || j| j| j| j| j| jd �}z@| �|�d @ }| j�t�d t |��� W d |v �r&t � | j � nd |v �r$t � | j � 0 dS ) zireads a list of terminated arguments, executes and writes the return code to the result channelr r� s filenorf s nonttys trues commandserverN�r} � s >is --cwd) r� r� r� rX rr rd rj rk � __class__�appendZdirstateZ_uiZ invalidateallZ resetstater � safehasattrro Z setconfigZrequestrn rm rt rl r� rp r r r �int�os�chdirri ) r r� �argsZcopieduiZuisrk rX r� �retr r r � runcommand> s@ � � zserver.runcommandc C s | j �tj� dS )z1writes the current encoding to the result channelN)rp r r r r r r � getencodingp s zserver.getencodingc C sJ | j �� d d� }|rB| j�|�}|r0|| � nt�td�| ��|dkS )Nr7 s unknown command %sr@ )ru rH � capabilitiesrV r rW r )r �cmd�handlerr r r �serveonet s zserver.serveone)s runcommands getencodingc C s� dd� t| j�� }|d7 }|dtj 7 }|d7 }| jrH|d| jj 7 }|dt�� 7 }t�t d�r~|d7 }|dt � d � 7 }| j � |� z| � � r�q�W n ty� Y d S 0 d S ) Ns capabilities: rS rF s encoding: s message-encoding: %s s pid: %ds getpgids pgid: %dr r )rO �sortedr� r rt r �getpidr r� r� �getpgidrn r r� r )r Zhellomsgr r r �serve� s" z server.serve)N)r, r- r. r/ r r~ r: r� r� r� r� r� r� r� r� r r r r r\ � s . 2 r\ c C s� | � dd�}|sdS t| �dd��}|dkr>|r>t�||�}nn|dkrVt�| j|�}nVt�t�|��}| � dd�}| � dd�}t � t j�|��}tj|t j�|�|||d�}| h} |r�| �|j� | �|j� | D ]} | �d|� q�dS ) 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. rR rb Ns track-logrc s max-log-filess max-log-size)�maxfiles�maxsize)rq �setrT r ZfileobjectloggerZferrr �abspathZ expandpath� configintZ configbytes�vfsmodr r� �path�dirnameZ filelogger�basename�addrj rX Z setlogger) rX rd re ZlogpathZtracked�loggerr� r� r Z targetuis�ur r r rs � s2 � rs c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� pipeservicec C s || _ || _d S r )rX rd )r rX rd r4 r r r r � s zpipeservice.__init__c C s d S r r r r r r �init� s zpipeservice.initc C sr | j }|�� �P\}}t|| j||�}z |�� W |�� W d � S |�� 0 W d � n1 sd0 Y d S r )rX Zprotectedfinoutr\ rd r� r~ )r rX r{ r| �svr r r �run� s �zpipeservice.runN)r, r- r. r r� r� r r r r r� � s r� c C s t �dd� t�� d S r8 )r� �setpgid�randomZseedr r r r �_initworkerprocess� s r� c C s� |� d�}|� d�}d }�z>z�||||||�}z�z |�� W nz tjyv } z | �td�|j � W Y d }~nJd }~0 ty� } z|jtjkr�� W Y d }~nd }~0 t y� Y n0 W |� � n |� � 0 W n8 |r�|j } n t |d�} | � t�t�� �� � Y n0 W |�� z |�� W n6 t�yV } z |jtjk�rB� W Y d }~n d }~0 0 nN|�� z |�� W n6 t�y� } z |jtjk�r�� W Y d }~n d }~0 0 0 d S )N�rb�wbs abort: %s r^ )Zmakefiler� r rW r �message�IOError�errnoZEPIPEr� r~ rm r r r Z strtolocal� traceback� format_exc�close) rX rd �conn�createcmdserverr} r{ r| r� �instrm r r r � _serverequest� sB ( � 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 )rX )r rX r r r r s zunixservicehandler.__init__c C s: t �||� |�tj� | j�td�| � | j�� d S )Ns listening at %s ) r ZbindunixsocketZlisten�socketZ SOMAXCONNrX �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 s dS )z9True if server should shut down; checked per pollintervalFr r r r r � shouldexit& s z unixservicehandler.shouldexitc C s dS )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)r\ rX )r rd r� r{ r| r} r r r r� - s z"unixservicehandler.createcmdserver) r, r- r. r/ � 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�s0t�td���| jsDt�td���|pNt |�| _ d | _ d | _ d | _ d | _t� | _d | _|�dd�}|dk r�t�td���t�||�| _tjr�t�� d S ) Ns addresss AF_UNIXs unsupported platforms' no socket path specified with --addressrR s max-repo-cacher s( negative max-repo-cache size not allowed)rX rd r� r r� r� r rW r r� �_servicehandler�_sock�_mainipc� _workeripc�_oldsigchldhandlerr� � _workerpids�_socketunlinkedr� r Z repoloader� _repoloaderr Zisdarwinr Zgui)r rX rd r4 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�rTt � tj� t�tj| j�}|| _d| _| j�� d S )Ns unblocksignalF)r� ZAF_UNIXr� Z socketpairZ SOCK_DGRAMr� r� r� r� r� r r� r Z unblocksignalrx �SIGCHLD�_sigchldhandlerr� r� r� �start)r �or r r r� Q s zunixforkingservice.initc C s | j s| j�| j� d| _ 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 r8 ) rx r� r� r� r� r� r� r� r� �stop� _reapworkersr r r r �_cleanupe s zunixforkingservice._cleanupc C s$ z| � � W | �� n | �� 0 d S r )� _mainloopr� r r r r r� o s zunixforkingservice.runc C s� d}| j }t�� }|�| jtj| j� |�| jtj| j� |sR|� � rR| � � d}z|j |j d�}W n6 t y� } z |jtjkr�� g }W Y d }~n d }~0 0 |s�|r:q�q:|D ]\}}|�|j|� q�q:|�� d S )NFT)�timeout)r� � selectorsZDefaultSelector�registerr� Z EVENT_READ�_acceptnewconnectionr� �_handlemainipcr� r� Zselectr� �OSErrorr� Z ETIMEDOUTr" Zfileobjr� )r Zexiting�h�selectorZeventsr� �keyZ_maskr r r r� u s2 � � z unixforkingservice._mainloopc C s( | j }z|�� \}}W nF tjy\ } z,|jd tjkrFW Y d }~d S � W Y d }~n d }~0 0 t�� }|r�z0| j � dd|� | j � |� |� � W |�� n |�� 0 n|zD|�� |�� | j�� | �|� |�� | j�� t�d� W n6 z | j jdd� W t�d� n t�d� 0 Y n0 d S )Nr rR s forked worker process (pid=%d) T)�forcer� )r� Zacceptr� r r� r� �EINTRr� �forkrX �logr� r� r� r� r� � _runworkerr� �_exitr� )r r� r� r� r� Z_addrr� �pidr r r r� � s8 � z'unixforkingservice._acceptnewconnectionc C sf z|� d�}W nF tjyT } z,|jd tjkr>W Y d}~dS � W Y d}~n d}~0 0 | j�|� dS )z#Process messages sent from a workeri � r N)Zrecvr� 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 rx �framer r r r� � s z"unixforkingservice._sigchldhandlerc C s� | j r�zt�d|�\}}W nV typ } z>|jtjkrBW Y d }~q |jtjkrP� | j �� W Y d }~d S d }~0 0 |dkr~d S | j� dd|� | j � |� q d S )Nr7 r rR s worker process exited (pid=%d) ) r� r� �waitpidr� r� r� ZECHILD�clearrX r� �discard)r �optionsr� Z_statusr� r r r r� � s z unixforkingservice._reapworkersc C sT t � t j| j� t� | j}z(t| j| j||j| j gd� W t � � n t � � 0 d S )Nr� ) rx r� r� r� r� r� rX rd r� � _reposetup�gcZcollect)r r� r� r r r r� � s �z unixforkingservice._runworkerc sl |� � s d S G � fdd�d|j�� � |_| j|_| j�|j�}|d u rJd 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 n" tjyB | j�dd� Y n0 d S )NrR s# failed to send repo root to master ) �superr� � _cmdserveripc�send�rootr� r rX r� r )r� �unixcmdserverrepor r r� � s �z>unixforkingservice._reposetup.<locals>.unixcmdserverrepo.close)r, r- r. r� � __classcell__r �r� )r� r r� � s r� s repocaches repo from cache: %s ) �localr� r� r� r� rV r� rX r� r Z copycache)r rX rd Z cachedrepor r� r r� � s z unixforkingservice._reposetup)N)r, r- r. r/ r r� r� r� r� r� r� r� r� r� r� r� r r r r r� 3 s %& r� )NN)+Z __future__r r� r� r� r� rx r� r r� r� Z BaseSelector� ImportErrorZ thirdpartyr Zi18nr r r r� r r r r r r r� Zutilsr r �objectr r1 r5 rU r[ r\ rs r� r� r� r� r� r r r r �<module> s> $ X� R )%"