File: //usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyc
�
1�3\c @@ s d d l m Z d d l Z d d l m Z d d l m Z m Z d d l m Z m
Z m Z m
Z
m Z m Z m Z m Z m Z m Z m Z d d l m Z m Z e j Z e j Z e d � Z e d � Z d
e e f Z d � Z d � Z d
� Z d � Z! d � Z" e j# � Z$ d d d � Z& e& d d d d �d � � Z' e& d d d d �d � � Z( e& d d d �d � � Z) e& d d d d �d � � Z* e& d d d d �d! � � Z+ d" d d# d$ d% d& g Z, d' � Z- e& d( d d �d) � � Z. e& d* d+ d d �d, � � Z
e& d- d. d d �d/ � � Z/ e& d0 d1 d d �d2 � � Z0 d3 � Z1 e& d% d4 d d �d5 � � Z2 e& d6 d d �d7 � � Z3 e& d8 d d �d9 � � Z4 e& d: d; d d �d< � � Z5 e& d" d= d d �d> � � Z6 e& d$ d? d d �d@ � � Z7 e& dA dB d d �dC � � Z8 e& d# dD d d �dE � � Z e& dF d d �dG � � Z9 e& dH d6 d d �dI � � Z: d S(J i ( t absolute_importNi ( t _( t hext nullid( t bundle2t changegroupt discoveryt encodingt errort exchanget pushkeyt pycompatt streamclonet utilt wireprototypes( t procutilt
stringutils/ incompatible Mercurial client; bundle2 requireds9 see https://www.mercurial-scm.org/wiki/IncompatibleClients %s
(%s)
c C@ sA x4 | j � D]& } | j d � r
| d j d � Sq
Wd d g S( s Returns 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.
s comp=i t ,t zlibt none( t getprotocapst
startswitht split( t protot cap( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt clientcompressionsupport- s c C@ s
| j d � S( s� Obtain 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.
t served( t filtered( t repoR t command( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt getdispatchrepo; s c C@ sA t | | | � } t | \ } } | j | � } | | | | � S( N( R t commandst getargs( R R R t funct spect args( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt dispatchD s c C@ sk i } x2 | D]* } | | k r
| | | | <| | =q
q
W| rg t j j d | d j | � f � n | S( Ns, warning: %s ignored unexpected arguments %s
R ( R t stderrt writet join( t cmdt keyst otherst optst k( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt optionsL s
c C@ s� | j } d | j k } | rG | j d d | � } | d k rG | Sn | j d d | � } | d k rm | S| r� | j d d � } | d k r� | Sn | j d d � S( s� Whether a bundle1 operation is allowed from the server.
Priority is:
1. server.bundle1gd.<action> (if generaldelta active)
2. server.bundle1.<action>
3. server.bundle1gd (if generaldelta active)
4. server.bundle1
t generaldeltat servers bundle1gd.%ss
bundle1.%st bundle1gdt bundle1N( t uit requirementst
configboolt None( R t actionR2 t gdt v( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt bundle1allowedW s
t pushc @ s� d � t j j � D� � � d k r8 � j t j � n � d k rZ t j d � � � n � d
k ro d � n t � t � s� t j d � � n � � � � f d � } | S( s� Decorator 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.
c S@ s, h | ]" \ } } | d d k r | � q S( t versioni ( ( t .0R, R8 ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pys <setcomp>� s t batchR: t pullsC invalid wire protocol permission; got %s; expected "push" or "pull"t s: arguments for version 1 commands must be declared as bytesc @ sK � t k r" t j d � � � n t j | d � d � d � �t � <| S( Ns+ %s command already registered for version 1R# t
transportst
permission( R R t ProgrammingErrorR t commandentry( R! ( R# t nameRA R@ ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt register� s
( R: R> N(
R t
TRANSPORTSt itemst addt SSHV2R RB R5 t
isinstancet bytes( RD R# RA RE ( ( R# RD RA R@ sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt wireprotocommandv s
R= s cmds *RA R> c C@ s# t j } | j d � } g } x�| j d � D]�} | j d d � \ } } i } xK | j d � D]: }
|
rb |
j d � \ } } | | � | | | � <qb qb Wt | \ }
} t | j } | d
k s� t � | j | � | r�| j � } i } xo | D]g } | d k rTi } x1 | j � D]# } | | k r | | | | <q q W| | d <q� | | | | <q� W|
| | g | D] } | | ^ qv� } n |
| | � } t | t j
� r�| St | t j t f � s�t � t | t j � r�| j
} n | j t j | � � q. Wt j d j | � � S( NR t ;t i R t =R: R> t *( R: R> ( R t unescapebatchargR R R RA t AssertionErrort checkpermR) RJ t ooberrort
bytesresponseRK t datat appendt escapebatchargR' ( R R t cmdsR* t unescapeargt rest pairt opR# t valst at nR8 R! R" t permR) RV R, t start keyt result( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR= � sD
, t betweent pairsc C@ s~ g | j d � D] } t j | d � ^ q } g } x1 | j | � D] } | j t j | � d � qD Wt j d j | � � S( NRN t -s
R? ( R R t
decodelistRe RW t
encodelistRU R' ( R R Rf t pt rt b( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyRe � s
. t branchmapc C@ s� | j � } g } xX | j � D]J \ } } t j t j | � � } t j | � } | j d | | f � q Wt j d j
| � � S( Ns %s %ss
( Rm t iteritemst urlreqt quoteR t fromlocalR Ri RW RU R' ( R R Rm t headst brancht nodest
branchnamet branchnodes( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyRm � s t branchesRt c C@ s_ t j | � } g } x1 | j | � D] } | j t j | � d � q% Wt j d j | � � S( Ns
R? ( R Rh Rw RW Ri RU R' ( R R Rt Rk Rl ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyRw � s
t clonebundlesR? c C@ s t j | j j d � � S( sZ Server 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.
s clonebundles.manifest( R RU t vfst tryread( R R ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyRx � s
t lookupR
t knownt getbundlet unbundlehashc C@ s t t � } t j d | � r. | j d � n t j | � r� | j j d d � rb | j d � n | j | j
@} | d h s� | j d � q� | j d d j t | � � � n | j j d d
� rt
j t
j | d d �� } | j d t j | � � n | j d
d j t
j � � | j j d d � ro| j t j � | j j d d � ro| j t j � qon | j | | � S( s) 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.
t changegroupsubsetR/ t preferuncompresseds stream-preferredt revlogv1t streams
streamreqs=%sR t experimentals bundle2-advertiset roles bundle2=s unbundle=%st narrowt narrowservebrokenellipses( t listt
wireprotocapsR t commandavailableRW R t allowservergenerationR2 R4 R3 t supportedformatsR' t sortedR t
encodecapst getrepocapsRo Rp t bundlepriorityR t NARROWCAPt ELLIPSESCAPt addcapabilities( R R t capst requiredformatst capsblob( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt
_capabilities s&
# t capabilitiesc C@ s+ t | | � } t j d j t | � � � S( NRN ( R� R RU R' R� ( R R R� ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR� * s R t rootsc @ sp t j | � } t j | d | d | j � �} t j | | d d � � t � f d � d � } t j d | � S( Nt missingrootst missingheadst 01t servec @ s
� j d � S( Ni � ( t read( ( t cg( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt <lambda>5 R? R? t gen( R Rh R t outgoingRr t changegroupmodt makechangegroupt itert streamres( R R R� Rt R� R� ( ( R� sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR / s R s bases headsc @ sy t j | � } t j | � } t j | d | d | �} t j | | d d � � t � f d � d � } t j d | � S( NR� R� R� R� c @ s
� j d � S( Ni � ( R� ( ( R� ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR� @ R? R? R� ( R Rh R R� R� R� R� R� ( R R t basesRr R� R� ( ( R� sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR 8 s t
debugwireargss one two *c C@ s= t d d d g | � } t j | j | | t j | � � � S( NR� t threet four( R- R RU R� R t strkwargs( R R t onet twoR* R+ ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR� C s c @ s d � } | j j d � } | s% d
St j | | � } t j | | � } | sS d
S| j � � j g | D] } � j | � ^ qi d t �� � j g | D] } � j | � ^ q� d t �� t
| � }
x�| D]�} | j d � } t j
j j | � }
| r | |
k r |
|
k r q� n d | k r�y | | d � } Wn t k
rUq� n X| j | � rkq� n t � � f d � | D� � r�q� n t � � � f d � | D� � r�q� q�n d | k ry | | d � } Wn t k
r�q� n Xt � � f d � | D� � sq� qn | d
} | j j d | � y | j j | � SWq� t k
rv| j j d | � q� q� Xq� Wd
S( s� Return 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.
c S@ s/ t g | j d � D] } | j d � ^ q � S( NRM R ( t setR t decode( t st h( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt decodehexstringZ s s pullbundles.manifestt inclusivet COMPRESSIONRr c 3@ s$ | ] } � j | � � k Vq d S( N( t rev( R< R� ( t clt
common_anc( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pys <genexpr>w s c 3@ s9 | ]/ } � j | � � k o0 � j | � � k Vq d S( N( R� ( R< R� ( R� R� t heads_anc( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pys <genexpr>y s R� c 3@ s$ | ] } � j | � � k Vq d S( N( R� ( R< R� ( R� R� ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pys <genexpr>� s t URLs sending pullbundle "%s"
s pullbundle "%s" not accessible
N( Ry Rz R5 R t parseclonebundlesmanifestt filterclonebundleentriest changelogt ancestorsR� t TrueR t getR
t compenginest _bundlenamest TypeErrort issubsett allt anyR2 t debugt opent IOError( R R R+ t clheadsRr t commonR� t manifestR[ R� t compformatst entryt compt altcompt bundle_headst bundle_basest path( ( R� R� R� sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt find_pullbundleK sX 11
RP c C@ s� t d t j j � | � } x� | j � D]� \ } } t j | } | d k rc t j | � | | <q( | d k r� t | j d � � | | <q( | d k r� t | j d � � | | <q( | d k r� | d k r� t | | <q
t
| � | | <q( | d k r( t d | � � q( q( Wt | d
� slt
j | j d � � sl| j d k rQt j t � St j t d
t �� qln t } yct | j j � � } t | j d t � � � } t | j d t � � � }
|
j t � | j j d d � r7d | j � k r7t | | | | | |
� } | r7t j d t! j" | � d t � Sn | j j d d � r�| j d t � } | r�|
r�| | k r�t j t# d � d
t# d � �� q�n t
j$ | d t% j& | � � \ }
} |
j d t � } Wn� t j k
r�} t
j | j d � � s1| j d k r+t j t% j' | � d � S� n t( j) | j � } d t% j' | � f g } g } | j* d k r�| j, d
| j* f � n | j- t( j. d | | � � | j/ � } t } n Xt j d | d | � S( NR} Rt t csvR t scsvt booleant 0t plains unknown getbundle option type %sR> t
bundlecapss http-v1t hintRr R� R/ t
pullbundles partial-pullR� t prefer_uncompressedt disablefullbundleR� s% server has pull-based clones disableds/ remove --pull if specified or upgrade MercurialR� t prefercompresseds
t messages error:abort(0 R- R t GETBUNDLE_ARGUMENTSR) Rn Rh R� R R� t Falset boolt KeyErrorR9 R t bundle2requestedR� RD RT t bundle2requiredR t Abortt bundle2requiredmaint bundle2requiredhintR� R� Rr t discardR R2 R4 R R� R� R
t
filechunkiterR t getbundlechunksR R� t bytestrR t bundle20R� R5 RW t addpartt
bundlepartt getchunks( R R R* R+ R, R8 t keytypeR� R� Rr R� t bundleR t infot chunkst exct bundlert manargst advargs( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR} � sx
Rr c C@ s& | j � } t j t j | � d � S( Ns
( Rr R RU Ri ( R R R� ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyRr � s t helloc C@ s# t | | � j } t j d | � S( sf Called 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: <token0> <token1> <token2>
s capabilities: %s
( R� RV R RU ( R R R� ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR� � s t listkeyst namespacec C@ s: t | j t j | � � j � � } t j t j | � � S( N( R� R� R t tolocalRG R RU t
pushkeymodt
encodekeys( R R R� t d( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR� � s $Rc c C@ sv y4 t j | � } | j | � } t | � } d } Wn( t k
r^ } t j | � } d } n Xt j d | | f � S( Ni i s %d %s
( R R� R{ R t ExceptionR t forcebytestrR RU ( R R Rc R, R` Rk t successt inst( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR{ � s
s nodes *c C@ s8 d j d � | j t j | � � D� � } t j | � S( NR? c s@ s! | ] } | r d p d Vq d S( t 1R� N( ( R< Rl ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pys <genexpr> s ( R' R| R Rh RU ( R R Rt R* R8 ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR| � s t protocapsR� c C@ s: | j t j k r- t | j d � � | _ n t j d � S( NRN t OK( RD R t SSHV1R� R t
_protocapsRU ( R R R� ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR s s namespace key old newc C@ s� t | � d k r^ t j | � | k r^ y | j d � t j | � } Wqm t k
rZ qm Xn t j | � } | j � �@ } | j t j | � t j | � t j | � | � p� t } Wd QX| r� | j
� n d } t j d t
| � | f � S( Ni s utf-8R? s %d
%s( t lenR t escapestrR� R R� t UnicodeDecodeErrort mayberedirectstdioR
R� t getvalueR RU t int( R R R� Rc t oldt newt outputRk ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR
s '
!t
stream_outc C@ s t j t j | � � S( s� If 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.
( R t streamreslegacyR t generatev1wireproto( R R ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR� s t unbundlec @ s� t j | � } | j � ��} y�t j | | d � d � } z�| j � � | j j d d � r� � f d � } t j � � � n� d) \ � � � � f d � } t j d d � \ } � | j j
d � � t j | t j d
� � � d } x � D] } � j | � q� W� j d � t j | j � d � } t | t j � r�t | d � r�| j d
k rnt j t � St j t d t �� n t j | | | d | j � � } t j! | d � r�t j" d | j# � � St j$ | | r�| j% � n d � SWd | � XWn�t j& t j t j' f k
r�}
t( |
d t) � s�y � Wq�t j k
r�t* j+ j d |
� |
j, d k r�t* j+ j d |
j, � n t* j+ j- � t j$ d | r�| j% � n d � St j' k
r�t j. t j/ |
� | r�| j% � n d � SXn t0 j1 | j � } x' t( |
d d* � D] } | j2 | � q WyYy � WnKt j3 k
r�}
t( |
d d � }
|
d k r�d |
j4 d d+ � k r�� n | j5 d � } | j6 d |
j7 � |
j8 d k r�| j6 d |
j8 d t) �n |
j9 d k r| j6 d |
j9 d t) �n |
j: d k r3| j6 d |
j: d t) �n |
j; d k r^| j6 d |
j; d t) �n |
j< d k r�| j6 d! |
j< d t) �q�n XWn t j& k
r}
| j5 d" � } |
j= d k r�| j6 d# |
j= � n |
j> r�| j6 d$ d% j? |
j> � � q�n� t j k
ry}
d& t@ jA |
� f g } g } |
j, d k rZ| jB d |
j, f � n | j2 t0 jC d' | | � � n8 t j' k
r�}
| j5 d( d& t@ jA |
� f g � n Xt j" d | j# � � SXWd QXd S(, Ns preparing changesc S@ s d S( N( R5 ( ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR� / R? R/ t streamunbundlec @ s x � D] } q Wd S( N( ( Rj ( t payload( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt cleanup3 s
c @ s- � r � j � n � r) t j � � n d S( N( t closet ost unlink( ( t fpt tempname( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR < s
t prefixs hg-unbundle-s" redirecting incoming bundle to %s
s wb+i R: s http-v1R� R� R� R� R? t duringunbundle2s
abort: %s
s (%s)
t _bundle2salvagedoutputt
_replycapsR
R s
error:pushkeys in-reply-toR� t mandatoryRc R R t rets error:unsupportedcontentt parttypet paramss R� s error:aborts error:pushraced( NN( ( (D R Rh R R t check_headst
getpayloadR2 R4 R
t chunkbufferR5 R t mkstempR� R t fdopent sysstrR& t seekt
readbundleRJ R� t cg1unpackerR9 RD RT R� R R� R� R� R t clientt safehasattrR R� t pushresR t BundleValueErrort PushRacedt getattrR� R R% R� t flusht pusherrR� R R� R� t
PushkeyFailedR� t newpartt addparamt partidR� Rc R R R" R# R$ R' R R RW R� ( R R Rr t their_headsR R t fdRk Rj R� R� R� t outt
remotecapst partt errpartR� R� ( ( R R R sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyR ( s�
!
$ " (; t
__future__R R t i18nR t nodeR R R? R R R� R R R R R
R� R R R
R t utilsR R t urlerrRo R� R� R� R R R$ R- R9 t commanddictR R5 RL R= Re Rm Rw Rx R� R� R� R R� R� R} Rr R� R� R{ R| R R� R ( ( ( sA /usr/lib64/python2.7/site-packages/mercurial/wireprotov1server.pyt <module> sT L / .
*
B L