` ddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZddlmZmZej:Zej<ZedZed Z d ee fzZ!d Z"dOd Z#dOd Z$dZ%dZ&ejNZ(dPdZ)e)ddddZ*e)ddddZ+e)dddZ,e)ddddZ-e)dd dd!Z.e)d"d#dd$Z/e)d%d&dd'Z0gd(Z1d)Z2e)d*dd+Z3e)d,d-dd.Z e)d/d0dd1Z4e)d2d3dd4Z5d5Z6e)d6d&dd7Z7e)d8dd9Z8e)d:dd;Z9e)dZ:e)d?d@ddAZ;e)dBdCddDZ<e)dEdFddGZ=e)dHdIddJZe)dKddLZ>e)dMd8ddNZ?y)QN)_)hex)getattr)bundle2 bundlecaches changegroup discoveryencodingerrorexchangehookpushkeypycompatrepoview requirements streamcloneutilwireprototypes)procutil stringutils/incompatible Mercurial client; bundle2 requireds9see https://www.mercurial-scm.org/wiki/IncompatibleClients%s (%s) c|jD])}|jds|ddjdcSddgS)aReturns a list of compression methods supported by the client. Returns a list of the compression methods supported by the client according to the protocol capabilities. If no such capability has been announced, fallback to the default of zlib and uncompressed. scomp=N,szlibsnone) getprotocaps startswithsplit)protocaps ?usr/lib/python3.12/site-packages/mercurial/wireprotov1server.pyclientcompressionsupportr!1sE!!# >>( #qr7==& &$ W c|jjdd}|r||dztjvr|dz }|j |S)zObtain the repo used for processing wire protocol commands. The intent of this function is to serve as a monkeypatch point for extensions that need commands to operate on different repo views under specialized circumstances. serversviews.hidden)uiconfigr filtertablefiltered)reporcommand accesshidden viewconfigs r getdispatchrepor-AsO 73J   "  #x';'; ;j == $$r"cnt||||}t|\}}|j|}|||g|S)N)r+)r-commandsgetargs)r)rr*r+funcspecargss r dispatchr4Vs@ 4l KD'"JD$ == D e #d ##r"ci}|D]}||vs||||<||=|r3tjjd|dj|fz|S)Ns,warning: %s ignored unexpected arguments %s r)rstderrwritejoin)cmdkeysothersoptsks r optionsr>_sc D  ;QiDGq  <DIIf%& '  Kr"c|j}tj|jv}|r|j dd|z}||S|j dd|z}||S|r|j dd}||S|j ddS)zWhether a bundle1 operation is allowed from the server. Priority is: 1. server.bundle1gd. (if generaldelta active) 2. server.bundle1. 3. server.bundle1gd (if generaldelta active) 4. server.bundle1 r$s bundle1gd.%ss bundle1.%ss bundle1gdsbundle1)r%requirementsmodGENERALDELTA_REQUIREMENTr configbool)r)actionr%gdvs r bundle1allowedrFms B  1 1T5F5F FB MM)_v%= > =H i!78A} MM)\ 2 =H ==J //r"pushc(tjjDchc]\}}|ddk(s|c}}dvrtjdzdt t stjdfd}|Scc}}w)aDecorator to declare a wire protocol command. ``name`` is the name of the wire protocol command being provided. ``args`` defines the named arguments accepted by the command. It is a space-delimited list of argument names. ``*`` denotes a special value that says to accept all named arguments. ``permission`` defines the permission type needed to run this command. Can be ``push`` or ``pull``. These roughly map to read-write and read-only, respectively. Default is to assume command requires ``push`` permissions because otherwise commands not declaring their permissions could modify a repository that is supposed to be read-only. sversionrrGpullsCinvalid wire protocol permission; got %s; expected "push" or "pull"r"s:arguments for version 1 commands must be declared as bytesctvrtjdztj|t<|S)Ns+%s command already registered for version 1)r3 transports permission)r/r ProgrammingErrorr commandentry)r1r3namerMrLs r registerz"wireprotocommand..registersL 8 ((>E (44 t z  r")r TRANSPORTSitemsr rN isinstancebytes)rPr3rMr=rErQrLs``` @r wireprotocommandrVs %//5577da1Z=A;M7J++$$ 13= >   | dE "$$ I    O;s BBsbatchscmds *rJ)rMc tj}g}|jdD]}|jdd\}}i} |jdD]*} | s| jd\} } || | || <,t|\} }t|j}|dvsJ|j ||rl|j}i}|D]9}|dk(r*i}| j D]}||vs| |||<||d<2| |||<;| ||g|Dcgc]}|| c}}n | ||}t|tjr|cSt|tjtfsJt|tjr |j}|jtj|tjdj|Scc}w)N; rr=rI*)runescapebatchargrr/rM checkpermr:rTooberror bytesresponserUdataappendescapebatchargr8)r)rcmdsr; unescapeargrespairopr3valsanrEr1r2permr:r`r=starkeyresults r batchros 11K C 4 ::dA&DD!Awwt}1'21~[^$"b\ d|&&))))  ::#?#?"GHHH fn:: ;[[F >0089K!N  ' ' # 77)@s G sbetweenspairsc6|jdDcgc]}tj|d}}g}|j|D])}|j tj |dz+tj dj|Scc}w)NrY- r")rr decodelistbetweenra encodelistr_r8)r)rpairsprbs r rtrts9>T9J K9JA^ & &q$ /9JE K A \\%  **1-56!  ' ' 44 LsB branchmapcD|j}g}|jD]X\}}tjt j |}t j|}|jd||fzZt jdj|S)Ns%s %srr) branchmaprSurlreqquoter fromlocalrrurar_r8)r)rr|headsbranchnodes branchname branchnodess r r|r|s I E"* \\("4"4V"<= $//6  X[ 99:+  ' ' 5(9 ::r"sbranchesnodesctj|}g}|j|D])}|jtj|dz+tj dj |S)Nrrr")rrsbranchesrarur_r8)r)rrrxrys r rrsa  % %e ,E A ]]5 ! **1-56"  ' ' 44r"sget_cached_bundle_inlinespathctj|jdr$tj|j|dd||jj t j }|jj ||}|jj|stjd|ztjj|}tjj|j|stjd|zd}||j|}tj|dS)z< Server command to send a clonebundle to the client spretransmit-inline-clone-bundleT)throwclonebundlepathsclonebundle %s does not exists'clonebundle %s is using an illegal pathc3K||5}tj|jd}tj|tj |D]}| dddy#1swYyxYww)N)osfstatfilenor uvarintencode filechunkiter)vfs bundle_pathflengthchunks r generatorz+get_cached_bundle_inline..generator#s_  XXahhj)!,F$$V, ,++A. /  s A=AA1( A=1A:6A=genprefer_uncompressed)rhashookr%rr8rBUNDLE_CACHE_DIRexistsr Abortrpathrealpathrr streamres)r)rr bundle_dirrclonebundles_dirrstreams r get_cached_bundle_inliner s  ||DGG?@ GG  .  |<<=JhhmmJ5O 88??? +kk:TABBww'' 3 77  O , 7 78H IkkDtKLLtxx 1F  # #D IIr"s clonebundlesr"ctj|}g}|jD]3}|jtjr#|j |5t jdj|S)zA legacy version of clonebundles_manifest This version filtered out new url scheme (like peer-bundle-cache://) to avoid confusion in older clients. rr) r get_manifest splitlinesrCLONEBUNDLESCHEMErarr_r8)r)rmanifest_contentsmodified_manifestlines r clonebundlesr.sn%11$7!,,. ??<99 :   &/  ' ' 3D(E FFr"sclonebundles_manifestr[cVtj|}tj|S)aServer command for returning info for available bundles to seed clones. Clients will parse this response and determine what bundle to fetch. Extensions may wrap this command to filter or dynamically emit data depending on the request. e.g. you could advertise URLs for the closest data center given the client's IP address. The only filter on the server side is filtering out inline clonebundles in case a client does not support them. Otherwise, older clients would retrieve and error out on those. )rrrr_)r)rr3rs r clonebundles_2r?s'%11$7  ' '(9 ::r")lookuprzpushkeyknown getbundles unbundlehashc tt}tjd|r|j dt j |r|jjddr|j dt j|}|tjhz s|j dn,|j ddjt|z|jjdd rQtjtj |d }|j d t"j%|z|j d djtj&z|jjdd rZ|j t(j*|jjddr|j t(j,|j/||S)a)return a list of capabilities for a repo This function exists to allow extensions to easily wrap capabilities computation - returns a lists: easy to alter - change done here will be propagated to both `capabilities` and `hello` command without any other action needed. changegroupsubsetr$spreferuncompressedsstream-preferredsstreams streamreqs=%srs experimentalsbundle2-advertise)rolesbundle2=s unbundle=%ssnarrowsnarrowservebrokenellipses)list wireprotocapsr/commandavailablerarallowservergenerationr%rBstreamed_requirementsr@REVLOGV1_REQUIREMENTr8sortedr encodecaps getrepocapsr}r~bundlepriorityr NARROWCAP ELLIPSESCAPaddcapabilities)r)rcapsrequiredformatscapsblobs r _capabilitiesr[sy  D  !5u= ()((. 77  i)> ? KK+ ,%;;DA/"F"F!GG KK " KK(499VO5L+MM N ww/+?@%%g&9&9$Y&OP K&,,x"889KK7+A+A!BBC ww/95 N,,- 77  o/K L KK22 3  t ,,r"s capabilitiesctt||}tjdjt |S)NrY)rrr_r8rr)rrs r capabilitiesrs, u %D  ' ' &,(? @@r"s changegroupsrootsctj|}tj|||j }t j ||ddtfdd}tj|S)N missingroots ancestorsof01servec&jdSNireadcgsr zchangegroup.. rwwu~r"r"r) rrsr outgoingrchangegroupmodmakechangegroupiterr)r)rrootsrrrrs @r r r sf  % %e ,E!! 5djjlH  ' 'hx HB %s +C  # # ,,r"rs bases headsctj|}tj|}tj|||}t j ||ddt fdd}tj|S)Nrrrc&jdSrrrsr rz#changegroupsubset..rr"r"r)rrsr rrrrr)r)rbasesrrrrs @r changegroupsubsetrsk  % %e ,E  % %e ,E!!$UNH  ' 'hx HB %s +C  # # ,,r" debugwireargss one two *c tdddg|}tj|j||fit j |S)Nrsthreesfour)r>rr_ debugwireargsr strkwargs)r)ronetwor;r<s r rrsM #h%8& AD  ' '3@x'9'9$'?@ r"cjd}|jjd}|sytj||}tj||d}|sy|j j j|D cgc]} j| c} dj|D cgc]} j| c} dt|} |D]} | jd} tjjj| } | r | | vr| | vrHd| vrK || d}|j|rjt!fd |Drt#fd |Drd | vr" || d }t!fd |Ds| d }|j$j'd|z |jj)|cSycc} wcc} w#t$rYwxYw#t$rY&wxYw#t*$r"|j$j'd|zYTwxYw)aReturn a file object for the first matching pullbundle. Pullbundles are specified in .hg/pullbundles.manifest similar to clonebundles. For each entry, the bundle specification is checked for compatibility: - Client features vs the BUNDLESPEC. - Revisions shared with the clients vs base revisions of the bundle. A bundle can be applied only if all its base revisions are known by the client. - At least one leaf of the bundle's DAG is missing on the client. - Every leaf of the bundle's DAG is part of node set the client wants. E.g. do not send a bundle of all changes if the client wants only one specific branch of many. cp|jdDchc]}tj|c}Scc}w)NrX)rbinascii unhexlify)shs r decodehexstringz(find_pullbundle..decodehexstrings-/0wwt}=}!""1%}===s3spullbundles.manifestNT) pullbundles) inclusives COMPRESSIONheadsc3DK|]}j|vywNrev.0rcl common_ancs r z"find_pullbundle..sE 266#;*,  c3nK|],}j|vxrj|v.ywrr)rrrr heads_ancs r rz"find_pullbundle..s;'Cs 9,NJ1NN's25sbasesc3DK|]}j|vywrrrs r rz"find_pullbundle..sILSrvvc{j0LrsURLssending pullbundle "%s" spullbundle "%s" not accessible )rtryreadrparseclonebundlesmanifestfilterclonebundleentries unfiltered changelog ancestorsrr!getr compengines _bundlenames TypeErrorissubsetallanyr%debugopenIOError)r)rr<clheadsrcommonrmanifestrer compformatsentrycompaltcomp bundle_heads bundle_basesrrrrs @@@r find_pullbundlers' >xx 78H   0 0x @C  / /ct LC    $ $B U;UcbffSkU;t LIf=fsrvvc{f=NJ*51Kyy(""//33D9 D +{0J  u  .uX? $$V,E EE' u  .uX? ILIIV}  2T9: 88==& &CJ Q<=      GGMM=D E  sB?G/G"* G'9 G7>H' G43G47 HH'H21H2rc*tdtjj|}|j D]\}}tj|}|dk(rtj |||<7|dk(rt |jd||<Z|dk(rt|jd||<}|dk(r|dk(rd||<t|||<|d k7std |zt|d sktj|jd sG|jd k(rtj t"St%j&t(t* t|j,j/}t|jdt}t|jdt} | j1|j2|j4j7ddrNd|j9vrrGETBUNDLE_ARGUMENTSr:rSrsrrsetboolKeyErrorrFr bundle2requestedrrPr^bundle2requiredr rbundle2requiredmainbundle2requiredhintrrdiscardnullidr%rBrrrrrrgetbundlechunksrrmessagerbundle20rraaddpart bundlepart getchunks)r)rr;r<r=rEkeytyperrrbundler infochunksprefercompressedexcbundlermanargsadvargss r getbundler;sr n88==? D 1 44Q7 h $//2DG  1774=)DG  !''$-(DG  "DyQq'Q  >HI I!$ $ (((-)@AzzZ'%..??++18KL L+!dnn**,-DHHXsu-.TXXi/0t{{# GG  y- 85#5#5#77%T5$OF%//**62 77  i)= >((5$/K6g.>kk>?MN  // ( &006 f 88$7>"  # # ,<(< ! ;;!((-)@AzzZ'%..s{{U/BCC ""477+ ,- 88  NNGSXX. /**>7GLM""$ !s.#C-LBLPAP /P5BP  Prcx|j}tjtj|dzS)Nrr)rrr_ru)r)rrs r rr?s. A  ' '(A(A!(Du(L MMr"shelloc^t||j}tjd|zS)afCalled as part of SSH handshake to obtain server info. Returns a list of lines describing interesting things about the server, in an RFC822-like format. Currently, the only one defined is ``capabilities``, which consists of a line of space separated tokens describing server abilities: capabilities: scapabilities: %s )rr`rr_rs r hellor>Es- e $ ) )D  ' '(=(D EEr"slistkeys namespacect|jtj|j }t j tj|Sr) rlistkeysr tolocalrSrr_ pushkeymod encodekeys)r)r namespaceds r rArAUsEt}}X--i89??ABA  ' ' (=(=a(@ AAr"rkeyc tj|}|j|}t|}d}tjd||fzS#t$r!}t j |}d}Yd}~?d}~wwxYw)Nrrs%d %s ) r rBlookupr Exceptionr forcebytestrrr_)r)rrmr=rjrxsuccessinsts r rIrI[su   S ! KKN F  ' ' gq\(A BB   # #D )s3A A9A44A9rsnodes *cdjd|jtj|D}tj|S)Nr"c30K|]}|xrdxsdyw)1rN)rrys r rzknown..js!$Pq dd$Ps)r8knownrrsr_)r)rrr;rEs r rRrRhsD $(JJ~/H/H/O$P A  ' ' **r"s protocapsscapsc|jtjk(rt|j d|_tj dS)NrYsOK)rPrSSHV1r#r _protocapsr_rs r protocapsrVps; zz^)))tzz$/0  ' ' ..r"rsnamespace key old newc@t|dk(r@tj||k7r( |jdt j |}nt j |}|j5}|jt j |t j |t j ||xsd}dddr|jnd}tjdt|fzS#t $rYwxYw#1swYOxYw)Nzutf-8Fr"s%d %s) lenr escapestrdecoder rBUnicodeDecodeErrormayberedirectstdiorgetvaluerr_int)r)rrErmoldnewoutputrxs r rrws 3x2~*..s3s:  JJw ""3'Cs#  ! ! #v LL  +  %  %      $#)V__ cF  ' ' SVV4D(D EE#"    $ #s&D4AD DDDs stream_outcRtjtj|S)zIf the server supports streaming clone, it advertises the "stream" capability with a value representing the version and flags of the repo it is serving. Client checks to see if it understands the format. )rstreamreslegacyrgeneratev1wireproto)r)rs r rrs  ) )+*I*I$*O PPr"sunbundlec  tj|}|j5} tj||dd} |j |j jddrfd}tjnd\fd}tjd \}|j jd ztj|tjd D]}j!|j#d tj$|j d}t'|t(j*rct-|d sW|j.dk(r)tj0t2|cdddSt5j6t8t:tj<|||d|j?} tj@| dr4tjB| jE|cdddStjF| |r|jInd|cdddS#|wxYw#t4jJt4j6t4jLf$r} tO| dds #t4j6$r} tPjRj!| jUtPjRjWtjFd |r|jIndcYd} ~ cYd} ~ cdddSd} ~ wt4jL$rLtjXtjZ| |r|jIndcYcYd} ~ cdddSwxYwt]j^|j } tO| ddD]} | ja|  #t4jb$r-} tO| dd} | d| jeddvr| jgd}|jid| jj| jl|jid| jld| jn|jid| jnd| jp|jid | jpd| jr|jid!| jrd| jt|jid"| jtdYd} ~ nd} ~ wwxYwnN#t4jJ$rz} | jgd#}| jv|jid$| jv| jxr+|jid%d&j{| jxYd} ~ nd} ~ wt4j6$rj} d'| j|fg}g}| j~|jd(| j~f| jat]jd)||Yd} ~ nKd} ~ wt4jL$r2} | jgd*d'tj| fgYd} ~ nd} ~ wwxYwtjB| jEcYd} ~ cdddSd} ~ wwxYw#1swYyxYw)+Nspreparing changescyrrQrQr"r rzunbundle..sdr"r$sstreamunbundlecD]}yrrQ)rwpayloads r cleanupzunbundle..cleanups")A ")r")NNcZrjrtjyyr)closerunlink)fptempnamesr rjzunbundle..cleanups#HHJ#IIh/$r"s hg-unbundle-)prefixs"redirecting incoming bundle to %s swb+rrGrrrr/rr"duringunbundle2F_bundle2salvagedoutputrQ _replycapsrserrors error:pushkeys in-reply-tor?) mandatoryrGsnewsoldsretserror:unsupportedcontentsparttypesparamsrr r!serror:pushraced)Drrsr]r check_heads getpayloadr%rBr chunkbufferrmkstempr rfdopensysstrr7seek readbundlerTr cg1unpackerrFrPr^r'r rr(r)unbundleclient safehasattrrdr1pushresr^BundleValueError PushRacedrrr6formatflushpusherrbytestrrr.r/ PushkeyFailedrnewpartaddparampartidrErmrar`retparttypeparamsr8r-rrar0rrK)r)rr their_headsrbrjfdrwrrxr7r8out remotecapsparterrpartr9r:rnriros @@@r rrs ++E2K  ! ! #vB K  {4H I"G8 **,77%%i1BC! ))'2B$.LB0 $,#3#3?#KLBGGMM>I2xv'>?B$ %GGAJ))$''2t<33(w7zzZ/ .66G" y $ #X +++2E%%#{Helln##Ay1*99akkmL  y $ #n&--Fv(  y $ #x && U__EE K3 159{{  OO))#**,7OO))+)116??,CU $ #Z)11 ((--3)] $ #Z &&tww/Gs$?@  "11g6G6G6IJ JG $ #|E K} $ #s&[I4E I(I4 A?I(I40'I(I4( I11I443['[6J77N2 A2M<N2=[[[AN2!["[#[1N22A[9O::T;D#T61U6T;;U>[Z A0W[Z A Y>[Z (Z[Z  &[1[2[[[[)F)NrG)@rri18nrnoderrrrrr rr r r r rrrCrrr@rrrutilsrrurlerrr}r(r)r'r!r-r4r>rF commanddictr/rVrortr|rrrrrrrrrrr;rr>rArIrRrVrrrQr"r rs8 $  JK@ #68K"LL  %*$ 0< &> % % ',`(I':*8;*8Z*h7;5<5,73;4;+xG<5=5-w7KJLJB/37; G< G *DWE;F;" &-V/g6A7A .(w?-@-&7K-L-"LWEFCL,9K:K\(w/N0N (w/ F0 F+|@BAB )V8 C9 C(J7;+<+,G</=/ *67KFLF8-G4Q5Q+xG<FK=FKr"