#jHdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl mZddlZddlZddlZddlmZddlmZmZmZmZmZmZddlmZmZmZddlm Z m!Z!m"Z"m#Z#dd l$m%Z%exseZ&ejNddd kZ(ejRjUe+ed  d Z,ejRjUed Z-ej\ddZ/ej\dZ0ej\dZ1dZ2dZ3ejRjide5e6e7e8fdZ9dZ:ejRjide e!fdZ;e,dZdZ?Gdd eZ@ej\d!ZAe,e-ejRjd"d#d$ZCejRjUe d% e,e-d&ZDejRjUed' ejRjid(d)d*+ejRjd,d-ZFejRjUe+ed. d/ ejRjide e!fd0ZGej\d1ZHej\d2ZIej\d3ZJejRjid4d5d6eKd7fd5d8eKd7fd9d:d5eKd7eKd7fdd6eKd7fdd8eKd7fd;deKd7eKd7ff d<ZLejRjid=d>d?ZMd@ZNdAZOdBZPy)CzTests for the HTTP server.N) HTTPStatus) DefaultGc)IS_LINUXIS_MACOS IS_WINDOWS SYS_PLATFORMbtonntob)IS_UID_GID_RESOLVABLEGateway HTTPServer)ANY_INTERFACE_IPV4ANY_INTERFACE_IPV6EPHEMERAL_PORTSUCCESSFUL_SUBPROCESS_EXIT) ThreadPool)AF_UNIXz:UNIX domain sockets are only available under UNIX-based OSreasonz9Peercreds lookup does not work under macOS/BSD currently.)abstractfile)paramsctrtjddjdit }|j |S)9Check that bound UNIX socket address is stored in server.z1`requests-unixsocket` is defunct under Python 3.8zunix_{request.param}_sock) PY38_OR_LOWERpytestskipformatlocalsgetfixturevalue)requestnames G/srv/dav/.venv/lib/python3.12/site-packages/cheroot/test/test_server.pyunix_sock_filer(1s?  GH - & - - 9 9D  " "4 ((c tstjtddj dt t tjfjS)z'Return an abstract UNIX socket address.z. does not support an abstract socket namespacer)scheroot-test-socket) rr r!r joinr struuiduuid4decoderr)r'unix_abstract_sockr0=sU  nJ K  88 & TZZ\" #    fh r)c#Ktj\}}|tj|tj|yw)zYield a UNIX file socket.N)tempfilemkstemposcloseunlink) tmp_sock_fhtmp_sock_fnames r'unix_file_sockr9Ls9#+"2"2"4K HH[IInsAA ctttft}|jrJ|j j rJ|j|jsJ|j j sJ|j j D]}|jrJ|j|j j rJ|jrJy)zBCheck that prepare() makes the server ready, and stop() clears it. bind_addrgatewayN) rrrr readyrequests_threadspreparestop) httpserverthrs r'test_prepare_makes_server_readyrEWs%~6J  ""+++ +        ' '' '""++yyyOO""+++ + r)cftttft}|j t j |j}|j|jd|jsJ|j|jd|jrJy)0Check that stop() interrupts running of serve().r;target?N) rrrr rA threadingThreadservestartr+is_aliverB)rC serve_threads r'test_stop_interrupts_serverQns%~6J ##:+;+;.serve_threadsE 4     41v& 233' 4sAAArHrJN)r-r.objectrrrr queueQueuerArKrLrNr+rO interrupt get_nowait)rRrPrCrXrYrZs` @@@@r'test_server_interruptr`s$DJJL#34M"H%~6J {{}H44##<8Lc  "" "#=1Jc$$&& &    $9 99 9r)ctttft}|j d}||j j _tj|j}|j|j|jtrdndz|jrJ|j j rJ|j#y)rGr;ctrU)KeyboardInterrupt)argskwargss r'raise_keyboard_interruptzTtest_serving_is_false_and_stop_returns_after_ctrlc..raise_keyboard_interruptsr)rHrN)rrrr rA _connections _selectorselectrKrLrMrNr+expiration_interval IS_SLOW_ENVrO_servingrB)rCrfrPs r'2test_serving_is_false_and_stop_returns_after_ctrlcrns%~6J  0HJ%%,##:+;+; 11 1r)cH|j|}|j|k(sJy)zBCheck that bound UNIX abstract socket address is stored in server.Nrv)rsr0rCs r'test_bind_addr_unix_abstractrzs*!!"45J   #5 55 5r)z/peer_creds/idsz/peer_creds/textsc$eZdZfdZdZxZS) _TestGatewayc|j}|j}t|j}|tk(rR|j |j |jf}|jdjtt|y|tk(r7|jdj|j|jfytt |GS)N|!)reqconnr uriPEERCRED_IDS_URIpeer_pidpeer_uidpeer_gid send_payloadr+mapr,PEERCRED_TEXTS_URI peer_user peer_groupsuperr|respond)selfrrreq_uri peer_creds __class__s r'rz_TestGateway.respondshhxxsww- & & t}}DJ   chhs3 ';< = ( (   chh'HI J\4022r)c~|j}d|_|j|jt |y)Ns200 OK)rstatusensure_headers_sentwriter )rpayloadrs r'rz_TestGateway.send_payloads.hh  ! $w- r))__name__ __module__ __qualname__rr __classcell__)rs@r'r|r|s  3!r)r|cL|j|}t|_d|_|S)z3Construct a test server with ``peercreds_enabled``.T)rrr|r=peercreds_enabledrws r'peercreds_enabled_serverrs*!!.1J%J#'J r)r)reruns reruns_delayc|}|j}t|tr|j}tj j |d}djdit}tjtjtjf}djtt|}t!j"5t%j&|t(z|}|j+|j,|k(sJt%j&|t.z|}|j0dk(sJ dddy#1swYyxYw) z2Check that ``PEERCRED`` lookup works when enabled.safehttp+unix://{quoted}r~timeoutiNr)r< isinstancebytesr/urllibparsequoter"r#r4getpidgetuidgetgidr+rr,requests_unixsocket monkeypatchr?getrraise_for_statustextr status_code) http_request_timeoutrrCr<quoted unix_base_uriexpected_peercredspeercreds_resppeercreds_text_resps r'test_peercreds_unix_sockr s$ *J$$I)U#$$& \\     3F1*11=FH=Mbiik299;>#c+=">?  ( ( * 6! , ,(  '')""&8888&ll . .( #..#555 6 6 6s A/EEzDModules `grp` and `pwd` are not available under the current platformc|}d|_|j}t|tr|j }t j j|d}djd it}ddl }ddl }|jtjj|j!tj"j$f}dj't)t*|}t-j.5t1j2|t4z|} | j7| j8|k(sJ dddy#1swYyxYw) z6Check that ``PEERCRED`` resolution works when enabled.TrrrrNrrr)peercreds_resolve_enabledr<rrr/rrrr"r#grppwdgetpwuidr4rpw_namegetgrgidrgr_namer+rr,rrr?rrrr) rrrCr<rrrrexpected_textcredsrs r'$test_peercreds_unix_sock_with_lookupr-s*J+/J($$I)U#$$& \\     3F1*11=FH=M RYY[!)) RYY[!))#c+=">?  ( ( *>&ll . .(  ,,."''+==== >>>s AEEz\This regression test is for a Linux bug, and the resource module is not available on Windowsresource_limit)ii)r)indirectmany_open_socketsc|jjfdt_|j_|j dt jdkDsJt fdjDsJy)a>Test the server does not crash with a high file-descriptor value. This test shouldn't cause a server crash when trying to access file-descriptor higher than 1024. The earlier implementation used to rely on ``select()`` syscall that doesn't support file descriptors with numbers higher than 1024. czjj|jj|SrU)filenosaddsocketfileno)r_old_process_connnative_process_conns r'rzAtest_high_number_of_file_descriptors..native_process_connss/##'' (:(:(<= &&r)/rc3(K|] }|k\ ywrUr).0fnrs r' z7test_high_number_of_file_descriptors..sJr^#JsN)server_instance process_connsetrconnectlenany)native_server_clientrrrs `@@r'$test_high_number_of_file_descriptorsrVs6-<<II'#&%8K((5  % "** +a // / J.A.I.IJ JJ Jr) SO_REUSEPORTz5socket.SO_REUSEPORT is not supported on this platformctj|tdd}tj|}|jtjtj d|j |tft|jddtd}|j|d}|j|j|j|jJy)z5Check that port initialized externally can be reused.rrqNrT)r<r= reuse_portrA)r getaddrinfor setsockopt SOL_SOCKETrbindr getsocknamer spyrArBr5 spy_exception)rsromockerfamilysserverrs r'test_reuse_portrs   8 ;A >F fALL""F$7$7;FFG^ $% --/"1%F **VY 'C NN KKMGGI    $$ $r)c#nKtj5ddddy#1swYyxYww)z7Disable garbage collection when this fixture is in use.N)rnogcrr)r' _garbage_binrs-     s5) 525c#lKtjdd}|j|j\}}|j |j|j dz|f |j |j |j||fy#|j |j||fwxYww)z7Set the resource limit two times bigger then requested.resourcez&The "resource" module is Unix-specificrrN)r importorskip getrlimit RLIMIT_NOFILE setrlimitparam)r%r soft_limit hard_limits r'rrs""7H &//0F0FGJ    J' Mmm 811J 3KL811J 3KLsA!B4$B2B4 B11B4c#jK|jdg} t|D]<}tj}|j||j |k\s<n|dj }||k\sJ||D]}|j y#|D]}|j wxYww)z4    B 66Z   66& && &r)rrerror) )rrmin=-1 must be > 0)rrr)rrqr)rrr)rrmin=0 must be > 0)rrr)rrqr)rrr)rqrKExpected an integer or the infinity value for the `max` argument but got 0.)rqrJzMExpected an integer or the infinity value for the `max` argument but got 0.5.)rrr)r1zMExpected an integer or the infinity value for the `max` argument but got '1'.)rrqzmax=1 must be > min=2ctjttf|5t d||dddy#1swYyxYw)zTest that a ThreadPool rejects invalid min/max values. The ThreadPool should raise an error with the proper message when initialized with an invalid min+max number of threads. matchNr)r raises ValueError TypeErrorrr s r'#test_threadpool_invalid_threadrangers>P  I.e <     s :Ac|jtddtd}|jtjt d5|jdddy#1swYyxYw)zTest for ThreadPool multi-start behavior. Tests that when calling start() on a ThreadPool multiple times raises a :exc:`RuntimeError` _spawn_workerc.tjdS)NT)r>)typesSimpleNamespace)rs r'z7test_threadpool_multistart_validation..>s%''d3r)N)rz%Threadpools can only be started once.r)setattrrrNr r RuntimeError)rr s r'%test_threadpool_multistart_validationr"3sb3 4 BHHJ 5   s A**A3c*d}t|tft}tdddd||_|j t j|j}|j|j|j|jj|j\}}|jjjdt jd|d |d }|j t"j$k(sJy) zCA server that can't keep up with requests returns a 503 HTTP error.z 127.0.0.1r;rqr)r r accepted_queue_sizeaccepted_queue_timeoutrrHNzhttp://:r)rrr rr?rArKrLrMrN addfinalizerrBr<_queuerVrrrSERVICE_UNAVAILABLE)r% localhostrCrP_hostportresponses r',test_overload_results_in_suitable_http_errorr/JsIn-J %   J##:+;+;rRs\      )NN?? ,$*   !$. kk(( "" G) kk(( F) +,)-)   .'(  !:!:H8 552266%(!7!*!!,6-6< ! > >D : ! ,-K. KB'' B % %$ MM2  85   %L   Be  E%L%,' Be  Be  E%L%,'( ')( ')!D  E!D  .BB