
    	^c.                         d dl Z d dlZd dlZd dlZd dlZd dlZd dlZddlmZ ddl	m
Z
 ddlmZmZm	Z	mZ ddlmZ d Ze j        d             Zd	 Z G d
 d          Zd ZdS )    N   )_)getattr)encodingerrorpycompatutil)procutilc                  
   t          j        t          j                              } t          j                            d          r>	 | dt          j        d          j	        z  z  } n# t          t          t          f$ r Y nw xY w| S )a
  Return a string which is used to differentiate pid namespaces

    It's useful to detect "dead" processes and remove stale locks with
    confidence. Typically it's just hostname. On modern linux, we include an
    extra Linux-specific pid namespace identifier.
    s   linuxs   /%xs   /proc/self/ns/pid)r   
strtolocalsocketgethostnamer   sysplatform
startswithosstatst_inoFileNotFoundErrorPermissionErrorNotADirectoryError)results    0/usr/lib/python3/dist-packages/mercurial/lock.py_getlockprefixr      s      !3!5!566F&&x00 	frw';<<CCCFF!?4FG 	 	 	D	Ms   A& &B ?B c               #   D  K   g di } d fd}	 dD ]5}t          t          |d          }|r|| vrt          j        |          | |<   6	 | D ]}t          j        ||           n# t          $ r Y nw xY wddV  d	 |                                 D ]\  }}t          j        ||           nW# t          $ r Y nKw xY w# d	 |                                 D ]\  }}t          j        ||           w # t          $ r Y w w xY wxY wr d                    dS dS )a@  Block signal interrupt while doing something critical

    This makes sure that the code block wrapped by this context manager won't
    be interrupted.

    For Windows developers: It appears not possible to guard time.sleep()
    from CTRL_C_EVENT, so please don't use time.sleep() to test if this is
    working.
    Fc                     | t          t          dd           k    s| t          t          dd           k    rt          t          j        )NSIGINTCTRL_C_EVENT)r   signalKeyboardInterruptr   SignalInterrupt)nums    r   raiseinterruptz)_delayedinterrupt.<locals>.raiseinterrupt=   sI    '&(D1111SGND=
 =
 6
 6
 $#''    c                 P    r                     |            d S  |            d S N)append)r!   frameassertedsigsblockedr"   s     r   	catchtermz$_delayedinterrupt.<locals>.catchtermE   s;     	 $$$$$N3r#   )s   CTRL_C_EVENTs   SIGINTs   SIGBREAKs   SIGHUPs   SIGTERMNTr   )r   r   	getsignal
