GPdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z ddlmZmZmZmZddlmZddlmZddlmZdd lmZmZmZdd lmZmZm Z m!Z!ejDe#Z$d Z%d Z&d Z'dZ(d0de e)de*fdZ+de*fdZ,defdZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5Gdde6Z7dZ8Gdd e6Z9Gd!d"e6Z:Gd#d$e6Z;Gd%d&e6ZGd*d+eZ?Gd,d-eZ@ejfd.ZBe#d/k(reByy)1zDHTTP server for dulwich that implements the git smart HTTP protocol.)BytesION)ListTupleOptional)WSGIRequestHandler ServerHandler WSGIServer make_server)parse_qs) log_utils)ReceivableProtocol)BaseRepoNotGitRepositoryRepo) DictBackendDEFAULT_HANDLERSgenerate_info_refsgenerate_objects_info_packsz200 OKz 404 Not Foundz 403 Forbiddenz500 Internal Server Error timestampreturncgd}gd}|tj}tj|dd\}}}}}}} d|| |||||||fzS)N)MonTueWedThuFriSatSun) NJanFebMarAprMayJunJulAugSepOctNovDecz#%s, %02d %3s %4d %02d:%02d:%02d GMD)timegmtime) rweekdaysmonthsyearmonthdayhhmmsswds lib/third_party/dulwich/web.pydate_time_stringr8FswAHFIIK '+{{9'=bq'A$D%b"b" 0  u     4 c`d|jd|jjdzS)aExtract the URL prefix from a regex match. Args: mat: A regex match object. Returns: The URL prefix, defined as the text before the match in the original string. Normalized to start with one leading slash and end with zero. /N)stringstartstrip)mats r7 url_prefixr@ks+ Mciik*005 55r9c6|jt|S)z>Get a Repo instance for the given backend and URL regex match.)open_repositoryr@)backendr?s r7get_reporDws  " ":c? 33r9c#*K||jdy |jt| |jd}|sn| |j y#t$r|j dY/wxYw#|j wxYww)a!Send a file-like object to the request output. Args: req: The HTTPGitRequest object to send output to. f: An open file-like object to send; will be closed. content_type: The MIME type for the file. Returns: Iterator over the contents of the file, as chunks. NzFile not foundi(zError reading file) not_foundrespondHTTP_OKreadIOErrorerrorclose)reqf content_typedatas r7 send_filerQ|s ymm,--  G\*66%=DJ    .ii,--.  s:B0A A> BA;8A>:A;;A>>BBcV|jdtjjS)Nr;)replaceospathsep)urls r7 _url_to_pathrXs ;;sBGGKK ((r9c|jt|j}tj d|t |t ||j|dS)NzSending plain text file %s text/plain)nocacherXgrouploggerinforQrDget_named_filerMrCr?rUs r7 get_text_filerasKKKM  $D KK,d3 S(7C0??E| TTr9c#K|jd|jdzjd}tjd|t ||j }|j |s|jdy ||j}|j|jtd|y#t$r|jdYywxYww)NasciizSending loose object %szObject not foundzError reading objectzapplication/x-git-loose-object)r\encoder]r^rD object_storecontains_looserFas_legacy_objectrJrK cache_foreverrGrH)rMrCr?shargrPs r7get_loose_objectrls 99Q<#))A, & . .w 7C KK)3/GS)66L  & &s +mm.//C 113KK9: J ii.//s*BC'C+C'C$!C'#C$$C'c|jt|j}tj d|t |t ||j|dS)NSending pack file %sz application/x-git-packed-objectsrjrXr\r]r^rQrDr_r`s r7 get_pack_filerpsU  $D KK&-  #--d3* r9c|jt|j}tj d|t |t ||j|dS)Nrnz$application/x-git-packed-objects-tocror`s r7 get_idx_filerrsU  $D KK&-  #--d3. r9c#Kt|jd}|jddgd} t||}|r|js|jj|jdd}||jdy|j|jtd|z}ttj |} ||t#|g| |d} | j$j'd |jdzd z| j$j'd| j)y|j|jtd t*j-d t/|D]} | y#t$r&}|j t |Yd}~yd}~wwxYww) N QUERY_STRINGservicerreUnsupported servicezapplication/x-%s-advertisementT) stateless_rpcadvertise_refss # service= rZzEmulating dumb info/refs)r environgetrDrrFstrdumbhandlersrf forbiddenr[rGrHr rrIr@protowrite_pkt_linehandler]r^r) rMrCr?paramsrurepoe handler_clswriterhandlertexts r7 get_info_refsrsy ckk.1 2FjjTF+A.G%sxxll&&w~~g'>E  -- 56 6   G%E%OP"79>>59  _      $$]W^^G5L%Lu%TU $$T*  G\* ./&t,DJ-7 mmCF##s./G  FEG  G!G=G GG c|j|jtdtj dt t ||S)NrZzEmulating dumb info/packs)r[rGrHr]r^rrD)rMrCr?s r7get_info_packsrs8KKMKK& KK+, &x'= >>r9ceZdZdZdZddZy)_LengthLimitedFilea Wrapper class to limit the length of reads from a file-like object. This is used to ensure EOF is read from the wsgi.input object once Content-Length bytes are read. This behavior is required by the WSGI spec but not implemented in wsgiref as of 2.5. c ||_||_yN)_input _bytes_avail)selfinput max_bytess r7__init__z_LengthLimitedFile.__init__s %r9c|jdkry|dk(s||jkDr |j}|xj|zc_|jj|S)Nrr9)rrrI)rsizes r7rIz_LengthLimitedFile.readsX    ! 2: 1 11$$D T!{{%%r9N)r)__name__ __module__ __qualname____doc__rrIr9r7rrs&&r9rc#ZK|jjd}tjd||jj |j dd}||jdy t|||j|jtd|z}t|j dj"|}||t%|g||}|j'y#t$r&}|jt|Yd}~yd}~wwxYww)Nr;zHandling service request for %srervzapplication/x-%s-result wsgi.input)rw)r\lstripr]r^r~r{rfrrDrrFr|r[rGrHr rzrIr@r) rMrCr?rurrrrrs r7handle_service_requestr siik  %G KK17;,,""7>>'#:DAKmm122#KKM KK!:W!D EE s{{<8==u EE'JsO#4e3OG NN mmCF##s1A7D+: C9A3D+9 D(D#D+#D((D+c eZdZdZddefdZdZeddfdede ede e e eeffd Z d ed e fd Zd ed e fd Zd ed e fdZddZddZy)HTTPGitRequestzzClass encapsulating the state of a single git HTTP request. :ivar environ: the WSGI environment for the request. Nr}cX||_||_||_||_g|_g|_yr)rzr}r~_start_response_cache_headers_headers)rrzstart_responser}r~s r7rzHTTPGitRequest.__init__$s.    -  r9c>|jj||fy)zAdd a header to the response.N)rappend)rnamevalues r7 add_headerzHTTPGitRequest.add_header,s dE]+r9statusrOheadersc|r|jj||r|jjd|f|jj|j|j ||jS)z9Begin a response with the given status and other headers.z Content-Type)rextendrrr)rrrOrs r7rGzHTTPGitRequest.respond0sb  MM  )  MM .,!? @ T001##FDMM::r9messagercg|_tjd||jtd|j dS)z;Begin a HTTP 404 response and return the text of a message.z Not found: %srZre)rr]r^rGHTTP_NOT_FOUNDrfrrs r7rFzHTTPGitRequest.not_found?5  OW- ^\2~~g&&r9cg|_tjd||jtd|j dS)z;Begin a HTTP 403 response and return the text of a message.z Forbidden: %srZre)rr]r^rGHTTP_FORBIDDENrfrs r7rzHTTPGitRequest.forbiddenFrr9cg|_tjd||jtd|j dS)z;Begin a HTTP 500 response and return the text of a message.z Error: %srZre)rr]rKrG HTTP_ERRORrfrs r7rKzHTTPGitRequest.errorMs5  ['* Z.~~g&&r9cgd|_y)z2Set the response to never be cached by the client.))ExpireszFri, 01 Jan 1980 00:00:00 GMT)Pragmazno-cache) Cache-Controlz$no-cache, max-age=0, must-revalidateN)r)rs r7r[zHTTPGitRequest.nocacheTs  r9crtj}dt|fdt|dzfdg|_y)z4Set the response to be cached forever by the client.Dateri3)rzpublic, max-age=31536000N)r,r8r)rnows r7rjzHTTPGitRequest.cache_forever\s:iik %c* + (x8 9 9 r9)FN)rN)rrrrboolrrrHr|rrrrGbytesrFrrKr[rjrr9r7rrs d, &*37 ; ;sm ;$uS#X/0 ;'''''''S'U'  r9rceZdZdZdej dfedej dfedej dfedej dfedej dfedej dfe dej d fe dej d fe d ej d fe d ej d fe i Z ddefdZdZy)HTTPGitApplicationz}Class encapsulating the state of a git WSGI application. :ivar backend: the Backend object backing this application GETz/HEAD$z /info/refs$z/objects/info/alternates$z/objects/info/http-alternates$z/objects/info/packs$z&/objects/([0-9a-f]{2})/([0-9a-f]{38})$z(/objects/pack/pack-([0-9a-f]{40})\.pack$z'/objects/pack/pack-([0-9a-f]{40})\.idx$POSTz/git-upload-pack$z/git-receive-pack$Nr}c||_||_tt|_||_||jj |yyr)rCr}dictrr~ fallback_appupdate)rrCr}r~rs r7rzHTTPGitApplication.__init__sB  -. (   MM  * r9c|d}|d}t|||j|j}d}|jj D]1\}}||k7r |j |} | s |j||f}n|0|j |j ||S|jdgS|||j S)N PATH_INFOREQUEST_METHOD)r}r~z#Sorry, that method is not supported) rr}r~serviceskeyssearchrrFrC) rrzrrUmethodrMrsmethodspathr?s r7__call__zHTTPGitApplication.__call__s{#)* ^$))dmm "mm002NGU& ,,t$C--7 3 ?  ,((.AA &KLMMsDLL#..r9)FNN)rrrrrecompilerarrrlrprrrrrrrrr9r7rrfs   8$%}   =)*M   678-   ;<=}   123^  BJJ? @    BJJB C    BJJA B   /013I 0124J'H,+d+/r9rceZdZdZdZdZy) GunzipFilterzkWSGI middleware that unzips gzip-encoded requests before passing on to the underlying application. c||_yrappr applications r7rzGunzipFilter.__init__ r9c~|jdddk(r> |dj|d}tjd|d|d<|d=d |vr|d =|j||S#tttf$rBt j d}tj|d||jdYwxYw) NHTTP_CONTENT_ENCODINGgziprirr)filenamefileobjmodeCONTENT_LENGTH) r{tellAttributeErrorrJNotImplementedErrortempfileSpooledTemporaryFileshutil copyfileobjseekrGzipFiler)rrzr wsgi_inputs r7rzGunzipFilter.__call__s ;;. 3v = # %**,$\2 %)MMz%GL !/07*,-xx00!#G-@A # &::;KL ""7<#8*E" #sA&&AB<;B<Nrrrrrrrr9r7rrs1r9rceZdZdZdZdZy)LimitedInputFilterzgWSGI middleware that limits the input length of a request to that specified in Content-Length. c||_yrrrs r7rzLimitedInputFilter.__init__rr9c|jdd}|rt|dt||d<|j||S)Nrrr)r{rintr)rrzrcontent_lengths r7rzLimitedInputFilter.__call__sI !%5r: $6 %s>':%GL !xx00r9Nrrr9r7rrs 1r9rcDt|i|}tt|}|S)zpFactory function to create an instance of HTTPGitApplication, correctly wrapped with needed middleware. )rrr)argskwargsr wrapped_apps r7make_wsgi_chainrs( d -f -C$\#%67K r9c"eZdZdZdZdZdZy)ServerHandlerLoggerz@ServerHandler that uses dulwich's logger for logging exceptions.c2tjd|yNz/Exception happened during processing of request)exc_infor] exceptionrrs r7 log_exceptionz!ServerHandlerLogger.log_exception =  r9c0tj|g|yrr]r^rformatrs r7 log_messagezServerHandlerLogger.log_message F"T"r9c(tj|yrr]rKrrs r7 log_errorzServerHandlerLogger.log_error dr9N)rrrrrrrrr9r7rrsJ #r9rc(eZdZdZdZdZdZdZy)WSGIRequestHandlerLoggerzEWSGIRequestHandler that uses dulwich's logger for logging exceptions.c2tjd|yrrrs r7rz&WSGIRequestHandlerLogger.log_exceptionr r9c0tj|g|yrr r s r7rz$WSGIRequestHandlerLogger.log_messagerr9c(tj|yrrrs r7rz"WSGIRequestHandlerLogger.log_errorrr9c@|jj|_|jsyt |j|j |j |j}||_|j|jjy)zHandle a single HTTP requestN) rfilereadlineraw_requestline parse_requestrwfile get_stderr get_environrequest_handlerrunserverget_app)rrs r7rzWSGIRequestHandlerLogger.handlesv $zz224!!# % JJ DOO$5t7G7G7I #' DKK'')*r9N)rrrrrrrrrr9r7rrsO # +r9rceZdZdZy)WSGIServerLoggercFtjdt|zy)zHandle an error. z7Exception happened during processing of request from %sN)r]rr|)rrequestclient_addresss r7 handle_errorzWSGIServerLogger.handle_errors  E.! " r9N)rrrr+rr9r7r'r's r9r'c>ddl}|j}|jddddd|jd d d td d |j |\}}t |dkDr|d}nt j}tjtdt|i}t|}t|j|j|t t"}t$j'd|j|j|j)y)z,Entry point for starting an HTTP git server.rNz-lz--listen_addresslisten_address localhostzBinding IP address.)destdefaulthelpz-pz--portporti@zPort to listen on.)r/typer0r1rcr;) handler_class server_classz'Listening for HTTP connections on %s:%d)optparse OptionParser add_optionr parse_argslenrTgetcwdr default_logging_configrrrr r-r2rr'r]r^ serve_forever) argvr6parseroptionsrgitdirrCrr$s r7mainrBs   " " $F    "      ! %%d+MGT 4y1}a $$&3V -.G ' "C   .% F KK1   r9__main__r)CriorrrrrTrsysr,typingrrrwsgiref.simple_serverrrr r urllib.parser dulwichr dulwich.protocolr dulwich.reporrrdulwich.serverrrrr getLoggerrr]rHrrrfloatr|r8r@rDrQrXrarlrprrrrobjectrrrrrrrrrr'r>rBrr9r7rPsm,K  (("    X &    ( ""3"J 6s 64h4 2)U"!H?&&.*E VE P:/:/z161@11* -  +1+: z hh*Z zFr9