o ]Lb/@sddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZmZm Z mZddlmZdd Zejd d Zd d ZGdddeZddZdS))absolute_importN)_)getattr)encodingerrorpycompatutil)procutilc Csxtt}tjdr:z |dtdj 7}W|St y9}z|j t j t j t jfvr.WYd}~|Sd}~ww|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. slinuxs/%xs/proc/self/ns/pidN)r strtolocalsocket gethostnamer sysplatform startswithosstatst_inoOSErrorerrnoENOENTEACCESENOTDIR)resultexr0/usr/lib/python3/dist-packages/mercurial/lock.py_getlockprefixs  rc#sgdi}ddfdd}zRdD]}tt|d}|r+||vr+t|||<qz |D]}t||q/Wn tyBYnwddVWdz|D] \}}t||qPWn&tyeYnwdz|D] \}}t||qmWwtyYwwrd dSdS) 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. FcSs*|ttddks|ttddkrttj)NSIGINT CTRL_C_EVENT)rsignalKeyboardInterruptrSignalInterrupt)numrrrraiseinterrupt?s z)_delayedinterrupt..raiseinterruptcsr |dS|dSN)append)r"frame assertedsigsblockedr#rr catchtermGs z$_delayedinterrupt..catchterm)s CTRL_C_EVENTsSIGINTsSIGBREAKsSIGHUPsSIGTERMNTr)rr getsignal ValueErroritems) orighandlersr*namer"handlerrr'r_delayedinterrupt0sP      r1c sHfdd}t||dg|Rddi||r|rdnd}d} |s$d} n|r(|} d} zWnCtjys} z5| |krE||j| j| | krP||j| j|| kr`ttj| j j | jt d| d7} WYd } ~ nd } ~ wwq+| _ j rd| krj krnn |td j n|d j jrS) 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.csdd|vr |dd\}}tdtjt|t|f}n tdjt|f}||dS)z=issue the usual "waiting on lock" message through any channel:rs5waiting for lock on %s held by process %r on host %r s"waiting for lock on %s held by %r N)splitrrbytestrdesc)printerlockerhostpidmsglrr printwarningys  ztrylock..printwarningrdolockFTrNsgot lock after %d seconds )lock_trylockrLockHelddebugr7warnr ETIMEDOUTfilenamer5timesleepdelayr acquirefn) uivfslocknametimeout warntimeoutargskwargsr=debugidx warningidxrIinstrr;rtrylockssF  rUc@seZdZdZdZ      dddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZdddZdS)r@aAn 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 Csv||_||_d|_||_||_||_||_|rt|_nt j |_g|_ | |_ |r7||_|jr9|dSdSdS)Nr)rLfheldrN releasefnrJr5r1_maybedelayedinterruptr nullcontextmanager postrelease_getpidr9r@rI) selfrLfnamerNrXrJr5 signalsafer>rrr__init__s$    z lock.__init__cCs|Sr$rr]rrr __enter__szlock.__enter__cCs(tdd|||fD}|j|ddS)Ncss|]}|duVqdSr$r).0arrr sz lock.__exit__..)success)allrelease)r]exc_type exc_valueexc_tbrfrrr__exit__sz lock.__exit__cCs(|jrtjdtddd|_|dS)Nz$use lock.release instead of del lock)category stacklevelr)rWwarningsrDDeprecationWarningrhrarrr__del__s z lock.__del__cCstSr$)r getpidrarrrr\sz lock._getpidc Cs~|j} z ||j|WStjy>}z#|dkr.td|dkr(|d8}WYd}~qttj|j|j |j d}~ww)NTrr) rNrArrBrGrHrrErFr5r7)r]rNrTrrrr@s    z lock.lockc Cs`|jr |jd7_dStjdurtt_dtj|jf}d}|js|r|d8}z!||j||jd|_Wdn1sCwYWnMt t fy}z?|j t j kr| }|durgWYd}~q||}|durtt j|j|j|j|n t|j |j|j|jWYd}~nd}~ww|js|s$|jstt j|j|j|jddS)Nrs%s:%d)rWr@_hostrr9rYrLmakelockrVrIOErrorrEEXIST _readlock _testlockrrBEAGAINjoinr5LockUnavailablestrerrorrF)r]rMretrywhyr7rrrrA sR         z lock._trylockc CsJz|j|jWSttfy$}z|jtjkrWYd}~dSd}~ww)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)rLreadlockrVrrxrr)r]rrrrrz3s zlock._readlockcCsx|durdSz |dd\}}Wn tyYdSw|tjkr"dSzt|}Wn ty2YdSwt|r:dSdS)NFr2rT)r3r,r@rvintr testpid)r]r7r8r9rrr_lockshouldbebroken@s"     zlock._lockshouldbebrokencCs||s|Sz:t|j|jddd$|}||s'|WdWS|j|jWdWdS1s:wYWdStjyM|YSw)Ns.breakr)rN)rr@rLrVrzunlinkr LockErrorr]r7rrrr{Qs  &zlock._testlockcCs|}||S)aoreturn 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. )rzr{rrrrtestlock`s z lock.testlockcCs|jdkr|jd8_dS|jdkr]d|_||jkrdSz|jr'|Wz |j|jWnty:Ynwz |j|jWwtyMYww|jD]}||qQd|_dSdS)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.rrN) rWr\r9rXrLrrVrr[)r]rfcallbackrrrrhns0       z lock.release)r?NNNTT)T)__name__ __module__ __qualname____doc__rvr`rbrlrrr\r@rArzrr{rrhrrrrr@s* ( r@cGs|D] }|dur |qdSr$)rh)locksr@rrrrhs rh) __future__r contextlibrrrr rGrpi18nrrrrrr utilsr rcontextmanagerr1rUobjectr@rhrrrrs&     B> Z