}j-M@ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZmZm Z mZmZmZddlmZmZGddZGd d ZGd d Zd diZdZGddZddZGddZdZdZ GddZ!GddZ"y)N)_)getattr)encodingerror loggingutilpycompat repocacheutilvfs)cborutilprocutilc2eZdZdZdZedZdZdZy)channeledoutputz^ Write data to out in the following format: data length (unsigned int), data c ||_||_yN)outchannel)selfrrs ;usr/lib/python3.12/site-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.write3s@  v{{64<<TCdJK rcL|dvr t|t|j|SN)isattyfilenotellseek)AttributeErrorrrrattrs r __getattr__zchanneledoutput.__getattr__:' 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__Ns$S'2 " !rc tj|}|||d<|jj|j |y)Nsdata)r byteskwargsr;r!r<)rr&optss rr!zchanneledmessage.writeSs<##D)   DM -.rc.t|j|Sr)rr;r.s rr0zchanneledmessage.__getattr__Ystzz4((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__ls rc d|jzSrrrs rrzchanneledinput.nameqrrc|dkrR|j}|j||j}|}|r$|j||j}||z }|r$|S|j||jSNr) maxchunksize_readrrsizesbufs rreadzchanneledinput.readusp !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-rrGr.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_selectmessageencoderrvs]}}\+?@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._readstr s6 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]G}|jtj|j dr4|j#ddddI|j%|dd||j |j |j&|j(|j*|j, } |j/|d z}|j0j3t5j6d t9|d |vr t;j<|j>yy#d |vr t;j<|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 resetstater safehasattrr setconfigrequestrrrrrrr!r"r#intoschdirr) rrargscopieduiuisrrrrrets r runcommandzserver.runcommand6s ~~77<<>j 99'DII [[**4;;7F"--FK JJv 4: :DIIL499--1 II # # %B MMO##DHHh7 UIw8HI  G  II HH II II II--   #'',s2C LL  v{{5#c(; <4" x4" s A G44&HcV|jjtjy)z1writes the current encoding to the result channelN)rr!rrs r getencodingzserver.getencodinghs 8,,-rc|jjdd}|rK|jj|}|r |||dk7St j t d|z|dk7S)Nrcsunknown command %sr)rr[ capabilitiesrprrqr)rcmdhandlers rserveonezserver.serveonelspkk""$Sb) ''++C0G czkk!$9":S"@AAczr)s runcommands getencodingc$ddjt|jz}|dz }|dtjzz }|dz }|jr|d|jjzz }|dt j zz }tjtdr |dz }|dtjd zz }|jj| |jr |jry #t$rYy wxYw) Nscapabilities: rmrYs encoding: smessage-encoding: %s spid: %dgetpgidspgid: %drr)rgsortedrrrrgetpidr rrrrr!rr)rhellomsgs rservez server.serve{s$tyy8I8I1J'KKEMH$5$555E 99 1DII4F4FF FHJ!222   B *  H  bjjm3 3H ! --/--/   s !D 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 rrrs,ii 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.runsh 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.newconnection src4t|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 Ncn||_||_|d|_tjt dst jtd|jst jtd|xs t||_ d|_ d|_ d|_ d|_t|_d|_|j%dd}|dkrt jtdt'j(|||_t,j.rt1j2yy) NsaddressAF_UNIXsunsupported platforms'no socket path specified with --addressrlsmax-repo-cachers(negative max-repo-cache size not allowed)rrrrr rr rrqrr_servicehandler_sock_mainipc _workeripc_oldsigchldhandlerr _workerpids_socketunlinkedrr repoloader _repoloaderr isdarwinrgui)rrrrrArmaxlens rrzunixforkingservice.__init__.s J'  2++a 789 9||++a JKL L&@* A:++a KLM M$//F;    LLN rc\tjtj|_tjtjtj}|\|_|_|jj|j|jtjtdr#tjtjtjtj|j }||_d|_|j&j)y)N unblocksignalF)r rr socketpair SOCK_DGRAMrr rrrr rrr*rSIGCHLD_sigchldhandlerr!r#r%start)ros rrzunixforkingservice.initGs]]6>>2    fnnf.?.? @)*& t '' DLLA   Ho 6  " "6>> 2 MM&..$*>*> ?"#$  rcv|js-|jj|jd|_yy)NT)r#rrrrs r _unlinksocketz unixforkingservice._unlinksocketVs0##  - -dll ;#'D $rcrtjtj|j|jj |j j |j j |j|jj|jdyrJ) rr-r!rrrr r2r%stop _reapworkersrs r_cleanupzunixforkingservice._cleanup[sv fnnd&=&=>      !rcp |j|jy#|jwxYwr) _mainloopr6rs rrzunixforkingservice.runes$  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._mainloopks  ,,. 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"ri18nrr rrrrrr r r rutilsr rrr9rErorvrxrrrrrrr7rrrrs   '':)):T'T'p 9 NNb&*R($ F??Dz.z.r