q&f]4ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZmZGddZGdd ZGd d Zd d iZdZGddZddZGddZdZdZGddZ GddZ!y)N)_)encodingerror loggingutilpycompat repocacheutilvfs)cborutilprocutilc2eZdZdZdZedZdZdZy)channeledoutputz^ Write data to out in the following format: data length (unsigned int), data c ||_||_yN)outchannel)selfrrs 9/usr/lib/python3/dist-packages/mercurial/commandserver.py__init__zchanneledoutput.__init__*s c d|jzSNs <%c-channel>rrs rnamezchanneledoutput.name.--rc |sy|jjtjd|jt ||z|jj y)N>cI)rwritestructpackrlenflush)rdatas rr zchanneledoutput.write2s@  v{{64<<TCdJK rcL|dvr t|t|j|SN)isattyfilenotellseek)AttributeErrorgetattrrrattrs r __getattr__zchanneledoutput.__getattr__9' 7 7 & &txx&&rN) __name__ __module__ __qualname____doc__rpropertyrr r0rrrr"s*..'rrc&eZdZdZdZdZdZdZy)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. TcBt|||_||_||_yr)r_coutr _encodefn)rrr encodenameencodefns rrzchanneledmessage.__init__Ms$S'2 " !rc tj|}|||d<|jj|j |y)Nsdata)r byteskwargsr;r r<)rr%optss rr zchanneledmessage.writeRs<##D)   DM -.rc.t|j|Sr)r-r;r.s rr0zchanneledmessage.__getattr__Xstzz4((rN)r2r3r4r5 structuredrr r0r7rrr9r9?sJ" / )rr9cVeZdZdZdZdZedZd dZdZ d dZ dZ d Z e Z d Zy ) 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 ic.||_||_||_yr)in_rr)rrGrrs rrzchanneledinput.__init__ks rc d|jzSrrrs rrzchanneledinput.nameprrc|dkrR|j}|j||j}|}|r$|j||j}||z }|r$|S|j||jSNr) maxchunksize_readrrsizesbufs rreadzchanneledinput.readtsp !8$$D 4.ACJJtT\\2qJ::dDLL1 1rcN|sy|dkDsJ|jjtjd|||jj |j j d}tjd|d}|sy|j j |S)Nrrr>I)rr r!r"r$rGrQunpack)rrNrlengths rrLzchanneledinput._readsaxx v{{67D9: q!uf-a088==( (rc|dkr`|j}|j|d}|}|r<|jds+|j|d}||z }|r|jds+|S|j|dS)NrL )rKrLendswithrMs rreadlinezchanneledinput.readlinesz !8$$D 4&ACAJJu-JJtT*qAJJu-J::dD) )rc|Srr7rs r__iter__zchanneledinput.__iter__s rc6|j}|st|Sr)r[ StopIteration)rls rnextzchanneledinput.nexts MMO rcL|dvr t|t|j|Sr')r,r-rGr.s rr0zchanneledinput.__getattr__r1rN))r2r3r4r5rKrr6rrQrLr[r]ra__next__r0r7rrrErE\sL L .. 2) * H'rrEscborcJdjtj|SNr)joinr streamencode)vs rrjssxx 5 5a 89rc|jdd}|D]}tj|}|s||fcStjddj |z)N cmdserversmessage-encodingss"no supported message encodings: %s ) configlist_messageencodersgetrAbortrg)uiencnamesnfs r_selectmessageencoderrvsb}}\+?@H    # a4K ++- (0CC rcXeZdZdZddZdZdZdZdZdZ d Z d Z d Z e e d Z d Zy)serverzm Listens for commands on fin, runs them and writes the output on a channel based stream to fout. Nc~tj|_|r*|j|_||_|j|_n||_dx|_|_||_t|d|_ t|d|_ t|d|_ t||d|_ t|d|_|jjdddk(rA|jj!|_t#|jd|j d|_|jd d d k(r!t'|\}}t)|d |||_||_|j-dd|_d|_|j.s8t3j2t2j4t2j6|_yy)NdeoIrrllog-)repofpuismessage-outputschannelmsshutdown-on-interrupt)rgetcwdcwdbaseuirrrrepoui_prereposetupsrcdebugcerrcoutrEcincresultconfigcopy setuploggingcmsgrvr9client configbool_shutdown_on_interrupt_old_inthandlersignalSIGINTSIG_IGN)rrrrfinfout prereposetupsencnameencfns rrzserver.__init__sc??$ kkDGDI''DKDG&* *DI +%dD1 #D$/ #D$/ !#tT2&tT2 77>>, /4 7ggllnDG t < 99U- .* <226NGU(tWeDDI ')mm 2' # $**#)==#OD +rcz|js/tjtj|jyy)z9release and restore resources taken during server sessionN)rrrrrs rcleanupzserver.cleanups(** MM&--)=)= >+rcR|sy|jj|}|st|Srf)rrQEOFError)rrNr%s rrLz server._reads+{{%N rcztjd|jdd}|sy|j|S)z[read a string from the channel format: data length (uint32), data rTrSrr)r!rUrL)rrVs r_readstrzserver._readstrs6 udjjm4Q7zz&!!rcL|j}|r|jdSgS)z6read a list of NULL separated strings from the channel)rsplit)rrOs r _readlistzserver._readlists# MMO 775> !Ircddlm}|jr|j|S tjtj|j |j|tjtjtj S#tj$rt$rYnwxYw tjtjtj n7#tjtjtj wxYw|jjtdy)Nrdispatchs interrupted! rc) rrrrrrrSignalInterruptKeyboardInterruptrrr)rreqrs r_dispatchcommandzserver._dispatchcommands  & &$$S) ) 9 MM&--)=)= >$$S) MM&-- 8$$         MM&-- 8FMM&-- 8  Q()*s#>BB52C,4B55C,,4D c &ddlm}|j}|jj }|g}|j r||j _|jj|j}|j|_|j||x|j _|j j_ |j j|D]=}|jt|jdr*|j!dddd?|j#|dd||j |j|j$|j&|j(|j* } |j-|d z}|j.j1t3j4d t7|d |vr t9j:|j<yy#d |vr t9j:|j<wwxYw) zireads a list of terminated arguments, executes and writes the return code to the result channelrrr)rsnonttystrues commandserverNrs>is--cwd)rrrrrrrrr __class__appenddirstate_ui invalidateall resetstatehasattrr setconfigrequestrrrrrrr r!r"intoschdirr) rrargscopieduiuisrrrrrets r runcommandzserver.runcommand5s ~~77<<>j 99'DII [[**4;;7F"--FK JJv 4: :DIIL499--1 II # # % JB MMO488X. UIw8HI  J G  II HH II II II--   #'',s2C LL  v{{5#c(; <4" x4" s 9A G**&HcV|jjtjy)z1writes the current encoding to the result channelN)rr rrs r getencodingzserver.getencodinggs 8,,-rc|jjdd}|rK|jj|}|r |||dk7St j t d|z|dk7S)Nrcsunknown command %sr)rr[ capabilitiesrprrqr)rcmdhandlers rserveonezserver.serveonekspkk""$Sb) ''++C0G czkk!$9":S"@AAczr)s runcommands getencodingcddjt|jz}|dz }|dtjzz }|dz }|jr|d|jjzz }|dt j zz }ttdr |dz }|dtjd zz }|jj| |jr |jry #t$rYy wxYw) Nscapabilities: rmrYs encoding: smessage-encoding: %s spid: %dgetpgidspgid: %drr)rgsortedrrrr getpidrrrrr rr)rhellomsgs rservez server.servezs$tyy8I8I1J'KKEMH$5$555E 99 1DII4F4FF FHJ!222 2y !  H  bjjm3 3H ! --/--/   s!C99 DDr)r2r3r4r5rrrLrrrrrrrrr7rrrxrxsH ,P\?  "40#d. $.{KLrrxc|jdd}|syt|jdd}|dk(r|rtj||}n|dk(r!tj|j |}nt jt j|}|jdd}|jdd}tjtjj|}tj |tjj#||||}|h} |r6| j%|j&| j%|j(| D]} | j+d|y) zSet 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. rlrNs track-logrs max-log-filess max-log-size)maxfilesmaxsize)rsetrnrfileobjectloggerferrr abspath expandpath configint configbytesvfsmodr rpathdirname fileloggerbasenameaddrrr setlogger) rrrrlogpathtrackedloggerrrr targetuisus rrrs1ii f-G "-- l;?..?jj12''  GG  W %   I  dkk" dgg * L&)*rceZdZdZdZdZy) pipeservicec ||_||_yr)rrr)rrrrrAs rrzpipeservice.__init__s rcyrr7rs rinitzpipeservice.inits rc|j}|j5\}}t||j||} |j |j cdddS#|j wxYw#1swYyxYwr)rrprotectedfinoutrxrrr)rrrrrsvs rrunzpipeservice.runsq WW   ! [c4DIIsD1B xxz       s"A8A# A8#A55A88BN)r2r3r4rrrr7rrrrs  rrcXtjddtjyrJ)rsetpgidrandomseedr7rr_initworkerprocessrsJJq! KKMrc|jd}|jd}d} ||||||} |j|j |j! |j!y#tj$r1}|jt d|j zYd}~rd}~wt $rYt$rYwxYw#|jwxYw#|r |j} n t|d} | jtjtjxYw#t $rYywxYw#|j! |j!w#t $rYwwxYwxYw)Nrbwbs abort: %s r{)makefilerrrqrmessageBrokenPipeErrorrrrrr r strtolocal traceback format_excclose) rrrconncreatecmdserverrrrrinstrs r _serverequestrsB -- C == D B T4dM B  HHJ JJL   JJL){{ 7 HHQ~&5 6 6       JJL 77D"4.D 8&&y';';'=>?       JJL   s C&A6C&%D=6C 'B50C5 CCC C CCC##C&&AD::E = E E  E>E/.E>/ E;8E>:E;;E>c8eZdZdZdZdZdZdZdZdZ dZ y) unixservicehandlerzSet 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||_yr)rr)rrrs rrzunixservicehandler.__init__s rctj|||jtj|j j td|z|j jy)Nslistening at %s ) r bindunixsocketlistensocket SOMAXCONNrrstatusrr$)rsockaddresss r bindsocketzunixservicehandler.bindsocketsL D'* F$$% q-.89  rc.tj|yr)runlink)rrs r unlinksocketzunixservicehandler.unlinksockets 'rcy)z9True if server should shut down; checked per pollintervalFr7rs r shouldexitzunixservicehandler.shouldexitsrcy)z/Called when main process notices new connectionNr7rs r newconnectionz unixservicehandler.newconnectionsrc4t|j||||S)zhCreate new command server instance; called in the process that serves for the current connection)rxrr)rrrrrrs rrz"unixservicehandler.createcmdserver"sdggtS$ >>r) r2r3r4r5 pollintervalrrrrrrr7rrrrs, L >?rrcZeZdZdZddZdZdZdZdZdZ d Z d Z d Z d Z d ZdZy)unixforkingservicezG Listens on unix domain socket and forks server per connection NcZ||_||_|d|_ttdst j td|jst j td|xs t||_ d|_ d|_ d|_ d|_ t|_d|_|j#dd}|dkrt j tdt%j&|||_t*j,rt/j0yy) NsaddressAF_UNIXsunsupported platforms'no socket path specified with --addressrlsmax-repo-cachers(negative max-repo-cache size not allowed)rrrrrr rrqrr_servicehandler_sock_mainipc _workeripc_oldsigchldhandlerr _workerpids_socketunlinkedrr repoloader _repoloaderrisdarwinr gui)rrrrrArmaxlens rrzunixforkingservice.__init__-s J' vy)++a 789 9||++a JKL L&@* A:++a KLM M$//F;    LLN rcHtjtj|_tjtjtj}|\|_|_|jj|j|jttdr#tjtjtjtj|j}||_d|_|j$j'y)N unblocksignalF)r rr socketpair SOCK_DGRAMrr rrrrr r*rSIGCHLD_sigchldhandlerr!r#r%start)ros rrzunixforkingservice.initFs]]6>>2    fnnf.?.? @)*& t '' DLLA 8_ -  " "6>> 2 MM&..$*>*> ?"#$  rcv|js-|jj|jd|_yy)NT)r#rrrrs r _unlinksocketz unixforkingservice._unlinksocketUs0##  - -dll ;#'D $rcrtjtj|j|jj |j j |j j |j|jj|jdyrJ) rr-r!rrrr r2r%stop _reapworkersrs r_cleanupzunixforkingservice._cleanupZsv fnnd&=&=>      !rcp |j|jy#|jwxYwr) _mainloopr6rs rrzunixforkingservice.runds$  NN  MMODMMOs#5cd}|j}tj}|j|jtj |j |j|jtj |j |s"|jr|jd}|j|j}|s|rn(G|D]!\}}|j|j|#m|jy)NFT)timeout)r selectorsDefaultSelectorregisterr EVENT_READ_acceptnewconnectionr_handlemainipcrr2selectrr%fileobjr)rexitinghselectoreventskey_masks rr8zunixforkingservice._mainloopjs  ,,. JJ ,,d.G.G   MM9//1D1D q||~ ""$__Q^^_>Fc\|jd}|jj|y)z#Process messages sent from a workeriN)recvr%load)rr rErs rr@z!unixforkingservice._handlemainipcs$yy d#rcB|jtjyr)r5rWNOHANG)rrframes rr.z"unixforkingservice._sigchldhandlers "**%rc:|jrf tjd|\}}|dk(ry|j j dd||jj||jreyy#t$r|jj YywxYw)Nrcrrlsworker process exited (pid=%d) )r"rwaitpidChildProcessErrorclearrrrMdiscard)roptionsrQ_statuss rr5zunixforkingservice._reapworkerss !zz"g6 W ax GGKK &I3 O    $ $S )%   &&( sA44#BBcZtjtj|jt|j} t |j |j||j|jgtjy#tjwxYw)Nr) rr-r!rrrrrrr _reposetupgccollect)rrrDs rrNzunixforkingservice._runworkersq fnnd&=&=>     !!#/   JJLBJJLs 9BB*cP|jsyGfdd|j|_|j|_|jj |j }|y|jjdd|j tj||y)Nc"eZdZfdZxZS)8unixforkingservice._reposetup..unixcmdserverrepoct| |jj|jy#t j $r|jjddYywxYw)Nrls#failed to send repo root to master ) superr _cmdserveripcsendrootr rrrrM)rrunixcmdserverrepos rrz>unixforkingservice._reposetup..unixcmdserverrepo.closesW'46&&++DII6||GGKK$&Ms%7/A)(A))r2r3r4r __classcell__)rrks@rrkres   rrks repocachesrepo from cache: %s ) localrr rhr%rprjrrrMr copycache)rrrr cachedreporks @rr`zunixforkingservice._reposetupszz|   +!__%%))$))4     L":DIIFJ-rr)r2r3r4r5rrr2r6rr8r?r@r.r5rNr`r7rrrr(sD2 !(  :"B$ & * .rr)NN)"rarrr;rr r!ri18nrrrrrrr r r rutilsr r rr9rErorvrxrrrrrrr7rrrrs   '':)):T'T'p 9 NNb&*R($ F??Dz.z.r