q&f| ddlZddlZddlZddlZddlZddlZddlmZmZm Z m Z  ddlm Z e jejdk(r9ddlZddlZej$ZdZdZdZd Zej.d j0Zd Zd Zd ZdZdZdZdZ dZ!dZ"ej$jFZ$GddejJZ&ejNejPZ)ejTjVZ,e,jZZ.ej^ejPejPej`ejPejPejbge._2ejbe._3e,jhZ4ejbge4_2ejje4_3e,jlZ6ejbej`ejPe)ejNe&ge6_2ejje6_3e,jnZ7ejbej`ejPe)ejNe&ge7_2ejje7_3e,jpZ8ge8_2ejPe8_3e,jrZ9ejPge9_2de9_3e,jtZ;ejPej`ejPejPejNej^ejPej`ge;_2ejPe;_3e,jxZe,ddZ?e?KejbejNe&e)ejPejjge?_2ejje?_3e,jZ@ejbejPejjge@_2ejPe@_3e,jZBe)ejjejjej^geB_2ejbeB_3e,jZCejbejNe&geC_2ejjeC_3dZDdZEeErdZFndZFdZGGddeHZIGddeIZJGddeIZKGd d!eIZLGd"d#eIZMGd$d%eIZNGd&d'eIZOGd(d)ZPGd*d+ZQGd,d-ePZRd.ZSGd/d0ePZTd@d1ZUGd2d3ePZVGd4d5eQZWGd6d7eWZXGd8d9eWZYGd:d;eYeXZZGd<d=eQZ[Gd>d?Z\y#e $r ddlmZ Y3wxYw)AN) capabilitiescompatencodingload)bser)pybserntli@iliicpeZdZdefdefdej fdej fdej fgZdZy) OVERLAPPEDInternal InternalHighOffset OffsetHighhEventcJd|_d|_d|_d|_d|_yNr)rrrrrselfs E/usr/lib/python3/dist-packages/hgext/fsmonitor/pywatchman/__init__.py__init__zOVERLAPPED.__init__Vs&DM !D DKDODKN) __name__ __module__ __qualname__ ULONG_PTRwintypesDWORDHANDLE_fields_rrrrrMsB  # Y ' x~~ & 8>> * x '   rrGetOverlappedResultEx Fc ~tdtjdtjd||ddzy)N[z%a, %d %b %Y %H:%M:%Sz] )printtimestrftimegmtimefmtargss rlogr2s1  5t{{}Ed1g   rcyNr&r/s rr2r2 rc tj}tttzt zd|d|dd |j t|S#t|wxYw)z7expand a win32 error code into a human readable messageNr)ctypesc_char_p FormatMessageFORMAT_MESSAGE_FROM_SYSTEMFORMAT_MESSAGE_ALLOCATE_BUFFERFORMAT_MESSAGE_IGNORE_INSERTSvalue LocalFree)errbufs r_win32_strerrorrAsZ // C" ( ) ' (       yy# #s A Ac eZdZddZdZdZy) WatchmanErrorNc ||_||_yr4)msgcmd)rrErFs rrzWatchmanError.__init__src||_yr4)rF)rrFs r setCommandzWatchmanError.setCommands rch|jr|jd|jS|jS)Nz, while executing )rFrErs r__str__zWatchmanError.__str__s% 88/3xxB BxxrNN)rrr rrHrJr&rrrCrCsrrCc eZdZy)BSERv1UnsupportedNrrr r&rrrMrMrrMc eZdZy) UseAfterForkNrNr&rrrQrQrOrrQc eZdZdfd ZxZS)WatchmanEnvironmentErrorcPtt| dj||||y)Nz{0}: errno={1} errmsg={2})superrSrformat)rrEerrnoerrmsgrF __class__s rrz!WatchmanEnvironmentError.__init__ s' &6 ' . .sE6 BC rr4rrr r __classcell__rYs@rrSrSs   rrSceZdZfdZxZS)SocketConnectErrorcTtt| d|d|||_||_y)Nzunable to connect to z: )rUr^rsockpathexc)rr`rarYs rrzSocketConnectError.__init__s+  $0-5s ; ! rrZr\s@rr^r^s rr^ceZdZdZy) SocketTimeoutaA specialized exception raised for socket timeouts during communication to/from watchman. This makes it easier to implement non-blocking loops as callers can easily distinguish between a routine timeout and an actual error condition. Note that catching WatchmanError will also catch this as it is a super-class, so backwards compatibility in exception handling is preserved. Nrrr __doc__r&rrrcrcsrrcc$eZdZdZdfd ZxZS) CommandErrorzRerror returned by watchman self.msg is the message returned by watchman. c4tt| d||y)Nzwatchman command error: )rUrgr)rrErFrYs rrzCommandError.__init__(s lD*,/ 13 rr4)rrr rerr[r\s@rrgrg"s   rrgc2eZdZdZdZdZdZdZdZdZ y) Transportz.communication transport to the watchman serverNct)z tear it downNotImplementedErrorrs rclosezTransport.close3 !##rct)zread size bytesrl)rsizes r readByteszTransport.readBytes7rorct)zwrite some datarl)rr@s rwritezTransport.write;rorcyr4r&rr=s r setTimeoutzTransport.setTimeout?r5rc|jg|_t|jdk(r=d|jdvr,|jdjdd\}}|g|_|S |jd}d|vr=dj |j}|jdd\}}|g|_||zS|jj |n)zread a line Maintains its own buffer, callers of the transport should not mix calls to readBytes and readLine. r rr r)r@lensplitrrjoinappend)rlinebresults rreadLinezTransport.readLineBs 88 DH txx=A %488A;"6 ))%3IT1sDHKt$Az$((+GGE1- q3}$ HHOOA r) rrr rer@rnrrrtrwrr&rrrjrj.s#8 C$$$ rrjc,eZdZdZdZdZdZdZdZy)Codecz.communication encoding for the watchman serverNc||_yr4) transport)rrs rrzCodec.__init__`s "rctr4rlrs rreceivez Codec.receivec !##rctr4rlrr1s rsendz Codec.sendfrrc:|jj|yr4)rrwrvs rrwzCodec.setTimeoutis !!%(r) rrr rerrrrrwr&rrrr[s8I#$$)rrc2eZdZdZdZdZdZdZdZdZ y)UnixSocketTransportz"local unix domain socket transportNc||_||_tjtjtj} |j |j|j |j||_y#tj$r+}|jt|j|d}~wwxYwr4) r`timeoutsocketAF_UNIX SOCK_STREAM settimeoutconnectsockerrorrnr^)rr`rres rrzUnixSocketTransport.__init__rs   }}V^^V-?-?@ 7 OODLL ) LL 'DI|| 7 JJL$T]]A6 6 7s=BB>&B99B>c`|jr"|jjd|_yyr4)rrnrs rrnzUnixSocketTransport.closes# 99 IIOO DI rc\||_|jj|jyr4)rrrrvs rrwzUnixSocketTransport.setTimeouts  T\\*rc |jj|g}|ds td|dS#tj$r t dwxYw)Nrempty watchman responseztimed out waiting for response)rrecvrCrrrc)rrqr@s rrrzUnixSocketTransport.readBytessZ B99>>$'(Cq6#$=>>q6M~~ B @A A Bs 03Ac |jj|y#tj$r t dwxYw)Nztimed out sending query command)rsendallrrrc)rdatas rrtzUnixSocketTransport.writes; C II  d #~~ C AB B Cs=) rrr rerrrnrwrrrtr&rrrrms%, D 7 +BCrrcxtd||dk7rt|j||}|tk(rn{|tk(rt tnb|t k(rt t y|tk(r t}tdt|yt}tdt|yt|||dS)aWindows 7 and earlier does not support GetOverlappedResultEx. The alternative is to use GetOverlappedResult and wait for read or write operation to complete. This is done be using CreateEvent and WaitForSingleObjectEx. CreateEvent, WaitForSingleObjectEx and GetOverlappedResult are all part of Windows API since WindowsXP. This is the exact same implementation that can be found in the watchman source code (see get_overlapped_result_ex_impl in stream_win.c). This way, maintenance should be simplified. z"Preparing to wait for maximum %dmsrFz WaitForSingleObjectEx failed: %szUnexpected error: %s) r2WaitForSingleObjectExr WAIT_OBJECT_0WAIT_IO_COMPLETION SetLastError WAIT_TIMEOUT WAIT_FAILED GetLastErrorrAGetOverlappedResult)pipeolapnbytesmillis alertablewaitReturnCoder?s r_get_overlapped_result_ex_implrs,f5 {.t{{FIN ] *  1 1 + ,  | +  & { *.C 2OC4H I.C &(< = tT65 99rc:eZdZdZdZdZdZdZdZdZ dZ y ) WindowsNamedPipeTransportzconnect to a named pipec ||_ttj|dz|_d|_t jrtj|}t|ttzddttd|_t!}|jt"k(s|jdk(r-d|_t%|j|j'd|t)dddd|_t!}|j*|j-d|t.|_tj2ddk(s |j0 t4|_yy) NrTFzCreateEvent failedWATCHMAN_WIN7_COMPAT1)r`intmathceilr_iobufrPYTHON3osfsencode CreateFile GENERIC_READ GENERIC_WRITE OPEN_EXISTINGFILE_FLAG_OVERLAPPEDrrINVALID_HANDLE_VALUEr^ _make_win_err CreateEvent _waitable_raise_win_errr'_get_overlapped_result_exgetenvr)rr`rr?s rrz"WindowsNamedPipeTransport.__init__s  499Wt^45  >>{{8,H  = (     n 99, , QDI$T]]D4F4Fr34OP P%T4=n >> !    4c :)>& II, - 4--5-KD *6rc&|j||r4)rrrEr?s rrz(WindowsNamedPipeTransport._raise_win_errs  c**rc6td||t|fzS)Nz%s win32 error code: %d %s)IOErrorrArs rrz'WindowsNamedPipeTransport._make_win_errs# (Coc6J+K K  rc|jr tdt|jd|_|jt|jd|_y)Nz Closing pipe)rr2 CloseHandlerrs rrnzWindowsNamedPipeTransport.closesA 99    " >> %  'rc*t|dz|_y)Nr)rrrvs rrwz$WindowsNamedPipeTransport.setTimeouts54<( rc|jrR|t|jk\r|j}d|_|S|jd|}|j|d|_|Stj|}t }|j |_td|t|j||d|}|s(t}|tk7r|jd|z|tj}|j!|j|||rdn |j"dsjt}t%|j||t&k(r#tdt)d|j"ztd||jd ||j*}|dk(r!t%|j|t-d |d|}t/||}||k(r|S||d|_|d|S) zA read can block for an unbounded amount of time, even if the kernel reports that the pipe handle is signalled, so we need to always perform our reads asynchronously Nzmade read buff of size %dzfailed to read %d bytesrTzGetOverlappedResultEx timedoutz%timed out after waiting %dms for readz&GetOverlappedResultEx reports error %dzerror while waiting for readz'Async read yielded 0 bytes; unpossible!)rrzr@r7create_string_bufferrrrr2ReadFilerrERROR_IO_PENDINGrr"r#rr CancelIoExrrcr=rmin) rrqresr@r immediater?nread returned_sizes rrrz#WindowsNamedPipeTransport.readBytess ;;s4;;''kk ++et$C++de,DKJ))$/|nn  '.TYYT4> .C&&##$=$DcJ -- IItUA d .C tyy$ 'l"45#;dllJ 8# >    > D  A: tyy$ 'CD D&5kE4( E !J-.) >M""rct}|j|_t|jt j |t|d|}|s=t}|tk7r*|jdt||jfz|tj}|j|j|||rdn |jdr"td|j |j St}t#|j||t$k(rt'd|jz|jdt|z|y)Nz%failed to write %d bytes to handle %rrTzmade write of %d bytesz&timed out after waiting %dms for writez)error while waiting for write of %d bytes)rrr WriteFilerr7r8rzrrrr"r#rrr2r=rrrc)rrrrr?nwrotes rrtzWindowsNamedPipeTransport.writeAs|nn  IIvt,c$it .C&&##;4y$)),-!  ) ) IItV)Qt  (&,, 7<< n 499d# , 84<<G   7#d) CS rN) rrr rerrrrnrwrrrtr&rrrrs,!!LF+ )B#H% rrcJ|r|StjjddS)NWATCHMAN_BINARYwatchman)renvironget)binpaths r_default_binpathris" ::>>+Z 88rc8eZdZdZdZdZd dZdZdZdZ dZ y) CLIProcessTransporta"open a pipe to the cli to talk to the service This intended to be used only in the test harness! The CLI is an oddball because we only support JSON input and cannot send multiple commands through the same instance, so we spawn a new process for each command. We disable server spawning for this implementation, again, because it is intended to be used only in our test harness. You really should not need to use the CLI transport for anything real. While the CLI can output in BSER, our Transport interface doesn't support telling this instance that it should do so. That effectively limits this implementation to JSON input and output only at this time. It is the responsibility of the caller to set the send and receive codecs appropriately. NTc@||_||_t||_yr4)r`rrr)rr`rrs rrzCLIProcessTransport.__init__s   '0 rcP|jr|jj|jj|jjj |jj j |jj d|_yyr4)procpidkillstdinrnstdoutwaitrs rrnzCLIProcessTransport.closesi 99yy}}(  IIOO ! ! # II   " " $ IINN DI rc|jr |jS|jdj|jddddddg}t j |tj tj |_|jS) Nz--sockname={0}z--logfile=/BOGUSz--statefile=/BOGUSz --no-spawnz --no-localz --no-prettyz-j)rr)rrrVr` subprocessPopenPIPErs r_connectzCLIProcessTransport._connectst 9999  LL  # #DMM 2       $$     yyrc|j|jjj|}|s t d|S)NzEOF on CLI process transport)rrrreadrC)rrqrs rrrzCLIProcessTransport.readBytess9 ii##D) >? ? rc|jr|jd|_|j}|jj |}|jjd|_|S)NFT)closedrnrrrt)rrrrs rrtzCLIProcessTransport.writesT ;; JJLDK}}jjt$   rr4) rrr rerrrrnrrrrtr&rrrrrs,& D F1 $rrc4eZdZdZfdZdZdZdZxZS) BserCodecz>#J#J 9zI;789||~ l "#0   ' 2 !D %&D " !D %&D "rcF|jjtg}|ds tdt j |d\}}}t |dr/t|j||_|j|z|_ t|d}||kDrD|j|jj||z |t|dz }||kDrDdj|} |j|}|S#t$r}td|zd}~wwxYw)Nrrrr rr)rrrr rCrr hasattrmaxrrrrzr}r|rr ) rr@recv_bser_versionrecv_bser_capabilitiesrrrrrs rrzBser2WithFallbackCodec.receive s~~'' 231v 9: ::>--A:O714 4 ( #D$5$57H ID  $ 6 69O OD 3q6{Tk JJt~~//t < = CBL DTk88C= J++h'CJ J Dq HI I Js1D D  DD ct|dr+tj||j|jd}ntj|}|j j |y)Nr)rr)r"rrrrrrtrs rrzBser2WithFallbackCodec.send$sQ 4 (**))!33C **d#C S!r)rrr rerrrr[r\s@rrrs'4J0 "rrceZdZdZy)ImmutableBser2CodecrNrdr&rrr(r(0s   rr(c2eZdZdZdZfdZdZdZxZS) JsonCodecz>**W%C S5[)r) rrr rer,rrrr[r\s@rr*r*7sF D *rr*ceZdZdZdZdZdZdZdZdZ iZ iZ gZ ddgZ dZdZdZ ddZdZdZdZd Zd Zd Zd Zd ZdZdZdZddZddZdZddZ dZ!y)clientz3Handles the communication with the watchman serviceNr2 subscriptionc ~||_||_||_t| |_t j |rt|tr||_ ny|xstjdxsd}|dk(rtjdk(r t|_ n8|dk(r t|_ n'|dk(rt|_ |d}||}nt!d|zt#|xstjdxsd}t#|xstjdxsd}|j%||_|j%||_|d urNt*j,r/t/j0|_t.j4|_yd|_d|_y||_|d urt.j4|_y||_y) NWATCHMAN_TRANSPORTlocalr clir,zinvalid transport %sWATCHMAN_ENCODINGrF)r`ruseImmutableBserrrinspectisclass issubclassrjrrrnamerrrrCstr_parseEncoding recvCodec sendCodecrrrget_local_encoding valueEncodingdefault_local_errors valueErrors) rr`rr sendEncoding recvEncodingr;rErGrs rrzclient.__init__ms!  0'0 ??9 %*Y *J&DN!ORYY/C%DOIG#4!:g%!4e#!4'#)L'#/L#$:Y$FGG  DBII&9: Df   DBII&9: Df ,,\:,,\: E !~~%-%@%@%B"#+#@#@ %)"#' !.D e##+#@#@ #. rcfd}|S)Nc@|jjSr4)rErG)rcodecrs r make_codecz)client._makeBSERCodec..make_codecsD$6$68H8HI Irr&)rrLrMs`` r_makeBSERCodeczclient._makeBSERCodecs JrcV|dk(r6|jr|jtS|jtS|dk(rQtj r t d|jr|jtS|jtS|dk(rtStd|z)Nrzbser-v1zrPython 3 does not support the BSER v1 encoding: specify "bser" or omit the sendEncoding and recvEncoding argumentsr,zinvalid encoding %s) r;rNr(rrrrMrrr*rC)rencs rrAzclient._parseEncodings &=$$**+>??&&'=> > I ~~'  $$**+=>>&&y1 1 F]  5 ;<  IJ JF# f  86'? JK Kj!! N H1 LM M NsBD D6#D11D6c |jr-|jtjk7r t dy|j |j |_i}|jtk(r|j|d<|j|j |jfi||_ |j|j|_ |j|j|_tj|_y)zestablish transport connectionz@do not re-use a connection after fork; open a new client insteadNr)recvConnrrgetpidrQr`rbrrrrtportrCsendConnrB)rkwargss rrzclient._connects ==xx299;&"V  ==  113DM >>0 0 $ F9 $T^^DMM4<<J6J tzz2 tzz2 99;rc$|jyr4rnrs r__del__zclient.__del__  rc&|j|Sr4)rrs r __enter__zclient.__enter__s  rc$|jyr4rj)rexc_type exc_value exc_tracebacks r__exit__zclient.__exit__rlrc||jr0|jjd|_d|_d|_yyr4)rfrnrdrgrs rrnz client.closes3 :: JJ   DJ DM DM rc|j|jj}|j|drt |d|j|dr|j j |d|j|dr|d}||jvrg|j|<|j|j |tjjtjj|d}||jvri|j|<||j|vrg|j||<|j||j ||S)areceive the next PDU from the watchman service If the client has activated subscriptions or logs then this PDU may be a unilateral PDU sent by the service to inform the client of a log event or subscription change. It may also simply be the response portion of a request initiated by query. There are clients in production that subscribe and call this in a loop to retrieve all subscription responses, so care should be taken when making changes here. rr2r5root) rrdrrRrglogsr}subsrr_normpathnormcase sub_by_root)rrsubrvs rrzclient.receive!s9 &&( == )vg/ / == ' II  VE] + == 0(C499$!# # IIcN ! !& )77##BGG$4$4VF^$DED4+++)+  &$**400.0  &s+   T "3 ' . .v 6 rcFd|vr|dry|jD]}||vsyy)N unilateralTF)r~)rrks risUnilateralResponsezclient.isUnilateralResponseHs6 3 3|#4 ACx rc0|j}|rg|_|S)zRetrieve buffered log data If remove is true the data will be removed from the buffer. Otherwise it will be left in the buffer )rw)rremovers rgetLogz client.getLogQs ii DI rc|tjjtjj|}||jvry||j|vry|j||}|r+|j||=||j vr |j |=|S||j vry|j |}|r |j |=|S)aRetrieve the data associated with a named subscription If remove is True (the default), the subscription data is removed from the buffer. Otherwise the data is returned but left in the buffer. Returns None if there is no data associated with `name` If root is not None, then only return the subscription data that matches both root and name. When used in this way, remove processing impacts both the unscoped and scoped stores for the subscription data. N)rr_ryrzr{rx)rr?rrvr|s rgetSubscriptionzclient.getSubscription\s  77##BGG$4$4T$:;D4+++4++D11""4(.C$$T*40499$ $J tyy iio  $ rctd|j |jj||j }|j |r"|j }|j |r"|S#t $r'}td|j|j|d}~wt$r}|j|d}~wwxYw)a?Send a query to the watchman service and return the response This call will block until the response is returned. If any unilateral responses are sent by the service in between the request-response they will be buffered up in the client object and NOT returned via this method. zcalling client.queryz,I/O error communicating with watchman daemonN) r2rrgrrrEnvironmentErrorrSrWstrerrorrCrH)rr1reeexs rqueryz client.querys "#   MM  t $,,.C++C0lln++C0J +>      MM$   s*AA=;A== C "B(( C 4CC c|jd|xsg|xsgd}|j|ds(tj||d|vrt |d|S)z!Perform a server capability checkr)rrrr)rrRr synthesizerg)rrrrs rcapabilityCheckzclient.capabilityChecks_jj HNBO }}S.1  # #C 2#~"3w<00 rcp|jj||jj|yr4)rdrwrgrvs rrwzclient.setTimeouts&   '   'r) Ng?NNNFFFN)T)TNrK)"rrr rer`rrCrBrgrdrxr{rwr~rfr;rrrNrArRrbrrkrnrsrnrrrrrrrwr&rrr4r4\s=HIIIHH DK D(J E C>/@ =( ("T4!%N !F@ (rr4r4)]r<rrrrr,rrrrrrr  ImportErrorr r?r7ctypes.wintypesr"rrrrc_void_pr=rr:r;r<rrrrINFINITErWPARAMr! StructurerPOINTERr#LPDWORDwindllkernel32 _kernel32 CreateFileArLPSTRLPVOIDr$argtypesrestyperBOOLrrrrFormatMessageAr9r>rgetattrr'r CreateEventArrr  _debuggingr2rAr/rCrMrQrSr^rcrgrjrrrrrrrrrr(r*r4r&rrrs:  22 ! MM 77d?HLM%M*6??2.44!+%/"$.!KLM#H"&&IV%% fnnX^^,G &&I&&JJ"J''K$OO,K"--K!!Hz" H }}H##Iz" I! I))LL#>>L))L%^^,LL,,Mx~~&M%NNM##I#77z" $ #+--#I/FM( OO FNN: &  NN MM * &)1 %%;; &" %-NN!((K   K #//K%%J#??NFNN:,FGJ!J      , I    =  } M  =  **Z))$(C)(CV%:Pf f R9F)FR$"$"N    >"Y>"B 02D "*"*JT(T([!  !sU U,+U,