>Y d dl Z d dlZd dlZddlmZ ddlmZmZmZm Z m
Z
m Z ddl m Z
ddlmZmZ e j" Ze j$ Ze j& ZdZdZd Zd
Ze
j0 Zd Z e
j4 e
j6 G d d
Z d Z d Z d Z d Z d Z!d Z"d Z# e
j4 e
j6 G d d Z$ddZ% G d d Z&y) N )_)encodingerrorpycompatutilwireprototypeswireprotov1server)r ) compression
stringutil s application/mercurial-0.1s application/mercurial-0.2s application/hg-errorc g }d} | j j d||fz }|n*|j t j | |dz
}Mdj | S )zjDecode a long value from multiple HTTP request headers.
Returns the value as a bytes, not a str.
r s %s-%d )headersgetappendr bytesurljoin)req headerprefixchunksivs =usr/lib/python3.12/site-packages/mercurial/wireprotoserver.pydecodevaluefromheadersr ) sk
F A
KKOOH a'8 8 9 9
h''*+ Q 88F r c p e Zd Zd Zed Zd Zd Zd Zd Z e
j d Z d Z
d Zd
Zy ) httpv1protocolhandlerc < || _ || _ || _ d | _ y N)_req_ui
_checkperm
_protocaps)selfr ui checkperms r __init__z httpv1protocolhandler.__init__< s # r c y)Ns http-v1 r$ s r namezhttpv1protocolhandler.nameB s r c
| j }i }|j }|D ]E }|dk( r3i }|j D ] }|dk7 s ||vs|| d ||< ||d< ;|| d ||< G |D cg c] }|| c}S c c}w )N * cmdr )_argssplitkeys)r$ args knownargsdatar1 kstarkeys r getargsz httpv1protocolhandler.getargsF s JJL zz|ADy$>>+Cf}D $-cN1$5S , "T
#A,q/Q "&&AQ&&&s 1 B c | j j j }t | j j j dd }|rK|j
t j | j j j | d |S t | j d }|j
t j |d |S )N
X-HgArgs-Postr T)keep_blank_valuess X-HgArg) r qsparams
asdictoflistsintr r updateurlreqparseqsbodyfhreadr )r$ r2 postlenargvalues r r/ zhttpv1protocolhandler._argsU s yy!!//1dii''++, r r r r
filechunkiterrB )r$ lengths r
getpayloadz httpv1protocolhandler.getpayloadj sb TYY&&'89: #dii''++,z8httpv1protocolhandler.addcapabilities.. s, "ALA Q / / 1 6 67s 79s compression=%s) r r% configint
configboolr supportedcompenginesr
SERVERROLEr )r$ repocaps compengines comptypess r addcapabilitiesz%httpv1protocolhandler.addcapabilities s H tww 0 0|j
d |_ t |j d< |j ddz y t |j fd
} t |||| y # |j $ r}|j D ] \ } }
|
|j | < |j
|j t! j" | |_ |j dt! j" | z Y d }~y d }~ww xY w) a Possibly process a wire protocol request.
If the current request is a wire protocol request, the request is
processed by this function.
``req`` is a ``parsedrequest`` instance.
``res`` is a ``wsgiresponse`` instance.
Returns a bool indicating if the request was serviced. If set, the caller
should stop processing the request, as a response has already been issued.
r commonr. Fi Content-Type 0
%s
s Not FoundTc | S r r) )rr r& rctxr s r z#handlewsgirequest.. s 9T3#=r N)hgwebr rk r< r| dispatchpath
statusmessagestatusHGTYPEr setbodybytesr r% _callhttp
ErrorResponsecoder forcebytestr) r r resr& hgwebcommonrk r{ protore r5 r s `` ` r handlewsgirequestr s5 - 99D
S\\!
,,v
C : ..s3
'- O$ l23 ! TWW=
E
B$S%- $ $ BIIDAq CKKN ..
FFJ++A.
j&=&=a&@@AA Bs 'B7 7EA>E Ec d|j v r|rt t j i fS t j
| }t
j | t j D ]U }|j j |v s i }| j dd|j z }|||d< t ||fc S d| j dd i}t t j d |fS )zDetermine the appropriate response type and compression settings.
Returns a tuple of (mediatype, compengine, engineopts).
s 0.2r` s %slevels levels zliblevels zlib)rJ HGTYPE2r _noopenginer
clientcompressionsupportr ri rj rc r+ rg r r rm )r% r prefer_uncompressed compformatsengineoptslevels r _httpresponsetyper s ##%% K 3 3 5r9 9 (@@G $99 &&
F &&(--< Y
V[[]0JK$%*DN ,,
"
bll9l; r |t
t |j@ y tC jD d | )
Nr r~ c 3 K |j j }dt | cxk rdk sJ J t j dt | | | D ] }| y w)Nr B)rc r+ lenstructpack)genr
engineoptsr+ chunks r genversion2z _callhttp..genversion2 sb &&(--3t9"s"""""kk$D **
EK s A%A'c | t k( rd_ nj | _ |j d< |j | |j | y y )Ns 200 Script output followsr )HTTP_OKr r r r
setbodygen)r contenttype bodybytesbodygenr r s r setresponsez _callhttp..setresponse& s\ 7?5CJ$2248CJ'2 O$ Y ' NN7 # r s: requested wire protocol command is not available over HTTP)r )r s %d
%sr Ts hgweb.protocol internal failure)NN)#r r r
rz commandavailabler HGERRTYPEr r&
permissionhashiddenaccessdispatch
isinstancebytesr r
bytesresponser4 streamreslegacyr streamresr r% r compressstreamr pushresr outputpusherrdrainooberrormessager ProgrammingError)rk r r r r{ r r accesshiddenrspr mediatyper r r s ` @r r r s , $ % % 6 6sE B
OO%..s3>>? ..tS9L
$
$T5#|
DC#u GVs3 C 55 6GVsxx8 C 77 8GVSWW5 C 11 2gg ): GGUC 3 3)
% 6: ##C4 c6:6CGY4 C // 0377CJJ//GVs3 C // 0377" GVs3 C 00 1GY#++>$$%GMMr c | j dt | z | j | | j y)z-Send a bytes response for protocol version 1.s %d
N)writer flush)rQ rI s r _sshv1respondbytesr d s, JJwU#$JJuJJLr c n | j }|j D ]
} || | j y r )r r r )rQ sourcer r s r _sshv1respondstreamr k s) JJE
e JJLr c |j d|z |j | j d | j y )Ns %s
-
)r r )rQ rR r s r _sshv1respondooberrorr r s1 JJzC JJLJJuJJLr c n e Zd ZdZd Zed Zd Zd Zd Z e
j d Z d Z
d Zd
Zy ) sshv1protocolhandlerz rC ) r$ r2 r4 r1 narglinearglr6 r5 vals r r8 z sshv1protocolhandler.getargs s zz|s4y!Aii((*3B/G ]]_FC$kk!$>"?#"EFFd{ s1v A "ii 0 0 23B 7G $]]_FC ))..Q0C #DI '
"T
iinnSV, S " "&&AQ&&&s / D>c | j S r )r# r* s r rJ z!sshv1protocolhandler.getprotocaps s r c #
K t | j d t | j j }|rD| j j | t | j j }|rCy y w)Nr )r r r> r r rC )r$ counts r rO z sshv1protocolhandler.getpayload sa 4::s+ DII&&()))..' ' **,-E s A>BBc # K d y wr r) r* s r rW z'sshv1protocolhandler.mayberedirectstdio s
s c r t j j dd j dd d }d|z S )Ns
SSH_CLIENTr rG r r s remote:ssh:)r environr r0 )r$ r] s r r] zsshv1protocolhandler.client s7 !!%%mS9??aHK &&r c | j t j k( r|j d |j d |S )Ns protocapsr_ )r+ r r r )r$ rk rl s r ro z$sshv1protocolhandler.addcapabilities s1 99,, , KK % H r c y r r) rq s r r& z sshv1protocolhandler.checkperm s r N)rs rt ru __doc__r' rv r+ r8 rJ rO rw rx rW r] ro r& r) r r r r y sV F $ $'* . '
r r c d}t | || }|j sC|dk( r |j d d }|sd}/t j j || } | s
t
|d ^t j |||| }
|j j j |j j j t |
t rt
||
ndt |
t j rt
||
j n2t |
t j" rt% ||
n
t |
t j& r
t% ||
nt |
t j( r&t
|d t
|d|
j* z nt |
t j, rt
||
j* nrt |
t j. r"t1 || j |
j2 n6t5 j6 d|
z |dk( ry t5 j6 d|z |j sBy y ) Ns protov1-servingr s shutdownr )r s %ds6 unhandled response type from wire protocol command: %ss unhandled ssh server state: %s) r is_setr r
rz r r r r% rQ r rR r r r r r4 r r r r r r r r r r r ) r% rk r rQ evr stater request availabler s r
_runsshserverr s $
E S$ /Eiik & &llnSb)G #)22CCI "4-#,,eW<C
GGLL GGLL #u%"4-C !=!=>"42C !9!9:#D#.C !?!?@#D#.C !7!78"4-"4 9C !7!78"41C !8!89%dBGGS[[A,,1367
k
! ((1E9
g iikkr c e Zd ZddZd Zd Zy) sshserverNc || _ || _ |j \ | _ | _ || _ |rQt
j || j dd | _ t
j ||j dd |_ y y )N oT)logdata e) r! _repo
protectfinoutr r
_accesshiddenr makeloggingfileobjectrR )r$ r% rk logfhr s r r' zsshserver.__init__ sy
" 0 0 2 4: ) 33tzz4DJ 00rwwdBG r c | j t j | j j | j
| j y r )
serveuntil threadingEventr!
restorefinoutr r r* s r
serve_foreverzsshserver.serve_forever s1 )* tyy$**5r c t | j | j | j | j || j
y)z%Serve until a threading.Event is set.N)r r! r r r r )r$ r s r r zsshserver.serveuntil" s- HHdjj$))TZZT=O=O
r )NF)rs rt ru r' r r r) r r r r s
6
r r )F)'rw r r i18nr r r r r r r
interfaces
interfaceutilutilsr r rS urlerrr@ r r r r r r implementerbaseprotocolhandlerr r| r r r r r r r r r r) r r r
s
.
==
%
& # " >==>g% g% ?g%\-CL!3HJNZ >==>D
D
?D
NJZ
r