ValueErroritems)orighandlersr*   namer!   handlerr(   r)   r"   s        @@@r   _delayedinterruptr1   .   s      LGL( ( (             
 		: 		:D &$--C :s,..$*$4S$9$9S!	# . .c9----. 	 	 	D	 
 	 , 2 2 4 4 , ,Wc7++++, 	 	 	D		 	 , 2 2 4 4 , ,Wc7++++, 	 	 	D	
  (|A'''''( (se   8C A+ *C +
A85C 7A88	C /B4 4
C CD/C87D8
DDDDc                    fd}t          ||dg|R ddi||r|rdnd}d}	|sd}	n|r|}	d}
	 	                                  n# t          j        $ r}|
|k    r || j        |j                   |
|	k    r || j        |j                   ||
k    r0t          j        t          j        |j	        j
        |j                  t          j        d           |
dz  }
Y d}~nd}~ww xY w|
_        j        r]d|	cxk    rj        k    r.n n+|                     t          d	          j        z             n|                     d	j        z             j        r                                 S )
zreturn an acquired lock or raise an a LockHeld exception

    This function is responsible to issue warnings and or debug messages about
    the held lock while trying to acquires it.c                 P   d|v rj|                     dd          \  }}t          d          t          j        j                  t          j        |          t          j        |          fz  }n+t          d          j        t          j        |          fz  } | |           dS )z=issue the usual "waiting on lock" message through any channel   :r   s5   waiting for lock on %s held by process %r on host %r
s"   waiting for lock on %s held by %r
N)splitr   r   bytestrdesc)printerlockerhostpidmsgls        r   printwarningztrylock.<locals>.printwarningw   s     6>>T1--ID#I   (( %% &&CC :;; ((? C 	r#   r   dolockFTr   Ns   got lock after %d seconds
)lock_trylockr   LockHelddebugr9   warnerrno	ETIMEDOUTfilenamer7   timesleepdelayr   	acquirefn)uivfslocknametimeoutwarntimeoutargskwargsr>   debugidx
warningidxrK   instr=   s               @r   trylockrW   q   s       & 	S(A====U=f==A 5W5qq2HJ !

	 ! 
E	JJLLL~ 
	 
	 
	  RXt{333
""RWdk222%nOT]AFDK   JqMMMQJEEEEEE
		  AGw ?
%%%%ag%%%%%GGA455?@@@@HH3ag=>>>{ 	Hs   A C'BC""C'c                   n    e Zd ZdZdZ	 	 	 	 	 	 ddZd Zd Zd Zd	 Z	d
 Z
d Zd Zd Zd Zd ZddZdS )rA   a  An advisory lock held by one process to control access to a set
    of files.  Non-cooperating processes or incorrectly written scripts
    can ignore Mercurial's locking scheme and stomp all over the
    repository, so don't do that.

    Typically used via localrepository.lock() to lock the repository
    store (.hg/store/) or localrepository.wlock() to lock everything
    else under .hg/.Nr@   Tc	                 \   || _         || _        d| _        || _        || _        || _        || _        |rt          | _        nt          j
        | _        g | _        |                                 | _        |r6|                                 | _        | j        r|                                  d S d S d S )Nr   )rN   fheldrP   	releasefnrL   r7   r1   _maybedelayedinterruptr	   nullcontextmanagerpostrelease_getpidr;   rA   rK   )	selfrN   fnamerP   r\   rL   r7   
signalsafer?   s	            r   __init__zlock.__init__   s     	""	 	B*;D''*.*AD'<<>> 	!DJ~ !     	! 	!! !r#   c                     | S r%    ra   s    r   	__enter__zlock.__enter__   s    r#   c                 j    t          d |||fD                       }|                     |           d S )Nc              3      K   | ]}|d u V  	d S r%   rf   ).0as     r   	<genexpr>z lock.__exit__.<locals>.<genexpr>   s&      GGAa4iGGGGGGr#   )success)allrelease)ra   exc_type	exc_valueexc_tbrn   s        r   __exit__zlock.__exit__   s?    GG(Iv)FGGGGGW%%%%%r#   c                     | j         r#t          j        dt          d           d| _         |                                  d S )Nz$use lock.release instead of del lock   )category
stacklevelr   )r[   warningsrE   DeprecationWarningrp   rg   s    r   __del__zlock.__del__   sH    9 		M6+    DIr#   c                 (    t          j                    S r%   )r
   getpidrg   s    r   r`   zlock._getpid   s       r#   c                 2   | j         }	 	 |                                  | j         |z
  S # t          j        $ r_}|dk    r$t	          j        d           |dk    r|dz  }Y d }~Yt          j        t          j        |j        | j	        |j
                  d }~ww xY w)NTr   r   )rP   rB   r   rC   rI   rJ   rF   rG   rH   r7   r9   )ra   rP   rV   s      r   rA   z	lock.lock   s    ,	|g-->   a<<JqMMM{{1HHHHnOT]DIt{  s   ( B%B!0BBc                    | j         r| xj         dz  c_         d S t          j        t                      t          _        dt          j        | j        fz  }d}| j         s;|r8|dz  }	 |                                 5  | j                            || j                   d| _         d d d            n# 1 swxY w Y   n# t          t          f$ r}|j        t          j        k    rv|                                 }|Y d }~|                     |          }|Ct          j        t          j        | j                            | j                  | j        |          n+t          j        |j        |j        |j        | j                  Y d }~nd }~ww xY w| j         s|8| j         sCt          j        t          j        | j                            | j                  | j        d          d S )Nr   s   %s:%d   r#   )r[   rA   _hostr   r;   r]   rN   makelockrZ   OSErrorIOErrorrF   EEXIST	_readlock	_testlockr   rC   EAGAINjoinr7   LockUnavailablestrerrorrH   )ra   rO   retrywhyr9   s        r   rB   zlock._trylock	  s.   9 	IINIIF:'))DJtz4844) 	 	QJE0022 " "H%%h777 !DI" " " " " " " " " " " " " " " W%   9,,!^^--F~ !^^F33F)#n!L HMM$&11 I"	   *  /	3<ty   *)))) ) 	 	2 y 	 .dhmmDF33TY  	 	sC   'B; ;(B/#B; /B33B; 6B37B; ;F+F<BFFc                 d    	 | j                             | j                  S # t          $ r Y dS w xY w)zread lock and return its value

        Returns None if no lock exists, pid for old-style locks, and host:pid
        for new-style locks.
        N)rN   readlockrZ   r   rg   s    r   r   zlock._readlock1  sA    	8$$TV,,,  	 	 	44	s   ! 
//c                     |dS 	 |                     dd          \  }}n# t          $ r Y dS w xY w|t          j        k    rdS 	 t	          |          }n# t          $ r Y dS w xY wt          j        |          rdS dS )NFr4   r   T)r5   r,   rA   r   intr
   testpid)ra   r9   r:   r;   s       r   _lockshouldbebrokenzlock._lockshouldbebroken<  s    >5	T1--ID## 	 	 	55	4:5	c((CC 	 	 	55	C   	5ts     
..A 
A"!A"c                    |                      |          s|S 	 t          | j        | j        dz   d          5  |                                 }|                      |          s|cd d d            S | j                            | j                   d d d            d S # 1 swxY w Y   d S # t          j        $ r |cY S w xY w)Ns   .breakr   )rP   )r   rA   rN   rZ   r   unlinkr   	LockErrorra   r9   s     r   r   zlock._testlockM  s4   ''// 	M	dh 2A>>> ( ())//77 "!( ( ( ( ( ( ( ( '''	( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (
  	 	 	MMM	s@   B) +B#B) 0BB) B  B) #B $B) )B=<B=c                 T    |                                  }|                     |          S )ao  return id of locker if lock is valid, else None.

        If old-style lock, we cannot tell what machine locker is on.
        with new-style lock, if locker is on this machine, we can
        see if locker is alive.  If locker is on this machine but
        not alive, we can safely break lock.

        The lock file is only deleted when None is returned.

        )r   r   r   s     r   testlockzlock.testlock\  s%     !!~~f%%%r#   c                    | j         dk    r| xj         dz  c_         dS | j         dk    rd| _         |                                 | j        k    rdS 	 | j        r|                                  	 | j                            | j                   nE# t          $ r Y n9w xY w# 	 | j                            | j                   w # t          $ r Y w w xY wxY w| j        D ]} ||           d| _        dS dS )zrelease the lock and execute callback function if any

        If the lock has been acquired multiple times, the actual release is
        delayed to the last release call.r   r   N)	r[   r`   r;   r\   rN   r   rZ   r   r_   )ra   rn   callbacks      r   rp   zlock.releasej  s)   
 9q==IINIIIIY!^^DI||~~))> %NN$$$HOODF++++   DHOODF++++   D !, " "!!!!#D% ^sB   B ,B 
BBCB>=C>
CC
CC)r@   NNNTT)T)__name__
__module____qualname____doc__r   rd   rh   rt   r{   r`   rA   rB   r   r   r   r   rp   rf   r#   r   rA   rA      s         " E ! ! ! !:  & & &  ! ! !   & & &P	 	 	  "  & & &$ $ $ $ $ $r#   rA   c                  <    | D ]}||                                  d S r%   )rp   )locksrA   s     r   rp   rp     s/      LLNNN r#   )
contextlibrF   r   r   r   rI   ry   i18nr   r   r    r   r   r	   utilsr
   r   contextmanagerr1   rW   rA   rp   rf   r#   r   <module>r      sF        				                                     ?( ?( ?(D; ; ;|T$ T$ T$ T$ T$ T$ T$ T$n    r#   